Enum DispatchAutodiffContext 

pub enum DispatchAutodiffContext {
    Disabled,
    Enabled(GradientCheckpointingStrategy),
}

The autodiff backend associated with a dispatch tensor.

This is backend metadata, not a statement that a float tensor requires gradients. Enabled float tensors may be tracked or untracked by autodiff.

Tensor inputs to one backend operation merge their autodiff contexts. A disabled context is compatible with an enabled one and is treated as a constant for that operation. Two enabled contexts must use the same gradient-checkpointing strategy.

Variants

Disabled

Route the tensor through its concrete backend.

Enabled(GradientCheckpointingStrategy)

Associate the tensor with an autodiff backend using the given strategy.

Trait Implementations

impl Clone for DispatchAutodiffContext

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for DispatchAutodiffContext

impl Debug for DispatchAutodiffContext

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

Formats the value using the given formatter.

impl Default for DispatchAutodiffContext

fn default() -> Self

Returns the “default value” for a type.

impl Eq for DispatchAutodiffContext

impl PartialEq for DispatchAutodiffContext

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for DispatchAutodiffContext