attribute [PreferCheckpoint]¶
Description¶
Mark a differentiable function to prefer checkpointing over recomputation when a value computed in the primal pass is needed during backward derivative propagation.
Signature¶
[PreferCheckpoint]
Mark a differentiable function to prefer checkpointing over recomputation when a value computed in the primal pass is needed during backward derivative propagation.
[PreferCheckpoint]