Expand description
Implementations of Gradient tapes and generic gradient containers.
Structs
Records gradient computations to execute later.
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.
Traits
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.