diff options
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 94edf6f..7e4cb1d 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -902,7 +902,9 @@ class TosaArgGen: if max(shape) < 64: # padding must be less than the kernel size bigPadding = bigKernel - 1 - paddings.update({x for x in itertools.product(*([[0, bigPadding]] * 4))}) + paddings.update( + {x for x in itertools.product(*([[0, bigPadding]] * 4))} + ) # There are too many parameter combinations, so generate them sparsely, # very sparse for negative tests |