commit fd5bec4cc0e6ef192a580621f225c971d605c261
author Cathal Corbett <cathal.corbett@arm.com> Thu Mar 03 15:13:23 2022 +0000
committer Cathal Corbett <cathal.corbett@arm.com> Thu Mar 03 15:13:32 2022 +0000
tree 1547c8a37c24ea99587d604335254629754b1ab0
parent 624b5a3b8e3978e9b3887918194d485b14674cea

Revert "Revert "IVGCVSW-6268 Add support of Unidirectional Sequence Lstm fp32/fp16 to Neon""

This reverts commit f87b90e4dbb906436cf205a2a19e199bfe9224ed.

Reason for revert: 22.02 release.

Change-Id: I1ca5a79a8957908f655a6c4e79eefa24c5aec645

src/backends/aclCommon/ArmComputeTensorUtils.cpp

diff --git a/src/backends/aclCommon/ArmComputeTensorUtils.cpp b/src/backends/aclCommon/ArmComputeTensorUtils.cpp
index 9ed7b7b..2dc6d2a 100644
--- a/src/backends/aclCommon/ArmComputeTensorUtils.cpp
+++ b/src/backends/aclCommon/ArmComputeTensorUtils.cpp
@@ -45,6 +45,38 @@
     }
 }
 
+armnn::DataType GetArmNNDataType(arm_compute::DataType dataType)
+{
+    switch(dataType)
+    {
+        case arm_compute::DataType::BFLOAT16:
+            return armnn::DataType::BFloat16;
+        case arm_compute::DataType::U8:
+            return armnn::DataType::Boolean;
+        case arm_compute::DataType::F16:
+            return armnn::DataType::Float16;
+        case arm_compute::DataType::F32:
+            return armnn::DataType::Float32;
+        case arm_compute::DataType::QASYMM8_SIGNED:
+            return armnn::DataType::QAsymmS8;
+        case arm_compute::DataType::QASYMM8:
+            return armnn::DataType::QAsymmU8;
+        case arm_compute::DataType::QSYMM16:
+            return armnn::DataType::QSymmS16;
+        case arm_compute::DataType::S64:
+            return armnn::DataType::Signed64;
+        case arm_compute::DataType::QSYMM8_PER_CHANNEL:
+            return armnn::DataType::QSymmS8;
+        case arm_compute::DataType::QSYMM8:
+            return armnn::DataType::QSymmS8;
+        case arm_compute::DataType::S32:
+            return armnn::DataType::Signed32;
+        default:
+            ARMNN_ASSERT_MSG(false, "Unknown data type");
+            return armnn::DataType::Float32;
+    }
+}
+
 arm_compute::Coordinates BuildArmComputeReductionCoordinates(size_t inputDimensions,
                                                              unsigned int originalInputRank,
                                                              const std::vector<unsigned int>& armnnAxes)