| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | renamed_device.h | 46 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | meta_support.cc | 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()
|
| H A D | diag_op.cc | 142 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
163 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
|
| H A D | sdca_ops.cc | 226 *context->device()->tensorflow_cpu_worker_threads(); in DoCompute()
284 *context->device()->tensorflow_cpu_worker_threads(); in Compute()
|
| H A D | searchsorted_op.cc | 55 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
84 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
|
| H A D | bincount_op.cc | 59 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()
|
| H A D | split_op.cc | 141 context->device()->tensorflow_cpu_worker_threads()->num_threads; in operator ()()
189 context->device()->tensorflow_cpu_worker_threads()->workers->ParallelFor( in operator ()()
|
| H A D | dynamic_stitch_op.cc | 290 c->device()->tensorflow_cpu_worker_threads()->num_threads > 1) { in Compute()
292 c->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
|
| H A D | pooling_ops_common.h | 220 *(context->device()->tensorflow_cpu_worker_threads()); in SpatialMaxPool()
485 *(context->device()->tensorflow_cpu_worker_threads());
659 *(context->device()->tensorflow_cpu_worker_threads());
|
| H A D | roll_op.cc | 168 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRoll()
296 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in DoRollWithMemcpy()
|
| H A D | split_v_op.cc | 235 context->device()->tensorflow_cpu_worker_threads()->num_threads; in operator ()()
281 context->device()->tensorflow_cpu_worker_threads()->workers, in operator ()()
|
| H A D | nth_element_op.cc | 116 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in operator ()()
|
| H A D | sparse_cross_op.cc | 801 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
872 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
947 auto* worker_threads = context->device()->tensorflow_cpu_worker_threads(); in Compute()
|
| H A D | population_count_op.cc | 121 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in operator ()()
|
| H A D | concat_lib_cpu.h | 53 auto worker_threads = d->tensorflow_cpu_worker_threads(); in ConcatCPUImpl()
|
| H A D | sobol_op.cc | 168 *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
|
| H A D | isotonic_regression_op.cc | 155 context->device()->tensorflow_cpu_worker_threads()->workers; in solve_pava_batch()
|
| H A D | random_op_cpu.h | 165 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads());
|
| H A D | deep_conv2d.cc | 490 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()
566 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()
1165 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_threadpool.h | 68 ->tensorflow_cpu_worker_threads()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | sparse_ordering_amd_op.cc | 94 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_diag_op.cc | 401 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
461 context->device()->tensorflow_cpu_worker_threads()->workers; in Compute()
|
| H A D | lu_op.cc | 122 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/sleep/ |
| H A D | sleep_kernel.cc | 64 auto thread_pool = ctx->device()->tensorflow_cpu_worker_threads()->workers; in ComputeAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.h | 143 virtual const CpuWorkerThreads* tensorflow_cpu_worker_threads() const { in tensorflow_cpu_worker_threads() function
|