Searched refs:broadcast2 (Results 1 – 3 of 3) sorted by relevance
40 HloInstruction* broadcast2 = in TEST_F() local46 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()50 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()60 HloInstruction* broadcast2 = in TEST_F() local66 EXPECT_EQ(broadcast2, computation->root_instruction()); in TEST_F()
345 HloInstruction* broadcast2 = builder.AddInstruction( in TEST_F() local348 HloInstruction::CreateReshape(reshape_shape, broadcast2)); in TEST_F()
485 auto broadcast2 = builder.AddInstruction( in TEST_F() local488 HloInstruction::CreateTuple({transpose, broadcast2})); in TEST_F()