Struct Checkpointer 

pub struct Checkpointer { /* private fields */ }

Struct responsible of fetching the output for a node in the autodiff graph during a backward pass

Implementations

impl Checkpointer

pub fn new( backward_states: BackwardStates, retro_forwards: RetroForwards, node_tree: NodeTree, ) -> Self

Constructs a new Checkpointer.

impl Checkpointer

pub fn retrieve_node_output<T>(&mut self, node_id: NodeId) -> T

where T: Clone + Send + 'static,

Gives the output of the given node, by recursively asking parents to compute themselves or give their pre-computed tensors.

pub fn is_empty(&self) -> bool

Checks if checkpointer has been drained adequately. Useful for testing

Trait Implementations

impl Debug for Checkpointer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.