RewriteRule

class onnxscript.rewriter.pattern.RewriteRule(target_pattern: GraphPattern | Callable, replacement_pattern: ReplacementPatternFunction | Callable, condition_function: Callable | None = None, matcher: PatternMatcher | Callable[[GraphPattern], PatternMatcher] | None = None, verbose: int = 0, name: str | None = None, remove_nodes: bool = True, graph_pre_visitor: Callable[[], None] | None = None, graph_post_visitor: Callable[[], None] | None = None, as_function: bool = False)[source]
try_rewrite(model: Model, graph_or_function: Graph | Function, node: Node, *, verbose: int | None = None, tracer: MatchingTracer | None = None) ReplacementSubgraph | None[source]

If the node matches the pattern, then replace the node with the replacement pattern.

apply_to_model(model: Model, *, commute: bool = False, verbose: int | None = None, tracer: MatchingTracer | None = None)[source]
commute() Sequence[RewriteRule][source]