Searched refs:ModuleSchedulerAlgorithm (Results 1 – 2 of 2) sorted by relevance
62 ModuleSchedulerAlgorithm; typedef66 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler(123 const ModuleSchedulerAlgorithm& algorithm = {},142 const ModuleSchedulerAlgorithm& algorithm = {});153 ModuleSchedulerAlgorithm algorithm_;
519 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler( in ComputationSchedulerToModuleScheduler()713 const ModuleSchedulerAlgorithm& algorithm, int64* peak_memory) { in ScheduleModule()746 const ModuleSchedulerAlgorithm& algorithm) in HloMemoryScheduler()