Home
last modified time | relevance | path

Searched refs:set_tensorflow_device_thread_pool (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device.cc224 set_tensorflow_device_thread_pool(thread_pool_.get()); in Init()
231 set_tensorflow_device_thread_pool(thread_pool); in Init()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drenamed_device.cc64 set_tensorflow_device_thread_pool(underlying_threadpool_.get()); in RenamedDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h283 void set_tensorflow_device_thread_pool(thread::ThreadPool* thread_pool) { in set_tensorflow_device_thread_pool() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc531 set_tensorflow_device_thread_pool(thread_pool_.get()); in Init()
538 set_tensorflow_device_thread_pool(thread_pool); in Init()