Caffe2 - C++ API
A deep learning, cross platform ML framework
caffe2::GetPrependDimGradient Class Reference
Inheritance diagram for caffe2::GetPrependDimGradient:
caffe2::GradientMakerBase

Additional Inherited Members

- Public Member Functions inherited from caffe2::GradientMakerBase
 GradientMakerBase (const OperatorDef &def, const vector< GradientWrapper > &g_output)
 
virtual bool CopyDeviceOption () const
 
virtual bool CopyEngine () const
 
virtual void VerifyOp () const
 
virtual GradientOpsMeta Get ()
 Returns the gradient ops meta. More...
 
const OperatorDef & Def () const
 
- Static Public Member Functions inherited from caffe2::GradientMakerBase
static CaffeMap< string, string > MatchGradsToParams (const OperatorDef &op)
 Returns map that returns the parameters that the gradients are for.
 
- Protected Member Functions inherited from caffe2::GradientMakerBase
string I (const int i)
 
string O (const int i)
 
string GI (const int i)
 
string GI_I (const int i)
 
string GI_V (const int i)
 
string GO (const int i)
 
string GO_I (const int i)
 
string GO_V (const int i)
 
const GradientWrapperGradOut (int i)
 
void SetDense (const int i, const string &name)
 
void SetSparse (const int i, const string &indices, const string &values)
 
- Static Protected Member Functions inherited from caffe2::GradientMakerBase
template<class... Args>
static vector< OperatorDef > SingleGradientDef (const Args &...args)
 a helper function to allow one to create one single operator def, which is usually the case for many simple operators.
 
- Protected Attributes inherited from caffe2::GradientMakerBase
const OperatorDef & def_
 
const vector< GradientWrapper > & g_output_
 
vector< GradientWrapperg_input_
 

Detailed Description

Definition at line 26 of file prepend_dim_op.cc.


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