jax
bb271aaf - [pallas:mosaic_gpu] Added `FragmentedArray.to_layout`

Commit

1 year ago

[pallas:mosaic_gpu] Added `FragmentedArray.to_layout` PiperOrigin-RevId: 686524192

Author

superbobry

Committer

Google-ML-Automation

Parents