| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| parallel_for.h | 25 void ParallelFor(int64 batch_size, int64 desired_parallelism,
|
| parallel_for.cc | 22 void ParallelFor(int64 batch_size, int64 desired_parallelism,
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| threadpool.cc | 95 void ParallelFor(int64 total, int64 cost_per_unit,
100 device.parallelFor(
128 void ThreadPool::ParallelFor(int64 total, int64 cost_per_unit,
130 impl_->ParallelFor(total, cost_per_unit, std::move(fn));
136 impl_->ParallelFor(total, cost_per_unit,
138 // ParallelFor may use the current thread to do some
|
| threadpool.h | 62 // ParallelFor shards the "total" units of work assuming each unit of work
72 void ParallelFor(int64 total, int64 cost_per_unit,
75 // Shards the "total" units of work. For more details, see "ParallelFor".
|
| threadpool_test.cc | 60 TEST(ThreadPool, ParallelFor) {
61 // Make ParallelFor use as many threads as possible.
71 pool.ParallelFor(kWorkItems, kHugeCost, [&work](int64 begin, int64 end) {
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/ |
| example_partitioner.cc | 51 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism,
84 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism,
|
| /external/tensorflow/tensorflow/core/util/ |
| work_sharder.cc | 35 workers->ParallelFor(total, cost_per_unit, work);
|
| example_proto_fast_parsing.cc | 406 void ParallelFor(const std::function<void(size_t)>& f, size_t n,
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
| multiple_additive_trees.cc | 76 boosted_trees::utils::ParallelFor(batch_size, worker_threads->NumThreads(),
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| hyperplane_lsh_probes.cc | 124 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor(
|
| /external/tensorflow/tensorflow/core/kernels/ |
| matrix_set_diag_op.cc | 134 thread_pool->ParallelFor(output.dimension(0), cost_per_batch,
|
| matrix_band_part_op.cc | 188 thread_pool->ParallelFor(total_rows, row_cost, std::move(compute_shard));
|
| dynamic_stitch_op.cc | 291 thread_pool->ParallelFor(indices_inputs.size(), bytes_processed,
|
| conv_ops_fused.cc | 91 thread_pool->ParallelFor(
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| masked_matmul_ops.cc | 184 worker_threads.workers->ParallelFor(num_nonzero_elements, cost_per_unit,
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| reduce_slice_ops.cc | 85 thread_pool->ParallelFor(size, std::max(bound / dim2, (Index)1), work); \
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| quantile_ops.cc | 285 boosted_trees::utils::ParallelFor(
698 boosted_trees::utils::ParallelFor(
[all...] |
| stats_accumulator_ops.cc | 443 boosted_trees::utils::ParallelFor(
507 boosted_trees::utils::ParallelFor(
[all...] |
