Home | History | Annotate | Download | only in products

Lines Matching refs:Index

74 template<typename Index> struct GemmParallelInfo
81 Index rhs_start;
82 Index rhs_length;
85 template<bool Condition, typename Functor, typename Index>
86 void parallelize_gemm(const Functor& func, Index rows, Index cols, bool transpose)
110 Index size = transpose ? cols : rows;
114 Index max_threads = std::max<Index>(1,size / 32);
117 Index threads = std::min<Index>(nbThreads(), max_threads);
128 GemmParallelInfo<Index>* info = new GemmParallelInfo<Index>[threads];
132 Index i = omp_get_thread_num();
134 Index actual_threads = omp_get_num_threads();
136 Index blockCols = (cols / actual_threads) & ~Index(0x3);
137 Index blockRows = (rows / actual_threads) & ~Index(0x7);
139 Index r0 = i*blockRows;
140 Index actualBlockRows = (i+1==actual_threads) ? rows-r0 : blockRows;
142 Index c0 = i*blockCols;
143 Index actualBlockCols = (i+1==actual_threads) ? cols-c0 : blockCols;