xla
1e4e5d23 - Update test_xla_sharding.py to take into account the number of devices

Commit
1 year ago
Update test_xla_sharding.py to take into account the number of devices
Author
Parents
Loading