Struct burn::backend::autodiff::checkpoint::strategy::BalancedCheckpointing

pub struct BalancedCheckpointing {}

Operation properties are as they are marked (compute or memory bound)

Trait Implementations

impl CheckpointStrategy for BalancedCheckpointing

fn compute_property<R>(retro_forward: R) -> ComputingProperty

where R: RetroForward,

An operation marked as memory bound is memory bound. When memory bound, an operation needs to save its RetroForward

fn checkpoint_parents<'a, B2, const D2: usize, A>( parents: A, builder: &mut CheckpointerBuilder, )

where B2: Backend, A: IntoIterator<Item = &'a AutodiffTensor<B2, D2>>,

An operation marked as memory bound is really memory bound. Since the operation may not checkpoint its parents but may need them indirectly if asked to recompute itself, the method needs to know the parent tensors to maybe checkpoint them

impl Clone for BalancedCheckpointing

fn clone(&self) -> BalancedCheckpointing

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BalancedCheckpointing

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

Formats the value using the given formatter.

impl Default for BalancedCheckpointing

fn default() -> BalancedCheckpointing

Returns the “default value” for a type.

impl Copy for BalancedCheckpointing