Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Static Public Member Functions
caffe2::MaxReducerGradient< T, Context > Class Template Reference
Inheritance diagram for caffe2::MaxReducerGradient< T, Context >:
caffe2::BaseReducerGradient

Public Types

using FixedDispatch = FixedValues< 1 >
 

Public Member Functions

 MaxReducerGradient (const Meta &, const T *s_grad, CPUContext *)
 
template<int FixedSize>
void fillGradWithMainInputAndForwardOutput (const Meta &meta, const T *data, T *data_grad, const T *forward_output, TIndex, Context *, const int)
 

Static Public Member Functions

static bool requiresDataInput (const OperatorDef &)
 
static bool requiresForwardOutput ()
 
- Static Public Member Functions inherited from caffe2::BaseReducerGradient
static constexpr std::array< int, 0 > originalInputs ()
 
static constexpr bool computeLength ()
 
static int numAuxInputsWithGrads (const OperatorDef &)
 
static bool requiresDataInput (const OperatorDef &)
 
static bool requiresForwardOutput ()
 

Detailed Description

template<typename T, class Context>
class caffe2::MaxReducerGradient< T, Context >

Definition at line 772 of file reducer_functors.h.


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