tfp.experimental.auto_batching.instructions.Function

A function subject to auto-batching, callable with FunctionCallOp.

tfp.experimental.auto_batching.instructions.Function(
    graph, vars_in, vars_out, type_inference, name=None
)

graph A ControlFlowGraph comprising the function's body.
vars_in List of string giving the names of the formal parameters of the function.
vars_out Pattern of string giving the name(s) of the variables the function returns. Ergo, functions must be canonicalized to place the return value(s) in the same-named variable(s) along every path to the exit.
type_inference A callable which takes a list of patterns of TensorTypes corresponding to the data types of vars_in. This callable must return a pattern of TensorTypes corresponding to the structure assembled by the return_vars.
name Optional string denoting this Function in printed output.

name