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
DeferredComputeClear(NDArrayHandle *arrays, const int num)mxnet::Imperative
DropGrads(const std::vector< NDArray * > &variables)mxnet::Imperative
Get()mxnet::Imperativestatic
get_opt_constraints() constmxnet::Imperativeinline
GetDeferredComputeSymbol(const std::vector< NDArray * > &outputs)mxnet::Imperative
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_deferred_compute() constmxnet::Imperativeinline
is_np_default_dtype() constmxnet::Imperativeinline
is_np_shape() constmxnet::Imperativeinline
is_recording() constmxnet::Imperativeinline
is_training() constmxnet::Imperativeinline
ListNonleafVariables(const nnvm::Symbol &sym) constmxnet::Imperative
MarkVariables(const std::vector< NDArray * > &variables, const std::vector< uint32_t > &grad_reqs, const std::vector< NDArray * > &gradients)mxnet::Imperative
NDArray classmxnet::Imperativefriend
PreferBulkExecInference()mxnet::Imperativeinlinestatic
PreferBulkExecTrain()mxnet::Imperativeinlinestatic
RecordDeferredCompute(nnvm::NodeAttrs &&attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs)mxnet::Imperative
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_deferred_compute(bool is_deferred_compute)mxnet::Imperativeinline
set_is_np_default_dtype(bool is_np_default_dtype)mxnet::Imperativeinline
set_is_np_shape(int is_np_shape)mxnet::Imperativeinline
set_is_recording(bool is_recording)mxnet::Imperativeinline
set_is_training(bool is_train)mxnet::Imperativeinline
set_opt_constraints(OptConstraint constraints)mxnet::Imperativeinline
SetDeferredComputeVariable(NDArrayHandle *arrays, SymbolHandle *variables, const int num)mxnet::Imperative