We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a53dcd1 commit 9922baeCopy full SHA for 9922bae
onnxscript/optimizer/_constant_folding.py
@@ -902,8 +902,8 @@ class FoldConstantsPass(ir.passes.InPlacePass):
902
output_size_limit: Maximum size of output tensors to fold.
903
should_fold: An optional function that takes a node and returns True if
904
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.
+ The function should return True/False value to indicate if this particular
+ node should be folded, or None to use the default folding rules.
907
"""
908
909
def __init__(
0 commit comments