Home
last modified time | relevance | path

Searched refs:tensorflow_cpu_worker_threads (Results 1 – 25 of 77) 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/
Ddiag_op.cc142 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
163 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()
Dsearchsorted_op.cc55 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
84 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
Dsdca_ops.cc226 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()
284 *context->device()->tensorflow_cpu_worker_threads(); 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 ()()
Ddynamic_stitch_op.cc290 c->device()->tensorflow_cpu_worker_threads()->num_threads > 1) { in Compute()
292 c->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
Dbincount_op.cc59 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
104 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
178 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
Dconcat_lib_cpu.h53 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl()
Dpopulation_count_op.cc121 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
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()
Dsobol_op.cc168 *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
Dpooling_ops_common.h220 *(context->device()->tensorflow_cpu_worker_threads()); in SpatialMaxPool()
485 *(context->device()->tensorflow_cpu_worker_threads());
659 *(context->device()->tensorflow_cpu_worker_threads());
Disotonic_regression_op.cc155 context->device()->tensorflow_cpu_worker_threads()->workers; in solve_pava_batch()
Drandom_op_cpu.h165 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads());
Dsplit_v_op.cc235 context->device()->tensorflow_cpu_worker_threads()->num_threads; in operator ()()
281 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/util/
Dmkl_threadpool.h68 ->tensorflow_cpu_worker_threads()
/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/examples/custom_ops_doc/sleep/
Dsleep_kernel.cc64 auto thread_pool = ctx->device()->tensorflow_cpu_worker_threads()->workers; in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
Dmatrix_diag_op.cc401 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
461 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h143 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc91 device->tensorflow_cpu_worker_threads()->num_threads); in BuildXlaCompilationCache()

1234