Searched refs:kAllGather (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_support.cc | 82 case HloOpcode::kAllGather: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | hlo_opcode.h | 51 V(kAllGather, "all-gather", 1) \
|
| D | all_gather_decomposer.cc | 144 if (hlo->opcode() != HloOpcode::kAllGather) { in Run()
|
| D | hlo_replication_analysis.cc | 59 hlo->opcode() == HloOpcode::kAllGather) { in DetermineHloInstructionIsReplicated()
|
| D | instruction_fusion.cc | 150 case HloOpcode::kAllGather: in IsExpensive()
|
| D | hlo_instruction.cc | 412 case HloOpcode::kAllGather: { in CreateFromProto()
1688 case HloOpcode::kAllGather: in CloneWithNewOperands()
2228 case HloOpcode::kAllGather: in IdenticalSlowPath()
3130 case HloOpcode::kAllGather: in Visit()
|
| D | hlo_graph_dumper.cc | 1064 case HloOpcode::kAllGather: in GetInstructionColor()
|
| D | sharding_propagation.cc | 179 case HloOpcode::kAllGather: in PickRepresentativeOperand()
|
| D | hlo_parser.cc | 163 case HloOpcode::kAllGather: in CanInferShape()
1196 case HloOpcode::kAllGather: { in ParseInstructionRhs()
|
| D | layout_assignment.cc | 2222 case HloOpcode::kAllGather: in InstructionCanChangeLayout()
|
| D | hlo_instructions.cc | 622 : HloCollectiveInstruction(HloOpcode::kAllGather, shape, {operand}, in HloAllGatherInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | schedule_aware_all_gather_cse.cc | 40 if (coll->opcode() == HloOpcode::kAllGather) { in MayConsiderAsAllGather()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 207 kAllGather = 4, enumerator
|
| D | nccl_manager.cc | 440 AddParticipant(std::move(participant), context, kAllGather, in AddToAllGather()
810 case kAllGather: { in LoopKernelLaunches()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 392 EXPECT_EQ(root->opcode(), HloOpcode::kAllGather); in TEST_F()
403 EXPECT_EQ(root->opcode(), HloOpcode::kAllGather); in TEST_F()
|
| D | xla_builder.cc | 2746 AddInstruction(std::move(instr), HloOpcode::kAllGather, {operand})); in AllGather()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 254 case HloOpcode::kAllGather: in EmitOp()
|