Module: tfp.experimental.auto_batching.lowering

View source on GitHub

Lowering the full IR to stack machine instructions.

Aliases:

At present, only one pass is needed to make the whole instruction language defined in instructions.py understandable by the virtual machine defined in virtual_machine.py, namely lowering FunctionCallOp instructions to sequences of push, pop, and goto.

Functions

lower_function_calls(...): Lowers a Program that may have (recursive) FunctionCallOp instructions.