#include <TensorPadding.h>
Public Types | |
enum | { IsAligned = true, PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess, Layout = TensorEvaluator<ArgType, Device>::Layout, CoordAccess = true, RawAccess = false } |
typedef XprType::CoeffReturnType | CoeffReturnType |
typedef DSizes< Index, NumDims > | Dimensions |
typedef XprType::Index | Index |
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
typedef XprType::Scalar | Scalar |
typedef TensorPaddingOp< PaddingDimensions, ArgType > | XprType |
Public Member Functions | |
EIGEN_DEVICE_FUNC 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 Scalar * | data () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions & | dimensions () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (Scalar *) |
template<int LoadMode> | |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE PacketReturnType | packet (Index index) const |
EIGEN_DEVICE_FUNC 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 = internal::unpacket_traits<PacketReturnType>::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 | |
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 84 of file TensorPadding.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::CoeffReturnType |
Definition at line 91 of file TensorPadding.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Dimensions |
Definition at line 89 of file TensorPadding.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Index |
Definition at line 87 of file TensorPadding.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::PacketReturnType |
Definition at line 92 of file TensorPadding.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::Scalar |
Definition at line 90 of file TensorPadding.h.
typedef TensorPaddingOp<PaddingDimensions, ArgType> Eigen::TensorEvaluator< const TensorPaddingOp< PaddingDimensions, ArgType >, Device >::XprType |
Definition at line 86 of file TensorPadding.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 95 of file TensorPadding.h.
|
inline |
Definition at line 103 of file TensorPadding.h.
|
inline |
Definition at line 142 of file TensorPadding.h.
|
inline |
Definition at line 146 of file TensorPadding.h.
|
inline |
Definition at line 189 of file TensorPadding.h.
|
inline |
Definition at line 201 of file TensorPadding.h.
|
inline |
Definition at line 136 of file TensorPadding.h.
|
inline |
Definition at line 138 of file TensorPadding.h.
|
inlineprivate |
Definition at line 217 of file TensorPadding.h.
|
inlineprivate |
Definition at line 204 of file TensorPadding.h.
|
inlineprivate |
Definition at line 227 of file TensorPadding.h.
|
inline |
Definition at line 181 of file TensorPadding.h.
|
inlineprotected |
Definition at line 258 of file TensorPadding.h.
|
inlineprotected |
Definition at line 315 of file TensorPadding.h.
|
inlineprotected |
Definition at line 373 of file TensorPadding.h.
|
inlineprivate |
Definition at line 238 of file TensorPadding.h.
|
protected |
Definition at line 383 of file TensorPadding.h.
|
protected |
Definition at line 386 of file TensorPadding.h.
|
protected |
Definition at line 385 of file TensorPadding.h.
|
protected |
Definition at line 384 of file TensorPadding.h.
|
protected |
Definition at line 387 of file TensorPadding.h.
|
protected |
Definition at line 389 of file TensorPadding.h.
|
static |
Definition at line 88 of file TensorPadding.h.
|
static |
Definition at line 93 of file TensorPadding.h.