Public Types | Public Member Functions | List of all members
Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment > Class Template Reference

#include <TensorContractionMapper.h>

Inheritance diagram for Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >:
Inheritance graph
[legend]

Public Types

typedef BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment > Base
 
typedef Scalar_ Scalar
 
typedef TensorContractionSubMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment > SubMapper
 
typedef SubMapper VectorMapper
 
- Public Types inherited from Eigen::internal::BaseTensorContractionMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >
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  
 

Public Member Functions

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 TensorContractionInputMapper (const Tensor &tensor, const nocontract_t &nocontract_strides, const nocontract_t &ij_strides, const contract_t &contract_strides, const contract_t &k_strides)
 
- Public Member Functions inherited from Eigen::internal::BaseTensorContractionMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >
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 HalfPacket loadHalfPacket (Index i, Index j) const
 
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< IndexcomputeIndexPair (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::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >

Definition at line 436 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 BaseTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment> Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::Base

Definition at line 441 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 Scalar_ Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::Scalar

Definition at line 440 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 TensorContractionSubMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment> Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::SubMapper

Definition at line 442 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 SubMapper Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::VectorMapper

Definition at line 443 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>
EIGEN_DEVICE_FUNC Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::TensorContractionInputMapper ( const Tensor tensor,
const nocontract_t &  nocontract_strides,
const nocontract_t &  ij_strides,
const contract_t &  contract_strides,
const contract_t &  k_strides 
)
inline

Definition at line 445 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>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE SubMapper Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::getSubMapper ( Index  i,
Index  j 
) const
inline

Definition at line 453 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>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE VectorMapper Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::getVectorMapper ( Index  i,
Index  j 
) const
inline

Definition at line 457 of file TensorContractionMapper.h.


The documentation for this class was generated from the following file:


hebiros
Author(s): Xavier Artache , Matthew Tesch
autogenerated on Thu Sep 3 2020 04:10:48