diff options
author | Won Jeon <won.jeon@arm.com> | 2023-08-10 10:33:01 +0000 |
---|---|---|
committer | Won Jeon <won.jeon@arm.com> | 2023-08-18 15:21:15 -0700 |
commit | a21b2e88d19d8cb11a9120d40bacbb594d600565 (patch) | |
tree | 3bc8a40db72a31c1e552a3bd6339627a1175686e /reference_model/src/tensor.cc | |
parent | e0247481eb1f83f6eb7161d3f7ac2690b180952a (diff) | |
download | reference_model-a21b2e88d19d8cb11a9120d40bacbb594d600565.tar.gz |
Add DIM operator to reference model
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iea11ee5d3d98773e9c5e9b827593c05afb41ce3b
Diffstat (limited to 'reference_model/src/tensor.cc')
-rw-r--r-- | reference_model/src/tensor.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/reference_model/src/tensor.cc b/reference_model/src/tensor.cc index 4982c99..1aabe5b 100644 --- a/reference_model/src/tensor.cc +++ b/reference_model/src/tensor.cc @@ -137,6 +137,7 @@ int TosaReference::Tensor::readFromNpyFile(const char* filename) nperror = NumpyUtilities::readFromNpyFile(filename, elements, i32databuf); break; case DType_INT48: + case DType_SHAPE: i64databuf = (int64_t*)calloc(sizeof(int64_t), elements); ASSERT_MEM(i64databuf); @@ -220,6 +221,7 @@ int TosaReference::Tensor::readFromNpyFile(const char* filename) } break; case TOSA_REF_TYPE_INT48: + case TOSA_REF_TYPE_SHAPE: if (setTensorValueInt64(elements, i64databuf)) { free(i64databuf); @@ -379,6 +381,7 @@ int TosaReference::Tensor::writeToNpyFile(const char* filename) const free(i32databuf); break; case TOSA_REF_TYPE_INT48: + case TOSA_REF_TYPE_SHAPE: i64databuf = (int64_t*)calloc(sizeof(int64_t), elements); ASSERT_MEM(i64databuf); @@ -672,6 +675,7 @@ int TosaReference::Tensor::readfromVector(const ArrayProxy<int64_t> vals) switch (getDtype()) { case TOSA_REF_TYPE_INT48: + case TOSA_REF_TYPE_SHAPE: if (vals.size() != elements) { WARNING("The input size (%ld) doesn't match the number of elements (%d) assigned to the tensor.", @@ -847,6 +851,7 @@ int TosaReference::Tensor::writeToVector(ArrayProxy<int64_t> vals) switch (getDtype()) { case TOSA_REF_TYPE_INT48: + case TOSA_REF_TYPE_SHAPE: if (vals.size() != elements) { WARNING("The output size (%ld) doesn't match the number of elements (%d) assigned to the tensor.", |