Searched refs:hlo_to_parallel_tasks (Results 1 – 2 of 2) sorted by relevance
181 HloToParallelTasks hlo_to_parallel_tasks; in Run() local182 ComputeTargetParallelTasks(module, &hlo_to_parallel_tasks); in Run()186 bool changed = AssignParallelTasks(module, hlo_to_parallel_tasks); in Run()194 HloModule* module, const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasks() argument196 hlo_to_parallel_tasks); in AssignParallelTasks()201 const HloToParallelTasks& hlo_to_parallel_tasks) { in AssignParallelTasksHelper() argument212 hlo_to_parallel_tasks); in AssignParallelTasksHelper()216 hlo_to_parallel_tasks); in AssignParallelTasksHelper()220 auto it = hlo_to_parallel_tasks.find(instruction); in AssignParallelTasksHelper()221 if (it == hlo_to_parallel_tasks.end()) { in AssignParallelTasksHelper()[all …]
97 const HloToParallelTasks& hlo_to_parallel_tasks);100 const HloToParallelTasks& hlo_to_parallel_tasks);105 HloToParallelTasks* hlo_to_parallel_tasks);