Searched defs:collective (Results 1 – 3 of 3) sorted by relevance
267 Status NcclManager::GetCommunicator(NcclManager::Collective* collective, in GetCommunicator()476 Collective* collective = collective_it->second; in SignalMultiNodeReady() local506 Collective* collective = nullptr; in AddParticipant() local601 Collective* collective) { in CheckReady()613 void NcclManager::RunCollective(Collective* collective) { in RunCollective()726 Collective* collective = next_launch.first; in LoopKernelLaunches() local
1037 auto* collective = xla::Cast<xla::HloCollectiveInstruction>(instr); in SetupCommonCollectiveOpAttributes() local
437 const HloCollectiveInstruction* collective = in IsLayoutConstrainedCollective() local