Skip to content
Snippets Groups Projects
Commit fb8acb5b authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added missing binding

parent 2b1fbc89
No related branches found
No related tags found
2 merge requests!414Update version 0.5.1 -> 0.6.0,!408[Add] Dropout Operator
Pipeline #69335 passed
...@@ -59,6 +59,7 @@ void init_Scheduler(py::module& m){ ...@@ -59,6 +59,7 @@ void init_Scheduler(py::module& m){
py::class_<SequentialScheduler, std::shared_ptr<SequentialScheduler>, Scheduler>(m, "SequentialScheduler") py::class_<SequentialScheduler, std::shared_ptr<SequentialScheduler>, Scheduler>(m, "SequentialScheduler")
.def(py::init<std::shared_ptr<GraphView>&, bool>(), py::arg("graph_view"), py::arg("reset_cp_model") = true) .def(py::init<std::shared_ptr<GraphView>&, bool>(), py::arg("graph_view"), py::arg("reset_cp_model") = true)
.def("set_scheduling_policy", &SequentialScheduler::setSchedulingPolicy, py::arg("policy"))
.def("get_sequential_static_scheduling", &SequentialScheduler::getSequentialStaticScheduling, py::arg("step") = 0) .def("get_sequential_static_scheduling", &SequentialScheduler::getSequentialStaticScheduling, py::arg("step") = 0)
.def("generate_memory", &SequentialScheduler::generateMemory, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false) .def("generate_memory", &SequentialScheduler::generateMemory, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false)
.def("generate_memory_auto_concat", &SequentialScheduler::generateMemoryAutoConcat, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false) .def("generate_memory_auto_concat", &SequentialScheduler::generateMemoryAutoConcat, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false)
......
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