Skip to content

Commit 4ef81a0

Browse files
committed
correction nn_ops_converters
1 parent 108bf26 commit 4ef81a0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

py/torch_tensorrt/fx/converters/nn_ops_converters.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,13 @@ def relu(network, submod, args, kwargs, layer_name):
2424
)
2525

2626

27+
@tensorrt_converter(torch.nn.functional.tanh)
2728
@tensorrt_converter(torch.nn.modules.activation.Tanh)
2829
def tanh(network, submod, args, kwargs, layer_name):
2930
# args/kwargs should have already been normalized to kwargs
3031
assert len(args) == 0
3132

32-
activation.sigmoid(
33+
return activation.tanh(
3334
network=network,
3435
target="torch.nn.modules.activation.Tanh",
3536
source_ir=SourceIR.NN,

0 commit comments

Comments
 (0)