From 9104cd559222b98f2b21f14d4fd561ed4a4e9bc2 Mon Sep 17 00:00:00 2001 From: Adnan AlSinan Date: Wed, 6 Apr 2022 16:19:31 +0100 Subject: Add support for int8 CpuPool3d - Add implementation for the CPU pooling 3d layer. - NDHWC data layout support. - Support QASYMM8/QASYMM8_SIGNED. - Add Pooling helper file for Pool3d/2d common functions. Resolves COMPMID-4668 Change-Id: Iadf042036b076099c2353d6e2fe9fc623bc263d8 Signed-off-by: Adnan AlSinan Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7387 Reviewed-by: Gunes Bayir Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/cpu/kernels/pool3d/list.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/cpu/kernels/pool3d/list.h') diff --git a/src/cpu/kernels/pool3d/list.h b/src/cpu/kernels/pool3d/list.h index ece780eb0b..3426360f93 100644 --- a/src/cpu/kernels/pool3d/list.h +++ b/src/cpu/kernels/pool3d/list.h @@ -31,6 +31,8 @@ namespace cpu #define DECLARE_POOLING_KERNEL(func_name) \ void func_name(const ITensor *src0, ITensor *dst0, Pooling3dLayerInfo &, const Window &window) +DECLARE_POOLING_KERNEL(neon_q8_pool3d); +DECLARE_POOLING_KERNEL(neon_q8_signed_pool3d); DECLARE_POOLING_KERNEL(neon_fp16_pool3d); DECLARE_POOLING_KERNEL(neon_fp32_pool3d); -- cgit v1.2.1