#include <TensorContraction.h>
Public Member Functions | |
template<bool lhs_inner_dim_contiguous, bool rhs_inner_dim_contiguous, bool rhs_inner_dim_reordered, int Alignment> | |
EIGEN_DEVICE_FUNC void | evalProduct (Scalar *buffer) const |
EIGEN_DEVICE_FUNC | TensorEvaluator (const XprType &op, const Device &device) |
Public Member Functions inherited from Eigen::TensorContractionEvaluatorBase< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device > > | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | cleanup () |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType | coeff (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | costPerCoeff (bool) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Scalar * | data () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions & | dimensions () const |
EIGEN_DEVICE_FUNC void | evalGemm (Scalar *buffer) const |
EIGEN_DEVICE_FUNC void | evalGemv (Scalar *buffer) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (Scalar *data) |
EIGEN_DEVICE_FUNC void | evalTo (Scalar *buffer) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packet (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE | TensorContractionEvaluatorBase (const XprType &op, const Device &device) |
Static Public Attributes | |
static const int | ContractDims = internal::array_size<Indices>::value |
static const int | LDims |
static const int | NumDims = LDims + RDims - 2 * ContractDims |
static const int | RDims |
Static Public Attributes inherited from Eigen::TensorContractionEvaluatorBase< TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device > > | |
static const int | ContractDims |
static const int | LDims |
static const int | NumDims |
static const int | RDims |
Definition at line 572 of file TensorContraction.h.
typedef TensorContractionEvaluatorBase<Self> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::Base |
Definition at line 576 of file TensorContraction.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::CoeffReturnType |
Definition at line 581 of file TensorContraction.h.
typedef array<Index, ContractDims> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::contract_t |
Definition at line 603 of file TensorContraction.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::Dimensions |
Definition at line 610 of file TensorContraction.h.
typedef internal::conditional< static_cast<int>Layout) == static_cast<int>ColMajor), LeftArgType, RightArgType>::type Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::EvalLeftArgType |
Definition at line 593 of file TensorContraction.h.
typedef internal::conditional< static_cast<int>Layout) == static_cast<int>ColMajor), RightArgType, LeftArgType>::type Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::EvalRightArgType |
Definition at line 595 of file TensorContraction.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::Index |
Definition at line 580 of file TensorContraction.h.
typedef array<Index, LDims - ContractDims> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::left_nocontract_t |
Definition at line 604 of file TensorContraction.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::PacketReturnType |
Definition at line 582 of file TensorContraction.h.
typedef array<Index, RDims - ContractDims> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::right_nocontract_t |
Definition at line 605 of file TensorContraction.h.
typedef internal::remove_const<typename XprType::Scalar>::type Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::Scalar |
Definition at line 579 of file TensorContraction.h.
typedef TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, Device> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::Self |
Definition at line 575 of file TensorContraction.h.
typedef TensorContractionOp<Indices, LeftArgType, RightArgType> Eigen::TensorEvaluator< const TensorContractionOp< Indices, LeftArgType, RightArgType >, Device >::XprType |
Definition at line 578 of file TensorContraction.h.
anonymous enum |
Enumerator | |
---|---|
Layout |
Definition at line 584 of file TensorContraction.h.
|
inline |
Definition at line 612 of file TensorContraction.h.
|
inline |
Definition at line 616 of file TensorContraction.h.
|
static |
Definition at line 601 of file TensorContraction.h.
|
static |
Definition at line 597 of file TensorContraction.h.
|
static |
Definition at line 607 of file TensorContraction.h.
|
static |
Definition at line 599 of file TensorContraction.h.