Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/data/
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()
Ddataset_utils.cc792 const auto num_threads = params.runner_threadpool_size; in CopyBatch()
889 return std::min(kAutotuneDefaultParallelism, ctx->runner_threadpool_size()); in GetAutotuneDefaultParallelism()
Droot_dataset.cc246 params.runner_threadpool_size = threadpool_size_; in CreateParams()
/external/tensorflow/tensorflow/core/framework/
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
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc271 params.runner_threadpool_size = pool->NumThreads(); in CreateParams()
505 params.runner_threadpool_size = dataset()->num_threads_; in GetNextInternal()
Dmap_and_batch_dataset_op.cc280 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
Dparse_example_dataset_op.cc427 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc368 const auto num_threads = ctx->runner_threadpool_size(); in CopyBatch()
Dparallel_batch_dataset_op.cc269 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
Dparallel_filter_dataset_op.cc199 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
Dparallel_map_dataset_op.cc306 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()