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_;
503 ModuleSchedulerAlgorithm ComputationSchedulerToModuleScheduler( in ComputationSchedulerToModuleScheduler()686 const ModuleSchedulerAlgorithm& algorithm, int64* peak_memory) { in ScheduleModule()718 const ModuleSchedulerAlgorithm& algorithm) in HloMemoryScheduler()