Module extension 

Eager AD support for out-of-tree extension primitives.

Structs

ExtensionCacheKey

A stable key for one extension-owned runtime cache entry.

ExtensionCacheLimits

Bounded retention limits for extension runtime caches.

ExtensionCacheStore

Bounded type-erased cache storage owned by an extension executor.

ExtensionExecutionContext

Backend and cache state passed to one extension execution.

ExtensionExecutor

Runtime owner for backend-specific extension dispatch and caches.

ExtensionRegistry

Registry of backend-specific extension runtime executors.

ExtensionRuleSet

Explicit, owned set of extension AD rules.

Enums

ExtensionCacheSelector

Selector used to inspect or clear extension cache entries.

ExtensionRegistryError

Errors returned from extension registries.

ExtensionRuntimeRegistryError

Errors returned by backend-parametric extension runtime registries.

Traits

ExtensionAdRule

AD rule provider for an extension family.

ExtensionOp

The contract every out-of-tree extension primitive must satisfy.

ExtensionRuntime

A backend-specific runtime executor for one extension family.

Functions

adopt_untracked_eager_value

Adopt an untracked eager tensor value produced by this runtime’s backend.

apply

Apply an extension op in the traced graph.

apply_eager

Apply an extension op to eager AD tensors.

apply_standard_op

Apply one standard tensor op eagerly and record it for AD when needed.

Derive Macros

ExtensionFamilyId

Derive an inherent FAMILY_ID constant for an extension payload type.