xla
3affe2a4 - use dim > 2 to suit TPU v3(otherwise can't be divide evenly)

Commit

2 years ago

use dim > 2 to suit TPU v3(otherwise can't be divide evenly)

References

JackCao/unordered_sharding_spec

#5305 - Supoort unordered sharding spec correctly

Author

JackCaoG

JackCaoG

Parents

Loading