Searched refs:node_copy (Results 1 – 25 of 41) sorted by relevance
12
| /external/executorch/backends/transforms/ |
| D | duplicate_dynamic_quant_chain.py | 82 new_get_item_node_1 = gm.graph.node_copy(get_item_node_1)
83 new_get_item_node_2 = gm.graph.node_copy(get_item_node_2)
84 new_chose_qparams_node = gm.graph.node_copy(chose_qparams_node)
113 new_node = gm.graph.node_copy(node)
135 new_node = gm.graph.node_copy(dq_node)
|
| D | decompose_sdpa.py | 81 subgraph_node = graph.node_copy(
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | graph_passes.py | 173 env[node.name] = new_graph.node_copy(node, load_arg)
196 env[node.name] = new_graph.node_copy(node, load_arg)
810 env_c[node_b.name] = graph_c.node_copy(node_b, load_arg)
839 env_c[node_b.name] = graph_c.node_copy(node_b, load_arg)
858 env_c[node_b.name] = graph_c.node_copy(node_b, load_arg)
872 env_c[node_b.name] = graph_c.node_copy(node_b, load_arg)
933 env_c[node_b.name] = graph_c.node_copy(node_b, load_arg)
1128 env_c[node_b.name] = graph_c.node_copy(node_b, load_arg)
|
| /external/pytorch/torch/fx/passes/dialect/common/ |
| D | cse_pass.py | 75 new_node = new_graph.node_copy(n, lambda x: env[x])
106 new_node = new_graph.node_copy(n, lambda x: env[x])
|
| /external/pytorch/torch/_functorch/ |
| D | compile_utils.py | 99 new_node = new_graph.node_copy(n, lambda x: env[x])
147 new_node = new_graph.node_copy(n, lambda x: env[x])
|
| D | fx_minifier.py | 294 new_node = new_graph.node_copy(node, lambda x: env[x])
382 new_node = new_graph.node_copy(node, lambda x: env[x])
396 new_node = new_graph.node_copy(node, lambda x: env[x])
|
| D | partitioners.py | 201 env[node] = new_graph.node_copy(node, lambda x: env[x])
203 env[node] = new_graph.node_copy(node, lambda x: env[x])
569 env[node] = new_graph.node_copy(node, lambda x: env[x])
590 env[node] = new_graph.node_copy(node, lambda x: env[x])
|
| /external/executorch/backends/qualcomm/_passes/ |
| D | decompose_einsum.py | 56 remap[decomposed_node] = graph.node_copy(
|
| /external/pytorch/test/fx/ |
| D | test_fx_node_hook.py | 52 node_a_copy = graph.node_copy(node_a)
|
| D | quantization.py | 101 return quantizer.quantized_graph.node_copy(node, load_arg)
331 r = env[node.name] = self.quantized_graph.node_copy(
340 env[node.name] = self.quantized_graph.node_copy(
|
| /external/executorch/backends/mediatek/_passes/ |
| D | decompose_scaled_dot_product_attention.py | 64 subgraph_node = graph.node_copy(
|
| /external/pytorch/test/expect/ |
| D | TestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect | 3 …et_attr', 'graph_copy', 'inserting_after', 'inserting_before', 'lint', 'node_copy', 'nodes', 'on_g…
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | duplicate_dq_pass.py | 39 new_node = gm.graph.node_copy(dq_node)
|
| /external/executorch/exir/ |
| D | lowered_backend_module.py | 407 new_node = new_graph.node_copy(param_node, lambda x: node_map[x])
410 new_node = new_graph.node_copy(buffer_node, lambda x: node_map[x])
413 new_node = new_graph.node_copy(input_node, lambda x: node_map[x])
421 new_node = new_graph.node_copy(node, lambda x: node_map[x])
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | graph_partition.cc | 47 Node* node_copy = nullptr; member
103 node_info.node_copy = graph->CopyNode(node); in PrepareSubgraphForFunctionConversion()
251 NodeBuilder::NodeOut(input_node.second.node_copy, 0); in BuildPartitionedCallOp()
|
| /external/pytorch/torch/fx/passes/ |
| D | graph_manipulation.py | 40 val_map[node] = new_graph.node_copy(node, lambda n: val_map[n])
|
| D | tools_common.py | 289 env[cur] = new_graph.node_copy(cur, lambda x: env[x])
|
| D | split_utils.py | 231 n = comp.graph.node_copy(node, remap_func)
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | fuse_handler.py | 117 node = fused_graph.node_copy(root_node, load_arg)
|
| D | fuse.py | 129 env[node.name] = fused_graph.node_copy(node, load_arg)
|
| /external/pytorch/torch/fx/passes/utils/ |
| D | fuser_utils.py | 154 new_node = subgraph.node_copy(node, remap_inputs)
|
| /external/pytorch/torch/fx/experimental/ |
| D | const_fold.py | 117 new_node = gm.graph.node_copy(inline_node, replacement_fn)
|
| /external/executorch/backends/cadence/aot/ |
| D | reorder_ops.py | 599 zero_point_copy = graph.node_copy(zero_point)
600 scale_copy = graph.node_copy(scale)
|
| /external/pytorch/torch/_export/passes/ |
| D | constant_folding.py | 287 new_node = new_graph.node_copy(node, lambda x: node_remapping[x])
|
| /external/pytorch/torch/_inductor/ |
| D | constant_folding.py | 335 new_node = new_graph.node_copy(node, lambda x: node_remapping[x])
|
12