|
typedef BaseTensorContractionMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ > | Base |
|
typedef Scalar_ | Scalar |
|
typedef TensorContractionSubMapper< Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ > | SubMapper |
|
typedef SubMapper | VectorMapper |
|
typedef SimpleTensorContractionMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, Alignment, MakePointer_ > | ParentMapper |
|
enum | |
|
|
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE const CoeffLoader< Tensor, Tensor::RawAccess, MakePointer_ > & | get_tensor () const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE SubMapper | getSubMapper (Index i, Index j) const |
|
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE VectorMapper | getVectorMapper (Index i, Index j) const |
|
EIGEN_DEVICE_FUNC | TensorContractionInputMapper (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 | BaseTensorContractionMapper (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_STRONG_INLINE internal::enable_if< internal::unpacket_traits< PacketT >::size==packet_size, PacketT >::type | load (Index i, Index j) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE internal::enable_if< internal::unpacket_traits< PacketT >::size!=packet_size, PacketT >::type | load (Index i, Index j) const |
|
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketT | loadPacket (Index i, Index j) const |
|
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 |
|
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< class > class MakePointer_ = MakePointer>
class Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >
Definition at line 524 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< class > class MakePointer_ = MakePointer>
typedef BaseTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_> Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::Base |
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< class > class MakePointer_ = MakePointer>
typedef Scalar_ Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::Scalar |
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< class > class MakePointer_ = MakePointer>
typedef TensorContractionSubMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_> Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::SubMapper |
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< class > class MakePointer_ = MakePointer>
typedef SubMapper Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::VectorMapper |
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< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::TensorContractionInputMapper |
( |
const Tensor & |
tensor, |
|
|
const nocontract_t & |
nocontract_strides, |
|
|
const nocontract_t & |
ij_strides, |
|
|
const contract_t & |
contract_strides, |
|
|
const contract_t & |
k_strides |
|
) |
| |
|
inline |
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< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE const CoeffLoader<Tensor, Tensor::RawAccess, MakePointer_>& Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::get_tensor |
( |
| ) |
const |
|
inline |
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< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE SubMapper Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::getSubMapper |
( |
Index |
i, |
|
|
Index |
j |
|
) |
| const |
|
inline |
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< class > class MakePointer_ = MakePointer>
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE VectorMapper Eigen::internal::TensorContractionInputMapper< Scalar_, Index, side, Tensor, nocontract_t, contract_t, packet_size, inner_dim_contiguous, inner_dim_reordered, Alignment, MakePointer_ >::getVectorMapper |
( |
Index |
i, |
|
|
Index |
j |
|
) |
| const |
|
inline |