diff options
Diffstat (limited to 'reference_model/src/model_runner.cc')
-rw-r--r-- | reference_model/src/model_runner.cc | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/reference_model/src/model_runner.cc b/reference_model/src/model_runner.cc index 2395a85..5c086e6 100644 --- a/reference_model/src/model_runner.cc +++ b/reference_model/src/model_runner.cc @@ -53,7 +53,7 @@ GraphStatus IModelRunner::run() } template <typename T> -int IModelRunner::setInput(std::string input_name, std::vector<T> vals) +int IModelRunner::setInput(std::string input_name, std::vector<T>& vals) { return model_runner_impl->setInput<T>(input_name, vals); } @@ -65,12 +65,14 @@ std::vector<T> IModelRunner::getOutput(std::string output_name) } // Template explicit specialization -template int IModelRunner::setInput<float>(std::string input_name, std::vector<float> vals); -template int IModelRunner::setInput<int32_t>(std::string input_name, std::vector<int32_t> vals); -template int IModelRunner::setInput<int64_t>(std::string input_name, std::vector<int64_t> vals); -template int IModelRunner::setInput<unsigned char>(std::string input_name, std::vector<unsigned char> vals); +template int IModelRunner::setInput<float>(std::string input_name, std::vector<float>& vals); +template int IModelRunner::setInput<half_float::half>(std::string input_name, std::vector<half_float::half>& vals); +template int IModelRunner::setInput<int32_t>(std::string input_name, std::vector<int32_t>& vals); +template int IModelRunner::setInput<int64_t>(std::string input_name, std::vector<int64_t>& vals); +template int IModelRunner::setInput<unsigned char>(std::string input_name, std::vector<unsigned char>& vals); template std::vector<float> IModelRunner::getOutput<float>(std::string output_name); +template std::vector<half_float::half> IModelRunner::getOutput<half_float::half>(std::string output_name); template std::vector<int32_t> IModelRunner::getOutput<int32_t>(std::string output_name); template std::vector<int64_t> IModelRunner::getOutput<int64_t>(std::string output_name); template std::vector<unsigned char> IModelRunner::getOutput<unsigned char>(std::string output_name);
\ No newline at end of file |