Home
last modified time | relevance | path

Searched refs:set_tensorflow_cpu_worker_threads (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dsingle_threaded_cpu_device.cc53 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in SingleThreadedCpuDevice()
H A Drenamed_device.cc63 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in RenamedDevice()
H A Dlocal_device.cc157 set_tensorflow_cpu_worker_threads(&tp_info->eigen_worker_threads_); in LocalDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Devaluation_utils.cc41 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in DeviceSimple()
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc54 set_tensorflow_cpu_worker_threads(&eigen_worker_threads_); in SimpleDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h139 void set_tensorflow_cpu_worker_threads(CpuWorkerThreads* t) { in set_tensorflow_cpu_worker_threads() function