IVGCVSW-3868 Add support of Int32 to Reshape layer
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I1bf116c462fe3428be31b6dbcd09b25c3a90727f
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index 8361812..f290cbd 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -1307,6 +1307,7 @@
{
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 465d45c..228f8a8 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -1283,10 +1283,11 @@
{
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
};