Searched refs:parallel_task_assignment (Results 1 – 3 of 3) sorted by relevance
249 ParallelTaskAssignment parallel_task_assignment(max_parallelism_, in ComputeTargetParallelTasks() local258 parallel_task_assignment.GetTargetParallelTaskCount(instruction); in ComputeTargetParallelTasks()
145 ":parallel_task_assignment",1038 name = "parallel_task_assignment",1039 srcs = ["parallel_task_assignment.cc"],1040 hdrs = ["parallel_task_assignment.h"],1060 ":parallel_task_assignment",
609 "//tensorflow/compiler/xla/service/cpu:parallel_task_assignment",