[incubator-tvm] branch main updated: Add BatchNormAttrs Rust bindings (#6678)

[tqchen]

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

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


The following commit(s) were added to refs/heads/main by this push:
     new f161a4d  Add BatchNormAttrs Rust bindings (#6678)
f161a4d is described below

commit f161a4d3f26beb222d52f24f2cdde1aa4878ec35
Author: Gus Smith <gusco...@gmail.com>
AuthorDate: Sat Oct 24 06:59:35 2020 -0700

    Add BatchNormAttrs Rust bindings (#6678)
---
 rust/tvm/src/ir/relay/attrs/nn.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/rust/tvm/src/ir/relay/attrs/nn.rs 
b/rust/tvm/src/ir/relay/attrs/nn.rs
index f743534..cb96f0f 100644
--- a/rust/tvm/src/ir/relay/attrs/nn.rs
+++ b/rust/tvm/src/ir/relay/attrs/nn.rs
@@ -94,3 +94,15 @@ pub struct SoftmaxAttrsNode {
     pub base: BaseAttrsNode,
     pub axis: i32,
 }
+
+#[repr(C)]
+#[derive(Object)]
+#[ref_name = "BatchNormAttrs"]
+#[type_key = "relay.attrs.BatchNormAttrs"]
+pub struct BatchNormAttrsNode {
+    pub base: BaseAttrsNode,
+    pub axis: i32,
+    pub epsilon: f64,
+    pub center: bool,
+    pub scale: bool,
+}