aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures/ReductionOperationFixture.h
diff options
context:
space:
mode:
authorPablo Marquez Tello <pablo.tello@arm.com>2023-07-27 18:02:37 +0100
committerPablo Marquez Tello <pablo.tello@arm.com>2023-08-01 12:03:06 +0000
commit4cb0bd488f70a07b222e1ed7008d888642dfec6f (patch)
tree00d61aa30e0e2d13a9661f65315d408c859f6dc2 /tests/validation/fixtures/ReductionOperationFixture.h
parent16b37527906c68885f81a8db35f9d6040d73efec (diff)
downloadComputeLibrary-4cb0bd488f70a07b222e1ed7008d888642dfec6f.tar.gz
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 <pablo.tello@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10003 Reviewed-by: Viet-Hoa Do <viet-hoa.do@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/fixtures/ReductionOperationFixture.h')
-rw-r--r--tests/validation/fixtures/ReductionOperationFixture.h2
1 files changed, 1 insertions, 1 deletions
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<T, T>(src, dst_shape, axis, op, quantization_info);
+ return reference::reduction_operation<T, T>(src, dst_shape, axis, op, data_type, quantization_info);
}
TensorType _target{};