tfp.experimental.auto_batching.type_inference.is_inferring

View source on GitHub

Returns whether type inference is running.

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).

inferring Python bool, True if this is called in the dynamic scope of type inference, otherwise False.