Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Ddataset.h291 runner_threadpool_size(ctx->runner_threadpool_size()), in Params()
308 runner_threadpool_size = thread_pool->NumThreads(); in Params()
310 runner_threadpool_size = port::NumSchedulableCPUs(); in Params()
340 int32 runner_threadpool_size = 0; member
398 int32 runner_threadpool_size() { return params_.runner_threadpool_size; } in runner_threadpool_size() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dthreadpool_dataset_op.cc220 params.runner_threadpool_size = pool->NumThreads(); in CreateParams()
413 params.runner_threadpool_size = dataset()->num_threads_; in GetNextInternal()
Dmap_and_batch_dataset_op.cc283 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
319 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
Dnuma_map_and_batch_dataset_op.cc224 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
267 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_iterator.cc78 num_parallel_calls_->value = ctx->runner_threadpool_size(); in Initialize()
110 /*max=*/ctx->runner_threadpool_size())}); in CreateNode()