Skip to content

Commit 9922bae

Browse files
committed
docs
Signed-off-by: Justin Chu <justinchuby@users.noreply.github.com>
1 parent a53dcd1 commit 9922bae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxscript/optimizer/_constant_folding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -902,8 +902,8 @@ class FoldConstantsPass(ir.passes.InPlacePass):
902902
output_size_limit: Maximum size of output tensors to fold.
903903
should_fold: An optional function that takes a node and returns True if
904904
the node should be considered for folding.
905-
The function should return (1) True to fold the node, (2) False to
906-
avoid folding the node, (3) None to use the default rules.
905+
The function should return True/False value to indicate if this particular
906+
node should be folded, or None to use the default folding rules.
907907
"""
908908

909909
def __init__(

0 commit comments

Comments
 (0)