NEML2 1.4.0
|
This is the complete list of members for ParameterStore, including all inherited members.
_nl_param_models | ParameterStore | protected |
_nl_params | ParameterStore | protected |
declare_parameter(const std::string &name, const T &rawval) | ParameterStore | protected |
declare_parameter(const std::string &name, const std::string &input_option_name, bool allow_nonlinear=false) | ParameterStore | protected |
get_parameter(const std::string &name) | ParameterStore | |
get_parameter(const std::string &name) const | ParameterStore | |
has_nl_param() const | ParameterStore | inline |
named_nonlinear_parameter_models(bool recursive=false) const | ParameterStore | virtual |
named_nonlinear_parameters(bool recursive=false) const | ParameterStore | virtual |
named_parameters() const | ParameterStore | inline |
named_parameters() | ParameterStore | |
nl_param(const std::string &) const | ParameterStore | |
ParameterStore(const OptionSet &options, NEML2Object *object) | ParameterStore | |
send_parameters_to(const torch::TensorOptions &options) | ParameterStore | protectedvirtual |
set_parameter(const std::string &, const Tensor &) | ParameterStore | |
set_parameters(const std::map< std::string, Tensor > &) | ParameterStore |