#include <TensorReduction.h>

Classes | |
| struct | BlockIteratorState |
Public Member Functions | |
| 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 vectorized) const |
| EIGEN_DEVICE_FUNC EvaluatorPointerType | data () const |
| const EIGEN_DEVICE_FUNC Device & | device () const |
| EIGEN_DEVICE_FUNC const EIGEN_STRONG_INLINE Dimensions & | dimensions () const |
| EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (EvaluatorPointerType data) |
| EIGEN_STRONG_INLINE bool | evalSubExprsIfNeededCommon (EvaluatorPointerType data) |
| const EIGEN_DEVICE_FUNC TensorEvaluator< ArgType, Device > & | impl () const |
| template<int LoadMode> | |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packet (Index index) const |
| EIGEN_STRONG_INLINE | TensorReductionEvaluatorBase (const XprType &op, const Device &device) |
Static Public Attributes | |
| static const bool | InputPacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess |
| static const int | NumInputDims = internal::array_size<InputDimensions>::value |
| static const int | NumOutputDims = NumInputDims - NumReducedDims |
| static const int | NumPreservedStrides = max_n_1<NumOutputDims>::size |
| static const int | NumReducedDims = internal::array_size<Dims>::value |
| static const Index | PacketSize = PacketType<CoeffReturnType, Device>::size |
| static const bool | PreservingInnerMostDims = internal::preserve_inner_most_dims<Dims, NumInputDims, Layout>::value |
| static const bool | ReducingInnerMostDims = internal::are_inner_most_dims<Dims, NumInputDims, Layout>::value |
| static const bool | RunningFullReduction = (NumOutputDims==0) |
Private Member Functions | |
| EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | firstInput (Index index) const |
Private Attributes | |
| const Device EIGEN_DEVICE_REF | m_device |
| Dimensions | m_dimensions |
| array< internal::TensorIntDivisor< Index >, NumOutputDims > | m_fastOutputStrides |
| TensorEvaluator< ArgType, Device > | m_impl |
| Index | m_numValuesToReduce |
| array< Index, NumOutputDims > | m_output_to_input_dim_map |
| array< Index, NumOutputDims > | m_outputStrides |
| array< Index, NumPreservedStrides > | m_preservedStrides |
| array< bool, NumInputDims > | m_reduced |
| array< Index, NumReducedDims > | m_reducedDims |
| array< Index, NumReducedDims > | m_reducedStrides |
| Op | m_reducer |
| EvaluatorPointerType | m_result |
Static Private Attributes | |
| static const bool | RunningOnGPU = false |
| static const bool | RunningOnSycl = false |
Friends | |
| template<typename S , typename O , typename D , bool V> | |
| struct | internal::FullReducer |
| template<int , typename , typename > | |
| struct | internal::GenericDimReducer |
| template<int , typename , typename , bool > | |
| struct | internal::InnerMostDimPreserver |
| template<typename , typename , bool , bool > | |
| struct | internal::InnerMostDimReducer |
| template<typename S , typename O , typename D > | |
| struct | internal::InnerReducer |
Definition at line 505 of file TensorReduction.h.
| typedef ArgType Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::ChildType |
Definition at line 511 of file TensorReduction.h.
| typedef internal::ReductionReturnType<Op, typename XprType::CoeffReturnType>::type Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::CoeffReturnType |
Definition at line 520 of file TensorReduction.h.
| typedef internal::conditional<NumOutputDims==0, Sizes<>, DSizes<Index, NumOutputDims> >::type Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::Dimensions |
Definition at line 516 of file TensorReduction.h.
| typedef Storage::Type Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::EvaluatorPointerType |
Definition at line 526 of file TensorReduction.h.
| typedef XprType::Index Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::Index |
Definition at line 510 of file TensorReduction.h.
| typedef TensorEvaluator<ArgType, Device>::Dimensions Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::InputDimensions |
Definition at line 512 of file TensorReduction.h.
| typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::PacketReturnType |
Definition at line 521 of file TensorReduction.h.
| typedef Dims Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::ReducedDims |
Definition at line 508 of file TensorReduction.h.
| typedef internal::reducer_traits<Op, Device> Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::ReducerTraits |
Definition at line 507 of file TensorReduction.h.
| typedef XprType::Scalar Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::Scalar |
Definition at line 517 of file TensorReduction.h.
| typedef internal::remove_const<Scalar>::type Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::ScalarNoConst |
Definition at line 542 of file TensorReduction.h.
| typedef TensorReductionEvaluatorBase<const TensorReductionOp<Op, Dims, ArgType, MakePointer_>, Device> Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::Self |
Definition at line 518 of file TensorReduction.h.
| typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::Storage |
Definition at line 525 of file TensorReduction.h.
| typedef internal::TensorBlockNotImplemented Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::TensorBlock |
Definition at line 545 of file TensorReduction.h.
| typedef Eigen::internal::traits<XprType>::PointerType Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::TensorPointerType |
Definition at line 524 of file TensorReduction.h.
| typedef TensorReductionOp<Op, Dims, ArgType, MakePointer_> Eigen::TensorReductionEvaluatorBase< const TensorReductionOp< Op, Dims, ArgType, MakePointer_ >, Device >::XprType |
Definition at line 509 of file TensorReduction.h.
| anonymous enum |
| Enumerator | |
|---|---|
| IsAligned | |
| PacketAccess | |
| BlockAccess | |
| PreferBlockAccess | |
| Layout | |
| CoordAccess | |
| RawAccess | |
Definition at line 532 of file TensorReduction.h.
|
inline |
Definition at line 552 of file TensorReduction.h.
|
inline |
Definition at line 757 of file TensorReduction.h.
|
inline |
Definition at line 765 of file TensorReduction.h.
|
inline |
Definition at line 828 of file TensorReduction.h.
|
inline |
Definition at line 839 of file TensorReduction.h.
|
inline |
Definition at line 841 of file TensorReduction.h.
|
inline |
Definition at line 631 of file TensorReduction.h.
|
inline |
Definition at line 752 of file TensorReduction.h.
|
inline |
Definition at line 634 of file TensorReduction.h.
|
inlineprivate |
Definition at line 887 of file TensorReduction.h.
|
inline |
Definition at line 840 of file TensorReduction.h.
|
inline |
Definition at line 785 of file TensorReduction.h.
|
friend |
Definition at line 854 of file TensorReduction.h.
|
friend |
Definition at line 851 of file TensorReduction.h.
|
friend |
Definition at line 853 of file TensorReduction.h.
|
friend |
Definition at line 852 of file TensorReduction.h.
|
friend |
Definition at line 877 of file TensorReduction.h.
|
static |
Definition at line 519 of file TensorReduction.h.
|
private |
Definition at line 966 of file TensorReduction.h.
|
private |
Definition at line 930 of file TensorReduction.h.
|
private |
Definition at line 933 of file TensorReduction.h.
|
private |
Definition at line 948 of file TensorReduction.h.
|
private |
Definition at line 938 of file TensorReduction.h.
|
private |
Definition at line 936 of file TensorReduction.h.
|
private |
Definition at line 932 of file TensorReduction.h.
|
private |
Definition at line 934 of file TensorReduction.h.
|
private |
Definition at line 928 of file TensorReduction.h.
|
private |
Definition at line 945 of file TensorReduction.h.
|
private |
Definition at line 942 of file TensorReduction.h.
|
private |
Definition at line 951 of file TensorReduction.h.
|
private |
Definition at line 964 of file TensorReduction.h.
|
static |
Definition at line 513 of file TensorReduction.h.
|
static |
Definition at line 515 of file TensorReduction.h.
|
static |
Definition at line 530 of file TensorReduction.h.
|
static |
Definition at line 514 of file TensorReduction.h.
|
static |
Definition at line 522 of file TensorReduction.h.
|
static |
Definition at line 549 of file TensorReduction.h.
|
static |
Definition at line 548 of file TensorReduction.h.
|
static |
Definition at line 550 of file TensorReduction.h.
|
staticprivate |
Definition at line 961 of file TensorReduction.h.
|
staticprivate |
Definition at line 962 of file TensorReduction.h.