[tvm] branch main updated: Missed out_layout field of conv1d attrs (#11325)

[wu...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

wuwei pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 0ed0f1d651 Missed out_layout field of conv1d attrs (#11325)
0ed0f1d651 is described below

commit 0ed0f1d65126dd24cf35566128fe1a097335b5a4
Author: apeskov <pe...@gmail.com>
AuthorDate: Mon May 16 20:11:58 2022 +0300

    Missed out_layout field of conv1d attrs (#11325)
    
    Signed-off-by: Alexander Peskov <pe...@gmail.com>
---
 include/tvm/relay/attrs/nn.h | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/include/tvm/relay/attrs/nn.h b/include/tvm/relay/attrs/nn.h
index 008baf9245..2cb33c774a 100644
--- a/include/tvm/relay/attrs/nn.h
+++ b/include/tvm/relay/attrs/nn.h
@@ -99,6 +99,12 @@ struct Conv1DAttrs : public tvm::AttrsNode<Conv1DAttrs> {
             "Dimension ordering of weight. Can be 'OIW', or 'WIO', etc."
             "'O', 'I', 'W' stands for num_filter, input_channel, and width"
             "dimensions respectively.");
+    TVM_ATTR_FIELD(out_layout)
+        .set_default("")
+        .describe(
+            "Dimension ordering of output. Can be 'NCW', 'NWC', etc."
+            "'N', 'C', 'W' stands for batch, channel, and width"
+            "dimensions respectively. Default to be same as input layout.");
 
     // use 0 bits to indicate none.
     TVM_ATTR_FIELD(out_dtype)