Home
last modified time | relevance | path

Searched defs:NSNodeTargetType (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/ns/
H A D_numeric_suite_fx.py335 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
387 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
480 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
536 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
682 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
712 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
/aosp_15_r20/external/pytorch/torch/ao/ns/fx/
H A Dgraph_matcher.py40 non_matchable_functions: Set[NSNodeTargetType],
41 non_matchable_modules: Set[NSNodeTargetType],
42 non_matchable_methods: Set[NSNodeTargetType],
184 type_a_related_to_b: Set[Tuple[NSNodeTargetType, NSNodeTargetType]], argument
253 base_name_to_sets_of_related_ops: Dict[str, Set[NSNodeTargetType]],
316 base_name_to_sets_of_related_ops: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
H A Dmappings.py458 base_name_to_sets_of_related_ops: Dict[str, Set[NSNodeTargetType]],
459 op: NSNodeTargetType,
468 base_name_to_sets_of_related_ops: Dict[str, Set[NSNodeTargetType]],
469 op: NSNodeTargetType,
470 related_op: Optional[NSNodeTargetType],
H A Dutils.py42 node_type_to_io_type_map: Dict[str, Set[NSNodeTargetType]],
164 node_type_to_io_type_map: Dict[str, Set[NSNodeTargetType]],
H A Dgraph_passes.py244 node_type_to_io_type_map: Dict[str, Set[NSNodeTargetType]],
716 node_type_to_io_type_map: Optional[Dict[str, Set[NSNodeTargetType]]] = None,
H A Dns_types.py65 NSNodeTargetType = Union[Callable, str] variable
H A Dpattern_utils.py20 base_name_to_sets_of_related_ops: Dict[str, Set[NSNodeTargetType]],