Searched refs:comp_f (Results 1 – 1 of 1) sorted by relevance
446 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F() local448 dot_f->shape(), HloInstruction::FusionKind::kCustom, {add, add}, comp_f)); in TEST_F()485 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F() local488 comp_f)); in TEST_F()506 auto new_fusion_root = comp_f->root_instruction(); in TEST_F()660 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F() local662 shape, HloInstruction::FusionKind::kCustom, {body_transpose}, comp_f)); in TEST_F()