Searched defs:NcclCliqueKey (Results 1 – 2 of 2) sorted by relevance
23 NcclCliqueKey::NcclCliqueKey(std::vector<GlobalDeviceId> devices) in NcclCliqueKey() function in xla::gpu::NcclCliqueKey
352 const std::function<void(const NcclCliqueKey&, const NcclClique&)>& fn) { in ForEach()