description: Infers the variable types of a given program.
tfp.experimental.auto_batching.type_inference.infer_types
Infers the variable types of a given program.
View aliases
Main aliases
tfp.experimental.auto_batching.frontend.ab_type_inference.infer_types
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.
|
Raises |
ValueError
|
If some types still remain incomplete after inference.
|