Searched defs:NcclStream (Results 1 – 1 of 1) sorted by relevance
47 struct NcclManager::NcclStream { struct in tensorflow::NcclManager49 NcclStream() {} in NcclStream() argument50 ~NcclStream() { in ~NcclStream()56 se::StreamExecutor* executor = nullptr;60 std::unique_ptr<se::Stream> stream;63 std::unique_ptr<Thread> thread;64 mutex mu;65 condition_variable cv;