From e9e12f033632a20ef6164d67bce1b1a49ad9a08b Mon Sep 17 00:00:00 2001
From: Zhi <5145158+zhiics@users.noreply.github.com>
Date: Tue, 18 Dec 2018 12:26:19 -0800
Subject: [PATCH] [Relay][doc] Update the description of returns in mxnet.py
(#2309)
---
python/tvm/relay/frontend/mxnet.py | 11 +++++------
1 file changed, 5 insertions(+), 6 deletions(-)
diff --git a/python/tvm/relay/frontend/mxnet.py b/python/tvm/relay/frontend/mxnet.py
index 77e97d26e..f61c65bba 100644
--- a/python/tvm/relay/frontend/mxnet.py
+++ b/python/tvm/relay/frontend/mxnet.py
@@ -343,7 +343,7 @@ _convert_map.update({k : _rename(k) for k in _identity_list})
def _from_mxnet_impl(symbol, shape_dict, dtype_info):
- """Convert mxnet symbol to nnvm implementation.
+ """Convert mxnet symbol to compatible relay Function.
Reconstruct a relay Function by traversing the mxnet symbol.
@@ -361,15 +361,14 @@ def _from_mxnet_impl(symbol, shape_dict, dtype_info):
Returns:
-------
- nnvm.sym.Symbol
- Converted symbol
+ func : tvm.relay.Function
+ Converted relay Function
"""
assert symbol is not None
jgraph = json.loads(symbol.tojson())
jnodes = jgraph["nodes"]
node_map = {}
-
for nid, node in enumerate(jnodes):
children = [node_map[e[0]][e[1]] for e in node["inputs"]]
attrs = StrAttrsDict(node.get("attrs", {}))
@@ -444,8 +443,8 @@ def from_mxnet(symbol,
Returns
-------
- sym : nnvm.Symbol
- Compatible nnvm symbol
+ sym : tvm.relay.Function
+ Compatible relay Function
params : dict of str to tvm.NDArray
The parameter dict to be used by nnvm
--
GitLab