Home
last modified time | relevance | path

Searched refs:matched_node_pattern (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dmatch_utils.py123 last_node, match_map, node_pattern, matched_node_pattern, pattern, match_value argument
128 matched_node_pattern,
137 last_node, match_map, n, matched_node_pattern, pattern, match_value
141 def record_match(pattern, node, last_node, matched_node_pattern, match_map): argument
146 record_match(s, node, last_node, matched_node_pattern, match_map)
159 matched_node_pattern.append(tuple(current_node_pattern))
161 matched_node_pattern.extend(list(current_node_pattern))
163 matched_node_pattern.append(current_node_pattern[0])
165 matched_node_pattern.append(node)
172 matched_node_pattern: List[Node] = []
[all …]
H A Dfuse.py94 matched_node_pattern,
108 root_node = root_node_getter(matched_node_pattern) # type: ignore[index]
114 extra_inputs = extra_inputs_getter(matched_node_pattern)
123 matched_node_pattern, # type: ignore[arg-type]
150 def apply_match(pattern, node, match, matched_node_pattern, node_to_subpattern): argument
159 matched_node_pattern.append(tuple(current_node_pattern))
163 matched_node_pattern.append(node)
172 matched_node_pattern,
180 matched_node_pattern: List[Node] = []
186 matched_node_pattern,
H A Dprepare.py390 matched_node_pattern: Optional[List[Node]],
400 assert matched_node_pattern is not None and len(matched_node_pattern) >= 1
408 root_node = root_node_getter(matched_node_pattern)
410 output_node = matched_node_pattern[0]
467 def _add_matched_node_name_to_set(matched_node_pattern: NodePattern, s: Set[str]):
468 if isinstance(matched_node_pattern, Node):
469 s.add(matched_node_pattern.name)
470 elif isinstance(matched_node_pattern, (list, tuple)):
471 for maybe_node in matched_node_pattern:
506 matched_node_pattern: NodePattern,
[all …]
H A Dfuse_handler.py43 matched_node_pattern: NodePattern,
62 matched_node_pattern: NodePattern,
98 matched_modules = get_modules(matched_node_pattern)
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/pruner/
H A Dmatch_utils.py42 matched_node_pattern: List[Node],
51 return matched_node_pattern + [node]
56 return matched_node_pattern + [node]
60 modules, tuple(rest), user, matched_node_pattern + [node]