xla
2c2f554a
- Add optimization barrier to enable the gradient checkpointing (#3482)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
Add optimization barrier to enable the gradient checkpointing (#3482) * Add optimization barrier * Make optimization_barrier take vector and add test for correctness
References
#3482 - Add optimization barrier to enable the gradient checkpointing
Author
JackCaoG
Parents
5d9c53d7
Loading