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, MakePointer_ > 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, MakePointer_ >:
Inheritance graph
[legend]

Public Types

typedef SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment, MakePointer_ > ParentMapper
 
- Public Types inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment, MakePointer_ >
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<typename PacketT , int >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketT load (Index i, Index j) const
 
template<typename PacketT , int >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketT 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, MakePointer_ >
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
 
const contract_t & contract_strides () const
 
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index firstAligned (Index size) const
 
const nocontract_t & ij_strides () const
 
const contract_t & k_strides () const
 
const nocontract_t & nocontract_strides () 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
 
const CoeffLoader< Tensor, Tensor::RawAccess, MakePointer_ > & tensor () const
 

Additional Inherited Members

- Protected Attributes inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment, MakePointer_ >
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, MakePointer_ > 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, template< class > class MakePointer_>
class Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >

Definition at line 377 of file TensorContractionMapper.h.

Member Typedef Documentation

◆ ParentMapper

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< class > class MakePointer_>
typedef SimpleTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, Alignment, MakePointer_> Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::ParentMapper

Definition at line 381 of file TensorContractionMapper.h.

Constructor & Destructor Documentation

◆ BaseTensorContractionMapper()

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< class > class MakePointer_>
EIGEN_DEVICE_FUNC Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, 1, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::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 384 of file TensorContractionMapper.h.

Member Function Documentation

◆ load()

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

Definition at line 398 of file TensorContractionMapper.h.

◆ loadPacket()

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

Definition at line 392 of file TensorContractionMapper.h.


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


gtsam
Author(s):
autogenerated on Sat Jun 1 2024 03:11:38