From 4596591344355b3fba01c404b120f3e39d9fe3cc Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Mon, 3 Jun 2024 10:01:05 +0000
Subject: [PATCH] Fix Slice_Op constructor binding
---
python_binding/operator/pybind_Slice.cpp | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/python_binding/operator/pybind_Slice.cpp b/python_binding/operator/pybind_Slice.cpp
index a024076e1..b87cc8da4 100644
--- a/python_binding/operator/pybind_Slice.cpp
+++ b/python_binding/operator/pybind_Slice.cpp
@@ -23,10 +23,12 @@ void init_Slice(py::module& m) {
py::class_<Slice_Op, std::shared_ptr<Slice_Op>, OperatorTensor>(m, "SliceOp", py::multiple_inheritance())
.def(py::init<const std::vector<std::int64_t>&,
const std::vector<std::int64_t>&,
- const std::vector<std::int8_t>&>(),
+ const std::vector<std::int8_t>&,
+ const std::vector<std::int64_t>&>(),
py::arg("starts"),
py::arg("ends"),
- py::arg("axes"))
+ py::arg("axes"),
+ py::arg("steps"))
.def_static("get_inputs_name", &Slice_Op::getInputsName)
.def_static("get_outputs_name", &Slice_Op::getOutputsName);
declare_registrable<Slice_Op>(m, "SliceOp");
--
GitLab