Searched refs:builder_body (Results 1 – 1 of 1) sorted by relevance
592 auto builder_body = HloComputation::Builder("body"); in TEST_F() local593 auto body_param = builder_body.AddInstruction( in TEST_F()596 builder_body.AddInstruction(CreateDot(shape, body_param, body_param)); in TEST_F()597 auto body = module->AddEmbeddedComputation(builder_body.Build()); in TEST_F()650 auto builder_body = HloComputation::Builder("body"); in TEST_F() local651 auto body_param = builder_body.AddInstruction( in TEST_F()653 auto body_transpose = builder_body.AddInstruction( in TEST_F()661 auto body_fusion = builder_body.AddInstruction(HloInstruction::CreateFusion( in TEST_F()663 auto body = module->AddEmbeddedComputation(builder_body.Build()); in TEST_F()725 auto builder_body = HloComputation::Builder("body"); in TEST_F() local[all …]