From 309ab5a73816879cc4ce4ee742ec40b51f2ff240 Mon Sep 17 00:00:00 2001
From: thibault allenet <thibault.allenet@cea.fr>
Date: Wed, 4 Dec 2024 13:37:07 +0000
Subject: [PATCH] Add binding for applyWeightInterleaving
---
python_binding/recipes/pybind_Recipes.cpp | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/python_binding/recipes/pybind_Recipes.cpp b/python_binding/recipes/pybind_Recipes.cpp
index f656af70d..21478a5b1 100644
--- a/python_binding/recipes/pybind_Recipes.cpp
+++ b/python_binding/recipes/pybind_Recipes.cpp
@@ -151,6 +151,14 @@ void init_Recipes(py::module &m)
:param node: Node which Operator will turn into a Generic Operator
:type graph_view: :py:class:`aidge_core.Node`
)mydelimiter");
+
+ m.def("apply_weightinterleaving", applyWeightInterleaving, py::arg("node"), R"mydelimiter(
+ Replace weight Producer linked to the given node with a weight producer with interleaving and format NHWC.
+ This recipe is specific to the ARM cortex-m export for low bit integer support.
+
+ :param node: Node which linked weights will recieve interleaving
+ :type graph_view: :py:class:`aidge_core.Node`
+ )mydelimiter");
}
} // namespace Aidge
--
GitLab