Searched refs:tensorflow_cpu_worker_threads (Results 1 – 25 of 68) sorted by relevance
123
49 const CpuWorkerThreads* tensorflow_cpu_worker_threads() const override { in tensorflow_cpu_worker_threads() function50 return underlying_->tensorflow_cpu_worker_threads(); in tensorflow_cpu_worker_threads()
141 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()161 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
91 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()
30 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in GetCpuCastFromBfloat()
31 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in GetCpuCastFromFloat()
216 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()271 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
85 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll()213 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy()
115 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
56 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
117 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
184 *(context->device()->tensorflow_cpu_worker_threads()); in SpatialMaxPool()431 *(context->device()->tensorflow_cpu_worker_threads());601 *(context->device()->tensorflow_cpu_worker_threads());
169 context->device()->tensorflow_cpu_worker_threads()->num_threads; in Compute()212 context->device()->tensorflow_cpu_worker_threads()->workers, in Compute()
132 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
227 context->device()->tensorflow_cpu_worker_threads()->num_threads; in Compute()270 context->device()->tensorflow_cpu_worker_threads()->workers, in Compute()
159 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
53 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in HandleCopies()
65 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
205 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()304 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
272 context->device()->tensorflow_cpu_worker_threads()->workers; in DoCompute()415 context->device()->tensorflow_cpu_worker_threads()->workers; in DoCompute()
284 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()441 *context->device()->tensorflow_cpu_worker_threads(); in Compute()697 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()840 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
89 auto worker_threads = ctx->device()->tensorflow_cpu_worker_threads(); in operator ()()
115 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()
124 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in Compute()
119 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads() function
117 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()319 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()