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