From 327225d3b2f716d5c62d801a7fafc7d377521f34 Mon Sep 17 00:00:00 2001 From: Manuel Bottini Date: Tue, 13 Apr 2021 13:09:30 +0100 Subject: Port NEDirectConvolutionLayer to new API Partially resolves: COMPMID-4009 Change-Id: I19ffb61c5c4541134a5028677d2d81228740e454 Signed-off-by: Manuel Bottini Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5419 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: SiCong Li Reviewed-by: Georgios Pinitas Reviewed-by: Michele Di Giorgio --- Android.bp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'Android.bp') diff --git a/Android.bp b/Android.bp index 7e2da8fe53..17281a49d1 100644 --- a/Android.bp +++ b/Android.bp @@ -176,8 +176,6 @@ cc_library_static { "src/core/NEON/kernels/NEDepthConvertLayerKernel.cpp", "src/core/NEON/kernels/NEDepthToSpaceLayerKernel.cpp", "src/core/NEON/kernels/NEDepthwiseConvolutionLayerNativeKernel.cpp", - "src/core/NEON/kernels/NEDirectConvolutionLayerKernel.cpp", - "src/core/NEON/kernels/NEDirectConvolutionLayerOutputStageKernel.cpp", "src/core/NEON/kernels/NEFFTDigitReverseKernel.cpp", "src/core/NEON/kernels/NEFFTRadixStageKernel.cpp", "src/core/NEON/kernels/NEFFTScaleKernel.cpp", @@ -301,6 +299,8 @@ cc_library_static { "src/core/cpu/kernels/CpuConcatenateWidthKernel.cpp", "src/core/cpu/kernels/CpuCopyKernel.cpp", "src/core/cpu/kernels/CpuDequantizationKernel.cpp", + "src/core/cpu/kernels/CpuDirectConvolutionKernel.cpp", + "src/core/cpu/kernels/CpuDirectConvolutionStageKernel.cpp", "src/core/cpu/kernels/CpuElementwiseKernel.cpp", "src/core/cpu/kernels/CpuElementwiseUnaryKernel.cpp", "src/core/cpu/kernels/CpuFillKernel.cpp", @@ -630,6 +630,7 @@ cc_library_static { "src/runtime/cpu/operators/CpuConcatenate.cpp", "src/runtime/cpu/operators/CpuCopy.cpp", "src/runtime/cpu/operators/CpuDequantization.cpp", + "src/runtime/cpu/operators/CpuDirectConvolution.cpp", "src/runtime/cpu/operators/CpuElementwise.cpp", "src/runtime/cpu/operators/CpuElementwiseUnary.cpp", "src/runtime/cpu/operators/CpuFill.cpp", -- cgit v1.2.1