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 /arm_compute/runtime/NEON | |
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 'arm_compute/runtime/NEON')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEConv3D.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEConv3D.h b/arm_compute/runtime/NEON/functions/NEConv3D.h index 2b3a45f0af..2a3c5351b0 100644 --- a/arm_compute/runtime/NEON/functions/NEConv3D.h +++ b/arm_compute/runtime/NEON/functions/NEConv3D.h @@ -66,6 +66,8 @@ public: * |:--------------|:------------------|:------|:--------------| * |F16 |F16 |F16 |F16 | * |F32 |F32 |F32 |F32 | + * |QASYMM8 |QASYMM8 |S32 |QASYMM8 | + * |QASYMM8_SIGNED |QASYMM8_SIGNED |S32 |QASYMM8_SIGNED | * * @param[in] input Source tensor. 4 lower dimensions represent a single input [IFM, width, height, depth], * while every optional dimension from 5 and above represent a batch of inputs. |