Fix input setting and invocation. (#5752)

Summary:
Pull Request resolved: #5752

.

Reviewed By: kirklandsign

Differential Revision: D63640035

fbshipit-source-id: 47b5952d3e77cbe65751c81af55960001a9552fd

Author: shoumikhin

extension/apple/Benchmark/Tests/Tests.mm

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