Searched refs:IsCrossReplicaAllReduce (Results 1 – 4 of 4) sorted by relevance
79 while (!next->IsCrossReplicaAllReduce()) { in MatchesArCrsPattern()307 if (next_0->IsCrossReplicaAllReduce()) { in KeepProvablyEqualInstructionGroups()331 while (!next->IsCrossReplicaAllReduce()) { in RewriteGraph()
1244 bool IsCrossReplicaAllReduce() const;
2186 bool HloInstruction::IsCrossReplicaAllReduce() const { in IsCrossReplicaAllReduce() function in xla::HloInstruction
1398 crs->IsCrossReplicaAllReduce() && in HandleAllReduce()1421 crs->IsCrossReplicaAllReduce(), NcclAllReduceThunk::NcclIsEnabled()); in HandleAllReduce()