Home
last modified time | relevance | path

Searched defs:sub_computation (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc229 HloComputation* sub_computation = in TEST_F() local
Dbuffer_assignment_test.cc1566 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local
2433 HloComputation* sub_computation; in TEST_F() local
Ddynamic_padder.cc1696 for (auto* sub_computation : hlo->called_computations()) { in DefaultAction() local
Dhlo_dataflow_analysis_test.cc2777 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local