#include <TensorContractionMapper.h>
Inheritance diagram for Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >:
Public Types | |
| typedef unpacket_traits< Packet >::half | HalfPacket |
| typedef Tensor::PacketReturnType | Packet |
| typedef SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment > | ParentMapper |
 Public Types inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment > | |
| enum | { DirectOffsets = CoeffLoader<Tensor, Tensor::RawAccess>::DirectOffsets } |
Public Member Functions | |
| EIGEN_DEVICE_FUNC | BaseTensorContractionMapper (const Tensor &tensor, const nocontract_t &nocontract_strides, const nocontract_t &ij_strides, const contract_t &contract_strides, const contract_t &k_strides) |
| template<int AlignmentType> | |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE HalfPacket | loadHalfPacket (Index i, Index j) const |
| template<int AlignmentType> | |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet | loadPacket (Index i, Index j) const |
| Public Member Functions inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment > | |
| 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 void | offsetBuffer (typename Tensor::Index offset) |
| 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 void | prefetch (Index) |
| 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 Index | stride () const |
Additional Inherited Members | |
| Protected Attributes inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment > | |
| const contract_t | m_contract_strides |
| const nocontract_t | m_ij_strides |
| const contract_t | m_k_strides |
| const nocontract_t | m_nocontract_strides |
| CoeffLoader< Tensor, Tensor::RawAccess > | m_tensor |
Detailed Description
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
class Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >
Definition at line 222 of file TensorContractionMapper.h.
Member Typedef Documentation
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
| typedef unpacket_traits<Packet>::half Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::HalfPacket |
Definition at line 236 of file TensorContractionMapper.h.
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
| typedef Tensor::PacketReturnType Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::Packet |
Definition at line 235 of file TensorContractionMapper.h.
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
| typedef SimpleTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment> Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::ParentMapper |
Definition at line 225 of file TensorContractionMapper.h.
Constructor & Destructor Documentation
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
|
inline |
Definition at line 228 of file TensorContractionMapper.h.
Member Function Documentation
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
template<int AlignmentType>
|
inline |
Definition at line 283 of file TensorContractionMapper.h.
template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
template<int AlignmentType>
|
inline |
Definition at line 240 of file TensorContractionMapper.h.
The documentation for this class was generated from the following file: