Module graph

Module graph 

Source

Structs§

GradClipConfig
InferenceConfig
Inference configuration (lowered from @inference).
IrAssert
A compile-time or runtime assertion from @assert.
IrGraph
IrNode
A node in the computation graph.
IrOutput
An output of a graph, preserving the user-facing name across optimisations.
IrParam
A learnable parameter in the graph.
IrProgram
The full program IR — lowered from the AST.
LrScheduleConfig
NodeId
Unique identifier for a node in the graph.
OptimizerConfig
TrainingConfig
Training configuration (lowered from @training).

Enums§

AttrValue
An attribute value (for custom ops and node attrs).
ConfigValue
A configuration value.
ConstantValue
A constant value embedded in the graph.
DType
Resolved data type.
Dim
A resolved tensor shape — dimensions are either concrete or symbolic.
InitStrategy
How to initialize a parameter.
IrHint
Execution hints attached to nodes.
IrType
The resolved type of a value in the graph.
OpKind
An operation in the computation graph.