Add TupleGetItemAttrs Rust bindings

apache · Oct 19, 2020 · fa49f9c · fa49f9c
1 parent 14a786d
commit fa49f9c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/rust/tvm/src/ir/relay/attrs/mod.rs b/rust/tvm/src/ir/relay/attrs/mod.rs
@@ -19,3 +19,15 @@
 
 pub mod nn;
 pub mod transform;
+
+use crate::ir::attrs::BaseAttrsNode;
+use tvm_macros::Object;
+
+#[repr(C)]
+#[derive(Object)]
+#[ref_name = "TupleGetItemAttrs"]
+#[type_key = "relay.attrs.TupleGetItemAttrs"]
+pub struct TupleGetItemAttrsNode {
+    pub base: BaseAttrsNode,
+    pub index: i32,
+}