Searched refs:CreateAllReduce (Results 1 – 15 of 15) sorted by relevance
75 all_reduces.push_back(b->AddInstruction(HloInstruction::CreateAllReduce( in MakeCrossReplicaReductions()219 auto all_reduce = b.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()223 b.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()245 auto crs0 = b.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()250 auto crs1 = b.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()
59 computation->AddInstruction(HloInstruction::CreateAllReduce( in Run()
72 auto ar = comp->AddInstruction(HloInstruction::CreateAllReduce( in DecomposeAllGather()
57 computation.AddInstruction(HloInstruction::CreateAllReduce( in ReplaceWithContiguousAllReduce()
83 combined = computation.AddInstruction(HloInstruction::CreateAllReduce( in CombineAllReduces()
201 computation->AddInstruction(HloInstruction::CreateAllReduce( in Run()
239 HloInstruction* crs = builder.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()
259 HloInstruction* crs = builder.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()
636 while_parent->AddInstruction(HloInstruction::CreateAllReduce( in CreateSinkedAllReduces()
213 builder.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F()
715 static std::unique_ptr<HloInstruction> CreateAllReduce(
515 CreateAllReduce(shape, all_operands(), computations(0), in CreateFromProto()1278 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateAllReduce( in CreateAllReduce() function in xla::HloInstruction
1494 return builder->AddInstruction(HloInstruction::CreateAllReduce( in CreateInstruction()
233 computation.AddInstruction(HloInstruction::CreateAllReduce( in TryDecomposeAllReduce()
4236 return b->AddInstruction(HloInstruction::CreateAllReduce( in GetDefaultCollectiveOpsCreator()4252 return b->AddInstruction(HloInstruction::CreateAllReduce( in GetDefaultCollectiveOpsCreator()