Searched refs:GetAllReduceGroup (Results 1 – 3 of 3) sorted by relevance
65 metadata_.GetAllReduceGroup(*predecessor->all_reduce_id())) { in GlobalPredecessors()86 metadata_.GetAllReduceGroup(*instruction->all_reduce_id()); in GlobalPredecessors()142 metadata_.GetAllReduceGroup(*successor->all_reduce_id())) { in GlobalSuccessors()164 metadata_.GetAllReduceGroup(*instruction->all_reduce_id()); in GlobalSuccessors()264 instruction_group = metadata_.GetAllReduceGroup(*hlo->all_reduce_id()); in VisitTopologicalOrder()
141 const std::vector<HloInstruction*>& GetAllReduceGroup(
84 for (HloInstruction* instr : GetAllReduceGroup(*hlo->all_reduce_id())) { in Build()234 const std::vector<HloInstruction*>& HloModuleGroupMetadata::GetAllReduceGroup( in GetAllReduceGroup() function in xla::HloModuleGroupMetadata