Fix Crash in inserting Q / DQ around MaxPool for TF adaptor (#997)

Signed-off-by: zehao-intel <zehao.huang@intel.com>

Author: zehao-intel

neural_compressor/adaptor/tf_utils/graph_rewriter/qdq/insert_qdq_pattern.py

@@ -250,7 +250,8 @@ def _insert_qdq_pattern_for_common_ops(self, original_node, is_asymmetric):
                 continue
 
             # the qdq in pattern dq+maxpool+q should be the same type in itex mode
-            if self.itex_mode and self.node_name_mapping[each_input_name].node.op == "MaxPool":
+            if self.itex_mode and each_input_name in self.node_name_mapping and \
+                    self.node_name_mapping[each_input_name].node.op == "MaxPool":
                 maxpool_node = self.graph_info[each_input_name].node
                 dtype = dtypes.DType(self.graph_info[maxpool_node.input[0]].node.attr["T"].type)
             elif self.node_name_mapping[original_node.name].node.op == "MatMul":

neural_compressor/adaptor/tf_utils/graph_util.py

@@ -425,6 +425,8 @@ def replace_const_node(self,
                                                                         outputs=target_node)
 
         for sub_node in target_node:
+            if not sub_node in self.node_name_details:
+                continue
             for index, each_node_name in enumerate(self.node_name_details[sub_node].node.input):
                 if each_node_name + ':0' == old_constant_node_name \
                         or each_node_name == old_constant_node_name: