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

View source on GitHub

Optimizing stack usage (pushes and pops).

Aliases:

Modules

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

Functions

fuse_pop_push(...): Fuses pop+push sequences in the given Program.