Public Types | Public Member Functions | Protected Attributes | List of all members
Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ > Class Template Reference

#include <TensorContractionMapper.h>

Inheritance diagram for Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >:
Inheritance graph
[legend]

Public Types

enum  { DirectOffsets = CoeffLoader<Tensor, Tensor::RawAccess, MakePointer_>::DirectOffsets }
 

Public Member Functions

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
 
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
 

Protected Attributes

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, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
class Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >

Definition at line 114 of file TensorContractionMapper.h.

Member Enumeration Documentation

◆ anonymous enum

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
anonymous enum
Enumerator
DirectOffsets 

Definition at line 128 of file TensorContractionMapper.h.

Constructor & Destructor Documentation

◆ SimpleTensorContractionMapper()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::SimpleTensorContractionMapper ( 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 117 of file TensorContractionMapper.h.

Member Function Documentation

◆ computeIndex()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::computeIndex ( Index  row,
Index  col 
) const
inline

Definition at line 151 of file TensorContractionMapper.h.

◆ computeIndexPair()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE IndexPair<Index> Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::computeIndexPair ( Index  row,
Index  col,
const Index  distance 
) const
inline

Definition at line 192 of file TensorContractionMapper.h.

◆ contract_strides()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const contract_t& Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::contract_strides ( ) const
inline

Definition at line 266 of file TensorContractionMapper.h.

◆ firstAligned()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::firstAligned ( Index  size) const
inline

Definition at line 241 of file TensorContractionMapper.h.

◆ ij_strides()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const nocontract_t& Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::ij_strides ( ) const
inline

Definition at line 265 of file TensorContractionMapper.h.

◆ k_strides()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const contract_t& Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::k_strides ( ) const
inline

Definition at line 267 of file TensorContractionMapper.h.

◆ nocontract_strides()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const nocontract_t& Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::nocontract_strides ( ) const
inline

Definition at line 262 of file TensorContractionMapper.h.

◆ offsetBuffer()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::offsetBuffer ( typename Tensor::Index  offset)
inline

Definition at line 132 of file TensorContractionMapper.h.

◆ operator()() [1/2]

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::operator() ( Index  row) const
inline

Definition at line 140 of file TensorContractionMapper.h.

◆ operator()() [2/2]

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::operator() ( Index  row,
Index  col 
) const
inline

Definition at line 146 of file TensorContractionMapper.h.

◆ prefetch()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::prefetch ( Index  )
inline

Definition at line 137 of file TensorContractionMapper.h.

◆ stride()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::stride ( ) const
inline

Definition at line 247 of file TensorContractionMapper.h.

◆ tensor()

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const CoeffLoader<Tensor, Tensor::RawAccess, MakePointer_>& Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::tensor ( ) const
inline

Definition at line 258 of file TensorContractionMapper.h.

Member Data Documentation

◆ m_contract_strides

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const contract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::m_contract_strides
protected

Definition at line 273 of file TensorContractionMapper.h.

◆ m_ij_strides

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const nocontract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::m_ij_strides
protected

Definition at line 272 of file TensorContractionMapper.h.

◆ m_k_strides

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const contract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::m_k_strides
protected

Definition at line 274 of file TensorContractionMapper.h.

◆ m_nocontract_strides

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
const nocontract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::m_nocontract_strides
protected

Definition at line 271 of file TensorContractionMapper.h.

◆ m_tensor

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, int Alignment, template< class > class MakePointer_ = MakePointer>
CoeffLoader<Tensor, Tensor::RawAccess, MakePointer_> Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ >::m_tensor
protected

Definition at line 270 of file TensorContractionMapper.h.

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

gtsam
Author(s):
autogenerated on Tue Jul 4 2023 02:45:22