ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.experimental.auto_batching.stack_optimization

Optimizing stack usage (pushes and pops).

Modules

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

Functions

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