1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
<!-- Copyright (c) 2020 ARM Limited. -->
<!-- -->
<!-- SPDX-License-Identifier: MIT -->
<!-- -->
<!-- HTML header for doxygen 1.8.13-->
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<meta name="generator" content="Doxygen 1.8.13"/>
<meta name="robots" content="NOINDEX, NOFOLLOW" />
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<title>ArmNN: src/armnn/SubgraphViewSelector.cpp Source File</title>
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="jquery.js"></script>
<script type="text/javascript" src="dynsections.js"></script>
<link href="navtree.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="resize.js"></script>
<script type="text/javascript" src="navtreedata.js"></script>
<script type="text/javascript" src="navtree.js"></script>
<script type="text/javascript">
$(document).ready(initResizable);
</script>
<link href="search/search.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="search/searchdata.js"></script>
<script type="text/javascript" src="search/search.js"></script>
<script type="text/x-mathjax-config">
MathJax.Hub.Config({
extensions: ["tex2jax.js"],
jax: ["input/TeX","output/HTML-CSS"],
});
</script><script type="text/javascript" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
<link href="stylesheet.css" rel="stylesheet" type="text/css"/>
</head>
<body>
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
<div id="titlearea">
<table cellspacing="0" cellpadding="0">
<tbody>
<tr style="height: 56px;">
<img alt="ArmNN" src="Arm_NN_horizontal_blue.png" style="max-width: 10rem; margin-top: .5rem; margin-left 10px"/>
<td style="padding-left: 0.5em;">
<div id="projectname">
 <span id="projectnumber">21.02</span>
