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

Public Types

enum  { DirectOffsets = CoeffLoader<Tensor, Tensor::RawAccess>::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
 
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
 

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 > 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>
class Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >

Definition at line 77 of file TensorContractionMapper.h.

Member Enumeration Documentation

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

Definition at line 91 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, int Alignment>
EIGEN_DEVICE_FUNC Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::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 80 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::computeIndex ( Index  row,
Index  col 
) const
inline

Definition at line 114 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, int Alignment>
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 >::computeIndexPair ( Index  row,
Index  col,
const Index  distance 
) const
inline

Definition at line 152 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::firstAligned ( Index  size) const
inline

Definition at line 198 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE void Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::offsetBuffer ( typename Tensor::Index  offset)
inline

Definition at line 95 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::operator() ( Index  row) const
inline

Definition at line 103 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::operator() ( Index  row,
Index  col 
) const
inline

Definition at line 109 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::prefetch ( Index  )
inline

Definition at line 100 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, int Alignment>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::stride ( ) const
inline

Definition at line 204 of file TensorContractionMapper.h.

Member Data Documentation

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

Definition at line 212 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, int Alignment>
const nocontract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::m_ij_strides
protected

Definition at line 211 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, int Alignment>
const contract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::m_k_strides
protected

Definition at line 213 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, int Alignment>
const nocontract_t Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::m_nocontract_strides
protected

Definition at line 210 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, int Alignment>
CoeffLoader<Tensor, Tensor::RawAccess> Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >::m_tensor
protected

Definition at line 209 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:46