Change torch_tensorrt.fx to torch_tensorrt.dynamo

pytorch · Jan 2, 2025 · 32caa07 · 32caa07
1 parent d336d40
commit 32caa07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/torch_tensorrt/dynamo/conversion/impl/grid.py b/py/torch_tensorrt/dynamo/conversion/impl/grid.py
@@ -4,8 +4,8 @@
 from torch.fx.node import Target
 from torch_tensorrt.dynamo._SourceIR import SourceIR
 from torch_tensorrt.dynamo.conversion._ConversionContext import ConversionContext
-from torch_tensorrt.fx.converters.converter_utils import set_layer_name
-from torch_tensorrt.fx.types import TRTTensor
+from torch_tensorrt.dynamo.conversion.converter_utils import set_layer_name
+from torch_tensorrt.dynamo.types import TRTTensor
 
 # bilinear, nearest, bicubic
 GridSamplerInterpolationMode = {