diff options
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 2 |
1 files changed, 1 insertions, 1 deletions
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, |