From 6d9c982a5aec543d1f7f198f0fee10a7a3a78ddb Mon Sep 17 00:00:00 2001 From: Sheri Zhang Date: Fri, 24 Sep 2021 16:02:57 +0100 Subject: Conv3d support * Add CpuDirectConv3d support for fp32 and fp16 * Dilation is not supported * Need decouple Partially resolve: COMPMID-4661 Signed-off-by: Sheri Zhang Change-Id: Ib1865b9ff328b684d131512b1baf77bc2f10318f Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6430 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena --- arm_compute/runtime/OperatorList.h | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'arm_compute/runtime/OperatorList.h') diff --git a/arm_compute/runtime/OperatorList.h b/arm_compute/runtime/OperatorList.h index 7b00cbb726..4646974148 100644 --- a/arm_compute/runtime/OperatorList.h +++ b/arm_compute/runtime/OperatorList.h @@ -206,6 +206,16 @@ * */ +/** Conv3D + * + * Description: + * Function to compute a 3d convolution layer. + * + * Equivalent Android NNAPI Op: + * ANEURALNETWORKS_CONV_3D + * + */ + /** Copy * * Description: -- cgit v1.2.1