diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2019-08-19 13:31:38 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2019-08-20 12:50:08 +0000 |
| commit | 9f0d5ecb52d5145a1fff51d1824ecfb8fe417023 (patch) | |
| tree | 0c936013aa340dce5c5db93ebb1ff3345ef1ab65 /src | |
| parent | 6f688fdd9b27523560fbff1a268a952417f4a807 (diff) | |
| download | ComputeLibrary-9f0d5ecb52d5145a1fff51d1824ecfb8fe417023.tar.gz | |
COMPMID-2597: Checking bias type in NEDeconvolutionLayer::validate even when bias == nullptr
Change-Id: If7fb2e6d11e4653208b58ebdbfcc284d6c5a2e9a
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1761
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src')
| -rw-r--r-- | src/runtime/NEON/functions/NEDeconvolutionLayer.cpp | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp index 4db1346654..2a09ba4285 100644 --- a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp @@ -73,13 +73,16 @@ Status NEDeconvolutionLayer::validate(const ITensorInfo *input, const ITensorInf info.pad().first, info.pad().second, stride_x, stride_y); ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input, weights); - if(is_data_type_quantized_asymmetric(input->data_type())) + if(bias != nullptr) { - ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(bias, 1, DataType::S32); - } - else - { - ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input, bias); + if(is_data_type_quantized_asymmetric(input->data_type())) + { + ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(bias, 1, DataType::S32); + } + else + { + ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input, bias); + } } if(output->tensor_shape().total_size() > 0) |