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.cc508 thread::ThreadPool* device_thread_pool = in StartParallelExecutors() local
510 if (!device_thread_pool) { in StartParallelExecutors()
514 std::bind(&thread::ThreadPool::Schedule, device_thread_pool, _1); in StartParallelExecutors()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc643 thread::ThreadPool* device_thread_pool = in RunInternal() local
647 if (!device_thread_pool) { in RunInternal()
650 args.runner = [this, device_thread_pool](Executor::Args::Closure c) { in RunInternal()
651 SchedClosure(device_thread_pool, std::move(c)); in RunInternal()