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