Searched defs:max_parallelism (Results 1 – 8 of 8) sorted by relevance
25 void SetPerThreadMaxParallelism(int max_parallelism) { in SetPerThreadMaxParallelism()32 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total, in Shard()56 const Runner& runner, int max_parallelism) { in Do()
74 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism()
32 SimpleCostModel(const int64 max_parallelism, in SimpleCostModel()53 DefaultCostModel(const int64 max_parallelism, in DefaultCostModel()65 int64 max_parallelism; in GetParallelTaskCount() local108 const int64 max_parallelism, in ParallelTaskAssignment()
68 ParallelTaskAssigner(const int64 max_parallelism, in ParallelTaskAssigner()
368 const int max_parallelism = in RunHloPassesAfterLayoutAssn() local
50 const int64 max_parallelism = total_workload / parallel_cell_size; in operator() local
267 std::function<void(std::function<void()>)> runner, int max_parallelism) { in RunnerWithMaxParallelism()
310 auto max_parallelism = dataset()->max_intra_op_parallelism_; in GetNextInternal() local