adding grid_sampler_2d and grid_sampler_3d

apbose · apbose · commit cb6168f7705e · 2023-11-09T15:35:59.000-08:00
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -331,6 +331,8 @@ def aten_ops_fmod(
 
 
 @dynamo_tensorrt_converter(torch.ops.aten.grid_sampler)  # type: ignore[misc]
+@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_2d)  # type: ignore[misc]
+@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_3d)  # type: ignore[misc]
 @enforce_tensor_types(
     {
         0: (TRTTensor,),

Original file line number	Diff line number	Diff line change
`@@ -331,6 +331,8 @@ def aten_ops_fmod(`
`331`	`331`
`332`	`332`
`333`	`333`	`@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler) # type: ignore[misc]`
	`334`	`+@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_2d) # type: ignore[misc]`
	`335`	`+@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_3d) # type: ignore[misc]`
`334`	`336`	`@enforce_tensor_types(`
`335`	`337`	`{`
`336`	`338`	`0: (TRTTensor,),`