module onnx_tools.optim._main_onnx_optim#

Short summary#

module mlprodict.onnx_tools.optim._main_onnx_optim

Calls all possible ONNX optimisations.

source on GitHub

Functions#

function

truncated documentation

onnx_optimisations

Calls several possible optimisations including onnx_remove_node().

Documentation#

Calls all possible ONNX optimisations.

source on GitHub

mlprodict.onnx_tools.optim._main_onnx_optim.onnx_optimisations(onnx_model, recursive=True, debug_info=None, **options)#

Calls several possible optimisations including onnx_remove_node.

Parameters:

  • onnx_model – onnx model

  • recursive – looks into subgraphs

  • debug_info – debug information (private)

  • options – additional options

Returns:

new onnx _model

source on GitHub