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

[Fix] wrong class

parent 576e99f1
No related branches found
No related tags found
No related merge requests found
...@@ -21,7 +21,7 @@ void init_OperatorTensor(py::module& m){ ...@@ -21,7 +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) .def("output_dims_forwarded", &OperatorTensor::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