onnxruntime/include
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
..
onnxruntime/core Localized Recompute for Gelu and AttentionDropout (#4402) 2020-08-04 21:48:15 -07:00