#include <TensorContractionMapper.h>
|
| enum | { DirectOffsets = CoeffLoader<Tensor, Tensor::RawAccess, MakePointer_>::DirectOffsets
} |
| |
|
| EIGEN_DEVICE_FUNC | SimpleTensorContractionMapper (const Tensor &tensor, const nocontract_t &nocontract_strides, const nocontract_t &ij_strides, const contract_t &contract_strides, const contract_t &k_strides) |
| |
| EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void | offsetBuffer (typename Tensor::Index offset) |
| |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | prefetch (Index) |
| |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar | operator() (Index row) const |
| |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar | operator() (Index row, Index col) const |
| |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | computeIndex (Index row, Index col) const |
| |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE IndexPair< Index > | computeIndexPair (Index row, Index col, const Index distance) const |
| |
| EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index | firstAligned (Index size) const |
| |
| EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index | stride () const |
| |
| const CoeffLoader< Tensor, Tensor::RawAccess, MakePointer_ > & | tensor () const |
| |
| const nocontract_t & | nocontract_strides () const |
| |
| const nocontract_t & | ij_strides () const |
| |
| const contract_t & | contract_strides () const |
| |
| const contract_t & | k_strides () const |
| |
◆ anonymous enum
◆ SimpleTensorContractionMapper()
| EIGEN_DEVICE_FUNC Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::SimpleTensorContractionMapper |
( |
const Tensor & |
tensor, |
|
|
const nocontract_t & |
nocontract_strides, |
|
|
const nocontract_t & |
ij_strides, |
|
|
const contract_t & |
contract_strides, |
|
|
const contract_t & |
k_strides |
|
) |
| |
|
inline |
◆ computeIndex()
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::computeIndex |
( |
Index |
row, |
|
|
Index |
col |
|
) |
| const |
|
inline |
◆ computeIndexPair()
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE IndexPair< Index > Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::computeIndexPair |
( |
Index |
row, |
|
|
Index |
col, |
|
|
const Index |
distance |
|
) |
| const |
|
inline |
◆ contract_strides()
◆ firstAligned()
◆ ij_strides()
◆ k_strides()
◆ nocontract_strides()
◆ offsetBuffer()
◆ operator()() [1/2]
◆ operator()() [2/2]
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::operator() |
( |
Index |
row, |
|
|
Index |
col |
|
) |
| const |
|
inline |
◆ prefetch()
◆ stride()
◆ tensor()
| const CoeffLoader< Tensor, Tensor::RawAccess, MakePointer_ > & Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::tensor |
( |
| ) |
const |
|
inline |
◆ m_contract_strides
◆ m_ij_strides
◆ m_k_strides
◆ m_nocontract_strides
◆ m_tensor
| CoeffLoader<Tensor, Tensor::RawAccess, MakePointer_> Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::m_tensor |
|
protected |
The documentation for this class was generated from the following file: