concrete.ml.onnx.onnx_utils

Previousconcrete.ml.onnx.onnx_model_manipulations Nextconcrete.ml.onnx.ops_impl

Last updated 2 years ago

Was this helpful?

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
ONNX_COMPARISON_OPS_TO_NUMPY_IMPL_BOOL
ONNX_OPS_TO_NUMPY_IMPL_BOOL
IMPLEMENTED_ONNX_OPS

function `get_attribute`

get_attribute(attribute: AttributeProto) → Any

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_name`

get_op_name(node)

Construct the qualified name of the ONNX operator.

Args:

node (Any): ONNX graph node

Returns:

result (str): qualified name

function `execute_onnx_with_numpy`

execute_onnx_with_numpy(graph: GraphProto, *inputs: ndarray) → Tuple[ndarray, ]

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.