Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc125 std::vector<std::forward_list<NodeBuilder::NodeOut>> out_nodes_map; in ReplaceBroadcast() local
130 if (out_nodes_map.size() <= dst_dev) { in ReplaceBroadcast()
131 out_nodes_map.resize(dst_dev + 1); in ReplaceBroadcast()
134 auto it = out_nodes_map.begin() + dst_dev; in ReplaceBroadcast()
149 for (const auto& out_nodes : out_nodes_map) { in ReplaceBroadcast()
191 for (const auto& out_node : out_nodes_map[send_dev]) { in ReplaceBroadcast()
201 out_nodes_map[send_dev].clear(); in ReplaceBroadcast()
215 for (int recv_dev = 0; recv_dev < out_nodes_map.size(); ++recv_dev) { in ReplaceBroadcast()
216 if (out_nodes_map[recv_dev].empty()) { in ReplaceBroadcast()
234 for (const auto& out_node : out_nodes_map[recv_dev]) { in ReplaceBroadcast()