Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_rematerialization_test_utils_test.cc71 auto body_computation = MakeRematerializableComputation(); in TEST_F() local
H A Dhlo_rematerialization_test.cc158 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() local
194 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() local
H A Dheap_simulator_test.cc66 HloComputation* body_computation = in TEST_F() local
163 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() local
908 HloComputation* body_computation = in TEST_F() local
H A Dhlo_live_range_test.cc308 HloComputation* body_computation = in TEST_F() local
H A Dhlo_dce_test.cc281 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() local
H A Dmemory_space_assignment_test.cc821 HloComputation* body_computation = in TEST_P() local
2630 HloComputation* body_computation = in TEST_P() local
2941 HloComputation* body_computation = in TEST_P() local
3046 HloComputation* body_computation = in TEST_P() local
3680 HloComputation* body_computation; in TEST_P() local
H A Dbuffer_assignment_test.cc834 HloComputation* body_computation = in TEST_F() local
1044 auto body_computation = in TEST_F() local
H A Dhlo_dataflow_analysis_test.cc2809 HloComputation* body_computation = in TEST_F() local