diff options
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 306bdc6706..2e639c4be4 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -137,10 +137,11 @@ enum class DataLayoutDimension /** Available ConvolutionMethod*/ enum class ConvolutionMethod { - GEMM, /**< Convolution using GEMM */ - DIRECT, /**< Direct convolution */ - WINOGRAD, /**< Convolution using Winograd */ - FFT /**< Convolution using FFT */ + GEMM, /**< Convolution using GEMM */ + GEMM_CONV2D, /**< Direct 2D GEMM convolution */ + DIRECT, /**< Direct convolution */ + WINOGRAD, /**< Convolution using Winograd */ + FFT /**< Convolution using FFT */ }; /** Available DepthwiseConvolutionFunction*/ |