Skip to content

Commit 97da0e6

Browse files
committed
Fix typo
1 parent d6fc716 commit 97da0e6

File tree

3 files changed

+4
-8
lines changed
  • include/tvm/relay/attrs
  • python/tvm/relay/op/nn
  • src/relay/op/nn

3 files changed

+4
-8
lines changed

include/tvm/relay/attrs/nn.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,6 @@ struct Conv2DTransposeAttrs : public tvm::AttrsNode<Conv2DTransposeAttrs> {
442442
}
443443
};
444444

445-
////
446445
/*! \brief Attributes used in dilate operator */
447446
struct DilateAttrs : public tvm::AttrsNode<DilateAttrs> {
448447
Array<IndexExpr> strides;
@@ -452,7 +451,6 @@ struct DilateAttrs : public tvm::AttrsNode<DilateAttrs> {
452451
.describe("Dilation stride on each dimension, 1 means no dilation.");
453452
}
454453
};
455-
////
456454

457455
/*! \brief Attributes used in 1D transposed convolution operator */
458456
struct Conv1DTransposeAttrs : public tvm::AttrsNode<Conv1DTransposeAttrs> {

python/tvm/relay/op/nn/nn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1355,7 +1355,7 @@ def dilate(data, strides):
13551355
data : tvm.relay.Expr
13561356
n-D, can be any layout.
13571357
1358-
strides : <tuple of n <int>
1358+
strides : <tuple of <int>
13591359
Dilation stride on each dimension, 1 means no dilation.
13601360
13611361
Returns

src/relay/op/nn/nn.cc

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -961,14 +961,13 @@ Do log on the data - do not accept logits.
961961
.add_type_rel("CrossEntropy", CrossEntropyRel);
962962

963963

964-
/////
965964
// relay.nn.dilate
966965
TVM_REGISTER_NODE_TYPE(DilateAttrs);
967966

968967
bool DilateRel(const Array<Type>& types,
969-
int num_inputs,
970-
const Attrs& attrs,
971-
const TypeReporter& reporter) {
968+
int num_inputs,
969+
const Attrs& attrs,
970+
const TypeReporter& reporter) {
972971
CHECK_EQ(types.size(), 2);
973972
const auto* x = types[0].as<TensorTypeNode>();
974973
const DilateAttrs* param = attrs.as<DilateAttrs>();
@@ -1009,7 +1008,6 @@ Dilate data with zeros.
10091008
.add_argument("x", "1D Tensor", "Data to dilate.")
10101009
.set_support_level(10)
10111010
.add_type_rel("Dilate", DilateRel);
1012-
/////
10131011

10141012
// Positional relay function to create cross_entropy_with_logits operator used by frontend FFI.
10151013
Expr MakeCrossEntropyWithLogits(Expr predictions, Expr targets) {

0 commit comments

Comments
 (0)