NEML2 1.4.0
|
This is the complete list of members for ImplicitUpdate, 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 |
_model | ImplicitUpdate | 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 |
_solver | ImplicitUpdate | 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 override | ImplicitUpdate | virtual |
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() | ImplicitUpdate | 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 | |
ImplicitUpdate(const OptionSet &name) | ImplicitUpdate | |
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 |
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) override | ImplicitUpdate | protectedvirtual |
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 |