[exo] Register Instance Norm fuse (#9169)

This commit registers Instance norm fusion to exo optimize pass

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>

diff --git a/compiler/exo/src/Passes.h b/compiler/exo/src/Passes.h
index fd20830..2a702d0 100644 (file)
--- a/compiler/exo/src/Passes.h
+++ b/compiler/exo/src/Passes.h
@@ -23,6 +23,7 @@
 #include "Pass/FoldReshapeOfConstPass.h"
 #include "Pass/FoldTransposeOfConstPass.h"
 #include "Pass/FuseBiasAddPass.h"
+#include "Pass/FuseInstanceNormPass.h"
 #include "Pass/FuseReluPass.h"
 #include "Pass/FuseRsqrtPass.h"
 #include "Pass/FuseSquaredDifferencePass.h"

diff --git a/compiler/exo/src/TFLOptimize.cpp b/compiler/exo/src/TFLOptimize.cpp
index 27d2168..f732fd0 100644 (file)
--- a/compiler/exo/src/TFLOptimize.cpp
+++ b/compiler/exo/src/TFLOptimize.cpp
@@ -43,6 +43,11 @@ void optimize(loco::Graph *g)
       phase.emplace_back(stdex::make_unique<FuseBiasAddPass>());
     }
 
+    if (get<Knob::UseFuseInstanceNormPass>())
+    {
+      phase.emplace_back(stdex::make_unique<FuseInstanceNormPass>());
+    }
+
     if (get<Knob::UseFuseReluPass>())
     {
       phase.emplace_back(stdex::make_unique<FuseReluPass>());