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