Defines a view into a logical tensor.
|
| typedef TensorRef< Storage_, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ > | Base |
| | Base tensor reference. More...
|
| |
| typedef TensorRef< typename platform::remove_const< Storage_ >::type const, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ > | ConstTensorRef |
| | Tensor reference to of constant value. More...
|
| |
| typedef Base | TensorRef |
| | Base tensor reference. More...
|
| |
| typedef Base::Storage | Storage |
| | Storage type. More...
|
| |
| typedef Base::Index | Index |
| | Index type. More...
|
| |
| typedef TensorRef::TensorCoord | TensorCoord |
| | Coordinate in logical tensor space. More...
|
| |
| typedef TensorRef::StorageCoord | StorageCoord |
| | Coordinate in storage n-D array. More...
|
| |
| typedef TensorRef::StrideVector | StrideVector |
| |
| typedef TensorView< typename platform::remove_const< Storage >::type const, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ > | ConstTensorView |
| | TensorView of constant value. More...
|
| |
| typedef TensorCoord | Coord_t |
| | Coordinate in logical tensor space. More...
|
| |
| typedef Base::LongIndex | Offset_t |
| | Type used to compute the offset of an element to the base of a tensor. More...
|
| |
| typedef TensorRef | TensorRef_t |
| | Base class. More...
|
| |
| typedef TensorRef::ConstTensorRef | ConstTensorRef_t |
| | TensorRef to const-valued type. More...
|
| |
| typedef Storage_ | Storage |
| | Data type of individual access. More...
|
| |
| typedef MapFunc_ | MapFunc |
| | Mapping function from logical coordinate to internal n-D array. More...
|
| |
| typedef Index_ | Index |
| | Index type. More...
|
| |
| typedef LongIndex_ | LongIndex |
| | Typically, strides in memory can be very large. More...
|
| |
| typedef Coord< kRank > | TensorCoord |
| | Coordinate in logical tensor space. More...
|
| |
| typedef Coord< kStorageRank > | StorageCoord |
| | Coordinate in storage n-D array. More...
|
| |
| typedef Coord< kStorageRank - 1 > | StrideVector |
| |
| typedef TensorRef< typename platform::remove_const< Storage >::type const, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ > | ConstTensorRef |
| | Tensor reference to of constant value. More...
|
| |
| typedef TensorCoord | Coord_t |
| | Coordinate in logical tensor space. More...
|
| |
|
| CUTLASS_HOST_DEVICE | TensorView () |
| | Default constructor. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorView (Base const &_ref, TensorCoord const &_size) |
| | Constructs a TensorView from a TensorRef and size. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorView (Storage *ptr, StrideVector const &stride, TensorCoord const &size) |
| | Constructs a TensorView from a pointer, a stride vector, and size. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorView (Storage *ptr, StorageCoord const &stride, TensorCoord const &size) |
| | Constructs a TensorView from a pointer, a stride vector, and size. More...
|
| |
| CUTLASS_HOST_DEVICE void | reset (Base const &_ref=Base(), TensorCoord const &_size=TensorCoord()) |
| | Updates the reference and size of a Tensor_view object. More...
|
| |
| CUTLASS_HOST_DEVICE TensorCoord const & | size () const |
| | Accesses the size. More...
|
| |
| CUTLASS_HOST_DEVICE Index | size (int dim) const |
| | Accesses the size. More...
|
| |
| CUTLASS_HOST_DEVICE TensorView & | operator= (TensorView const &_tensor) |
| | Assigns the Tensor_view. More...
|
| |
| CUTLASS_HOST_DEVICE bool | contains (TensorCoord const &coord) const |
| | Determines whether a location is within a tensor. More...
|
| |
| CUTLASS_HOST_DEVICE TensorRef | ref () const |
| | Returns a TensorRef pointing to the first element of the tensor. More...
|
| |
| CUTLASS_HOST_DEVICE ConstTensorRef | const_ref () const |
| | Returns a TensorRef pointing to the first element of the tensor. More...
|
| |
| CUTLASS_HOST_DEVICE TensorView | subview (TensorCoord const &location, TensorCoord size) const |
| | Returns a Tensor_view given location and size quantities. More...
|
| |
| CUTLASS_HOST_DEVICE size_t | capacity () const |
| | Returns the number of scalar elements needed to store tensor. More...
|
| |
| CUTLASS_HOST_DEVICE TensorView | operator+ (TensorCoord const &b) const |
| | Returns a TensorView offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE TensorView & | operator+= (TensorCoord const &b) |
| | Returns a TensorRef offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE TensorView | operator- (TensorCoord const &b) const |
| | Returns a TensorRef offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE TensorView & | operator-= (TensorCoord const &b) |
| | Returns a TensorRef offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorRef (Storage *ptr=nullptr) |
| | Helper for 1-D memory. All higher ranks are projected onto the fastest changing rank. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorRef (Storage *ptr, Index ldm) |
| | Helper to construct from a pointer and single stride element for 2-D pitch linear memory. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorRef (Storage *ptr, StrideVector const &stride) |
| | Constructs from a single pointer and stride vector. More...
|
| |
| CUTLASS_HOST_DEVICE | TensorRef (Storage *ptr, StorageCoord const &stride) |
| |
| CUTLASS_HOST_DEVICE | TensorRef (TensorRef< typename platform::remove_const< Storage >::type, kRank, MapFunc, kStorageRank, Index, LongIndex > const &ref) |
| | Enables conversion from TensorRef of non-const type. More...
|
| |
| CUTLASS_HOST_DEVICE ConstTensorRef | const_ref () const |
| | Returns a reference to constant-valued tensor. More...
|
| |
| CUTLASS_HOST_DEVICE void | reset (Storage *ptr=nullptr) |
| | Updates only the pointer. More...
|
| |
| CUTLASS_HOST_DEVICE void | reset (Storage *ptr, StorageCoord const &stride) |
| | Updates the pointer, stride, and location within a TensorRef. More...
|
| |
| CUTLASS_HOST_DEVICE bool | good () const |
| | Returns true if the TensorRef may be safely accessed. More...
|
| |
| CUTLASS_HOST_DEVICE Storage * | data () const |
| | Returns the pointer to referenced data. More...
|
| |
| CUTLASS_HOST_DEVICE StorageCoord | stride () const |
| | Returns the stride of the tensor. More...
|
| |
| CUTLASS_HOST_DEVICE Index | stride (int dim) const |
| | Returns the stride of the tensor in the given dimension. More...
|
| |
| CUTLASS_HOST_DEVICE Index | leading_dim (int idx=0) const |
| | Returns the maximum stride element as the 'leading dimension'. More...
|
| |
| CUTLASS_HOST_DEVICE StorageCoord | map (TensorCoord const &coord) const |
| | Maps a logical coordinate to an n-D array in memory. More...
|
| |
| CUTLASS_HOST_DEVICE LongIndex | offset (TensorCoord const &coord) const |
| | Computes the offset of an index from the origin of the tensor. More...
|
| |
| CUTLASS_HOST_DEVICE Storage & | at (TensorCoord const &coord) const |
| | Returns a reference to the element at a given Coord. More...
|
| |
| CUTLASS_HOST_DEVICE Storage & | at (LongIndex idx) const |
| | Returns a reference to the element at a given linear index. More...
|
| |
| CUTLASS_HOST_DEVICE Storage & | operator[] (TensorCoord const &coord) const |
| | Returns a reference to the element at a given Coord. More...
|
| |
| CUTLASS_HOST_DEVICE Storage & | operator[] (LongIndex idx) const |
| | Returns a reference to the element at a given linear index. More...
|
| |
| CUTLASS_HOST_DEVICE TensorRef & | add_pointer_offset (LongIndex delta) |
| | Adds an offset to each pointer. More...
|
| |
| CUTLASS_HOST_DEVICE TensorRef | operator+ (TensorCoord const &b) const |
| | Returns a TensorRef offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE TensorRef & | operator+= (TensorCoord const &b) |
| | Returns a TensorRef offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE TensorRef | operator- (TensorCoord const &b) const |
| | Returns a TensorRef offset by a given amount. More...
|
| |
| CUTLASS_HOST_DEVICE TensorRef & | operator-= (TensorCoord const &b) |
| | Returns a TensorRef offset by a given amount. More...
|
| |