jax
13541e9f - Make blocked_fold_in consistent when the block sizes induce padding

Commit

278 days ago

Make blocked_fold_in consistent when the block sizes induce padding Add coverage for padded shapes to unit tests. PiperOrigin-RevId: 738029476

References

Author

Google-ML-Automation

Committer

Google-ML-Automation

Parents