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