Home
last modified time | relevance | path

Searched refs:UpdateEntryComputationLayout (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dhlo_test_base.cc133 UpdateEntryComputationLayout(module.get()); in ParseAndReturnVerifiedModule()
145 UpdateEntryComputationLayout(module.get()); in ParseAndReturnVerifiedModule()
152 UpdateEntryComputationLayout(module); in AddEntryComputationAndUpdateEntryComputationLayout()
156 void HloTestBase::UpdateEntryComputationLayout(HloModule* module) { in UpdateEntryComputationLayout() function in xla::HloTestBase
157 xla::UpdateEntryComputationLayout( in UpdateEntryComputationLayout()
H A Dhlo_test_base.h360 void UpdateEntryComputationLayout(HloModule* module);
H A Dbuffer_donation_test.cc66 UpdateEntryComputationLayout(hlo_module.get()); in RunAndCheck()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_runner.cc100 xla::UpdateEntryComputationLayout(module.get(), in Execute()
170 xla::UpdateEntryComputationLayout(&executable->module(), in ExecuteWithDeviceBuffers()
456 xla::UpdateEntryComputationLayout(module.get(), in CreateExecutable()
H A Dhlo_module_util.h48 void UpdateEntryComputationLayout(
H A Dhlo_module_util.cc149 void UpdateEntryComputationLayout( in UpdateEntryComputationLayout() function
H A Dservice.cc310 UpdateEntryComputationLayout( in BuildExecutables()
816 UpdateEntryComputationLayout( in BuildExecutable()
1200 UpdateEntryComputationLayout( in GetComputationGraphStats()