Searched refs:gather_instruction (Results 1 – 2 of 2) sorted by relevance
347 auto gather_instruction = Cast<HloGatherInstruction>(instruction); in ImportInstructionImpl() local351 gather_instruction->gather_dimension_numbers(), builder_))); in ImportInstructionImpl()354 gather_instruction->gather_slice_sizes().begin(), in ImportInstructionImpl()355 gather_instruction->gather_slice_sizes().end()); in ImportInstructionImpl()360 builder_->getBoolAttr(gather_instruction->indices_are_sorted()))); in ImportInstructionImpl()
1497 HloInstruction* gather_instruction = builder.AddInstruction( in TEST_F() local1510 EXPECT_EQ(gather_instruction->ToString(), in TEST_F()1533 HloInstruction* gather_instruction = builder.AddInstruction( in TEST_F() local1546 EXPECT_EQ(gather_instruction->ToString(), in TEST_F()