caffe2::WeightedSumReducerGradient< T, Context > Class Template Reference

Inheritance diagram for caffe2::WeightedSumReducerGradient< T, Context >:

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, int64_t offset, Context *context, const int)
template<int FixedSize>
void	fillGradWithMainInput (const Meta &meta, const T *data, T *data_grad, int64_t 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 504 of file reducer_functors.h.

---

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

• caffe2/operators/reducer_functors.h