This is an automated email from the ASF dual-hosted git repository.
masahi 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 05f11ad Add dilation to MaxPool2DAttrs Rust bindings (#9215)
05f11ad is described below
commit 05f11ade86d8856d77fa2da157d6b763f244d5a1
Author: Gus Smith <[email protected]>
AuthorDate: Thu Oct 7 01:22:08 2021 -0700
Add dilation to MaxPool2DAttrs Rust bindings (#9215)
Gets Rust bindings back in sync. Dilation must have been added to
MaxPool2DAttrs on the C++ side, without it being added on the Rust side. There
should probably be tests to catch this!
---
rust/tvm/src/ir/relay/attrs/nn.rs | 1 +
1 file changed, 1 insertion(+)
diff --git a/rust/tvm/src/ir/relay/attrs/nn.rs
b/rust/tvm/src/ir/relay/attrs/nn.rs
index 04320d1..f003ae6 100644
--- a/rust/tvm/src/ir/relay/attrs/nn.rs
+++ b/rust/tvm/src/ir/relay/attrs/nn.rs
@@ -94,6 +94,7 @@ pub struct MaxPool2DAttrsNode {
pub pool_size: Array<IndexExpr>,
pub strides: Array<IndexExpr>,
pub padding: Array<IndexExpr>,
+ pub dilation: Array<IndexExpr>,
pub layout: TString,
pub ceil_mode: bool,
}