xla
5caaaee8 - Expose mark_sharding_with_gradients as a public API. (#8826)

Commit

1 year ago

Expose mark_sharding_with_gradients as a public API. (#8826)

References

#8826 - Expose mark_sharding_with_gradients as a public API.

Author

iwknow

iwknow

Parents

Loading