HomeSort by relevance Sort by last modified time
    Searched full:unroll_times (Results 1 - 1 of 1) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorReductionCuda.h 383 const int unroll_times = 16; local
384 eigen_assert(NumPerThread % unroll_times == 0);
409 for (Index j = 0; j < NumPerThread; j += unroll_times) {
410 const Index last_col = col_begin + blockDim.x * (j + unroll_times - 1);
420 for (int k = 0; k < unroll_times; ++k) {
453 const int unroll_times = 16; local
454 eigen_assert(NumPerThread % unroll_times == 0);
455 eigen_assert(unroll_times % 2 == 0);
486 for (Index j = 0; j < NumPerThread; j += unroll_times) {
487 const Index last_col = col_begin + blockDim.x * (j + unroll_times - 1) * 2
    [all...]

Completed in 179 milliseconds