Searched refs:node_to_subpattern (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | fuse.py | 96 node_to_subpattern,
99 if node_to_subpattern is not None:
100 node_subpattern = node_to_subpattern.get(node, None)
147 node_to_subpattern: Dict[Node, Any] = {}
150 def apply_match(pattern, node, match, matched_node_pattern, node_to_subpattern): argument
154 apply_match(s, node, match, current_node_pattern, node_to_subpattern)
157 subpattern, arg, match, current_node_pattern, node_to_subpattern
167 node_to_subpattern[node] = pattern
174 node_to_subpattern,
187 node_to_subpattern,
|