module onnxrt.optim.onnx_optimisation_identity

Short summary

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