Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
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: