tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] MarisaKirisame commented on a change in pull request #5098: [TOPI][RELAY]Global MaxPool3d and AvgPool3d topi & relay implementation
Date Fri, 20 Mar 2020 02:35:33 GMT
MarisaKirisame commented on a change in pull request #5098: [TOPI][RELAY]Global MaxPool3d and
AvgPool3d topi & relay implementation
URL: https://github.com/apache/incubator-tvm/pull/5098#discussion_r395415481
 
 

 ##########
 File path: python/tvm/relay/op/nn/nn.py
 ##########
 @@ -943,6 +943,76 @@ def global_avg_pool2d(data,
     return _make.global_avg_pool2d(data, layout)
 
 
+def global_max_pool3d(data,
+                      layout="NCDHW"):
+    r"""3D global maximum pooling operator.
+
+    This operator takes data as input and does 3D max value calculation
+    across each window represented by DxWxH.
+
+
+    In the default case, where the data_layout is `NCDHW`
+    a data Tensor with shape `(batch_size, in_channels, depth, height, width)`,
+    to produce an output Tensor with the following rule:
+
+    with data of shape (b, c, d, h, w)
+
+    .. math::
+
+        \mbox{out}(b, c, 1, 1, 1)  =  \max_{l=0, \ldots, d},  \max_{m=0, \ldots, h},
+              \max_{n=0, \ldots, w} \mbox{datam}(b, c, l, m, n)
+
+    Parameters
+    ----------
+    data : tvm.relay.Expr
+        The input data to the operator.
+
+    layout : str, optional
 
 Review comment:
   Optional[str]
   ```suggestion
       layout : str, optional
   ```

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services

Mime
View raw message