diff options
Diffstat (limited to 'reference_model/include/model_runner.h')
-rw-r--r-- | reference_model/include/model_runner.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/reference_model/include/model_runner.h b/reference_model/include/model_runner.h index 4629467..4335794 100644 --- a/reference_model/include/model_runner.h +++ b/reference_model/include/model_runner.h @@ -68,7 +68,7 @@ public: * NOTE: setInput() must be called for each input tensor before run() is called. */ template <typename T> - int setInput(std::string input_name, std::vector<T> vals); + int setInput(std::string input_name, std::vector<T>& vals); /* * Retrieve the output tensors from the graph after running. |