From f28bd0a0946e339a4bbd50f9490d986ab8af022e Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Tue, 21 Mar 2023 18:25:00 +0000
Subject: Update Argmax's rank maximum to MAX_RANK

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Idc68dfee54f1855782e57a8855b4e4120e25e74d
---
 chapters/tensor_ops.adoc | 2 +-
 tosa.xml                 | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/chapters/tensor_ops.adoc b/chapters/tensor_ops.adoc
index cb268fa..508b680 100644
--- a/chapters/tensor_ops.adoc
+++ b/chapters/tensor_ops.adoc
@@ -17,7 +17,7 @@ include::{generated}/operators/ARGMAX.adoc[]
 
 [source,c++]
 ----
-ERROR_IF(axis < 0 || axis >= rank(shape1) || rank(shape1) > 4);
+ERROR_IF(axis < 0 || axis >= rank(shape1));
 if (axis == 0) {
     left_shape = [];
 } else {
diff --git a/tosa.xml b/tosa.xml
index 7fb641d..6576db1 100644
--- a/tosa.xml
+++ b/tosa.xml
@@ -17,7 +17,8 @@
         <name>ARGMAX</name>
         <arguments>
           <argument category="input" name="input" type="in_t*" shape="shape1">
-            <description>Input tensor with rank from 1 to 4</description>
+            <description>Input tensor</description>
+            <levellimit value="rank(shape1)" limit="MAX_RANK"/>
           </argument>
           <argument category="attribute" name="axis" type="int32_t" shape="-">
             <description>Axis in range from 0 to rank(shape1)-1</description>
-- 
cgit v1.2.1