From cf87f509fc23d02c56569f794a3fb59e1b8be277 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Tue, 2 Feb 2021 14:59:09 +0000 Subject: Tweak scheduling use of SQDMULH in quantized AVG pooling Resolves COMPMID-4195 Change-Id: Ie5116c1ddddccafba40432fd4b5245bb27890a88 Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4997 Reviewed-by: TeresaARM Reviewed-by: Manuel Bottini Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- .../arm_conv/pooling/kernels/a64_u8q_nhwc_avg_generic_depthfirst.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/core/NEON/kernels/arm_conv/pooling/kernels/a64_u8q_nhwc_avg_generic_depthfirst.hpp') diff --git a/src/core/NEON/kernels/arm_conv/pooling/kernels/a64_u8q_nhwc_avg_generic_depthfirst.hpp b/src/core/NEON/kernels/arm_conv/pooling/kernels/a64_u8q_nhwc_avg_generic_depthfirst.hpp index 114eacf450..d46658f080 100644 --- a/src/core/NEON/kernels/arm_conv/pooling/kernels/a64_u8q_nhwc_avg_generic_depthfirst.hpp +++ b/src/core/NEON/kernels/arm_conv/pooling/kernels/a64_u8q_nhwc_avg_generic_depthfirst.hpp @@ -42,6 +42,7 @@ struct a64_u8q_nhwc_avg_generic_depthfirst constexpr static PoolingType pooling_type(void) { return PoolingType::AVERAGE; } + kern_type kernel = a64_u8q_nhwc_avg_generic_depthfirst_impl; a64_u8q_nhwc_avg_generic_depthfirst(const CPUInfo *) {} -- cgit v1.2.1