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

Public Types

using FixedDispatch = FixedValues< 1 >
 

Public Member Functions

 SumReducerGradient (const Meta &, const T *s_grad, CPUContext *)
 
template<int FixedSize>
void fillGrad (const Meta &meta, T *data_grad, int64_t offset, Context *context, const int length)
 

Additional Inherited Members

- 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::SumReducerGradient< T, Context >

Definition at line 423 of file reducer_functors.h.


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