mxnet
|
This is the complete list of members for mxnet::Imperative, including all inherited members.
Backward(const std::vector< NDArray * > &outputs, const std::vector< NDArray * > &ograds, const std::vector< NDArray * > &variables, bool is_train, bool retain_graph, bool create_graph) | mxnet::Imperative | |
BulkExecMaxNodeTrainBwd() | mxnet::Imperative | inlinestatic |
BulkExecMaxNodeTrainFwd() | mxnet::Imperative | inlinestatic |
Get() | mxnet::Imperative | static |
Invoke(const Context &default_ctx, const nnvm::NodeAttrs &attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs) | mxnet::Imperative | |
InvokeOp(const Context &ctx, const nnvm::NodeAttrs &attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs, const std::vector< OpReqType > &req, const DispatchMode dispatch_mode, OpStatePtr state=OpStatePtr()) | mxnet::Imperative | |
is_np_shape() const | mxnet::Imperative | inline |
is_recording() const | mxnet::Imperative | inline |
is_training() const | mxnet::Imperative | inline |
MarkVariables(const std::vector< NDArray * > &variables, const std::vector< mx_uint > &grad_reqs, const std::vector< NDArray * > &gradients) | mxnet::Imperative | |
NDArray class | mxnet::Imperative | friend |
PreferBulkExecInference() | mxnet::Imperative | inlinestatic |
PreferBulkExecTrain() | mxnet::Imperative | inlinestatic |
RecordOp(nnvm::NodeAttrs &&attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs, const OpStatePtr &state=OpStatePtr(), std::vector< bool > *p_save_inputs=nullptr, std::vector< bool > *p_save_outputs=nullptr) | mxnet::Imperative | |
set_is_np_shape(bool is_np_shape) | mxnet::Imperative | inline |
set_is_recording(bool is_recording) | mxnet::Imperative | inline |
set_is_training(bool is_train) | mxnet::Imperative | inline |