TensorContractionThreadPool.h | 43 const Index num_threads; member in struct:Eigen::internal::packRhsAndKernelArg 205 int num_threads = TensorCostModel<ThreadPoolDevice>::numThreads( local 210 if (n == 1) num_threads = 1; 212 if (num_threads == 1) { 226 shard_by_col = shardByCol(m, n, num_threads); 230 blocking(k, m, n, num_threads); 237 blocking(k, m, n, num_threads); 257 gm = coarsenM(m, n, bm, bn, bk, gn, num_threads, shard_by_col); 258 gn = coarsenN(m, n, bm, bn, bk, gm, num_threads, shard_by_col); 260 gn = coarsenN(m, n, bm, bn, bk, gm, num_threads, shard_by_col) 796 const Index num_threads = this->m_device.numThreads(); local [all...] |