Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h79 thread::ThreadPool* tensorflow_device_thread_pool() override { in tensorflow_device_thread_pool() function
84 underlying_device_->tensorflow_device_thread_pool() != nullptr) { in tensorflow_device_thread_pool()
85 return Device::tensorflow_device_thread_pool(); in tensorflow_device_thread_pool()
87 return underlying_device_->tensorflow_device_thread_pool(); in tensorflow_device_thread_pool()
Dfunction.cc489 pool = device_->tensorflow_device_thread_pool(); in FunctionLibraryRuntimeImpl()
Dprocess_function_library_runtime.cc1175 thread::ThreadPool* pool = flr->device()->tensorflow_device_thread_pool(); in RunMultiDevice()
Ddirect_session.cc701 item.device->tensorflow_device_thread_pool(); in RunInternal()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h172 virtual thread::ThreadPool* tensorflow_device_thread_pool() { in tensorflow_device_thread_pool() function
Ddataset.h366 ctx->device()->tensorflow_device_thread_pool(); in Params()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc550 unit.device->tensorflow_device_thread_pool(); in StartParallelExecutors()