Skip to content
Snippets Groups Projects

Low bit support for ARM Cortex-M export

Merged Thibault Allenet requested to merge low_bit_support_arm into dev
1 file
+ 8
0
Compare changes
  • Side-by-side
  • Inline
@@ -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']]:
Loading