CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
Classes | Namespaces
tensor.h File Reference

Defines layout functions used by TensorRef and derived classes for common 4-D and 5-D tensor formats. More...

#include "assert.h"
#include "cutlass/cutlass.h"
#include "cutlass/fast_math.h"
#include "cutlass/layout/matrix.h"
#include "cutlass/coord.h"
#include "cutlass/tensor_coord.h"
Include dependency graph for tensor.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  cutlass::layout::TensorNHWC
 Mapping function for 4-D NHWC tensors. More...
 
class  cutlass::layout::TensorNCHW
 Mapping function for 4-D NCHW tensors. More...
 
class  cutlass::layout::TensorNCxHWx< Interleave >
 Mapping function for 4-D NC/xHWx tensors. More...
 
class  cutlass::layout::TensorCxRSKx< Interleave >
 Mapping function for 4-D CxRSKx tensors. More...
 

Namespaces

 cutlass
 
 cutlass::layout
 

Detailed Description

Layout functions map logical coordinates to linear memory. They often require additional data to describe strides between elements.

Layout functions must implement all members in the public interface of IdentityTensorLayout<> defined in cutlass/tensor_ref.h.