diff options
author | Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-04-23 10:40:44 +0100 |
---|---|---|
committer | VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-04-29 11:37:29 +0000 |
commit | 8ec0bb6d9027bb7505d6fa0eada42a52c6e1073b (patch) | |
tree | 71ac1c10ca882032b6c289233752cb1d0407fb67 /src/runtime/CL/functions/CLFFTConvolutionLayer.cpp | |
parent | 14c4e0f1c3e3bfebb33eb1ceb8156b3a090f4d11 (diff) | |
download | ComputeLibrary-8ec0bb6d9027bb7505d6fa0eada42a52c6e1073b.tar.gz |
COMPMID-2117 : Use FFT convolution if output feature map depth is less than input
Change-Id: I588f467258aaa86a9fab8eb96603e02f24362d05
Signed-off-by: Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1010
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLFFTConvolutionLayer.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLFFTConvolutionLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp b/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp index 6a89b9d304..77097d2b63 100644 --- a/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLFFTConvolutionLayer.cpp @@ -279,7 +279,7 @@ Status CLFFTConvolutionLayer::validate(const ITensorInfo *input, const ITensorIn if((output != nullptr) && (output->total_size() != 0)) { ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input, output); - ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(input, output); + ARM_COMPUTE_RETURN_ERROR_ON((input->tensor_shape()[idx_height] != output->tensor_shape()[idx_height]) || (input->tensor_shape()[idx_width] != output->tensor_shape()[idx_width])); // Validate Activation Layer if(act_info.enabled()) |