NEML2 1.4.0
|
This is the complete list of members for VariableStore, including all inherited members.
allocate_variables(TorchShapeRef batch_shape, const torch::TensorOptions &options, bool in, bool out, bool dout_din, bool d2out_din2) | VariableStore | protectedvirtual |
cache(TorchShapeRef batch_shape) | VariableStore | protectedvirtual |
declare_axis(const std::string &name) | VariableStore | |
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_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) | VariableStore | protectedvirtual |
get_input_variable(const VariableName &name) | VariableStore | inline |
get_input_variable(const VariableName &name) const | VariableStore | inline |
get_output_variable(const VariableName &name) | VariableStore | inline |
get_output_variable(const VariableName &name) const | VariableStore | inline |
input_axis() | VariableStore | inline |
input_axis() const | VariableStore | inline |
input_storage() | VariableStore | inline |
input_storage() const | VariableStore | inline |
input_view(const VariableName &) | VariableStore | |
input_views() | VariableStore | inline |
input_views() const | VariableStore | 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 |
reinit_input_views() | VariableStore | protectedvirtual |
reinit_output_views(bool out, bool dout_din=true, bool d2out_din2=true) | VariableStore | protectedvirtual |
second_derivative_storage() | VariableStore | inline |
second_derivative_storage() const | VariableStore | inline |
setup_input_views() | VariableStore | protectedvirtual |
setup_layout() | VariableStore | virtual |
setup_output_views() | VariableStore | protectedvirtual |
VariableStore(const OptionSet &options, NEML2Object *object) | VariableStore |