diff options
-rw-r--r-- | verif/tosa_test_gen.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index b3731ce..cf9e06a 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -348,7 +348,7 @@ class TosaArgGen: shape = shapeList[0] for a in range(0, len(shape)): - axes.append(("axis_{}".format(a), [a])) + axes.append(("axis{}".format(a), [a])) return axes @staticmethod @@ -575,7 +575,7 @@ class TosaArgGen: arg_list.append(("perm{}_shift{}".format(p, shift), [shift])) else: - arg_list.append(("shift0", [0])) + arg_list.append(("perm0_shift0", [0])) return arg_list |