concrete.ml.onnx.onnx_utils.md
module concrete.ml.onnx.onnx_utils
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
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
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_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_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
Last updated