tfp.experimental.auto_batching.stackless.is_running
Stay organized with collections
Save and categorize content based on your preferences.
Returns whether the stackless machine is running a computation.
tfp.experimental.auto_batching.stackless.is_running()
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 |
|---|
running
|
Python bool, True if this is called in the dynamic scope of
stackless running, 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.
Last updated 2023-11-21 UTC.
[null,null,["Last updated 2023-11-21 UTC."],[],[],null,["# tfp.experimental.auto_batching.stackless.is_running\n\n\u003cbr /\u003e\n\n|-----------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/probability/blob/v0.23.0/tensorflow_probability/python/experimental/auto_batching/stackless.py#L284-L296) |\n\nReturns whether the stackless machine is running a computation.\n\n#### View aliases\n\n\n**Main aliases**\n\n[`tfp.experimental.auto_batching.frontend.st.is_running`](https://www.tensorflow.org/probability/api_docs/python/tfp/experimental/auto_batching/stackless/is_running)\n\n\u003cbr /\u003e\n\n tfp.experimental.auto_batching.stackless.is_running()\n\nThis can be useful for writing special primitives that change their behavior\ndepending on whether they are being staged, run stackless, inferred (see\n[`type_inference.is_inferring`](../../../../tfp/experimental/auto_batching/type_inference/is_inferring)), or none of the above (i.e., dry-run execution,\nsee [`frontend.Context.batch`](../../../../tfp/experimental/auto_batching/Context#batch)).\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|-----------|-------------------------------------------------------------------------------------------------------|\n| `running` | Python `bool`, `True` if this is called in the dynamic scope of stackless running, otherwise `False`. |\n\n\u003cbr /\u003e"]]
View source on GitHub