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 /arm_compute/graph.h | |
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 'arm_compute/graph.h')
0 files changed, 0 insertions, 0 deletions