diff options
| author | Freddie Liardet <frederick.liardet@arm.com> | 2021-10-18 13:28:57 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-10-22 13:54:17 +0000 |
| commit | f727ef49e2109bdac105dd6575d2e336adf780a3 (patch) | |
| tree | bc7e9381fab039cd1281bf189188ece318a4a955 /src/cpu/kernels/conv3d/neon/list.h | |
| parent | 00ff6641b70382a7882b8a04fe67fb4a93ee5675 (diff) | |
| download | ComputeLibrary-f727ef49e2109bdac105dd6575d2e336adf780a3.tar.gz | |
Add uint8/int8 support to cpu conv3d
Add support for qasymm8/qasymm8_signed in cpu conv3d.
Resolves: COMPMID-4665
Signed-off-by: Freddie Liardet <frederick.liardet@arm.com>
Change-Id: I2450bb6f24969745c8b936f4b657bd406b788c57
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6478
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/cpu/kernels/conv3d/neon/list.h')
| -rw-r--r-- | src/cpu/kernels/conv3d/neon/list.h | 2 |
1 files changed, 2 insertions, 0 deletions
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 |