From 529195f3ae752192d5f205345235bf537f472cbc Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Fri, 14 Jan 2022 12:56:35 +0000 Subject: 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 Change-Id: I64203e174300d23eedeb22bddefe07e931c4eff3 --- src/backends/reference/RefLayerSupport.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/backends/reference/RefLayerSupport.cpp') diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 0ac2ddc2a9..4d4f014671 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -2087,14 +2087,15 @@ bool RefLayerSupport::IsStackSupported(const std::vector& inp IgnoreUnused(descriptor); bool supported = true; - std::array supportedTypes = + std::array supportedTypes = { DataType::BFloat16, DataType::Float32, DataType::Float16, DataType::QAsymmS8, DataType::QAsymmU8, - DataType::QSymmS16 + DataType::QSymmS16, + DataType::Signed32 }; supported &= CheckSupportRule(TypeAnyOf(output, supportedTypes), reasonIfUnsupported, -- cgit v1.2.1