Searched defs:MatchContext (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | pattern_matcher.py | 164 ctx: MatchContext,
303 class MatchContext: class
351 def _match(self, node: torch.fx.Node, ctx: MatchContext) -> MatchResult:
367 self, ctx: MatchContext, searched: Set[torch.fx.Node]
387 def _match(self, node: NodeOrConstant, ctx: MatchContext) -> MatchResult:
396 def _match(self, node: NodeOrConstant, ctx: MatchContext) -> MatchResult:
418 def _match(self, node: NodeOrConstant, ctx: MatchContext) -> MatchResult:
440 def _match(self, node: NodeOrConstant, ctx: MatchContext) -> MatchResult:
505 self, ctx: MatchContext, searched: Set[torch.fx.Node]
516 def _match_users(self, node: torch.fx.Node, ctx: MatchContext) -> bool:
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | split_cat.py | 470 def _match(self, node: torch.fx.Node, ctx: MatchContext):
1135 def find_anchor_nodes(self, ctx: MatchContext, searched: Set[torch.fx.Node]):
|