diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-06-19 16:11:53 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:14:20 +0100 |
commit | ce093143ec7b554edefc533c90e45c80946cde51 (patch) | |
tree | 1e4aa13ba3fe10c93ca42e2f5477bd2c4888324e /arm_compute/core/CL/kernels/CLPoolingLayerKernel.h | |
parent | 4c2938ed50a78753bfbdbb2f3cbf43f5fed779f9 (diff) | |
download | ComputeLibrary-ce093143ec7b554edefc533c90e45c80946cde51.tar.gz |
COMPMID-403:Add support for 7x7 pooling on CL.
Change-Id: I3c2c8d7e8e61d7737170cb1568900ce4ac337068
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/78181
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Diffstat (limited to 'arm_compute/core/CL/kernels/CLPoolingLayerKernel.h')
-rw-r--r-- | arm_compute/core/CL/kernels/CLPoolingLayerKernel.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h b/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h index 546a40b15e..6c5091ff9e 100644 --- a/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h +++ b/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h @@ -52,6 +52,7 @@ public: * @param[in] input Source tensor. Data types supported: F16, F32. * @param[out] output Destination tensor. Data types supported: Same as @p input. * @param[in] pool_info Contains pooling operation information described in @ref PoolingLayerInfo. + * Supported pooling sizes : 2, 3 and 7 */ void configure(const ICLTensor *input, ICLTensor *output, const PoolingLayerInfo &pool_info); |