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) | mxnet::Imperative::CachedOp | explicit |
Forward(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 |