jax
74ffed90 - Switch the order of sharding and memory kind custom call

Commit

1 year ago

Switch the order of sharding and memory kind custom call PiperOrigin-RevId: 634505383

References

Author

yashk2810

Committer

a-googler

Parents