Package-level declarations

Types

Link copied to clipboard
class GraphSink(graph: ComputeGraph) : OpSink

Builds/upserts nodes and edges in a ComputeGraph incrementally from OpTrace. Maintains a mapping from TensorRef.id to its producing node/output and spec.

Link copied to clipboard

Writes traces into DefaultExecutionTape.

Link copied to clipboard
class TraceToGraphBuilder(graph: ComputeGraph, session: TraceSession? = null)

Shared builder to convert OpTrace streams into a ComputeGraph. Used by both GraphSink (online) and DefaultExecutionTape.toComputeGraph() (offline).