Searched refs:get_normalized_nth_input (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
| H A D | graph_passes.py | 16 get_normalized_nth_input,
37 node_to_use_for_fqn = get_normalized_nth_input(node, gm, 0)
112 new_graph.output(map_arg(get_normalized_nth_input(node, gm, 0), load_arg))
130 node_arg = get_normalized_nth_input(node, gm, node_arg_idx)
423 get_normalized_nth_input(node_a, gm_a, 0), gm_a, gm_b, graph_c
434 get_normalized_nth_input(node_a, gm_a, 0), gm_a, gm_b, graph_c
442 (arg_copy, get_normalized_nth_input(node_a, gm_a, 1)),
469 cur_node = get_normalized_nth_input(cur_node, gm_a, 0) # type: ignore[assignment]
551 cur_node = get_normalized_nth_input(cur_node, gm_a, 0) # type: ignore[assignment]
881 prev_node_b = get_normalized_nth_input(node_b, gm_b, 0)
[all …]
|
| H A D | utils.py | 62 first_arg = get_normalized_nth_input(node, gm, 0)
87 first_arg = get_normalized_nth_input(node, gm, 0)
114 prev_node = get_normalized_nth_input(node, gm, 0)
129 prev_node = get_normalized_nth_input(node, gm, 0)
138 cur_node_dtype_target = get_normalized_nth_input(node, gm, 1)
146 first_arg = get_normalized_nth_input(node, gm, 0)
170 prev_node = get_normalized_nth_input(node, gm, 0)
178 scale_node = get_normalized_nth_input(node, gm, scale_arg_idx)
179 zp_node = get_normalized_nth_input(node, gm, zp_arg_idx)
507 def get_normalized_nth_input(node: Node, gm: GraphModule, idx: int) -> Node: function
|
| H A D | n_shadows_utils.py | 12 get_normalized_nth_input,
639 prev_node = get_normalized_nth_input(first_node, mt, 0)
|