[tvm] branch main updated: Add Conv3D bindings (#11381)

[ma...@apache.org]

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 3248793dd8 Add Conv3D bindings (#11381)
3248793dd8 is described below

commit 3248793dd8043e8fd68a4d2d104d61f1f0e71f61
Author: Andrew Cheung <43...@users.noreply.github.com>
AuthorDate: Fri May 20 02:03:24 2022 -0700

    Add Conv3D bindings (#11381)
---
 rust/tvm/src/ir/relay/attrs/nn.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/rust/tvm/src/ir/relay/attrs/nn.rs b/rust/tvm/src/ir/relay/attrs/nn.rs
index c9d2911133..040939d4f6 100644
--- a/rust/tvm/src/ir/relay/attrs/nn.rs
+++ b/rust/tvm/src/ir/relay/attrs/nn.rs
@@ -75,6 +75,25 @@ pub struct Conv2DAttrsNode {
     pub out_dtype: DataType,
 }
 
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "Conv3DAttrs"]
+#[type_key = "relay.attrs.Conv3DAttrs"]
+pub struct Conv3DAttrsNode {
+    pub base: BaseAttrsNode,
+    pub strides: Array<IndexExpr>,
+    pub padding: Array<IndexExpr>,
+    pub dilation: Array<IndexExpr>,
+    pub groups: i32,
+    pub channels: IndexExpr,
+    pub kernel_size: Array<IndexExpr>,
+    pub data_layout: TString,
+    pub kernel_layout: TString,
+    pub out_layout: TString,
+    pub auto_scheduler_rewritten_layout: TString,
+    pub out_dtype: DataType,
+}
+
 #[repr(C)]
 #[derive(Object, Debug)]
 #[ref_name = "BiasAddAttrs"]