#include <TensorArgMax.h>
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 EIGEN_STRONG_INLINE const 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<InputDimensions>::value |
Protected Attributes | |
TensorEvaluator< const TensorReductionOp< ReduceOp, Dims, const TensorIndexTupleOp< ArgType > >, Device > | m_impl |
TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device > | m_orig_impl |
const Index | m_return_dim |
Index | m_stride_div |
Index | m_stride_mod |
StrideDims | m_strides |
Private Member Functions | |
EIGEN_DEVICE_FUNC void | gen_strides (const InputDimensions &dims, StrideDims &strides) |
Definition at line 214 of file TensorArgMax.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::CoeffReturnType |
Definition at line 219 of file TensorArgMax.h.
typedef TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Device>::Dimensions Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Dimensions |
Definition at line 221 of file TensorArgMax.h.
typedef Storage::Type Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::EvaluatorPointerType |
Definition at line 226 of file TensorArgMax.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Index |
Definition at line 217 of file TensorArgMax.h.
typedef TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::InputDimensions |
Definition at line 222 of file TensorArgMax.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Scalar |
Definition at line 218 of file TensorArgMax.h.
typedef StorageMemory<CoeffReturnType, Device> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Storage |
Definition at line 225 of file TensorArgMax.h.
typedef array<Index, NumDims> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::StrideDims |
Definition at line 224 of file TensorArgMax.h.
typedef internal::TensorBlockNotImplemented Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::TensorBlock |
Definition at line 240 of file TensorArgMax.h.
typedef StorageMemory<TupleType, Device> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::TupleStorageMem |
Definition at line 227 of file TensorArgMax.h.
typedef TensorIndexTupleOp<ArgType>::CoeffReturnType Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::TupleType |
Definition at line 220 of file TensorArgMax.h.
typedef TensorTupleReducerOp<ReduceOp, Dims, ArgType> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::XprType |
Definition at line 216 of file TensorArgMax.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
PreferBlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 229 of file TensorArgMax.h.
|
inline |
Definition at line 243 of file TensorArgMax.h.
|
inline |
Definition at line 270 of file TensorArgMax.h.
|
inline |
Definition at line 274 of file TensorArgMax.h.
|
inline |
Definition at line 288 of file TensorArgMax.h.
|
inline |
Definition at line 279 of file TensorArgMax.h.
|
inline |
Definition at line 262 of file TensorArgMax.h.
|
inline |
Definition at line 266 of file TensorArgMax.h.
|
inlineprivate |
Definition at line 296 of file TensorArgMax.h.
|
protected |
Definition at line 320 of file TensorArgMax.h.
|
protected |
Definition at line 319 of file TensorArgMax.h.
|
protected |
Definition at line 321 of file TensorArgMax.h.
|
protected |
Definition at line 324 of file TensorArgMax.h.
|
protected |
Definition at line 323 of file TensorArgMax.h.
|
protected |
Definition at line 322 of file TensorArgMax.h.
|
static |
Definition at line 223 of file TensorArgMax.h.