xla
3413bd18 - Implement all_gather with native XLA primitive. (#3275)

Commit

4 years ago

Implement all_gather with native XLA primitive. (#3275) * Implement all_gather with native XLA primitive. * Enable GPU test.

References

Author

hjm-aws

Parents