Skip to content
Snippets Groups Projects

feat/script_compare_layers_aidge_onnx

5 files
+ 82
32
Compare changes
  • Side-by-side
  • Inline
Files
5
@@ -22,6 +22,21 @@ def generic_export(
opset:int = None,
enable_custom_op: bool = False,
verbose: bool = False) -> None:
"""Function to export a :py:class:`aidge_core.GenericOperator` to an ONNX node
:param aidge_node: Aidge node containing a :py:class:`aidge_core.GenericOperator`
:type aidge_node: aidge_core.Node
:param node_inputs_name: list of names of inputs node
:type node_inputs_name: list[str]
:param node_outputs_name: list of names of outputs node
:type node_outputs_name: list[str]
:param opset: opset to use for the export, defaults to None
:type opset: int, optional
:param enable_custom_op: If True, the export will not fait if the type associated to the :py:class:`aidge_core.GenericOperator` is not , defaults to False
:type enable_custom_op: bool, optional
:param verbose: If true, verbosity is activated, defaults to False
:type verbose: bool, optional
"""
aidge_operator = aidge_node.get_operator()
if not issubclass(type(aidge_operator), aidge_core.GenericOperatorOp):
Loading