diff options
| author | Jerry Ge <jerry.ge@arm.com> | 2024-01-02 22:29:08 +0000 |
|---|---|---|
| committer | Jerry Ge <jerry.ge@arm.com> | 2024-01-25 17:42:12 +0000 |
| commit | c5291695f04901e8abbc26dad6cba10e2c7685f8 (patch) | |
| tree | 6ba3ea7399e9e5ee679ca2ef4bf9770c8e5046dc /reference_model/src/model_runner.cc | |
| parent | 74342e522ec61e85fde64fe801da9e750b3e2d86 (diff) | |
| download | reference_model-c5291695f04901e8abbc26dad6cba10e2c7685f8.tar.gz | |
Save Int8/UInt8 reference outputs to native dtypes
* Int8/UInt8 reference outputs were previously saved to INT32
* Save those in their native dtypes and updated other affected code
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Id65fc8773150d3c56bc2c72789a6a0d3c78cd363
Diffstat (limited to 'reference_model/src/model_runner.cc')
| -rw-r--r-- | reference_model/src/model_runner.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/reference_model/src/model_runner.cc b/reference_model/src/model_runner.cc index 6f65202..28ad72c 100644 --- a/reference_model/src/model_runner.cc +++ b/reference_model/src/model_runner.cc @@ -82,12 +82,16 @@ int IModelRunner::getOutput(std::string output_name, uint8_t* raw_ptr, size_t si // Template explicit specialization 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<int8_t>(std::string input_name, std::vector<int8_t>& vals); +template int IModelRunner::setInput<int16_t>(std::string input_name, std::vector<int16_t>& 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<int8_t> IModelRunner::getOutput<int8_t>(std::string output_name); +template std::vector<int16_t> IModelRunner::getOutput<int16_t>(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); |