xla
Add Checkpoint api to use optimization barrier
#3524
Merged

Commits
  • Add PyTorch/XLA version of checkpoint with optimization barrier
    JackCaoG committed 3 years ago
  • Add checkpoint code
    JackCaoG committed 3 years ago
  • update test
    JackCaoG committed 3 years ago
Loading