onnxruntime/tools/python
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
..
util Update to allow large models to be checked for mobile support. (#18357) 2023-11-17 07:20:16 +10:00
__init__.py
check_onnx_model_mobile_usability.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
convert_onnx_models_to_ort.py Add ability to create ort format models from training offline utility (#16360) 2023-06-21 18:51:43 -07:00
create_reduced_build_config.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
dump_ort_model.py Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -07:00
dump_subgraphs.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
example_operator_perf_test.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
find_optimizer_opset_version_updates_required.py Bump linter versions (#18341) 2023-11-08 13:04:40 -08:00
fix_long_lines.py Add tool to fix lines > 120 chars. (#18293) 2023-11-09 10:12:57 +10:00
gen_contrib_doc.py [Better Engineering] Bump ruff to 0.0.278 and fix new lint errors (#16789) 2023-07-21 12:53:41 -07:00
gen_opkernel_doc.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
gen_ort_mobile_pkg_doc.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
get_submodules.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
onnx2tfevents.py Disable PERF* rules in ruff to allow better readability (#16834) 2023-07-25 15:38:22 -07:00
onnx_test_data_utils.py Various test infra updates from testing Azure ops with MAUI test app (#17262) 2023-08-27 09:35:00 +10:00
ort_test_dir_utils.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
PythonTools.md Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
remove_initializer_from_input.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
run_adb.py [React Native CI] Record more info to debug E2E test (#13329) 2022-10-18 17:21:28 -07:00
run_android_emulator.py Attempt to make the usage of the Android emulator in CIs more robust (#17903) 2023-10-15 08:42:36 +10:00
run_CIs_for_external_pr.py Add mobile CIs to list run by script for external PRs. (#18094) 2023-11-01 09:25:48 +10:00
sparsify_initializers.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
update_version.py [js] add API that allows to get package version (#16207) 2023-06-09 16:18:53 -07:00