From aa95ddc2abb7cef0b2edd03f7c4c9d9c6b9d7cf4 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 21 Jul 2020 22:45:13 +0100 Subject: COMPMID-3535: 9x9 Direct convolution support for CL and NHWC * Supported strides 1 and 2 Signed-off-by: Georgios Pinitas Change-Id: I4b9f087c0c328234159b2d1eacc2e465b3bb3c54 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3603 Reviewed-by: Michele Di Giorgio Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'arm_compute/core/CL') diff --git a/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h b/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h index b01ce6c0e8..5281a0c306 100644 --- a/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h +++ b/arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h @@ -54,7 +54,7 @@ public: * 1x1 convolution with stride_x = 1/2/3, stride_y = 1/2/3 * 3x3 convolution with stride_x = 1/2, stride_y = 1/2 * 5x5 convolution with stride_x = 1/2, stride_y = 1/2 - * 9x9 convolution with stride_x = 1/2, stride_y = 1/2, data_layout=NHWC + * 9x9 convolution with stride_x = 1/2, stride_y = 1/2 * * @param[in] input The input tensor to convolve. 3 lower dimensions represent a single input [width, height, IFM], * while every optional dimension from 4 and above represent a batch of inputs. Data types supported: QASYMM8_SIGNED/QASYMM8/F16/F32. @@ -74,7 +74,7 @@ public: * 1x1 convolution with stride_x = 1/2/3, stride_y = 1/2/3 * 3x3 convolution with stride_x = 1/2, stride_y = 1/2 * 5x5 convolution with stride_x = 1/2, stride_y = 1/2 - * 9x9 convolution with stride_x = 1/2, stride_y = 1/2, data_layout=NHWC + * 9x9 convolution with stride_x = 1/2, stride_y = 1/2 * * @param[in] compile_context The compile context to be used. * @param[in] input The input tensor to convolve. 3 lower dimensions represent a single input [width, height, IFM], -- cgit v1.2.1