From 1ffa75a8c88571e1fd0a5d285bd3fb301cb091e9 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 14 Jan 2021 15:42:55 +0900 Subject: [PATCH] add more doc --- python/tvm/relay/op/op.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/op/op.py b/python/tvm/relay/op/op.py index 58f8f47e7fa23..d6f3fe733a0aa 100644 --- 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