Searched refs:collective_nodes (Results 1 – 1 of 1) sorted by relevance
27 const Graph* graph, std::vector<Node*>* collective_nodes, in DiscoverDataDependencies() argument37 auto node_leave = [collective_nodes, instance_keys, data_dependencies, in DiscoverDataDependencies()46 collective_nodes->push_back(node); in DiscoverDataDependencies()71 const std::vector<Node*>& collective_nodes, in CreateControlDependencies() argument77 for (int i = 0; i < collective_nodes.size() - 1; i++) { in CreateControlDependencies()78 if (!collective_nodes[i]->IsCollective() || in CreateControlDependencies()79 collective_nodes[i]->type_string() != "CollectiveReduce") { in CreateControlDependencies()81 collective_nodes[i]->DebugString()); in CreateControlDependencies()83 const auto& deps_i = (*data_dependencies)[collective_nodes[i]]; in CreateControlDependencies()84 for (int j = i + 1; j < collective_nodes.size(); j++) { in CreateControlDependencies()[all …]