![]() |
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
: Pythonbool
,True
if this is called in the dynamic scope of VM staging, otherwiseFalse
.