Searched defs:max_parallelism (Results 1 – 10 of 10) sorted by relevance
34 SimpleCostModel(const int64_t max_parallelism, in SimpleCostModel()56 DefaultCostModel(const int64_t max_parallelism, in DefaultCostModel()68 int64_t max_parallelism; in GetParallelTaskCount() local113 const int64_t max_parallelism, in ParallelTaskAssignment()
69 ParallelTaskAssigner(const int64_t max_parallelism, in ParallelTaskAssigner()
672 const int max_parallelism = in RunHloPassesAfterLayoutAssn() local
25 void SetPerThreadMaxParallelism(int max_parallelism) { in SetPerThreadMaxParallelism()32 void Shard(int max_parallelism, thread::ThreadPool* workers, int64_t total, in Shard()58 const Runner& runner, int max_parallelism) { in Do()
76 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism()
130 const Eigen::Index max_parallelism = total_workload / parallel_cell_size; in operator() local237 const Eigen::Index max_parallelism = outer_dim * middle_dim; in operator() local
344 int32 max_parallelism = 5; field
381 int32 max_parallelism = 8; field
361 auto max_parallelism = dataset()->max_intra_op_parallelism_; in GetNextInternal() local
390 std::function<void(std::function<void()>)> runner, int max_parallelism) { in RunnerWithMaxParallelism()