Searched refs:broadcast2 (Results 1 – 5 of 5) sorted by relevance
41 HloInstruction* broadcast2 = in TEST_F() local47 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()51 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()61 HloInstruction* broadcast2 = in TEST_F() local67 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
20 // CHECK-LABEL: broadcast2
347 HloInstruction* broadcast2 = builder.AddInstruction( in TEST_F() local350 HloInstruction::CreateReshape(reshape_shape, broadcast2)); in TEST_F()
485 auto broadcast2 = builder.AddInstruction( in TEST_F() local488 HloInstruction::CreateTuple({transpose, broadcast2})); in TEST_F()
3447 auto broadcast2 = computation_->AddInstruction( in HandlePad() local3449 return ReplaceInstruction(pad, broadcast2); in HandlePad()