From ab7fd9146a56ac5fb3a061acfa51a25b82294f9d Mon Sep 17 00:00:00 2001
From: Wissam Boussella <wissam.boussella@cea.fr>
Date: Wed, 19 Mar 2025 10:47:28 +0100
Subject: [PATCH] add pybind
---
python_binding/data/pybind_DataFormat.cpp | 1 +
1 file changed, 1 insertion(+)
diff --git a/python_binding/data/pybind_DataFormat.cpp b/python_binding/data/pybind_DataFormat.cpp
index 5308fb302..d41251153 100644
--- a/python_binding/data/pybind_DataFormat.cpp
+++ b/python_binding/data/pybind_DataFormat.cpp
@@ -67,6 +67,7 @@ void init_DataFormat(py::module& m) {
bindEnum<DataFormat>(m, "dformat");
m.def("format_as", (const char* (*)(DataFormat)) &format_as, py::arg("df"));
m.def("get_permutation_mapping", &getPermutationMapping, py::arg("src"), py::arg("dst"));
+ m.def("get_transpose_output_data_format", &getTransposeOutputDataFormat, py::arg("inputDataFormat"), py::arg("outputDimsOrder"));
}
} // namespace Aidge
--
GitLab