TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

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.