diff options
author | Jerry Ge <jerry.ge@arm.com> | 2023-03-21 18:25:00 +0000 |
---|---|---|
committer | Jerry Ge <jerry.ge@arm.com> | 2023-03-21 19:25:21 +0000 |
commit | f28bd0a0946e339a4bbd50f9490d986ab8af022e (patch) | |
tree | fb016bcfd036c992dde7bd82270f4ec3abc4d6e0 | |
parent | 6577bcd5e7ba3a053ba8fcb7a41df5fae6c9651b (diff) | |
download | specification-f28bd0a0946e339a4bbd50f9490d986ab8af022e.tar.gz |
Update Argmax's rank maximum to MAX_RANK
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Idc68dfee54f1855782e57a8855b4e4120e25e74d
-rw-r--r-- | chapters/tensor_ops.adoc | 2 | ||||
-rw-r--r-- | 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 { @@ -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> |