Module: tfp.experimental.auto_batching.lowering

Lowering the full IR to stack machine instructions.

View aliases

Main aliases

tfp.experimental.auto_batching.frontend.lowering

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2023-11-21 UTC.