diff options
author | Dana Zlotnik <dana.zlotnik@arm.com> | 2021-12-21 13:34:42 +0200 |
---|---|---|
committer | Dana Zlotnik <dana.zlotnik@arm.com> | 2022-01-13 13:25:13 +0000 |
commit | c48a3e5431ac48fbbd53522e34c99ea4f4ce3e41 (patch) | |
tree | 5a72cc1ea1c4ac545bda07e95ddc14878a649983 /Android.bp | |
parent | b1812636bd16c522cf6ac8d4caed94c9cf35c1c5 (diff) | |
download | ComputeLibrary-c48a3e5431ac48fbbd53522e34c99ea4f4ce3e41.tar.gz |
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>
Diffstat (limited to 'Android.bp')
-rw-r--r-- | Android.bp | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/Android.bp b/Android.bp index 735566e569..0ce9f09290 100644 --- a/Android.bp +++ b/Android.bp @@ -505,7 +505,17 @@ cc_library_static { "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", |