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
diff --git a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
index b887721..e11f293 100644
--- a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
+++ b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
@@ -19,9 +19,10 @@
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: