From f28bd0a0946e339a4bbd50f9490d986ab8af022e Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Tue, 21 Mar 2023 18:25:00 +0000 Subject: Update Argmax's rank maximum to MAX_RANK Signed-off-by: Jerry Ge Change-Id: Idc68dfee54f1855782e57a8855b4e4120e25e74d --- chapters/tensor_ops.adoc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'chapters') 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 { -- cgit v1.2.1