caffe2::ArgMaxReducer< Context > Struct Template Reference

Public Member Functions
template<typename T >
bool	operator() (const int prev_size, const int next_size, const int n, const T *X, int64_t *Y, Context *context) const
template<>
bool	operator() (const int prev_size, const int next_size, const int n, const T *X, int64_t *Y, CPUContext *context) const

Detailed Description

template<class Context>
struct caffe2::ArgMaxReducer< Context >

Definition at line 75 of file arg_ops.h.

---

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

• caffe2/operators/arg_ops.h