fix doc warning

apache · Jun 9, 2020 · ccac1c2 · ccac1c2
1 parent 7e699cc
commit ccac1c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/api/python/relay/transform.rst b/docs/api/python/relay/transform.rst
@@ -21,4 +21,5 @@ tvm.relay.transform
 .. automodule:: tvm.relay.transform
     :members:
     :imported-members:
+    :exclude-members: tvm.tir.IntImm
     :autosummary:
diff --git a/python/tvm/relay/transform/transform.py b/python/tvm/relay/transform/transform.py
@@ -854,7 +854,7 @@ def DenseToSparse(weight_name, weight_shape):
     weight_name: Array[String]
       Names of weights which qualified sparse contrains
 
-    weight_shape: Array[Array[IntImm]]
+    weight_shape: Array[Array[tvm.tir.IntImm]]
       Weights shape in BSR format.
 
     Returns