Struct CustomBackwardOp 

pub struct CustomBackwardOp {
    pub handler: Arc<dyn CustomBackward>,
    pub input_versions: Vec<VersionSnapshot>,
    pub saved_storages: Vec<StorageHandle>,
    pub saved_layouts: Vec<Layout>,
    pub saved_shapes: Vec<Vec<usize>>,
}

Backward state for a custom op: the user’s handler + saved tensors.

Fields

handler: Arc<dyn CustomBackward>

The user’s backward implementation.

input_versions: Vec<VersionSnapshot>

Version snapshots for each input (for mutation checking).

saved_storages: Vec<StorageHandle>

Tensors saved during forward for use in backward.

saved_layouts: Vec<Layout>

saved_shapes: Vec<Vec<usize>>

Trait Implementations

impl Debug for CustomBackwardOp

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

Formats the value using the given formatter.