diff --git a/rust/tvm/src/ir/relay/attrs/nn.rs b/rust/tvm/src/ir/relay/attrs/nn.rs index e1c572ae3..c9d291113 100644 --- 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, }