|
|
vector< OperatorDef > | GetGradientDefs () override |
| |
|
| 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 562 of file recurrent_op_cudnn.cc.
The documentation for this struct was generated from the following file: