From e8a30ae56b67e184f2ad456fa229991f18373ba9 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 26 Dec 2023 20:22:32 -0600 Subject: [PATCH] feat(api): add optimization flag to ORT extended optimizations --- api/onnx_web/params.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index 5a84aa1a..f4b74659 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -161,6 +161,9 @@ class DeviceParams: 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