diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-07-01 12:20:56 +0100 |
---|---|---|
committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-07-13 13:42:25 +0000 |
commit | 96f977e43f452a75f2658b820791cb3d3da9c0a3 (patch) | |
tree | fe279f0573d871c051bb49acf4b83f50b29a1647 /Android.bp | |
parent | 04b39e8e56112dabf6f5746117354680a9985841 (diff) | |
download | ComputeLibrary-96f977e43f452a75f2658b820791cb3d3da9c0a3.tar.gz |
Port NEWinogradConvolutionLayer
Rename to CpuWinogradConv2d
Allow memory to be injected externally
Change-Id: I1f0a26ea533e326a7c63df86e708895c31752a39
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5926
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 8b6dd8392c..0778b636ce 100644 --- a/Android.bp +++ b/Android.bp @@ -178,7 +178,6 @@ cc_library_static { "src/core/NEON/kernels/NEStridedSliceKernel.cpp", "src/core/NEON/kernels/NETileKernel.cpp", "src/core/NEON/kernels/NEWeightsReshapeKernel.cpp", - "src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.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", @@ -285,6 +284,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/CpuWinogradConv2dKernel.cpp", "src/core/cpu/kernels/activation/neon/fp16.cpp", "src/core/cpu/kernels/activation/neon/fp32.cpp", "src/core/cpu/kernels/activation/neon/qasymm8.cpp", @@ -653,6 +653,7 @@ cc_library_static { "src/runtime/cpu/operators/CpuSoftmax.cpp", "src/runtime/cpu/operators/CpuSub.cpp", "src/runtime/cpu/operators/CpuTranspose.cpp", + "src/runtime/cpu/operators/CpuWinogradConv2d.cpp", "src/runtime/cpu/operators/internal/CpuGemmAssemblyDispatch.cpp", "src/runtime/gpu/cl/operators/ClActivation.cpp", "src/runtime/gpu/cl/operators/ClAdd.cpp", |