diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEPooling3dLayer.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEPooling3dLayer.h b/arm_compute/runtime/NEON/functions/NEPooling3dLayer.h index 7b31f916f6..4c5eb58e05 100644 --- a/arm_compute/runtime/NEON/functions/NEPooling3dLayer.h +++ b/arm_compute/runtime/NEON/functions/NEPooling3dLayer.h @@ -64,10 +64,12 @@ public: * |:--------------|:--------------| * |F16 |F16 | * |F32 |F32 | + * |QASYMM8 |QASYMM8 | + * |QASYMM8_SIGNED |QASYMM8_SIGNED | * * @note Source tensor is padded with -inf for MAX pooling and 0 otherwise * - * @param[in] input Source tensor. Data types supported: F16/F32. + * @param[in] input Source tensor. Data types supported: F16/F32/QASYMM8/QASYMM8_SIGNED. * @param[out] output Destination tensor. * @param[in] pool_info Contains pooling operation information described in @ref Pooling3dLayerInfo. */ @@ -75,7 +77,7 @@ public: /** Static function to check if given info will lead to a valid configuration of @ref NEPooling3dLayer * * - * @param[in] input Source tensor info. Data types supported: F16/F32. + * @param[in] input Source tensor info. Data types supported: F16/F32/QASYMM8/QASYMM8_SIGNED. * @param[in] output Destination tensor info. * @param[in] pool_info Contains pooling operation information described in @ref Pooling3dLayerInfo. * |