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

[COREML]multiple output support, reshape, split ops added #6296

Merged
merged 2 commits into from
Aug 21, 2020

Conversation

siju-samuel
Copy link
Member

Reshape, split support added in coreml

@FrozenGene @masahi please help me to review this PR. TIA

@siju-samuel siju-samuel requested a review from FrozenGene August 18, 2020 13:34
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)
Copy link
Contributor

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.

@@ -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
Copy link
Contributor

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?

@ZihengJiang ZihengJiang merged commit 4876813 into apache:master Aug 21, 2020
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
* [COREML]multiple output support, reshape, split ops added

* Review comments addressed
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
* [COREML]multiple output support, reshape, split ops added

* Review comments addressed
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
* [COREML]multiple output support, reshape, split ops added

* Review comments addressed
electriclilies pushed a commit to electriclilies/tvm that referenced this pull request Aug 26, 2020
* [COREML]multiple output support, reshape, split ops added

* Review comments addressed
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Sep 2, 2020
* [COREML]multiple output support, reshape, split ops added

* Review comments addressed
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Sep 3, 2020
* [COREML]multiple output support, reshape, split ops added

* Review comments addressed
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.

4 participants