diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-09-27 13:50:00 +0100 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-10-13 18:21:15 +0100 |
commit | bc2a3db54ecee48fe2236f7fc03da8fd07d81ca0 (patch) | |
tree | c3908f23c369fd3226e840f81c3ba4b49cc409a0 /reference_model/src/tensor.cc | |
parent | 93d4390f9aa5c4369f889e1cd336aa4e809ff6a7 (diff) | |
download | reference_model-bc2a3db54ecee48fe2236f7fc03da8fd07d81ca0.tar.gz |
Rename FLOAT type to FP32
Update tensor operations naming to state input type as TxT in
all cases. Effects CONV2D, CONV3D, DEPTHWISE_CONV2D,
FULLY_CONNECTED, TRANSPOSE_CONV2D.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ic959acfcb3aa0a910b33b774a5a85fac08219205
Diffstat (limited to 'reference_model/src/tensor.cc')
-rw-r--r-- | reference_model/src/tensor.cc | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/reference_model/src/tensor.cc b/reference_model/src/tensor.cc index cbe12a9..8e65a27 100644 --- a/reference_model/src/tensor.cc +++ b/reference_model/src/tensor.cc @@ -93,7 +93,7 @@ int TosaReference::Tensor::readFromNpyFile(const char* filename) switch (getDtype()) { - case DType_FLOAT: + case DType_FP32: fdatabuf = (float*)calloc(sizeof(float), elements); ASSERT_MEM(fdatabuf); @@ -161,8 +161,8 @@ int TosaReference::Tensor::readFromNpyFile(const char* filename) for (uint32_t i=0; i < elements; i++) { fdatabuf[i] = half_float::half_cast<float, half_float::half>(f16databuf[i]); } - // Fall through to DType_FLOAT case - case DType_FLOAT: + // Fall through to DType_FP32 case + case DType_FP32: if (setTensorValueFloat(elements, fdatabuf)) { if (f16databuf) @@ -229,7 +229,7 @@ int TosaReference::Tensor::writeToNpyFile(const char* filename) const switch (getDtype()) { - case DType_FLOAT: + case DType_FP32: fdatabuf = (float*)calloc(sizeof(float), elements); ASSERT_MEM(fdatabuf); @@ -429,7 +429,7 @@ int TosaReference::Tensor::readfromVector(const std::vector<float>& vals) uint32_t elements = getElementCount(); switch (getDtype()) { - case DType_FLOAT: + case DType_FP32: if (vals.size() != elements) { WARNING("The input size (%ld) doesn't match the number of elements (%d) assigned to the tensor.", @@ -532,7 +532,7 @@ int TosaReference::Tensor::writeToVector(std::vector<float>& vals) switch (getDtype()) { - case DType_FLOAT: + case DType_FP32: if (vals.size() != elements) { WARNING("The output size (%ld) doesn't match the number of elements (%d) assigned to the tensor.", |