IVGCVSW-8300 Fix for CTS Float16 tests

* Fix for Neon IsLayerSupported to properly check for multiple Quantization Scales

Signed-off-by: Orlaith Monahan <orlaith.monahan@arm.com>
Change-Id: I9f4558cbd62ce1657adb5025ac16c2b5d69d12b1

1 files changed, 16 insertions, 5 deletions

diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp
index 0298c7c552..b6db52342e 100644
--- a/src/backends/neon/NeonLayerSupport.cpp
+++ b/src/backends/neon/NeonLayerSupport.cpp
@@ -102,11 +102,22 @@ const TensorInfo OverrideDataType(const TensorInfo& info, Optional<DataType> typ
     {
         return info;
     }
-    return TensorInfo(info.GetShape(),
-                      type.value(),
-                      info.GetQuantizationScale(),
-                      info.GetQuantizationOffset(),
-                      info.IsConstant());
+    if (info.HasMultipleQuantizationScales())
+    {
+        return TensorInfo(info.GetShape(),
+                          type.value(),
+                          info.GetQuantizationScales(),
+                          info.GetQuantizationDim().value(),
+                          info.IsConstant());
+    }
+    else
+    {
+        return TensorInfo(info.GetShape(),
+                          type.value(),
+                          info.GetQuantizationScale(),
+                          info.GetQuantizationOffset(),
+                          info.IsConstant());
+    }
 }
 
 template< typename ... Args>