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

Fix Slice_Op constructor binding

parent 24a1e8d3
No related branches found
No related tags found
No related merge requests found
...@@ -23,10 +23,12 @@ void init_Slice(py::module& m) { ...@@ -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()) py::class_<Slice_Op, std::shared_ptr<Slice_Op>, OperatorTensor>(m, "SliceOp", py::multiple_inheritance())
.def(py::init<const std::vector<std::int64_t>&, .def(py::init<const std::vector<std::int64_t>&,
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("starts"),
py::arg("ends"), py::arg("ends"),
py::arg("axes")) py::arg("axes"),
py::arg("steps"))
.def_static("get_inputs_name", &Slice_Op::getInputsName) .def_static("get_inputs_name", &Slice_Op::getInputsName)
.def_static("get_outputs_name", &Slice_Op::getOutputsName); .def_static("get_outputs_name", &Slice_Op::getOutputsName);
declare_registrable<Slice_Op>(m, "SliceOp"); declare_registrable<Slice_Op>(m, "SliceOp");
......
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