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, end: Array, strides: Array, slice_mode: TVMString, axes: Array) -> Expr; + + #[name("relay.op._make.expand_dims")] + pub fn expand_dims(data: Expr, axis: i32, num_newaxis: i32) -> Expr; }