Public Types | Public Member Functions | List of all members
Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, 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, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >:
Inheritance graph
[legend]

Public Types

typedef unpacket_traits< Packet >::half HalfPacket
 
typedef Tensor::PacketReturnType Packet
 
typedef SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment > ParentMapper
 
- Public Types inherited from Eigen::internal::SimpleTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment >
enum  { DirectOffsets = CoeffLoader<Tensor, Tensor::RawAccess>::DirectOffsets }
 

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

Definition at line 222 of file TensorContractionMapper.h.

Member Typedef Documentation

template<typename Scalar, typename Index, int side, typename Tensor, typename nocontract_t, typename contract_t, int packet_size, bool inner_dim_contiguous, bool inner_dim_reordered, int Alignment>
typedef unpacket_traits<Packet>::half Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::HalfPacket

Definition at line 236 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, bool inner_dim_reordered, int Alignment>
typedef Tensor::PacketReturnType Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::Packet

Definition at line 235 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, bool inner_dim_reordered, int Alignment>
typedef SimpleTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment> Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::ParentMapper

Definition at line 225 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, bool inner_dim_reordered, int Alignment>
EIGEN_DEVICE_FUNC Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, 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 228 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, bool inner_dim_reordered, int Alignment>
template<int AlignmentType>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE HalfPacket Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::loadHalfPacket ( Index  i,
Index  j 
) const
inline

Definition at line 283 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, bool inner_dim_reordered, int Alignment>
template<int AlignmentType>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet Eigen::internal::BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment >::loadPacket ( Index  i,
Index  j 
) const
inline

Definition at line 240 of file TensorContractionMapper.h.


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


gtsam
Author(s):
autogenerated on Sat May 8 2021 02:55:49