Home
last modified time | relevance | path

Searched refs:runner_threadpool_size (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.h285 int64 runner_threadpool_size; member
290 runner_threadpool_size = ctx->runner_threadpool_size(); in CopyBatchParams()
296 runner_threadpool_size = GetRunnerThreadpoolSizeFromOpKernelContext(ctx); in CopyBatchParams()
H A Ddataset_utils.cc791 const auto num_threads = params.runner_threadpool_size; in CopyBatch()
888 return std::min(kAutotuneDefaultParallelism, ctx->runner_threadpool_size()); in GetAutotuneDefaultParallelism()
H A Droot_dataset.cc246 params.runner_threadpool_size = threadpool_size_; in CreateParams()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset.h408 runner_threadpool_size(ctx->runner_threadpool_size()), in Params()
425 runner_threadpool_size = GetRunnerThreadpoolSizeFromOpKernelContext(ctx); in Params()
483 int32 runner_threadpool_size = 0; member
546 int32 runner_threadpool_size() { return params_.runner_threadpool_size; } in runner_threadpool_size() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dthreadpool_dataset_op.cc271 params.runner_threadpool_size = pool->NumThreads(); in CreateParams()
505 params.runner_threadpool_size = dataset()->num_threads_; in GetNextInternal()
H A Dmap_and_batch_dataset_op.cc280 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
H A Dparse_example_dataset_op.cc427 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dpadded_batch_dataset_op.cc368 const auto num_threads = ctx->runner_threadpool_size(); in CopyBatch()
H A Dparallel_batch_dataset_op.cc269 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
H A Dparallel_filter_dataset_op.cc199 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
H A Dparallel_map_dataset_op.cc306 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()