Skip to content
Snippets Groups Projects
Commit 7a1808be authored by Cyril Moineau's avatar Cyril Moineau
Browse files

[Update] Add new function to binding.

parent 352bdfb6
No related branches found
No related tags found
No related merge requests found
...@@ -51,9 +51,18 @@ void init_GraphView(py::module& m) { ...@@ -51,9 +51,18 @@ void init_GraphView(py::module& m) {
Include a Node to the current GraphView object. Include a Node to the current GraphView object.
:param other_node: Node to add :param other_node: Node to add
:type oth_Node: Node :type other_node: Node
:param includeLearnableParameter: include non-data inputs, like weights and biases. Default True. :param include_learnable_parameters: include non-data inputs, like weights and biases, default True.
:type includeLearnableParameter: bool :type include_learnable_parameters: bool, optional
)mydelimiter")
.def("add", (void (GraphView::*)(std::shared_ptr<GraphView>)) & GraphView::add,
py::arg("other_graph"),
R"mydelimiter(
Include a GraphView to the current GraphView object.
:param other_graph: GraphView to add
:type otherGraph: GraphView
)mydelimiter") )mydelimiter")
.def("add_child", .def("add_child",
...@@ -105,4 +114,4 @@ void init_GraphView(py::module& m) { ...@@ -105,4 +114,4 @@ void init_GraphView(py::module& m) {
// }) // })
; ;
} }
} // namespace Aidge } // namespace Aidge
\ No newline at end of file
...@@ -21,6 +21,8 @@ void init_Operator(py::module& m){ ...@@ -21,6 +21,8 @@ void init_Operator(py::module& m){
.def("output", &Operator::output, py::arg("outputIdx")) .def("output", &Operator::output, py::arg("outputIdx"))
.def("input", &Operator::input, py::arg("inputIdx")) .def("input", &Operator::input, py::arg("inputIdx"))
.def("nb_data_inputs", &Operator::nbDataInputs) .def("nb_data_inputs", &Operator::nbDataInputs)
.def("nb_outputs", &Operator::nbOutputs)
.def("output_dims_forwarded", &Operator::outputDimsForwarded)
.def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data")) .def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data"))
.def("set_datatype", &Operator::setDatatype, py::arg("datatype")) .def("set_datatype", &Operator::setDatatype, py::arg("datatype"))
.def("set_backend", &Operator::setBackend, py::arg("name")) .def("set_backend", &Operator::setBackend, py::arg("name"))
......
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