[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 <[email protected]>
Samsung · Dec 2, 2024 · c84a20c · c84a20c
1 parent 39c39a3
commit c84a20c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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();
 }