diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-05-25 15:26:38 +0100 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-05-26 09:53:44 +0100 |
commit | f7f78ae236e623a57919f9450e8b2043e681ddb3 (patch) | |
tree | 0456c0006fbce5efbebe93818398b3a0cd7cd76c /reference_model/src/tensor.cc | |
parent | 0e6218e22f25901aa208fbec44c9b14e14a68ba7 (diff) | |
download | reference_model-f7f78ae236e623a57919f9450e8b2043e681ddb3.tar.gz |
Add support for uint16_t to RESCALE
Update ref-model RESCALE op to support UINT16 conversions
Add testing for RESCALE UINT16 and ERROR_IFs
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ic6e6e53de1f0b054bedb9e6ba3856e7475498aba
Diffstat (limited to 'reference_model/src/tensor.cc')
-rw-r--r-- | reference_model/src/tensor.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/reference_model/src/tensor.cc b/reference_model/src/tensor.cc index f2a3a98..36ace48 100644 --- a/reference_model/src/tensor.cc +++ b/reference_model/src/tensor.cc @@ -102,6 +102,7 @@ int TosaReference::Tensor::readFromNpyFile(const char* filename) case DType_INT4: case DType_INT8: case DType_INT16: + case DType_UINT16: i32databuf = (int32_t*)calloc(sizeof(int32_t), elements); ASSERT_MEM(i32databuf); @@ -157,6 +158,7 @@ int TosaReference::Tensor::readFromNpyFile(const char* filename) case DType_INT4: case DType_INT8: case DType_INT16: + case DType_UINT16: if (setTensorValueInt32(elements, i32databuf)) { free(i32databuf); @@ -225,6 +227,7 @@ int TosaReference::Tensor::writeToNpyFile(const char* filename) const case DType_INT4: case DType_INT8: case DType_INT16: + case DType_UINT16: i32databuf = (int32_t*)calloc(sizeof(int32_t), elements); ASSERT_MEM(i32databuf); |