diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-01-03 11:10:25 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-01-10 16:24:26 +0000 |
| commit | aea14c63e2efeda9d5f7492099389d439c65204f (patch) | |
| tree | 176a6181bbf00e4df078d5da0a17dd44f248958e /tests/validation/reference/ReductionOperation.h | |
| parent | c10bc0b5db5169a6ccea02a1aaefe34f082709e5 (diff) | |
| download | ComputeLibrary-aea14c63e2efeda9d5f7492099389d439c65204f.tar.gz | |
COMPMID-1764 NEON: Implement ArgMax/ArgMin
Change-Id: Ibe23aa90b36ffd8553d1d1c35fada5d300fab829
Reviewed-on: https://review.mlplatform.org/475
Reviewed-by: Isabella Gottardi <isabella.gottardi@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Diffstat (limited to 'tests/validation/reference/ReductionOperation.h')
| -rw-r--r-- | tests/validation/reference/ReductionOperation.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/validation/reference/ReductionOperation.h b/tests/validation/reference/ReductionOperation.h index 859b57aa7b..9f7050f551 100644 --- a/tests/validation/reference/ReductionOperation.h +++ b/tests/validation/reference/ReductionOperation.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2018 ARM Limited. + * Copyright (c) 2017-2019 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -35,10 +35,10 @@ namespace validation { namespace reference { -template <typename T> -SimpleTensor<T> reduction_operation(const SimpleTensor<T> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op); +template <typename T, typename OT> +SimpleTensor<OT> reduction_operation(const SimpleTensor<T> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op); } // namespace reference } // namespace validation } // namespace test } // namespace arm_compute -#endif /* __ARM_COMPUTE_TEST_FLOOR_H__ */ +#endif /* __ARM_COMPUTE_TEST_REDUCTION_OPERATION_H__ */ |