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()58 const Runner& runner, int max_parallelism) { in Do()
76 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism()
33 SimpleCostModel(const int64 max_parallelism, in SimpleCostModel()54 DefaultCostModel(const int64 max_parallelism, in DefaultCostModel()66 int64 max_parallelism; in GetParallelTaskCount() local110 const int64 max_parallelism, in ParallelTaskAssignment()
68 ParallelTaskAssigner(const int64 max_parallelism, in ParallelTaskAssigner()
430 const int max_parallelism = in RunHloPassesAfterLayoutAssn() local
130 const Eigen::Index max_parallelism = total_workload / parallel_cell_size; in operator() local242 const Eigen::Index max_parallelism = outer_dim * middle_dim; in operator() local
346 auto max_parallelism = dataset()->max_intra_op_parallelism_; in GetNextInternal() local
468 std::function<void(std::function<void()>)> runner, int max_parallelism) { in RunnerWithMaxParallelism()