diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-01-20 16:41:12 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-02-09 18:25:46 +0000 |
| commit | 373b407558f99eb4bba632c170d03d807941dd2a (patch) | |
| tree | 448bb0225fa8b5fdfa48ddee973ec0b51a115f44 /Android.bp | |
| parent | 4841c97170b85be0706b65d424e967e561cef932 (diff) | |
| download | ComputeLibrary-373b407558f99eb4bba632c170d03d807941dd2a.tar.gz | |
Make Softmax kernels and operator stateless
COMPMID-3997
Change-Id: I3a3cc76d8247dd769d9a5e6e171d718ea909312c
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4986
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@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 31bc14b6b3..04f9d93c65 100644 --- a/Android.bp +++ b/Android.bp @@ -300,7 +300,6 @@ cc_library_static { "src/core/NEON/kernels/NESobel3x3Kernel.cpp", "src/core/NEON/kernels/NESobel5x5Kernel.cpp", "src/core/NEON/kernels/NESobel7x7Kernel.cpp", - "src/core/NEON/kernels/NESoftmaxLayerKernel.cpp", "src/core/NEON/kernels/NESpaceToBatchLayerKernel.cpp", "src/core/NEON/kernels/NESpaceToDepthLayerKernel.cpp", "src/core/NEON/kernels/NEStackLayerKernel.cpp", @@ -405,6 +404,7 @@ cc_library_static { "src/core/cpu/kernels/CpuPoolingAssemblyWrapperKernel.cpp", "src/core/cpu/kernels/CpuPoolingKernel.cpp", "src/core/cpu/kernels/CpuReshapeKernel.cpp", + "src/core/cpu/kernels/CpuSoftmaxKernel.cpp", "src/core/cpu/kernels/CpuSubKernel.cpp", "src/core/cpu/kernels/activation/NEON/fp16.cpp", "src/core/cpu/kernels/activation/NEON/fp32.cpp", @@ -801,6 +801,7 @@ cc_library_static { "src/runtime/cpu/operators/CpuPooling.cpp", "src/runtime/cpu/operators/CpuPoolingAssemblyDispatch.cpp", "src/runtime/cpu/operators/CpuReshape.cpp", + "src/runtime/cpu/operators/CpuSoftmax.cpp", "src/runtime/cpu/operators/CpuSub.cpp", "src/runtime/gpu/cl/operators/ClActivation.cpp", "src/runtime/gpu/cl/operators/ClAdd.cpp", |