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

Module: tfp.experimental.auto_batching.type_inference

View source on GitHub

Type inference pass on functional control flow graph.

Aliases:

Until converged, we propagate type information (dtype and shape) from inputs toward outputs.

Functions

infer_types(...): Infers the variable types of a given program.

infer_types_from_signature(...): Infers the variable types of a given program.

is_inferring(...): Returns whether type inference is running.

signature(...): Computes a type signature for the given inputs.

type_of_pattern(...): Returns the instructions.Type of val.