Home
last modified time | relevance | path

Searched refs:src_node (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc51 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input() argument
52 inputs_.emplace_back(src_node, src_index); in Input()
54 if (GetOutputType(src_node, src_index, &dt)) { in Input()
55 def_builder_.Input(src_node->name(), src_index, dt); in Input()
85 NodeBuilder& NodeBuilder::ControlInput(Node* src_node) { in ControlInput() argument
86 control_inputs_.emplace_back(src_node); in ControlInput()
87 def_builder_.ControlInput(src_node->name()); in ControlInput()
94 for (const Node* src_node : src_nodes) { in ControlInputs() local
95 def_builder_.ControlInput(src_node->name()); in ControlInputs()
Dcontrol_flow.cc94 const Node* src_node = g->source_node(); in BuildControlFlowInfo() local
95 ControlFlowInfo& src_info = (*info)[src_node->id()]; in BuildControlFlowInfo()
96 src_info.frame = src_node; in BuildControlFlowInfo()
97 src_info.parent_frame = src_node; in BuildControlFlowInfo()
101 ready.push_back(src_node); in BuildControlFlowInfo()
Dcollective_order.cc100 Node* src_node = collective_nodes[src_idx]; in CreateControlDependencies() local
102 VLOG(1) << "Adding control dependency from node " << src_node->name() in CreateControlDependencies()
105 (*dependency_edges)[src_node].insert(dst_node); in CreateControlDependencies()
106 auto& src_paths = all_paths[src_node]; in CreateControlDependencies()
153 Node* src_node = pair.first; in InsertControlDependencies() local
155 graph->AddControlEdge(src_node, dst_node); in InsertControlDependencies()
Dnode_builder.h91 NodeBuilder& Input(Node* src_node, int src_index = 0);
98 NodeBuilder& ControlInput(Node* src_node);
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc96 NodeDefBuilder& NodeDefBuilder::Input(StringPiece src_node, int src_index, in Input() argument
99 if (arg != nullptr) SingleInput(arg, src_node, src_index, dt); in Input()
116 StringPiece src_node, int src_index, in SingleInput() argument
118 AddInput(src_node, src_index); in SingleInput()
173 void NodeDefBuilder::AddInput(StringPiece src_node, int src_index) { in AddInput() argument
174 if (src_node.empty()) { in AddInput()
176 } else if (src_node[0] == '^') { in AddInput()
178 strings::StrCat("Non-control input starting with ^: ", src_node)); in AddInput()
180 node_def_.add_input(strings::StrCat(src_node, ":", src_index)); in AddInput()
182 node_def_.add_input(string(src_node)); in AddInput()
[all …]
Dnode_def_builder.h80 NodeDefBuilder& Input(StringPiece src_node, int src_index, DataType dt);
90 NodeDefBuilder& ControlInput(StringPiece src_node);
159 void SingleInput(const OpDef::ArgDef* input_arg, StringPiece src_node,
165 void AddInput(StringPiece src_node, int src_index);
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc133 for (Node* src_node : graph->nodes()) { in InsertNodes()
136 for (const Edge* edge : src_node->out_edges()) { in InsertNodes()
146 for (int src_output_slot = 0; src_output_slot < src_node->num_outputs(); in InsertNodes()
149 strings::StrCat(src_node->name(), ":", src_output_slot); in InsertNodes()
170 const DataType src_dt = src_node->output_type(src_output_slot); in InsertNodes()
172 TF_RETURN_IF_ERROR(MemoryTypeForOutput(device_type, graph, src_node, in InsertNodes()
185 src_node->name(), src_output_slot, src_dt, tensor_name, in InsertNodes()
195 graph->AddEdge(src_node, src_output_slot, copy_node, 0); in InsertNodes()
225 const bool is_ref = IsRefType(src_node->output_type(src_output_slot)); in InsertNodes()
239 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc471 Node* src_node = edge->src(); in RecordArg() local
476 OutputTensor(src_node, src_slot), args_by_src_.size()); in RecordArg()
481 absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp, in RecordArg()
482 NodeDebugInfo(src_node->def())); in RecordArg()
492 src_arg_pairs->push_back({src_node, arg}); in RecordArg()
506 Node* src_node = edge->src(); in RecordControlResult() local
507 Node* src_image = node_images.at(src_node); in RecordControlResult()
515 Node* src_node = edge->src(); in RecordResult() local
516 Node* src_image = node_images.at(src_node); in RecordResult()
521 results_.emplace(OutputTensor(src_node, src_slot), results_.size()); in RecordResult()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc198 const NodeDef* src_node = node_map.GetNode(src); in IsNodesDirectlyConnected() local
200 EXPECT_TRUE(src_node != nullptr) << src << " node not found"; in IsNodesDirectlyConnected()
202 return src_node && dst_node && dst_node->input(position) == src_node->name(); in IsNodesDirectlyConnected()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc61 Node* src_node = edge->src(); in ReplaceReduce() local
65 int send_dev = src_node->assigned_device_name_index(); in ReplaceReduce()
67 recv_builder.Input(src_node); in ReplaceReduce()
73 .Input(src_node) in ReplaceReduce()
Dnccl_manager_test.cc211 TensorShape shape, int src_node, int src_rank, in MakeBroadcastTestCase() argument
221 if (node == src_node && local_rank == src_rank) { in MakeBroadcastTestCase()
364 const int src_node, const int src_local_rank, in RunMultiNodeBroadcastTest() argument
367 const int src_global_rank = src_node * num_ranks_per_node + src_local_rank; in RunMultiNodeBroadcastTest()
373 num_nodes, num_ranks_per_node, TensorShape({5, 6}), src_node, in RunMultiNodeBroadcastTest()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.cc1177 Node* src_node; in Convert() local
1184 src_node = iter->second.node; in Convert()
1186 if (src_node == nullptr) has_data_back_edge = true; in Convert()
1191 src_node = iter->second; in Convert()
1195 if (src_node != nullptr && src_index >= src_node->num_outputs()) { in Convert()
1199 << " which has " << src_node->num_outputs() << " outputs."; in Convert()
1201 if (src_node->type_string() == "If" || in Convert()
1202 src_node->type_string() == "StatelessIf" || in Convert()
1203 src_node->type_string() == "While" || in Convert()
1204 src_node->type_string() == "StatelessWhile") { in Convert()
[all …]
Dprocess_function_library_runtime.cc453 Node* src_node = it->src(); in PinArgsAndRets() local
454 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
456 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
457 VLOG(3) << "Considering src: " << src_node->name() in PinArgsAndRets()
461 src_node->IsIdentity()) { in PinArgsAndRets()
464 TF_RETURN_IF_ERROR(src_node->input_node(0, &input_node)); in PinArgsAndRets()
465 src_node = input_node; in PinArgsAndRets()
467 src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
468 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
469 VLOG(3) << "Considering src: " << src_node->name() in PinArgsAndRets()
[all …]
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_cluster.cc115 for (const std::unique_ptr<tensorflow::NodeDef>& src_node : in MaybeResolveClusters() local
118 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
/external/mesa3d/src/compiler/nir/
Dnir_from_ssa.c402 merge_node *src_node = get_merge_node(src->src.ssa, state); in coalesce_phi_nodes_block() local
403 if (src_node->set != dest_node->set) in coalesce_phi_nodes_block()
404 merge_merge_sets(dest_node->set, src_node->set); in coalesce_phi_nodes_block()
429 merge_node *src_node = get_merge_node(entry->src.ssa, state); in aggressive_coalesce_parallel_copy() local
432 if (src_node->set == dest_node->set) in aggressive_coalesce_parallel_copy()
435 if (!merge_sets_interfere(src_node->set, dest_node->set)) in aggressive_coalesce_parallel_copy()
436 merge_merge_sets(src_node->set, dest_node->set); in aggressive_coalesce_parallel_copy()
Dnir_lower_io_to_temporaries.c55 foreach_two_lists(dest_node, dest_vars, src_node, src_vars) { in emit_copies()
57 nir_variable *src = exec_node_data(nir_variable, src_node, node); in emit_copies()
Dnir_opt_find_array_copies.c488 struct match_node *src_node = in handle_write() local
493 if (src_node->last_overwritten <= dst_node->first_src_read) { in handle_write()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc775 const Node* src_node = in_edge->src(); in ClusterizeNodes() local
777 node_names.count(src_node->name()) <= 0 && !src_node->IsSource(); in ClusterizeNodes()
780 strings::StrCat(src_node->name(), ":", in_edge->src_output()); in ClusterizeNodes()
781 CHECK_EQ(1, src_node->num_outputs()) in ClusterizeNodes()
783 << src_node->name(); in ClusterizeNodes()
914 const Node* src_node = edge->src(); in BuildClusterByBorder() local
915 CHECK_NOTNULL(src_node); in BuildClusterByBorder()
920 if (tid.first == src_node->name() && tid.second == src_port) { in BuildClusterByBorder()
922 border_input_nodes.insert(src_node); in BuildClusterByBorder()
925 if (visited.insert(src_node).second) { in BuildClusterByBorder()
[all …]
/external/e2fsprogs/lib/ext2fs/
Dblkmap64_rb.c244 struct rb_node *dest_node, *src_node, *dest_last, **n; in rb_copy_bmap() local
256 src_node = ext2fs_rb_first(&src_bp->root); in rb_copy_bmap()
257 while (src_node) { in rb_copy_bmap()
258 src_ext = node_to_extent(src_node); in rb_copy_bmap()
278 src_node = ext2fs_rb_next(src_node); in rb_copy_bmap()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc397 utils::MutableNodeView* src_node, in UpdateFanoutEdgesWithOp() argument
400 const auto* output_shape_attr = src_node->GetAttr(kAttrOutputShape); in UpdateFanoutEdgesWithOp()
409 " in", src_node->GetName()), in UpdateFanoutEdgesWithOp()
413 src_node, kAttrOutputShape, shape_attr_copy); in UpdateFanoutEdgesWithOp()
416 const bool is_in_frame = context->frames.IsInFrame(*src_node->node()); in UpdateFanoutEdgesWithOp()
421 const auto& fanouts_src_port = src_node->GetRegularFanout(src_port); in UpdateFanoutEdgesWithOp()
430 GetFanoutNameFormat(src_node->GetName(), src_port, in UpdateFanoutEdgesWithOp()
435 src_node, fanout.node_view())); in UpdateFanoutEdgesWithOp()
490 utils::MutableNodeView* src_node, utils::MutableNodeView* dst_node) { in UpdateEdge() argument
491 DCHECK(src_node != nullptr); in UpdateEdge()
[all …]
Dgeneric_layout_optimizer_transposer.h139 utils::MutableNodeView* src_node,
178 utils::MutableNodeView* src_node,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc229 for (Node* src_node : input_dependencies) { in AddHostDisconnectNode()
230 graph->AddControlEdge(src_node, host_disconnect_node); in AddHostDisconnectNode()
/external/mesa3d/src/freedreno/ir3/
Dir3_nir_lower_tess.c863 foreach_two_lists(dest_node, &state->emit_outputs, src_node, &state->old_outputs) { in lower_gs_block()
865 nir_variable *src = exec_node_data(nir_variable, src_node, node); in lower_gs_block()
983 foreach_two_lists(dest_node, &state.new_outputs, src_node, &state.emit_outputs) { in ir3_nir_lower_gs()
985 nir_variable *src = exec_node_data(nir_variable, src_node, node); in ir3_nir_lower_gs()
/external/vulkan-validation-layers/layers/
Ddescriptor_sets.cpp1716 auto src_node = GetSetNode(src_set); in ValidateUpdateDescriptorSets() local
1719 assert(src_node); in ValidateUpdateDescriptorSets()
1723 … if (!ValidateCopyUpdate(&p_cds[i], dst_node, src_node, func_name, &error_code, &error_str)) { in ValidateUpdateDescriptorSets()
1754 auto src_node = dev_data->GetSetNode(src_set); in PerformUpdateDescriptorSets() local
1756 if (src_node && dst_node) { in PerformUpdateDescriptorSets()
1757 dst_node->PerformCopyUpdate(&p_cds[i], src_node); in PerformUpdateDescriptorSets()
/external/angle/third_party/abseil-cpp/absl/container/internal/
Dbtree.h905 btree_node *src_node, allocator_type *alloc) {
906 transfer(slot(dest_i), src_node->slot(src_i), alloc);
912 const size_type src_i, btree_node *src_node,
914 for (slot_type *src = src_node->slot(src_i), *end = src + n,
924 const size_type src_i, btree_node *src_node,
926 for (slot_type *src = src_node->slot(src_i + n - 1), *end = src - n,

12