Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Ddataset.h350 runner_threadpool_size(ctx->runner_threadpool_size()), in Params()
368 runner_threadpool_size = thread_pool->NumThreads(); in Params()
372 runner_threadpool_size = kDefaultRunnerThreadpoolSize; in Params()
417 int32 runner_threadpool_size = 0; member
466 int32 runner_threadpool_size() { return params_.runner_threadpool_size; } in runner_threadpool_size() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc244 params.runner_threadpool_size = pool->NumThreads(); in CreateParams()
474 params.runner_threadpool_size = dataset()->num_threads_; in GetNextInternal()
Dmap_and_batch_dataset_op.cc219 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
275 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
Dparse_example_dataset_op.cc383 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
425 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.cc184 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
238 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
Dparallel_map_dataset_op.cc224 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
272 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()