File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
neural_compressor/adaptor/tf_utils/graph_rewriter/generic Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -108,6 +108,11 @@ def get_optimized_model(self, itex_mode=False):
108
108
109
109
self ._tmp_graph_def = SplitSharedInputOptimizer (self ._tmp_graph_def ).do_transformation ()
110
110
111
+ # Put FuseDecomposedBNOptimizer before GraphFoldConstantOptimizer
112
+ # The 'Sub' op in the small decomposed ops of BN will be converted to const by GraphFoldConstantOptimizer.
113
+ # Then the FuseDecomposedBNOptimizer can't fuse the small decomposed ops to BN.
114
+ self ._tmp_graph_def = FuseDecomposedBNOptimizer (self ._tmp_graph_def ).do_transformation ()
115
+
111
116
# disable fold constant for itex qdq mode
112
117
if not itex_mode :
113
118
self ._tmp_graph_def = GraphFoldConstantOptimizer (self ._tmp_graph_def ).do_transformation ()
@@ -121,8 +126,6 @@ def get_optimized_model(self, itex_mode=False):
121
126
122
127
self ._tmp_graph_def = GraphCseOptimizer (self ._tmp_graph_def ).do_transformation ()
123
128
124
- self ._tmp_graph_def = FuseDecomposedBNOptimizer (self ._tmp_graph_def ).do_transformation ()
125
-
126
129
self ._tmp_graph_def = FoldBatchNormNodesOptimizer (
127
130
self ._tmp_graph_def ).do_transformation ()
128
131
You can’t perform that action at this time.
0 commit comments