Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc159 Node* sync_node = graph->AddNode(sync_def, &status); in AddSynchronizationNode() local
163 sync_node->set_assigned_device_name(device_name); in AddSynchronizationNode()
166 graph->AddControlEdge(node, sync_node); in AddSynchronizationNode()
169 graph->AddEdge(wait_node, 0, sync_node, 0); in AddSynchronizationNode()
174 graph->AddControlEdge(sync_node, dep.dst); in AddSynchronizationNode()
176 graph->AddEdge(sync_node, dep.src_output, dep.dst, dep.dst_input); in AddSynchronizationNode()