Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_group_util.cc62 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()
H A Dreduce_scatter_utils.cc81 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()
H A Dar_crs_combiner.cc98 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()
H A Dhlo_module_group_metadata.cc81 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()
H A Dhlo_instruction.h1328 if (!hlo.IsCrossModuleAllReduce()) { in AbslHashValue()
1659 bool IsCrossModuleAllReduce() const;
H A Dconditional_code_motion.cc225 if (instructions[0]->IsCrossModuleAllReduce()) { in InstructionWithinBranchIdentical()
229 if (!instruction->IsCrossModuleAllReduce()) { in InstructionWithinBranchIdentical()
H A Dlayout_assignment.cc640 } else if (instruction->IsCrossModuleAllReduce()) { in AddMandatoryConstraints()
2313 if (instruction->IsCrossModuleAllReduce()) { in ConstrainChannelLayouts()
H A Dwhile_loop_concat_code_motion.cc517 (!hlos[0]->shape().IsArray() || hlos[0]->IsCrossModuleAllReduce())) { in GroupHlosForConcat()
H A Dhlo_instruction.cc3048 bool HloInstruction::IsCrossModuleAllReduce() const { in IsCrossModuleAllReduce() function in xla::HloInstruction