Home
last modified time | relevance | path

Searched refs:node_pattern (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dquantize_handler.py35 def _default_root_node_getter(node_pattern): argument
36 if node_pattern is None:
37 return node_pattern
38 while not isinstance(node_pattern, Node):
39 node_pattern = node_pattern[-1]
40 return node_pattern
49 node_pattern: NodePattern,
58 self.node_pattern = node_pattern
62 self.root_node = root_node_getter(node_pattern)
115 node_pattern: NodePattern,
[all …]
H A Dfuse.py85 def default_root_node_getter(node_pattern): argument
86 while not isinstance(node_pattern[-1], Node):
87 node_pattern = node_pattern[-1]
88 return node_pattern[-1]
H A Dmatch_utils.py123 last_node, match_map, node_pattern, matched_node_pattern, pattern, match_value argument
125 if isinstance(node_pattern, Node):
126 match_map[node_pattern.name] = (
132 elif not isinstance(node_pattern, Iterable):
135 for n in node_pattern:
H A Dprepare.py519 for node_pattern in matched_node_pattern:
521 node_pattern,
/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A D_qnnpack_pt2e.py112 def root_node_getter(node_pattern): argument
113 getitem, maxpool, index = node_pattern
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py6256 def root_node_getter(node_pattern): argument
6257 reshape, transpose, shape = node_pattern