Searched refs:AssignParallelTasksHelper (Results 1 – 2 of 2) sorted by relevance
195 return AssignParallelTasksHelper(module, module->entry_computation(), in AssignParallelTasks()199 bool ParallelTaskAssigner::AssignParallelTasksHelper( in AssignParallelTasksHelper() function in xla::cpu::ParallelTaskAssigner211 changed |= AssignParallelTasksHelper(module, instruction->while_body(), in AssignParallelTasksHelper()215 changed |= AssignParallelTasksHelper(module, instruction->to_apply(), in AssignParallelTasksHelper()
98 bool AssignParallelTasksHelper(