[tvm] branch main updated: fix pooling documents (#12201)

[wu...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

wuwei pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 88bbb40540 fix pooling documents (#12201)
88bbb40540 is described below

commit 88bbb405408f9d6ae13da7a35341ed83ce8bc9f1
Author: Siyuan Feng <Hz...@sjtu.edu.cn>
AuthorDate: Thu Jul 28 01:36:22 2022 +0800

    fix pooling documents (#12201)
---
 python/tvm/topi/nn/pooling.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/python/tvm/topi/nn/pooling.py b/python/tvm/topi/nn/pooling.py
index df38889807..b12c492ed8 100644
--- a/python/tvm/topi/nn/pooling.py
+++ b/python/tvm/topi/nn/pooling.py
@@ -205,6 +205,9 @@ def pool1d(
     stride : list/tuple of one int or int
         Stride size, [stride_width]
 
+    dilation: list/tuple of two ints
+        Dilation size, [dilation_height, dilation_width]
+
     padding : list/tuple of two ints
         Pad size, [pad_left, pad_right]
 
@@ -282,6 +285,9 @@ def pool2d(
     stride : list/tuple of two ints
         Stride size, [stride_height, stride_width]
 
+    dilation: list/tuple of two ints
+        Dilation size, [dilation_height, dilation_width]
+
     padding : list/tuple of four ints
         Pad size, [pad_top, pad_left, pad_bottom, pad_right]]
 
@@ -351,6 +357,9 @@ def pool3d(
     stride : list/tuple of three ints
         Stride size, [stride_depth, stride_height, stride_width]
 
+    dilation: list/tuple of two ints
+        Dilation size, [dilation_height, dilation_width]
+
     padding : list/tuple of six ints
         Pad size, [pad_front, pad_top, pad_left, pad_back, pad_bottom, pad_right]