TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfp.experimental.auto_batching.type_inference.infer_types

View source on GitHub

Infers the variable types of a given program.

Aliases:

tfp.experimental.auto_batching.type_inference.infer_types(
    program,
    inputs,
    backend
)

Args:

  • program: instructions.Program whose types to infer.
  • inputs: A list of backend-compatible tensors aligned with program.vars_in.
  • backend: Backend implementation.

Returns:

Raises:

  • ValueError: If some types still remain incomplete after inference.