Searched refs:max_parallelism_ (Results 1 – 3 of 3) sorted by relevance
34 : max_parallelism_(max_parallelism), shape_size_(shape_size) {} in SimpleCostModel()42 return std::min(max_parallelism_, in GetParallelTaskCount()47 const int64 max_parallelism_; member in xla::cpu::SimpleCostModel56 : max_parallelism_(max_parallelism), in DefaultCostModel()84 max_parallelism = max_parallelism_; in GetParallelTaskCount()102 const int64 max_parallelism_; member in xla::cpu::DefaultCostModel237 ParallelTaskAssignment parallel_task_assignment(max_parallelism_, in ComputeTargetParallelTasks()
71 : max_parallelism_(max_parallelism), in ParallelTaskAssigner()101 int64 max_parallelism_; variable
33 const int max_parallelism_ = 10; member in xla::__anon073d40dd0111::ParallelTaskAssignmentTest43 return cpu::ParallelTaskAssigner(max_parallelism_, shape_size_func_, in RunParallelTaskAssigner()