TensorPlay
Reference guides
latest (dev)
Copy
View Markdown

Latest development documentation · Updated 2026-10-08

tensorplay.export

Classes

tensorplay.export.AdditionalInputs

Infer dynamic shape markers from representative input sets.

tensorplay.export.ConstantArgument

tensorplay.export.Constraint

A range restriction attached to one input dimension.

tensorplay.export.ConstraintsExceededError

A runtime input violated the declared dynamic-shape contract.

tensorplay.export.CustomDecompTable

A validated mutable mapping from graph targets to replacement callables.

tensorplay.export.CustomObjArgument

tensorplay.export.DerivedDim

tensorplay.export.Dim

A named symbolic dimension with an optional finite range.

tensorplay.export.EqualityConstraint

Ties several input sites to one shared dimension size.

tensorplay.export.ExportBackwardSignature

tensorplay.export.ExportGraphSignature

Describe lifted state, user values, mutations, and graph outputs.

tensorplay.export.ExportedProgram

A validated graph together with its state and example bindings.

tensorplay.export.FlatArgsAdapter

Adapt one flattened argument layout into another layout.

tensorplay.export.GraphSignature

Compact signature retained for callers that build signatures directly.

tensorplay.export.InputKind

tensorplay.export.InputSpec

tensorplay.export.ModuleCallEntry

tensorplay.export.ModuleCallSignature

tensorplay.export.OutputKind

tensorplay.export.OutputSpec

tensorplay.export.PT2ArchiveContents

tensorplay.export.ShapesCollection

Associate shape specifications with tensor objects by identity.

tensorplay.export.SymBoolArgument

tensorplay.export.SymFloatArgument

tensorplay.export.SymIntArgument

tensorplay.export.TensorArgument

tensorplay.export.TokenArgument

tensorplay.export.UnflattenedModule

Executable module view retaining the captured root module hierarchy.

tensorplay.export.WeightType

Role a packaged weight plays in the captured program.

Functions

tensorplay.export.default_decompositions

Return the mutable table of built-in graph rewrites.

tensorplay.export.dims

Construct several named dimensions with shared bounds.

tensorplay.export.draft_export

Capture a program, reporting failures instead of raising them.

tensorplay.export.export

Capture a callable and return an executable graph program.

tensorplay.export.export_for_training

Capture a callable while retaining its mutable training state.

tensorplay.export.load

Load an ExportedProgram previously written by save().

tensorplay.export.load_pt2

tensorplay.export.package_pt2

tensorplay.export.refine_dynamic_shapes_from_suggested_fixes

Apply suggested fixes (range refinements, specializations, relations).

tensorplay.export.register_dataclass

Register a dataclass as a flattenable graph value.

tensorplay.export.save

Save an ExportedProgram to a file or writable buffer.

tensorplay.export.unflatten

Build an executable module view from an exported program.

On this page

Ask DeepWiki