We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cd7e98d commit f83615aCopy full SHA for f83615a
py/torch_tensorrt/dynamo/conversion/impl/activation/ops.py
@@ -7,14 +7,15 @@
7
from torch import Tensor
8
from torch.fx.node import Target
9
from torch_tensorrt.dynamo._SourceIR import SourceIR
10
-from torch_tensorrt.dynamo.conversion.impl.activation.base import convert_activation
11
from torch_tensorrt.fx.converters.converter_utils import (
12
get_trt_plugin,
13
mark_as_int8_layer,
14
set_layer_name,
15
)
16
from torch_tensorrt.fx.types import TRTNetwork, TRTPluginFieldCollection, TRTTensor
17
+from .base import convert_activation
18
+
19
20
def gelu(
21
network: TRTNetwork,
0 commit comments