From 7930db48e12dd3a14c1971f41f5b83527efea281 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Thu, 22 Nov 2018 17:36:28 +0000 Subject: COMPMID-1728 CL: Implement ArgMax/ArgMin Change-Id: I7eae2e55cc0b0b7bbebb7617299daaca6f75f40c Reviewed-on: https://review.mlplatform.org/292 Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- utils/TypePrinter.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'utils') diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 91000cb702..2b81192a44 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -1342,6 +1342,12 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ReductionOperation & case ReductionOperation::MEAN_SUM: os << "MEAN_SUM"; break; + case ReductionOperation::ARG_IDX_MAX: + os << "ARG_IDX_MAX"; + break; + case ReductionOperation::ARG_IDX_MIN: + os << "ARG_IDX_MIN"; + break; default: ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); } -- cgit v1.2.1