#include <TensorPadding.h>
Classes | |
struct | BlockIteratorState |
Public Types | |
enum | { IsAligned = true, PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, BlockAccess = TensorEvaluator<ArgType, Device>::RawAccess, PreferBlockAccess = true, Layout = TensorEvaluator<ArgType, Device>::Layout, CoordAccess = true, RawAccess = false } |
typedef XprType::CoeffReturnType | CoeffReturnType |
typedef DSizes< Index, NumDims > | Dimensions |
typedef Storage::Type | EvaluatorPointerType |
typedef XprType::Index | Index |
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
typedef XprType::Scalar | Scalar |
typedef internal::remove_const< Scalar >::type | ScalarNoConst |
typedef StorageMemory< CoeffReturnType, Device > | Storage |
typedef internal::TensorMaterializedBlock< ScalarNoConst, NumDims, Layout, Index > | TensorBlock |
typedef internal::TensorBlockDescriptor< NumDims, Index > | TensorBlockDesc |
typedef internal::TensorBlockScratchAllocator< Device > | TensorBlockScratch |
typedef TensorPaddingOp< PaddingDimensions, ArgType > | XprType |
Public Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorBlock | block (TensorBlockDesc &desc, TensorBlockScratch &scratch, bool=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 EvaluatorPointerType | 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<PaddingDimensions>::value |
static const int | PacketSize = PacketType<CoeffReturnType, Device>::size |
Protected Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packetColMajor (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packetRowMajor (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packetWithPossibleZero (Index index) const |
Protected Attributes | |
const Device EIGEN_DEVICE_REF | m_device |
Dimensions | m_dimensions |
TensorEvaluator< ArgType, Device > | m_impl |
array< Index, NumDims > | m_inputStrides |
array< Index, NumDims+1 > | m_outputStrides |
PaddingDimensions | m_padding |
Scalar | m_paddingValue |
Private Member Functions | |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE bool | isLeftPaddingCompileTimeZero (int dim_index) const |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE bool | isPaddingAtIndexForDim (Index index, int dim_index) const |
EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE bool | isRightPaddingCompileTimeZero (int dim_index) const |
void | updateCostPerDimension (TensorOpCost &cost, int i, bool first) const |
Definition at line 85 of file TensorPadding.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::CoeffReturnType |
Definition at line 92 of file TensorPadding.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Dimensions |
Definition at line 90 of file TensorPadding.h.
typedef Storage::Type Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::EvaluatorPointerType |
Definition at line 96 of file TensorPadding.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Index |
Definition at line 88 of file TensorPadding.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::PacketReturnType |
Definition at line 93 of file TensorPadding.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Scalar |
Definition at line 91 of file TensorPadding.h.
typedef internal::remove_const<Scalar>::type Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::ScalarNoConst |
Definition at line 108 of file TensorPadding.h.
typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Storage |
Definition at line 95 of file TensorPadding.h.
typedef internal::TensorMaterializedBlock<ScalarNoConst, NumDims, Layout, Index> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::TensorBlock |
Definition at line 116 of file TensorPadding.h.
typedef internal::TensorBlockDescriptor<NumDims, Index> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::TensorBlockDesc |
Definition at line 111 of file TensorPadding.h.
typedef internal::TensorBlockScratchAllocator<Device> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::TensorBlockScratch |
Definition at line 112 of file TensorPadding.h.
typedef TensorPaddingOp<PaddingDimensions, ArgType> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::XprType |
Definition at line 87 of file TensorPadding.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
PreferBlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 98 of file TensorPadding.h.
|
inline |
Definition at line 119 of file TensorPadding.h.
|
inline |
Definition at line 239 of file TensorPadding.h.
|
inline |
Definition at line 167 of file TensorPadding.h.
|
inline |
Definition at line 171 of file TensorPadding.h.
|
inline |
Definition at line 216 of file TensorPadding.h.
|
inline |
Definition at line 484 of file TensorPadding.h.
|
inline |
Definition at line 152 of file TensorPadding.h.
|
inline |
Definition at line 154 of file TensorPadding.h.
|
inline |
Definition at line 231 of file TensorPadding.h.
|
inlineprivate |
Definition at line 524 of file TensorPadding.h.
|
inlineprivate |
Definition at line 511 of file TensorPadding.h.
|
inlineprivate |
Definition at line 534 of file TensorPadding.h.
|
inline |
Definition at line 208 of file TensorPadding.h.
|
inlineprotected |
Definition at line 565 of file TensorPadding.h.
|
inlineprotected |
Definition at line 623 of file TensorPadding.h.
|
inlineprotected |
Definition at line 681 of file TensorPadding.h.
|
inlineprivate |
Definition at line 545 of file TensorPadding.h.
|
protected |
Definition at line 700 of file TensorPadding.h.
|
protected |
Definition at line 692 of file TensorPadding.h.
|
protected |
Definition at line 695 of file TensorPadding.h.
|
protected |
Definition at line 694 of file TensorPadding.h.
|
protected |
Definition at line 693 of file TensorPadding.h.
|
protected |
Definition at line 696 of file TensorPadding.h.
|
protected |
Definition at line 698 of file TensorPadding.h.
|
static |
Definition at line 89 of file TensorPadding.h.
|
static |
Definition at line 94 of file TensorPadding.h.