diff options
Diffstat (limited to 'verif/frameworks/arg_gen.py')
| -rw-r--r-- | verif/frameworks/arg_gen.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/verif/frameworks/arg_gen.py b/verif/frameworks/arg_gen.py index 18e8976..8604f0b 100644 --- a/verif/frameworks/arg_gen.py +++ b/verif/frameworks/arg_gen.py @@ -587,6 +587,30 @@ class ArgGen: ) return arg_list + def agResize(op, shapes, rng): + args = [] + for mode in ["nearest", "bilinear"]: + for align_corners in [True, False]: + for half_pixel in [True, False]: + # If half_pixel_centers is True, align_corners must be False. + if ( + (mode == "bilinear") + and (align_corners is True) + and (half_pixel is True) + ): + continue + + for i in range(1, 4): + args.append( + [ + "_{}_align{}_half{}_scale{}".format( + mode, int(align_corners), int(half_pixel), i + ), + [mode, align_corners, half_pixel, i], + ] + ) + return args + def agFill(op, shapes, rng): values = [] for i in range(4): |