Skip to main content

Module rules

Module rules 

Source
Expand description

Primitive AD rule contract consumed by tidu.

This module defines the graph-level rule surface used by crate::try_linearize, crate::try_linear_transpose, and eager transpose helpers. It is intentionally narrower than Julia ChainRules: downstream primitive sets implement JVP and transpose emission for computegraph graph primitives.

Enums§

ADRuleError
Error returned when an AD rule cannot be emitted.
ADRuleKind
Identifies which AD rule failed or is unavailable.
PrimitiveValue
Reference to a value available to a primitive AD rule.

Traits§

ADKey
Constraint on GraphOperation::InputKey for AD use.
Primitive
Extends GraphOperation with primitive JVP and transpose rules for AD.
PrimitiveBuilder
Builder used by primitive JVP and transpose rules to append primitive applications.

Type Aliases§

ADRuleResult
Result type used by fallible AD rule emission.
DiffPassId
Unique identifier for a crate::linearize call.