diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-11-22 17:36:28 +0000 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-11-30 15:46:49 +0000 |
| commit | 7930db48e12dd3a14c1971f41f5b83527efea281 (patch) | |
| tree | d17899ba82203423320bfa8d2dea1e07b045c898 /utils | |
| parent | 95abfddfa08ab85d4f88c6f4d2e077969178f2d5 (diff) | |
| download | ComputeLibrary-7930db48e12dd3a14c1971f41f5b83527efea281.tar.gz | |
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>
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/TypePrinter.h | 6 |
1 files changed, 6 insertions, 0 deletions
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!"); } |