From f727ef49e2109bdac105dd6575d2e336adf780a3 Mon Sep 17 00:00:00 2001 From: Freddie Liardet Date: Mon, 18 Oct 2021 13:28:57 +0100 Subject: Add uint8/int8 support to cpu conv3d Add support for qasymm8/qasymm8_signed in cpu conv3d. Resolves: COMPMID-4665 Signed-off-by: Freddie Liardet Change-Id: I2450bb6f24969745c8b936f4b657bd406b788c57 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6478 Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena Comments-Addressed: Arm Jenkins --- src/cpu/kernels/conv3d/neon/list.h | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/cpu/kernels/conv3d/neon/list.h') diff --git a/src/cpu/kernels/conv3d/neon/list.h b/src/cpu/kernels/conv3d/neon/list.h index b24785a48f..3e2db664d7 100644 --- a/src/cpu/kernels/conv3d/neon/list.h +++ b/src/cpu/kernels/conv3d/neon/list.h @@ -29,6 +29,7 @@ #include "arm_compute/runtime/FunctionDescriptors.h" #include "src/core/NEON/wrapper/wrapper.h" #include "src/core/helpers/WindowHelpers.h" +#include "src/cpu/kernels/conv3d/neon/quantized.h" namespace arm_compute { @@ -171,6 +172,7 @@ void directconv3d_float_neon_ndhwc(const ITensor *src0, const ITensor *src1, con }, out); } + } // namespace cpu } // namespace arm_compute #endif // SRC_CORE_NEON_KERNELS_CONV3D_LIST_H \ No newline at end of file -- cgit v1.2.1