diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-10-18 18:02:48 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-10-19 17:41:40 +0100 |
commit | 2e4d889fb036d1c0a34503400a3f45cfc6f9f3e1 (patch) | |
tree | 083a125f75dc4e338bf9382da538742b4cd5bfa7 /reference_model/include | |
parent | 4196491afc23d375b5476b05be16defeed4eadad (diff) | |
download | reference_model-2e4d889fb036d1c0a34503400a3f45cfc6f9f3e1.tar.gz |
Add FP16 support to IModelRunner
* Added specific FP16 readfromVector and writeToVector methods.
* Added FP16 support to float readfromVector and writeToVector methods.
* Added missing reference to IModelRunner::setInput.
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I6b66468737e672afc925ccad4fb710fbb9427c14
Diffstat (limited to 'reference_model/include')
-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. |