tfp.experimental.auto_batching.type_inference.infer_types_from_signature

View source on GitHub

Infers the variable types of a given program.

Aliases:

tfp.experimental.auto_batching.type_inference.infer_types_from_signature(
    program,
    sig,
    backend
)

Args:

Returns:

Raises:

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