HomeSort by relevance Sort by last modified time
    Searched refs:Communicator (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/nccl/
nccl_manager.h 38 // NCCL manager is used to make the asynchronous communicator calls and to
115 // operation key, number of participants, and communicator key.
180 struct Communicator;
184 // Gets the `Communicator` object that will be used to enqueue NCCL kernels
185 // for `collective`, and returns it via `communicator`.
190 Status GetCommunicator(Collective* collective, Communicator** communicator);
226 std::vector<std::unique_ptr<Communicator>> communicators_;
nccl_manager.cc 83 struct NcclManager::Communicator {
85 explicit Communicator(std::vector<CommunicatorMember> members,
153 Communicator* communicator = nullptr; member in struct:tensorflow::NcclManager::Collective
173 // Guarded by the mutex of the containing Communicator.
196 NcclManager::Communicator** communicator) {
207 // For single-node collectives, we identify a communicator uniquely by the
210 // communicator for GPUs 0, 1, and 2.
213 // the same as getting the communicator connecting the devices in th
    [all...]

Completed in 1754 milliseconds