Home
last modified time | relevance | path

Searched refs:ComputationSchedulerToModuleScheduler (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_memory_scheduler_test.cc148 ComputationSchedulerToModuleScheduler(ListMemoryScheduler), in TEST_F()
200 ComputationSchedulerToModuleScheduler( in TEST_F()
249 ComputationSchedulerToModuleScheduler(ListMemoryScheduler))); in TEST_F()
300 ComputationSchedulerToModuleScheduler(ListMemoryScheduler))); in TEST_F()
Dhlo_memory_scheduler.cc519 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler( in ComputationSchedulerToModuleScheduler() function
668 ComputationSchedulerToModuleScheduler(ListMemoryScheduler, {})( in DefaultModuleScheduler()
677 ComputationSchedulerToModuleScheduler(DFSMemoryScheduler, {})( in DefaultModuleScheduler()
685 ComputationSchedulerToModuleScheduler(PostOrderMemoryScheduler, {})( in DefaultModuleScheduler()
Dhlo_memory_scheduler.h66 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler(
Dhlo_rematerialization_test.cc49 ComputationSchedulerToModuleScheduler(DefaultMemoryScheduler)); in RunHloRematerialization()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule.cc331 ComputationSchedulerToModuleScheduler( in Build()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc628 ComputationSchedulerToModuleScheduler( in RunHloPassesAndBufferAssignement()
741 ComputationSchedulerToModuleScheduler( in RunBackend()