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

  /external/tensorflow/tensorflow/core/nccl/
nccl_manager.h 182 struct NcclStream;
212 void LoopKernelLaunches(NcclStream* stream);
223 std::map<se::StreamExecutor*, std::vector<std::unique_ptr<NcclStream>>>
nccl_manager.cc 47 struct NcclManager::NcclStream {
49 NcclStream() {}
50 ~NcclStream() {
80 NcclStream* nccl_stream = nullptr;
264 std::set<NcclStream*> used_streams;
276 NcclStream* nccl_stream = nullptr;
284 nccl_stream = new NcclStream();
505 NcclStream* nccl_stream = collective->communicator->members[i].nccl_stream;
532 NcclStream* nccl_stream =
541 void NcclManager::LoopKernelLaunches(NcclStream* nccl_stream)
    [all...]

Completed in 210 milliseconds