diff --git a/python_binding/operator/pybind_BitShift.cpp b/python_binding/operator/pybind_BitShift.cpp index 9313f46c333e2c62779a04c47df1e7378647530e..4efb2c96fc683a34923accf14238005104eb5132 100644 --- a/python_binding/operator/pybind_BitShift.cpp +++ b/python_binding/operator/pybind_BitShift.cpp @@ -30,11 +30,11 @@ BitShiftDirection enum. :param direction: direction of the bit shift (BitShiftDirection.Left or BitShiftDirection.Right) :type direction: BitShiftDirection + :param rounding: flag to apply bitshift rounding + :type rounding: boolean :param name: name of the node. )mydelimiter") - .def(py::init<BitShift_Op::BitShiftDirection>(), py::arg("direction")) - .def("direction", &BitShift_Op::direction, "Get the direction of the bit shift (left or right).") - .def("rounding", &BitShift_Op::rounding, "Apply bitshift rounding") + .def(py::init<BitShift_Op::BitShiftDirection,bool>(), py::arg("direction"),py::arg("rounding")) .def_static("get_inputs_name", &BitShift_Op::getInputsName, "Get the names of the input tensors.") .def_static("get_outputs_name", &BitShift_Op::getOutputsName, "Get the names of the output tensors.") .def_static("attributes_name", []() {