Searched refs:PatternExpr (Results 1 – 25 of 25) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | pattern_matcher.py | 154 pattern: PatternExpr
165 pattern: PatternExpr, argument
308 outputs: List[Optional[PatternExpr]]
309 pattern_to_node: Dict[PatternExpr, Optional[torch.fx.Node]]
315 outputs: List[Optional[PatternExpr]], argument
316 pattern_to_node: Optional[Dict[PatternExpr, torch.fx.Node]] = None, argument
325 def match(self, pattern: PatternExpr, node: NodeOrConstant) -> MatchResult: argument
337 def filter_multi_user_patterns(self) -> Dict[PatternExpr, torch.fx.Node]:
345 class PatternExpr(ABC): class
381 class Arg(PatternExpr):
[all …]
|
| /external/pytorch/torch/_inductor/fx_passes/serialized_patterns/ |
| D | mm_pattern.py | 27 PatternExpr,
|
| D | bmm_pattern.py | 27 PatternExpr,
|
| D | addmm_pattern.py | 27 PatternExpr,
|
| D | _sfdp_pattern_13.py | 27 PatternExpr,
|
| D | _sfdp_pattern_5.py | 27 PatternExpr,
|
| D | _sfdp_pattern_1.py | 27 PatternExpr,
|
| D | _sfdp_pattern_2.py | 27 PatternExpr,
|
| D | _sfdp_pattern_3.py | 27 PatternExpr,
|
| D | _sfdp_pattern_4.py | 27 PatternExpr,
|
| D | _sfdp_pattern_11.py | 27 PatternExpr,
|
| D | _sfdp_pattern_6.py | 27 PatternExpr,
|
| D | _sfdp_pattern_10.py | 27 PatternExpr,
|
| D | _sfdp_pattern_14.py | 27 PatternExpr,
|
| D | _sfdp_pattern_12.py | 27 PatternExpr,
|
| D | _sfdp_pattern_9.py | 27 PatternExpr,
|
| D | _sfdp_pattern_8.py | 27 PatternExpr,
|
| D | _sfdp_pattern_7.py | 27 PatternExpr,
|
| D | _sfdp_pattern_19.py | 27 PatternExpr,
|
| D | _sfdp_pattern_15.py | 27 PatternExpr,
|
| D | _sfdp_pattern_17.py | 27 PatternExpr,
|
| D | _sfdp_pattern_18.py | 27 PatternExpr,
|
| D | _sfdp_pattern_16.py | 27 PatternExpr,
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | micro_pipeline_tp.py | 17 PatternExpr,
221 def reduce_scatter_template(inp: PatternExpr): argument
|
| D | split_cat.py | 24 PatternExpr,
1140 if isinstance(pattern, PatternExpr):
|