jax

b78bd232 - [Mosaic TPU] Add compatibility rules for the updated DenseElementsAttr encoding

[Mosaic TPU] Add compatibility rules for the updated DenseElementsAttr encoding MLIR has recently changed the encoding of i1 DenseElementsAttr, from a bitpacked vector to a vector of bytes. More importantly, a true splat used to be encoded by a single 0xff byte, while now it is 0x01. This requires us to downgrade the constants that might appear in our IR, because the old compiler does not recognize 0x01 as a splat. I don't think we've used i1 DenseElementsAttrs for anything beyond that. PiperOrigin-RevId: 871810024