TR-mbed 1.0
Loading...
Searching...
No Matches
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_ >:
Eigen::internal::BaseTensorContractionMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer > Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer >

Public Types

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

Public Member Functions

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< 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 Index stride () const
 
const CoeffLoader< Tensor, Tensor::RawAccess, MakePointer_ > & tensor () const
 
const nocontract_tnocontract_strides () const
 
const nocontract_tij_strides () const
 
const contract_t & contract_strides () const
 
const contract_t & k_strides () const
 

Protected Attributes

CoeffLoader< Tensor, Tensor::RawAccess, MakePointer_m_tensor
 
const nocontract_t m_nocontract_strides
 
const nocontract_t m_ij_strides
 
const contract_t m_contract_strides
 
const contract_t m_k_strides
 

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 

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

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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

◆ 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

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

◆ 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

◆ 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

◆ 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

◆ 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

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