cutlass::IdentityTensorLayout< Rank > Class Template Reference

`#include <tensor_ref.h>`

## Public Types

using | Index = int32_t
| Index type used for coordinates. More...
using | LongIndex = int64_t
| Long index type used for offsets. More...
using | TensorCoord = Coord< kRank, Index >
| Logical coordinate. More...
using | Stride = Coord< kStrideRank, Index >
| Stride vector. More...

## Public Member Functions

CUTLASS_HOST_DEVICE | IdentityTensorLayout (Stride const &stride=Stride())
CUTLASS_HOST_DEVICE LongIndex | operator() (Coord< Rank > const &coord) const
| Returns the offset of a coordinate in linear memory. More...
CUTLASS_HOST_DEVICE Stride | stride () const
| Returns the stride of the layout. More...
CUTLASS_HOST_DEVICE Stride & | stride ()
| Returns the stride of the layout. More...
CUTLASS_HOST_DEVICE LongIndex | capacity (TensorCoord const &size) const
| Compute the number of contiguous elements needed to store a tensor with the given size. More...

## Static Public Attributes

static int const | kRank = Rank
| Logical rank of tensor. More...
static int const | kStrideRank = Rank
| Rank of stride vector. More...

## Detailed Description

### template<int Rank>
class cutlass::IdentityTensorLayout< Rank >

Default layout function from coordinates in a tensor's index space into the n-D array held in memory.

All layout functions must define at least the members shown in IdentityTensorLayout<>.

## Member Typedef Documentation

template<int Rank>

using cutlass::IdentityTensorLayout< Rank >::Index = int32_t

template<int Rank>

using cutlass::IdentityTensorLayout< Rank >::LongIndex = int64_t

template<int Rank>

using cutlass::IdentityTensorLayout< Rank >::Stride = Coord<kStrideRank, Index>

template<int Rank>

using cutlass::IdentityTensorLayout< Rank >::TensorCoord = Coord<kRank, Index>

## Constructor & Destructor Documentation

template<int Rank>

CUTLASS_HOST_DEVICE cutlass::IdentityTensorLayout< Rank >::IdentityTensorLayout | ( | Stride const & | stride = `Stride()` | ) |

inline

## Member Function Documentation

template<int Rank>

CUTLASS_HOST_DEVICE LongIndex cutlass::IdentityTensorLayout< Rank >::capacity | ( | TensorCoord const & | size | ) | const

inline

template<int Rank>

CUTLASS_HOST_DEVICE LongIndex cutlass::IdentityTensorLayout< Rank >::operator() | ( | Coord< Rank > const & | coord | ) | const

inline

template<int Rank>

CUTLASS_HOST_DEVICE Stride cutlass::IdentityTensorLayout< Rank >::stride | ( |  | ) | const

inline

template<int Rank>

CUTLASS_HOST_DEVICE Stride& cutlass::IdentityTensorLayout< Rank >::stride | ( |  | ) |

inline

## Member Data Documentation

template<int Rank>

int const cutlass::IdentityTensorLayout< Rank >::kRank = Rank

static

template<int Rank>

int const cutlass::IdentityTensorLayout< Rank >::kStrideRank = Rank

static

The documentation for this class was generated from the following file:

- tensor_ref.h