Searched defs:hlo_to_parallel_tasks (Results 1 – 1 of 1) sorted by relevance
181 HloToParallelTasks hlo_to_parallel_tasks; in Run() local194 HloModule* module, const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasks()201 const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasksHelper()254 HloModule* module, HloToParallelTasks* hlo_to_parallel_tasks) { in ComputeTargetParallelTasks()