Fix ITEX resnetv2_50 tuning accuracy issue (#283)

Signed-off-by: Lv, Liang1 <liang1.lv@intel.com>
intel · Dec 15, 2022 · ae1e05d · ae1e05d
1 parent 88101e5
commit ae1e05d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_compressor/adaptor/tf_utils/graph_converter.py b/neural_compressor/adaptor/tf_utils/graph_converter.py
@@ -434,8 +434,8 @@ def _search_y_pattern_for_itex(self):
         g = GraphAnalyzer()
         g.graph = self._fp32_model.graph_def
         g.parse_graph()
-        y_pattern = [['Conv2D', 'MatMul'], ['BiasAdd'], ['Add', 'AddV2'], ('Relu',)]
-        y_pattern_variant = [['MaxPool', 'AvgPool'], ['Add', 'AddV2'], ('Relu',)]
+        y_pattern = [['Conv2D', 'MatMul'], ['BiasAdd'], ['Add', 'AddV2', 'AddN'], ('Relu',)]
+        y_pattern_variant = [['MaxPool', 'AvgPool'], ['Add', 'AddV2', 'AddN'], ('Relu',)]
         target_nodes = g.query_fusion_pattern_nodes(y_pattern)
         target_nodes_variant = g.query_fusion_pattern_nodes(y_pattern_variant)