Searched refs:parallel_task_assignment (Results 1 – 3 of 3) sorted by relevance
255 ParallelTaskAssignment parallel_task_assignment(max_parallelism_, in ComputeTargetParallelTasks() local264 parallel_task_assignment.GetTargetParallelTaskCount(instruction); in ComputeTargetParallelTasks()
191 ":parallel_task_assignment",1294 name = "parallel_task_assignment",1295 srcs = ["parallel_task_assignment.cc"],1296 hdrs = ["parallel_task_assignment.h"],1317 ":parallel_task_assignment",
807 "//tensorflow/compiler/xla/service/cpu:parallel_task_assignment",