|
mxnet
|
This is the complete list of members for mxnet::Imperative::CachedOp, including all inherited members.
| Backward(const bool retain_graph, const OpStatePtr &state, const std::vector< NDArray * > &inputs, const std::vector< OpReqType > &reqs, const std::vector< NDArray * > &outputs) | mxnet::Imperative::CachedOp | |
| CachedOp(const nnvm::Symbol &sym, const std::vector< std::pair< std::string, std::string > > &kwargs) | mxnet::Imperative::CachedOp | |
| Forward(const std::shared_ptr< CachedOp > &op_ptr, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs) | mxnet::Imperative::CachedOp | |
| GetBackwardGraph(const OpStatePtr &state, const std::vector< OpReqType > &reqs, const std::vector< NDArray * > &inputs) | mxnet::Imperative::CachedOp | |
| GetForwardGraph(const bool recording, const std::vector< NDArray * > &inputs) | mxnet::Imperative::CachedOp | |
| Gradient(const nnvm::NodePtr &node, const std::vector< nnvm::NodeEntry > &ograds) | mxnet::Imperative::CachedOp | |
| mutable_input_nodes() | mxnet::Imperative::CachedOp | inline |
| num_backward_inputs() | mxnet::Imperative::CachedOp | inline |
| num_inputs() | mxnet::Imperative::CachedOp | inline |
| num_outputs() | mxnet::Imperative::CachedOp | inline |
| save_inputs() | mxnet::Imperative::CachedOp | inline |
| save_outputs() | mxnet::Imperative::CachedOp | inline |
1.8.11