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 --- arm_compute/core/Types.h | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'arm_compute/core/Types.h') diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 5ddd207100..7db2f5fddf 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -545,9 +545,11 @@ enum class NonLinearFilterFunction : unsigned /** Available reduction operations */ enum class ReductionOperation { - SUM_SQUARE, /**< Sum of squares */ - SUM, /**< Sum */ - MEAN_SUM, /**< Mean of sum */ + SUM_SQUARE, /**< Sum of squares */ + SUM, /**< Sum */ + MEAN_SUM, /**< Mean of sum */ + ARG_IDX_MAX, /**< Index of the max value */ + ARG_IDX_MIN /**< Index of the min value */ }; /** The normalization type used for the normalization layer */ -- cgit v1.2.1