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(TensorShapeRef batch_shape, const torch::TensorOptions &options, bool in, bool out, bool dout_din, bool d2out_din2)VariableStoreprotectedvirtual
cache(TensorShapeRef batch_shape)VariableStoreprotectedvirtual
declare_axis(const std::string &name)VariableStore
declare_input_variable(S &&... name)VariableStoreinlineprotected
declare_input_variable(Size sz, TensorType t, S &&... name)VariableStoreinlineprotected
declare_input_variable_list(Size list_size, S &&... name)VariableStoreinlineprotected
declare_output_variable(S &&... name)VariableStoreinlineprotected
declare_output_variable(Size sz, TensorType t, S &&... name)VariableStoreinlineprotected
declare_output_variable_list(Size 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, Size sz) constVariableStoreinlineprotected
derivative_storage()VariableStoreinline
derivative_storage() constVariableStoreinline
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_type(const VariableName &) constVariableStore
input_variable(const VariableName &)VariableStore
input_variables()VariableStoreinline
input_variables() constVariableStoreinline
output_axis()VariableStoreinline
output_axis() constVariableStoreinline
output_storage()VariableStoreinline
output_storage() constVariableStoreinline
output_type(const VariableName &) constVariableStore
output_variable(const VariableName &)VariableStore
output_variables()VariableStoreinline
output_variables() constVariableStoreinline
second_derivative_storage()VariableStoreinline
second_derivative_storage() constVariableStoreinline
setup_input_views(VariableStore *host=nullptr)VariableStoreprotectedvirtual
setup_layout()VariableStorevirtual
setup_output_views(bool out, bool dout_din, bool d2out_din2)VariableStoreprotectedvirtual
VariableStore(const OptionSet &options, Model *object)VariableStore
zero(bool dout_din, bool d2out_din2)VariableStoreprotectedvirtual