| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.cc | 135 void ThreadPool::ParallelFor(int64 total, in ParallelFor() function in tensorflow::thread::ThreadPool
141 ParallelFor(total, *scheduling_params.cost_per_unit(), fn); in ParallelFor()
158 ParallelFor(total, in TransformRangeConcurrently()
203 void ThreadPool::ParallelFor(int64 total, int64 cost_per_unit, in ParallelFor() function in tensorflow::thread::ThreadPool
234 ParallelFor(total, scheduling_params, [this, &fn](int64 start, int64 limit) { in ParallelForWithWorkerId()
|
| D | threadpool.h | 168 void ParallelFor(int64 total, int64 cost_per_unit,
173 void ParallelFor(int64 total, const SchedulingParams& scheduling_params,
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | threadpool_test.cc | 74 threads->ParallelFor( in RunWithFixedBlockSize()
261 TEST(ThreadPool, ParallelFor) { in TEST() argument
273 pool.ParallelFor(kWorkItems, kHugeCost, in TEST()
299 pool.ParallelFor( in TEST()
420 pool.ParallelFor(total, cost_per_unit, in BM_ParallelFor()
|
| /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 | 503 void ParallelFor(const std::function<void(size_t)>& f, size_t n, in ParallelFor() function
1233 ParallelFor(ProcessMiniBatch, num_minibatches, thread_pool); in FastParseExample()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | isotonic_regression_op.cc | 157 thread_pool->ParallelFor( in solve_pava_batch()
|
| D | dynamic_stitch_op.cc | 304 thread_pool->ParallelFor(indices_inputs.size(), bytes_processed, in Compute()
|
| D | split_op.cc | 189 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in operator ()()
|
| D | ragged_cross_op.cc | 371 thread_pool->ParallelFor(batch_size, cost_per_batch, do_work); in Compute()
|
| D | conv_ops_fused_image_transform.cc | 93 thread_pool->ParallelFor( in FusedConvParallelFor()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 380 thread_pool->ParallelFor(num_batches, cost_per_batch, in Compute()
440 thread_pool->ParallelFor(num_batches, cost_per_batch, in Compute()
|
| D | matrix_band_part_op.cc | 190 thread_pool->ParallelFor(total_rows, row_cost, std::move(compute_shard)); in operator ()()
|
| D | matrix_set_diag_op.cc | 252 thread_pool->ParallelFor(output.dimension(0), cost_per_batch, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 298 worker_threads.workers->ParallelFor( in SparseDenseMatMulWithoutTransposedLHS()
|