Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Drun_handler.cc85 explicit Impl(int num_inter_op_threads) in Impl() argument
88 Env::Default(), ThreadOptions(), "inter_op", num_inter_op_threads)), in Impl()
98 num_inter_op_threads); in Impl()
99 int kStealDomainSize = std::min(6, num_inter_op_threads); in Impl()
101 for (int i = 0; i < num_inter_op_threads; ++i) { in Impl()
103 if (steal_end + kStealDomainSize < num_inter_op_threads) { in Impl()
107 steal_end = num_inter_op_threads; in Impl()
258 RunHandlerPool::RunHandlerPool(int num_inter_op_threads) in RunHandlerPool() argument
259 : impl_(new Impl(num_inter_op_threads)) {} in RunHandlerPool()
Drun_handler.h48 explicit RunHandlerPool(int num_inter_op_threads);