tfp.experimental.auto_batching.type_inference.is_inferring

View source on GitHub

Returns whether type inference is running.

Aliases:

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.