Model Member List

This is the complete list of members for Model, including all inherited members.

_AD_1st_deriv	Model	protected
_AD_2nd_deriv	Model	protected
_autoscale	NonlinearSystem	protected
_autoscale_miter	NonlinearSystem	protected
_autoscale_tol	NonlinearSystem	protected
_col_scaling	NonlinearSystem	protected
_Jacobian	NonlinearSystem	protected
_ndof	NonlinearSystem	protected
_registered_data	Data	protected
_registered_models	Model	protected
_residual	NonlinearSystem	protected
_row_scaling	NonlinearSystem	protected
_scaled_Jacobian	NonlinearSystem	protected
_scaled_residual	NonlinearSystem	protected
_scaling_matrices_initialized	NonlinearSystem	protected
_solution	NonlinearSystem	protected
allocate_variables(int deriv_order, bool options_changed)	Model	protectedvirtual
allocate_variables(TorchShapeRef batch_shape, const torch::TensorOptions &options, bool in, bool out, bool dout_din, bool d2out_din2)	Model	protectedvirtual
assemble(bool residual, bool Jacobian) override	Model	protectedvirtual
batch_dim() const	Model	inline
batch_sizes() const	Model	inline
BufferStore(const OptionSet &options, NEML2Object *object)	BufferStore
cache(TorchShapeRef batch_shape) override	Model	protectedvirtual
cache(const torch::TensorOptions &options)	Model	protectedvirtual
cache(TorchShapeRef batch_shape)	Model	protectedvirtual
check_AD_limitation() const	Model	virtual
ComposedModel	Model	friend
consumed_items() const override	Model	virtual
Data(const OptionSet &options)	Data
declare_axis(const std::string &name)	VariableStore
declare_buffer(const std::string &name, const T &rawval)	BufferStore	protected
declare_buffer(const std::string &name, const std::string &input_option_name)	BufferStore	protected
declare_input_variable(S &&... name)	VariableStore	inlineprotected
declare_input_variable(TorchSize sz, S &&... name)	VariableStore	inlineprotected
declare_input_variable_list(TorchSize list_size, S &&... name)	VariableStore	inlineprotected
declare_output_variable(S &&... name)	VariableStore	inlineprotected
declare_output_variable(TorchSize sz, S &&... name)	VariableStore	inlineprotected
declare_output_variable_list(TorchSize list_size, S &&... name)	VariableStore	inlineprotected
declare_parameter(const std::string &name, const T &rawval)	ParameterStore	protected
declare_parameter(const std::string &name, const std::string &input_option_name)	ParameterStore	protected
declare_subaxis(LabeledAxis &axis, const VariableName &subaxis) const	VariableStore	inlineprotected
declare_variable(LabeledAxis &axis, const VariableName &var) const	VariableStore	inlineprotected
declare_variable(LabeledAxis &axis, const VariableName &var, TorchSize sz) const	VariableStore	inlineprotected
derivative_storage()	VariableStore	inline
derivative_storage() const	VariableStore	inline
detach_and_zero(bool out, bool dout_din=true, bool d2out_din2=true) override	Model	protectedvirtual
disable_automatic_scaling(OptionSet &options)	NonlinearSystem	static
doc() const	NEML2Object	inline
enable_automatic_scaling(OptionSet &options)	NonlinearSystem	static
expected_options()	Model	static
get_buffer(const std::string &name)	BufferStore
get_d2output_dinput2()	Model	virtual
get_doutput_dinput()	Model	virtual
get_input_variable(const VariableName &name)	VariableStore	inline
get_input_variable(const VariableName &name) const	VariableStore	inline
get_output()	Model	virtual
get_output_variable(const VariableName &name)	VariableStore	inline
get_output_variable(const VariableName &name) const	VariableStore	inline
get_parameter(const std::string &name)	ParameterStore
has_nl_param() const	ParameterStore	inline
host() const	NEML2Object
host()	NEML2Object
init_scaling(const bool verbose=false)	NonlinearSystem	virtual
input_axis()	VariableStore	inline
input_axis() const	VariableStore	inline
input_options() const	NEML2Object	inline
input_requires_grad_(bool req=true)	Model
input_storage()	VariableStore	inline
input_storage() const	VariableStore	inline
input_view(const VariableName &)	VariableStore
input_views()	VariableStore	inline
input_views() const	VariableStore	inline
is_nonlinear_system() const	Model	inlinevirtual
Jacobian(const BatchTensor &x)	NonlinearSystem
Jacobian()	NonlinearSystem
Jacobian_view() const	NonlinearSystem	inline
Model(const OptionSet &options)	Model
name() const	NEML2Object	inline
named_buffers() const	BufferStore	inline
named_buffers()	BufferStore
named_parameters() const	ParameterStore	inline
named_parameters()	ParameterStore
NEML2Object(const OptionSet &options)	NEML2Object
nl_param(const std::string &) const	ParameterStore
nl_params() const	ParameterStore	inline
NonlinearSystem(const OptionSet &options)	NonlinearSystem
options() const	Model	inline
output_axis()	VariableStore	inline
output_axis() const	VariableStore	inline
output_storage()	VariableStore	inline
output_storage() const	VariableStore	inline
output_view(const VariableName &)	VariableStore
output_views()	VariableStore	inline
output_views() const	VariableStore	inline
ParameterStore	Model	friend
neml2::ParameterStore::ParameterStore(const OptionSet &options, NEML2Object *object)	ParameterStore
path() const	NEML2Object	inline
preflight() const	Model	virtual
provided_items() const override	Model	virtual
register_data(const std::string &name)	Data	inlineprotected
register_model(const std::string &name, int extra_deriv_order=0, bool nonlinear=false, bool merge_input=true)	Model	inlineprotected
registered_data() const	Data	inline
registered_model(const std::string &name) const	Model
registered_models() const	Model	inline
reinit(TorchShapeRef batch_shape, int deriv_order=0, const torch::Device &device=torch::kCPU, const torch::Dtype &dtype=NEML2_DTYPE)	Model	virtual
reinit(const BatchTensor &tensor, int deriv_order)	Model	virtual
reinit_input_views() override	Model	protectedvirtual
reinit_output_views(bool out, bool dout_din=true, bool d2out_din2=true) override	Model	protectedvirtual
requires_2nd_grad() const	Model	inline
requires_grad() const	Model	inline
residual(const BatchTensor &x)	NonlinearSystem
residual()	NonlinearSystem
residual_and_Jacobian(const BatchTensor &x)	NonlinearSystem
residual_and_Jacobian()	NonlinearSystem
residual_norm() const	NonlinearSystem
residual_view() const	NonlinearSystem	inline
scale_direction(const BatchTensor &p) const	NonlinearSystem
scale_Jacobian(const BatchTensor &J) const	NonlinearSystem
scale_residual(const BatchTensor &r) const	NonlinearSystem
second_derivative_storage()	VariableStore	inline
second_derivative_storage() const	VariableStore	inline
send_buffers_to(const torch::TensorOptions &options)	BufferStore	protectedvirtual
send_parameters_to(const torch::TensorOptions &options)	ParameterStore	protectedvirtual
set_input(const LabeledVector &in)	Model	virtual
set_solution(const BatchTensor &x) override	Model	protectedvirtual
set_value(bool out, bool dout_din, bool d2out_din2)=0	Model	protectedpure virtual
setup() override	Model	protectedvirtual
setup_input_views() override	Model	protectedvirtual
setup_layout()	VariableStore	virtual
setup_output_views() override	Model	protectedvirtual
setup_submodel_input_views()	Model	protectedvirtual
setup_submodel_output_views()	Model	protectedvirtual
solution() const	NonlinearSystem	inlinevirtual
SOLVING enum value	Model
stage	Model	static
Stage enum name	Model
type() const	NEML2Object	inline
UPDATING enum value	Model
use_AD_derivatives(bool first=true, bool second=true)	Model
using_AD_1st_derivative() const	Model	inline
using_AD_2nd_derivative() const	Model	inline
value(const LabeledVector &in)	Model	virtual
value()	Model	virtual
value_and_dvalue(const LabeledVector &in)	Model	virtual
value_and_dvalue()	Model	virtual
value_and_dvalue_and_d2value(const LabeledVector &in)	Model	virtual
value_and_dvalue_and_d2value()	Model	virtual
VariableStore(const OptionSet &options, NEML2Object *object)	VariableStore
~NEML2Object()=default	NEML2Object	virtual