onnxruntime/include/onnxruntime/core
Wei-Sheng Chin de9da123cf
Enable static memory planning for pipeline. (#4204)
* Enable static memory planning for pipeline.
1. We fix a bug when resolving symbolic shape for scalars.
2. We pass the original inputs to all pipeline stages so that
   the symbolic shapes can be resolved.

* Further Improvements
1. Address comments.
2. Further reduce activation size by ~50% when pipeline is on.
   This is done by removing all but one gradient tensor from the last
   RecordEvent in the backward pass.

* Address a comment

* Fix Windows build
2020-06-12 21:43:50 -07:00
..
common Fix Windows Inbox build failing on 1) building raw api tests and 2) referencing _winml namespace in onnxruntime.dll (#3872) 2020-05-08 15:59:16 -07:00
framework Handle mem pattern allocation failure better. Make BFCArena behavior more consistent (#4062) 2020-06-05 18:54:01 +10:00
graph Enable static memory planning for pipeline. (#4204) 2020-06-12 21:43:50 -07:00
optimizer Introduce training changes. 2020-03-11 14:39:03 -07:00
platform Parallel all the activations ops (#3722) 2020-05-05 01:18:17 -07:00
providers Add ArmNN Execution Provider (#3714) 2020-06-03 22:57:51 +05:30
session Update documentation + Update mlas threading lib to use the new TrySimpleParallelFor. (#3779) 2020-05-01 00:23:06 -07:00