Searched refs:inter_op_parallelism_threads (Results 1 – 16 of 16) sorted by relevance
75 int32_t inter_op_parallelism_threads = in InitComputePool() local76 options.config.inter_op_parallelism_threads(); in InitComputePool()77 if (inter_op_parallelism_threads == 0) { in InitComputePool()78 inter_op_parallelism_threads = DefaultNumInterOpThreads(); in InitComputePool()81 Env::Default(), ThreadOptions(), "Compute", inter_op_parallelism_threads, in InitComputePool()126 const int32_t inter_op = options.config.inter_op_parallelism_threads(); in NumInterOpThreadsFromSessionOptions()
345 if (options_.config.inter_op_parallelism_threads() < 0 || in DirectSession()346 (options_.config.inter_op_parallelism_threads() == 0 && in DirectSession()
131 return context.context().inter_op_parallelism_threads144 context.context().inter_op_parallelism_threads = num_threads
144 context.context().inter_op_parallelism_threads)
1078 config.inter_op_parallelism_threads = self._inter_op_parallelism_threads1819 def inter_op_parallelism_threads(self): member in Context1820 return self.config.inter_op_parallelism_threads1822 @inter_op_parallelism_threads.setter1823 def inter_op_parallelism_threads(self, num_threads): member in Context
126 session_config.inter_op_parallelism_threads += 1825 session_config.inter_op_parallelism_threads)835 session_config.inter_op_parallelism_threads)
136 if worker_config.inter_op_parallelism_threads < num_workers + 1:137 worker_config.inter_op_parallelism_threads = num_workers + 1
260 worker_config.inter_op_parallelism_threads = 4 # max num_workers + 1
122 inter_op_parallelism_threads=inter_op, use_per_session_threads=True)
19 name: "inter_op_parallelism_threads"
370 // inter_op_parallelism_threads is not configured), but still partitioning455 // inter_op_parallelism_threads available in each process.463 int32 inter_op_parallelism_threads = 5; field473 // inter_op_parallelism_threads.
713 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
577 config.inter_op_parallelism_threads = 1607 config.inter_op_parallelism_threads = 1637 config.inter_op_parallelism_threads = 1698 config.inter_op_parallelism_threads = 1741 config.inter_op_parallelism_threads = 1
179 inter_op_parallelism_threads=1, use_per_session_threads=True)
1437 config.inter_op_parallelism_threads = -1
2648 ::int32_t inter_op_parallelism_threads() const; in Swap()6309 inline ::int32_t ConfigProto::inter_op_parallelism_threads() const { in inter_op_parallelism_threads() function