</div>
</td>
</tr>
</tbody>
</table>
</div>
<!-- end header part -->
<!-- Generated by Doxygen 1.8.13 -->
<script type="text/javascript">
var searchBox = new SearchBox("searchBox", "search",false,'Search');
</script>
<script type="text/javascript" src="menudata.js"></script>
<script type="text/javascript" src="menu.js"></script>
<script type="text/javascript">
$(function() {
initMenu('',true,false,'search.php','Search');
$(document).ready(function() { init_search(); });
});
</script>
<div id="main-nav"></div>
</div><!-- top -->
<div id="side-nav" class="ui-resizable side-nav-resizable">
<div id="nav-tree">
<div id="nav-tree-contents">
<div id="nav-sync" class="sync"></div>
</div>
</div>
<div id="splitbar" style="-moz-user-select:none;"
class="ui-resizable-handle">
</div>
</div>
<script type="text/javascript">
$(document).ready(function(){initNavTree('_subgraph_view_selector_8cpp_source.xhtml','');});
</script>
<div id="doc-content">
<!-- window showing the filter options -->
<div id="MSearchSelectWindow"
onmouseover="return searchBox.OnSearchSelectShow()"
onmouseout="return searchBox.OnSearchSelectHide()"
onkeydown="return searchBox.OnSearchSelectKey(event)">
</div>
<!-- iframe showing the search results (closed by default) -->
<div id="MSearchResultsWindow">
<iframe src="javascript:void(0)" frameborder="0"
name="MSearchResults" id="MSearchResults">
</iframe>
</div>
<div class="header">
<div class="headertitle">
<div class="title">SubgraphViewSelector.cpp</div> </div>
</div><!--header-->
<div class="contents">
<a href="_subgraph_view_selector_8cpp.xhtml">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">//</span></div><div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment">// Copyright © 2017 Arm Ltd. All rights reserved.</span></div><div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment">// SPDX-License-Identifier: MIT</span></div><div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment">//</span></div><div class="line"><a name="l00005"></a><span class="lineno"> 5</span> </div><div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="preprocessor">#include "<a class="code" href="_subgraph_view_selector_8hpp.xhtml">SubgraphViewSelector.hpp</a>"</span></div><div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="preprocessor">#include "<a class="code" href="_graph_8hpp.xhtml">Graph.hpp</a>"</span></div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span> </div><div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="preprocessor">#include <<a class="code" href="_assert_8hpp.xhtml">armnn/utility/Assert.hpp</a>></span></div><div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="preprocessor">#include <<a class="code" href="_ignore_unused_8hpp.xhtml">armnn/utility/IgnoreUnused.hpp</a>></span></div><div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="preprocessor">#include <<a class="code" href="_polymorphic_downcast_8hpp.xhtml">armnn/utility/PolymorphicDowncast.hpp</a>></span></div><div class="line"><a name="l00012"></a><span class="lineno"> 12</span> </div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="preprocessor">#include <algorithm></span></div><div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="preprocessor">#include <map></span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="preprocessor">#include <queue></span></div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="preprocessor">#include <unordered_set></span></div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span> </div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="keyword">namespace </span><a class="code" href="namespacearmnn.xhtml">armnn</a></div><div class="line"><a name="l00019"></a><span class="lineno"> 19</span> {</div><div class="line"><a name="l00020"></a><span class="lineno"> 20</span> </div><div class="line"><a name="l00021"></a><span class="lineno"> 21</span> <span class="keyword">namespace</span></div><div class="line"><a name="l00022"></a><span class="lineno"> 22</span> {</div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span> <span class="comment"></span></div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span> <span class="comment">/// Intermediate data-structure to store the subgraph that a layer has been assigned to.</span></div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span> <span class="comment">/// This is a "disjoint set" data structure that allows efficient merging of subgraphs,</span></div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span> <span class="comment">/// which is a key part of the algorithm. Subgraphs are arranged in singly-linked trees</span></div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="comment">/// (with each node storing a pointer to its parent). Subgraphs in the same tree are considered</span></div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span> <span class="comment">/// to have been merged. Merging subgraphs is performed by attaching one tree to another,</span></div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="comment">/// which is a simple pointer update.</span></div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="comment">///</span></div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="comment">/// NOTE: Due to the way this is stored, it is almost never correct to directly compare pointers</span></div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="comment">/// to two PartialSubgraphs to check if two layers belong in the same subgraph. Instead you</span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="comment">/// should use IsMergedWith().</span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="comment">///</span></div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="comment">/// This structure also stores information about the dependencies of each subgraph, which is needed</span></div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="comment">/// to determine whether certain subgraphs can be merged. Checking whether a subgraph</span></div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="comment">/// depends on another subgraph is a frequent operation in the algorithm (see AssignSplitId) and so this is optimized</span></div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="comment">/// in preference to the merging of subgraphs. This leads to an approach where each subgraph stores</span></div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="comment">/// a set of all the subgraphs it depends on (for a fast lookup). In order to efficiently update this</span></div><div class="line"><a name="l00040"></a><span class="lineno"> 40</span> <span class="comment">/// set as subgraphs are merged means we also store a set of subgraphs which *depend on us* (i.e. the</span></div><div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="comment">/// complement of our dependencies).</span></div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span> <span class="comment"></span><span class="keyword">class </span>PartialSubgraph</div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span> {</div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span> <span class="keyword">public</span>:<span class="comment"></span></div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span> <span class="comment"> /// If this subgraph has been merged with another then there is an agreed "representative" for the combined</span></div><div class="line"><a name="l00046"></a><span class="lineno"> 46</span> <span class="comment"> /// subgraph, which uniquely identifies the subgraph.</span></div><div class="line"><a name="l00047"></a><span class="lineno"> 47</span> <span class="comment"></span> PartialSubgraph* GetRepresentative()</div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span>  {</div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span>  <span class="comment">// Recurse up the tree to find the root node.</span></div><div class="line"><a name="l00050"></a><span class="lineno"> 50</span>  <span class="keywordflow">if</span> (m_Parent == <span class="keyword">nullptr</span>)</div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>  {</div><div class="line"><a name="l00052"></a><span class="lineno"> 52</span>  <span class="keywordflow">return</span> <span class="keyword">this</span>;</div><div class="line"><a name="l00053"></a><span class="lineno"> 53</span>  }</div><div class="line"><a name="l00054"></a><span class="lineno"> 54</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00055"></a><span class="lineno"> 55</span>  {</div><div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  PartialSubgraph* result = m_Parent->GetRepresentative();</div><div class="line"><a name="l00057"></a><span class="lineno"> 57</span>  <span class="comment">// Update our parent pointer to point directly to the root in order to speed up future calls to this method.</span></div><div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  <span class="comment">// This essentially "flattens" the tree.</span></div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  m_Parent = result;</div><div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  <span class="keywordflow">return</span> result;</div><div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  }</div><div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  }</div><div class="line"><a name="l00063"></a><span class="lineno"> 63</span> <span class="comment"></span></div><div class="line"><a name="l00064"></a><span class="lineno"> 64</span> <span class="comment"> /// Merges this subgraph with another.</span></div><div class="line"><a name="l00065"></a><span class="lineno"> 65</span> <span class="comment"></span> <span class="keywordtype">void</span> MergeWith(PartialSubgraph* other)</div><div class="line"><a name="l00066"></a><span class="lineno"> 66</span>  {</div><div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  <span class="keywordflow">if</span> (m_Parent == <span class="keyword">nullptr</span>)</div><div class="line"><a name="l00068"></a><span class="lineno"> 68</span>  {</div><div class="line"><a name="l00069"></a><span class="lineno"> 69</span>  other = other->GetRepresentative();</div><div class="line"><a name="l00070"></a><span class="lineno"> 70</span>  <span class="keywordflow">if</span> (<span class="keyword">this</span> == other)</div><div class="line"><a name="l00071"></a><span class="lineno"> 71</span>  {</div><div class="line"><a name="l00072"></a><span class="lineno"> 72</span>  <span class="comment">// Already merged - no-op</span></div><div class="line"><a name="l00073"></a><span class="lineno"> 73</span>  <span class="keywordflow">return</span>;</div><div class="line"><a name="l00074"></a><span class="lineno"> 74</span>  }</div><div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  m_Parent = other;</div><div class="line"><a name="l00076"></a><span class="lineno"> 76</span> </div><div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  <span class="comment">// Update others' dependency sets to point to the new representative rather than us.</span></div><div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  <span class="comment">// Keeping these up-to-date means we can rely on these sets containing representatives when</span></div><div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  <span class="comment">// we perform a lookup in HasAntecedent() and so don't need to resolve the representative for each element</span></div><div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  <span class="comment">// of the set. See description at the top of this class for more rationale.</span></div><div class="line"><a name="l00081"></a><span class="lineno"> 81</span>  <span class="keywordflow">for</span> (PartialSubgraph* a : m_Antecedents)</div><div class="line"><a name="l00082"></a><span class="lineno"> 82</span>  {</div><div class="line"><a name="l00083"></a><span class="lineno"> 83</span>  <span class="keywordtype">size_t</span> numErased = a->m_Dependants.erase(<span class="keyword">this</span>);</div><div class="line"><a name="l00084"></a><span class="lineno"> 84</span>  <a class="code" href="_assert_8hpp.xhtml#a5698be69cbd5dfe6c28fcd9867e8cbed">ARMNN_ASSERT</a>(numErased == 1);</div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  <a class="code" href="namespacearmnn.xhtml#a44affeeb090c3c6a3062830562672e84">IgnoreUnused</a>(numErased);</div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  a->m_Dependants.insert(m_Parent);</div><div class="line"><a name="l00087"></a><span class="lineno"> 87</span>  }</div><div class="line"><a name="l00088"></a><span class="lineno"> 88</span>  <span class="keywordflow">for</span> (PartialSubgraph* a : m_Dependants)</div><div class="line"><a name="l00089"></a><span class="lineno"> 89</span>  {</div><div class="line"><a name="l00090"></a><span class="lineno"> 90</span>  <span class="keywordtype">size_t</span> numErased = a->m_Antecedents.erase(<span class="keyword">this</span>);</div><div class="line"><a name="l00091"></a><span class="lineno"> 91</span>  <a class="code" href="_assert_8hpp.xhtml#a5698be69cbd5dfe6c28fcd9867e8cbed">ARMNN_ASSERT</a>(numErased == 1);</div><div class="line"><a name="l00092"></a><span class="lineno"> 92</span>  <a class="code" href="namespacearmnn.xhtml#a44affeeb090c3c6a3062830562672e84">IgnoreUnused</a>(numErased);</div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  a->m_Antecedents.insert(m_Parent);</div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span>  }</div><div class="line"><a name="l00095"></a><span class="lineno"> 95</span> </div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span>  <span class="comment">// Merge our dependency sets into our new representative.</span></div><div class="line"><a name="l00097"></a><span class="lineno"> 97</span>  <span class="comment">// We no longer need to maintain our own sets, as requests will always be forwarded to the representative.</span></div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  m_Parent->m_Antecedents.insert(m_Antecedents.begin(), m_Antecedents.end());</div><div class="line"><a name="l00099"></a><span class="lineno"> 99</span>  m_Antecedents.clear();</div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  m_Parent->m_Dependants.insert(m_Dependants.begin(), m_Dependants.end());</div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span>  m_Dependants.clear();</div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  }</div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>  {</div><div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  <span class="comment">// Defer request to the representative</span></div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>  GetRepresentative()->MergeWith(other);</div><div class="line"><a name="l00107"></a><span class="lineno"> 107</span>  }</div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span>  }</div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span> <span class="comment"></span></div><div class="line"><a name="l00110"></a><span class="lineno"> 110</span> <span class="comment"> /// Checks if this subgraph has been merged with the given subgraph.</span></div><div class="line"><a name="l00111"></a><span class="lineno"> 111</span> <span class="comment"></span> <span class="keywordtype">bool</span> IsMergedWith(PartialSubgraph* other)</div><div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  {</div><div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  <span class="keywordflow">return</span> GetRepresentative() == other->GetRepresentative();</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  }</div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span> <span class="comment"></span></div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span> <span class="comment"> /// Marks the given subgraph as a direct antecedent (dependency) of this one.</span></div><div class="line"><a name="l00117"></a><span class="lineno"> 117</span> <span class="comment"></span> <span class="keywordtype">void</span> AddDirectAntecedent(PartialSubgraph* antecedent)</div><div class="line"><a name="l00118"></a><span class="lineno"> 118</span>  {</div><div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <span class="keywordflow">if</span> (m_Parent == <span class="keyword">nullptr</span>)</div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  {</div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  antecedent = antecedent->GetRepresentative();</div><div class="line"><a name="l00122"></a><span class="lineno"> 122</span> </div><div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  m_Antecedents.insert(antecedent);</div><div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  <span class="comment">// Also record all of its antecedents, so that we end up with direct and indirect antecedents.</span></div><div class="line"><a name="l00125"></a><span class="lineno"> 125</span>  <span class="comment">// This makes the lookup in HasAntecedent() faster.</span></div><div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  m_Antecedents.insert(antecedent->m_Antecedents.begin(), antecedent->m_Antecedents.end());</div><div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  <span class="comment">// All of our dependents also need to include the new antecedents</span></div><div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  <span class="keywordflow">for</span> (PartialSubgraph* d : m_Dependants)</div><div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  {</div><div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  d->m_Antecedents.insert(antecedent);</div><div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  d->m_Antecedents.insert(antecedent->m_Antecedents.begin(), antecedent->m_Antecedents.end());</div><div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  }</div><div class="line"><a name="l00133"></a><span class="lineno"> 133</span> </div><div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  <span class="comment">// Store reverse dependencies as well, required so that we can efficiently navigate the graph</span></div><div class="line"><a name="l00135"></a><span class="lineno"> 135</span>  <span class="comment">// when making updates.</span></div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  antecedent->m_Dependants.insert(<span class="keyword">this</span>);</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  antecedent->m_Dependants.insert(m_Dependants.begin(), m_Dependants.end());</div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  <span class="keywordflow">for</span> (PartialSubgraph* a : antecedent->m_Antecedents)</div><div class="line"><a name="l00139"></a><span class="lineno"> 139</span>  {</div><div class="line"><a name="l00140"></a><span class="lineno"> 140</span>  a->m_Dependants.insert(<span class="keyword">this</span>);</div><div class="line"><a name="l00141"></a><span class="lineno"> 141</span>  a->m_Dependants.insert(m_Dependants.begin(), m_Dependants.end());</div><div class="line"><a name="l00142"></a><span class="lineno"> 142</span>  }</div><div class="line"><a name="l00143"></a><span class="lineno"> 143</span>  }</div><div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00145"></a><span class="lineno"> 145</span>  {</div><div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  <span class="comment">// Defer request to the representative</span></div><div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  GetRepresentative()->AddDirectAntecedent(antecedent);</div><div class="line"><a name="l00148"></a><span class="lineno"> 148</span>  }</div><div class="line"><a name="l00149"></a><span class="lineno"> 149</span>  }</div><div class="line"><a name="l00150"></a><span class="lineno"> 150</span> <span class="comment"></span></div><div class="line"><a name="l00151"></a><span class="lineno"> 151</span> <span class="comment"> /// Checks if this subgraph is dependent on the given subgraph, either directly or indirectly.</span></div><div class="line"><a name="l00152"></a><span class="lineno"> 152</span> <span class="comment"></span> <span class="keywordtype">bool</span> HasAntecedent(PartialSubgraph* antecedent)</div><div class="line"><a name="l00153"></a><span class="lineno"> 153</span>  {</div><div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  <span class="keywordflow">if</span> (m_Parent == <span class="keyword">nullptr</span>)</div><div class="line"><a name="l00155"></a><span class="lineno"> 155</span>  {</div><div class="line"><a name="l00156"></a><span class="lineno"> 156</span>  antecedent = antecedent->GetRepresentative();</div><div class="line"><a name="l00157"></a><span class="lineno"> 157</span>  <span class="comment">// Thanks to keeping this set updated in MergeWith and AddDirectAntecedent, we can do an efficient lookup.</span></div><div class="line"><a name="l00158"></a><span class="lineno"> 158</span>  <span class="keywordflow">return</span> m_Antecedents.count(antecedent) > 0;</div><div class="line"><a name="l00159"></a><span class="lineno"> 159</span>  }</div><div class="line"><a name="l00160"></a><span class="lineno"> 160</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00161"></a><span class="lineno"> 161</span>  {</div><div class="line"><a name="l00162"></a><span class="lineno"> 162</span>  <span class="comment">// Defer request to the representative</span></div><div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  <span class="keywordflow">return</span> GetRepresentative()->HasAntecedent(antecedent);</div><div class="line"><a name="l00164"></a><span class="lineno"> 164</span>  }</div><div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  }</div><div class="line"><a name="l00166"></a><span class="lineno"> 166</span> </div><div class="line"><a name="l00167"></a><span class="lineno"> 167</span> <span class="keyword">private</span>:<span class="comment"></span></div><div class="line"><a name="l00168"></a><span class="lineno"> 168</span> <span class="comment"> /// Pointer to the parent node in the tree. If this is null then we are the representative for our merged subgraph.</span></div><div class="line"><a name="l00169"></a><span class="lineno"> 169</span> <span class="comment"></span> PartialSubgraph* m_Parent;<span class="comment"></span></div><div class="line"><a name="l00170"></a><span class="lineno"> 170</span> <span class="comment"> /// The representatives of all the subgraphs which we depend on, either directly or indirectly.</span></div><div class="line"><a name="l00171"></a><span class="lineno"> 171</span> <span class="comment"></span> std::unordered_set<PartialSubgraph*> m_Antecedents;<span class="comment"></span></div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span> <span class="comment"> /// The representatives of all the subgraphs which depend on us, either directly or indirectly.</span></div><div class="line"><a name="l00173"></a><span class="lineno"> 173</span> <span class="comment"></span> std::unordered_set<PartialSubgraph*> m_Dependants;</div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span> };</div><div class="line"><a name="l00175"></a><span class="lineno"> 175</span> <span class="comment"></span></div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span> <span class="comment">/// Intermediate data structure to store information associated with a particular layer.</span></div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span> <span class="comment"></span><span class="keyword">struct </span>LayerSelectionInfo</div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span> {</div><div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  <span class="keyword">using</span> LayerInfoContainer = std::map<Layer*, LayerSelectionInfo>;</div><div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  <span class="keyword">using</span> LayerInfoQueue = std::queue<LayerSelectionInfo*>;</div><div class="line"><a name="l00181"></a><span class="lineno"> 181</span> </div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span>  LayerSelectionInfo(<a class="code" href="namespacearmnn_serializer.xhtml#a9a8118be7780e95363d631cbca7e7800">Layer</a>* layer, <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#a341a1b47a5f71edaebf81f3a5f0f4c55">SubgraphViewSelector::LayerSelectorFunction</a>& selector)</div><div class="line"><a name="l00183"></a><span class="lineno"> 183</span>  : <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>{layer}</div><div class="line"><a name="l00184"></a><span class="lineno"> 184</span>  , <a class="code" href="_subgraph_view_selector_8cpp.xhtml#a3a672da6a5207243cf3e5c933872b5bf">m_Subgraph</a>{<span class="keyword">nullptr</span>}</div><div class="line"><a name="l00185"></a><span class="lineno"> 185</span>  , <a class="code" href="_subgraph_view_selector_8cpp.xhtml#aa479d97c404e5996e29b009c60b22e8b">m_IsSelected</a>{selector(*layer)}</div><div class="line"><a name="l00186"></a><span class="lineno"> 186</span>  , <a class="code" href="_subgraph_view_selector_8cpp.xhtml#a849c95a020d031ec405d7d21021d87c2">m_IsProcessed</a>(<span class="keyword">false</span>)</div><div class="line"><a name="l00187"></a><span class="lineno"> 187</span>  {</div><div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  }</div><div class="line"><a name="l00189"></a><span class="lineno"> 189</span> </div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  <span class="keywordtype">bool</span> IsInputLayer()<span class="keyword"> const</span></div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span> <span class="keyword"> </span>{</div><div class="line"><a name="l00192"></a><span class="lineno"> 192</span>  <span class="keywordflow">return</span> <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>->GetType() == <a class="code" href="namespacearmnn.xhtml#a56943a0946e5f15e5e58054b8e7a04a4a324118a6721dd6b8a9b9f4e327df2bf5">armnn::LayerType::Input</a> || <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>->GetType() == <a class="code" href="namespacearmnn.xhtml#a56943a0946e5f15e5e58054b8e7a04a4acb17869fe51048b5a5c4c6106551a255">armnn::LayerType::Constant</a>;</div><div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  }</div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span> </div><div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  <span class="keywordtype">void</span> CollectNonSelectedInputs(LayerSelectionInfo::LayerInfoContainer& layerInfos,</div><div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  <a class="code" href="classarmnn_1_1_subgraph_view.xhtml#a5cc65e15002dbc33a5c8a7d6680e9a9d">SubgraphView::InputSlots</a>& inputSlots)</div><div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  {</div><div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>&& slot = <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>->BeginInputSlots(); slot != <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>->EndInputSlots(); ++slot)</div><div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  {</div><div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  OutputSlot* parentLayerOutputSlot = slot->GetConnectedOutputSlot();</div><div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  <a class="code" href="_assert_8hpp.xhtml#a91c4dfde57907d7698c7531785690a7f">ARMNN_ASSERT_MSG</a>(parentLayerOutputSlot != <span class="keyword">nullptr</span>, <span class="stringliteral">"The input slots must be connected here."</span>);</div><div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  <span class="keywordflow">if</span> (parentLayerOutputSlot)</div><div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  {</div><div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  <a class="code" href="namespacearmnn_serializer.xhtml#a9a8118be7780e95363d631cbca7e7800">Layer</a>& parentLayer = parentLayerOutputSlot->GetOwningLayer();</div><div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  <span class="keyword">auto</span> parentInfo = layerInfos.find(&parentLayer);</div><div class="line"><a name="l00206"></a><span class="lineno"> 206</span>  <span class="keywordflow">if</span> (parentInfo == layerInfos.end() ||</div><div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  !<a class="code" href="_subgraph_view_selector_8cpp.xhtml#a3a672da6a5207243cf3e5c933872b5bf">m_Subgraph</a>->IsMergedWith(parentInfo->second.m_Subgraph.get()))</div><div class="line"><a name="l00208"></a><span class="lineno"> 208</span>  {</div><div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  <span class="comment">// Avoid collecting duplicate input slots</span></div><div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  InputSlot* inputSlot = &(*slot);</div><div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  <span class="keywordflow">if</span> (std::find(inputSlots.begin(), inputSlots.end(), inputSlot) == inputSlots.end())</div><div class="line"><a name="l00212"></a><span class="lineno"> 212</span>  {</div><div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  inputSlots.push_back(inputSlot);</div><div class="line"><a name="l00214"></a><span class="lineno"> 214</span>  }</div><div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  }</div><div class="line"><a name="l00216"></a><span class="lineno"> 216</span>  }</div><div class="line"><a name="l00217"></a><span class="lineno"> 217</span>  }</div><div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  }</div><div class="line"><a name="l00219"></a><span class="lineno"> 219</span> </div><div class="line"><a name="l00220"></a><span class="lineno"> 220</span>  <span class="keywordtype">void</span> CollectNonSelectedOutputSlots(LayerSelectionInfo::LayerInfoContainer& layerInfos,</div><div class="line"><a name="l00221"></a><span class="lineno"> 221</span>  <a class="code" href="classarmnn_1_1_subgraph_view.xhtml#a78293334750ec5279eb9c96d56deaf08">SubgraphView::OutputSlots</a>& outputSlots)</div><div class="line"><a name="l00222"></a><span class="lineno"> 222</span>  {</div><div class="line"><a name="l00223"></a><span class="lineno"> 223</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>&& slot = <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>->BeginOutputSlots(); slot != <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>->EndOutputSlots(); ++slot)</div><div class="line"><a name="l00224"></a><span class="lineno"> 224</span>  {</div><div class="line"><a name="l00225"></a><span class="lineno"> 225</span>  <span class="keywordflow">for</span> (InputSlot* childLayerInputSlot : slot->GetConnections())</div><div class="line"><a name="l00226"></a><span class="lineno"> 226</span>  {</div><div class="line"><a name="l00227"></a><span class="lineno"> 227</span>  <a class="code" href="namespacearmnn_serializer.xhtml#a9a8118be7780e95363d631cbca7e7800">Layer</a>& childLayer = childLayerInputSlot->GetOwningLayer();</div><div class="line"><a name="l00228"></a><span class="lineno"> 228</span>  <span class="keyword">auto</span> childInfo = layerInfos.find(&childLayer);</div><div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  <span class="keywordflow">if</span> (childInfo == layerInfos.end() ||</div><div class="line"><a name="l00230"></a><span class="lineno"> 230</span>  !<a class="code" href="_subgraph_view_selector_8cpp.xhtml#a3a672da6a5207243cf3e5c933872b5bf">m_Subgraph</a>->IsMergedWith(childInfo->second.m_Subgraph.get()))</div><div class="line"><a name="l00231"></a><span class="lineno"> 231</span>  {</div><div class="line"><a name="l00232"></a><span class="lineno"> 232</span>  <span class="comment">// Avoid collecting duplicate output slots</span></div><div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  OutputSlot* outputSlot = &(*slot);</div><div class="line"><a name="l00234"></a><span class="lineno"> 234</span>  <span class="keywordflow">if</span> (std::find(outputSlots.begin(), outputSlots.end(), outputSlot) == outputSlots.end())</div><div class="line"><a name="l00235"></a><span class="lineno"> 235</span>  {</div><div class="line"><a name="l00236"></a><span class="lineno"> 236</span>  outputSlots.push_back(outputSlot);</div><div class="line"><a name="l00237"></a><span class="lineno"> 237</span>  }</div><div class="line"><a name="l00238"></a><span class="lineno"> 238</span>  }</div><div class="line"><a name="l00239"></a><span class="lineno"> 239</span>  }</div><div class="line"><a name="l00240"></a><span class="lineno"> 240</span>  }</div><div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  }</div><div class="line"><a name="l00242"></a><span class="lineno"> 242</span> </div><div class="line"><a name="l00243"></a><span class="lineno"><a class="line" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7"> 243</a></span>  <a class="code" href="namespacearmnn_serializer.xhtml#a9a8118be7780e95363d631cbca7e7800">Layer</a>* <a class="code" href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a>;<span class="comment"></span></div><div class="line"><a name="l00244"></a><span class="lineno"> 244</span> <span class="comment"> /// Which subgraph this layer has been assigned to. Only valid once m_IsProcessed is true.</span></div><div class="line"><a name="l00245"></a><span class="lineno"> 245</span> <span class="comment"> /// Two layers with different m_Subgraph pointers may in fact have been merged into the same subgraph -</span></div><div class="line"><a name="l00246"></a><span class="lineno"> 246</span> <span class="comment"> /// see the description of the PartialSubgraph class.</span></div><div class="line"><a name="l00247"></a><span class="lineno"><a class="line" href="_subgraph_view_selector_8cpp.xhtml#a3a672da6a5207243cf3e5c933872b5bf"> 247</a></span> <span class="comment"></span> std::shared_ptr<PartialSubgraph> <a class="code" href="_subgraph_view_selector_8cpp.xhtml#a3a672da6a5207243cf3e5c933872b5bf">m_Subgraph</a>;</div><div class="line"><a name="l00248"></a><span class="lineno"><a class="line" href="_subgraph_view_selector_8cpp.xhtml#aa479d97c404e5996e29b009c60b22e8b"> 248</a></span>  <span class="keywordtype">bool</span> <a class="code" href="_subgraph_view_selector_8cpp.xhtml#aa479d97c404e5996e29b009c60b22e8b">m_IsSelected</a>;</div><div class="line"><a name="l00249"></a><span class="lineno"><a class="line" href="_subgraph_view_selector_8cpp.xhtml#a849c95a020d031ec405d7d21021d87c2"> 249</a></span>  <span class="keywordtype">bool</span> <a class="code" href="_subgraph_view_selector_8cpp.xhtml#a849c95a020d031ec405d7d21021d87c2">m_IsProcessed</a>;</div><div class="line"><a name="l00250"></a><span class="lineno"> 250</span> };</div><div class="line"><a name="l00251"></a><span class="lineno"> 251</span> </div><div class="line"><a name="l00252"></a><span class="lineno"> 252</span> } <span class="comment">// namespace <anonymous></span></div><div class="line"><a name="l00253"></a><span class="lineno"> 253</span> </div><div class="line"><a name="l00254"></a><span class="lineno"> 254</span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#aaf71a63dbbc776f8961b0f4fdb9da021">SubgraphViewSelector::Subgraphs</a></div><div class="line"><a name="l00255"></a><span class="lineno"><a class="line" href="classarmnn_1_1_subgraph_view_selector.xhtml#a3730b0a6006f0d87f894a44e01869d90"> 255</a></span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#a3730b0a6006f0d87f894a44e01869d90">SubgraphViewSelector::SelectSubgraphs</a>(<a class="code" href="classarmnn_1_1_graph.xhtml">Graph</a>& graph, <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#a341a1b47a5f71edaebf81f3a5f0f4c55">LayerSelectorFunction</a>& selector)</div><div class="line"><a name="l00256"></a><span class="lineno"> 256</span> {</div><div class="line"><a name="l00257"></a><span class="lineno"> 257</span>  <a class="code" href="classarmnn_1_1_subgraph_view.xhtml">SubgraphView</a> subgraph(graph);</div><div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  <span class="keywordflow">return</span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#a3730b0a6006f0d87f894a44e01869d90">SubgraphViewSelector::SelectSubgraphs</a>(subgraph, selector);</div><div class="line"><a name="l00259"></a><span class="lineno"> 259</span> }</div><div class="line"><a name="l00260"></a><span class="lineno"> 260</span> </div><div class="line"><a name="l00261"></a><span class="lineno"> 261</span> </div><div class="line"><a name="l00262"></a><span class="lineno"> 262</span> <span class="keyword">template</span><<span class="keyword">typename</span> Delegate></div><div class="line"><a name="l00263"></a><span class="lineno"><a class="line" href="namespacearmnn.xhtml#afce94270d9c4a51cd0c4ac6a58af4e26"> 263</a></span> <span class="keywordtype">void</span> <a class="code" href="namespacearmnn.xhtml#afce94270d9c4a51cd0c4ac6a58af4e26">ForEachLayerInput</a>(LayerSelectionInfo::LayerInfoContainer& layerInfos,</div><div class="line"><a name="l00264"></a><span class="lineno"> 264</span>  LayerSelectionInfo& layerInfo,</div><div class="line"><a name="l00265"></a><span class="lineno"> 265</span>  Delegate <span class="keyword">function</span>)</div><div class="line"><a name="l00266"></a><span class="lineno"> 266</span> {</div><div class="line"><a name="l00267"></a><span class="lineno"> 267</span>  <a class="code" href="classarmnn_1_1_layer.xhtml">Layer</a>& layer = *layerInfo.m_Layer;</div><div class="line"><a name="l00268"></a><span class="lineno"> 268</span> </div><div class="line"><a name="l00269"></a><span class="lineno"> 269</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span> inputSlot : layer.<a class="code" href="classarmnn_1_1_layer.xhtml#af5f530544d09a44d726f24702b67b35b">GetInputSlots</a>())</div><div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  {</div><div class="line"><a name="l00271"></a><span class="lineno"> 271</span>  <span class="keyword">auto</span> connectedInput = PolymorphicDowncast<OutputSlot*>(inputSlot.GetConnection());</div><div class="line"><a name="l00272"></a><span class="lineno"> 272</span>  <a class="code" href="_assert_8hpp.xhtml#a91c4dfde57907d7698c7531785690a7f">ARMNN_ASSERT_MSG</a>(connectedInput, <span class="stringliteral">"Dangling input slot detected."</span>);</div><div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  <a class="code" href="classarmnn_1_1_layer.xhtml">Layer</a>& inputLayer = connectedInput->GetOwningLayer();</div><div class="line"><a name="l00274"></a><span class="lineno"> 274</span> </div><div class="line"><a name="l00275"></a><span class="lineno"> 275</span>  <span class="keyword">auto</span> parentInfo = layerInfos.find(&inputLayer);</div><div class="line"><a name="l00276"></a><span class="lineno"> 276</span>  <span class="keywordflow">if</span> (parentInfo != layerInfos.end())</div><div class="line"><a name="l00277"></a><span class="lineno"> 277</span>  {</div><div class="line"><a name="l00278"></a><span class="lineno"> 278</span>  <span class="keyword">function</span>(parentInfo->second);</div><div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  }</div><div class="line"><a name="l00280"></a><span class="lineno"> 280</span>  }</div><div class="line"><a name="l00281"></a><span class="lineno"> 281</span> }</div><div class="line"><a name="l00282"></a><span class="lineno"> 282</span> </div><div class="line"><a name="l00283"></a><span class="lineno"> 283</span> <span class="keyword">template</span><<span class="keyword">typename</span> Delegate></div><div class="line"><a name="l00284"></a><span class="lineno"><a class="line" href="namespacearmnn.xhtml#a49538fa883b70c944e437d65d6628eec"> 284</a></span> <span class="keywordtype">void</span> <a class="code" href="namespacearmnn.xhtml#a49538fa883b70c944e437d65d6628eec">ForEachLayerOutput</a>(LayerSelectionInfo::LayerInfoContainer& layerInfos,</div><div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  LayerSelectionInfo& layerInfo,</div><div class="line"><a name="l00286"></a><span class="lineno"> 286</span>  Delegate <span class="keyword">function</span>)</div><div class="line"><a name="l00287"></a><span class="lineno"> 287</span> {</div><div class="line"><a name="l00288"></a><span class="lineno"> 288</span>  <a class="code" href="classarmnn_1_1_layer.xhtml">Layer</a>& layer= *layerInfo.m_Layer;</div><div class="line"><a name="l00289"></a><span class="lineno"> 289</span> </div><div class="line"><a name="l00290"></a><span class="lineno"> 290</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& outputSlot : layer.<a class="code" href="classarmnn_1_1_layer.xhtml#a98cdff4e0b45f4c80bfcedaf926e16e0">GetOutputSlots</a>())</div><div class="line"><a name="l00291"></a><span class="lineno"> 291</span>  {</div><div class="line"><a name="l00292"></a><span class="lineno"> 292</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& output : outputSlot.GetConnections())</div><div class="line"><a name="l00293"></a><span class="lineno"> 293</span>  {</div><div class="line"><a name="l00294"></a><span class="lineno"> 294</span>  <a class="code" href="classarmnn_1_1_layer.xhtml">Layer</a>& childLayer = output->GetOwningLayer();</div><div class="line"><a name="l00295"></a><span class="lineno"> 295</span> </div><div class="line"><a name="l00296"></a><span class="lineno"> 296</span>  <span class="keyword">auto</span> childInfo = layerInfos.find(&childLayer);</div><div class="line"><a name="l00297"></a><span class="lineno"> 297</span>  <span class="keywordflow">if</span> (childInfo != layerInfos.end())</div><div class="line"><a name="l00298"></a><span class="lineno"> 298</span>  {</div><div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  <span class="keyword">function</span>(childInfo->second);</div><div class="line"><a name="l00300"></a><span class="lineno"> 300</span>  }</div><div class="line"><a name="l00301"></a><span class="lineno"> 301</span>  }</div><div class="line"><a name="l00302"></a><span class="lineno"> 302</span>  }</div><div class="line"><a name="l00303"></a><span class="lineno"> 303</span> }</div><div class="line"><a name="l00304"></a><span class="lineno"> 304</span> </div><div class="line"><a name="l00305"></a><span class="lineno"><a class="line" href="namespacearmnn.xhtml#a09ff1f6670d27d3b41e5b5d35a6c9f37"> 305</a></span> <span class="keywordtype">void</span> <a class="code" href="namespacearmnn.xhtml#a09ff1f6670d27d3b41e5b5d35a6c9f37">AssignSplitId</a>(LayerSelectionInfo::LayerInfoContainer& layerInfos, LayerSelectionInfo& layerInfo)</div><div class="line"><a name="l00306"></a><span class="lineno"> 306</span> {</div><div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  <span class="comment">// Check each input to see if we can attach ourselves to any of the subgraphs that have already been assigned.</span></div><div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  <a class="code" href="namespacearmnn.xhtml#afce94270d9c4a51cd0c4ac6a58af4e26">ForEachLayerInput</a>(layerInfos, layerInfo, [&](LayerSelectionInfo& parentInfo)</div><div class="line"><a name="l00309"></a><span class="lineno"> 309</span>  {</div><div class="line"><a name="l00310"></a><span class="lineno"> 310</span>  <span class="comment">// We can only attach ourselves to the subgraph from this input if there isn't a cut here.</span></div><div class="line"><a name="l00311"></a><span class="lineno"> 311</span>  <span class="keywordflow">if</span> (layerInfo.m_IsSelected == parentInfo.m_IsSelected)</div><div class="line"><a name="l00312"></a><span class="lineno"> 312</span>  {</div><div class="line"><a name="l00313"></a><span class="lineno"> 313</span>  <span class="comment">// We also need to check that merging into this subgraph won't cause a dependency cycle between subgraphs.</span></div><div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  <span class="comment">// This will be the case if the subgraph that we will become part of is already a dependency</span></div><div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  <span class="comment">// of one of the subgraphs that are input to this layer, e.g:</span></div><div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  <span class="comment">//</span></div><div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  <span class="comment">// 0 | The numbers (0, 1) are the subgraph IDs of each layer and we are looking at layer X.</span></div><div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  <span class="comment">// / \ |</span></div><div class="line"><a name="l00319"></a><span class="lineno"> 319</span>  <span class="comment">// 1 0 | We can't merge X into subgraph 0, because the left-hand input already depends on subgraph 0.</span></div><div class="line"><a name="l00320"></a><span class="lineno"> 320</span>  <span class="comment">// \ / | We can however merge X into subgraph 1.</span></div><div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  <span class="comment">// X |</span></div><div class="line"><a name="l00322"></a><span class="lineno"> 322</span>  <span class="comment">//</span></div><div class="line"><a name="l00323"></a><span class="lineno"> 323</span>  bool dependenciesOk = true;</div><div class="line"><a name="l00324"></a><span class="lineno"> 324</span>  ForEachLayerInput(layerInfos, layerInfo, [&](LayerSelectionInfo& otherParentInfo)</div><div class="line"><a name="l00325"></a><span class="lineno"> 325</span>  {</div><div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  <span class="comment">// We call HasAntecedent() ~ n^2 times, where n is the number of inputs to this layer.</span></div><div class="line"><a name="l00327"></a><span class="lineno"> 327</span>  <span class="comment">// Hence it is important that this is efficient - see PartialSubgraph class description.</span></div><div class="line"><a name="l00328"></a><span class="lineno"> 328</span>  if (otherParentInfo.m_Subgraph->HasAntecedent(parentInfo.m_Subgraph.get()))</div><div class="line"><a name="l00329"></a><span class="lineno"> 329</span>  {</div><div class="line"><a name="l00330"></a><span class="lineno"> 330</span>  dependenciesOk = false;</div><div class="line"><a name="l00331"></a><span class="lineno"> 331</span>  }</div><div class="line"><a name="l00332"></a><span class="lineno"> 332</span>  });</div><div class="line"><a name="l00333"></a><span class="lineno"> 333</span> </div><div class="line"><a name="l00334"></a><span class="lineno"> 334</span>  <span class="keywordflow">if</span> (dependenciesOk)</div><div class="line"><a name="l00335"></a><span class="lineno"> 335</span>  {</div><div class="line"><a name="l00336"></a><span class="lineno"> 336</span>  <span class="comment">// Merge into the subgraph of this input. If we have already been merged into another subgraph</span></div><div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  <span class="comment">// (from another input of this layer), then merge both of them together.</span></div><div class="line"><a name="l00338"></a><span class="lineno"> 338</span>  <span class="keywordflow">if</span> (layerInfo.m_Subgraph == <span class="keyword">nullptr</span>)</div><div class="line"><a name="l00339"></a><span class="lineno"> 339</span>  {</div><div class="line"><a name="l00340"></a><span class="lineno"> 340</span>  layerInfo.m_Subgraph = parentInfo.m_Subgraph;</div><div class="line"><a name="l00341"></a><span class="lineno"> 341</span>  }</div><div class="line"><a name="l00342"></a><span class="lineno"> 342</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00343"></a><span class="lineno"> 343</span>  {</div><div class="line"><a name="l00344"></a><span class="lineno"> 344</span>  <span class="comment">// We call MergeWith() ~ n times, where n is the number of inputs to this layer.</span></div><div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  <span class="comment">// Therefore it does not need to be as performant as HasAntecedent().</span></div><div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  layerInfo.m_Subgraph->MergeWith(parentInfo.m_Subgraph.get());</div><div class="line"><a name="l00347"></a><span class="lineno"> 347</span>  }</div><div class="line"><a name="l00348"></a><span class="lineno"> 348</span>  }</div><div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  }</div><div class="line"><a name="l00350"></a><span class="lineno"> 350</span>  });</div><div class="line"><a name="l00351"></a><span class="lineno"> 351</span> </div><div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  <span class="comment">// If we weren't able to merge into an existing subgraph then we need to make a new one</span></div><div class="line"><a name="l00353"></a><span class="lineno"> 353</span>  <span class="keywordflow">if</span> (layerInfo.m_Subgraph == <span class="keyword">nullptr</span>)</div><div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  {</div><div class="line"><a name="l00355"></a><span class="lineno"> 355</span>  layerInfo.m_Subgraph = std::make_shared<PartialSubgraph>();</div><div class="line"><a name="l00356"></a><span class="lineno"> 356</span>  }</div><div class="line"><a name="l00357"></a><span class="lineno"> 357</span> </div><div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  <span class="comment">// Record dependencies of the chosen subgraph based on the inputs of this layer.</span></div><div class="line"><a name="l00359"></a><span class="lineno"> 359</span>  <a class="code" href="namespacearmnn.xhtml#afce94270d9c4a51cd0c4ac6a58af4e26">ForEachLayerInput</a>(layerInfos, layerInfo, [&](LayerSelectionInfo& parentInfo)</div><div class="line"><a name="l00360"></a><span class="lineno"> 360</span>  {</div><div class="line"><a name="l00361"></a><span class="lineno"> 361</span>  <span class="comment">// These functions are called ~n times, where n is the number of inputs to this layer.</span></div><div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  <span class="comment">// Therefore it does not need to be as performant as HasAntecedent().</span></div><div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  <span class="keywordflow">if</span> (!layerInfo.m_Subgraph->IsMergedWith(parentInfo.m_Subgraph.get()))</div><div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  {</div><div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  layerInfo.m_Subgraph->AddDirectAntecedent(parentInfo.m_Subgraph.get());</div><div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  }</div><div class="line"><a name="l00367"></a><span class="lineno"> 367</span>  });</div><div class="line"><a name="l00368"></a><span class="lineno"> 368</span> }</div><div class="line"><a name="l00369"></a><span class="lineno"> 369</span> </div><div class="line"><a name="l00370"></a><span class="lineno"><a class="line" href="namespacearmnn.xhtml#a6b10dc0d12c7f4a52ad01b9975dbe908"> 370</a></span> <span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.xhtml#a6b10dc0d12c7f4a52ad01b9975dbe908">IsReadyForSplitAssignment</a>(LayerSelectionInfo::LayerInfoContainer& layerInfos, LayerSelectionInfo& layerInfo)</div><div class="line"><a name="l00371"></a><span class="lineno"> 371</span> {</div><div class="line"><a name="l00372"></a><span class="lineno"> 372</span>  <span class="keywordtype">bool</span> ready = <span class="keyword">true</span>;</div><div class="line"><a name="l00373"></a><span class="lineno"> 373</span>  <a class="code" href="namespacearmnn.xhtml#afce94270d9c4a51cd0c4ac6a58af4e26">ForEachLayerInput</a>(layerInfos, layerInfo,</div><div class="line"><a name="l00374"></a><span class="lineno"> 374</span>  [&ready](LayerSelectionInfo& parentInfo)</div><div class="line"><a name="l00375"></a><span class="lineno"> 375</span>  {</div><div class="line"><a name="l00376"></a><span class="lineno"> 376</span>  <span class="keywordflow">if</span> (!parentInfo.m_IsProcessed)</div><div class="line"><a name="l00377"></a><span class="lineno"> 377</span>  {</div><div class="line"><a name="l00378"></a><span class="lineno"> 378</span>  ready = false;</div><div class="line"><a name="l00379"></a><span class="lineno"> 379</span>  }</div><div class="line"><a name="l00380"></a><span class="lineno"> 380</span>  });</div><div class="line"><a name="l00381"></a><span class="lineno"> 381</span>  <span class="keywordflow">return</span> ready;</div><div class="line"><a name="l00382"></a><span class="lineno"> 382</span> }</div><div class="line"><a name="l00383"></a><span class="lineno"> 383</span> </div><div class="line"><a name="l00384"></a><span class="lineno"> 384</span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#aaf71a63dbbc776f8961b0f4fdb9da021">SubgraphViewSelector::Subgraphs</a></div><div class="line"><a name="l00385"></a><span class="lineno"><a class="line" href="classarmnn_1_1_subgraph_view_selector.xhtml#ad28bb1b624cde39578cfefa8b7809aef"> 385</a></span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#a3730b0a6006f0d87f894a44e01869d90">SubgraphViewSelector::SelectSubgraphs</a>(<a class="code" href="classarmnn_1_1_subgraph_view.xhtml">SubgraphView</a>& subgraph, <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#a341a1b47a5f71edaebf81f3a5f0f4c55">LayerSelectorFunction</a>& selector)</div><div class="line"><a name="l00386"></a><span class="lineno"> 386</span> {</div><div class="line"><a name="l00387"></a><span class="lineno"> 387</span>  LayerSelectionInfo::LayerInfoContainer layerInfos;</div><div class="line"><a name="l00388"></a><span class="lineno"> 388</span> </div><div class="line"><a name="l00389"></a><span class="lineno"> 389</span>  LayerSelectionInfo::LayerInfoQueue processQueue;</div><div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& layer : subgraph)</div><div class="line"><a name="l00391"></a><span class="lineno"> 391</span>  {</div><div class="line"><a name="l00392"></a><span class="lineno"> 392</span>  <span class="keyword">auto</span> emplaced = layerInfos.emplace(layer, LayerSelectionInfo{layer, selector});</div><div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  LayerSelectionInfo& layerInfo = emplaced.first->second;</div><div class="line"><a name="l00394"></a><span class="lineno"> 394</span> </div><div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  <span class="comment">// Start with Input type layers</span></div><div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  <span class="keywordflow">if</span> (layerInfo.IsInputLayer())</div><div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  {</div><div class="line"><a name="l00398"></a><span class="lineno"> 398</span>  processQueue.push(&layerInfo);</div><div class="line"><a name="l00399"></a><span class="lineno"> 399</span>  }</div><div class="line"><a name="l00400"></a><span class="lineno"> 400</span>  }</div><div class="line"><a name="l00401"></a><span class="lineno"> 401</span> </div><div class="line"><a name="l00402"></a><span class="lineno"> 402</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_subgraph_view.xhtml#a5cc65e15002dbc33a5c8a7d6680e9a9d">SubgraphView::InputSlots</a>& subgraphInputSlots = subgraph.GetInputSlots();</div><div class="line"><a name="l00403"></a><span class="lineno"> 403</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& inputSlot : subgraphInputSlots)</div><div class="line"><a name="l00404"></a><span class="lineno"> 404</span>  {</div><div class="line"><a name="l00405"></a><span class="lineno"> 405</span>  <a class="code" href="classarmnn_1_1_layer.xhtml">Layer</a>& layer = inputSlot->GetOwningLayer();</div><div class="line"><a name="l00406"></a><span class="lineno"> 406</span>  <span class="keyword">auto</span> emplaced = layerInfos.emplace(&layer, LayerSelectionInfo{&layer, selector});</div><div class="line"><a name="l00407"></a><span class="lineno"> 407</span>  LayerSelectionInfo& layerInfo = emplaced.first->second;</div><div class="line"><a name="l00408"></a><span class="lineno"> 408</span> </div><div class="line"><a name="l00409"></a><span class="lineno"> 409</span>  processQueue.push(&layerInfo);</div><div class="line"><a name="l00410"></a><span class="lineno"> 410</span>  }</div><div class="line"><a name="l00411"></a><span class="lineno"> 411</span> </div><div class="line"><a name="l00412"></a><span class="lineno"> 412</span>  <span class="keywordflow">while</span> (!processQueue.empty())</div><div class="line"><a name="l00413"></a><span class="lineno"> 413</span>  {</div><div class="line"><a name="l00414"></a><span class="lineno"> 414</span>  LayerSelectionInfo& layerInfo = *processQueue.front();</div><div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  processQueue.pop(); <span class="comment">// remove front from queue</span></div><div class="line"><a name="l00416"></a><span class="lineno"> 416</span> </div><div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  <span class="comment">// This layerInfo may have been added to the queue multiple times, so skip if we have already processed it</span></div><div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  <span class="keywordflow">if</span> (!layerInfo.m_IsProcessed)</div><div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  {</div><div class="line"><a name="l00420"></a><span class="lineno"> 420</span>  <span class="comment">// Only process this layerInfo if all inputs have been processed</span></div><div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn.xhtml#a6b10dc0d12c7f4a52ad01b9975dbe908">IsReadyForSplitAssignment</a>(layerInfos, layerInfo))</div><div class="line"><a name="l00422"></a><span class="lineno"> 422</span>  {</div><div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  <span class="comment">// Put back of the process queue if we can't process it just yet</span></div><div class="line"><a name="l00424"></a><span class="lineno"> 424</span>  processQueue.push(&layerInfo);</div><div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  <span class="keywordflow">continue</span>; <span class="comment">// Skip to next iteration</span></div><div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  }</div><div class="line"><a name="l00427"></a><span class="lineno"> 427</span> </div><div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="comment">// Now we do the processing</span></div><div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  <a class="code" href="namespacearmnn.xhtml#a09ff1f6670d27d3b41e5b5d35a6c9f37">AssignSplitId</a>(layerInfos, layerInfo);</div><div class="line"><a name="l00430"></a><span class="lineno"> 430</span> </div><div class="line"><a name="l00431"></a><span class="lineno"> 431</span>  <span class="comment">// Queue any child nodes for processing</span></div><div class="line"><a name="l00432"></a><span class="lineno"> 432</span>  <a class="code" href="namespacearmnn.xhtml#a49538fa883b70c944e437d65d6628eec">ForEachLayerOutput</a>(layerInfos, layerInfo, [&processQueue](LayerSelectionInfo& childInfo)</div><div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  {</div><div class="line"><a name="l00434"></a><span class="lineno"> 434</span>  processQueue.push(&childInfo);</div><div class="line"><a name="l00435"></a><span class="lineno"> 435</span>  });</div><div class="line"><a name="l00436"></a><span class="lineno"> 436</span> </div><div class="line"><a name="l00437"></a><span class="lineno"> 437</span>  <span class="comment">// We don't need to process this node again</span></div><div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  layerInfo.m_IsProcessed = <span class="keyword">true</span>;</div><div class="line"><a name="l00439"></a><span class="lineno"> 439</span>  }</div><div class="line"><a name="l00440"></a><span class="lineno"> 440</span>  }</div><div class="line"><a name="l00441"></a><span class="lineno"> 441</span> </div><div class="line"><a name="l00442"></a><span class="lineno"> 442</span>  <span class="comment">// Collect all selected layers keyed by subgraph representative into a map</span></div><div class="line"><a name="l00443"></a><span class="lineno"> 443</span>  <span class="keyword">using</span> SelectionInfoPtrs = std::vector<LayerSelectionInfo*>;</div><div class="line"><a name="l00444"></a><span class="lineno"> 444</span>  std::map<PartialSubgraph*, SelectionInfoPtrs> splitMap;</div><div class="line"><a name="l00445"></a><span class="lineno"> 445</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& <a class="code" href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">info</a> : layerInfos)</div><div class="line"><a name="l00446"></a><span class="lineno"> 446</span>  {</div><div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">info</a>.second.m_IsSelected)</div><div class="line"><a name="l00448"></a><span class="lineno"> 448</span>  {</div><div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  <span class="keyword">auto</span> it = splitMap.find(<a class="code" href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">info</a>.second.m_Subgraph->GetRepresentative());</div><div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  <span class="keywordflow">if</span> (it == splitMap.end())</div><div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  {</div><div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  splitMap.insert(</div><div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  std::make_pair(<a class="code" href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">info</a>.second.m_Subgraph->GetRepresentative(), SelectionInfoPtrs{&<a class="code" href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">info</a>.second}));</div><div class="line"><a name="l00454"></a><span class="lineno"> 454</span>  }</div><div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  {</div><div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  it->second.push_back(&<a class="code" href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">info</a>.second);</div><div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  }</div><div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  }</div><div class="line"><a name="l00460"></a><span class="lineno"> 460</span>  }</div><div class="line"><a name="l00461"></a><span class="lineno"> 461</span> </div><div class="line"><a name="l00462"></a><span class="lineno"> 462</span>  <span class="comment">// Now each entry in splitMap represents a subgraph</span></div><div class="line"><a name="l00463"></a><span class="lineno"> 463</span>  <a class="code" href="classarmnn_1_1_subgraph_view_selector.xhtml#aaf71a63dbbc776f8961b0f4fdb9da021">Subgraphs</a> result;</div><div class="line"><a name="l00464"></a><span class="lineno"> 464</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& splitGraph : splitMap)</div><div class="line"><a name="l00465"></a><span class="lineno"> 465</span>  {</div><div class="line"><a name="l00466"></a><span class="lineno"> 466</span>  <a class="code" href="classarmnn_1_1_subgraph_view.xhtml#a5cc65e15002dbc33a5c8a7d6680e9a9d">SubgraphView::InputSlots</a> inputs;</div><div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  <a class="code" href="classarmnn_1_1_subgraph_view.xhtml#a78293334750ec5279eb9c96d56deaf08">SubgraphView::OutputSlots</a> outputs;</div><div class="line"><a name="l00468"></a><span class="lineno"> 468</span>  <a class="code" href="classarmnn_1_1_subgraph_view.xhtml#a74798938fdaeae75c8adfa4a7439e7f9">SubgraphView::Layers</a> layers;</div><div class="line"><a name="l00469"></a><span class="lineno"> 469</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>&& infoPtr : splitGraph.second)</div><div class="line"><a name="l00470"></a><span class="lineno"> 470</span>  {</div><div class="line"><a name="l00471"></a><span class="lineno"> 471</span>  infoPtr->CollectNonSelectedInputs(layerInfos, inputs);</div><div class="line"><a name="l00472"></a><span class="lineno"> 472</span>  infoPtr->CollectNonSelectedOutputSlots(layerInfos, outputs);</div><div class="line"><a name="l00473"></a><span class="lineno"> 473</span>  layers.push_back(infoPtr->m_Layer);</div><div class="line"><a name="l00474"></a><span class="lineno"> 474</span>  }</div><div class="line"><a name="l00475"></a><span class="lineno"> 475</span>  <span class="comment">// Create a new sub-graph with the new lists of input/output slots and layer</span></div><div class="line"><a name="l00476"></a><span class="lineno"> 476</span>  result.emplace_back(std::make_unique<SubgraphView>(std::move(inputs),</div><div class="line"><a name="l00477"></a><span class="lineno"> 477</span>  std::move(outputs),</div><div class="line"><a name="l00478"></a><span class="lineno"> 478</span>  std::move(layers)));</div><div class="line"><a name="l00479"></a><span class="lineno"> 479</span>  }</div><div class="line"><a name="l00480"></a><span class="lineno"> 480</span> </div><div class="line"><a name="l00481"></a><span class="lineno"> 481</span>  <span class="keywordflow">return</span> result;</div><div class="line"><a name="l00482"></a><span class="lineno"> 482</span> }</div><div class="line"><a name="l00483"></a><span class="lineno"> 483</span> </div><div class="line"><a name="l00484"></a><span class="lineno"> 484</span> } <span class="comment">// namespace armnn</span></div><div class="ttc" id="_ignore_unused_8hpp_xhtml"><div class="ttname"><a href="_ignore_unused_8hpp.xhtml">IgnoreUnused.hpp</a></div></div>
<div class="ttc" id="_subgraph_view_selector_8cpp_xhtml_aa479d97c404e5996e29b009c60b22e8b"><div class="ttname"><a href="_subgraph_view_selector_8cpp.xhtml#aa479d97c404e5996e29b009c60b22e8b">m_IsSelected</a></div><div class="ttdeci">bool m_IsSelected</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00248">SubgraphViewSelector.cpp:248</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_afce94270d9c4a51cd0c4ac6a58af4e26"><div class="ttname"><a href="namespacearmnn.xhtml#afce94270d9c4a51cd0c4ac6a58af4e26">armnn::ForEachLayerInput</a></div><div class="ttdeci">void ForEachLayerInput(LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo, Delegate function)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00263">SubgraphViewSelector.cpp:263</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a09ff1f6670d27d3b41e5b5d35a6c9f37"><div class="ttname"><a href="namespacearmnn.xhtml#a09ff1f6670d27d3b41e5b5d35a6c9f37">armnn::AssignSplitId</a></div><div class="ttdeci">void AssignSplitId(LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00305">SubgraphViewSelector.cpp:305</a></div></div>
<div class="ttc" id="namespacearmnn_serializer_xhtml_a9a8118be7780e95363d631cbca7e7800"><div class="ttname"><a href="namespacearmnn_serializer.xhtml#a9a8118be7780e95363d631cbca7e7800">armnnSerializer::Layer</a></div><div class="ttdeci">Layer</div><div class="ttdef"><b>Definition:</b> <a href="_armnn_schema__generated_8h_source.xhtml#l01144">ArmnnSchema_generated.h:1144</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_xhtml_a78293334750ec5279eb9c96d56deaf08"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.xhtml#a78293334750ec5279eb9c96d56deaf08">armnn::SubgraphView::OutputSlots</a></div><div class="ttdeci">std::vector< OutputSlot * > OutputSlots</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_8hpp_source.xhtml#l00039">SubgraphView.hpp:39</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_selector_xhtml_a341a1b47a5f71edaebf81f3a5f0f4c55"><div class="ttname"><a href="classarmnn_1_1_subgraph_view_selector.xhtml#a341a1b47a5f71edaebf81f3a5f0f4c55">armnn::SubgraphViewSelector::LayerSelectorFunction</a></div><div class="ttdeci">std::function< bool(const Layer &)> LayerSelectorFunction</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8hpp_source.xhtml#l00026">SubgraphViewSelector.hpp:26</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml"><div class="ttname"><a href="namespacearmnn.xhtml">armnn</a></div><div class="ttdoc">Copyright (c) 2021 ARM Limited and Contributors. </div><div class="ttdef"><b>Definition:</b> <a href="01__00__software__tools_8dox_source.xhtml#l00006">01_00_software_tools.dox:6</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a44affeeb090c3c6a3062830562672e84"><div class="ttname"><a href="namespacearmnn.xhtml#a44affeeb090c3c6a3062830562672e84">armnn::IgnoreUnused</a></div><div class="ttdeci">void IgnoreUnused(Ts &&...)</div><div class="ttdef"><b>Definition:</b> <a href="_ignore_unused_8hpp_source.xhtml#l00014">IgnoreUnused.hpp:14</a></div></div>
<div class="ttc" id="classarmnn_1_1_layer_xhtml_af5f530544d09a44d726f24702b67b35b"><div class="ttname"><a href="classarmnn_1_1_layer.xhtml#af5f530544d09a44d726f24702b67b35b">armnn::Layer::GetInputSlots</a></div><div class="ttdeci">const std::vector< InputSlot > & GetInputSlots() const</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.xhtml#l00237">Layer.hpp:237</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a6b10dc0d12c7f4a52ad01b9975dbe908"><div class="ttname"><a href="namespacearmnn.xhtml#a6b10dc0d12c7f4a52ad01b9975dbe908">armnn::IsReadyForSplitAssignment</a></div><div class="ttdeci">bool IsReadyForSplitAssignment(LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00370">SubgraphViewSelector.cpp:370</a></div></div>
<div class="ttc" id="_polymorphic_downcast_8hpp_xhtml"><div class="ttname"><a href="_polymorphic_downcast_8hpp.xhtml">PolymorphicDowncast.hpp</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_xhtml"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.xhtml">armnn::SubgraphView</a></div><div class="ttdoc">The SubgraphView class represents a subgraph of a Graph. </div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_8hpp_source.xhtml#l00023">SubgraphView.hpp:23</a></div></div>
<div class="ttc" id="_subgraph_view_selector_8cpp_xhtml_ae7fc37b88ff10e9294ed1c72f2a25ac7"><div class="ttname"><a href="_subgraph_view_selector_8cpp.xhtml#ae7fc37b88ff10e9294ed1c72f2a25ac7">m_Layer</a></div><div class="ttdeci">Layer * m_Layer</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00243">SubgraphViewSelector.cpp:243</a></div></div>
<div class="ttc" id="_assert_8hpp_xhtml_a91c4dfde57907d7698c7531785690a7f"><div class="ttname"><a href="_assert_8hpp.xhtml#a91c4dfde57907d7698c7531785690a7f">ARMNN_ASSERT_MSG</a></div><div class="ttdeci">#define ARMNN_ASSERT_MSG(COND, MSG)</div><div class="ttdef"><b>Definition:</b> <a href="_assert_8hpp_source.xhtml#l00015">Assert.hpp:15</a></div></div>
<div class="ttc" id="_graph_8hpp_xhtml"><div class="ttname"><a href="_graph_8hpp.xhtml">Graph.hpp</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_selector_xhtml_aaf71a63dbbc776f8961b0f4fdb9da021"><div class="ttname"><a href="classarmnn_1_1_subgraph_view_selector.xhtml#aaf71a63dbbc776f8961b0f4fdb9da021">armnn::SubgraphViewSelector::Subgraphs</a></div><div class="ttdeci">std::vector< SubgraphViewPtr > Subgraphs</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8hpp_source.xhtml#l00025">SubgraphViewSelector.hpp:25</a></div></div>
<div class="ttc" id="_assert_8hpp_xhtml_a5698be69cbd5dfe6c28fcd9867e8cbed"><div class="ttname"><a href="_assert_8hpp.xhtml#a5698be69cbd5dfe6c28fcd9867e8cbed">ARMNN_ASSERT</a></div><div class="ttdeci">#define ARMNN_ASSERT(COND)</div><div class="ttdef"><b>Definition:</b> <a href="_assert_8hpp_source.xhtml#l00014">Assert.hpp:14</a></div></div>
<div class="ttc" id="classarmnn_1_1_graph_xhtml"><div class="ttname"><a href="classarmnn_1_1_graph.xhtml">armnn::Graph</a></div><div class="ttdef"><b>Definition:</b> <a href="_graph_8hpp_source.xhtml#l00029">Graph.hpp:29</a></div></div>
<div class="ttc" id="classarmnn_1_1_layer_xhtml_a98cdff4e0b45f4c80bfcedaf926e16e0"><div class="ttname"><a href="classarmnn_1_1_layer.xhtml#a98cdff4e0b45f4c80bfcedaf926e16e0">armnn::Layer::GetOutputSlots</a></div><div class="ttdeci">const std::vector< OutputSlot > & GetOutputSlots() const</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.xhtml#l00238">Layer.hpp:238</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_xhtml_a5cc65e15002dbc33a5c8a7d6680e9a9d"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.xhtml#a5cc65e15002dbc33a5c8a7d6680e9a9d">armnn::SubgraphView::InputSlots</a></div><div class="ttdeci">std::vector< InputSlot * > InputSlots</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_8hpp_source.xhtml#l00038">SubgraphView.hpp:38</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a56943a0946e5f15e5e58054b8e7a04a4acb17869fe51048b5a5c4c6106551a255"><div class="ttname"><a href="namespacearmnn.xhtml#a56943a0946e5f15e5e58054b8e7a04a4acb17869fe51048b5a5c4c6106551a255">armnn::LayerType::Constant</a></div></div>
<div class="ttc" id="_assert_8hpp_xhtml"><div class="ttname"><a href="_assert_8hpp.xhtml">Assert.hpp</a></div></div>
<div class="ttc" id="_subgraph_view_selector_8cpp_xhtml_a849c95a020d031ec405d7d21021d87c2"><div class="ttname"><a href="_subgraph_view_selector_8cpp.xhtml#a849c95a020d031ec405d7d21021d87c2">m_IsProcessed</a></div><div class="ttdeci">bool m_IsProcessed</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00249">SubgraphViewSelector.cpp:249</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_selector_xhtml_a3730b0a6006f0d87f894a44e01869d90"><div class="ttname"><a href="classarmnn_1_1_subgraph_view_selector.xhtml#a3730b0a6006f0d87f894a44e01869d90">armnn::SubgraphViewSelector::SelectSubgraphs</a></div><div class="ttdeci">static Subgraphs SelectSubgraphs(Graph &graph, const LayerSelectorFunction &selector)</div><div class="ttdoc">Selects subgraphs from a graph based on the selector function and the algorithm. </div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00255">SubgraphViewSelector.cpp:255</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c"><div class="ttname"><a href="namespacearmnn.xhtml#a4dc0adc6737b5944e7671bee71788407acaf9b6b99962bf5c2264824231d7a40c">armnn::BoostLogSeverityMapping::info</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a56943a0946e5f15e5e58054b8e7a04a4a324118a6721dd6b8a9b9f4e327df2bf5"><div class="ttname"><a href="namespacearmnn.xhtml#a56943a0946e5f15e5e58054b8e7a04a4a324118a6721dd6b8a9b9f4e327df2bf5">armnn::LayerType::Input</a></div></div>
<div class="ttc" id="_subgraph_view_selector_8cpp_xhtml_a3a672da6a5207243cf3e5c933872b5bf"><div class="ttname"><a href="_subgraph_view_selector_8cpp.xhtml#a3a672da6a5207243cf3e5c933872b5bf">m_Subgraph</a></div><div class="ttdeci">std::shared_ptr< PartialSubgraph > m_Subgraph</div><div class="ttdoc">Which subgraph this layer has been assigned to. </div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00247">SubgraphViewSelector.cpp:247</a></div></div>
<div class="ttc" id="classarmnn_1_1_subgraph_view_xhtml_a74798938fdaeae75c8adfa4a7439e7f9"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.xhtml#a74798938fdaeae75c8adfa4a7439e7f9">armnn::SubgraphView::Layers</a></div><div class="ttdeci">std::list< Layer * > Layers</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_8hpp_source.xhtml#l00040">SubgraphView.hpp:40</a></div></div>
<div class="ttc" id="_subgraph_view_selector_8hpp_xhtml"><div class="ttname"><a href="_subgraph_view_selector_8hpp.xhtml">SubgraphViewSelector.hpp</a></div></div>
<div class="ttc" id="namespacearmnn_xhtml_a49538fa883b70c944e437d65d6628eec"><div class="ttname"><a href="namespacearmnn.xhtml#a49538fa883b70c944e437d65d6628eec">armnn::ForEachLayerOutput</a></div><div class="ttdeci">void ForEachLayerOutput(LayerSelectionInfo::LayerInfoContainer &layerInfos, LayerSelectionInfo &layerInfo, Delegate function)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_selector_8cpp_source.xhtml#l00284">SubgraphViewSelector.cpp:284</a></div></div>
<div class="ttc" id="classarmnn_1_1_layer_xhtml"><div class="ttname"><a href="classarmnn_1_1_layer.xhtml">armnn::Layer</a></div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.xhtml#l00210">Layer.hpp:210</a></div></div>
</div><!-- fragment --></div><!-- contents -->
</div><!-- doc-content -->
<!-- start footer part -->
<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
<ul>
<li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.xhtml">src</a></li><li class="navelem"><a class="el" href="dir_e0a84d05c80a2ef4231141dcbbeac5c8.xhtml">armnn</a></li><li class="navelem"><a class="el" href="_subgraph_view_selector_8cpp.xhtml">SubgraphViewSelector.cpp</a></li>
<li class="footer">Generated on Thu Feb 25 2021 17:27:29 for ArmNN by
<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.13 </li>
</ul>
</div>
</body>
</html>
|
