Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add ML Program support for more operators #19527

Merged
merged 12 commits into from
Mar 1, 2024

Conversation

skottmckay
Copy link
Contributor

Description

Add support for:

  • Clip/Relu/Relu6
  • Add/Mul/Div/Sub/Pow
  • GlobalAveragePool/GlobalMaxPool/AveragePool/MaxPool
  • Reshape
  • Gemm/MatMul

Fix some build issues/warnings from changes.

Fix a couple of potential issues with the Resize op as well (noticed due to change to reject inputs with empty data at a higher level).

Motivation and Context

Enable mobilenetv2 with ML Program

- Clip/Relu/Relu6
- Add/Mul/Div/Sub/Pow
- GlobalAveragePool/GlobalMaxPool/AveragePool/MaxPool
- Reshape
- Gemm/MatMul

Fix some build issues/warnings from changes.

Fix a couple of potential issues with the Resize op as well (noticed due to change to reject inputs with empty data at a higher level).
Comment on lines +21 to +27
# if op.type == 'const':
# if op.attributes["name"].immediateValue.tensor.strings.values[0] == "conv_0_pad_type_0":
# print(f"Conv pad_type={op.attributes['val'].immediateValue.tensor.strings.values}")
#
# if op.type == 'conv':
# #print(op)
# pass

Check notice

Code scanning / CodeQL

Commented-out code Note

This comment appears to contain commented-out code.
edgchen1
edgchen1 previously approved these changes Feb 28, 2024
@skottmckay skottmckay merged commit 2a857d9 into main Mar 1, 2024
87 of 91 checks passed
@skottmckay skottmckay deleted the skottmckay/CoremL_MLProgram_MoreOps_PR branch March 1, 2024 00:23
zz002 pushed a commit to zz002/onnxruntime that referenced this pull request Mar 7, 2024
### Description
<!-- Describe your changes. -->

Add support for:
- Clip/Relu/Relu6
- Add/Mul/Div/Sub/Pow
- GlobalAveragePool/GlobalMaxPool/AveragePool/MaxPool
- Reshape
- Gemm/MatMul

Fix some build issues/warnings from changes.

Fix a couple of potential issues with the Resize op as well (noticed due
to change to reject inputs with empty data at a higher level).

### 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. -->
Enable mobilenetv2 with ML Program
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants