Module eager 

Generic eager reverse-mode recording for Primitive frontends.

Structs

EagerInput

Input descriptor for recording one eager graph invocation.

EagerOutput

Per-output trace metadata returned by Recorder::record_graph.

RecordedGraph

Graph invocation recorded as one eager reverse-mode trace node.

Recorder

Stateful eager operation recorder.

Trace

Opaque handle to an eager reverse-mode trace node.

Enums

EagerRecordError

Error returned when eager graph recording receives inconsistent metadata.

Traits

BackwardExecutor

Downstream execution hooks for eager backward.

ForwardExecutor

Downstream execution hooks for eager forward-mode AD.

KeySource

Caller-provided source of stable eager value keys.

Functions

backward

Execute reverse-mode AD over an eager trace.

try_forward

Execute forward-mode AD over an eager trace for one output value.

Type Aliases

EagerRecordResult

Result type used by eager graph recording APIs.

ForwardTangentInput

Tangent seed aligned with one linearized graph input.