A function subject to auto-batching, callable with FunctionCallOp.
tfp.experimental.auto_batching.instructions.Function(
graph, vars_in, vars_out, type_inference, name=None
)
Args |
|---|
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.
|
View source on GitHub