Struct LinearizedGraph 

pub struct LinearizedGraph<Op: GraphOperation> { /* private fields */ }

Graph produced by linearizing a primitive computation graph.

Implementations

impl<Op: GraphOperation> LinearizedGraph<Op>

pub fn as_graph(&self) -> &Graph<Op>

Borrow the strictly-linear graph representation.

pub fn residual_graph(&self) -> &Graph<Op>

Borrow the residual graph referenced by the linear graph.

pub fn into_graph(self) -> Graph<Op>

Consume this value and return the strictly-linear graph representation.

pub fn into_graphs(self) -> (Graph<Op>, Arc<Graph<Op>>)

Consume this value and return both graph parts.

pub fn tangent_inputs(&self) -> &[(Op::InputKey, LocalValueId)]

Tangent input keys and local value ids.

pub fn tangent_outputs(&self) -> &[Option<LocalValueId>]

Tangent outputs aligned with requested primal outputs.