Searched refs:NcclCliqueCache (Results 1 – 3 of 3) sorted by relevance
27 NcclCliqueCache().ForEach([&](const NcclCliqueKey& k, const NcclClique&) { in DevicesWithOpenNcclChannels()
108 NcclCliqueMap& NcclCliqueCache() { in NcclCliqueCache() function232 maybe_clique_ = NcclCliqueCache().GetOrTryCreateIfAbsent( in RunCollectiveOp()
147 NcclCliqueMap& NcclCliqueCache();