Searched refs:set_thread_pool (Results 1 – 6 of 6) sorted by relevance
267 matmul.set_thread_pool(&device); in TEST()339 matmul.set_thread_pool(&device); in TEST()364 muladd.set_thread_pool(&device); in TEST()471 fn.set_thread_pool(&device); in TEST()499 fn.set_thread_pool(&device); in TEST()525 fn.set_thread_pool(&device); in TEST()539 fn.set_thread_pool(&device); in TEST()564 fn.set_thread_pool(&device); in TEST()594 fn.set_thread_pool(&device); in TEST()687 fn.set_thread_pool(&device); in TEST()
66 computation.set_thread_pool(&device); in TEST()78 computation.set_thread_pool(&device); in BM_NAME()
37 computation.set_thread_pool(&device);
82 model.set_thread_pool(&device); in TEST()
130 void set_thread_pool(const Eigen::ThreadPoolDevice* pool) { in set_thread_pool() function
228 matmul.set_thread_pool(&device);