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/src/tensor.h | |
| 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/src/tensor.h')
| -rw-r--r-- | reference_model/src/tensor.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h index efd7e62..4efbf84 100644 --- a/reference_model/src/tensor.h +++ b/reference_model/src/tensor.h @@ -229,11 +229,13 @@ public: virtual int copyValueFrom(Tensor* tensor) = 0; virtual int readfromVector(const std::vector<float>& vals); + virtual int readfromVector(const std::vector<half_float::half>& vals); virtual int readfromVector(const std::vector<int32_t>& vals); virtual int readfromVector(const std::vector<int64_t>& vals); virtual int readfromVector(const std::vector<unsigned char>& vals); virtual int writeToVector(std::vector<float>& vals); + virtual int writeToVector(std::vector<half_float::half>& vals); virtual int writeToVector(std::vector<int32_t>& vals); virtual int writeToVector(std::vector<int64_t>& vals); virtual int writeToVector(std::vector<unsigned char>& vals); |