From c84a20c0e0ce8004be365799591975dce6129326 Mon Sep 17 00:00:00 2001
From: SaeHie Park <saehie.park@gmail.com>
Date: Mon, 2 Dec 2024 15:10:50 +0900
Subject: [PATCH] [luci/pass] Adjust input channel for 3D InstNorm

This will adjust input channel as second dimension for 3D Instance Norm,
in FuseInstanceNormPass.

ONE-DCO-1.0-Signed-off-by: SaeHie Park <saehie.park@gmail.com>
---
 compiler/luci/pass/src/FuseInstanceNormPass.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/compiler/luci/pass/src/FuseInstanceNormPass.cpp b/compiler/luci/pass/src/FuseInstanceNormPass.cpp
index 5427e1fe69c..4bbf8192967 100644
--- a/compiler/luci/pass/src/FuseInstanceNormPass.cpp
+++ b/compiler/luci/pass/src/FuseInstanceNormPass.cpp
@@ -1075,6 +1075,11 @@ uint32_t PostFusion::input_channel(void)
   if (input_rank < 1)
     return 0;
 
+  if (input_rank == 3)
+  {
+    // use dim 1
+    return input->dim(1).value();
+  }
   // assume channel-last
   return input->dim(input_rank - 1).value();
 }