From 4cb0bd488f70a07b222e1ed7008d888642dfec6f Mon Sep 17 00:00:00 2001 From: Pablo Marquez Tello Date: Thu, 27 Jul 2023 18:02:37 +0100 Subject: Improved testing for ArgMinMax * ArgMinMax output was fixed to S32, this patch makes the changes required to allow other output types like U64/S64 * Made changes to the ArgMinMax fixture and tests to allow specifying output data type. * Made changes to the reference reduction_operation to allow specifying the output type * Added tests case to output S64 for the CL backend. * Added missing test cases in the neon backend. * Partially resolves MLCE-1089 Change-Id: I6f1cbc7093669d12c2a3aff6974cf19d83b2ecda Signed-off-by: Pablo Marquez Tello Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10003 Reviewed-by: Viet-Hoa Do Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- tests/validation/fixtures/ReductionOperationFixture.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/validation/fixtures/ReductionOperationFixture.h') diff --git a/tests/validation/fixtures/ReductionOperationFixture.h b/tests/validation/fixtures/ReductionOperationFixture.h index 36e6309d6f..b44f299486 100644 --- a/tests/validation/fixtures/ReductionOperationFixture.h +++ b/tests/validation/fixtures/ReductionOperationFixture.h @@ -134,7 +134,7 @@ protected: // Fill reference fill(src); - return reference::reduction_operation(src, dst_shape, axis, op, quantization_info); + return reference::reduction_operation(src, dst_shape, axis, op, data_type, quantization_info); } TensorType _target{}; -- cgit v1.2.1