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.