#include <TensorCustomOp.h>
Public Types | |
enum | { IsAligned = false, PacketAccess = (internal::packet_traits<Scalar>::size > 1), BlockAccess = 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 internal::traits< ArgType >::Index | Index |
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
typedef internal::remove_const< typename ArgType::Scalar >::type | Scalar |
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 CoeffReturnType * | data () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions & | dimensions () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (CoeffReturnType *data) |
template<int LoadMode> | |
EIGEN_DEVICE_FUNC PacketReturnType | packet (Index index) const |
EIGEN_DEVICE_FUNC 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 = internal::unpacket_traits<PacketReturnType>::size |
Protected Member Functions | |
EIGEN_DEVICE_FUNC void | evalTo (Scalar *data) |
Protected Attributes | |
const Device & | m_device |
Dimensions | m_dimensions |
const ArgType | m_op |
CoeffReturnType * | m_result |
Definition at line 80 of file TensorCustomOp.h.
typedef TensorCustomUnaryOp<CustomUnaryFunc, XprType> Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::ArgType |
Definition at line 82 of file TensorCustomOp.h.
typedef internal::remove_const<typename XprType::CoeffReturnType>::type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::CoeffReturnType |
Definition at line 87 of file TensorCustomOp.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Dimensions |
Definition at line 85 of file TensorCustomOp.h.
typedef internal::traits<ArgType>::Index Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Index |
Definition at line 83 of file TensorCustomOp.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::PacketReturnType |
Definition at line 88 of file TensorCustomOp.h.
typedef internal::remove_const<typename ArgType::Scalar>::type Eigen::TensorEvaluator< const TensorCustomUnaryOp< CustomUnaryFunc, XprType >, Device >::Scalar |
Definition at line 86 of file TensorCustomOp.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 91 of file TensorCustomOp.h.
|
inline |
Definition at line 100 of file TensorCustomOp.h.
|
inline |
Definition at line 120 of file TensorCustomOp.h.
|
inline |
Definition at line 127 of file TensorCustomOp.h.
|
inline |
Definition at line 136 of file TensorCustomOp.h.
|
inline |
Definition at line 141 of file TensorCustomOp.h.
|
inline |
Definition at line 106 of file TensorCustomOp.h.
|
inline |
Definition at line 108 of file TensorCustomOp.h.
|
inlineprotected |
Definition at line 144 of file TensorCustomOp.h.
|
inline |
Definition at line 132 of file TensorCustomOp.h.
|
protected |
Definition at line 152 of file TensorCustomOp.h.
|
protected |
Definition at line 150 of file TensorCustomOp.h.
|
protected |
Definition at line 151 of file TensorCustomOp.h.
|
protected |
Definition at line 153 of file TensorCustomOp.h.
|
static |
Definition at line 84 of file TensorCustomOp.h.
|
static |
Definition at line 89 of file TensorCustomOp.h.