Home
last modified time | relevance | path

Searched refs:IsCrossModuleAllReduce (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_util.cc63 if (predecessor->IsCrossModuleAllReduce()) { in GlobalPredecessors()
84 } else if (instruction->IsCrossModuleAllReduce()) { in GlobalPredecessors()
140 if (successor->IsCrossModuleAllReduce()) { in GlobalSuccessors()
162 } else if (instruction->IsCrossModuleAllReduce()) { in GlobalSuccessors()
263 } else if (hlo->IsCrossModuleAllReduce()) { in VisitTopologicalOrder()
Dar_crs_combiner.cc72 if (!instruction->IsCrossModuleAllReduce() || in MatchesArCrsPattern()
200 if (i1->IsCrossModuleAllReduce()) { in InstructionsComputeSameValue()
347 if (other_operand->IsCrossModuleAllReduce() && in RewriteGraph()
Dhlo_module_group_metadata.cc79 if (IsChannelInstruction(hlo) || hlo->IsCrossModuleAllReduce()) { in Build()
83 } else if (hlo->IsCrossModuleAllReduce()) { in Build()
205 hlo->IsCrossModuleAllReduce(); in InstructionCommunicates()
317 if (hlo->IsCrossModuleAllReduce()) { in RecordInstructions()
Dbfloat16_conversion_folding.cc180 if (crs->IsCrossModuleAllReduce()) { in HandleAllReduce()
Dhlo_instruction.h924 if (!IsCrossModuleAllReduce()) {
1241 bool IsCrossModuleAllReduce() const;
Dlayout_assignment.cc509 } else if (instruction->IsCrossModuleAllReduce()) { in AddMandatoryConstraints()
1860 } else if (instruction->IsCrossModuleAllReduce()) { in ConstrainChannelLayouts()
Dhlo_instruction.cc1859 if (!IsCrossModuleAllReduce()) { in Hash()
2182 bool HloInstruction::IsCrossModuleAllReduce() const { in IsCrossModuleAllReduce() function in xla::HloInstruction