tensorplay documentation — TensorPlay Download

Star–⇄ browse all packagesc

Latest development documentation · Updated 2026-09-08. A documentation snapshot for package 1.0.0.dev20260909 is not available.

On this page

Functions 6

checkpoint_sequential

functionFull reference ↗

tensorplay.utils.checkpoint.checkpoint_sequential(function: Callable[[...], Any], chunks: int, *args: Any, **kwargs: Any) → Any[source]

checkpoint

functionFull reference ↗

tensorplay.utils.checkpoint.checkpoint(function: Callable[[...], Any] | None = None, *args: Any, use_reentrant: bool | None = None, preserve_rng_state: bool = True, context_fn: Callable[[], Any] | None = None, determinism_check: str = 'default', debug: bool = False, early_stop: bool | None = None, respect_saved_tensors_hooks: bool | None = None, **kwargs: Any) → Any[source]

create_selective_checkpoint_contexts

functionFull reference ↗

tensorplay.utils.checkpoint.create_selective_checkpoint_contexts(policy_fn_or_list: Callable[[...], Any] | list[Any], allow_cache_entry_mutation: bool = False)[source]

set_checkpoint_debug_enabled

functionFull reference ↗

tensorplay.utils.checkpoint.set_checkpoint_debug_enabled(enabled: bool | None)[source]

set_checkpoint_early_stop

functionFull reference ↗

tensorplay.utils.checkpoint.set_checkpoint_early_stop(enabled: bool)[source]

set_device_states

functionFull reference ↗

tensorplay.utils.checkpoint.set_device_states(devices: Iterable[int], states: Iterable[Any], *, device_type: str | None = None) → None[source]

Classes 3

CheckpointPolicy

classFull reference ↗

class tensorplay.utils.checkpoint.CheckpointPolicy(*values)[source]

GraphExecGroup

classFull reference ↗

class tensorplay.utils.checkpoint.GraphExecGroup[source]

SelectiveCheckpointContext

classFull reference ↗

class tensorplay.utils.checkpoint.SelectiveCheckpointContext(*, is_recompute: bool, op_output: Any = None)[source]