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

Hotfix to add default arguments to ReduceMean

parent d02fe9c2
No related branches found
No related tags found
1 merge request!212Version 0.3.0
Pipeline #53839 passed
...@@ -40,7 +40,7 @@ void declare_ReduceMeanOp(py::module &m) { ...@@ -40,7 +40,7 @@ void declare_ReduceMeanOp(py::module &m) {
if False, the operatpr reduces all the dimensions. if False, the operatpr reduces all the dimensions.
:type noop_with_empty_axes: bool :type noop_with_empty_axes: bool
)mydelimiter") )mydelimiter")
.def(py::init<std::vector<std::int32_t>, bool, bool>(), py::arg("axes"), py::arg("keep_dims"), py::arg("noop_with_empty_axes")) .def(py::init<std::vector<std::int32_t>, bool, bool>(), py::arg("axes") = std::vector<std::int32_t>(), py::arg("keep_dims") = true, py::arg("noop_with_empty_axes") = false)
.def_static("get_inputs_name", &ReduceMean_Op::getInputsName) .def_static("get_inputs_name", &ReduceMean_Op::getInputsName)
.def_static("get_outputs_name", &ReduceMean_Op::getOutputsName) .def_static("get_outputs_name", &ReduceMean_Op::getOutputsName)
; ;
......
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