aboutsummaryrefslogtreecommitdiff
path: root/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2023-09-29 17:00:38 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2023-10-03 09:40:37 +0000
commitad323af0e9b47e53d366b85cdf74927f88748d40 (patch)
treec4251570a54ad9c123e4946bc1c69cbdbbf08253 /src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
parentedefbc282556021b7ea4db5d88d4cce2f9782b15 (diff)
downloadarmnn-ad323af0e9b47e53d366b85cdf74927f88748d40.tar.gz
IVGCVSW-8033 Update Arm NN to use TOSA v0.60.0
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Change-Id: If1504534fbd7d6b317fdb93083dcdbd8b827764b
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp')
-rw-r--r--src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp11
1 files changed, 10 insertions, 1 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp b/src/backends/tosaCommon/operatorMappings/TosaOperatorUtils.hpp
index dccb29892c..3e106e1fd5 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 @@ inline std::string TosaDTypeToString(DType tosaDType)
return "DType_UINT16";
case DType_FP16:
return "DType_FP16";
+ case DType_BF16:
+ return "DType_BF16";
}
return "";
}
@@ -273,6 +278,10 @@ inline std::string TosaOpToString(Op tosaOp)
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 "";
}