Searched refs:scheduling_params (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.cc | 136 const SchedulingParams& scheduling_params, in ParallelFor() argument
138 switch (scheduling_params.strategy()) { in ParallelFor()
140 if (scheduling_params.cost_per_unit().has_value()) { in ParallelFor()
141 ParallelFor(total, *scheduling_params.cost_per_unit(), fn); in ParallelFor()
146 if (scheduling_params.block_size().has_value()) { in ParallelFor()
148 total, *scheduling_params.block_size(), fn); in ParallelFor()
232 int64 total, const SchedulingParams& scheduling_params, in ParallelForWithWorkerId() argument
234 ParallelFor(total, scheduling_params, [this, &fn](int64 start, int64 limit) { in ParallelForWithWorkerId()
|
| D | threadpool.h | 173 void ParallelFor(int64 total, const SchedulingParams& scheduling_params,
203 int64 total, const SchedulingParams& scheduling_params,
|