Searched refs:data_dependencies (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order.cc | 29 absl::flat_hash_map<Node*, absl::flat_hash_set<int32>>* data_dependencies) { in DiscoverDataDependencies() argument
37 auto node_leave = [collective_nodes, instance_keys, data_dependencies, in DiscoverDataDependencies()
51 data_dependencies->reserve(data_dependencies->size() + 1 + in DiscoverDataDependencies()
53 const auto& node_deps = (*data_dependencies)[node]; in DiscoverDataDependencies()
55 auto& child_deps = (*data_dependencies)[out_edge->dst()]; in DiscoverDataDependencies()
73 absl::flat_hash_map<Node*, absl::flat_hash_set<int32>>* data_dependencies, in CreateControlDependencies() argument
83 const auto& deps_i = (*data_dependencies)[collective_nodes[i]]; in CreateControlDependencies()
95 const auto& deps_j = (*data_dependencies)[collective_nodes[j]]; in CreateControlDependencies()
189 absl::flat_hash_map<Node*, absl::flat_hash_set<int32>> data_dependencies; in OrderCollectives() local
191 graph, &collective_nodes, &instance_keys, &data_dependencies)); in OrderCollectives()
[all …]
|