xla
df680da7 - [SPMD] Use xs.Mesh in test_2d_tensor_3d_mesh (#5295)

Commit

2 years ago

[SPMD] Use xs.Mesh in test_2d_tensor_3d_mesh (#5295) * use mesh in test_2d_tensor_3d_mesh * remove attributes patch

References

Author

khatwanimohit

Parents