Skip to content

Commit

Permalink
update scale description and rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
Xingyu Zhou committed Oct 27, 2019
1 parent 4ed9ece commit 643c903
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 @@ -395,9 +395,9 @@ struct UpSamplingAttrs : public tvm::AttrsNode<UpSamplingAttrs> {

TVM_DECLARE_ATTRS(UpSamplingAttrs, "relay.attrs.UpSamplingAttrs") {
TVM_ATTR_FIELD(scale_h)
.describe("Should be true to preserve the values at the corner pixels");
.describe("The upsampling factor for height");
TVM_ATTR_FIELD(scale_w)
.describe("Should be true to preserve the values at the corner pixels");
.describe("The upsampling factor for width");
TVM_ATTR_FIELD(layout).set_default("NCHW")
.describe("Dimension ordering of input data. Can be 'NCHW', 'NHWC', etc."
"'N', 'C', 'H', 'W' stands for batch, channel, height, and width"
Expand Down

0 comments on commit 643c903

Please sign in to comment.