Module dfdx::gradients

Expand description

Implementations of Gradient tapes and generic gradient containers.

Structs

Records gradient computations to execute later.

A generic container for keeping variable sized arrays associated with a UniqueId.

Contains nothing. When Tape::add_backward_op is called, this struct does nothing.

Contains a boxed GradientTape. When Tape::add_backward_op is called, this function passes the operation directly to GradientTape::add_backward_op.

Represents something that can be updated with GradientProvider.

Represents something that can return a gradient for a given key.

Something that can add a gradient operation to GradientTape.