onnxruntime/tools/python/util/mobile_helpers
Scott McKay 159fe9d4f3
Update to mobile model usability checker (#19843)
### Description
<!-- Describe your changes. -->

- Add check for CoreML MLProgram supported ops
- Only check usability with ORT Mobile package if requested
- this package will be deprecated so info is a) of minimal value and b)
can be confusing.
- Output more things at INFO level
- a lot of meaningful info was only output at DEBUG level. The default
INFO level is more useful
  - dump full partition info at DEBUG level
- Check subgraphs fully
  - CoreML can handle a subgraph
- TBD if we want to add support for adding a subgraph to the parent
graph for Loop and If nodes
    - most likely will be required for simple If nodes to be performant
- Check 5D CoreML limitation

### 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. -->
Improve helper tools

---------

Co-authored-by: Edward Chen <18449977+edgchen1@users.noreply.github.com>
2024-06-18 07:50:33 +10:00
..
test Update to mobile model usability checker (#19843) 2024-06-18 07:50:33 +10:00
__init__.py
check_model_can_use_ort_mobile_pkg.py Update to mobile model usability checker (#19843) 2024-06-18 07:50:33 +10:00
usability_checker.py Update to mobile model usability checker (#19843) 2024-06-18 07:50:33 +10:00