#include <TensorShuffling.h>
Classes | |
struct | PacketLoader |
struct | PacketLoader< LoadMode, Self, true > |
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 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 = 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 Index | GetBlockOutputIndex (Index input_index, const DSizes< Index, NumDims > &input_block_strides, const DSizes< Index, NumDims > &output_block_strides, const DSizes< internal::TensorIntDivisor< Index >, NumDims > &fast_input_block_strides) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | srcCoeff (Index index) const |
Protected Attributes | |
const Device EIGEN_DEVICE_REF | m_device |
Dimensions | m_dimensions |
array< internal::TensorIntDivisor< Index >, NumDims > | m_fastOutputStrides |
TensorEvaluator< ArgType, Device > | m_impl |
array< Index, NumDims > | m_inputStrides |
array< Index, NumDims > | m_inverseShuffle |
bool | m_is_identity |
array< Index, NumDims > | m_outputStrides |
array< int, NumDims > | m_shuffle |
array< Index, NumDims > | m_unshuffledInputStrides |
Definition at line 86 of file TensorShuffling.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::CoeffReturnType |
Definition at line 94 of file TensorShuffling.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::Dimensions |
Definition at line 92 of file TensorShuffling.h.
typedef Storage::Type Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::EvaluatorPointerType |
Definition at line 98 of file TensorShuffling.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::Index |
Definition at line 90 of file TensorShuffling.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::PacketReturnType |
Definition at line 95 of file TensorShuffling.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::Scalar |
Definition at line 93 of file TensorShuffling.h.
typedef internal::remove_const<Scalar>::type Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::ScalarNoConst |
Definition at line 110 of file TensorShuffling.h.
typedef TensorEvaluator<const TensorShufflingOp<Shuffle, ArgType>, Device> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::Self |
Definition at line 88 of file TensorShuffling.h.
typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::Storage |
Definition at line 97 of file TensorShuffling.h.
typedef internal::TensorMaterializedBlock<ScalarNoConst, NumDims, Layout, Index> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::TensorBlock |
Definition at line 118 of file TensorShuffling.h.
typedef internal::TensorBlockDescriptor<NumDims, Index> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::TensorBlockDesc |
Definition at line 113 of file TensorShuffling.h.
typedef internal::TensorBlockScratchAllocator<Device> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::TensorBlockScratch |
Definition at line 114 of file TensorShuffling.h.
typedef TensorShufflingOp<Shuffle, ArgType> Eigen::TensorEvaluator< const TensorShufflingOp< Shuffle, ArgType >, Device >::XprType |
Definition at line 89 of file TensorShuffling.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
PreferBlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 100 of file TensorShuffling.h.
|
inline |
Definition at line 121 of file TensorShuffling.h.
|
inline |
Definition at line 256 of file TensorShuffling.h.
|
inline |
Definition at line 180 of file TensorShuffling.h.
|
inline |
Definition at line 184 of file TensorShuffling.h.
|
inline |
Definition at line 281 of file TensorShuffling.h.
|
inline |
Definition at line 290 of file TensorShuffling.h.
|
inline |
Definition at line 165 of file TensorShuffling.h.
|
inline |
Definition at line 167 of file TensorShuffling.h.
|
inlineprotected |
Definition at line 299 of file TensorShuffling.h.
|
inline |
Definition at line 234 of file TensorShuffling.h.
|
inline |
Definition at line 226 of file TensorShuffling.h.
|
inlineprotected |
Definition at line 324 of file TensorShuffling.h.
|
protected |
Definition at line 352 of file TensorShuffling.h.
|
protected |
Definition at line 343 of file TensorShuffling.h.
|
protected |
Definition at line 348 of file TensorShuffling.h.
|
protected |
Definition at line 353 of file TensorShuffling.h.
|
protected |
Definition at line 349 of file TensorShuffling.h.
|
protected |
Definition at line 346 of file TensorShuffling.h.
|
protected |
Definition at line 344 of file TensorShuffling.h.
|
protected |
Definition at line 347 of file TensorShuffling.h.
|
protected |
Definition at line 345 of file TensorShuffling.h.
|
protected |
Definition at line 350 of file TensorShuffling.h.
|
static |
Definition at line 91 of file TensorShuffling.h.
|
static |
Definition at line 96 of file TensorShuffling.h.