This is an automated email from the ASF dual-hosted git repository.
zhaowu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 3c2aa1a [DOCSTRING]missing function parameters updated (#5228)
3c2aa1a is described below
commit 3c2aa1aa830c8c889a77ba56105b06243648d843
Author: Samuel <siju.sam...@huawei.com>
AuthorDate: Fri Apr 3 19:11:28 2020 +0530
[DOCSTRING]missing function parameters updated (#5228)
---
python/tvm/relay/op/nn/nn.py | 28 ++++++++++++++++++++++++++--
1 file changed, 26 insertions(+), 2 deletions(-)
diff --git a/python/tvm/relay/op/nn/nn.py b/python/tvm/relay/op/nn/nn.py
index c62b1cf..64148bb 100644
--- a/python/tvm/relay/op/nn/nn.py
+++ b/python/tvm/relay/op/nn/nn.py
@@ -475,8 +475,8 @@ def log_softmax(data, axis=-1):
data: tvm.relay.Expr
The input data to the operator.
- axis: int
- The axis to sum over when computing softmax
+ axis: int, optional
+ The axis to sum over when computing log softmax
Returns
-------
@@ -510,6 +510,9 @@ def max_pool1d(data,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : int or tuple of int, optional
The strides of pooling.
@@ -567,6 +570,9 @@ def max_pool2d(data,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : tuple of int, optional
The strides of pooling.
@@ -612,6 +618,9 @@ def max_pool3d(data,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : tuple of int, optional
The strides of pooling.
@@ -658,6 +667,9 @@ def avg_pool1d(data,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : int or tuple of int, optional
The strides of pooling.
@@ -720,6 +732,9 @@ def avg_pool2d(data,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : tuple of int, optional
The strides of pooling.
@@ -769,6 +784,9 @@ def avg_pool3d(data,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : tuple of int, optional
The strides of pooling.
@@ -811,6 +829,9 @@ def max_pool2d_grad(out_grad,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : tuple of int, optional
The strides of pooling.
@@ -851,6 +872,9 @@ def avg_pool2d_grad(out_grad,
data : tvm.relay.Expr
The input data to the operator.
+ pool_size : int or tuple of int, optional
+ The size of window for pooling.
+
strides : tuple of int, optional
The strides of pooling.
