TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfp.experimental.auto_batching.virtual_machine.is_staging

View source on GitHub

Returns whether the virtual machine is staging a computation.

Aliases:

tfp.experimental.auto_batching.virtual_machine.is_staging()

This can be useful for writing special primitives that change their behavior depending on whether they are being staged, run stackless, inferred (see type_inference.is_inferring), or none of the above (i.e., dry-run execution, see frontend.Context.batch).

Returns:

  • staging: Python bool, True if this is called in the dynamic scope of VM staging, otherwise False.