mxnet
mxnet::Imperative::CachedOp Member List

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::CachedOpinline
num_backward_inputs()mxnet::Imperative::CachedOpinline
num_inputs()mxnet::Imperative::CachedOpinline
num_outputs()mxnet::Imperative::CachedOpinline
save_inputs()mxnet::Imperative::CachedOpinline
save_outputs()mxnet::Imperative::CachedOpinline