description: The auto-batching VM itself.
The auto-batching VM itself.
Main aliases
tfp.experimental.auto_batching.frontend.vm
execute(...): Executes or stages a complete auto-batching VM program.
execute(...)
is_staging(...): Returns whether the virtual machine is staging a computation.
is_staging(...)