jax

1d2ef4cc - [Pallas:TPU] Drop the buffer count to 1 when the pltpu.emit_pipeline operand window is constant

[Pallas:TPU] Drop the buffer count to 1 when the pltpu.emit_pipeline operand window is constant There's not point in allocating the second buffer if we're never going to use it. PiperOrigin-RevId: 881949066