Expand description
Graph-level intermediate representation for multi-operation models.
Extends the single-kernel IR to support directed acyclic graphs (DAGs) of operations, enabling transpilation of production ML models with 50-500+ operations.
Structs§
- Compute
Graph - A computation graph representing a multi-operation model.
- EdgeId
- A unique identifier for an edge (tensor) in the computation graph.
- Graph
Module - Extends
crate::Modulewith an optional computation graph. - Graph
Node - A node in the computation graph.
- NodeId
- A unique identifier for a node in the computation graph.
- Tensor
Info - Metadata about a tensor edge in the graph.
Enums§
- Activation
Function - Activation function that can be fused into a preceding operation.
- GraphOp
- The operation type for a graph node.