From c30fbe8630d41e21549af61c21b038767e04b827 Mon Sep 17 00:00:00 2001
From: Oliver Bause <oliver.bause@student.uni-tuebingen.de>
Date: Mon, 13 Nov 2023 13:52:53 +0100
Subject: [PATCH] Added 'expand_dims' to transform external calls

---
 rust/tvm/src/ir/relay/transform.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/rust/tvm/src/ir/relay/transform.rs b/rust/tvm/src/ir/relay/transform.rs
index 1cb33a50e878..736a1a5e6f91 100644
--- a/rust/tvm/src/ir/relay/transform.rs
+++ b/rust/tvm/src/ir/relay/transform.rs
@@ -19,4 +19,7 @@ external! {
 
     #[name("relay.op._make.strided_slice")]
     pub fn strided_slice(data: Expr, begin: Array<PrimExpr>, end: Array<PrimExpr>, strides: Array<PrimExpr>, slice_mode: TVMString, axes: Array<PrimExpr>) -> Expr;
+
+    #[name("relay.op._make.expand_dims")]
+    pub fn expand_dims(data: Expr, axis: i32, num_newaxis: i32) -> Expr;
 }