diff options
Diffstat (limited to 'verif/tosa_test_gen.py')
-rw-r--r-- | verif/tosa_test_gen.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index 0e57a7b..24d7b7b 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -683,7 +683,7 @@ class TosaArgGen: # Randomly generate legal output dimensions and shift # and then compute the stride and offset based on them - output_dims = [ testGen.randInt(), testGen.randInt() ] + output_dims = [ testGen.randInt(1), testGen.randInt(1) ] in_center_h = (ifm_shape[1] - 1) / 2.0 in_center_w = (ifm_shape[2] - 1) / 2.0 out_center_h = (output_dims[0] - 1) / 2.0 |