| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| reinterpret_string_to_float_op.cc | 65 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); variable
66 int num_threads = worker_threads->num_threads;
76 Shard(num_threads, worker_threads->workers, num_data, 100, work);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| concat_lib_cpu.h | 45 auto worker_threads = d->tensorflow_cpu_worker_threads(); local
46 int num_threads = std::min(4, worker_threads->num_threads);
125 Shard(worker_threads->num_threads, worker_threads->workers, output->size(),
|
| diag_op.cc | 141 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); local
142 Shard(worker_threads.num_threads, worker_threads.workers, size, 5 * size,
161 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); local
162 Shard(worker_threads.num_threads, worker_threads.workers, size, 5,
|
| gather_functor.h | 53 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); local
105 Shard(worker_threads->num_threads, worker_threads->workers,
|
| multinomial_op.cc | 65 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); local
123 Shard(worker_threads.num_threads, worker_threads.workers, batch_size, cost,
|
| nth_element_op.cc | 115 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); local
119 Shard(worker_threads.num_threads, worker_threads.workers, num_rows,
|
| population_count_op.cc | 117 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); local
118 Shard(worker_threads.num_threads, worker_threads.workers, total_shards,
|
| roll_op.cc | 85 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); local
88 Shard(worker_threads->num_threads, worker_threads->workers, num_elements,
213 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); local
218 Shard(worker_threads->num_threads, worker_threads->workers, total_work,
|
| adjust_hue_op.cc | 215 const DeviceBase::CpuWorkerThreads& worker_threads = variable
217 Shard(worker_threads.num_threads, worker_threads.workers, channel_count,
|
| adjust_saturation_op.cc | 192 const DeviceBase::CpuWorkerThreads& worker_threads = variable
194 Shard(worker_threads.num_threads, worker_threads.workers, channel_count,
|
| compare_and_bitpack_op.cc | 159 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); local
160 Shard(worker_threads.num_threads, worker_threads.workers, total_shards,
|
| linalg_ops_common.cc | 108 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); local
109 Shard(worker_threads.num_threads, worker_threads.workers,
|
| parameterized_truncated_normal_op.cc | 60 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); local
232 Shard(worker_threads.num_threads, worker_threads.workers, num_batches,
|
| random_poisson_op.cc | 267 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); local
268 Shard(worker_threads.num_threads, worker_threads.workers,
|
| reverse_op.cc | 80 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); local
81 Shard(worker_threads->num_threads, worker_threads->workers, N, cost_per_unit,
|
| sdca_ops.cc | 215 const DeviceBase::CpuWorkerThreads& worker_threads = local
218 Shard(worker_threads.num_threads, worker_threads.workers,
270 const DeviceBase::CpuWorkerThreads& worker_threads = variable
272 Shard(worker_threads.num_threads, worker_threads.workers,
|
| topk_op.cc | 222 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); local
223 Shard(worker_threads.num_threads, worker_threads.workers, num_rows,
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| masked_matmul_ops.cc | 171 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); variable
184 worker_threads.workers->ParallelFor(num_nonzero_elements, cost_per_unit,
|
| wals_solver_ops.cc | 146 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); variable
255 worker_threads.workers->Schedule(std::bind(work, shards[i]));
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| adjust_hsv_in_yiq_op.cc | 114 const DeviceBase::CpuWorkerThreads& worker_threads = variable
116 Shard(worker_threads.num_threads, worker_threads.workers, channel_count,
|
| segmentation_ops.cc | 89 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); local
102 Shard(worker_threads->num_threads, worker_threads->workers,
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| beam_search_ops.cc | 160 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); local
161 Shard(worker_threads.num_threads, worker_threads.workers,
|
| /external/tensorflow/tensorflow/contrib/resampler/kernels/ |
| resampler_ops.cc | 117 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); local
118 ::tensorflow::Shard(worker_threads.num_threads, worker_threads.workers,
319 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); local
322 ::tensorflow::Shard(worker_threads.num_threads, worker_threads.workers,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| rpcbench_test.cc | 42 static thread::ThreadPool* worker_threads; member in namespace:tensorflow
67 worker_threads = new thread::ThreadPool(Env::Default(), "worker_threads", n);
69 worker_threads->Schedule([worker_idx, n, num_cpus, num_gpus, &port] {
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| prediction_ops.cc | 271 thread::ThreadPool* const worker_threads = member in class:tensorflow::boosted_trees::GradientTreesPredictionOp::thread
291 worker_threads, output_predictions);
295 batch_features, worker_threads, output_predictions);
414 thread::ThreadPool* const worker_threads = member in class:tensorflow::boosted_trees::GradientTreesPartitionExamplesOp::thread
417 tree_config, batch_features, worker_threads->NumThreads(),
418 worker_threads, partition_ids_t->vec<int32>().data());
|
