Searched defs:hlo_to_parallel_tasks (Results 1 – 1 of 1) sorted by relevance
163 HloToParallelTasks hlo_to_parallel_tasks; in Run() local176 HloModule* module, const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasks()183 const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasksHelper()236 HloModule* module, HloToParallelTasks* hlo_to_parallel_tasks) { in ComputeTargetParallelTasks()