| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 33 int64 cost_per_unit, std::function<void(int64, int64)> work) { in Shard() argument
45 workers->ParallelFor(total, cost_per_unit, work); in Shard()
49 total, cost_per_unit, work, in Shard()
57 void Sharder::Do(int64 total, int64 cost_per_unit, const Work& work, in Do() argument
59 cost_per_unit = std::max(int64{1}, cost_per_unit); in Do()
69 total * cost_per_unit / kMinCostPerShard)); in Do()
|
| D | work_sharder_test.cc | 30 void RunSharding(int64 num_workers, int64 total, int64 cost_per_unit, in RunSharding() argument
37 Shard(num_workers, threads, total, cost_per_unit, in RunSharding()
50 LOG(INFO) << num_workers << " " << total << " " << cost_per_unit << " " in RunSharding()
68 for (auto cost_per_unit : {0, 1, 11, 102, 1003, 10005, 1000007}) { in TEST()
71 RunSharding(workers, total, cost_per_unit, maxp, &threads); in TEST()
82 const int64 cost_per_unit = 10; in TEST() local
84 Shard(workers, &threads, total_elements, cost_per_unit, in TEST()
|
| D | work_sharder.h | 60 int64 cost_per_unit, std::function<void(int64, int64)> work);
97 static void Do(int64 total, int64 cost_per_unit, const Work& work,
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.h | 77 absl::optional<int64> cost_per_unit, in SchedulingParams() argument
80 cost_per_unit_(cost_per_unit), in SchedulingParams()
84 absl::optional<int64> cost_per_unit() const { return cost_per_unit_; } in cost_per_unit() function
168 void ParallelFor(int64 total, int64 cost_per_unit,
197 int64 total, int64 cost_per_unit,
|
| D | threadpool.cc | 140 if (scheduling_params.cost_per_unit().has_value()) { in ParallelFor()
141 ParallelFor(total, *scheduling_params.cost_per_unit(), fn); in ParallelFor()
203 void ThreadPool::ParallelFor(int64 total, int64 cost_per_unit, in ParallelFor() argument
208 total, Eigen::TensorOpCost(0, 0, cost_per_unit), in ParallelFor()
213 int64 total, int64 cost_per_unit, in ParallelForWithWorkerId() argument
219 Eigen::TensorOpCost(0, 0, cost_per_unit), in ParallelForWithWorkerId()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.h | 36 int64 cost_per_unit, ElementCopier copier, in ConcatCPUImpl() argument
51 const int64 estimated_total_cost = output->size() * cost_per_unit; in ConcatCPUImpl()
130 cost_per_unit, work); in ConcatCPUImpl()
|
| D | concat_lib_cpu.cc | 104 int64 cost_per_unit = EstimateBytesPerElement<T>(inputs); in ConcatCPU() local
105 ConcatCPUImpl<T>(d, inputs, cost_per_unit, MemCpyCopier<T>(), output); in ConcatCPU()
|
| D | reverse_op.cc | 76 const int64 cost_per_unit = input.NumElements() / N; in ReverseRows() local
78 Shard(worker_threads->num_threads, worker_threads->workers, N, cost_per_unit, in ReverseRows()
|
| D | matmul_op_impl.h | 237 const int64 cost_per_unit =
248 (batch_size == 1 || cost_per_unit > kMaxCostOuterParallelism)) {
259 cost_per_unit,
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | banded_triangular_solve_op.cc | 173 const int64 cost_per_unit = in Launch() local
195 cost_per_unit, in Launch()
|
| D | matrix_triangular_solve_op_impl.h | 125 const int64 cost_per_unit =
141 cost_per_unit,
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2020-014.md | 13 int64 cost_per_unit, std::function<void(int64, int64)> work);
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | threadpool_test.cc | 413 static void BM_ParallelFor(int iters, int total, int cost_per_unit) { in BM_ParallelFor() argument
420 pool.ParallelFor(total, cost_per_unit, in BM_ParallelFor()
|