Searched refs:CreateAllReduceStart (Results 1 – 5 of 5) sorted by relevance
65 computation->AddInstruction(HloInstruction::CreateAllReduceStart( in Run()
744 static std::unique_ptr<HloInstruction> CreateAllReduceStart(
528 CreateAllReduceStart(shape, all_operands(), computations(0), in CreateFromProto()1301 HloInstruction::CreateAllReduceStart( in CreateAllReduceStart() function in xla::HloInstruction
1505 return builder->AddInstruction(HloInstruction::CreateAllReduceStart( in CreateInstruction()
204 builder.AddInstruction(HloInstruction::CreateAllReduceStart( in TEST_F()