diff options
-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 8c18e67..3014c81 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -4732,7 +4732,7 @@ class OutputShaper: @staticmethod def dimOp(ser, rng, a, axis, error_name=None): - output_shape = [1] + output_shape = [] if error_name == ErrorIf.WrongOutputType: all_dtypes = [ |