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) {
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");
......
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