ArmNN
NotReleased
SubgraphViewSelector.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#pragma once
6
7
#include "
SubgraphView.hpp
"
8
#include <functional>
9
#include <memory>
10
11
namespace
armnn
12
{
13
14
class
Layer;
15
class
Graph;
16
21
class
SubgraphViewSelector
final
22
{
23
public
:
24
using
SubgraphViewPtr
= std::unique_ptr<SubgraphView>;
25
using
Subgraphs
= std::vector<SubgraphViewPtr>;
26
using
LayerSelectorFunction
= std::function<bool(const Layer&)>;
27
32
static
Subgraphs
SelectSubgraphs
(
Graph
& graph,
const
LayerSelectorFunction
& selector);
33
static
Subgraphs
SelectSubgraphs
(
SubgraphView
& subgraph,
const
LayerSelectorFunction
& selector);
34
35
private
:
36
// this is a utility class, don't construct or copy
37
SubgraphViewSelector
() =
delete
;
38
SubgraphViewSelector
(
const
SubgraphViewSelector
&) =
delete
;
39
SubgraphViewSelector
& operator=(
const
SubgraphViewSelector
&) =
delete
;
40
};
41
42
}
// namespace armnn
SubgraphView.hpp
armnn::SubgraphViewSelector::SelectSubgraphs
static Subgraphs SelectSubgraphs(Graph &graph, const LayerSelectorFunction &selector)
Definition:
SubgraphViewSelector.cpp:251
armnn::SubgraphViewSelector::LayerSelectorFunction
std::function< bool(const Layer &)> LayerSelectorFunction
Definition:
SubgraphViewSelector.hpp:26
armnn::SubgraphViewSelector::Subgraphs
std::vector< SubgraphViewPtr > Subgraphs
Definition:
SubgraphViewSelector.hpp:25
armnn::SubgraphView
Definition:
SubgraphView.hpp:23
armnn
Definition:
BackendHelper.hpp:11
armnn::SubgraphViewSelector
Definition:
SubgraphViewSelector.hpp:21
armnn::Graph
Definition:
Graph.hpp:29
armnn::SubgraphViewSelector::SubgraphViewPtr
std::unique_ptr< SubgraphView > SubgraphViewPtr
Definition:
SubgraphViewSelector.hpp:24
src
armnn
SubgraphViewSelector.hpp
Generated on Fri Mar 13 2020 16:06:54 for ArmNN by
1.8.13