From 0718ee9f26219da5a1362c1060e7db3feae8b3ce Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Fri, 13 Sep 2019 16:53:38 +0100
Subject: IVGCVSW-3868 Add support of Int32 to Reshape layer

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I1bf116c462fe3428be31b6dbcd09b25c3a90727f
---
 src/backends/backendsCommon/WorkloadData.cpp | 1 +
 src/backends/reference/RefLayerSupport.cpp   | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 8361812697..f290cbd9cf 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -1307,6 +1307,7 @@ void ReshapeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
     {
         DataType::Float32,
         DataType::Float16,
+        DataType::Signed32,
         DataType::QuantisedAsymm8,
         DataType::QuantisedSymm16
     };
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 465d45cbae..228f8a8809 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -1283,10 +1283,11 @@ bool RefLayerSupport::IsReshapeSupported(const TensorInfo& input,
 {
     ignore_unused(descriptor);
     // Define supported output types.
-    std::array<DataType,4> supportedOutputTypes =
+    std::array<DataType,5> supportedOutputTypes =
     {
         DataType::Float32,
         DataType::Float16,
+        DataType::Signed32,
         DataType::QuantisedAsymm8,
         DataType::QuantisedSymm16
     };
-- 
cgit v1.2.1