xla
3a239e64 - Add all_gather_out and reduce_scatter_out (#3359)

Commit

3 years ago

Add all_gather_out and reduce_scatter_out (#3359) * Add all_gather_out and reduce_scatter_out * linter * Remove unnecessary else

References

Author

JackCaoG

Parents