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.