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 /arm_compute/core/Types.h | |
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 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 8 |
1 files changed, 5 insertions, 3 deletions
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 */ |