Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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()
H A Dhlo_memory_scheduler.cc526 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler( in ComputationSchedulerToModuleScheduler() function
680 ComputationSchedulerToModuleScheduler(ListMemoryScheduler, {})( in DefaultModuleScheduler()
689 ComputationSchedulerToModuleScheduler(DFSMemoryScheduler, {})( in DefaultModuleScheduler()
697 ComputationSchedulerToModuleScheduler(PostOrderMemoryScheduler, {})( in DefaultModuleScheduler()
H A Dhlo_memory_scheduler.h67 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler(
H A Dhlo_rematerialization_test.cc49 ComputationSchedulerToModuleScheduler(DefaultMemoryScheduler)); in RunHloRematerialization()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_hlo_schedule.cc161 ComputationSchedulerToModuleScheduler( in ScheduleGpuModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc841 ComputationSchedulerToModuleScheduler( in AssignBuffers()
1169 ComputationSchedulerToModuleScheduler( in CompileLegacyCpuExecutable()