Struct Checkpointer
pub struct Checkpointer { /* private fields */ }Expand description
Struct responsible of fetching the output for a node in the autodiff graph during a backward pass
Implementations§
§impl Checkpointer
impl Checkpointer
pub fn new(
backward_states: BackwardStates,
retro_forwards: RetroForwards,
node_tree: NodeTree,
) -> Checkpointer
pub fn new( backward_states: BackwardStates, retro_forwards: RetroForwards, node_tree: NodeTree, ) -> Checkpointer
Constructs a new Checkpointer.
§impl Checkpointer
impl Checkpointer
pub fn retrieve_node_output<T>(&mut self, node_id: NodeID) -> T
pub fn retrieve_node_output<T>(&mut self, node_id: NodeID) -> T
Gives the output of the given node, by recursively asking parents to compute themselves or give their pre-computed tensors.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for Checkpointer
impl !RefUnwindSafe for Checkpointer
impl !Send for Checkpointer
impl !Sync for Checkpointer
impl Unpin for Checkpointer
impl !UnwindSafe for Checkpointer
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
source§impl<T> IntoEither for T
impl<T> IntoEither for T
source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moresource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more