more rust bindings

fix

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

@@ -129,3 +129,16 @@ pub struct AvgPool2DAttrsNode {
     pub ceil_mode: bool,
     pub count_include_pad: bool
 }
+
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "UpSamplingAttrs"]
+#[type_key = "relay.attrs.UpSamplingAttrs"]
+pub struct UpSamplingAttrsNode {
+  pub base: BaseAttrsNode,
+  pub scale_h: f64,
+  pub scale_w: f64,
+  pub layout: TString,
+  pub method: TString,
+  pub align_corners: bool
+}

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

@@ -18,8 +18,13 @@
  */
 
 use crate::ir::attrs::BaseAttrsNode;
+use crate::ir::PrimExpr;
+use crate::runtime::array::Array;
+use crate::runtime::ObjectRef;
 use tvm_macros::Object;
 
+type IndexExpr = PrimExpr;
+
 #[repr(C)]
 #[derive(Object, Debug)]
 #[ref_name = "ExpandDimsAttrs"]
@@ -29,3 +34,50 @@ pub struct ExpandDimsAttrsNode {
     pub axis: i32,
     pub num_newaxis: i32,
 }
+
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "ConcatenateAttrs"]
+#[type_key = "relay.attrs.ConcatenateAttrs"]
+pub struct ConcatenateAttrsNode {
+  pub base: BaseAttrsNode,
+  pub axis: i32
+}
+
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "ReshapeAttrs"]
+#[type_key = "relay.attrs.ReshapeAttrs"]
+pub struct ReshapeAttrsNode {
+  pub base: BaseAttrsNode,
+  pub newshape: Array<IndexExpr>,
+  pub reverse: bool
+}
+
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "SplitAttrs"]
+#[type_key = "relay.attrs.SplitAttrs"]
+pub struct SplitAttrsNode {
+  pub base: BaseAttrsNode,
+  pub indices_or_sections: ObjectRef,
+  pub axis: i32
+}
+
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "TransposeAttrs"]
+#[type_key = "relay.attrs.TransposeAttrs"]
+pub struct TransposeAttrsNode {
+  pub base: BaseAttrsNode,
+  pub axes: Array<IndexExpr>
+}
+
+#[repr(C)]
+#[derive(Object, Debug)]
+#[ref_name = "SqueezeAttrs"]
+#[type_key = "relay.attrs.SqueezeAttrs"]
+pub struct SqueezeAttrsNode {
+  pub base: BaseAttrsNode,
+  pub axis: Array<IndexExpr>
+}