We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
align_corners=False
1 parent f43be5b commit 0a075f0Copy full SHA for 0a075f0
py/torch_tensorrt/fx/converters/acc_ops_converters.py
@@ -3630,7 +3630,7 @@ def acc_ops_interpolate(
3630
else:
3631
layer.resize_mode = trt.ResizeMode.NEAREST
3632
3633
- if align_corners != None:
+ if (align_corners is not None) and align_corners:
3634
layer.coordinate_transformation = (
3635
trt.ResizeCoordinateTransformation.ALIGN_CORNERS
3636
)
py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py
@@ -43,6 +43,14 @@ class TestInterpolateConverter(AccTestCase):
43
("bilinear"),
44
(None),
45
), # linear for 4D only
46
+ (
47
+ "4d_dim_scale_bilinear_align_corners_bool",
48
+ (2, 3, 4, 5),
49
+ (None),
50
+ (2),
51
+ ("bilinear"),
52
+ (False),
53
+ ), # linear for 4D only
54
(
55
"4d_dim_scale_align",
56
(2, 3, 4, 5),
0 commit comments