Fix tutorial input field name and rename PatternMatchContext to MatchContext

Co-authored-by: gramalingam <10075881+gramalingam@users.noreply.github.com>

Author: Copilot

docs/tutorial/rewriter/conditional_rewrite.md

@@ -51,13 +51,13 @@ The final graph with the applied rewrite looks as follows:
 
 ![broadcast_rewrite](examples/img/broadcast_02.png){align=center}
 
-# Using PatternMatchContext for Advanced Condition Checking
+# Using MatchContext for Advanced Condition Checking
 
-The `context` parameter passed to condition functions is an instance of {py:class}`onnxscript.rewriter.PatternMatchContext`, which provides access to additional information about the pattern match that can be useful for sophisticated condition checking.
+The `context` parameter passed to condition functions is an instance of {py:class}`onnxscript.rewriter.MatchContext`, which provides access to additional information about the pattern match that can be useful for sophisticated condition checking.
 
-## PatternMatchContext Properties
+## MatchContext Properties
 
-The PatternMatchContext provides the following read-only properties:
+The MatchContext provides the following read-only properties:
 
 - `model`: The entire ONNX model being matched
 - `graph_or_function`: The specific graph or function being matched
@@ -67,11 +67,11 @@ The PatternMatchContext provides the following read-only properties:
 
 ## Example Usage
 
-Here's an example showing how to use the PatternMatchContext to implement more sophisticated condition checking:
+Here's an example showing how to use the MatchContext to implement more sophisticated condition checking:
 
 ```python
 def advanced_condition_check(context, x, y, **_):
-    """Example condition function using PatternMatchContext."""
+    """Example condition function using MatchContext."""
 
     # Access the main node of the pattern match
     main_node = context.main_root_node
@@ -82,7 +82,7 @@ def advanced_condition_check(context, x, y, **_):
 
     # Access the broader graph context and check that x occurs as a graph-input
     model = context.model
-    if x not in model.graph.input:
+    if x not in model.graph.inputs:
         return False
 
     # You can inspect the matched nodes for advanced validation

onnxscript/rewriter/__init__.py

@@ -9,7 +9,7 @@
     "rewrite",
     "RewritePass",
     "MatchResult",
-    "PatternMatchContext",
+    "MatchContext",
     "RewriteRule",
     "RewriteRuleClassBase",
     "RewriteRuleSet",
@@ -32,7 +32,7 @@
     pattern,
     redundant_scatter_nd,
 )
-from onnxscript.rewriter._basics import MatchingTracer, MatchResult, MatchStatus, PatternMatchContext
+from onnxscript.rewriter._basics import MatchingTracer, MatchResult, MatchStatus, MatchContext
 from onnxscript.rewriter._rewrite_rule import (
     RewriterContext,
     RewriteRule,

onnxscript/rewriter/_basics.py

@@ -340,7 +340,7 @@ def print(self):
         print(separator)
 
 
-class PatternMatchContext:
+class MatchContext:
     """A read-only context containing information about a pattern match.
     
     This class captures information about the context describing a match to a given pattern,

onnxscript/rewriter/_rewrite_rule.py

@@ -131,7 +131,7 @@ def match(
             remove_nodes=check_nodes_are_removable,
         )
         if match:
-            context = _basics.PatternMatchContext(model, graph_or_function, node, match)
+            context = _basics.MatchContext(model, graph_or_function, node, match)
             for var in self._target_pattern.inputs:
                 if var.name is not None:
                     if var.name not in match.bindings:

…t/rewriter/pattern_match_context_test.py onnxscript/rewriter/match_context_test.pyonnxscript/rewriter/pattern_match_context_test.py renamed to onnxscript/rewriter/match_context_test.py onnxscript/rewriter/pattern_match_context_test.py renamed to onnxscript/rewriter/match_context_test.py

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
-"""Test for PatternMatchContext functionality."""
+"""Test for MatchContext functionality."""
 
 import unittest
 
@@ -10,9 +10,9 @@
 from onnxscript.rewriter import pattern
 
 
-class PatternMatchContextTest(unittest.TestCase):
+class MatchContextTest(unittest.TestCase):
     def test_context_usage_in_condition_function(self):
-        """Test that PatternMatchContext can be meaningfully used in condition functions."""
+        """Test that MatchContext can be meaningfully used in condition functions."""
 
         model_proto = onnx.parser.parse_model(
             """