Home
last modified time | relevance | path

Searched refs:device_thread_pool (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc549 thread::ThreadPool* device_thread_pool = in StartParallelExecutors() local
551 if (!device_thread_pool) { in StartParallelExecutors()
555 std::bind(&thread::ThreadPool::Schedule, device_thread_pool, _1); in StartParallelExecutors()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc700 thread::ThreadPool* device_thread_pool = in RunInternal() local
704 if (!device_thread_pool) { in RunInternal()
707 args->runner = [device_thread_pool](Executor::Args::Closure c) { in RunInternal()
708 device_thread_pool->Schedule(std::move(c)); in RunInternal()