onnxruntime/tools/python/util
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
..
android Attempt to make the usage of the Android emulator in CIs more robust (#17903) 2023-10-15 08:42:36 +10:00
mobile_helpers Update to allow large models to be checked for mobile support. (#18357) 2023-11-17 07:20:16 +10:00
ort_format_model [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
qdq_helpers Graph transformer to ensure unique DQ nodes for QDQ node units (#15145) 2023-03-31 08:39:43 +10:00
test Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
__init__.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
__init__append.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
add_openvino_win_libs.py Enable model subgraph execution in OVEP and setting the OpenVINO dll's to the path from the OpenVINO pypi packge in OVEP and fix OVEP windows io buffer sample (#16147) 2023-06-16 19:47:09 -07:00
check_onnx_model_mobile_usability.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
convert_onnx_models_to_ort.py Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -07:00
file_utils.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
get_azcopy.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
logger.py Add tool to fix lines > 120 chars. (#18293) 2023-11-09 10:12:57 +10:00
make_dynamic_shape_fixed.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
onnx_model_utils.py Update to allow large models to be checked for mobile support. (#18357) 2023-11-17 07:20:16 +10:00
optimize_onnx_model.py Fix url in the code documentation (graph optimizations) (#16770) 2023-07-20 07:02:22 -07:00
platform_helpers.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
pytorch_export_helpers.py Enable pylint and numpy rules (#15218) 2023-03-27 20:37:53 -07:00
reduced_build_config_parser.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
run.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
update_onnx_opset.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00