Fix tutorial example and add display method to PatternMatchContext

- Fix tutorial to use `x in model.graph.input` instead of checking input names
- Add display method to PatternMatchContext with in_graph_order parameter
- Fix test assertion to use model.graph[2] instead of model.graph.node[2]

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

Author: Copilot

docs/tutorial/rewriter/conditional_rewrite.md

@@ -82,8 +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
-    input_names = [input.name for input in model.graph.input]
-    if x not in input_names:
+    if x not in model.graph.input:
         return False
 
     # You can inspect the matched nodes for advanced validation

onnxscript/rewriter/_basics.py

@@ -393,6 +393,28 @@ def nodes(self) -> Sequence[ir.Node]:
         """All the nodes of the matching subgraph."""
         return self._match_result.nodes
 
+    def display(self, *, in_graph_order: bool = True) -> None:
+        """Display the nodes in the pattern match context.
+        
+        Args:
+            in_graph_order: If True, display nodes in the order they appear in the
+                graph/function. If False, display nodes in the order they appear
+                in the match result.
+        """
+        nodes = self.nodes
+        if not nodes:
+            return
+            
+        if in_graph_order:
+            # Display nodes in same order as in graph/function
+            for node in self._graph_or_function:
+                if node in nodes:
+                    node.display()
+        else:
+            # Display nodes in match order
+            for node in nodes:
+                node.display()
+
 
 class MatchingTracer:
     """A debugging helper class to trace the matching of a pattern against a graph.

onnxscript/rewriter/pattern_match_context_test.py

@@ -31,7 +31,7 @@ def condition_using_context(context, x, y):
             # Use context to check properties of the match
             self.assertIs(context.model, model)
             self.assertIs(context.graph_or_function, model.graph)
-            self.assertIs(context.main_root_node, model.graph.node[2])
+            self.assertIs(context.main_root_node, model.graph[2])
 
             # Verify that we can inspect the matched nodes
             self.assertEqual(len(context.nodes), 2)