diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-10-10 12:43:20 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2022-10-27 12:03:26 +0100 |
commit | da824cc0211beb69b6e94a8b1e5c76e8c4eda3a1 (patch) | |
tree | e13f9077fa9f4f8267555b2666d010d219fb6f3c /src/backends/tosaCommon/operatorMappings | |
parent | d646b5504351ab3c3ab8152553465c8e36947e5b (diff) | |
download | armnn-da824cc0211beb69b6e94a8b1e5c76e8c4eda3a1.tar.gz |
IVGCVSW-7273 Fix TOSA Serialization Library integration
* half.hpp is now required as an include for numpy_utils.h
* DType_FLOAT was changed to DType_FP32.
* DType_FP16 support has been added to ArmNNToDType utility function.
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: Ib25be9a5abfbd9db2a90b3e42d592259a6df0e01
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings')
-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: |