xla
50144950 - Wrap implicitly-replicated data in PjRtShardedData (#4971)

Commit

2 years ago

Wrap implicitly-replicated data in PjRtShardedData (#4971)

References

#4971 - Wrap implicitly-replicated data in PjRtShardedData

Author

jonb377

jonb377

Parents

Loading