Public Types | Public Member Functions | Static Public Attributes | Protected Attributes | Private Member Functions | List of all members
Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device > Struct Template Reference

#include <TensorArgMax.h>

Public Types

enum  {
  IsAligned = false, PacketAccess = false, BlockAccess = false, Layout = TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Device>::Layout,
  CoordAccess = false, RawAccess = false
}
 
typedef XprType::CoeffReturnType CoeffReturnType
 
typedef TensorEvaluator< const TensorReductionOp< ReduceOp, Dims, const TensorIndexTupleOp< ArgType > >, Device >::Dimensions Dimensions
 
typedef XprType::Index Index
 
typedef TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device >::Dimensions InputDimensions
 
typedef XprType::Scalar Scalar
 
typedef array< Index, NumDims > StrideDims
 
typedef TensorIndexTupleOp< ArgType >::CoeffReturnType TupleType
 
typedef TensorTupleReducerOp< ReduceOp, Dims, 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<InputDimensions>::value
 

Protected Attributes

TensorEvaluator< const TensorReductionOp< ReduceOp, Dims, const TensorIndexTupleOp< ArgType > >, Device > m_impl
 
TensorEvaluator< const TensorIndexTupleOp< ArgType >, Device > m_orig_impl
 
const int 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)
 

Detailed Description

template<typename ReduceOp, typename Dims, typename ArgType, typename Device>
struct Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >

Definition at line 201 of file TensorArgMax.h.

Member Typedef Documentation

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::CoeffReturnType

Definition at line 206 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Device>::Dimensions Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Dimensions

Definition at line 208 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef XprType::Index Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Index

Definition at line 204 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef TensorEvaluator<const TensorIndexTupleOp<ArgType> , Device>::Dimensions Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::InputDimensions

Definition at line 209 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::Scalar

Definition at line 205 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef array<Index, NumDims> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::StrideDims

Definition at line 211 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef TensorIndexTupleOp<ArgType>::CoeffReturnType Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::TupleType

Definition at line 207 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
typedef TensorTupleReducerOp<ReduceOp, Dims, ArgType> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::XprType

Definition at line 203 of file TensorArgMax.h.

Member Enumeration Documentation

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
anonymous enum
Enumerator
IsAligned 
PacketAccess 
BlockAccess 
Layout 
CoordAccess 
RawAccess 

Definition at line 213 of file TensorArgMax.h.

Constructor & Destructor Documentation

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::TensorEvaluator ( const XprType &  op,
const Device &  device 
)
inline

Definition at line 222 of file TensorArgMax.h.

Member Function Documentation

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::cleanup ( )
inline

Definition at line 246 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::coeff ( Index  index) const
inline

Definition at line 250 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::costPerCoeff ( bool  vectorized) const
inline

Definition at line 258 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC Scalar* Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::data ( ) const
inline

Definition at line 255 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::dimensions ( ) const
inline

Definition at line 238 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::evalSubExprsIfNeeded ( Scalar *  )
inline

Definition at line 242 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
EIGEN_DEVICE_FUNC void Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::gen_strides ( const InputDimensions &  dims,
StrideDims &  strides 
)
inlineprivate

Definition at line 266 of file TensorArgMax.h.

Member Data Documentation

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
TensorEvaluator<const TensorReductionOp<ReduceOp, Dims, const TensorIndexTupleOp<ArgType> >, Device> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::m_impl
protected

Definition at line 290 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
TensorEvaluator<const TensorIndexTupleOp<ArgType>, Device> Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::m_orig_impl
protected

Definition at line 289 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
const int Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::m_return_dim
protected

Definition at line 291 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
Index Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::m_stride_div
protected

Definition at line 294 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
Index Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::m_stride_mod
protected

Definition at line 293 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
StrideDims Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::m_strides
protected

Definition at line 292 of file TensorArgMax.h.

template<typename ReduceOp , typename Dims , typename ArgType , typename Device >
const int Eigen::TensorEvaluator< const TensorTupleReducerOp< ReduceOp, Dims, ArgType >, Device >::NumDims = internal::array_size<InputDimensions>::value
static

Definition at line 210 of file TensorArgMax.h.

The documentation for this struct was generated from the following file:

hebiros
Author(s): Xavier Artache , Matthew Tesch
autogenerated on Thu Sep 3 2020 04:10:24