Skip to content

Commit

Permalink
[ add ] PaddAttrs and format
Browse files Browse the repository at this point in the history
  • Loading branch information
AD1024 authored and gussmith23 committed Dec 29, 2021
1 parent 251a4ec commit 2c93393
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 2 deletions.
2 changes: 1 addition & 1 deletion rust/tvm/src/ir/relay/attrs/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -18,5 +18,5 @@
*/

pub mod nn;
pub mod transform;
pub mod reduce;
pub mod transform;
10 changes: 10 additions & 0 deletions rust/tvm/src/ir/relay/attrs/nn.rs
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,16 @@ use tvm_macros::Object;

type IndexExpr = PrimExpr;

#[repr(C)]
#[derive(Object, Debug)]
#[ref_name = "PadAttrs"]
#[type_key = "relay.attrs.PadAttrs"]
pub struct PadAttrsNode {
pub base: BaseAttrsNode,
pub pad_width: Array<Array<IndexExpr>>,
pub pad_mode: TString,
}

#[repr(C)]
#[derive(Object, Debug)]
#[ref_name = "Conv1DAttrs"]
Expand Down
2 changes: 1 addition & 1 deletion rust/tvm/src/ir/relay/attrs/reduce.rs
Original file line number Diff line number Diff line change
Expand Up @@ -45,4 +45,4 @@ pub struct VarianceAttrsNode {
pub keepdims: bool,
pub exclude: bool,
pub unbiased: bool,
}
}

0 comments on commit 2c93393

Please sign in to comment.