Skip to content

Commit

Permalink
review comments
Browse files Browse the repository at this point in the history
  • Loading branch information
siju-samuel committed Oct 9, 2018
1 parent 7bd19f3 commit 054496a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions include/tvm/relay/attrs/nn.h
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@ struct LRNAttrs : public tvm::AttrsNode<LRNAttrs> {
TVM_ATTR_FIELD(size).set_default(5)
.describe("The size of the local region to be considered for normalization.");
TVM_ATTR_FIELD(axis).set_default(1)
.describe("Input data layout channel axis");
.describe("Axis of input data layout channel.");
TVM_ATTR_FIELD(bias).set_default(2)
.describe("The offset parameter to avoid division by 0.");
TVM_ATTR_FIELD(alpha).set_default(0.0001)
Expand All @@ -207,7 +207,7 @@ struct L2NormalizeAttrs : public tvm::AttrsNode<L2NormalizeAttrs> {
TVM_ATTR_FIELD(eps)
.describe("A lower bound value for the norm, to avoid division by 0.");
TVM_ATTR_FIELD(axis)
.describe("axis over the normalization applied");
.describe("Axis over the normalization applied.");
}
};

Expand Down

0 comments on commit 054496a

Please sign in to comment.