Searched refs:IsCrossReplicaAllReduce (Results 1 – 5 of 5) sorted by relevance
174 while (!next->IsCrossReplicaAllReduce()) { in MatchesArCrsPattern()487 if (next_0->IsCrossReplicaAllReduce()) { in KeepProvablyEqualInstructionGroupsMPMD()527 if (next->IsCrossReplicaAllReduce()) { in KeepProvablyEqualInstructionGroupsSPMD()550 while (!next->IsCrossReplicaAllReduce()) { in RewriteGraph()
63 if (!inst->IsCrossReplicaAllReduce()) { in Run()
1494 bool IsCrossReplicaAllReduce() const;
2684 bool HloInstruction::IsCrossReplicaAllReduce() const { in IsCrossReplicaAllReduce() function in xla::HloInstruction
2231 if (hlo->IsCrossReplicaAllReduce() && hlo->operand_count() == 1) { in HandleAllReduce()