diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-05-15 13:24:00 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-05-15 15:06:13 +0000 |
commit | 68c6a7948270ce6cc00cdfb3f51e9bf9b9838832 (patch) | |
tree | 3905718ebe8ddd0a22665fe5b13c21930a80c612 /src/runtime/CL | |
parent | 6a98a6e322bfb03f98ac9c4dfdc932ec4bea1fd7 (diff) | |
download | ComputeLibrary-68c6a7948270ce6cc00cdfb3f51e9bf9b9838832.tar.gz |
COMPMID-2331: Fix FFTConvolutionLayer when no bias is used.
Change-Id: I33c53e97dc8c7f5ebc195055aba8310b384a79b2
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1136
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL')
-rw-r--r-- | src/runtime/CL/functions/CLFFTConvolutionLayer.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
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; |