#include <TensorCustomOp.h>
Public Types | |
| enum | { IsAligned = false, PacketAccess = (PacketType<CoeffReturnType, Device>::size > 1), BlockAccess = false, PreferBlockAccess = false, Layout = TensorEvaluator<XprType, Device>::Layout, CoordAccess = false, RawAccess = false } |
| typedef TensorCustomUnaryOp< CustomUnaryFunc, XprType > | ArgType |
| typedef internal::remove_const< typename XprType::CoeffReturnType >::type | CoeffReturnType |
| typedef DSizes< Index, NumDims > | Dimensions |
| typedef Storage::Type | EvaluatorPointerType |
| typedef internal::traits< ArgType >::Index | Index |
| typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
| typedef internal::remove_const< typename ArgType::Scalar >::type | Scalar |
| typedef StorageMemory< CoeffReturnType, Device > | Storage |
| typedef internal::TensorBlockNotImplemented | TensorBlock |
| typedef Eigen::internal::traits< XprType >::PointerType | TensorPointerType |
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 |
| EIGEN_DEVICE_FUNC const EIGEN_STRONG_INLINE Dimensions & | dimensions () const |
| EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (EvaluatorPointerType data) |
| template<int LoadMode> | |
| EIGEN_DEVICE_FUNC PacketReturnType | packet (Index index) const |
| EIGEN_STRONG_INLINE | TensorEvaluator (const ArgType &op, const Device &device) |
Static Public Attributes | |
| static const int | NumDims = internal::traits<ArgType>::NumDimensions |
| static const int | PacketSize = PacketType<CoeffReturnType, Device>::size |
Protected Member Functions | |
| void | evalTo (EvaluatorPointerType data) |
Protected Attributes | |
| const Device EIGEN_DEVICE_REF | m_device |
| Dimensions | m_dimensions |
| const ArgType | m_op |
| EvaluatorPointerType | m_result |
Definition at line 81 of file TensorCustomOp.h.
| typedef TensorCustomUnaryOp<CustomUnaryFunc, XprType> Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::ArgType |
Definition at line 83 of file TensorCustomOp.h.
| typedef internal::remove_const<typename XprType::CoeffReturnType>::type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::CoeffReturnType |
Definition at line 88 of file TensorCustomOp.h.
| typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Dimensions |
Definition at line 86 of file TensorCustomOp.h.
| typedef Storage::Type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::EvaluatorPointerType |
Definition at line 93 of file TensorCustomOp.h.
| typedef internal::traits<ArgType>::Index Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Index |
Definition at line 84 of file TensorCustomOp.h.
| typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::PacketReturnType |
Definition at line 89 of file TensorCustomOp.h.
| typedef internal::remove_const<typename ArgType::Scalar>::type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Scalar |
Definition at line 87 of file TensorCustomOp.h.
| typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Storage |
Definition at line 92 of file TensorCustomOp.h.
| typedef internal::TensorBlockNotImplemented Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::TensorBlock |
Definition at line 106 of file TensorCustomOp.h.
| typedef Eigen::internal::traits<XprType>::PointerType Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::TensorPointerType |
Definition at line 91 of file TensorCustomOp.h.
| anonymous enum |
| Enumerator | |
|---|---|
| IsAligned | |
| PacketAccess | |
| BlockAccess | |
| PreferBlockAccess | |
| Layout | |
| CoordAccess | |
| RawAccess | |
Definition at line 95 of file TensorCustomOp.h.
|
inline |
Definition at line 109 of file TensorCustomOp.h.
|
inline |
Definition at line 129 of file TensorCustomOp.h.
|
inline |
Definition at line 136 of file TensorCustomOp.h.
|
inline |
Definition at line 145 of file TensorCustomOp.h.
|
inline |
Definition at line 150 of file TensorCustomOp.h.
|
inline |
Definition at line 115 of file TensorCustomOp.h.
|
inline |
Definition at line 117 of file TensorCustomOp.h.
|
inlineprotected |
Definition at line 160 of file TensorCustomOp.h.
|
inline |
Definition at line 141 of file TensorCustomOp.h.
|
protected |
Definition at line 167 of file TensorCustomOp.h.
|
protected |
Definition at line 165 of file TensorCustomOp.h.
|
protected |
Definition at line 166 of file TensorCustomOp.h.
|
protected |
Definition at line 168 of file TensorCustomOp.h.
|
static |
Definition at line 85 of file TensorCustomOp.h.
|
static |
Definition at line 90 of file TensorCustomOp.h.