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.


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


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