Home
last modified time | relevance | path

Searched refs:worker_threads (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc203 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
205 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
264 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
266 Shard(worker_threads.num_threads, worker_threads.workers, num_streams, in Compute()
312 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
314 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
365 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
367 Shard(worker_threads.num_threads, worker_threads.workers, num_streams, in Compute()
424 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
426 Shard(worker_threads.num_threads, worker_threads.workers, num_streams, in Compute()
[all …]
Dprediction_ops.cc169 thread::ThreadPool* const worker_threads = in Compute() local
171 Shard(worker_threads->NumThreads(), worker_threads, batch_size, in Compute()
263 thread::ThreadPool* const worker_threads = in Compute() local
265 Shard(worker_threads->NumThreads(), worker_threads, batch_size, in Compute()
383 thread::ThreadPool* const worker_threads = in Compute() local
385 Shard(worker_threads->NumThreads(), worker_threads, batch_size, in Compute()
/external/pdfium/third_party/libopenjpeg20/
Dthread.c606 opj_worker_thread_t* worker_threads; member
696 tp->worker_threads = (opj_worker_thread_t*) opj_calloc((size_t)num_threads, in opj_thread_pool_setup()
698 if (tp->worker_threads == NULL) { in opj_thread_pool_setup()
704 tp->worker_threads[i].tp = tp; in opj_thread_pool_setup()
706 tp->worker_threads[i].mutex = opj_mutex_create(); in opj_thread_pool_setup()
707 if (tp->worker_threads[i].mutex == NULL) { in opj_thread_pool_setup()
713 tp->worker_threads[i].cond = opj_cond_create(); in opj_thread_pool_setup()
714 if (tp->worker_threads[i].cond == NULL) { in opj_thread_pool_setup()
715 opj_mutex_destroy(tp->worker_threads[i].mutex); in opj_thread_pool_setup()
721 tp->worker_threads[i].marked_as_waiting = OPJ_FALSE; in opj_thread_pool_setup()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc371 const DeviceBase::CpuWorkerThreads& worker_threads = in Initialize() local
374 worker_threads, num_examples, num_sparse_features, weights, in Initialize()
378 worker_threads, num_examples, num_dense_features, weights, in Initialize()
381 worker_threads, num_examples, num_sparse_features, num_dense_features, in Initialize()
387 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation() argument
461 Shard(worker_threads.num_threads, worker_threads.workers, num_sparse_features, in CreateSparseFeatureRepresentation()
467 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation() argument
493 Shard(worker_threads.num_threads, worker_threads.workers, num_dense_features, in CreateDenseFeatureRepresentation()
499 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample() argument
541 Shard(worker_threads.num_threads, worker_threads.workers, num_examples, in ComputeSquaredNormPerExample()
Ddiag_op.cc141 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
142 Shard(worker_threads.num_threads, worker_threads.workers, size, 5 * size, in operator ()()
161 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
162 Shard(worker_threads.num_threads, worker_threads.workers, size, 5, in operator ()()
Dsdca_ops.cc224 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
227 Shard(worker_threads.num_threads, worker_threads.workers, in DoCompute()
282 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
284 Shard(worker_threads.num_threads, worker_threads.workers, in Compute()
Droll_op.cc86 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll() local
89 Shard(worker_threads->num_threads, worker_threads->workers, num_elements, in DoRoll()
214 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy() local
219 Shard(worker_threads->num_threads, worker_threads->workers, total_work, in DoRollWithMemcpy()
Dconcat_lib_cpu.h48 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl() local
49 int num_threads = std::min(4, worker_threads->num_threads); in ConcatCPUImpl()
128 Shard(worker_threads->num_threads, worker_threads->workers, output->size(), in ConcatCPUImpl()
Dnth_element_op.cc115 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
119 Shard(worker_threads.num_threads, worker_threads.workers, num_rows, in operator ()()
Dpopulation_count_op.cc117 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()() local
118 Shard(worker_threads.num_threads, worker_threads.workers, total_shards, in operator ()()
Dpooling_ops_common.h183 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPool() local
253 Shard(worker_threads.num_threads, worker_threads.workers, in SpatialMaxPool()
430 const DeviceBase::CpuWorkerThreads& worker_threads =
500 Shard(worker_threads.num_threads, worker_threads.workers,
600 const DeviceBase::CpuWorkerThreads& worker_threads =
602 Shard(worker_threads.num_threads, worker_threads.workers,
Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
123 Shard(worker_threads.num_threads, worker_threads.workers, in Compute()
Dcompare_and_bitpack_op.cc159 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()() local
160 Shard(worker_threads.num_threads, worker_threads.workers, total_shards, in operator ()()
Dgather_functor.h55 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopies() local
107 Shard(worker_threads->num_threads, worker_threads->workers, in HandleCopies()
/external/autotest/client/site_tests/power_Thermal/
Dpower_Thermal.py105 thread_count = len(self.worker_threads)
108 self.worker_threads.append(new_thread)
318 self.worker_threads = []
349 logging.info('stopping %d thread(s)' % len(self.worker_threads))
351 while self.worker_threads:
352 t = self.worker_threads.pop()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dreinterpret_string_to_float_op.cc65 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
66 int num_threads = worker_threads->num_threads; in Compute()
76 Shard(num_threads, worker_threads->workers, num_data, 100, work); in Compute()
Dmodel_ops.cc205 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
206 int num_threads = worker_threads->num_threads; in Compute()
219 Shard(num_threads, worker_threads->workers, num_data, costPerTraverse, in Compute()
304 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
305 int num_threads = worker_threads->num_threads; in Compute()
314 Shard(num_threads, worker_threads->workers, num_data, costPerTraverse, in Compute()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dprediction_ops.cc55 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
56 const int32 num_threads = worker_threads->num_threads; in Compute()
73 Shard(num_threads, worker_threads->workers, batch_size, cost_per_traverse, in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees.cc28 tensorflow::thread::ThreadPool* const worker_threads, in Predict() argument
86 boosted_trees::utils::ParallelFor(batch_size, worker_threads->NumThreads(), in Predict()
87 worker_threads, update_predictions); in Predict()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc287 thread::ThreadPool* const worker_threads = in DoCompute() local
307 worker_threads, output_predictions, in DoCompute()
312 batch_features, worker_threads, output_predictions, in DoCompute()
452 thread::ThreadPool* const worker_threads = in DoCompute() local
455 tree_config, batch_features, worker_threads->NumThreads(), in DoCompute()
456 worker_threads, partition_ids_t->vec<int32>().data()); in DoCompute()
Dquantile_ops.cc290 thread::ThreadPool* const worker_threads = in Compute() local
293 resource_handle_list.size(), worker_threads->NumThreads(), in Compute()
294 worker_threads, in Compute()
448 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
450 Shard(worker_threads.num_threads, worker_threads.workers, num_features, in Compute()
704 thread::ThreadPool* const worker_threads = in Compute() local
707 resource_handle_list.size(), worker_threads->NumThreads(), in Compute()
708 worker_threads, in Compute()
848 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
850 Shard(worker_threads.num_threads, worker_threads.workers, num_features, in Compute()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.cc89 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in operator ()() local
102 Shard(worker_threads->num_threads, worker_threads->workers, in operator ()()
Dadjust_hsv_in_yiq_op.cc114 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
116 Shard(worker_threads.num_threads, worker_threads.workers, channel_count, in DoCompute()
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops.cc117 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
118 ::tensorflow::Shard(worker_threads.num_threads, worker_threads.workers, in operator ()()
319 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
322 ::tensorflow::Shard(worker_threads.num_threads, worker_threads.workers, in operator ()()
/external/tensorflow/tensorflow/contrib/seq2seq/kernels/
Dbeam_search_ops.cc160 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
161 Shard(worker_threads.num_threads, worker_threads.workers, in operator ()()

123