|
CUTLASS_HOST_DEVICE | TileCoord () |
| Default ctor. More...
|
|
CUTLASS_HOST_DEVICE | TileCoord (Coord< 3, Index > const &coord) |
| Constructs from Coord<3> and infers coord[kC] = 0. More...
|
|
CUTLASS_HOST_DEVICE | TileCoord (Coord< 4, Index > const &coord) |
| Constructs from Coord<4> More...
|
|
CUTLASS_HOST_DEVICE | TileCoord (Index coord[4]) |
| Constructs from an array of coordinate elements. More...
|
|
CUTLASS_HOST_DEVICE | TileCoord (Index d, Index h, Index w, Index c) |
| Helper to construct from a row and column. More...
|
|
CUTLASS_HOST_DEVICE Index const & | d () const |
| Returns the D element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index & | d () |
| Returns the D element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index const & | h () const |
| Returns the H element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index & | h () |
| Returns the H element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index const & | w () const |
| Returns the W element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index & | w () |
| Returns the W element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index const & | c () const |
| Returns the Celement of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Index & | c () |
| Returns the C element of the coordinate. More...
|
|
CUTLASS_HOST_DEVICE Coord< 2 > | hw () const |
| Gets H and W dimensions as a Coord<2> More...
|
|
CUTLASS_HOST_DEVICE Coord< 3 > | hwc () const |
| Gets H, W, and C dimensions as a Coord<3> More...
|
|
CUTLASS_HOST_DEVICE Coord< 3 > | dhw () const |
| Gets D, H, and W dimensions as a Coord<3> More...
|
|
CUTLASS_HOST_DEVICE TileCoord | operator+ (Base const &b) const |
| Element-wise addition. More...
|
|
CUTLASS_HOST_DEVICE TileCoord | operator- (Base const &b) const |
| Element-wise subtraction. More...
|
|
CUTLASS_HOST_DEVICE TileCoord | operator* (Base const &b) const |
| Element-wise multiplication. More...
|
|
CUTLASS_HOST_DEVICE TileCoord | operator/ (Base const &b) const |
| Element-wise division. More...
|
|
CUTLASS_HOST_DEVICE TileCoord & | operator+= (Base const &b) |
| In-place addition. More...
|
|
CUTLASS_HOST_DEVICE TileCoord & | operator-= (Base const &b) |
| In-place subtraction. More...
|
|
CUTLASS_HOST_DEVICE TileCoord & | operator*= (Base const &b) |
| In-place multiplication. More...
|
|
CUTLASS_HOST_DEVICE TileCoord & | operator/= (Base const &b) |
| In-place division. More...
|
|
CUTLASS_HOST_DEVICE | Coord (Index value=0) |
| Default ctor initializes uniformly. More...
|
|
CUTLASS_HOST_DEVICE | Coord (Index _idx[]) |
| Constructs from an array of integers. More...
|
|
CUTLASS_HOST_DEVICE | Coord (Coord< kRank > const &coord) |
| Constructs from an array of integers. More...
|
|
CUTLASS_HOST_DEVICE Coord< Slice > | slice (int start=0, Index identity=0) const |
|
CUTLASS_HOST_DEVICE | operator bool () const |
| Returns true if Coord is non-zero. More...
|
|
CUTLASS_HOST_DEVICE bool | operator! () const |
| Returns true if Coord is uniformly zero. More...
|
|
CUTLASS_HOST_DEVICE Coord | operator+ (Coord const &b) const |
| Element-wise addition. More...
|
|
CUTLASS_HOST_DEVICE Coord | operator- (Coord const &b) const |
| Element-wise subtraction. More...
|
|
CUTLASS_HOST_DEVICE Coord | operator* (Coord const &b) const |
| Element-wise multiplication. More...
|
|
CUTLASS_HOST_DEVICE Coord | operator/ (Coord const &b) const |
| Element-wise division. More...
|
|
CUTLASS_HOST_DEVICE Coord & | operator+= (Coord const &b) |
| In-place addition. More...
|
|
CUTLASS_HOST_DEVICE Coord & | operator-= (Coord const &b) |
| In-place subtraction. More...
|
|
CUTLASS_HOST_DEVICE Coord & | operator*= (Coord const &b) |
| In-place multiplication. More...
|
|
CUTLASS_HOST_DEVICE Coord & | operator/= (Coord const &b) |
| In-place division. More...
|
|
CUTLASS_HOST_DEVICE Index & | operator[] (int dim) |
| Member access operator. More...
|
|
CUTLASS_HOST_DEVICE Index const & | operator[] (int dim) const |
| Member access operator. More...
|
|
CUTLASS_HOST_DEVICE T | dot (Coord const &b, T sum) const |
| Computes the dot product of two Coord instances. More...
|
|
CUTLASS_HOST_DEVICE T | dot (Coord const &b) const |
| Computes the dot product of two Coord instances. More...
|
|
CUTLASS_HOST_DEVICE Index & | at () |
| Gets the index of a given Coord element. More...
|
|
CUTLASS_HOST_DEVICE Index & | at (int dim) |
| Access via index; may limit unrolling potential. More...
|
|
CUTLASS_HOST_DEVICE Index const & | at () const |
| Gets the index of a given Coord element. More...
|
|
CUTLASS_HOST_DEVICE Index const & | at (int dim) const |
| Access via index; may limit unrolling potential. More...
|
|
CUTLASS_HOST_DEVICE bool | operator== (Coord< kRank > const &b) const |
| Determines if two Coord<> objects are equal. More...
|
|
CUTLASS_HOST_DEVICE bool | operator!= (Coord< kRank > const &b) const |
| Not equal. More...
|
|
CUTLASS_HOST_DEVICE Coord & | clamp (Coord< kRank > const &max, Coord< kRank > const &min=Coord< kRank >()) |
| Clamps a coordinate to a range specified by maximum and minimum values. More...
|
|
CUTLASS_HOST_DEVICE Index | count () const |
| Returns the product of all elements. More...
|
|
CUTLASS_HOST_DEVICE bool | operator< (Coord< kRank > const &b) const |
| Less than operator. More...
|
|
CUTLASS_HOST_DEVICE bool | operator<= (Coord< kRank > const &b) const |
| Less than or equals operator. More...
|
|
template<typename Index_ = int>
struct cutlass::TileCoord< Index_ >
TileCoord wraps Coord<4, int> to provide a helper for accessing named dimensions. Classes expecting a coordinate in the rank=4 index space of a CUTLASS tile structure should use TileCoord.