diff options
| author | Sheri Zhang <sheri.zhang@arm.com> | 2021-04-22 14:41:12 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-04-28 12:52:32 +0000 |
| commit | a47dcc229d912d4e4bb5afa37220d20451f243a7 (patch) | |
| tree | f8b296701fbdebfc7d29abc09144c49619bcca1c /arm_compute/runtime/CL/functions/CLPoolingLayer.h | |
| parent | 2b7fee089c76226bfafcae77ba49f1eddb1e01da (diff) | |
| download | ComputeLibrary-a47dcc229d912d4e4bb5afa37220d20451f243a7.tar.gz | |
Update operator list documentation
All the common information for the operators are stored in OperatorList.h.
All data type and data layout information for the operators are store in the function header files.
Partially resolve: COMPMID-4199
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I272948cfb3f84e42232a82dd84c0158d84642099
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5511
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/CL/functions/CLPoolingLayer.h')
| -rw-r--r-- | arm_compute/runtime/CL/functions/CLPoolingLayer.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/arm_compute/runtime/CL/functions/CLPoolingLayer.h b/arm_compute/runtime/CL/functions/CLPoolingLayer.h index ec1c4eb768..902feca234 100644 --- a/arm_compute/runtime/CL/functions/CLPoolingLayer.h +++ b/arm_compute/runtime/CL/functions/CLPoolingLayer.h @@ -54,6 +54,18 @@ public: CLPoolingLayer &operator=(CLPoolingLayer &&) = default; /** Set the input and output tensors. * + * Valid data layouts: + * - NHWC + * - NCHW + * + * Valid data type configurations: + * |src |dst | + * |:--------------|:--------------| + * |QASYMM8 |QASYMM8 | + * |QASYMM8_SIGNED |QASYMM8_SIGNED | + * |F16 |F16 | + * |F32 |F32 | + * * @param[in,out] input Source tensor. (Written to only when padding != 0) Data types supported: QASYMM8/QASYMM8_SIGNED/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. |