shingjan commented on code in PR #12485:
URL: https://github.com/apache/tvm/pull/12485#discussion_r951888906
##########
python/tvm/relay/frontend/pytorch.py:
##########
@@ -3919,7 +3950,8 @@ def _handel_nested_input(inputs):
out_names = _get_output_names(op_node)
outputs.update(zip(out_names, relay_out))
else:
- assert op_node.outputsSize() == 1
+ # The node_name of embedding_bag is like "22_23_24_25"
Review Comment:
this change looks very clean to me. We may want to check if this node is
`embedding_bag` before we do the split. My concern is that this change may not
work on other ops that return more than one outputs.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
