diff --git a/aidge_core/export_utils/export_registry.py b/aidge_core/export_utils/export_registry.py index e5b6b2098cd760c4d425b96caf7b41cc8e82c46e..8927ae5169978da81e39912ebd4e26e2655137ad 100644 --- a/aidge_core/export_utils/export_registry.py +++ b/aidge_core/export_utils/export_registry.py @@ -80,6 +80,14 @@ class ExportLib(aidge_core.OperatorImpl): aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint32])) aidge_core.register_Tensor([self._name, aidge_core.dtype.uint64], aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint64])) + aidge_core.register_Tensor([self._name, aidge_core.dtype.int4], + aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.int4])) + aidge_core.register_Tensor([self._name, aidge_core.dtype.uint4], + aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.uint4])) + aidge_core.register_Tensor([self._name, aidge_core.dtype.dual_int4], + aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_int4])) + aidge_core.register_Tensor([self._name, aidge_core.dtype.dual_uint4], + aidge_core.get_key_value_Tensor(["cpu", aidge_core.dtype.dual_uint4])) @classproperty def _export_node_registry(cls) -> Dict[str, List['ExportNode']]: