Searched refs:Collective (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 179 struct Collective;
190 Status GetCommunicator(Collective* collective, Communicator** communicator);
207 Collective* CheckReady(const string& collective_key, Collective* collective)
211 void RunCollective(Collective* collective);
217 std::unordered_map<string, std::unique_ptr<Collective>> collectives_
|
| D | nccl_manager.cc | 67 std::deque<std::pair<Collective*, int>> pending_launches_ GUARDED_BY(mu);
130 struct NcclManager::Collective { struct in tensorflow::NcclManager
131 Collective(DataType data_type_in, CollectiveType type_in, in Collective() function
195 Status NcclManager::GetCommunicator(NcclManager::Collective* collective, in GetCommunicator()
381 Collective* to_run = nullptr; in SignalMultiNodeReady()
386 Collective* collective = collective_it->second.get(); in SignalMultiNodeReady()
399 Collective* to_run = nullptr; in AddParticipant()
404 Collective* collective = nullptr; in AddParticipant()
406 auto collective_unique_ptr = absl::make_unique<Collective>( in AddParticipant()
472 NcclManager::Collective* NcclManager::CheckReady(const string& collective_key, in CheckReady()
[all …]
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 432 // Raw data transfers in support of Collective Ops.
491 // Collective Op dynamic group resolution messages.
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 190 // Collective permute field.
|
| /external/cldr/tools/java/org/unicode/cldr/util/data/external/ |
| D | 2013-1_UNLOCODE_CodeListPart1.csv | 3908 ,"BE","ABK","Albertkanaal ports","Albertkanaal ports",,"1-------","AI","0307",,,"Collective entry r…
|