Searched refs:NcclStream (Results 1 – 2 of 2) sorted by relevance
47 struct NcclManager::NcclStream { struct in tensorflow::NcclManager49 NcclStream() {} in NcclStream() argument50 ~NcclStream() { in ~NcclStream() argument80 NcclStream* nccl_stream = nullptr;264 std::set<NcclStream*> used_streams; in GetCommunicator()276 NcclStream* nccl_stream = nullptr; in GetCommunicator()284 nccl_stream = new NcclStream(); in GetCommunicator()505 NcclStream* nccl_stream = collective->communicator->members[i].nccl_stream; in RunCollective()532 NcclStream* nccl_stream = in RunCollective()541 void NcclManager::LoopKernelLaunches(NcclStream* nccl_stream) { in LoopKernelLaunches()
182 struct NcclStream;212 void LoopKernelLaunches(NcclStream* stream);223 std::map<se::StreamExecutor*, std::vector<std::unique_ptr<NcclStream>>>