jax

3ad2cd96 - [Mosaic] Prove divisibility of loop induction vars

[Mosaic] Prove divisibility of loop induction vars This helps when a slice offset is based on a loop induction var that is divisible by the tile size. PiperOrigin-RevId: 885061108