diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-07-21 22:45:13 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-07-22 11:37:50 +0000 |
commit | aa95ddc2abb7cef0b2edd03f7c4c9d9c6b9d7cf4 (patch) | |
tree | 7fb62d8550934c628438b1aae6de053a3f923609 /arm_compute/core/CL/kernels | |
parent | f932d2c8409831cb9cb97a2eb65be93ad4709cd6 (diff) | |
download | ComputeLibrary-aa95ddc2abb7cef0b2edd03f7c4c9d9c6b9d7cf4.tar.gz |
COMPMID-3535: 9x9 Direct convolution support for CL and NHWC
* Supported strides 1 and 2
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I4b9f087c0c328234159b2d1eacc2e465b3bb3c54
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3603
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/CL/kernels')
-rw-r--r-- | arm_compute/core/CL/kernels/CLDirectConvolutionLayerKernel.h | 4 |
1 files changed, 2 insertions, 2 deletions
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], |