ExportGraphSignature

class tensorplay.export.ExportGraphSignature(input_specs: list[InputSpec], output_specs: list[OutputSpec])[source]

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

property assertion_dep_token: Mapping[int, str] | None

Position of the assertion dependency token output, if present.

clone() → ExportGraphSignature[source]

Deep copy: specs and argument records are duplicated, not shared.

get_param_to_buffer() → Mapping[str, str][source]

Map parameter targets to the buffer targets holding their gradients.

Gradients are declared as GRADIENT_TO_PARAMETER outputs; a gradient for a parameter whose optimizer state lives in a buffer binds the two targets under the parameter’s FQN.

get_replace_hook(replace_inputs: bool = False)[source]

Build a rename hook suitable for graph rewriting passes.

is_buffer(name: str) → bool[source]

Whether name is a placeholder carrying a lifted buffer.

is_param(name: str) → bool[source]

Whether name is a placeholder carrying a lifted parameter.

replace_all_uses(old: str, new: str) → None[source]

Rename a graph value across every input and output spec.