Update nn.rs (#10063)

uwsampl · Jan 31, 2022 · 5029c9d · 5029c9d
1 parent 779dc51
commit 5029c9d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rust/tvm/src/ir/relay/attrs/nn.rs b/rust/tvm/src/ir/relay/attrs/nn.rs
@@ -103,6 +103,7 @@ pub struct MatmulAttrsNode {
 pub struct DenseAttrsNode {
     pub base: BaseAttrsNode,
     pub units: IndexExpr,
+    pub auto_scheduler_rewritten_layout: TString,
     pub out_dtype: DataType,
 }