From 8ec0bb6d9027bb7505d6fa0eada42a52c6e1073b Mon Sep 17 00:00:00 2001 From: Vidhya Sudhan Loganathan Date: Tue, 23 Apr 2019 10:40:44 +0100 Subject: COMPMID-2117 : Use FFT convolution if output feature map depth is less than input Change-Id: I588f467258aaa86a9fab8eb96603e02f24362d05 Signed-off-by: Vidhya Sudhan Loganathan Reviewed-on: https://review.mlplatform.org/c/1010 Comments-Addressed: Arm Jenkins Reviewed-by: Gian Marco Iodice Tested-by: Arm Jenkins --- arm_compute/core/Types.h | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'arm_compute/core/Types.h') 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 */ -- cgit v1.2.1