Home
last modified time | relevance | path

Searched refs:tensorflow_cpu_worker_threads (Results 1 – 25 of 76) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc204 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
265 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
313 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
366 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
425 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
502 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
Dprediction_ops.cc170 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
264 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
384 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h50 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads() function
51 return underlying_->tensorflow_cpu_worker_threads(); in tensorflow_cpu_worker_threads()
/external/tensorflow/tensorflow/core/kernels/
Ddiag_op.cc141 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
161 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
Dmeta_support.cc91 return tf_context->device()->tensorflow_cpu_worker_threads()->num_threads; in GetWorkersCount()
155 auto& workers = *(tf_context->device()->tensorflow_cpu_worker_threads()); in QuantizedGemmImpl()
168 auto& workers = *(tf_context->device()->tensorflow_cpu_worker_threads()); in MultiThreadTransform1D()
Dsdca_ops.cc225 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()
283 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
Droll_op.cc86 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll()
214 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy()
Dnth_element_op.cc115 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
Dbincount_op.cc56 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
Dpopulation_count_op.cc117 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
Dpooling_ops_common.h184 *(context->device()->tensorflow_cpu_worker_threads()); in SpatialMaxPool()
431 *(context->device()->tensorflow_cpu_worker_threads());
601 *(context->device()->tensorflow_cpu_worker_threads());
Dmatrix_set_diag_op.cc132 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
Dsplit_op.cc144 context->device()->tensorflow_cpu_worker_threads()->num_threads; in operator ()()
192 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in operator ()()
Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
Dcompare_and_bitpack_op.cc159 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dreinterpret_string_to_float_op.cc65 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
Dmodel_ops.cc205 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
304 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dprediction_ops.cc55 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc288 context->device()->tensorflow_cpu_worker_threads()->workers; in DoCompute()
453 context->device()->tensorflow_cpu_worker_threads()->workers; in DoCompute()
Dquantile_ops.cc291 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
449 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
705 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
849 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.cc89 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in operator ()()
Dadjust_hsv_in_yiq_op.cc115 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()
/external/tensorflow/tensorflow/contrib/nearest_neighbor/kernels/
Dhyperplane_lsh_probes.cc124 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in Compute()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h136 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads() function
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops.cc117 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()
319 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()

1234