diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2019-08-19 13:31:38 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-08-20 17:05:45 +0100 |
| commit | d2532e00deebede5e2a6148e79d4aa5c8bae25c5 (patch) | |
| tree | 7e505baba46f550fcd5f50cfad65529ca1c81160 /src | |
| parent | 7c2d92c477e893b797d8db0a1bb1beffe7c26a63 (diff) | |
| download | ComputeLibrary-d2532e00deebede5e2a6148e79d4aa5c8bae25c5.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) |