add more doc

apache · Jan 14, 2021 · f658556 · f658556
1 parent 81ef7b1
commit f658556
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/relay/op/op.py b/python/tvm/relay/op/op.py
@@ -365,7 +365,9 @@ def register_shape_func(op_name, data_dependant, shape_func=None, level=10):
         The name of the op.
 
     data_dependant : bool or list of bool
-        Whether the shape function depends on input data.
+        Whether the shape function depends on input data. If this is a list of bool,
+        the length of the list must be the same as the number of arguments of this op.
+        The list specifies per-input data dependence of the op.
 
     shape_func : function (attrs: Attrs, inputs: List[Tensor], out_ndims: List[IndexExpr])
                  -> shape_tensors: List<Tensor>