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
2 merge requests!46Remove Operator reference to Tensor,!20Draft: Introduction of Tiling
......@@ -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"))
......
......@@ -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)
;
}
}
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