|
NEML2 2.0.0
|
This is the complete list of members for ParameterStore, including all inherited members.
| assign_parameter_stack(jit::Stack &stack) | ParameterStore | protected |
| collect_parameter_stack() const | ParameterStore | protected |
| declare_parameter(const std::string &name, const T &rawval) | ParameterStore | protected |
| declare_parameter(const std::string &name, const TensorName< T > &tensorname, bool allow_nonlinear) | 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) const | ParameterStore | inline |
| get_parameter(const std::string &name) | ParameterStore | |
| named_parameters() const | ParameterStore | inline |
| named_parameters() | ParameterStore | |
| operator=(const ParameterStore &)=delete | ParameterStore | |
| operator=(ParameterStore &&)=delete | ParameterStore | |
| ParameterStore(Model *object) | ParameterStore | |
| ParameterStore(const ParameterStore &)=delete | ParameterStore | |
| ParameterStore(ParameterStore &&)=delete | ParameterStore | |
| send_parameters_to(const TensorOptions &options) | ParameterStore | protectedvirtual |
| set_parameter(const std::string &, const Tensor &) | ParameterStore | |
| set_parameters(const std::map< std::string, Tensor > &) | ParameterStore | |
| ~ParameterStore()=default | ParameterStore | virtual |