diff options
| author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-10-13 17:40:32 +0100 |
|---|---|---|
| committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-10-31 10:16:25 +0000 |
| commit | 29254aeb11a76c86449c2f38587e9144b2f2aacb (patch) | |
| tree | ca2df26e81c2417b34768ac325e0f7200b5265df /Android.bp | |
| parent | e5362e7e5dbccf81c5296a7e77154e11e1a14d2f (diff) | |
| download | ComputeLibrary-29254aeb11a76c86449c2f38587e9144b2f2aacb.tar.gz | |
Optimize CL softmax
* The new softmax implementation consists of only a single kernel.
- There are 2 versions of softmax, one for the x dimension
and one for any other dimensions.
- Softmax kernel handles both native and quantized data type.
Resolves: COMPMID-6447
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I4a9ae5bc63f78aebeaa85ee48a0d102c9c245eda
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10489
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'Android.bp')
| -rw-r--r-- | Android.bp | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/Android.bp b/Android.bp index dc35143f76..7681205770 100644 --- a/Android.bp +++ b/Android.bp @@ -68,7 +68,6 @@ opencl_srcs = [ "src/core/CL/cl_kernels/common/select.cl", "src/core/CL/cl_kernels/common/slice_ops.cl", "src/core/CL/cl_kernels/common/softmax_layer.cl", - "src/core/CL/cl_kernels/common/softmax_layer_quantized.cl", "src/core/CL/cl_kernels/common/stack_layer.cl", "src/core/CL/cl_kernels/common/tile.cl", "src/core/CL/cl_kernels/common/transpose.cl", |