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 prepared extension execution.

Enums

ExtensionCacheSelector

Selector used to inspect or clear extension cache entries.

Traits

ExtensionOp

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

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_eager_with_extension_session

Apply an extension op to eager tensors through a direct prepared-operation callback receiving a non-owning backend session.

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.