From 0bd4ec89d52cc1fd36e92dff2fb496b3550ee7f5 Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Mon, 1 May 2023 18:36:43 +0000
Subject: Refactor ref_model rank checking and add level check to argmax

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: Iad035b31d5e5e83040068e6311501490765bfff7
---
 verif/generator/tosa_test_gen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'verif/generator')

diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index 65bdeb7..c8c22c2 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -2614,7 +2614,7 @@ class TosaTestGen:
         "argmax": {
             "op": Op.ARGMAX,
             "operands": (1, 0),
-            "rank": (1, 4),
+            "rank": (1, 6),
             "build_fcn": (
                 build_argmax,
                 TosaTensorGen.tgBasic,
-- 
cgit v1.2.1