Home
last modified time | relevance | path

Searched defs:InternalMatch (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/passes/utils/
H A Dmatcher_utils.py36 class InternalMatch: class
166 def _remove_overlapping_matches(self, matches: List[InternalMatch]) -> List[InternalMatch]:
184 def _match_literals(self, pn: Any, gn: Any, match: InternalMatch) -> bool:
203 def _match_nodes(self, pn: Node, gn: Node, match: InternalMatch) -> bool:
/aosp_15_r20/external/executorch/backends/xnnpack/_passes/
H A Dconvert_to_sdpa.py25 def get_scale(self, match: InternalMatch) -> Optional[float]:
54 def assert_2d_mask(self, match: InternalMatch) -> None:
66 match: InternalMatch,
H A Dconvert_to_upsample_bilinear2d.py22 internal_match: InternalMatch,
/aosp_15_r20/external/pytorch/torch/fx/
H A Dsubgraph_rewriter.py211 match_filters: Optional[List[Callable[["InternalMatch", Graph, Graph], bool]]] = None,
231 match_filters: Optional[List[Callable[["InternalMatch", Graph, Graph], bool]]] = None,
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/
H A Dqat_utils.py340 match: "InternalMatch",
355 match: "InternalMatch",