Searched refs:body_transpose (Results 1 – 1 of 1) sorted by relevance
653 auto body_transpose = builder_body.AddInstruction( in TEST_F() local662 shape, HloInstruction::FusionKind::kCustom, {body_transpose}, comp_f)); in TEST_F()676 EXPECT_FALSE(OutputsBF16(body_transpose)); in TEST_F()736 auto body_transpose = builder_body.AddInstruction( in TEST_F() local739 HloInstruction::CreateTuple({body_dot1, body_transpose})); in TEST_F()761 EXPECT_FALSE(OutputsBF16(body_transpose)); in TEST_F()