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.signature

View source on GitHub

Computes a type signature for the given inputs.

Aliases:

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

Args:

  • program: instructions.Program for whose inputs to compute the signature.
  • inputs: A list of backend-compatible tensors aligned with program.vars_in.
  • backend: Backend implementation.

Returns: