commit 7930db48e12dd3a14c1971f41f5b83527efea281
author Michalis Spyrou <michalis.spyrou@arm.com> Thu Nov 22 17:36:28 2018 +0000
committer Michalis Spyrou <michalis.spyrou@arm.com> Fri Nov 30 15:46:49 2018 +0000
tree d17899ba82203423320bfa8d2dea1e07b045c898
parent 95abfddfa08ab85d4f88c6f4d2e077969178f2d5

COMPMID-1728 CL: Implement ArgMax/ArgMin

Change-Id: I7eae2e55cc0b0b7bbebb7617299daaca6f75f40c
Reviewed-on: https://review.mlplatform.org/292
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>

utils/TypePrinter.h

diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h
index 91000cb..2b81192 100644
--- a/utils/TypePrinter.h
+++ b/utils/TypePrinter.h
@@ -1342,6 +1342,12 @@
         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!");
     }