diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2019-08-14 11:33:11 +0100 |
|---|---|---|
| committer | mike.kelly <mike.kelly@arm.com> | 2019-08-14 14:03:13 +0000 |
| commit | 4992c344c888774da0e5cffd4b607828bf46b74f (patch) | |
| tree | 1546984e53e9d4701f2e847e17f426f0dcbb38a8 | |
| parent | 97520101bb98a30315dd6e31198b08ba050b58c0 (diff) | |
| download | armnn-4992c344c888774da0e5cffd4b607828bf46b74f.tar.gz | |
IVGCVSW-3633 Fixed possible driver crash
* Fixed possible driver crash due to supportedTypes arrays with the wrong size.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I52702b84089d02d3c63ff7aa5923a9f5628bec91
| -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 |