#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.