#include <TensorArgMax.h>
Public Types | |
enum | { IsAligned = false, PacketAccess = false, BlockAccess = false, Layout = TensorEvaluator<ArgType, Device>::Layout, CoordAccess = false, RawAccess = false } |
typedef XprType::CoeffReturnType | CoeffReturnType |
typedef TensorEvaluator< ArgType, Device >::Dimensions | Dimensions |
typedef XprType::Index | Index |
typedef XprType::Scalar | Scalar |
typedef TensorIndexTupleOp< 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 *) |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE | TensorEvaluator (const XprType &op, const Device &device) |
Static Public Attributes | |
static const int | NumDims = internal::array_size<Dimensions>::value |
Protected Attributes | |
TensorEvaluator< ArgType, Device > | m_impl |
Definition at line 76 of file TensorArgMax.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::CoeffReturnType |
Definition at line 81 of file TensorArgMax.h.
typedef TensorEvaluator<ArgType, Device>::Dimensions Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::Dimensions |
Definition at line 83 of file TensorArgMax.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::Index |
Definition at line 79 of file TensorArgMax.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::Scalar |
Definition at line 80 of file TensorArgMax.h.
typedef TensorIndexTupleOp<ArgType> Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::XprType |
Definition at line 78 of file TensorArgMax.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 86 of file TensorArgMax.h.
|
inline |
Definition at line 95 of file TensorArgMax.h.
|
inline |
Definition at line 106 of file TensorArgMax.h.
|
inline |
Definition at line 110 of file TensorArgMax.h.
|
inline |
Definition at line 116 of file TensorArgMax.h.
|
inline |
Definition at line 120 of file TensorArgMax.h.
|
inline |
Definition at line 98 of file TensorArgMax.h.
|
inline |
Definition at line 102 of file TensorArgMax.h.
|
protected |
Definition at line 123 of file TensorArgMax.h.
|
static |
Definition at line 84 of file TensorArgMax.h.