mxnet
|
This is the complete list of members for mxnet::Executor, including all inherited members.
arg_grad_map() const =0 | mxnet::Executor | pure virtual |
aux_state_map() const =0 | mxnet::Executor | pure virtual |
Backward(const std::vector< NDArray > &head_grads, bool is_train=true)=0 | mxnet::Executor | pure virtual |
Bind(nnvm::Symbol symbol, const Context &default_ctx, const std::map< std::string, Context > &group2ctx, const std::vector< NDArray > &in_args, const std::vector< NDArray > &arg_grad_store, const std::vector< OpReqType > &grad_req_type, const std::vector< NDArray > &aux_states, Executor *shared_exec=NULL) | mxnet::Executor | static |
Forward(bool is_train)=0 | mxnet::Executor | pure virtual |
in_arg_map() const =0 | mxnet::Executor | pure virtual |
MonitorCallback typedef | mxnet::Executor | |
outputs() const =0 | mxnet::Executor | pure virtual |
PartialForward(bool is_train, int step, int *step_left)=0 | mxnet::Executor | pure virtual |
Print(std::ostream &os) const | mxnet::Executor | inlinevirtual |
Reshape(const bool partial_shaping, const bool allow_up_sizing, const Context &default_ctx, const std::map< std::string, Context > &ctx_map, const std::unordered_map< std::string, TShape > &provided_arg_shapes, std::vector< NDArray > *in_args, std::vector< NDArray > *arg_grads, std::vector< NDArray > *aux_states)=0 | mxnet::Executor | pure virtual |
SetMonitorCallback(const MonitorCallback &callback) | mxnet::Executor | inlinevirtual |
SimpleBind(nnvm::Symbol symbol, const Context &default_ctx, const std::map< std::string, Context > &group2ctx, const std::vector< Context > &in_arg_ctxes, const std::vector< Context > &arg_grad_ctxes, const std::vector< Context > &aux_state_ctxes, const std::unordered_map< std::string, TShape > &arg_shape_map, const std::unordered_map< std::string, int > &arg_dtype_map, const std::unordered_map< std::string, int > &arg_stype_map, const std::vector< OpReqType > &grad_req_types, const std::unordered_set< std::string > ¶m_names, std::vector< NDArray > *in_args, std::vector< NDArray > *arg_grads, std::vector< NDArray > *aux_states, std::unordered_map< std::string, NDArray > *shared_data_arrays=nullptr, Executor *shared_exec=nullptr) | mxnet::Executor | static |
~Executor() | mxnet::Executor | inlinevirtual |