aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp b/src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp
index 962402549..08230074c 100644
--- a/src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp
+++ b/src/runtime/NEON/functions/NEFFTConvolutionLayer.cpp
@@ -118,8 +118,11 @@ void NEFFTConvolutionLayer::configure(ITensor *input, const ITensor *weights, co
ITensor *output_to_use = _has_bias ? &_bias_output : output;
// Permute bias
- _permute_bias_func.configure(biases, &_permuted_bias, PermutationVector(1U, 2U, 0U));
- _permuted_bias.info()->set_data_layout(DataLayout::NCHW);
+ if(biases != nullptr)
+ {
+ _permute_bias_func.configure(biases, &_permuted_bias, PermutationVector(1U, 2U, 0U));
+ _permuted_bias.info()->set_data_layout(DataLayout::NCHW);
+ }
// Permute input if needed
_needs_permute = input->info()->data_layout() == DataLayout::NHWC;