Skip to content

Commit 941b04a

Browse files
committed
fix bugs
1 parent 31435e2 commit 941b04a

File tree

1 file changed

+1
-3
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/slice

1 file changed

+1
-3
lines changed

py/torch_tensorrt/dynamo/conversion/impl/slice/ops.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import math
22
from typing import Optional
33

4+
import numpy as np
45
import tensorrt as trt
5-
import torch
66
from torch.fx.node import Target
77
from torch_tensorrt.dynamo._SourceIR import SourceIR
88
from torch_tensorrt.dynamo.conversion import impl
@@ -13,11 +13,9 @@
1313
)
1414
from torch_tensorrt.dynamo.conversion.impl.slice.base import slice
1515
from torch_tensorrt.fx.converters.converter_utils import (
16-
Frameworks,
1716
has_dynamic_shape,
1817
prepend_ones,
1918
set_layer_name,
20-
unified_dtype_converter,
2119
)
2220
from torch_tensorrt.fx.types import Shape, TRTTensor
2321

0 commit comments

Comments
 (0)