Save the date! Google I/O returns May 18-20 Register now

tfp.experimental.auto_batching.type_inference.signature

Computes a type signature for the given inputs.

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.

sig A list of (patterns of) instructions.TensorType aligned with program.vars_in.