Add BatchNormAttrs Rust bindings (apache#6678)

trevor-m · Dec 2, 2020 · 3045b74 · 3045b74
1 parent e268e99
commit 3045b74
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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,
+}