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

Data Structures

struct  Meta
 

Public Types

using FixedDispatch = FixedValues< 1 >
 

Public Member Functions

 WeightedSumReducerGradient (const Meta &, const T *s_grad, CPUContext *)
 
template<int FixedSize>
void fillGrad (const Meta &meta, T *data_grad, TIndex offset, Context *context, const int)
 
template<int FixedSize>
void fillGradWithMainInput (const Meta &meta, const T *data, T *data_grad, TIndex offset, Context *context, const int)
 

Static Public Member Functions

static constexpr std::array< int, 1 > originalInputs ()
 
static int numAuxInputsWithGrads (const OperatorDef &def)
 
static bool requiresDataInput (const OperatorDef &def)
 
- 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::WeightedSumReducerGradient< T, Context >

Definition at line 524 of file reducer_functors.h.


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