Cutlass CUDA Templates for Linear Algebra Subroutines and Solvers

Classes | Namespaces | Functions
tensor_ref_collection.h File Reference

Introduces TensorRefCollection concept and defines TensorRefBatch and TensorRefArray. More...

`#include "cutlass/tensor_ref.h"`

Go to the source code of this file.

## Classes
struct | cutlass::TensorRefBatchStrided< Storage_, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ >
class | cutlass::TensorRefBatchStrided< Storage_, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ >::ConstIterator
| Constant iterator over tensors implied by TensorRefBatchStrided. More...
struct | cutlass::TensorRefArray< Storage_, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ >
class | cutlass::TensorRefArray< Storage_, Rank_, MapFunc_, StorageRank_, Index_, LongIndex_ >::ConstIterator
| TensorRefIterator over TensorRef objects in TensorRefArray. More...

## Namespaces
| cutlass

## Functions
template<typename TensorRef_ >
CUTLASS_HOST_DEVICE TensorRefBatchStrided< typename TensorRef_::Storage, TensorRef_::kRank, typename TensorRef_::MapFunc, TensorRef_::kStorageGrank, typename TensorRef_::Index, typename TensorRef_::LongIndex > | cutlass::make_TensorRefBatchStrided (TensorRef_ const &ref, typename TensorRef_::LongIndex batch_stride=0)
| Helper to construct a TensorRefBatchStrided<> object using type deduction. More...