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]