Skip to content

Commit 6a26856

Browse files
authored
fix missing return in sigmoid nn convert
1 parent eaa5ef0 commit 6a26856

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/fx/converters/nn_ops_converters.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def sigmoid(network, submod, args, kwargs, layer_name):
2929
# args/kwargs should have already been normalized to kwargs
3030
assert len(args) == 0
3131

32-
activation.sigmoid(
32+
return activation.sigmoid(
3333
network=network,
3434
target="torch.nn.modules.activation.Sigmoid",
3535
source_ir=SourceIR.NN,

0 commit comments

Comments
 (0)