diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-11-14 13:43:46 +0000 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2020-11-17 02:04:09 +0000 |
commit | 9d07e00917d276c4cd4a3b5c3a08df3de9d82dbc (patch) | |
tree | e5906b33efaf5174b52832b0d9212f90f82d2a17 /src | |
parent | c1734380bf656a8f8da1accadf67ccdca4f30b52 (diff) | |
download | armnn-9d07e00917d276c4cd4a3b5c3a08df3de9d82dbc.tar.gz |
MLCE-278 issue with signed-int8 quantized model
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I144ebfca524f4cdee9cc82eef3995c6b32bfc40b
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/aclCommon/ArmComputeSubgraphUtils.hpp | 3 | ||||
-rw-r--r-- | src/backends/backendsCommon/LayerSupportRules.hpp | 5 | ||||
-rw-r--r-- | 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<armnn::DataType> 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<armnn::DataType> 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<armnn::DataType> 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: |