| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_reachability.cc | 103 if (input->opcode() == HloOpcode::kAllReduce && input->channel_id()) { in Build()
136 case HloOpcode::kAllReduce: { in Build()
|
| D | all_reduce_combiner.cc | 66 TF_RET_CHECK(hlo->opcode() == HloOpcode::kAllReduce); in CombineAllReduces()
188 if (instruction->opcode() != HloOpcode::kAllReduce) { in CreateComputationGroups()
227 if (instruction->opcode() != HloOpcode::kAllReduce) { in CreateComputationGroups()
|
| D | bfloat16_conversion_folding_test.cc | 41 hlo.opcode() == HloOpcode::kAllReduce) { in SupportsBF16Operand()
52 hlo.opcode() == HloOpcode::kAllReduce) { in SupportsBF16Output()
61 hlo.opcode() == HloOpcode::kAllReduce) { in SupportsMixedPrecisions()
|
| D | hlo_opcode_test.cc | 53 case HloOpcode::kAllReduce: in TEST()
|
| D | hlo_query.cc | 127 if (hlo->opcode() == HloOpcode::kAllReduce && in ContainsLayoutConstrainedAllReduce()
|
| D | hlo_replication_analysis.cc | 58 if (hlo->opcode() == HloOpcode::kAllReduce || in DetermineHloInstructionIsReplicated()
75 if (hlo->opcode() == HloOpcode::kAllReduce) { in DetermineHloInstructionIsReplicated()
|
| D | hlo_opcode.h | 52 V(kAllReduce, "all-reduce", kHloOpcodeIsVariadic) \
|
| D | all_reduce_combiner_test.cc | 51 if (hlo->opcode() == HloOpcode::kAllReduce) { in AllReduceCount()
144 ASSERT_TRUE(combined->opcode() == HloOpcode::kAllReduce); in TEST_F()
|
| D | hlo_element_type_converter.cc | 145 opcode == HloOpcode::kAllReduce || opcode == HloOpcode::kFusion || in Run()
|
| D | hlo_instruction.cc | 430 case HloOpcode::kAllReduce: { in CreateFromProto()
1534 case HloOpcode::kAllReduce: in HasSideEffectNoRecurse()
1689 case HloOpcode::kAllReduce: in CloneWithNewOperands()
2229 case HloOpcode::kAllReduce: in IdenticalSlowPath()
2434 case HloOpcode::kAllReduce: in to_apply()
2455 case HloOpcode::kAllReduce: in set_to_apply()
2681 return opcode() == HloOpcode::kAllReduce && channel_id(); in IsCrossModuleAllReduce()
2685 return opcode() == HloOpcode::kAllReduce && !channel_id(); in IsCrossReplicaAllReduce()
2825 opcode() == HloOpcode::kAllReduce || in ExtraAttributesToString()
2884 case HloOpcode::kAllReduce: in ExtraAttributesToString()
[all …]
|
| D | conditional_code_motion.cc | 167 case HloOpcode::kAllReduce: in WorthHoisting()
182 case HloOpcode::kAllReduce: in WorthHoisting()
933 if (opcode == HloOpcode::kAllReduce && !is_inside_branch) { in WorthHoisting()
|
| D | call_graph.cc | 61 case HloOpcode::kAllReduce: in GetInstructionCallContext()
|
| D | hlo_computation.cc | 414 case HloOpcode::kAllReduce: in ComputeInstructionPostOrder()
470 case HloOpcode::kAllReduce: { in ComputeChannelDependencies()
|
| D | sharding_propagation.cc | 180 case HloOpcode::kAllReduce: in PickRepresentativeOperand()
323 case HloOpcode::kAllReduce: in SupportSpatialPartitioning()
1594 (opcode == HloOpcode::kAllReduce && in CheckAndUpdateDeviceAssignmentsInWhileBody()
|
| D | bfloat16_normalization.cc | 510 hlo->opcode() == HloOpcode::kAllReduce) && in DefaultAction()
|
| D | instruction_fusion.cc | 151 case HloOpcode::kAllReduce: in IsExpensive()
|
| D | hlo_verifier.cc | 43 case HloOpcode::kAllReduce: in IsCallerInstruction()
1083 case HloOpcode::kAllReduce: in CheckMixedPrecisionOperands()
1516 if (instruction->opcode() != HloOpcode::kAllReduce) { in VerifyLayoutConstrainedAllReduce()
|
| D | layout_assignment.cc | 1326 if (user->opcode() == HloOpcode::kAllReduce) { in PropagateOperandConstraint()
1495 if (instruction->opcode() == HloOpcode::kAllReduce) { in PropagateBufferConstraintToOperands()
2276 case HloOpcode::kAllReduce: in InstructionCanChangeLayout()
|
| D | hlo_module.cc | 168 case HloOpcode::kAllReduce: in ReplaceComputations()
|
| D | bfloat16_propagation.cc | 316 (use.instruction->opcode() == HloOpcode::kAllReduce && in AllUsersConsumeBF16()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | schedule_aware_all_gather_cse.cc | 44 if (coll->opcode() == HloOpcode::kAllReduce && coll->shape().IsArray()) { in MayConsiderAsAllGather()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 204 kAllReduce = 1, enumerator
|
| D | nccl_manager.cc | 435 AddParticipant(std::move(participant), context, kAllReduce, reduction_op); in AddToAllReduce()
736 case kAllReduce: { in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 334 return xla::HloOpcode::kAllReduce; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 554 case HloOpcode::kAllReduce: { in ImportInstructionImpl()
|