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/common_runtime/
Drenamed_device.h46 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads() function
48 return Device::tensorflow_cpu_worker_threads(); in tensorflow_cpu_worker_threads()
50 return underlying_device_->tensorflow_cpu_worker_threads(); in tensorflow_cpu_worker_threads()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc205 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
264 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
331 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
378 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
430 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
489 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
566 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
Dprediction_ops.cc185 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
275 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
392 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
/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.cc226 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()
284 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
Dbincount_op.cc57 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
102 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
143 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
Dnth_element_op.cc116 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
Dsplit_op.cc141 context->device()->tensorflow_cpu_worker_threads()->num_threads; in operator ()()
189 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in operator ()()
Dpopulation_count_op.cc117 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
Dconcat_lib_cpu.h53 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl()
Dsobol_op.cc161 *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
Droll_op.cc168 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll()
296 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy()
Drandom_op_cpu.h165 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads());
Disotonic_regression_op.cc155 context->device()->tensorflow_cpu_worker_threads()->workers; in solve_pava_batch()
Dpooling_ops_common.h222 *(context->device()->tensorflow_cpu_worker_threads()); in SpatialMaxPool()
483 *(context->device()->tensorflow_cpu_worker_threads());
653 *(context->device()->tensorflow_cpu_worker_threads());
Dcompare_and_bitpack_op.cc159 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
Dsplit_v_op.cc203 context->device()->tensorflow_cpu_worker_threads()->num_threads; in operator ()()
246 context->device()->tensorflow_cpu_worker_threads()->workers, in operator ()()
Dgather_functor_batched.h56 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopiesBatched()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dprediction_ops.cc54 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_ordering_amd_op.cc94 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_threadpool.h66 ->tensorflow_cpu_worker_threads() in MklDnnThreadPool()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc62 device->tensorflow_cpu_worker_threads()->num_threads); in BuildXlaCompilationCache()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h139 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads() function

1234