Searched defs:PatternExpr (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | pattern_matcher.py | 165 pattern: PatternExpr,
315 outputs: List[Optional[PatternExpr]],
316 pattern_to_node: Optional[Dict[PatternExpr, torch.fx.Node]] = None,
325 def match(self, pattern: PatternExpr, node: NodeOrConstant) -> MatchResult:
345 class PatternExpr(ABC): class
760 def __init__(self, pattern: PatternExpr, partial: bool = False) -> None:
805 def __init__(self, outputs: Sequence[Optional[PatternExpr]]) -> None:
845 self, pattern: PatternExpr, ctx: MatchContext
927 def run(obj: PatternExpr, output_name: str = "output") -> str:
1214 search_fn_pattern: Union[PatternExpr, None] = None,
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | micro_pipeline_tp.py | 221 def reduce_scatter_template(inp: PatternExpr):
|