#include <TensorArgMax.h>
Public Types | |
enum | { IsAligned = false, PacketAccess = false, BlockAccess = false, PreferBlockAccess = TensorEvaluator<ArgType, Device>::PreferBlockAccess, Layout = TensorEvaluator<ArgType, Device>::Layout, CoordAccess = false, RawAccess = false } |
typedef XprType::CoeffReturnType | CoeffReturnType |
typedef TensorEvaluator< ArgType, Device >::Dimensions | Dimensions |
typedef Storage::Type | EvaluatorPointerType |
typedef XprType::Index | Index |
typedef XprType::Scalar | Scalar |
typedef StorageMemory< CoeffReturnType, Device > | Storage |
typedef internal::TensorBlockNotImplemented | TensorBlock |
typedef TensorIndexTupleOp< ArgType > | XprType |
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) |
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 Storage::Type Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::EvaluatorPointerType |
Definition at line 86 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 StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::Storage |
Definition at line 85 of file TensorArgMax.h.
typedef internal::TensorBlockNotImplemented Eigen::TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::TensorBlock |
Definition at line 99 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 | |
PreferBlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 88 of file TensorArgMax.h.
|
inline |
Definition at line 102 of file TensorArgMax.h.
|
inline |
Definition at line 113 of file TensorArgMax.h.
|
inline |
Definition at line 117 of file TensorArgMax.h.
|
inline |
Definition at line 123 of file TensorArgMax.h.
|
inline |
Definition at line 127 of file TensorArgMax.h.
|
inline |
Definition at line 105 of file TensorArgMax.h.
|
inline |
Definition at line 109 of file TensorArgMax.h.
|
protected |
Definition at line 136 of file TensorArgMax.h.
|
static |
Definition at line 84 of file TensorArgMax.h.