Module semantic_extension 

Semantic-program automatic-differentiation rules for extension operations.

Rules in this module are owned explicitly by crate::AdContext. They operate on opaque semantic ProgramValue tokens and never expose computegraph node keys or execution-program slots.

Structs

ResidualSpec

Per-rule declaration of which primal inputs/outputs a transpose (VJP) rule reads as full tensor residuals versus which need only shape/dtype metadata.

SemanticExtensionRuleSet

Explicit clone-on-write set of semantic extension AD rules.

SemanticLinearTransposeRequest

Ordered inputs for one semantic linear-transpose rule.

SemanticLinearizeRequest

Ordered inputs for one semantic extension linearization rule.

SemanticLinearizeResult

Output of one semantic extension linearization rule.

SemanticPrimalVjpRequest

Ordered inputs for one direct semantic primal-VJP rule.

Enums

AdValue

One optional semantic AD value.

PrimalValueKind

Whether a checked semantic AD request refers to a primal input or output.

SemanticAdError

Failures while dispatching or building semantic extension AD.

SemanticAdRuleRole

Semantic extension AD rule role.

SemanticExtensionRegistryError

Registration failures for semantic extension AD rules.

Traits

SemanticLinearTransposeRule

Transpose rule for an extension viewed as a linear map.

SemanticLinearizeRule

Definitional JVP rule for one extension family.

SemanticPrimalVjpRule

Optional direct VJP rule expressed against primal semantic values.