From 2cd7a37af612723b712ca37d5fa99e17a84d7425 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Tue, 12 May 2020 21:03:56 +0100 Subject: COMPMID-3464: Address NESoftmaxLayer failures for QASYMM8_SIGNED Normalization with the maximum value was causing results to wrap-around As a work-around we use saturating intrinsics to perform the operation Signed-off-by: Georgios Pinitas Change-Id: I719b7ac7ad274dc2ae339bc4a055f9200134ed97 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3184 Tested-by: Arm Jenkins Reviewed-by: Sang-Hoon Park Comments-Addressed: Arm Jenkins --- src/core/NEON/kernels/NESoftmaxLayerKernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/NESoftmaxLayerKernel.cpp') diff --git a/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp b/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp index 790c8bacc5..41bf03ad1d 100644 --- a/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp +++ b/src/core/NEON/kernels/NESoftmaxLayerKernel.cpp @@ -311,7 +311,7 @@ void logits_1d_softmax_qasymm8(const ITensor &in, const ITensor &max, void *cons for(; x <= (input_width - vec_size); x += vec_size) { auto vec_elements = wrapper::vloadq(in_ptr + x); - vec_elements = wrapper::vsub(vec_max, vec_elements); + vec_elements = wrapper::vqsub(vec_max, vec_elements); auto vec_elements_flt = convert_int_to_float(vec_elements); if(is_log) -- cgit v1.2.1