View source on GitHub
|
Returns whether type inference is running.
tfp.experimental.auto_batching.type_inference.is_inferring()
This can be useful for writing special primitives that change their behavior
depending on whether they are being inferred, staged (see
virtual_machine.is_staging), or neither (i.e., dry-run execution, see
frontend.Context.batch).
Returns | |
|---|---|
inferring
|
Python bool, True if this is called in the dynamic scope of
type inference, otherwise False.
|