Skip to main content

Module context

Module context 

Source
Expand description

AD context for guard-based shape resolution and value metadata queries.

During AD graph construction, linalg rules such as SVD, QR, and LU need concrete matrix dimensions to choose between structurally different subgraphs. ShapeGuardContext records those dimension comparisons as guards so cached AD graphs can later be invalidated when the observed shape relationship changes.

Structs§

ShapeGuard
A recorded dimension comparison made during AD graph construction.
ShapeGuardContext
AD context providing dimension resolution, guard recording, and value metadata.
TensorMeta
Per-value tensor metadata used by AD rules.

Enums§

MetadataRegistryError
Error returned when the process-global AD metadata registry is unavailable.
ShapeGuardError
Error returned when shape-guard metadata cannot be resolved.

Functions§

lookup_global_metadata
Look up a single metadata entry from the global registry.

Type Aliases§

ShapeGuardResult
Result type used by shape-guard metadata queries.