From 1be8b82974da1090107d0318e846430a605b8b40 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 15 May 2019 13:24:00 +0100 Subject: COMPMID-2331: Fix FFTConvolutionLayer when no bias is used. Change-Id: I33c53e97dc8c7f5ebc195055aba8310b384a79b2 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/1136 Reviewed-by: Michalis Spyrou Comments-Addressed: Arm Jenkins Reviewed-by: Giuseppe Rossini Tested-by: Arm Jenkins --- src/runtime/CL/functions/CLFFTConvolutionLayer.cpp | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'src/runtime/CL/functions/CLFFTConvolutionLayer.cpp') diff --git a/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp b/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp index 77097d2b63..afb1cab520 100644 --- a/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp @@ -119,8 +119,11 @@ void CLFFTConvolutionLayer::configure(ICLTensor *input, const ICLTensor *weights ICLTensor *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; -- cgit v1.2.1