Module: tfp.experimental.auto_batching.lowering

View source on GitHub

Lowering the full IR to stack machine instructions.

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.