NEML2 1.4.0
|
This is the complete list of members for CopyVariable< T >, including all inherited members.
_autoscale | NonlinearSystem | protected |
_autoscale_miter | NonlinearSystem | protected |
_autoscale_tol | NonlinearSystem | protected |
_col_scaling | NonlinearSystem | protected |
_from | CopyVariable< T > | protected |
_Jacobian | NonlinearSystem | protected |
_ndof | NonlinearSystem | protected |
_nl_param_models | ParameterStore | protected |
_nl_params | ParameterStore | 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 |
_to | CopyVariable< T > | protected |
allocate_variables(bool in, bool out) | Model | protectedvirtual |
allocate_variables(TensorShapeRef 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(TensorShapeRef batch_shape, int deriv_order, const torch::Device &device, const torch::Dtype &dtype) | Model | protectedvirtual |
cache(TensorShapeRef batch_shape) | Model | protectedvirtual |
check_AD_limitation() const | Model | protectedvirtual |
check_inplace_dbg() | Model | protected |
check_input(const LabeledVector &in) const | Model | protectedvirtual |
consumed_items() const override | Model | virtual |
CopyVariable(const OptionSet &options) | CopyVariable< T > | |
d2value() | Model | virtual |
d2value(const LabeledVector &in) | 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(Size sz, TensorType t, S &&... name) | VariableStore | inlineprotected |
declare_input_variable_list(Size list_size, S &&... name) | VariableStore | inlineprotected |
declare_output_variable(S &&... name) | VariableStore | inlineprotected |
declare_output_variable(Size sz, TensorType t, S &&... name) | VariableStore | inlineprotected |
declare_output_variable_list(Size 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, bool allow_nonlinear=false) | 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, Size sz) const | VariableStore | inlineprotected |
derivative_storage() | VariableStore | inline |
derivative_storage() const | VariableStore | inline |
device() const | Model | inline |
diagnose(std::vector< Diagnosis > &) const override | Model | virtual |
diagnose_nl_sys(std::vector< Diagnosis > &diagnoses) const | Model | |
diagnostic_assert(std::vector< Diagnosis > &diagnoses, bool assertion, Args &&... args) const | DiagnosticsInterface | |
diagnostic_assert_force(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
diagnostic_assert_old_force(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
diagnostic_assert_old_state(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
diagnostic_assert_residual(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
diagnostic_assert_state(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
diagnostic_check_input_variable(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
diagnostic_check_output_variable(std::vector< Diagnosis > &diagnoses, const VariableBase &v) const | DiagnosticsInterface | |
DiagnosticsInterface(NEML2Object *object) | DiagnosticsInterface | |
disable_automatic_scaling(OptionSet &options) | NonlinearSystem | static |
doc() const | NEML2Object | inline |
dvalue() | Model | virtual |
dvalue(const LabeledVector &in) | Model | virtual |
dvalue_and_d2value() | Model | virtual |
dvalue_and_d2value(const LabeledVector &in) | Model | virtual |
enable_automatic_scaling(OptionSet &options) | NonlinearSystem | static |
expected_options() | CopyVariable< T > | static |
get_buffer(const std::string &name) | BufferStore | |
get_d2output_dinput2() | Model | protectedvirtual |
get_doutput_dinput() | Model | protectedvirtual |
get_input_variable(const VariableName &name) | VariableStore | inline |
get_input_variable(const VariableName &name) const | VariableStore | inline |
get_Jacobian() const | NonlinearSystem | inline |
get_output() | Model | protectedvirtual |
get_output_variable(const VariableName &name) | VariableStore | inline |
get_output_variable(const VariableName &name) const | VariableStore | inline |
get_parameter(const std::string &name) | ParameterStore | |
get_parameter(const std::string &name) const | ParameterStore | |
get_residual() const | NonlinearSystem | inline |
get_system_matrices() const | Model | |
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_storage() | VariableStore | inline |
input_storage() const | VariableStore | inline |
input_type(const VariableName &) const | VariableStore | |
input_variable(const VariableName &) | VariableStore | |
input_variables() | VariableStore | inline |
input_variables() const | VariableStore | inline |
is_AD_disabled() const final | NonlinearSystem | inlinevirtual |
is_AD_enabled() const override | Model | inlinevirtual |
is_nonlinear_system() const | Model | inlinevirtual |
Jacobian(const Tensor &x) | NonlinearSystem | |
Jacobian() | NonlinearSystem | |
Model(const OptionSet &options) | Model | |
name() const | NEML2Object | inline |
named_buffers() const | BufferStore | inline |
named_buffers() | BufferStore | |
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 | |
NEML2Object(const OptionSet &options) | NEML2Object | |
nl_param(const std::string &) const | ParameterStore | |
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_type(const VariableName &) const | VariableStore | |
output_variable(const VariableName &) | VariableStore | |
output_variables() | VariableStore | inline |
output_variables() const | VariableStore | inline |
neml2::ParameterStore::ParameterStore(const OptionSet &options, NEML2Object *object) | ParameterStore | |
path() const | NEML2Object | inline |
prepare() | Model | |
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(const Tensor &tensor, int deriv_order) | Model | virtual |
reinit(TensorShapeRef batch_shape={}, int deriv_order=0, const torch::Device &device=default_device(), const torch::Dtype &dtype=default_dtype()) | Model | virtual |
reinit(bool in, bool out) | Model | protectedvirtual |
requires_2nd_grad() const | Model | inlineprotected |
requires_grad() const | Model | inlineprotected |
residual(const Tensor &x) | NonlinearSystem | |
residual() | NonlinearSystem | |
residual_and_Jacobian(const Tensor &x) | NonlinearSystem | |
residual_and_Jacobian() | NonlinearSystem | |
residual_norm() const | NonlinearSystem | |
scalar_type() const | Model | inline |
scale_direction(const Tensor &p) const | NonlinearSystem | |
scale_Jacobian(const Tensor &J) const | NonlinearSystem | |
scale_residual(const Tensor &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 | protectedvirtual |
set_parameter(const std::string &, const Tensor &) | ParameterStore | |
set_parameters(const std::map< std::string, Tensor > &) | ParameterStore | |
set_solution(const Tensor &x) override | Model | protectedvirtual |
set_value(bool out, bool dout_din, bool d2out_din2) override | CopyVariable< T > | protectedvirtual |
setup() override | Model | protectedvirtual |
setup_input_views(VariableStore *host=nullptr) override | Model | protectedvirtual |
setup_layout() | VariableStore | virtual |
setup_nonlinear_system() | Model | protectedvirtual |
setup_output_views() | Model | protectedvirtual |
setup_output_views(bool out, bool dout_din, bool d2out_din2) | Model | protectedvirtual |
setup_submodel_input_views(VariableStore *host) | Model | protectedvirtual |
setup_submodel_output_views() | Model | protectedvirtual |
solution() const | NonlinearSystem | inlinevirtual |
type() const | NEML2Object | inline |
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() | Model | virtual |
value(const LabeledVector &in) | Model | virtual |
value_and_dvalue() | Model | virtual |
value_and_dvalue(const LabeledVector &in) | Model | virtual |
value_and_dvalue_and_d2value() | Model | virtual |
value_and_dvalue_and_d2value(const LabeledVector &in) | Model | virtual |
VariableStore(const OptionSet &options, Model *object) | VariableStore | |
zero() | Model | protectedvirtual |
zero(bool dout_din, bool d2out_din2) | Model | protectedvirtual |
~NEML2Object()=default | NEML2Object | virtual |