diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2021-09-24 16:02:57 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-10-15 15:53:10 +0000 |
commit | 6d9c982a5aec543d1f7f198f0fee10a7a3a78ddb (patch) | |
tree | 43627d9cebdd9196b8cc350f81b4c2137db92177 /arm_compute/runtime/OperatorList.h | |
parent | 273c28cc225d8edc7921397a754649529610c4e2 (diff) | |
download | ComputeLibrary-6d9c982a5aec543d1f7f198f0fee10a7a3a78ddb.tar.gz |
Conv3d support
* Add CpuDirectConv3d support for fp32 and fp16
* Dilation is not supported
* Need decouple
Partially resolve: COMPMID-4661
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: Ib1865b9ff328b684d131512b1baf77bc2f10318f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6430
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Diffstat (limited to 'arm_compute/runtime/OperatorList.h')
-rw-r--r-- | arm_compute/runtime/OperatorList.h | 10 |
1 files changed, 10 insertions, 0 deletions
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: |