code: def is_staging():
      """Returns whether the virtual machine is staging a computation.
  
      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`.
      """
      return _staging
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/virtual_machine.py
imports: import contextlib
  import functools
  from absl import logging
  import six
  from tensorflow_probability.python.experimental.auto_batching import instructions
label_desc: Use image node_id 3 for calling a global function with example usage: is_staging() and returns: _staging