module onnx_tools.optim.onnx_optimisation_identity#

Short summary#

module mlprodict.onnx_tools.optim.onnx_optimisation_identity

Optimisation of ONNX graphs.

source on GitHub

Functions#

function

truncated documentation

onnx_remove_node_identity

Removes as many Identity nodes as possible. The function looks into every node and subgraphs if recursive is …

Documentation#

Optimisation of ONNX graphs.

source on GitHub

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

Removes as many Identity nodes as possible. The function looks into every node and subgraphs if recursive is True for identity node. Unless such a node directy connects one input to one output, it will be removed and every other node gets its inputs or outputs accordingly renamed.

Parameters:

  • onnx_model – onnx model

  • recursive – looks into subgraphs

  • debug_info – debug information (private)

  • options – additional options (unused)

Returns:

new onnx _model

source on GitHub