typo fix

apache · Mar 12, 2020 · dd3d3c7 · dd3d3c7
1 parent a629ec8
commit dd3d3c7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/include/tvm/relay/attrs/nn.h b/include/tvm/relay/attrs/nn.h
@@ -537,7 +537,7 @@ struct AdaptivePool3DAttrs : public tvm::AttrsNode<AdaptivePool3DAttrs> {
       .describe("Output depth, height and width.");
     TVM_ATTR_FIELD(layout).set_default("NCDHW")
       .describe("Dimension ordering of data and weight. Can be 'NCDHW', 'NDHWC', etc."
-                  "'N', 'D', 'C', 'H', 'W' stands for batch, dpeth channel, height, and width"
+                  "'N', 'C', 'D', 'H', 'W' stands for batch, channel, depth, height, and width"
                   "dimensions respectively. Convolution is applied on 'D', 'H' and"
                   "'W' dimensions.");
   }

diff --git a/python/tvm/relay/op/nn/nn.py b/python/tvm/relay/op/nn/nn.py
@@ -2394,7 +2394,7 @@ def adaptive_max_pool3d(data,
         as output depth, height and width.
 
         If a single integer is provided for output_size, the output size is
-        (N x C x output_size, output_size x output_size) for any input (NCDHW).
+        (N x C x output_size x output_size x output_size) for any input (NCDHW).
 
         If a tuple of integers (depth, height, width) are provided for output_size,
         the output size is (N x C x depth x height x width) for any input (NCDHW).
@@ -2440,7 +2440,7 @@ def adaptive_avg_pool3d(data,
         as output depth, height and width.
 
         If a single integer is provided for output_size, the output size is
-        (N x C x output_size, output_size x output_size) for any input (NCDHW).
+        (N x C x output_size x output_size x output_size) for any input (NCDHW).
 
         If a tuple of integers (depth, height, width) are provided for output_size,
         the output size is (N x C x depth x height x width) for any input (NCDHW).

diff --git a/src/relay/op/nn/pooling.cc b/src/relay/op/nn/pooling.cc
@@ -699,7 +699,7 @@ RELAY_REGISTER_OP("nn.adaptive_max_pool3d")
                    If a single integer is provided for output_size, the output size is
                    (N x C x output_size x output_size x output_size) for any input (NCDHW).
                    If a tuple of integers (depth, height, width) are provided for output_size,
-                   the output size is (N x C x depth, height x width) for any input (NCDHW).
+                   the output size is (N x C x depth x height x width) for any input (NCDHW).
 - **out**: This depends on the `layout` parameter. Output is 5D array of shape
            (batch_size, channels, output_depth, output_height, output_width)  if `layout` is `NCDHW`.
 
@@ -736,7 +736,7 @@ RELAY_REGISTER_OP("nn.adaptive_avg_pool3d")
                    If a single integer is provided for output_size, the output size is
                    (N x C x output_size x output_size x output_size) for any input (NCDHW).
                    If a tuple of integers (depth, height, width) are provided for output_size,
-                   the output size is (N x C x depth, height x width) for any input (NCDHW).
+                   the output size is (N x C x depth x height x width) for any input (NCDHW).
 - **out**: This depends on the `layout` parameter. Output is 5D array of shape
            (batch_size, channels, output_depth, output_height, output_width)  if `layout` is `NCDHW`.
 )code" TVM_ADD_FILELINE)