Home
last modified time | relevance | path

Searched refs:control_node (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc96 Status MergeReadVariableOps(Node* handle_op, Node* control_node, in MergeReadVariableOps() argument
116 if (control_node) graph->AddControlEdge(control_node, node); in MergeReadVariableOps()
177 Node* control_node = nullptr; in Run() local
183 control_node = e->src(); in Run()
189 read_var_ops[std::pair<Node*, Node*>(var_handle_op, control_node)] in Run()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc81 const NodeDef& control_node) { in CanDedupControlWithRegularInput() argument
82 return !IsIdentityConsumingSwitch(graph, control_node); in CanDedupControlWithRegularInput()
91 NodeDef* control_node = graph.GetNode(control_node_name); in CanDedupControlWithRegularInput() local
92 if (control_node == nullptr) { in CanDedupControlWithRegularInput()
95 return CanDedupControlWithRegularInput(graph, *control_node); in CanDedupControlWithRegularInput()
1033 NodeDef* control_node = GetControllingFaninToAdd( in AddControllingFanin() local
1038 if (control_node == nullptr) { in AddControllingFanin()
1039 control_node = GetOrCreateIdentityConsumingSwitch(fanin_port); in AddControllingFanin()
1041 AddFaninInternal(node, {control_node, Graph::kControlSlot}); in AddControllingFanin()
1442 NodeDef* control_node = in UpdateAllRegularFaninsToControlling() local
[all …]