Fix input setting and invocation. (#5888)

Fix input setting and invocation. (#5752)

Summary:
Pull Request resolved: #5752

.

Reviewed By: kirklandsign

Differential Revision: D63640035

fbshipit-source-id: 47b5952d3e77cbe65751c81af55960001a9552fd
(cherry picked from commit 68548e5)

Co-authored-by: Anthony Shoumikhin <[email protected]>

Author: pytorchbot

extension/apple/Benchmark/Tests/Tests.mm

@@ -98,7 +98,8 @@ + (void)initialize {
               const auto sizes = tensor_meta->sizes();
               tensors.emplace_back(ones({sizes.begin(), sizes.end()},
                                         tensor_meta->scalar_type()));
-              inputs.emplace_back(tensors.back());
+              XCTAssertEqual(module->set_input(tensors.back(), index),
+                             Error::Ok);
             } break;
             default:
               XCTFail("Unsupported tag %i at input %d", *input_tag, index);
@@ -110,7 +111,7 @@ + (void)initialize {
           ]
                             options:XCTMeasureOptions.defaultOptions
                               block:^{
-                                XCTAssertEqual(module->forward(inputs).error(),
+                                XCTAssertEqual(module->forward().error(),
                                                Error::Ok);
                               }];
         });