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 /Android.bp | |
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 'Android.bp')
-rw-r--r-- | Android.bp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/Android.bp b/Android.bp index 36d392de57..ccfb2c707f 100644 --- a/Android.bp +++ b/Android.bp @@ -397,6 +397,7 @@ cc_library_static { "src/cpu/kernels/CpuDequantizeKernel.cpp", "src/cpu/kernels/CpuDirectConv2dKernel.cpp", "src/cpu/kernels/CpuDirectConv2dOutputStageKernel.cpp", + "src/cpu/kernels/CpuDirectConv3dKernel.cpp", "src/cpu/kernels/CpuElementwiseKernel.cpp", "src/cpu/kernels/CpuElementwiseUnaryKernel.cpp", "src/cpu/kernels/CpuFillKernel.cpp", @@ -477,6 +478,7 @@ cc_library_static { "src/cpu/operators/CpuDepthwiseConv2dAssemblyDispatch.cpp", "src/cpu/operators/CpuDequantize.cpp", "src/cpu/operators/CpuDirectConv2d.cpp", + "src/cpu/operators/CpuDirectConv3d.cpp", "src/cpu/operators/CpuElementwise.cpp", "src/cpu/operators/CpuElementwiseUnary.cpp", "src/cpu/operators/CpuFill.cpp", @@ -736,6 +738,7 @@ cc_library_static { "src/runtime/NEON/functions/NECast.cpp", "src/runtime/NEON/functions/NEChannelShuffleLayer.cpp", "src/runtime/NEON/functions/NEConcatenateLayer.cpp", + "src/runtime/NEON/functions/NEConv3D.cpp", "src/runtime/NEON/functions/NEConvertFullyConnectedWeights.cpp", "src/runtime/NEON/functions/NEConvolutionLayer.cpp", "src/runtime/NEON/functions/NECopy.cpp", |