diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2021-07-06 15:01:35 +0100 |
---|---|---|
committer | Manuel Bottini <manuel.bottini@arm.com> | 2021-07-15 16:27:56 +0000 |
commit | 29599d01a8f498e33b9c6995bd879473dc02e077 (patch) | |
tree | 2dcbd399c5cf0a87edbd1885e29e8cb1ed1ce9bc /Android.bp | |
parent | 13ef1763d6eef9606afaed90fb258d1a4577f15b (diff) | |
download | ComputeLibrary-29599d01a8f498e33b9c6995bd879473dc02e077.tar.gz |
Port NEGEMMConvolutionLayer
Details:
port NEWeightsReshapeKernel to CpuWeightsReshapeKernel
port NEGEMMConvolutionLayer to CpuGEMMConvolutionLayer
Resolves: COMPMID-4509
Change-Id: I3c7051e2c3f6d808a7ccb898aad70e5b221b9dc3
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5938
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'Android.bp')
-rw-r--r-- | Android.bp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/Android.bp b/Android.bp index b32a9da895..f2934cb37d 100644 --- a/Android.bp +++ b/Android.bp @@ -176,7 +176,6 @@ cc_library_static { "src/core/NEON/kernels/NEStackLayerKernel.cpp", "src/core/NEON/kernels/NEStridedSliceKernel.cpp", "src/core/NEON/kernels/NETileKernel.cpp", - "src/core/NEON/kernels/NEWeightsReshapeKernel.cpp", "src/core/NEON/kernels/arm_conv/depthwise/depthwise_fp16.cpp", "src/core/NEON/kernels/arm_conv/depthwise/depthwise_fp32.cpp", "src/core/NEON/kernels/arm_conv/depthwise/depthwise_s8q.cpp", @@ -283,6 +282,7 @@ cc_library_static { "src/core/cpu/kernels/CpuSoftmaxKernel.cpp", "src/core/cpu/kernels/CpuSubKernel.cpp", "src/core/cpu/kernels/CpuTransposeKernel.cpp", + "src/core/cpu/kernels/CpuWeightsReshapeKernel.cpp", "src/core/cpu/kernels/CpuWinogradConv2dKernel.cpp", "src/core/cpu/kernels/activation/neon/fp16.cpp", "src/core/cpu/kernels/activation/neon/fp32.cpp", @@ -639,6 +639,7 @@ cc_library_static { "src/runtime/cpu/operators/CpuFlatten.cpp", "src/runtime/cpu/operators/CpuFloor.cpp", "src/runtime/cpu/operators/CpuGemm.cpp", + "src/runtime/cpu/operators/CpuGemmConvolution.cpp", "src/runtime/cpu/operators/CpuGemmDirectConv2d.cpp", "src/runtime/cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp", "src/runtime/cpu/operators/CpuGemmLowpOutputStage.cpp", |