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

add graphviewhelper python binding

parent 52a4d93d
No related branches found
No related tags found
No related merge requests found
......@@ -28,6 +28,8 @@ namespace Aidge {
*/
std::set<std::shared_ptr<Tensor>> producers(std::shared_ptr<GraphView> graphview);
// TODO: change for every Tensor of Operator Producer not constant
/**
* @brief Getter for every ``Tensor`` owned by an ``Operator`` inside the provided ``GraphView``.
* @note An ``Operator`` owns its output ``Tensor``s.
......
......@@ -68,6 +68,7 @@ void init_GraphRegex(py::module&);
void init_MatchSolution(py::module&);
void init_Recipes(py::module&);
void init_GraphViewHelper(py::module&);
void init_Scheduler(py::module&);
void init_TensorUtils(py::module&);
......@@ -129,6 +130,7 @@ void init_Aidge(py::module& m) {
init_MatchSolution(m);
init_Recipes(m);
init_GraphViewHelper(m);
init_Scheduler(m);
init_TensorUtils(m);
init_Filler(m);
......
/********************************************************************************
* Copyright (c) 2023 CEA-List
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* SPDX-License-Identifier: EPL-2.0
*
********************************************************************************/
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <memory>
#include <set>
#include "aidge/graph/GraphView.hpp"
#include "aidge/data/Tensor.hpp"
#include "aidge/recipes/GraphViewHelper.hpp"
namespace py = pybind11;
namespace Aidge {
void init_GraphViewHelper(py::module &m) {
m.def("producers", &producers, py::arg("graphview"));
}
} // namespace Aidge
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