Source code for mlprodict.onnxrt.optim._main_onnx_optim
"""
Calls all possible :epkg:`ONNX` optimisations.
:githublink:`%|py|5`
"""
from .onnx_optimisation import onnx_remove_node
[docs]def onnx_optimisations(onnx_model, recursive=True, debug_info=None, **options):
"""
Calls several possible optimisations including
:func:`onnx_remove_node <mlprodict.onnxrt.optim.onnx_optimisation.onnx_remove_node>`.
:param onnx_model: onnx model
:param recursive: looks into subgraphs
:param debug_info: debug information (private)
:param options: additional options
:return: new onnx _model
:githublink:`%|py|18`
"""
new_model = onnx_remove_node(
onnx_model, recursive=recursive, debug_info=debug_info,
**options)
return new_model