Module: tfp.experimental.auto_batching.type_inference

Type inference pass on functional control flow graph.

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.