NEML2 1.4.0
Loading...
Searching...
No Matches
VariableStore Member List

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)VariableStoreprotectedvirtual
cache(TorchShapeRef batch_shape)VariableStoreprotectedvirtual
declare_axis(const std::string &name)VariableStore
declare_input_variable(S &&... name)VariableStoreinlineprotected
declare_input_variable(TorchSize sz, S &&... name)VariableStoreinlineprotected
declare_input_variable_list(TorchSize list_size, S &&... name)VariableStoreinlineprotected
declare_output_variable(S &&... name)VariableStoreinlineprotected
declare_output_variable(TorchSize sz, S &&... name)VariableStoreinlineprotected
declare_output_variable_list(TorchSize list_size, S &&... name)VariableStoreinlineprotected
declare_subaxis(LabeledAxis &axis, const VariableName &subaxis) constVariableStoreinlineprotected
declare_variable(LabeledAxis &axis, const VariableName &var) constVariableStoreinlineprotected
declare_variable(LabeledAxis &axis, const VariableName &var, TorchSize sz) constVariableStoreinlineprotected
derivative_storage()VariableStoreinline
derivative_storage() constVariableStoreinline
detach_and_zero(bool out, bool dout_din=true, bool d2out_din2=true)VariableStoreprotectedvirtual
get_input_variable(const VariableName &name)VariableStoreinline
get_input_variable(const VariableName &name) constVariableStoreinline
get_output_variable(const VariableName &name)VariableStoreinline
get_output_variable(const VariableName &name) constVariableStoreinline
input_axis()VariableStoreinline
input_axis() constVariableStoreinline
input_storage()VariableStoreinline
input_storage() constVariableStoreinline
input_view(const VariableName &)VariableStore
input_views()VariableStoreinline
input_views() constVariableStoreinline
output_axis()VariableStoreinline
output_axis() constVariableStoreinline
output_storage()VariableStoreinline
output_storage() constVariableStoreinline
output_view(const VariableName &)VariableStore
output_views()VariableStoreinline
output_views() constVariableStoreinline
reinit_input_views()VariableStoreprotectedvirtual
reinit_output_views(bool out, bool dout_din=true, bool d2out_din2=true)VariableStoreprotectedvirtual
second_derivative_storage()VariableStoreinline
second_derivative_storage() constVariableStoreinline
setup_input_views()VariableStoreprotectedvirtual
setup_layout()VariableStorevirtual
setup_output_views()VariableStoreprotectedvirtual
VariableStore(const OptionSet &options, NEML2Object *object)VariableStore