-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
[COREML]multiple output support, reshape, split ops added #6296
Conversation
outname = outnames if isinstance(outnames, _base.string_types) else outnames[0] | ||
etab.set_expr(outname, outs, force_override=True) | ||
else: | ||
assert len(outnames) == len(outs) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
minor suggestion: it would be helpful to have a message for the assertion here.
python/tvm/relay/frontend/coreml.py
Outdated
@@ -474,7 +484,7 @@ def coreml_op_to_relay(op, inname, outname, etab): | |||
inname : str or list of str | |||
Name of the input Relay expression. | |||
|
|||
outname : str | |||
outnames : str |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this also expected to be str or list of str
, based on usage on line 509?
* [COREML]multiple output support, reshape, split ops added * Review comments addressed
* [COREML]multiple output support, reshape, split ops added * Review comments addressed
* [COREML]multiple output support, reshape, split ops added * Review comments addressed
* [COREML]multiple output support, reshape, split ops added * Review comments addressed
* [COREML]multiple output support, reshape, split ops added * Review comments addressed
* [COREML]multiple output support, reshape, split ops added * Review comments addressed
Reshape, split support added in coreml
@FrozenGene @masahi please help me to review this PR. TIA