This is an automated email from the ASF dual-hosted git repository.
comaniac pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 08d7345 fix Relay build docstring (#7963)
08d7345 is described below
commit 08d73454fd5ad118d45d699025e168368c8fbf5f
Author: Andrew Liu <[email protected]>
AuthorDate: Sat May 1 22:31:56 2021 -0700
fix Relay build docstring (#7963)
---
python/tvm/relay/build_module.py | 20 ++++++++++----------
1 file changed, 10 insertions(+), 10 deletions(-)
diff --git a/python/tvm/relay/build_module.py b/python/tvm/relay/build_module.py
index ed59ad9..6eb684e 100644
--- a/python/tvm/relay/build_module.py
+++ b/python/tvm/relay/build_module.py
@@ -111,8 +111,14 @@ class BuildModule(object):
Returns
-------
- factory_module :
tvm.relay.backend.graph_executor_factory.GraphExecutorFactoryModule
- The runtime factory for the TVM graph executor.
+ graph_json : str
+ The json string that can be accepted by graph executor.
+
+ mod : tvm.Module
+ The module containing necessary libraries.
+
+ params : dict
+ The parameters of the final graph.
"""
target = _update_target(target)
target, target_host = Target.check_and_update_host_consist(
@@ -245,14 +251,8 @@ def build(ir_mod, target=None, target_host=None,
params=None, mod_name="default"
Returns
-------
- graph_json : str
- The json string that can be accepted by graph executor.
-
- mod : tvm.Module
- The module containing necessary libraries.
-
- params : dict
- The parameters of the final graph.
+ factory_module :
tvm.relay.backend.graph_executor_factory.GraphExecutorFactoryModule
+ The runtime factory for the TVM graph executor.
"""
# pylint: enable=line-too-long
# fmt: on
