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 /arm_compute | |
| 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 'arm_compute')
| -rw-r--r-- | arm_compute/core/Types.h | 7 | ||||
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLConvolutionLayer.h | 4 |
2 files changed, 7 insertions, 4 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 5a469a6c9a..24e91bd3c5 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -128,9 +128,10 @@ enum class DataLayoutDimension /** Available ConvolutionMethod*/ enum class ConvolutionMethod { - GEMM, /**< Convolution using GEMM */ - DIRECT, /**< Direct convolution */ - WINOGRAD /**< Convolution using Winograd */ + GEMM, /**< Convolution using GEMM */ + DIRECT, /**< Direct convolution */ + WINOGRAD, /**< Convolution using Winograd */ + FFT /**< Convolution using FFT */ }; /** Padding mode to use for PadLayer */ diff --git a/arm_compute/runtime/CL/functions/CLConvolutionLayer.h b/arm_compute/runtime/CL/functions/CLConvolutionLayer.h index 8270e9723e..96f69f21d5 100644 --- a/arm_compute/runtime/CL/functions/CLConvolutionLayer.h +++ b/arm_compute/runtime/CL/functions/CLConvolutionLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -25,6 +25,7 @@ #define __ARM_COMPUTE_CLCONVOLUTIONLAYER_H__ #include "arm_compute/runtime/CL/functions/CLDirectConvolutionLayer.h" +#include "arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h" #include "arm_compute/runtime/CL/functions/CLGEMMConvolutionLayer.h" #include "arm_compute/runtime/CL/functions/CLWinogradConvolutionLayer.h" #include "arm_compute/runtime/IFunction.h" @@ -39,6 +40,7 @@ namespace arm_compute * -# @ref CLGEMMConvolutionLayer * -# @ref CLWinogradConvolutionLayer * -# @ref CLDirectConvolutionLayer + * -# @ref CLFFTConvolutionLayer */ class CLConvolutionLayer : public IFunction { |