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

This is the complete list of members for VariableBase, including all inherited members.

_argsVariableBaseprotected
_batch_sizesVariableBaseprotected
_d2value_dVariableBaseprotected
_derivative_storageVariableBaseprotected
_dvalue_dVariableBaseprotected
_nameVariableBaseprotected
_raw_valueVariableBaseprotected
_second_derivative_storageVariableBaseprotected
_value_storageVariableBaseprotected
add_arg(const VariableBase &arg)VariableBaseinline
args() constVariableBaseinline
base_dim() constVariableBaseinline
base_sizes() const =0VariableBasepure virtual
base_storage() constVariableBaseinline
batch_dim() constVariableBaseinline
batch_sizes() constVariableBaseinline
cache(TorchShapeRef batch_shape)VariableBasevirtual
clear_args()VariableBaseinline
d(const VariableBase &x)VariableBase
d(const VariableBase &x1, const VariableBase &x2)VariableBase
derivative_storage() constVariableBase
name() constVariableBaseinline
raw_value() constVariableBaseinline
reinit_views(bool out, bool dout_din, bool d2out_din2)VariableBasevirtual
requires_grad_(bool req=true)=0VariableBasepure virtual
second_derivative_storage() constVariableBase
setup_views(const LabeledVector *value, const LabeledMatrix *deriv=nullptr, const LabeledTensor3D *secderiv=nullptr)VariableBase
setup_views(const VariableBase *other)VariableBase
sizes() const =0VariableBasepure virtual
tensor() const =0VariableBasepure virtual
value_storage() constVariableBase
VariableBase(const VariableName &name_in)VariableBaseinline
~VariableBase()=defaultVariableBasevirtual