moved self._update_value_info_for_graph_output() call outside the for…

… loop Signed-off-by: Harsh Peswani <[email protected]>
quic · Dec 20, 2024 · 46934c1 · 46934c1
1 parent dbce7f1
commit 46934c1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TrainingExtensions/onnx/src/python/aimet_onnx/sequential_mse/seq_mse.py b/TrainingExtensions/onnx/src/python/aimet_onnx/sequential_mse/seq_mse.py
@@ -205,7 +205,8 @@ def _update_value_info(self):
             if node.op_type == "QcQuantizeOp":
                 self._update_value_info_for_output(node)
                 self._update_value_info_for_input(node)
-                self._update_value_info_for_graph_output()
+
+        self._update_value_info_for_graph_output()
 
     def _fill_static_tensor_name_to_proto(self):
         """