Back to Cutlass

CUTLASS: tensor_view.h File Reference

docs/tensor__view_8h.html

4.4.22.1 KB
Original Source

| | CUTLASS

CUDA Templates for Linear Algebra Subroutines and Solvers |

Classes | Namespaces | Functions

tensor_view.h File Reference

Defines a structure containing strides and a pointer to tensor data. More...

#include <cmath>
#include "cutlass/cutlass.h"
#include "cutlass/tensor_ref.h"

Include dependency graph for tensor_view.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

|

Classes

| | class | cutlass::TensorView< Element_, Layout_ > | | |

|

Namespaces

| | | cutlass | | |

|

Functions

| | template<typename Element , typename Layout > | | CUTLASS_HOST_DEVICE TensorView< Element, Layout > | cutlass::make_TensorView (Element *ptr, Layout const &layout, typename Layout::TensorCoord const &extent) | | | Constructs a TensorRef, deducing types from arguments. More...
| | |

Detailed Description

TensorView is derived from TensorRef and contributes bounds to the tensor's index space. Thus, it is a complete mathematical object and may be used in tensor algorithms. It is decoupled from data storage and is therefore lightweight and may be embedded in larger tensor objects or memory structures.

See cutlass/tensor_ref.h for more details about the mapping of the logical tensor index space to linear memory.


Generated by 1.8.11