Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
H A 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
H A Dsearchsorted_op.cc55 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
84 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
H A Dsdca_ops.cc225 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
283 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
H A 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
H A Dparameterized_truncated_normal_op.cc70 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
333 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
H A Dsparse_cross_op.cc801 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
872 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
947 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute() local
H A Ddeep_conv2d.cc490 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
566 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
1165 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
H A Dmaxpooling_op.cc97 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolWithArgMaxHelper() local
593 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPoolGradGrad() local
1032 const DeviceBase::CpuWorkerThreads& worker_threads = in launch() local
H A Dnth_element_op.cc116 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()() local
H A Dconcat_lib_cpu.h53 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl() local
H A Dgather_functor.h55 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopies() local
H A Dgather_functor_batched.h56 auto* worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopiesBatched() local
H A Dpopulation_count_op.cc121 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()() local
H A Dsobol_op.cc167 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
/aosp_15_r20/external/rust/android-crates-io/crates/tokio-macros/src/
Dentry.rs55 worker_threads: Option<usize>, field
74 worker_threads: Option<(usize, Span)>, field
122 let worker_threads = parse_int(worker_threads, span, "worker_threads")?; in set_worker_threads() localVariable
180 let worker_threads = match (flavor, self.worker_threads) { in build() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_kernels.cc145 Expected<Eigen::ThreadPoolInterface*> worker_threads = in CompileImpl() local
287 Expected<Eigen::ThreadPoolInterface*> worker_threads = in ExecuteImpl() local
H A Dtf_jitrt_kernels.cc304 Expected<Eigen::ThreadPoolInterface*> worker_threads = in CompileImpl() local
679 Expected<Eigen::ThreadPoolInterface*> worker_threads = in ExecuteImpl() local
/aosp_15_r20/external/crosvm/cros_async/src/blocking/
H A Dpool.rs31 worker_threads: Slab<JoinHandle<()>>, field
294 let mut worker_threads = mem::replace(&mut state.worker_threads, Slab::new()); in shutdown() localVariable
/aosp_15_r20/external/rust/android-crates-io/crates/tokio/src/runtime/
Dbuilder.rs65 worker_threads: Option<usize>, field
412 pub fn worker_threads(&mut self, val: usize) -> &mut Self { in worker_threads() method
/aosp_15_r20/external/rust/android-crates-io/crates/vhost-user-backend/src/
Dhandler.rs90 worker_threads: Vec<thread::JoinHandle<VringEpollResult<()>>>, field
113 let mut worker_threads = Vec::new(); in new() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_ordering_amd_op.cc94 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute() local
H A Dmat_mul_op.cc301 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in SparseDenseMatMulWithoutTransposedLHS() local
351 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in SparseDenseMatMulWithTransposedLHS() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dcrop_and_resize_op.cc342 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()() local
559 const DeviceBase::CpuWorkerThreads& worker_threads = in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dlu_op.cc122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local

123