Home
last modified time | relevance | path

Searched defs:worker_threads (Results 1 – 25 of 59) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc203 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
264 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
312 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
365 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
424 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
501 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
Dprediction_ops.cc169 thread::ThreadPool* const worker_threads = in Compute() local
263 thread::ThreadPool* const worker_threads = in Compute() local
383 thread::ThreadPool* const worker_threads = in Compute() local
/external/tensorflow/tensorflow/core/kernels/
Ddiag_op.cc141 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
161 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dsdca_internal.cc371 const DeviceBase::CpuWorkerThreads& worker_threads = in Initialize() local
387 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation()
467 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation()
499 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample()
Dsdca_ops.cc224 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
282 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
Dnth_element_op.cc115 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Droll_op.cc86 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll() local
214 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy() local
Dgather_functor.h55 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopies() local
Dpopulation_count_op.cc117 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dcompare_and_bitpack_op.cc159 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
Dconcat_lib_cpu.h48 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl() local
Dquantized_matmul_op.cc54 auto& worker_threads = in GemmlowpMultiply() local
Drandom_poisson_op.cc267 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dlinalg_ops_common.cc108 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
/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
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees.cc28 tensorflow::thread::ThreadPool* const worker_threads, in Predict()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dprediction_ops.cc55 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.cc89 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in operator ()() local
Dadjust_hsv_in_yiq_op.cc114 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc287 thread::ThreadPool* const worker_threads = in DoCompute() local
452 thread::ThreadPool* const worker_threads = in DoCompute() local
Dquantile_ops.cc290 thread::ThreadPool* const worker_threads = in Compute() local
448 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
704 thread::ThreadPool* const worker_threads = in Compute() local
848 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
/external/tensorflow/tensorflow/contrib/seq2seq/kernels/
Dbeam_search_ops.cc160 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops.cc117 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
319 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dmasked_matmul_ops.cc171 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local

123