mxnet
mxnet::Imperative Member List

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::Imperativeinlinestatic
BulkExecMaxNodeTrainFwd()mxnet::Imperativeinlinestatic
Get()mxnet::Imperativestatic
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::Imperativeinline
is_recording() const mxnet::Imperativeinline
is_training() const mxnet::Imperativeinline
MarkVariables(const std::vector< NDArray * > &variables, const std::vector< mx_uint > &grad_reqs, const std::vector< NDArray * > &gradients)mxnet::Imperative
NDArray classmxnet::Imperativefriend
PreferBulkExecInference()mxnet::Imperativeinlinestatic
PreferBulkExecTrain()mxnet::Imperativeinlinestatic
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::Imperativeinline
set_is_recording(bool is_recording)mxnet::Imperativeinline
set_is_training(bool is_train)mxnet::Imperativeinline