xla
1f8f8aea - Added xm.save() utility API to save the model's state without incurring into

Commit

6 years ago

Added xm.save() utility API to save the model's state without incurring into potential stalls due to asymmetric computations across TPU devices.

References

Author

dlibenzi

Committer

dlibenzi

Parents