|
typedef Scalar_ | Scalar |
|
typedef BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment > | Base |
|
typedef TensorContractionSubMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment > | SubMapper |
|
typedef SubMapper | VectorMapper |
|
typedef SimpleTensorContractionMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment > | ParentMapper |
|
typedef Tensor::PacketReturnType | Packet |
|
typedef unpacket_traits< Packet >::half | HalfPacket |
|
enum | |
|
|
EIGEN_DEVICE_FUNC | TensorContractionInputMapper (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_STRONG_INLINE SubMapper | getSubMapper (Index i, Index j) const |
|
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE VectorMapper | getVectorMapper (Index i, Index j) const |
|
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) |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet | loadPacket (Index i, Index j) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE HalfPacket | loadHalfPacket (Index i, Index j) const |
|
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 |
|