From c4c3a79f054209840d592bfe4c97a00b6278e9a7 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Wed, 22 Nov 2023 17:20:08 +0000
Subject: [PATCH] [Fix] place 'outputDimsForwarded' python binding in
 'Aidge::OperatorTensor' file

---
 python_binding/operator/pybind_Operator.cpp       | 1 -
 python_binding/operator/pybind_OperatorTensor.cpp | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/python_binding/operator/pybind_Operator.cpp b/python_binding/operator/pybind_Operator.cpp
index 559c4b0d6..f9482eda2 100644
--- a/python_binding/operator/pybind_Operator.cpp
+++ b/python_binding/operator/pybind_Operator.cpp
@@ -27,7 +27,6 @@ void init_Operator(py::module& m){
     .def("nb_data", &Operator::nbData)
     .def("nb_param", &Operator::nbParam)
     .def("nb_outputs", &Operator::nbOutputs)
-    .def("output_dims_forwarded", &Operator::outputDimsForwarded)
     .def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data"))
     .def("set_datatype", &Operator::setDataType, py::arg("dataType"))
     .def("set_backend", &Operator::setBackend, py::arg("name"))
diff --git a/python_binding/operator/pybind_OperatorTensor.cpp b/python_binding/operator/pybind_OperatorTensor.cpp
index 605ee27f4..f5c95b2c6 100644
--- a/python_binding/operator/pybind_OperatorTensor.cpp
+++ b/python_binding/operator/pybind_OperatorTensor.cpp
@@ -21,6 +21,7 @@ void init_OperatorTensor(py::module& m){
     py::class_<OperatorTensor, std::shared_ptr<OperatorTensor>, Operator>(m, "OperatorTensor")
     .def("get_output", &OperatorTensor::getOutput, py::arg("outputIdx"))
     .def("get_input", &OperatorTensor::getInput, py::arg("inputIdx"))
+    .def("output_dims_forwarded", &Operator::outputDimsForwarded)
     ;
 }
 }
-- 
GitLab