Skip to content
Snippets Groups Projects
Commit c4c3a79f authored by Maxence Naud's avatar Maxence Naud
Browse files

[Fix] place 'outputDimsForwarded' python binding in 'Aidge::OperatorTensor' file

parent b1f7a9d5
No related branches found
No related tags found
No related merge requests found
...@@ -27,7 +27,6 @@ void init_Operator(py::module& m){ ...@@ -27,7 +27,6 @@ void init_Operator(py::module& m){
.def("nb_data", &Operator::nbData) .def("nb_data", &Operator::nbData)
.def("nb_param", &Operator::nbParam) .def("nb_param", &Operator::nbParam)
.def("nb_outputs", &Operator::nbOutputs) .def("nb_outputs", &Operator::nbOutputs)
.def("output_dims_forwarded", &Operator::outputDimsForwarded)
.def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data")) .def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data"))
.def("set_datatype", &Operator::setDataType, py::arg("dataType")) .def("set_datatype", &Operator::setDataType, py::arg("dataType"))
.def("set_backend", &Operator::setBackend, py::arg("name")) .def("set_backend", &Operator::setBackend, py::arg("name"))
......
...@@ -21,6 +21,7 @@ void init_OperatorTensor(py::module& m){ ...@@ -21,6 +21,7 @@ void init_OperatorTensor(py::module& m){
py::class_<OperatorTensor, std::shared_ptr<OperatorTensor>, Operator>(m, "OperatorTensor") py::class_<OperatorTensor, std::shared_ptr<OperatorTensor>, Operator>(m, "OperatorTensor")
.def("get_output", &OperatorTensor::getOutput, py::arg("outputIdx")) .def("get_output", &OperatorTensor::getOutput, py::arg("outputIdx"))
.def("get_input", &OperatorTensor::getInput, py::arg("inputIdx")) .def("get_input", &OperatorTensor::getInput, py::arg("inputIdx"))
.def("output_dims_forwarded", &Operator::outputDimsForwarded)
; ;
} }
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment