module onnxrt.optim._main_onnx_optim

Short summary

module mlprodict.onnxrt.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.onnxrt.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