onnxruntime/tools/python/util/mobile_helpers
Scott McKay e7a524fea9
Update to allow large models to be checked for mobile support. (#18357)
### Description
<!-- Describe your changes. -->
Update usability checker and related infrastructure to support checking
models > 2GB.
- Add ability to set flag to keep initializers as external data
- we optimize the model as part of the checking so need to write out a
new copy.
- Handle issue with ONNX shape inferencing silently failing
- use API that supports large models but requires writing the model to a
new file
  - automate cleanup of that copy of the model

### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
Allow analysis of LLMs to determine gaps for mobile usage.

---------

Co-authored-by: Edward Chen <18449977+edgchen1@users.noreply.github.com>
2023-11-17 07:20:16 +10:00
..
test
__init__.py
check_model_can_use_ort_mobile_pkg.py Update to allow large models to be checked for mobile support. (#18357) 2023-11-17 07:20:16 +10:00
usability_checker.py Update to allow large models to be checked for mobile support. (#18357) 2023-11-17 07:20:16 +10:00