| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | sparse_cholesky_op.cc | 110 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| H A D | csr_sparse_matrix_to_dense_op.cc | 114 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| H A D | csr_sparse_matrix_to_sparse_tensor_op.cc | 127 auto worker_threads = *(c->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_matmul_op.cc | 57 auto& worker_threads = in GemmlowpMultiply() local
|
| H A D | stateless_random_gamma_op.cc | 183 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Fill() local
|
| H A D | depthwise_conv_grad_op.cc | 469 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
941 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
|
| H A D | random_poisson_op.cc | 271 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
|
| H A D | topk_op.cc | 231 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| H A D | multinomial_op.cc | 80 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
|
| H A D | reverse_op.cc | 77 auto worker_threads = context->device()->tensorflow_cpu_worker_threads(); in ReverseRows() local
|
| H A D | random_op.cc | 316 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| H A D | pooling_ops_common.h | 219 const DeviceBase::CpuWorkerThreads& worker_threads = in SpatialMaxPool() local
|
| H A D | depthwise_conv_op.cc | 246 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
|
| H A D | conv_grad_ops_3d.cc | 458 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
1061 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| H A D | ctc_decoder_ops.cc | 258 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
|
| H A D | random_binomial_op.cc | 181 auto worker_threads = *(ctx->device()->tensorflow_cpu_worker_threads()); in operator ()() local
|
| H A D | avgpooling_op.cc | 394 const DeviceBase::CpuWorkerThreads& worker_threads = in Compute() local
|
| H A D | quantized_conv_ops.cc | 436 auto& worker_threads = in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | linalg_ops_common.cc | 114 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Compute() local
|
| H A D | banded_triangular_solve_op.cc | 175 auto worker_threads = *(context->device()->tensorflow_cpu_worker_threads()); in Launch() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | rpcbench_test.cc | 42 static thread::ThreadPool* worker_threads; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A D | c_api_multi_client_test.cc | 86 std::vector<std::thread> worker_threads; in StartWorkers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | adjust_hue_op.cc | 215 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
|
| H A D | adjust_saturation_op.cc | 194 const DeviceBase::CpuWorkerThreads& worker_threads = in DoCompute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_qmatmul_op.cc | 456 const DeviceBase::CpuWorkerThreads& worker_threads = in GetBiasHandle() local
|