tfp.experimental.auto_batching.type_inference.is_inferring

Stay organized with collections Save and categorize content based on your preferences.

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.