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>
diff --git a/Android.bp b/Android.bp
index 31bc14b..04f9d93 100644
--- a/Android.bp
+++ b/Android.bp
@@ -300,7 +300,6 @@
"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 @@
"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 @@
"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",