onnxruntime/orttraining
Sherlock eb0f57f0e4
Localized Recompute for Gelu and AttentionDropout (#4402)
* Gelu Activation Recompute Draft

* Prototype for localized recompute

* Introduce localized_recompute rewriter

* Command line args for enabling recompute

* Add logger to Gradient Graph Builder

* use const when possible
2020-08-04 21:48:15 -07:00
..
orttraining Localized Recompute for Gelu and AttentionDropout (#4402) 2020-08-04 21:48:15 -07:00
pytorch_frontend_examples Remove ORTModel from frontend API (#3825) 2020-05-11 18:20:33 -07:00
tools Update convergence baseline for ci_test. (#4465) 2020-07-09 15:29:36 +08:00