Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drenamed_device.h80 thread::ThreadPool* tensorflow_device_thread_pool() override { in tensorflow_device_thread_pool() function
85 underlying_device_->tensorflow_device_thread_pool() != nullptr) { in tensorflow_device_thread_pool()
86 return Device::tensorflow_device_thread_pool(); in tensorflow_device_thread_pool()
88 return underlying_device_->tensorflow_device_thread_pool(); in tensorflow_device_thread_pool()
H A Dprocess_function_library_runtime.cc1313 thread::ThreadPool* pool = flr->device()->tensorflow_device_thread_pool(); in RunMultiDeviceSync()
1438 thread::ThreadPool* pool = flr->device()->tensorflow_device_thread_pool(); in RunMultiDeviceAsync()
H A Dfunction.cc499 pool = device_->tensorflow_device_thread_pool(); in FunctionLibraryRuntimeImpl()
H A Ddirect_session.cc717 item.device->tensorflow_device_thread_pool(); in RunInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h176 virtual thread::ThreadPool* tensorflow_device_thread_pool() { in tensorflow_device_thread_pool() function
H A Ddataset.cc392 ctx->device()->tensorflow_device_thread_pool(); in GetRunnerThreadpoolSizeFromOpKernelContext()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc794 thread::ThreadPool* tensorflow_device_thread_pool() override { in tensorflow_device_thread_pool() function in tensorflow::tfd::__anond671884e0711::DeviceWithCustomAllocator
795 return device_->tensorflow_device_thread_pool(); in tensorflow_device_thread_pool()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dgraph_mgr.cc559 unit.device->tensorflow_device_thread_pool(); in StartParallelExecutors()