#include <TensorChipping.h>
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorBlock | block (TensorBlockDesc &desc, TensorBlockScratch &scratch, bool root_of_expr_ast=false) const |
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 EIGEN_STRONG_INLINE Storage::Type | data () const |
EIGEN_DEVICE_FUNC const EIGEN_STRONG_INLINE Dimensions & | dimensions () const |
EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (EvaluatorPointerType) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE internal::TensorBlockResourceRequirements | getResourceRequirements () const |
template<int LoadMode> | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packet (Index index) const |
EIGEN_STRONG_INLINE | TensorEvaluator (const XprType &op, const Device &device) |
Static Public Attributes | |
static const int | NumDims = NumInputDims-1 |
static const int | NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value |
static const int | PacketSize = PacketType<CoeffReturnType, Device>::size |
Protected Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool | isInnerChipping () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool | isOuterChipping () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | srcCoeff (Index index) const |
Protected Attributes | |
const Device EIGEN_DEVICE_REF | m_device |
const internal::DimensionId< DimId > | m_dim |
Dimensions | m_dimensions |
TensorEvaluator< ArgType, Device > | m_impl |
Index | m_inputOffset |
Index | m_inputStride |
Index | m_stride |
Definition at line 115 of file TensorChipping.h.
typedef TensorEvaluator<const ArgType, Device>::TensorBlock Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::ArgTensorBlock |
Definition at line 160 of file TensorChipping.h.
typedef internal::TensorBlockDescriptor<NumInputDims, Index> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::ArgTensorBlockDesc |
Definition at line 158 of file TensorChipping.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::CoeffReturnType |
Definition at line 123 of file TensorChipping.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::Dimensions |
Definition at line 121 of file TensorChipping.h.
typedef Storage::Type Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::EvaluatorPointerType |
Definition at line 127 of file TensorChipping.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::Index |
Definition at line 120 of file TensorChipping.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::PacketReturnType |
Definition at line 124 of file TensorChipping.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::Scalar |
Definition at line 122 of file TensorChipping.h.
typedef internal::remove_const<Scalar>::type Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::ScalarNoConst |
Definition at line 151 of file TensorChipping.h.
typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::Storage |
Definition at line 126 of file TensorChipping.h.
typedef internal::TensorMaterializedBlock<ScalarNoConst, NumDims, Layout, Index> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::TensorBlock |
Definition at line 164 of file TensorChipping.h.
typedef internal::TensorBlockDescriptor<NumDims, Index> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::TensorBlockDesc |
Definition at line 154 of file TensorChipping.h.
typedef internal::TensorBlockScratchAllocator<Device> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::TensorBlockScratch |
Definition at line 155 of file TensorChipping.h.
typedef TensorChippingOp<DimId, ArgType> Eigen::TensorEvaluator< const TensorChippingOp< DimId, ArgType >, Device >::XprType |
Definition at line 117 of file TensorChipping.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
Layout | |
PacketAccess | |
BlockAccess | |
IsOuterChipping | |
IsInnerChipping | |
PreferBlockAccess | |
CoordAccess | |
RawAccess |
Definition at line 129 of file TensorChipping.h.
|
inline |
Definition at line 167 of file TensorChipping.h.
|
inline |
Definition at line 290 of file TensorChipping.h.
|
inline |
Definition at line 208 of file TensorChipping.h.
|
inline |
Definition at line 212 of file TensorChipping.h.
|
inline |
Definition at line 260 of file TensorChipping.h.
|
inline |
Definition at line 349 of file TensorChipping.h.
|
inline |
Definition at line 201 of file TensorChipping.h.
|
inline |
Definition at line 203 of file TensorChipping.h.
|
inline |
Definition at line 282 of file TensorChipping.h.
|
inlineprotected |
Definition at line 386 of file TensorChipping.h.
|
inlineprotected |
Definition at line 392 of file TensorChipping.h.
|
inline |
Definition at line 218 of file TensorChipping.h.
|
inlineprotected |
Definition at line 365 of file TensorChipping.h.
|
protected |
Definition at line 404 of file TensorChipping.h.
|
protected |
Definition at line 403 of file TensorChipping.h.
|
protected |
Definition at line 398 of file TensorChipping.h.
|
protected |
Definition at line 402 of file TensorChipping.h.
|
protected |
Definition at line 400 of file TensorChipping.h.
|
protected |
Definition at line 401 of file TensorChipping.h.
|
protected |
Definition at line 399 of file TensorChipping.h.
|
static |
Definition at line 119 of file TensorChipping.h.
|
static |
Definition at line 118 of file TensorChipping.h.
|
static |
Definition at line 125 of file TensorChipping.h.