Searched refs:UpdateEntryComputationLayout (Results 1 – 7 of 7) sorted by relevance
48 void UpdateEntryComputationLayout(
100 xla::UpdateEntryComputationLayout(module.get(), in Execute()170 xla::UpdateEntryComputationLayout(&executable->module(), in ExecuteWithDeviceBuffers()456 xla::UpdateEntryComputationLayout(module.get(), in CreateExecutable()
149 void UpdateEntryComputationLayout( in UpdateEntryComputationLayout() function
310 UpdateEntryComputationLayout( in BuildExecutables()816 UpdateEntryComputationLayout( in BuildExecutable()1200 UpdateEntryComputationLayout( in GetComputationGraphStats()
133 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::HloTestBase157 xla::UpdateEntryComputationLayout( in UpdateEntryComputationLayout()
360 void UpdateEntryComputationLayout(HloModule* module);
66 UpdateEntryComputationLayout(hlo_module.get()); in RunAndCheck()