Searched refs:hlo_to_parallel_tasks (Results 1 – 2 of 2) sorted by relevance
163 HloToParallelTasks hlo_to_parallel_tasks; in Run() local164 ComputeTargetParallelTasks(module, &hlo_to_parallel_tasks); in Run()168 bool changed = AssignParallelTasks(module, hlo_to_parallel_tasks); in Run()176 HloModule* module, const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasks() argument178 hlo_to_parallel_tasks); in AssignParallelTasks()183 const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasksHelper() argument194 hlo_to_parallel_tasks); in AssignParallelTasksHelper()198 hlo_to_parallel_tasks); in AssignParallelTasksHelper()202 auto it = hlo_to_parallel_tasks.find(instruction); in AssignParallelTasksHelper()203 if (it == hlo_to_parallel_tasks.end()) { in AssignParallelTasksHelper()[all …]
91 const HloToParallelTasks& hlo_to_parallel_tasks);94 const HloToParallelTasks& hlo_to_parallel_tasks);99 HloToParallelTasks* hlo_to_parallel_tasks);