Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc100 const std::vector<Node*>& host_configuration_nodes, in AddWaitNode() argument
106 AddNodeAttr("N", static_cast<int32>(host_configuration_nodes.size()), in AddWaitNode()
109 if (!host_configuration_nodes.empty()) { in AddWaitNode()
110 MergeDebugInfo(NodeDebugInfo(host_configuration_nodes[0]->def()), in AddWaitNode()
121 for (int i = 0; i < host_configuration_nodes.size(); ++i) { in AddWaitNode()
122 graph->AddEdge(host_configuration_nodes[i], 0, *wait_node, i); in AddWaitNode()
306 std::vector<Node*> host_configuration_nodes; in Run() local
312 host_configuration_nodes.push_back(host_configuration_node); in Run()
320 host_configuration_nodes, graph, in Run()