Home
last modified time | relevance | path

Searched refs:body_transpose (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc653 auto body_transpose = builder_body.AddInstruction( in TEST_F() local
662 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() local
739 HloInstruction::CreateTuple({body_dot1, body_transpose})); in TEST_F()
761 EXPECT_FALSE(OutputsBF16(body_transpose)); in TEST_F()