We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 31435e2 commit 941b04aCopy full SHA for 941b04a
py/torch_tensorrt/dynamo/conversion/impl/slice/ops.py
@@ -1,8 +1,8 @@
1
import math
2
from typing import Optional
3
4
+import numpy as np
5
import tensorrt as trt
-import torch
6
from torch.fx.node import Target
7
from torch_tensorrt.dynamo._SourceIR import SourceIR
8
from torch_tensorrt.dynamo.conversion import impl
@@ -13,11 +13,9 @@
13
)
14
from torch_tensorrt.dynamo.conversion.impl.slice.base import slice
15
from torch_tensorrt.fx.converters.converter_utils import (
16
- Frameworks,
17
has_dynamic_shape,
18
prepend_ones,
19
set_layer_name,
20
- unified_dtype_converter,
21
22
from torch_tensorrt.fx.types import Shape, TRTTensor
23
0 commit comments