Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
Public Member Functions | Static Public Attributes | List of all members
cutlass::IdentityTensorMapFunc< Rank > Struct Template Reference

#include <tensor_ref.h>

Public Member Functions

CUTLASS_HOST_DEVICE Coord< Rank > operator() (Coord< Rank > const &coord) const
 

Static Public Attributes

static int const kStorageRank = Rank
 

Detailed Description

template<int Rank>
struct cutlass::IdentityTensorMapFunc< Rank >

Default mapping function from coordinates in a tensor's index space into the n-D array held in memory. Assumes StorageRank = Rank

Member Function Documentation

◆ operator()()

template<int Rank>
CUTLASS_HOST_DEVICE Coord<Rank> cutlass::IdentityTensorMapFunc< Rank >::operator() ( Coord< Rank > const &  coord) const
inline

Member Data Documentation

◆ kStorageRank

template<int Rank>
int const cutlass::IdentityTensorMapFunc< Rank >::kStorageRank = Rank
static

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