jax
318ae893 - [Pallas TPU] Relax windowing restriction when lowering mapped grids

Commit

1 year ago

[Pallas TPU] Relax windowing restriction when lowering mapped grids PiperOrigin-RevId: 621330022

References

Author

sharadmv

Committer

a-googler

Parents