diff options
Diffstat (limited to 'src/armnn/SubGraphSelector.hpp')
-rw-r--r-- | src/armnn/SubGraphSelector.hpp | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/src/armnn/SubGraphSelector.hpp b/src/armnn/SubGraphSelector.hpp index f96b986a2e..371ba78f18 100644 --- a/src/armnn/SubGraphSelector.hpp +++ b/src/armnn/SubGraphSelector.hpp @@ -21,14 +21,12 @@ public: using SubGraphs = std::vector<SubGraphPtr>; using LayerSelectorFunction = std::function<bool(const Layer&)>; - /// Selects subgraphs of a graph based on the selector function - /// and the algorithm. Since the SubGraphs returns modifiable pointers - /// the input and output slots of the graph: - /// 1) the graph cannot be const - /// 2) the caller need to make sure that the SubGraphs lifetime is - /// shorter than the graph's - static SubGraphs SelectSubGraphs(Graph& graph, - const LayerSelectorFunction& selector); + /// Selects subgraphs from a graph based on the selector function and the algorithm. + /// Since the SubGraphs object returns modifiable pointers to the input and output slots of the graph: + /// 1) the graph/sub-graph cannot be const + /// 2) the caller needs to make sure that the SubGraphs lifetime is shorter than the parent graph's + static SubGraphs SelectSubGraphs(Graph& graph, const LayerSelectorFunction& selector); + static SubGraphs SelectSubGraphs(SubGraph& subGraph, const LayerSelectorFunction& selector); private: // this is a utility class, don't construct or copy |