Skip to content

Commit 3045b74

Browse files
gussmith23Trevor Morris
authored andcommitted
Add BatchNormAttrs Rust bindings (apache#6678)
1 parent e268e99 commit 3045b74

File tree

1 file changed

+12
-0
lines changed
  • rust/tvm/src/ir/relay/attrs

1 file changed

+12
-0
lines changed

rust/tvm/src/ir/relay/attrs/nn.rs

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,3 +94,15 @@ pub struct SoftmaxAttrsNode {
9494
pub base: BaseAttrsNode,
9595
pub axis: i32,
9696
}
97+
98+
#[repr(C)]
99+
#[derive(Object)]
100+
#[ref_name = "BatchNormAttrs"]
101+
#[type_key = "relay.attrs.BatchNormAttrs"]
102+
pub struct BatchNormAttrsNode {
103+
pub base: BaseAttrsNode,
104+
pub axis: i32,
105+
pub epsilon: f64,
106+
pub center: bool,
107+
pub scale: bool,
108+
}

0 commit comments

Comments
 (0)