View source on GitHub |
Computes a type signature for the given inputs
.
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 | |
---|---|
sig
|
A list of (patterns of) instructions.TensorType aligned with
program.vars_in .
|