jax
4459991d - Merge pull request #18961 from mattjj:issue18955

Commit

2 years ago

Merge pull request #18961 from mattjj:issue18955 PiperOrigin-RevId: 590655131

References

#18961 - [shard-map] register cumsum et al with generic rules

Author

a-googler

a-googler

Parents

Loading