tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From zha...@apache.org
Subject [incubator-tvm] branch master updated: [DOCSTRING]missing function parameters updated (#5228)
Date Fri, 03 Apr 2020 13:41:39 GMT
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.samuel@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.
 


Mime
View raw message