Skip to content

ArgmaxTraits

#include <reductions.hh>
struct ArgmaxTraits

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

Traits to find the indices of maxima along axes.

Name Kind Owner
identity function Declared here
fill function Declared here
transform function Declared here
reduce function Declared here
store function Declared here
atomic function Declared here
dual_atomic function Declared here
AccumT typedef Declared here
OutT typedef Declared here
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 >

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


using OutT = std::int64_t

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