diff options
| -rw-r--r-- | verif/conformance/tosa_base_profile_ops_info.json | 4 | ||||
| -rw-r--r-- | verif/generator/tosa_test_gen.py | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/verif/conformance/tosa_base_profile_ops_info.json b/verif/conformance/tosa_base_profile_ops_info.json index d05b881..2af4c47 100644 --- a/verif/conformance/tosa_base_profile_ops_info.json +++ b/verif/conformance/tosa_base_profile_ops_info.json @@ -1901,7 +1901,9 @@ "--tensor-dim-range", "1,16", "--target-rank", - "4" + "4", + "--target-rank", + "5" ], [ "--target-shape", diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index ce3f81f..56a34e5 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -3653,7 +3653,6 @@ class TosaTestGen: "rescale": { "op": Op.RESCALE, "operands": (1, 0), - "rank": (1, 4), "build_fcn": ( build_rescale, TosaTensorGen.tgBasic, @@ -3677,7 +3676,6 @@ class TosaTestGen: TosaErrorValidator.evScaleNotTrue, TosaErrorValidator.evWrongInputType, TosaErrorValidator.evWrongOutputType, - TosaErrorValidator.evWrongRank, TosaErrorValidator.evWrongInputList, TosaErrorValidator.evWrongOutputList, ), |