| /external/tensorflow/tensorflow/core/lib/core/ |
| D | threadpool.cc | 96 void ParallelFor(int64 total, int64 cost_per_unit, in ParallelFor() function
179 void ThreadPool::ParallelFor(int64 total, int64 cost_per_unit, in ParallelFor() function in tensorflow::thread::ThreadPool
181 impl_->ParallelFor(total, cost_per_unit, std::move(fn)); in ParallelFor()
187 impl_->ParallelFor(total, cost_per_unit, in ParallelForWithWorkerId()
|
| D | threadpool_test.cc | 125 TEST(ThreadPool, ParallelFor) { in TEST() argument
137 pool.ParallelFor(kWorkItems, kHugeCost, in TEST()
247 pool.ParallelFor( in BM_ParallelFor()
|
| D | threadpool.h | 94 void ParallelFor(int64 total, int64 cost_per_unit,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/ |
| D | example_partitioner.cc | 51 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism, in UpdatePartitions()
84 boosted_trees::utils::ParallelFor(batch_size, desired_parallelism, in PartitionExamples()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | parallel_for.h | 25 void ParallelFor(int64 batch_size, int64 desired_parallelism,
|
| D | parallel_for.cc | 22 void ParallelFor(int64 batch_size, int64 desired_parallelism, in ParallelFor() function
|
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 45 workers->ParallelFor(total, cost_per_unit, work); in Shard()
|
| D | example_proto_fast_parsing.cc | 428 void ParallelFor(const std::function<void(size_t)>& f, size_t n, in ParallelFor() function
1069 ParallelFor(ProcessMiniBatch, num_minibatches, thread_pool); in FastParseExample()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
| D | multiple_additive_trees.cc | 86 boosted_trees::utils::ParallelFor(batch_size, worker_threads->NumThreads(), in Predict()
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/ |
| D | hyperplane_lsh_probes.cc | 124 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matrix_set_diag_op.cc | 134 thread_pool->ParallelFor(output.dimension(0), cost_per_batch, in Compute()
|
| D | matrix_band_part_op.cc | 188 thread_pool->ParallelFor(total_rows, row_cost, std::move(compute_shard)); in operator ()()
|
| D | dynamic_stitch_op.cc | 303 thread_pool->ParallelFor(indices_inputs.size(), bytes_processed, in Compute()
|
| D | split_op.cc | 192 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in operator ()()
|
| D | conv_ops_fused_image_transform.cc | 92 thread_pool->ParallelFor( in FusedConvParallelFor()
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | pmf_to_cdf_op.cc | 71 thread_pool->ParallelFor( in Compute()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 184 worker_threads.workers->ParallelFor(num_nonzero_elements, cost_per_unit, in Compute()
|
| /external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/ |
| D | reduce_slice_ops.cc | 90 thread_pool->ParallelFor(size, std::max(bound / dim2, (Index)1), work); \
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 443 boosted_trees::utils::ParallelFor( in Compute()
508 boosted_trees::utils::ParallelFor( in Compute()
|
| D | quantile_ops.cc | 292 boosted_trees::utils::ParallelFor( in Compute()
706 boosted_trees::utils::ParallelFor( in Compute()
|