Home
last modified time | relevance | path

Searched refs:CreateAllReduceStart (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dasync_collective_creator.cc65 computation->AddInstruction(HloInstruction::CreateAllReduceStart( in Run()
Dhlo_instruction.h744 static std::unique_ptr<HloInstruction> CreateAllReduceStart(
Dhlo_instruction.cc528 CreateAllReduceStart(shape, all_operands(), computations(0), in CreateFromProto()
1301 HloInstruction::CreateAllReduceStart( in CreateAllReduceStart() function in xla::HloInstruction
Dhlo_parser.cc1505 return builder->AddInstruction(HloInstruction::CreateAllReduceStart( in CreateInstruction()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule_test.cc204 builder.AddInstruction(HloInstruction::CreateAllReduceStart( in TEST_F()