Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
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... | |