Fix typo

apache · Apr 14, 2020 · 97da0e6 · 97da0e6
1 parent d6fc716
commit 97da0e6
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/include/tvm/relay/attrs/nn.h b/include/tvm/relay/attrs/nn.h
@@ -442,7 +442,6 @@ struct Conv2DTransposeAttrs : public tvm::AttrsNode<Conv2DTransposeAttrs> {
   }
 };
 
-////
 /*! \brief Attributes used in dilate operator */
 struct DilateAttrs : public tvm::AttrsNode<DilateAttrs> {
   Array<IndexExpr> strides;
@@ -452,7 +451,6 @@ struct DilateAttrs : public tvm::AttrsNode<DilateAttrs> {
       .describe("Dilation stride on each dimension, 1 means no dilation.");
   }
 };
-////
 
 /*! \brief Attributes used in 1D transposed convolution operator */
 struct Conv1DTransposeAttrs : public tvm::AttrsNode<Conv1DTransposeAttrs> {

diff --git a/python/tvm/relay/op/nn/nn.py b/python/tvm/relay/op/nn/nn.py
@@ -1355,7 +1355,7 @@ def dilate(data, strides):
     data : tvm.relay.Expr
         n-D, can be any layout.
 
-    strides : <tuple of n <int>
+    strides : <tuple of <int>
         Dilation stride on each dimension, 1 means no dilation.
 
     Returns

diff --git a/src/relay/op/nn/nn.cc b/src/relay/op/nn/nn.cc
@@ -961,14 +961,13 @@ Do log on the data - do not accept logits.
 .add_type_rel("CrossEntropy", CrossEntropyRel);
 
 
-/////
 // relay.nn.dilate
 TVM_REGISTER_NODE_TYPE(DilateAttrs);
 
 bool DilateRel(const Array<Type>& types,
-                    int num_inputs,
-                    const Attrs& attrs,
-                    const TypeReporter& reporter) {
+			   int num_inputs,
+			   const Attrs& attrs,
+			   const TypeReporter& reporter) {
   CHECK_EQ(types.size(), 2);
   const auto* x = types[0].as<TensorTypeNode>();
   const DilateAttrs* param = attrs.as<DilateAttrs>();
@@ -1009,7 +1008,6 @@ Dilate data with zeros.
 .add_argument("x", "1D Tensor", "Data to dilate.")
 .set_support_level(10)
 .add_type_rel("Dilate", DilateRel);
-/////
 
 // Positional relay function to create cross_entropy_with_logits operator used by frontend FFI.
 Expr MakeCrossEntropyWithLogits(Expr predictions, Expr targets) {