jax
d6a84cc5 - Pallas GPU no longer assumes that all slices have stride 1

Commit

1 year ago

Pallas GPU no longer assumes that all slices have stride 1 Fixes #20895. PiperOrigin-RevId: 639031975

References

Author

superbobry

Committer

a-googler

Parents