onnxruntime/onnxruntime
Scott McKay dc50aa42d5
Refactor session state finalization and kernel lookup usage (#4763)
* Refactor SessionState to support coming de/serialization changes
  - move more parts into SessionState to simplify usage
  - do the kernel lookup once instead of multiple times from different places
  - rename finalize_session_state.* to session_state_utils.* as the finalization logic is now inside SessionState

* Fix some build issues

* Move subgraph session state creation into SessionState. It's not needed by GraphPartitioner any more so we can delay the creation until later. Fixes issue where EP may have removed the subgraph during partitioning when taking a control flow node, and SessionState thought the subgraph was still valid.

* Address PR comments

* Clarify a comment
2020-08-20 12:19:38 +10:00
..
contrib_ops Check status of element-wise op prepare functions. (#4830) 2020-08-17 16:11:30 -07:00
core Refactor session state finalization and kernel lookup usage (#4763) 2020-08-20 12:19:38 +10:00
featurizers_ops/cpu Initial checkin (#3791) 2020-05-01 14:58:49 -07:00
gsl
python Update quantization notebook to pytorch 1.6 (#4834) 2020-08-18 14:20:46 -07:00
test Refactor session state finalization and kernel lookup usage (#4763) 2020-08-20 12:19:38 +10:00
tool/etw
.style.yapf Enable running PEP8 on python scripts using flake8 (#3928) 2020-05-15 07:15:06 +10:00
__init__.py Add new PytTrch front-end (#4815) 2020-08-17 09:45:25 -07:00
ReformatSource.ps1
ReformatSourcePython.bat Update GPT2 Model Benchmark Script to Support IO Binding (#4088) 2020-06-01 15:07:48 -07:00
VSCodeCoverage.runsettings