Commit 9ba95ec4 by Joshua Z. Zhang Committed by Tianqi Chen

[FRONTEND] fix same var used in single op (#383)

* fix same var used in single op

* revert to older version
parent a3a9dbeb
......@@ -314,7 +314,7 @@ def _from_mxnet_impl(symbol, graph):
childs = symbol.get_children()
if childs:
op_name = symbol.attr('op_name')
childs = [_from_mxnet_impl(c, graph) for c in childs]
childs = [_from_mxnet_impl(childs[i], graph) for i in range(len(childs.list_outputs()))]
childs = [x for y in childs for x in _as_list(y)] # expand group symbol
node = _convert_symbol(op_name, childs, attr)
else:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment