Updated docstring in the partioning function

Change-Id: Ibb76dd67673d84238818f31eefe5e99b401bbae3
apache · Aug 9, 2021 · ba74d16 · ba74d16
1 parent 063dcef
commit ba74d16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tvm/relay/op/contrib/cmsisnn.py b/python/tvm/relay/op/contrib/cmsisnn.py
@@ -37,7 +37,8 @@ def partition_for_cmsisnn(mod, params=None, **opts):
 
     Returns
     -------
-    ret : annotated and partitioned module.
+    ret : Module
+        annotated and partitioned module.
     """
     if params:
         mod["main"] = bind_params_by_name(mod["main"], params)