CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Helper to perform for-each operation.
#include <tensor_foreach.h>
Public Member Functions | |
TensorForEachHelper (Func &func, Coord< Rank > const &extent, Coord< Rank > &coord) | |
Constructor for general rank. More... | |
Static Public Attributes | |
static int const | kActiveRank = Rank - RankRemaining - 1 |
Index of the active rank. More... | |
|
inline |
|
static |