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 |
DeferredComputeClear(NDArrayHandle *arrays, const int num) | mxnet::Imperative | |
DropGrads(const std::vector< NDArray * > &variables) | mxnet::Imperative | |
Get() | mxnet::Imperative | static |
get_opt_constraints() const | mxnet::Imperative | inline |
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() const | mxnet::Imperative | inline |
is_np_default_dtype() const | mxnet::Imperative | inline |
is_np_shape() const | mxnet::Imperative | inline |
is_recording() const | mxnet::Imperative | inline |
is_training() const | mxnet::Imperative | inline |
ListNonleafVariables(const nnvm::Symbol &sym) const | mxnet::Imperative | |
MarkVariables(const std::vector< NDArray * > &variables, const std::vector< uint32_t > &grad_reqs, const std::vector< NDArray * > &gradients) | mxnet::Imperative | |
NDArray class | mxnet::Imperative | friend |
PreferBulkExecInference() | mxnet::Imperative | inlinestatic |
PreferBulkExecTrain() | mxnet::Imperative | inlinestatic |
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::Imperative | inline |
set_is_np_default_dtype(bool is_np_default_dtype) | mxnet::Imperative | inline |
set_is_np_shape(int is_np_shape) | mxnet::Imperative | inline |
set_is_recording(bool is_recording) | mxnet::Imperative | inline |
set_is_training(bool is_train) | mxnet::Imperative | inline |
set_opt_constraints(OptConstraint constraints) | mxnet::Imperative | inline |
SetDeferredComputeVariable(NDArrayHandle *arrays, SymbolHandle *variables, const int num) | mxnet::Imperative |