Home
last modified time | relevance | path

Searched defs:body_computation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test_utils_test.cc71 auto body_computation = MakeRematerializableComputation(); in TEST_F() local
Dhlo_rematerialization_test.cc158 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() local
194 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() local
Dhlo_live_range_test.cc308 HloComputation* body_computation = in TEST_F() local
Dhlo_dce_test.cc283 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() local
Dheap_simulator_test.cc67 HloComputation* body_computation = in TEST_F() local
164 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() local
908 HloComputation* body_computation = in TEST_F() local
Dmemory_space_assignment_test.cc786 HloComputation* body_computation = in TEST_P() local
2669 HloComputation* body_computation = in TEST_P() local
2980 HloComputation* body_computation = in TEST_P() local
3085 HloComputation* body_computation = in TEST_P() local
3719 HloComputation* body_computation; in TEST_P() local
Dbuffer_assignment_test.cc815 HloComputation* body_computation = in TEST_F() local
1025 auto body_computation = in TEST_F() local
Dhlo_dataflow_analysis_test.cc2744 HloComputation* body_computation = in TEST_F() local