Skip to content

T
tic
Commit 9ba95ec4 by Joshua Z. Zhang Committed by Tianqi Chen
Options

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

* fix same var used in single op

* revert to older version
parent a3a9dbeb
Showing with 1 additions and 1 deletions
nnvm/python/nnvm/frontend/mxnet.py
...@@ -314,7 +314,7 @@ def _from_mxnet_impl(symbol, graph): ...@@ -314,7 +314,7 @@ def _from_mxnet_impl(symbol, graph):
childs = symbol.get_children() childs = symbol.get_children()
if childs: if childs:
op_name = symbol.attr('op_name') 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 childs = [x for y in childs for x in _as_list(y)] # expand group symbol
node = _convert_symbol(op_name, childs, attr) node = _convert_symbol(op_name, childs, attr)
else: 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