File size: 1,053 Bytes
6a22ec9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
from __future__ import annotations
from onnxscript.ir import _tape
from onnxscript.rewriter._basics import MatchingTracer, MatchResult, MatchStatus
from onnxscript.rewriter._matcher import PatternMatcher, SimplePatternMatcher
from onnxscript.rewriter._pattern_ir import (
ANY_VALUE,
AttrVar,
Constant,
OpsetPatternBuilder,
OrValue,
Var,
pattern_builder,
torch_module_op,
)
from onnxscript.rewriter._rewrite_rule import (
Pattern,
PatternBase,
RewriteRule,
RewriteRuleClassBase,
RewriteRuleSet,
)
RewriterContext = _tape.Builder
__all__ = [
"ANY_VALUE",
"AttrVar",
"OrValue",
"Constant",
"OpsetPatternBuilder",
"pattern_builder",
"PatternBase",
"Pattern",
"RewriteRule",
"RewriteRuleClassBase",
"RewriteRuleSet",
"RewriterContext",
"MatchingTracer",
"MatchResult",
"MatchStatus",
"PatternMatcher",
"SimplePatternMatcher",
"torch_module_op",
"Var",
]
|