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.frontend.vm.is_staging

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.