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/ops/image.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/ops/image.cc')
-rw-r--r-- | reference_model/src/ops/image.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/reference_model/src/ops/image.cc b/reference_model/src/ops/image.cc index 891261b..cf1d9f7 100644 --- a/reference_model/src/ops/image.cc +++ b/reference_model/src/ops/image.cc @@ -63,7 +63,7 @@ int OpResize<InDtype, OutDtype, resize_t>::checkTensorAttributes() if (this->mode == ResizeMode_BILINEAR) { - if (OutDtype != DType_INT32 && OutDtype != DType_INT48 && OutDtype != DType_FLOAT && OutDtype != DType_FP16) + if (OutDtype != DType_INT32 && OutDtype != DType_INT48 && OutDtype != DType_FP32 && OutDtype != DType_FP16) { printNodeValidationError("OpResize: invalid data type for BILINEAR"); return 1; @@ -71,7 +71,7 @@ int OpResize<InDtype, OutDtype, resize_t>::checkTensorAttributes() } else { - if (OutDtype != DType_INT8 && OutDtype != DType_INT16 && OutDtype != DType_FLOAT && OutDtype != DType_FP16) + if (OutDtype != DType_INT8 && OutDtype != DType_INT16 && OutDtype != DType_FP32 && OutDtype != DType_FP16) { printNodeValidationError("OpResize: invalid data type for NEAREST"); return 1; @@ -225,4 +225,4 @@ DEF_INSTANTIATE_THREE_TYPE(OpResize, INT8, INT8, int16_t); DEF_INSTANTIATE_THREE_TYPE(OpResize, INT16, INT48, int16_t); DEF_INSTANTIATE_THREE_TYPE(OpResize, INT16, INT16, int16_t); DEF_INSTANTIATE_THREE_TYPE(OpResize, FP16, FP16, float); -DEF_INSTANTIATE_THREE_TYPE(OpResize, FLOAT, FLOAT, float); +DEF_INSTANTIATE_THREE_TYPE(OpResize, FP32, FP32, float); |