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.frontend

View source on GitHub

AutoGraph-based auto-batching frontend.


ab_type_inference module: Type inference pass on functional control flow graph.

allocation_strategy module: Live variable analysis.

dsl module: Python-embedded DSL frontend for authoring autobatchable IR programs.

instructions module: Instruction language for auto-batching virtual machine.

lowering module: Lowering the full IR to stack machine instructions.

st module: A stackless auto-batching VM.

stack module: Optimizing stack usage (pushes and pops).

tf_backend module: TensorFlow (graph) backend for auto-batching VM.

vm module: The auto-batching VM itself.


class Context: Context object for auto-batching multiple Python functions together.


truthy(...): Normalizes Tensor ranks for use in if conditions.

Other Members