From 32caa07852bc224fabc81b180f76e62f17b036ba Mon Sep 17 00:00:00 2001 From: HolyWu Date: Fri, 3 Jan 2025 07:41:30 +0800 Subject: [PATCH] Change torch_tensorrt.fx to torch_tensorrt.dynamo --- py/torch_tensorrt/dynamo/conversion/impl/grid.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/torch_tensorrt/dynamo/conversion/impl/grid.py b/py/torch_tensorrt/dynamo/conversion/impl/grid.py index b4f6101ef4..302d286237 100644 --- 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 = {