Home
last modified time | relevance | path

Searched refs:CreateGather (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction_test.cc1369 HloInstruction::CreateGather(gather_result_shape, input, start_indices, in TEST_F()
1404 HloInstruction::CreateGather(gather_result_shape, input, start_indices, in TEST_F()
Dhlo_instruction.h725 static std::unique_ptr<HloInstruction> CreateGather(
Dhlo_instruction.cc532 instruction = CreateGather(shape, operands(0), operands(1), in CreateFromProto()
1306 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateGather( in CreateGather() function in xla::HloInstruction
Dhlo_parser.cc1643 instruction = builder->AddInstruction(HloInstruction::CreateGather( in ParseInstructionRhs()