Struct burn::backend::autodiff::checkpoint::state::BackwardStates

pub struct BackwardStates { /* private fields */ }

Links NodeIDs to their current state

Implementations

impl BackwardStates

pub fn new(map: HashMap<NodeID, State>) -> BackwardStates

Constructs a new BackwardStates.

impl BackwardStates

pub fn get_state<T>(&mut self, node_id: &NodeID) -> T

where T: Clone + Send + 'static,

Returns the output in the state of the given NodeID, and decrements the number of times this state is required. This function always gives ownership of the output, but will clone it if needed for further uses.

pub fn save<T>(&mut self, node_id: NodeID, saved_output: T)

where T: Clone + Send + 'static,

Saves the output to the state of the given NodeID.

Trait Implementations

impl Debug for BackwardStates

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

Formats the value using the given formatter.

impl Default for BackwardStates

fn default() -> BackwardStates

Returns the “default value” for a type.