Skip to main content

GradientList

Type Alias GradientList 

Source
pub type GradientList = Vec<Option<NodeId>>;
Expand description

Result of a backward rule: one gradient NodeId per forward input slot, with None for inputs that aren’t differentiable (e.g. the index tensor of IndexSelect, or the block table of PagedAttn).

Length must match the forward op’s inputs slice.

Aliased Type§

pub struct GradientList { /* private fields */ }