Home
last modified time | relevance | path

Searched refs:compute_pool (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_util.cc61 static thread::ThreadPool* compute_pool = InitComputePool(options); in ComputePool() local
62 return compute_pool; in ComputePool()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_env.h66 thread::ThreadPool* compute_pool = nullptr; member
Dgraph_mgr.cc144 worker_env_->compute_pool, cluster_flr)); in InitItem()
500 thread::ThreadPool* pool = worker_env_->compute_pool; in StartParallelExecutors()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/
Dgrpc_eager_service_impl.h48 env_->compute_pool->Schedule([this, call]() { \
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc271 worker_env_2->compute_pool->Schedule([this, step_id, parsed, done_cb]() { in AddRequest()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc249 worker_env_.compute_pool = ComputePool(sess_opts); in Init()
Dgrpc_worker_service.cc173 worker_->env()->compute_pool->Schedule(std::move(f)); in Schedule()