diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py index 32b9691f1f..c566d9de0a 100644 --- a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py +++ b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py @@ -2639,6 +2639,7 @@ def aten_ops_pad( ) +@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.default) @dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.vec) def upsample_nearest2d( ctx: ConversionContext, @@ -2660,6 +2661,7 @@ def upsample_nearest2d( ) +@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.default) @dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.vec) def upsample_bilinear2d( ctx: ConversionContext,