Searched refs:kAllReduce (Results 1 – 22 of 22) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_reachability.cc | 88 if (input->opcode() == HloOpcode::kAllReduce && input->all_reduce_id()) { in Build()
119 case HloOpcode::kAllReduce: { in Build()
|
| 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 | dynamic_padder.cc | 78 case HloOpcode::kAllReduce: in ChooseIdentityValue()
|
| D | hlo_opcode.h | 55 V(kAllReduce, "all-reduce", kHloOpcodeIsVariadic) \
|
| D | hlo_instruction.cc | 375 case HloOpcode::kAllReduce: { in CreateFromProto()
1248 case HloOpcode::kAllReduce: in HasSideEffectNoRecurse()
1372 case HloOpcode::kAllReduce: in CloneWithNewOperands()
1826 case HloOpcode::kAllReduce: in IdenticalSlowPath()
1994 case HloOpcode::kAllReduce: in to_apply()
2014 case HloOpcode::kAllReduce: in set_to_apply()
2183 return opcode() == HloOpcode::kAllReduce && all_reduce_id(); in IsCrossModuleAllReduce()
2187 return opcode() == HloOpcode::kAllReduce && !all_reduce_id(); in IsCrossReplicaAllReduce()
2309 opcode() == HloOpcode::kAllReduce || in ExtraAttributesToString()
2356 case HloOpcode::kAllReduce: in ExtraAttributesToString()
[all …]
|
| D | hlo_element_type_converter.cc | 145 opcode == HloOpcode::kAllReduce || opcode == HloOpcode::kFusion || in Run()
|
| D | call_graph.cc | 61 case HloOpcode::kAllReduce: in GetInstructionCallContext()
|
| D | bfloat16_normalization.cc | 392 hlo->opcode() == HloOpcode::kAllReduce) && in DefaultAction()
|
| D | hlo_computation.cc | 346 case HloOpcode::kAllReduce: in ComputeInstructionPostOrder()
405 case HloOpcode::kAllReduce: { in ComputeChannelDependencies()
|
| D | instruction_fusion.cc | 127 case HloOpcode::kAllReduce: in IsExpensive()
|
| D | hlo_verifier.cc | 47 case HloOpcode::kAllReduce: in IsCallerInstruction()
765 case HloOpcode::kAllReduce: in CheckMixedPrecisionOperands()
|
| D | bfloat16_propagation.cc | 279 (use.instruction->opcode() == HloOpcode::kAllReduce && in AllUsersConsumeBF16()
|
| D | hlo_rematerialization.cc | 75 case HloOpcode::kAllReduce: in IsRematerializable()
|
| D | hlo_graph_dumper.cc | 1040 case HloOpcode::kAllReduce: in GetInstructionColor()
|
| D | buffer_assignment.cc | 188 case HloOpcode::kAllReduce: in GatherComputationsByAllocationType()
|
| D | layout_assignment.cc | 1996 case HloOpcode::kAllReduce: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 523 : HloCollectiveInstruction(HloOpcode::kAllReduce, shape, operands, in HloAllReduceInstruction()
|
| D | hlo_parser.cc | 817 case HloOpcode::kAllReduce: { in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 174 kAllReduce = 1, enumerator
|
| D | nccl_manager.cc | 346 AddParticipant(std::move(participant), context, kAllReduce, reduction_op); in AddToAllReduce()
571 case kAllReduce: { in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 227 case HloOpcode::kAllReduce: in IsConstantVisitor()
2155 return AddInstruction(std::move(instr), HloOpcode::kAllReduce, {operand}); in CrossReplicaSum()
|