We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent eaa5ef0 commit 6a26856Copy full SHA for 6a26856
py/torch_tensorrt/fx/converters/nn_ops_converters.py
@@ -29,7 +29,7 @@ def sigmoid(network, submod, args, kwargs, layer_name):
29
# args/kwargs should have already been normalized to kwargs
30
assert len(args) == 0
31
32
- activation.sigmoid(
+ return activation.sigmoid(
33
network=network,
34
target="torch.nn.modules.activation.Sigmoid",
35
source_ir=SourceIR.NN,
0 commit comments