docs

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

onnxscript/optimizer/_constant_folding.py

@@ -916,7 +916,8 @@ def merge_dims(dim1, dim2):
     return ir.Shape([merge_dims(dim1, dim2) for dim1, dim2 in zip(shape1, shape2)])
 
 
-def _record_contributing_values(original_node: ir.Node, replacement: Replacement):
+def _record_contributing_values(original_node: ir.Node, replacement: Replacement) -> None:
+    """Record the set of original input values that contributed to the constant-folded outputs."""
     folded_from: set[str] = set()
     for input in original_node.inputs:
         if input is None: