concrete.ml.onnx.onnx_utils.md

module concrete.ml.onnx.onnx_utils

Utils to interpret an ONNX model with numpy.

Global Variables

• ATTR_TYPES

• ATTR_GETTERS

• ONNX_OPS_TO_NUMPY_IMPL

• ONNX_COMPARISON_OPS_TO_NUMPY_IMPL_FLOAT

function get_attribute

Get the attribute from an ONNX AttributeProto.

Args:

• attribute (onnx.AttributeProto): The attribute to retrieve the value from.

Returns:

• Any: The stored attribute value.

function get_op_type

Construct the qualified type name of the ONNX operator.

Args:

• node (Any): ONNX graph node

Returns:

• result (str): qualified name

function execute_onnx_with_numpy

Execute the provided ONNX graph on the given inputs.

Args:

• graph (onnx.GraphProto): The ONNX graph to execute.

• *inputs: The inputs of the graph.

Returns:

• Tuple[numpy.ndarray]: The result of the graph's execution.

function remove_initializer_from_input

Remove initializers from model inputs.

In some cases, ONNX initializers may appear, erroneously, as graph inputs. This function searches all model inputs and removes those that are initializers.

Args:

• model (onnx.ModelProto): the model to clean

Returns:

• onnx.ModelProto: the cleaned model