diff options
-rw-r--r-- | src/backends/backendsCommon/LayerSupportRules.hpp | 2 | ||||
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 5 |
2 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/LayerSupportRules.hpp b/src/backends/backendsCommon/LayerSupportRules.hpp index 3a2ae06f5a..03bec53353 100644 --- a/src/backends/backendsCommon/LayerSupportRules.hpp +++ b/src/backends/backendsCommon/LayerSupportRules.hpp @@ -27,6 +27,8 @@ inline armnn::Optional<armnn::DataType> GetBiasTypeFromWeightsType(armnn::Option return armnn::DataType::Signed32; case armnn::DataType::QSymmS16: return armnn::DataType::Signed32; + case armnn::DataType::QAsymmS8: + return armnn::DataType::Signed32; default: BOOST_ASSERT_MSG(false, "GetBiasTypeFromWeightsType(): Unsupported data type."); } diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 7b25a436e9..607c86b112 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -939,13 +939,14 @@ bool RefLayerSupport::IsFullyConnectedSupported(const TensorInfo& input, if (descriptor.m_BiasEnabled) { // Defined supported types for bias - std::array<DataType, 4> + std::array<DataType, 5> supportedBiasTypes = { DataType::BFloat16, DataType::Float32, DataType::Float16, - DataType::Signed32 + DataType::Signed32, + DataType::QAsymmS8 }; supported &= CheckSupportRule(TypeAnyOf(biases, supportedBiasTypes), reasonIfUnsupported, |