From fb65bcf27eb77f593abd345aed0e884a440c827f Mon Sep 17 00:00:00 2001
From: Eric Kunze <eric.kunze@arm.com>
Date: Mon, 11 Mar 2024 21:20:42 -0700
Subject: Tighten ARGMAX definition

Define behavior if multiple values are tied for the maximum.
Define behavior if NaN values are encountered.

Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: Id05a680cd11a331ba84a8e38f9a7c77a71bcbb78
---
 chapters/tensor_ops.adoc | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'chapters/tensor_ops.adoc')

diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc
index 3de5150..a36a7f1 100644
--- a/chapters/tensor_ops.adoc
+++ b/chapters/tensor_ops.adoc
@@ -12,6 +12,9 @@
 ==== ARGMAX
 
 This returns the index with the largest value across the given axis of the input tensor.
+If multiple locations have equal values, returns the first match along the search axis.
+NaN values always compare as greater than non-NaN values.
+If an entire axis consists of NaN, the last location must be returned.
 
 include::{generated}/operators/ARGMAX.adoc[]
 
-- 
cgit v1.2.1

