Skip to content
Snippets Groups Projects
Commit 9cb10213 authored by Thibault Allenet's avatar Thibault Allenet
Browse files

Update export registry with new tensor entries

parent a7efeb7e
No related branches found
No related tags found
No related merge requests found
from aidge_core.export_utils import ExportLib from aidge_core.export_utils import ExportLib
from aidge_export_arm_cortexm.utils import ROOT
class ExportLibAidgeARM(ExportLib): class ExportLibAidgeARM(ExportLib):
_name="aidge_arm" _name="aidge_arm"
def __init__(self, operator):
super(ExportLibAidgeARM, self).__init__(operator)
self.forward_template = str(ROOT / "_Aidge_Arm" / "templates" / "forward_call" / "forward.jinja")
# TODO ugly fix for Tensor registration issue... # TODO ugly fix for Tensor registration issue...
import aidge_core import aidge_core
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.float32], aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.float32],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.float32])) aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.float32]))
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.int32],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int32]))
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.int8],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int8]))
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.int4],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int4]))
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.uint4],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint4]))
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.dual_int4],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_int4]))
aidge_core.register_Tensor(["aidge_arm", aidge_core.dtype.dual_uint4],
aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_uint4]))
class ExportLibCMSISNN(ExportLib): class ExportLibCMSISNN(ExportLib):
_name="export_cmsisnn" _name="export_cmsisnn"
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment