diff options
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp')
| -rw-r--r-- | src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp index b887721648..e11f293b12 100644 --- a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp +++ b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp @@ -19,9 +19,10 @@ DType ArmNNToDType(const DataType& type) switch (type) { case DataType::Float16: - case DataType::Float32: case DataType::BFloat16: - return DType_FLOAT; + return DType_FP16; + case DataType::Float32: + return DType_FP32; case DataType::QAsymmU8: return DType_UINT8; case DataType::QSymmS8: |