<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/probability/blob/v0.25.0/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py) |

The auto-batching VM itself.

#### View aliases


**Main aliases**

[`tfp.experimental.auto_batching.frontend.vm`](https://www.tensorflow.org/probability/api_docs/python/tfp/experimental/auto_batching/virtual_machine)

<br />

## Functions

[`execute(...)`](https://www.tensorflow.org/probability/api_docs/python/tfp/experimental/auto_batching/virtual_machine/execute): Executes or stages a complete auto-batching VM program.

[`is_staging(...)`](https://www.tensorflow.org/probability/api_docs/python/tfp/experimental/auto_batching/virtual_machine/is_staging): Returns whether the virtual machine is staging a computation.