From 9d07e00917d276c4cd4a3b5c3a08df3de9d82dbc Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Sat, 14 Nov 2020 13:43:46 +0000 Subject: MLCE-278 issue with signed-int8 quantized model Signed-off-by: Teresa Charlin Change-Id: I144ebfca524f4cdee9cc82eef3995c6b32bfc40b --- src/backends/aclCommon/ArmComputeSubgraphUtils.hpp | 3 +-- src/backends/backendsCommon/LayerSupportRules.hpp | 5 ++--- src/backends/backendsCommon/test/WorkloadTestUtils.hpp | 3 +-- 3 files changed, 4 insertions(+), 7 deletions(-) diff --git a/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp b/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp index 79744ecf97..d969d5bc31 100644 --- a/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp +++ b/src/backends/aclCommon/ArmComputeSubgraphUtils.hpp @@ -71,9 +71,8 @@ inline armnn::Optional GetOptionalBiasTypeFromWeightsType(armnn 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 ddecc82172..e7aef0a140 100644 --- a/src/backends/backendsCommon/LayerSupportRules.hpp +++ b/src/backends/backendsCommon/LayerSupportRules.hpp @@ -23,12 +23,11 @@ inline armnn::Optional GetBiasTypeFromWeightsType(armnn::Option case armnn::DataType::Float16: case armnn::DataType::Float32: return weightsType; + case armnn::DataType::QAsymmS8: case armnn::DataType::QAsymmU8: - return armnn::DataType::Signed32; + case armnn::DataType::QSymmS8: case armnn::DataType::QSymmS16: return armnn::DataType::Signed32; - case armnn::DataType::QAsymmS8: - 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 583239122d..3173561a94 100644 --- a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp +++ b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp @@ -100,9 +100,8 @@ inline armnn::Optional GetBiasTypeFromWeightsType(armnn::Option 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: -- cgit v1.2.1