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...