COMPMID-3442: Add support of negative axis in NESoftmaxLayer and reference code
authorSheri Zhang <sheri.zhang@arm.com>
Tue, 5 May 2020 10:47:36 +0000 (11:47 +0100)
committerSheri Zhang <sheri.zhang@arm.com>
Tue, 12 May 2020 13:12:58 +0000 (13:12 +0000)
commit1f567afcdfb2919fab417f0060155deda7132df8
tree79631c4b121b89ff261156c41d1cc217afd891fc
parentc630e94d143ac5f46381f53a4994b29ea7ef2ac0
COMPMID-3442: Add support of negative axis in NESoftmaxLayer and reference code

Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I285cc3b74ac0a45f0ad5830baed5237cea568f15
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3147
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h
arm_compute/runtime/NEON/functions/NESoftmaxLayer.h
src/runtime/NEON/functions/NESoftmaxLayer.cpp
tests/validation/NEON/SoftmaxLayer.cpp
tests/validation/fixtures/SoftmaxLayerFixture.h
tests/validation/reference/LogSoftmaxLayer.cpp
tests/validation/reference/LogSoftmaxLayer.h
tests/validation/reference/SoftmaxLayer.cpp
tests/validation/reference/SoftmaxLayer.h