IVGCVSW-8033 Update Arm NN to use TOSA v0.60.0
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: If1504534fbd7d6b317fdb93083dcdbd8b827764b
diff --git a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
index dccb298..3e106e1 100644
--- a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
+++ b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
@@ -16,14 +16,17 @@
using namespace armnn;
using namespace tosa;
+const std::string mainName = "main";
+
// Function to return Tosa datatype from input ArmNN datatype.
inline DType ArmNNToDType(const DataType& type)
{
switch (type)
{
case DataType::Float16:
- case DataType::BFloat16:
return DType_FP16;
+ case DataType::BFloat16:
+ return DType_BF16;
case DataType::Float32:
return DType_FP32;
case DataType::QAsymmU8:
@@ -126,6 +129,8 @@
return "DType_UINT16";
case DType_FP16:
return "DType_FP16";
+ case DType_BF16:
+ return "DType_BF16";
}
return "";
}
@@ -273,6 +278,10 @@
return "Op_COND_IF";
case Op_WHILE_LOOP:
return "Op_WHILE_LOOP";
+ case Op_FFT2D:
+ return "Op_FFT2D";
+ case Op_RFFT2D:
+ return "Op_RFFT2D";
}
return "";
}