[converter] stop tracking ops with used constants (#250)

alibaba · Sep 7, 2023 · 075b97d · 075b97d
1 parent 8db9029
commit 075b97d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tinynn/converter/base.py b/tinynn/converter/base.py
@@ -414,10 +414,10 @@ def init_operations(self):
                     if self.common_graph.has_nested_names(n):
                         nested_names = self.common_graph.get_list_expanded_names(n)
                         for x in nested_names:
-                            if x in self.common_graph.tensor_map:
+                            if x in self.common_graph.tensor_map and self.common_graph.tensor_map[x].buffer is None:
                                 no_track_flag = False
                                 break
-                    elif n in self.common_graph.tensor_map:
+                    elif n in self.common_graph.tensor_map and self.common_graph.tensor_map[n].buffer is None:
                         no_track_flag = False
                         break
                 if no_track_flag: