commit afa4e3a489df01d7f3b4d0c3f0a89a5251976bd5
author Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Tue Apr 02 11:41:45 2019 +0100
committer Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Tue Apr 02 11:48:03 2019 +0100
tree 710bd1abc201ac06a28b4e8c9385215a94be3440
parent a0beb3b8aeb8bbea906609e0f50a250c33cde10b

IVGCVSW-2875 Enable QuantisedSymm16 support for ref Dequantize workload

Change-Id: I4b1c1f963214433ad67d5a311fbea5f21177f6a2
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>

src/backends/reference/RefLayerSupport.cpp

diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 4d164d5..d2cf6f9 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -392,10 +392,27 @@
                                             const TensorInfo& output,
                                             Optional<std::string&> reasonIfUnsupported) const
 {
-    return IsSupportedForDataTypeRef(reasonIfUnsupported,
-                                     input.GetDataType(),
-                                     &FalseFunc<>,
-                                     &TrueFunc<>);
+   bool supported = true;
+
+    std::array<DataType,2> supportedInputTypes = {
+        DataType::QuantisedAsymm8,
+        DataType::QuantisedSymm16
+    };
+
+    supported &= CheckSupportRule(TypeAnyOf(input, supportedInputTypes), reasonIfUnsupported,
+                                  "Reference dequantize: input type not supported.");
+
+    std::array<DataType,2> supportedOutputTypes = {
+        DataType::Float32,
+    };
+
+    supported &= CheckSupportRule(TypeAnyOf(output, supportedOutputTypes), reasonIfUnsupported,
+                                  "Reference dequantize: output type not supported.");
+
+    supported &= CheckSupportRule(ShapesAreSameTotalSize(input, output), reasonIfUnsupported,
+                                  "Reference dequantize: input and output shapes have different num total elements.");
+
+    return supported;
 }
 
 bool RefLayerSupport::IsDetectionPostProcessSupported(const armnn::TensorInfo& input0,