We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent eb59a26 commit e878a8cCopy full SHA for e878a8c
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