module onnxrt.onnx_inference_manipulations

Short summary

module mlprodict.onnxrt.onnx_inference_manipulations

Implements a class able to compute the predictions from on an ONNX model.

source on GitHub

Functions

function	truncated documentation
enumerate_model_node_outputs	Enumerates all the nodes of a model.
select_model_inputs_outputs	Takes a model and changes its outputs.

Documentation

Implements a class able to compute the predictions from on an ONNX model.

source on GitHub

mlprodict.onnxrt.onnx_inference_manipulations.enumerate_model_node_outputs(model, add_node=False)

Enumerates all the nodes of a model.

Parameters

• model – ONNX graph

• add_node – if False, the function enumerates all output names from every node, otherwise, it enumerates tuple (output name, node)

Returns

enumerator

source on GitHub

mlprodict.onnxrt.onnx_inference_manipulations.select_model_inputs_outputs(model, outputs=None, inputs=None)

Takes a model and changes its outputs.

Parameters

• model – ONNX model

• inputs – new inputs, same ones if None

• outputs – new outputs, same ones if None

Returns

modified model

The function removes unneeded files.

source on GitHub