Searched refs:AssignParallelTasks (Results 1 – 2 of 2) sorted by relevance
186 bool changed = AssignParallelTasks(module, hlo_to_parallel_tasks); in Run()193 bool ParallelTaskAssigner::AssignParallelTasks( in AssignParallelTasks() function in xla::cpu::ParallelTaskAssigner
96 bool AssignParallelTasks(HloModule* module,