CUTLASS: tensor.h File Reference

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

• Main Page

• Modules

• Namespaces

• Classes

• Files

• File List

• File Members

• include

• cutlass

• layout

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.

---

Generated by 1.8.11