fix a bug in find_output_consumers

Author: sdpython

tests/backend_test_base.py

@@ -73,8 +73,6 @@ def run_onnxruntime(self, model_path, inputs, output_names):
         # opt.log_severity_level = 0
         # opt.log_verbosity_level = 255
         # opt.enable_profiling = True
-        with open(model_path, 'rb') as f:
-            onx = onnx.load(f)
         try:
             m = rt.InferenceSession(model_path, opt)
         except Fail as e:

tf2onnx/graph.py

@@ -1286,7 +1286,10 @@ def find_output_consumers(self, output_name):
             if output_name in node.input:
                 nodes.append(node)
 
-            # find consumers in sub graphs
+        for node in self.get_nodes():
+            # find consumers in sub graphs,
+            # should we keep an index of nodes including
+            # a subgraphs?
             body_graphs = node.get_body_graphs()
             if body_graphs:
                 for g in body_graphs.values():