Searched refs:b_gte (Results 1 – 1 of 1) sorted by relevance
953 HloInstruction* b_gte = builder.AddInstruction( in TEST_F() local955 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, a_gte, b_gte)); in TEST_F()974 EXPECT_TRUE(OutputsBF16(b_gte)); in TEST_F()994 HloInstruction* b_gte = builder.AddInstruction( in TEST_F() local999 HloInstruction::CreateTranspose(shape, b_gte, {0, 1})); in TEST_F()1014 EXPECT_FALSE(OutputsBF16(b_gte)); in TEST_F()