Searched refs:ModuleSchedulerAlgorithm (Results 1 – 2 of 2) sorted by relevance
55 ModuleSchedulerAlgorithm; typedef59 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler(116 const ModuleSchedulerAlgorithm& algorithm = {},134 const ModuleSchedulerAlgorithm& algorithm = {});145 ModuleSchedulerAlgorithm algorithm_;
515 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler( in ComputationSchedulerToModuleScheduler()698 const ModuleSchedulerAlgorithm& algorithm, int64* peak_memory) { in ScheduleModule()730 const ModuleSchedulerAlgorithm& algorithm) in HloMemoryScheduler()