NEML2 1.4.0
|
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) | VariableStore | protectedvirtual |
cache(TensorShapeRef batch_shape) | VariableStore | protectedvirtual |
declare_axis(const std::string &name) | VariableStore | |
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_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 |
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_type(const VariableName &) const | VariableStore | |
input_variable(const VariableName &) | VariableStore | |
input_variables() | VariableStore | inline |
input_variables() const | VariableStore | 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 |
second_derivative_storage() | VariableStore | inline |
second_derivative_storage() const | VariableStore | inline |
setup_input_views(VariableStore *host=nullptr) | VariableStore | protectedvirtual |
setup_layout() | VariableStore | virtual |
setup_output_views(bool out, bool dout_din, bool d2out_din2) | VariableStore | protectedvirtual |
VariableStore(const OptionSet &options, Model *object) | VariableStore | |
zero(bool dout_din, bool d2out_din2) | VariableStore | protectedvirtual |