Searched refs:a_gte (Results 1 – 1 of 1) sorted by relevance
951 HloInstruction* a_gte = builder.AddInstruction( in TEST_F() local955 HloInstruction* dot = builder.AddInstruction(CreateDot(shape, a_gte, b_gte)); in TEST_F()973 EXPECT_TRUE(OutputsBF16(a_gte)); in TEST_F()992 HloInstruction* a_gte = builder.AddInstruction( in TEST_F() local997 HloInstruction::CreateTranspose(shape, a_gte, {0, 1})); in TEST_F()1013 EXPECT_FALSE(OutputsBF16(a_gte)); in TEST_F()