feat(api): add optimization flag to ORT extended optimizations

ssube · Dec 27, 2023 · e8a30ae · e8a30ae
1 parent 006f983
commit e8a30ae
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py
@@ -161,6 +161,9 @@ def sess_options(self, cache=True) -> SessionOptions:
         elif "onnx-graph-basic" in self.optimizations:
             logger.debug("enabling basic ONNX graph optimizations")
             sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_BASIC
+        elif "onnx-graph-extended" in self.optimizations:
+            logger.debug("enabling extended ONNX graph optimizations")
+            sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_EXTENDED
         elif "onnx-graph-all" in self.optimizations:
             logger.debug("enabling all ONNX graph optimizations")
             sess.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL