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.