Searched refs:ComputeTargetParallelTasks (Results 1 – 2 of 2) sorted by relevance
182 ComputeTargetParallelTasks(module, &hlo_to_parallel_tasks); in Run()253 void ParallelTaskAssigner::ComputeTargetParallelTasks( in ComputeTargetParallelTasks() function in xla::cpu::ParallelTaskAssigner
104 void ComputeTargetParallelTasks(HloModule* module,