Searched refs:IsCrossModuleAllReduce (Results 1 – 9 of 9) sorted by relevance
62 if (predecessor->IsCrossModuleAllReduce()) { in GlobalPredecessors()83 } else if (instruction->IsCrossModuleAllReduce()) { in GlobalPredecessors()139 if (successor->IsCrossModuleAllReduce()) { in GlobalSuccessors()161 } else if (instruction->IsCrossModuleAllReduce()) { in GlobalSuccessors()258 } else if (hlo->IsCrossModuleAllReduce()) { in VisitTopologicalOrder()
81 if (!ar->IsCrossModuleAllReduce() || !ar->use_global_device_ids()) { in IsPerIdOffsets()134 (ar->IsCrossModuleAllReduce() && !ar->use_global_device_ids()); in IsPerIdOffset()274 (ar->IsCrossModuleAllReduce() && in MatchReduceScatter()324 if (!ar->IsCrossModuleAllReduce()) { in MatchReduceScatter()
98 CHECK(all_reduce->IsCrossModuleAllReduce()); in HasCombinableReplicaGroup()167 if (instruction->IsCrossModuleAllReduce() && in MatchesArCrsPattern()374 if (i1->IsCrossModuleAllReduce()) { in InstructionsComputeSameValue()567 if (other_operand->IsCrossModuleAllReduce() && in RewriteGraph()
81 if (IsChannelInstruction(hlo) || hlo->IsCrossModuleAllReduce()) { in Build()85 } else if (hlo->IsCrossModuleAllReduce()) { in Build()223 hlo->IsCrossModuleAllReduce(); in InstructionCommunicates()335 if (hlo->IsCrossModuleAllReduce()) { in RecordInstructions()
1328 if (!hlo.IsCrossModuleAllReduce()) { in AbslHashValue()1659 bool IsCrossModuleAllReduce() const;
225 if (instructions[0]->IsCrossModuleAllReduce()) { in InstructionWithinBranchIdentical()229 if (!instruction->IsCrossModuleAllReduce()) { in InstructionWithinBranchIdentical()
640 } else if (instruction->IsCrossModuleAllReduce()) { in AddMandatoryConstraints()2313 if (instruction->IsCrossModuleAllReduce()) { in ConstrainChannelLayouts()
517 (!hlos[0]->shape().IsArray() || hlos[0]->IsCrossModuleAllReduce())) { in GroupHlosForConcat()
3048 bool HloInstruction::IsCrossModuleAllReduce() const { in IsCrossModuleAllReduce() function in xla::HloInstruction