onnxruntime
b03fb82a - Transformer layer-wise Recompute (#4526)

Commit
5 years ago
Transformer layer-wise Recompute (#4526) * Build Recomputation Graph * Make topological sort to run FW nodes first * Pattern match start and end of transformer layer * Topological sort with Priority * Add logger to Gradient Graph Builder * Use Logger * Introduce Execution Order
Author
Parents
Loading