From 16a499fe3a63dd8df1e79f4d2d2ebfed3b7b4040 Mon Sep 17 00:00:00 2001
From: Noam ZERAH <noam.zerah@cea.fr>
Date: Wed, 5 Mar 2025 14:45:35 +0000
Subject: [PATCH] Updating constructor in python binding
---
python_binding/operator/pybind_BitShift.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/python_binding/operator/pybind_BitShift.cpp b/python_binding/operator/pybind_BitShift.cpp
index 9313f46c3..4efb2c96f 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", []() {
--
GitLab