Skip to content

Commit b880985

Browse files
apbosegs-olive
authored andcommitted
correcting nn_ops_converter.py
1 parent 733c1ba commit b880985

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def hardtanh(network, submod, args, kwargs, layer_name):
3030
# args/kwargs should have already been normalized to kwargs
3131
assert len(args) == 0
3232

33-
activation.hardtanh(
33+
return activation.hardtanh(
3434
network=network,
3535
target="torch.nn.modules.activation.Hardtanh",
3636
source_ir=SourceIR.NN,

0 commit comments

Comments
 (0)