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

#include <TensorContractionMapper.h>

Inheritance diagram for Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >:
Inheritance graph
[legend]

Public Types

typedef Tensor::PacketReturnType Packet
 
typedef SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment > ParentMapper
 
- Public Types inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment >
enum  
 

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 >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet loadHalfPacket (Index i, Index j) const
 
template<int >
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, 1, 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, 1, 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, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
class Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >

Definition at line 305 of file TensorContractionMapper.h.

Member Typedef Documentation

template<typename Scalar , typename Index , int side, typename Tensor , typename nocontract_t , typename contract_t , bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
typedef Tensor::PacketReturnType Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >::Packet

Definition at line 318 of file TensorContractionMapper.h.

template<typename Scalar , typename Index , int side, typename Tensor , typename nocontract_t , typename contract_t , bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
typedef SimpleTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment> Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >::ParentMapper

Definition at line 308 of file TensorContractionMapper.h.

Constructor & Destructor Documentation

template<typename Scalar , typename Index , int side, typename Tensor , typename nocontract_t , typename contract_t , bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
EIGEN_DEVICE_FUNC Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >::BaseTensorContractionMapper ( 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 311 of file TensorContractionMapper.h.

Member Function Documentation

template<typename Scalar , typename Index , int side, typename Tensor , typename nocontract_t , typename contract_t , bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
template<int >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >::loadHalfPacket ( Index  i,
Index  j 
) const
inline

Definition at line 326 of file TensorContractionMapper.h.

template<typename Scalar , typename Index , int side, typename Tensor , typename nocontract_t , typename contract_t , bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
template<int >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment >::loadPacket ( Index  i,
Index  j 
) const
inline

Definition at line 320 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:30