commit 529195f3ae752192d5f205345235bf537f472cbc
author Sadik Armagan <sadik.armagan@arm.com> Fri Jan 14 12:56:35 2022 +0000
committer TeresaARM <teresa.charlinreyes@arm.com> Tue Jan 18 15:15:58 2022 +0000
tree 2ee9dba45e0c181138725607133c6f4bf1a6828a
parent 611c7fb97412230d5cefee047081455fb60db06c

MLCE-736 'Adding support for TfLite Models'

* Added constant input supports for Pack/Stack, Concatenation operators
* Added Int32 support to Pack/Stack operator on CpuRef
* Removed unsupported operator from TfLite Delegate

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I64203e174300d23eedeb22bddefe07e931c4eff3

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 0ac2ddc..4d4f014 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -2087,14 +2087,15 @@
     IgnoreUnused(descriptor);
 
     bool supported = true;
-    std::array<DataType,6> supportedTypes =
+    std::array<DataType,7> supportedTypes =
     {
         DataType::BFloat16,
         DataType::Float32,
         DataType::Float16,
         DataType::QAsymmS8,
         DataType::QAsymmU8,
-        DataType::QSymmS16
+        DataType::QSymmS16,
+        DataType::Signed32
     };
 
     supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported,