Searched refs:a_trans (Results 1 – 1 of 1) sorted by relevance
943 HloInstruction* a_trans = in TEST_F() local948 builder.AddInstruction(HloInstruction::CreateTuple({a_trans, b_trans})); in TEST_F()971 EXPECT_TRUE(OutputsBF16(a_trans)); in TEST_F()996 HloInstruction* a_trans = builder.AddInstruction( in TEST_F() local1001 builder.AddInstruction(CreateDot(shape, a_trans, b_trans)); in TEST_F()1011 EXPECT_TRUE(OutputsBF16(a_trans)); in TEST_F()