Trait dfdx::gradients::Tape

pub trait Tape {
    const OWNS_TAPE: bool;

    fn add_backward_op<F: 'static + FnOnce(&mut Gradients)>(
        &mut self,
        operation: F
    );
}

Something that can add a gradient operation to GradientTape.

Required Associated Constants

const OWNS_TAPE: bool

Whether this object currently owns the GradientTape. This is known at compile time.

Required Methods

fn add_backward_op<F: 'static + FnOnce(&mut Gradients)>(&mut self, operation: F)

Implementors

impl Tape for NoneTape

const OWNS_TAPE: bool = false

impl Tape for OwnedTape

const OWNS_TAPE: bool = true