Searched refs:max_parallelism (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 25 void SetPerThreadMaxParallelism(int max_parallelism) { in SetPerThreadMaxParallelism() argument
26 CHECK_LE(0, max_parallelism); in SetPerThreadMaxParallelism()
27 per_thread_max_parallism = max_parallelism; in SetPerThreadMaxParallelism()
32 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total, in Shard() argument
38 max_parallelism = std::min(max_parallelism, GetPerThreadMaxParallelism()); in Shard()
39 if (max_parallelism <= 1) { in Shard()
44 if (max_parallelism >= workers->NumThreads()) { in Shard()
50 max_parallelism); in Shard()
56 const Runner& runner, int max_parallelism) { in Do() argument
66 std::max<int>(1, std::min(static_cast<int64>(max_parallelism), in Do()
|
| D | work_sharder.h | 57 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total,
68 void SetPerThreadMaxParallelism(int max_parallelism);
74 ScopedPerThreadMaxParallelism(int max_parallelism) in ScopedPerThreadMaxParallelism() argument
76 SetPerThreadMaxParallelism(max_parallelism); in ScopedPerThreadMaxParallelism()
96 const Runner& runner, int max_parallelism);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 32 SimpleCostModel(const int64 max_parallelism, in SimpleCostModel() argument
34 : max_parallelism_(max_parallelism), shape_size_(shape_size) {} in SimpleCostModel()
53 DefaultCostModel(const int64 max_parallelism, in DefaultCostModel() argument
56 : max_parallelism_(max_parallelism), in DefaultCostModel()
65 int64 max_parallelism; in GetParallelTaskCount() local
77 max_parallelism = in GetParallelTaskCount()
84 max_parallelism = max_parallelism_; in GetParallelTaskCount()
97 return std::min(max_parallelism, in GetParallelTaskCount()
108 const int64 max_parallelism, in ParallelTaskAssignment() argument
112 VLOG(1) << "ParallelTaskAssignment max_parallelism: " << max_parallelism; in ParallelTaskAssignment()
[all …]
|
| D | parallel_task_assignment.h | 41 ParallelTaskAssignment(const int64 max_parallelism,
68 ParallelTaskAssigner(const int64 max_parallelism, in ParallelTaskAssigner() argument
71 : max_parallelism_(max_parallelism), in ParallelTaskAssigner()
|
| D | cpu_compiler.cc | 368 const int max_parallelism = in RunHloPassesAfterLayoutAssn() local
379 max_parallelism, ShapeSizeBytesFunction(), target_machine_features); in RunHloPassesAfterLayoutAssn()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.cc | 267 std::function<void(std::function<void()>)> runner, int max_parallelism) { in RunnerWithMaxParallelism() argument
269 [max_parallelism]( in RunnerWithMaxParallelism()
274 [max_parallelism](const std::function<void()>& fn) { in RunnerWithMaxParallelism()
275 ScopedPerThreadMaxParallelism scope(max_parallelism); in RunnerWithMaxParallelism()
|
| D | map_defun_op.cc | 154 int max_parallelism) in ComputeOptions()
160 if (max_parallelism >= 1) { in ComputeOptions()
161 runner = RunnerWithMaxParallelism(*ctx->runner(), max_parallelism); in ComputeOptions()
|
| D | dataset_utils.h | 111 std::function<void(std::function<void()>)> runner, int max_parallelism);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | redux_functor.h | 50 const int64 max_parallelism = total_workload / parallel_cell_size; in operator() local
56 std::min(max_parallelism, Eigen::divup(total_workload, min_block_size)); in operator()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 310 auto max_parallelism = dataset()->max_intra_op_parallelism_; in GetNextInternal() local
312 RunnerWithMaxParallelism(*ctx->runner(), max_parallelism); in GetNextInternal()
|