Skip to content

ArgmaxTraits

#include <ncarray/reductions.hh>
struct ArgmaxTraits

Defined in src/lib/ncarray/reductions.hh:186

Traits to find the indices of maxima along axes.

Return Name Description
NCA_HDAccumT< T > identity static inline
NCA_HD auto fill static inline
NCA_HDAccumT< T > transform static inline
NCA_HDAccumT< T > reduce static inline
NCA_HD auto store static inline
NCA_D void atomic static inline
NCA_D void dual_atomic static inline

static inline

template<typename T> static inline NCA_HDAccumT< T > identity()

Defined in src/lib/ncarray/reductions.hh:194


static inline

template<typename T> static inline NCA_HD auto fill()

Defined in src/lib/ncarray/reductions.hh:199


static inline

template<typename T> static inline NCA_HDAccumT< T > transform(ssize_t idx, T val)

Defined in src/lib/ncarray/reductions.hh:202


static inline

template<typename T> static inline NCA_HDAccumT< T > reduce(AccumT< T > a, AccumT< T > b)

Defined in src/lib/ncarray/reductions.hh:207


static inline

template<typename T> static inline NCA_HD auto store(AccumT< T > res, double ddof = 0.0)

Defined in src/lib/ncarray/reductions.hh:215


static inline

template<typename T> static inline NCA_D void atomic(AccumT< T > * dest, AccumT< T > val)

Defined in src/lib/ncarray/reductions.hh:220


static inline

template<typename OutT, typename SrcT> static inline NCA_D void dual_atomic(AccumT< SrcT > * scratch, AccumT< SrcT > val, OutT * res, double ddof = 0.0)

Defined in src/lib/ncarray/reductions.hh:227

Name Description
AccumT
OutT

using AccumT = KeyValPair< ssize_t, T >

Type: KeyValPair< ssize_t, T >

Defined in src/lib/ncarray/reductions.hh:188


using OutT = std::int64_t

Defined in src/lib/ncarray/reductions.hh:191