Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h260 void set_tensorflow_device_thread_pool(thread::ThreadPool* thread_pool) { in set_tensorflow_device_thread_pool() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc431 set_tensorflow_device_thread_pool(thread_pool_.get()); in Init()
435 set_tensorflow_device_thread_pool(thread_pool); in Init()