diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index 5a84aa1a6..f4b746591 100644 --- 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