Decouple CpuSoftmaxKernel
Resolves COMPMID-4633
Change-Id: I9f93b28fbc3b18ccaeb453596dc8e0eddfe06b6a
Signed-off-by: Dana Zlotnik <dana.zlotnik@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6861
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/Android.bp b/Android.bp
index 735566e..0ce9f09 100644
--- a/Android.bp
+++ b/Android.bp
@@ -505,7 +505,17 @@
"src/cpu/kernels/scale/sve/integer.cpp",
"src/cpu/kernels/scale/sve/qasymm8.cpp",
"src/cpu/kernels/scale/sve/qasymm8_signed.cpp",
- "src/cpu/kernels/softmax/impl/sve/impl.cpp",
+ "src/cpu/kernels/softmax/generic/neon/fp16.cpp",
+ "src/cpu/kernels/softmax/generic/neon/fp32.cpp",
+ "src/cpu/kernels/softmax/generic/neon/qasymm8.cpp",
+ "src/cpu/kernels/softmax/generic/neon/qasymm8_signed.cpp",
+ "src/cpu/kernels/softmax/generic/sve/fp16.cpp",
+ "src/cpu/kernels/softmax/generic/sve/fp32.cpp",
+ "src/cpu/kernels/softmax/generic/sve/impl.cpp",
+ "src/cpu/kernels/softmax/generic/sve/qasymm8.cpp",
+ "src/cpu/kernels/softmax/generic/sve/qasymm8_signed.cpp",
+ "src/cpu/kernels/softmax/generic/sve2/qasymm8.cpp",
+ "src/cpu/kernels/softmax/generic/sve2/qasymm8_signed.cpp",
"src/cpu/kernels/sub/neon/qasymm8.cpp",
"src/cpu/kernels/sub/neon/qasymm8_signed.cpp",
"src/cpu/kernels/sub/neon/qsymm16.cpp",