diff options
-rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 56ca437b21..22d7914ab9 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -372,7 +372,7 @@ bool RefLayerSupport::IsConvolution2dSupported(const TensorInfo& input, if (biases.has_value()) { - std::array<DataType,3> biasesSupportedTypes = { + std::array<DataType,2> biasesSupportedTypes = { DataType::Float32, DataType::Signed32 }; @@ -471,8 +471,8 @@ bool RefLayerSupport::IsDequantizeSupported(const TensorInfo& input, supported &= CheckSupportRule(TypeAnyOf(input, supportedInputTypes), reasonIfUnsupported, "Reference dequantize: input type not supported."); - std::array<DataType,2> supportedOutputTypes = { - DataType::Float32, + std::array<DataType,1> supportedOutputTypes = { + DataType::Float32 }; supported &= CheckSupportRule(TypeAnyOf(output, supportedOutputTypes), reasonIfUnsupported, @@ -491,7 +491,7 @@ bool RefLayerSupport::IsDetectionPostProcessSupported(const armnn::TensorInfo& i { bool supported = true; - std::vector<DataType> supportedInputTypes = + std::array<DataType,3> supportedInputTypes = { DataType::Float32, DataType::QuantisedAsymm8, @@ -1603,7 +1603,7 @@ bool RefLayerSupport::IsTransposeConvolution2dSupported(const TensorInfo& input, if (biases.has_value()) { - std::array<DataType,3> biasesSupportedTypes = + std::array<DataType,2> biasesSupportedTypes = { DataType::Float32, DataType::Signed32 |