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]¶