Searched defs:sub_computation (Results 1 – 4 of 4) sorted by relevance
229 HloComputation* sub_computation = in TEST_F() local
1643 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local2511 HloComputation* sub_computation; in TEST_F() local
2042 for (auto* sub_computation : hlo->called_computations()) { in DefaultAction() local
2842 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local