Google I/O returns May 18-20! Reserve space and build your schedule Register now

Module: tfp.experimental.auto_batching.lowering

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.