Searched defs:NcclStream (Results 1 – 1 of 1) sorted by relevance
72 struct NcclManager::NcclStream : public core::RefCounted { struct in tensorflow::NcclManager77 se::StreamExecutor* executor = nullptr;83 se::Stream* stream = nullptr;85 std::unique_ptr<se::Stream> stream;92 mutex mu;93 condition_variable cv;