|
| GradientMakerBase (const OperatorDef &def, const vector< GradientWrapper > &g_output) |
|
virtual bool | CopyDeviceOption () const |
|
virtual bool | CopyEngine () const |
|
virtual bool | CopyArguments () const |
|
virtual void | VerifyOp () const |
|
virtual GradientOpsMeta | Get () |
| Returns the gradient ops meta. More...
|
|
const OperatorDef & | Def () const |
|
static CaffeMap< string, string > | MatchGradsToParams (const OperatorDef &op) |
| Returns map that returns the parameters that the gradients are for.
|
|
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 GradientWrapper & | GradOut (int i) |
|
void | SetDense (const int i, const string &name) |
|
void | SetSparse (const int i, const string &indices, const string &values) |
|
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.
|
|
const OperatorDef & | def_ |
|
const vector< GradientWrapper > & | g_output_ |
|
vector< GradientWrapper > | g_input_ |
|
Definition at line 343 of file top_k.cc.
The documentation for this class was generated from the following file: