diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-04-20 15:20:31 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2021-04-20 18:23:28 +0000 |
commit | 3294449a6dfc0053d89188bb2445e03cfddd2501 (patch) | |
tree | c640d324131d74138a702832eb9655c190b7acca | |
parent | 778b95cb755880ab4adc972fbf3b7022a99b63f9 (diff) | |
download | ComputeLibrary-3294449a6dfc0053d89188bb2445e03cfddd2501.tar.gz |
Fix biases check in NEDepthwise
Check if biases are null before passing them to
CpuDepthwiseNativeKernel.
Resolves: COMPMID-4389
Change-Id: I29acbdefb75b9e81c293801a265e9b850d8d00b9
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5464
Reviewed-by: TeresaARM <teresa.charlinreyes@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r-- | src/runtime/NEON/functions/NEDepthwiseConvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEDepthwiseConvolutionLayer.cpp b/src/runtime/NEON/functions/NEDepthwiseConvolutionLayer.cpp index e1ceb0f083..1d7b1c677b 100644 --- a/src/runtime/NEON/functions/NEDepthwiseConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEDepthwiseConvolutionLayer.cpp @@ -255,7 +255,7 @@ void NEDepthwiseConvolutionLayer::NEDepthwiseConvolutionLayerGeneric::configure( } auto depthwise_conv_kernel = std::make_unique<cpu::kernels::CpuDepthwiseConvolutionNativeKernel>(); - depthwise_conv_kernel->configure(input_to_use->info(), weights_to_use->info(), biases->info(), output_to_use->info(), info); + depthwise_conv_kernel->configure(input_to_use->info(), weights_to_use->info(), biases == nullptr ? nullptr : biases->info(), output_to_use->info(), info); if(_impl->is_nchw) { |