tensorplay.compiler API

Capture a callable as one graph operation without entering its body.

Declare that a callable returns one stable value across captures.

Compile a callable through the TensorPlay capture and lowering pipeline.

Suspend the public capture state for a dynamic Python region.

Prevent capture from entering a callable.

Declare that a callable must not be absorbed into a captured graph.

The tracer is meant to treat calls to fn as region boundaries: the surrounding pieces are captured as separate graphs and fn itself runs eagerly between them. Under eager execution the marker is inert metadata, so the callable is returned unchanged.

Mark a module method as an entry point visible to graph capture.

Return the BackendCapabilities a backend declares.

Return the currently selected default backend.

Return whether the current Python frame is being captured.

Return whether an export capture session is active.

Return registered backend names accepted by compile().

Backends whose optional dependencies are missing are hidden unless include_unavailable is set; selecting one by name produces an error naming what to install.

Return the optimization options each compile mode selects.

With mode set, returns that mode’s option patch; with mode unset, returns the full mode-to-options mapping. Unknown modes raise.

Resolve a backend name or validate a backend callable.

Mark a tensor (or one dimension of it) as fixed for shape policies.

A marked dimension is meant to be treated as a compile-time constant rather than a symbolic size. Eager execution has no dynamic shape environment, so the marker records nothing here and the input is returned unchanged; it matters only to a capture that resolves symbolic sizes.

Mark a callable as a reusable nested graph region.

Capture a callable as an opaque operation with structured arguments.

Mark a method declaration as an overload during inspection.

Register a backend in the TensorPlay compiler registry.

Register a backend tagged for diagnostics.

Register a backend tagged for experimental use.

Clear compiler specializations and capture-time constant values.

Set the backend used when compile() receives no backend name.

Register a graph-time implementation for a callable.

Remove a named backend from the compiler registry.

Mark a callable as an eager-only helper.