diff --git a/aidge_export_arm_cortexm/export.py b/aidge_export_arm_cortexm/export.py index 5179937e86974de4e3134f9652e99ef2918751bc..26e7d87f1259a16b8f3d3bdd85e7e1f640f43de0 100644 --- a/aidge_export_arm_cortexm/export.py +++ b/aidge_export_arm_cortexm/export.py @@ -3,7 +3,7 @@ import os import shutil from pathlib import Path import numpy as np - +from aidge_core.export_utils.data_conversion import aidge2c from aidge_core.export_utils.code_generation import * from aidge_export_arm_cortexm.utils import (ROOT, AVAILABLE_BOARDS, has_board, \ OPERATORS_REGISTRY, supported_operators) @@ -98,7 +98,7 @@ def export(export_folder_name, for node in graphview.get_nodes(): if node.type() == "Producer": if not node.get_operator().get_attr("Constant"): - export_type = aidge_datatype2ctype(node.get_operator().get_output(0).dtype()) + export_type = aidge2c(node.get_operator().get_output(0).dtype()) list_inputs_name.append((export_type, node.name())) # Get output nodes @@ -108,9 +108,9 @@ def export(export_folder_name, if len(node.get_children()) == 0: if node.get_operator().has_attr("DataType"): # Temporary fix because impossible to set DataType of a generic operator - export_type = aidge_datatype2ctype(node.get_operator().get_attr("DataType")) + export_type = aidge2c(node.get_operator().get_attr("DataType")) else: - export_type = aidge_datatype2ctype(node.get_operator().get_output(0).dtype()) + export_type = aidge2c(node.get_operator().get_output(0).dtype()) list_outputs_name.append((export_type, node.name()))