View source on GitHub |
Infers the variable types of a given program.
tfp.experimental.auto_batching.type_inference.infer_types(
program, inputs, backend
)
Args | |
---|---|
program
|
instructions.Program whose types to infer.
|
inputs
|
A list of backend-compatible tensors aligned with
program.vars_in .
|
backend
|
Backend implementation. |
Returns | |
---|---|
typed
|
instructions.Program with types inferred.
|
Raises | |
---|---|
ValueError
|
If some types still remain incomplete after inference. |