tfp.experimental.auto_batching.type_inference.infer_types_from_signature

Infers the variable types of a given program.

Main aliases

tfp.experimental.auto_batching.type_inference.infer_types_from_signature(
    program, sig, backend
)

Args
`program`	`instructions.Program` whose types to infer.
`sig`	A `list` of (patterns of) `instructions.TensorType` 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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2023-11-21 UTC.

Args