Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc207 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
267 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
334 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
381 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
433 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
492 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
570 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
Dprediction_ops.cc184 thread::ThreadPool* const worker_threads = in Compute() local
274 thread::ThreadPool* const worker_threads = in Compute() local
391 thread::ThreadPool* const worker_threads = in Compute() local
/external/rust/crates/tokio-macros/src/
Dentry.rs30 worker_threads: Option<usize>, field
45 worker_threads: Option<(usize, Span)>, field
89 let worker_threads = parse_int(worker_threads, span, "worker_threads")?; in set_worker_threads() localVariable
119 let worker_threads = match (flavor, self.worker_threads) { in build() localVariable
/external/tensorflow/tensorflow/core/kernels/
Ddiag_op.cc142 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
163 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dsdca_internal.cc405 const DeviceBase::CpuWorkerThreads& worker_threads = in Initialize() local
421 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateSparseFeatureRepresentation()
508 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in CreateDenseFeatureRepresentation()
540 const DeviceBase::CpuWorkerThreads& worker_threads, const int num_examples, in ComputeSquaredNormPerExample()
Dsdca_ops.cc225 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
283 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
Droll_op.cc168 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll() local
296 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy() local
Dnth_element_op.cc116 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dgather_functor.h55 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopies() local
Dconcat_lib_cpu.h53 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl() local
Dpopulation_count_op.cc121 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dgather_functor_batched.h56 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopiesBatched() local
Dsobol_op.cc160 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
Dquantized_matmul_op.cc54 auto& worker_threads = in GemmlowpMultiply() local
Dparameterized_truncated_normal_op.cc69 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
332 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
Dstateless_random_gamma_op.cc183 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Fill() local
Drandom_poisson_op.cc271 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
/external/rust/crates/tokio/src/runtime/
Dbuilder.rs56 worker_threads: Option<usize>, field
225 pub fn worker_threads(&mut self, val: usize) -> &mut Self { in worker_threads() method
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_ordering_amd_op.cc94 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute() local
Dsparse_cholesky_op.cc110 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_kernels.cc145 Expected<Eigen::ThreadPoolInterface*> worker_threads = in CompileImpl() local
287 Expected<Eigen::ThreadPoolInterface*> worker_threads = in ExecuteImpl() local
/external/crosvm/common/cros_asyncv2/src/blocking/
Dpool.rs28 worker_threads: Slab<JoinHandle<()>>, field
284 let mut worker_threads = mem::take(&mut state.worker_threads); in shutdown() localVariable
/external/crosvm/cros_async/src/blocking/
Dpool.rs30 worker_threads: Slab<JoinHandle<()>>, field
286 let mut worker_threads = mem::replace(&mut state.worker_threads, Slab::new()); in shutdown() localVariable
/external/crosvm/common/cros_async/src/blocking/
Dpool.rs30 worker_threads: Slab<JoinHandle<()>>, field
286 let mut worker_threads = mem::replace(&mut state.worker_threads, Slab::new()); in shutdown() localVariable
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local

123