From ccac1c2f54bfa817d7aa2aa8e29e44bf6408803c Mon Sep 17 00:00:00 2001
From: Cody Yu <comaniac0422@gmail.com>
Date: Tue, 9 Jun 2020 00:53:48 +0000
Subject: [PATCH] fix doc warning

---
 docs/api/python/relay/transform.rst     | 1 +
 python/tvm/relay/transform/transform.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/docs/api/python/relay/transform.rst b/docs/api/python/relay/transform.rst
index c66904d8bcbac..4a0f25ed6c2cc 100644
--- 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
index 8f4ec10465007..aa9f1e95179b6 100644
--- 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