diff --git a/python_binding/operator/pybind_OperatorTensor.cpp b/python_binding/operator/pybind_OperatorTensor.cpp index f5c95b2c6d1ba0ae7b23e9748896b26f973333a7..ce34dea158e6df1466db415b2539962c2113d42b 100644 --- a/python_binding/operator/pybind_OperatorTensor.cpp +++ b/python_binding/operator/pybind_OperatorTensor.cpp @@ -21,7 +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) + .def("output_dims_forwarded", &OperatorTensor::outputDimsForwarded) ; } }