tree 9985ca8ad1910d48a84aa9781fe3156e614ff5f4
parent f1f1f87132690a8061801ef1a4638d637c780df7
author Gunes Bayir <gunes.bayir@arm.com> 1712700784 +0100
committer Gunes Bayir <gunes.bayir@arm.com> 1712840325 +0000

Add SME2 implementation of softmax for FP16

In addition to the softmax kernel, this patch fixes minor issues in the fp32 implementation.

Resolves: COMPMID-6920

Change-Id: Ibbd9f0af5f2a93fba0e92d72ba437279c34149d3
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11402
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>