MLCE-278 issue with signed-int8 quantized model
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I144ebfca524f4cdee9cc82eef3995c6b32bfc40b
diff --git a/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp b/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp
index 79744ec..d969d5b 100644
--- a/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp
+++ b/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp
@@ -71,9 +71,8 @@
case armnn::DataType::Float32:
return weightsType;
case armnn::DataType::QAsymmS8:
- return armnn::DataType::Signed32;
case armnn::DataType::QAsymmU8:
- return armnn::DataType::Signed32;
+ case armnn::DataType::QSymmS8:
case armnn::DataType::QSymmS16:
return armnn::DataType::Signed32;
default:
diff --git a/src/backends/backendsCommon/LayerSupportRules.hpp b/src/backends/backendsCommon/LayerSupportRules.hpp
index ddecc82..e7aef0a 100644
--- a/src/backends/backendsCommon/LayerSupportRules.hpp
+++ b/src/backends/backendsCommon/LayerSupportRules.hpp
@@ -23,11 +23,10 @@
case armnn::DataType::Float16:
case armnn::DataType::Float32:
return weightsType;
- case armnn::DataType::QAsymmU8:
- return armnn::DataType::Signed32;
- case armnn::DataType::QSymmS16:
- return armnn::DataType::Signed32;
case armnn::DataType::QAsymmS8:
+ case armnn::DataType::QAsymmU8:
+ case armnn::DataType::QSymmS8:
+ case armnn::DataType::QSymmS16:
return armnn::DataType::Signed32;
default:
ARMNN_ASSERT_MSG(false, "GetBiasTypeFromWeightsType(): Unsupported data type.");
diff --git a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
index 5832391..3173561 100644
--- a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
+++ b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
@@ -100,9 +100,8 @@
case armnn::DataType::Float32:
return weightsType;
case armnn::DataType::QAsymmS8:
- return armnn::DataType::Signed32;
case armnn::DataType::QAsymmU8:
- return armnn::DataType::Signed32;
+ case armnn::DataType::QSymmS8:
case armnn::DataType::QSymmS16:
return armnn::DataType::Signed32;
default: