Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 3e6676da92c228196c6cc7e6fe4e8891ac1b387c / . / Documentation / classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html
blob: df37a9cedcecd5bdd8aa687ed80582c2567ee7c2 [file] [log] [blame]
Ryan OSheade36e4a2020-03-13 16:26:19 +0000[diff] [blame]1<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
2<html xmlns="http://www.w3.org/1999/xhtml">
3<head>
4<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
5<meta http-equiv="X-UA-Compatible" content="IE=9"/>
6<meta name="generator" content="Doxygen 1.8.13"/>
7<meta name="viewport" content="width=device-width, initial-scale=1"/>
8<title>ArmNN: OptimizeConsecutiveReshapesImpl Class Reference</title>
9<link href="tabs.css" rel="stylesheet" type="text/css"/>
10<script type="text/javascript" src="jquery.js"></script>
11<script type="text/javascript" src="dynsections.js"></script>
12<link href="navtree.css" rel="stylesheet" type="text/css"/>
13<script type="text/javascript" src="resize.js"></script>
14<script type="text/javascript" src="navtreedata.js"></script>
15<script type="text/javascript" src="navtree.js"></script>
16<script type="text/javascript">
17 $(document).ready(initResizable);
18</script>
19<link href="search/search.css" rel="stylesheet" type="text/css"/>
20<script type="text/javascript" src="search/searchdata.js"></script>
21<script type="text/javascript" src="search/search.js"></script>
22<link href="doxygen.css" rel="stylesheet" type="text/css" />
23</head>
24<body>
25<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
26<div id="titlearea">
27<table cellspacing="0" cellpadding="0">
28 <tbody>
29 <tr style="height: 56px;">
30 <td id="projectalign" style="padding-left: 0.5em;">
31 <div id="projectname">ArmNN
32 &#160;<span id="projectnumber">NotReleased</span>
33 </div>
34 </td>
35 </tr>
36 </tbody>
37</table>
38</div>
39<!-- end header part -->
40<!-- Generated by Doxygen 1.8.13 -->
41<script type="text/javascript">
42var searchBox = new SearchBox("searchBox", "search",false,'Search');
43</script>
44<script type="text/javascript" src="menudata.js"></script>
45<script type="text/javascript" src="menu.js"></script>
46<script type="text/javascript">
47$(function() {
48 initMenu('',true,false,'search.php','Search');
49 $(document).ready(function() { init_search(); });
50});
51</script>
52<div id="main-nav"></div>
53</div><!-- top -->
54<div id="side-nav" class="ui-resizable side-nav-resizable">
55 <div id="nav-tree">
56 <div id="nav-tree-contents">
57 <div id="nav-sync" class="sync"></div>
58 </div>
59 </div>
60 <div id="splitbar" style="-moz-user-select:none;"
61 class="ui-resizable-handle">
62 </div>
63</div>
64<script type="text/javascript">
65$(document).ready(function(){initNavTree('classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html','');});
66</script>
67<div id="doc-content">
68<!-- window showing the filter options -->
69<div id="MSearchSelectWindow"
70 onmouseover="return searchBox.OnSearchSelectShow()"
71 onmouseout="return searchBox.OnSearchSelectHide()"
72 onkeydown="return searchBox.OnSearchSelectKey(event)">
73</div>
74
75<!-- iframe showing the search results (closed by default) -->
76<div id="MSearchResultsWindow">
77<iframe src="javascript:void(0)" frameborder="0"
78 name="MSearchResults" id="MSearchResults">
79</iframe>
80</div>
81
82<div class="header">
83 <div class="summary">
84<a href="#pub-methods">Public Member Functions</a> &#124;
85<a href="#pro-methods">Protected Member Functions</a> &#124;
86<a href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl-members.html">List of all members</a> </div>
87 <div class="headertitle">
88<div class="title">OptimizeConsecutiveReshapesImpl Class Reference</div> </div>
89</div><!--header-->
90<div class="contents">
91
92<p><code>#include &lt;<a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html">OptimizeConsecutiveReshapes.hpp</a>&gt;</code></p>
93<table class="memberdecls">
94<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-methods"></a>
95Public Member Functions</h2></td></tr>
96<tr class="memitem:a5a8476ffc04ce7460bb09ad50d1d23de"><td class="memItemLeft" align="right" valign="top">void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html#a5a8476ffc04ce7460bb09ad50d1d23de">Run</a> (<a class="el" href="classarmnn_1_1_graph.html">Graph</a> &amp;graph, <a class="el" href="classarmnn_1_1_input_slot.html">InputSlot</a> &amp;connection) const</td></tr>
97<tr class="separator:a5a8476ffc04ce7460bb09ad50d1d23de"><td class="memSeparator" colspan="2">&#160;</td></tr>
98</table><table class="memberdecls">
99<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pro-methods"></a>
100Protected Member Functions</h2></td></tr>
101<tr class="memitem:ae344e9caa07713681429bd8eea772ebf"><td class="memItemLeft" align="right" valign="top">&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html#ae344e9caa07713681429bd8eea772ebf">OptimizeConsecutiveReshapesImpl</a> ()=default</td></tr>
102<tr class="separator:ae344e9caa07713681429bd8eea772ebf"><td class="memSeparator" colspan="2">&#160;</td></tr>
103<tr class="memitem:accd18a7450d91508ab6e12379507e01d"><td class="memItemLeft" align="right" valign="top">&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html#accd18a7450d91508ab6e12379507e01d">~OptimizeConsecutiveReshapesImpl</a> ()=default</td></tr>
104<tr class="separator:accd18a7450d91508ab6e12379507e01d"><td class="memSeparator" colspan="2">&#160;</td></tr>
105</table>
106<a name="details" id="details"></a><h2 class="groupheader">Detailed Description</h2>
107<div class="textblock">
108<p class="definition">Definition at line <a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html#l00014">14</a> of file <a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html">OptimizeConsecutiveReshapes.hpp</a>.</p>
109</div><h2 class="groupheader">Constructor &amp; Destructor Documentation</h2>
110<a id="ae344e9caa07713681429bd8eea772ebf"></a>
111<h2 class="memtitle"><span class="permalink"><a href="#ae344e9caa07713681429bd8eea772ebf">&#9670;&nbsp;</a></span>OptimizeConsecutiveReshapesImpl()</h2>
112
113<div class="memitem">
114<div class="memproto">
115<table class="mlabels">
116 <tr>
117 <td class="mlabels-left">
118 <table class="memname">
119 <tr>
120 <td class="memname"><a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html">OptimizeConsecutiveReshapesImpl</a> </td>
121 <td>(</td>
122 <td class="paramname"></td><td>)</td>
123 <td></td>
124 </tr>
125 </table>
126 </td>
127 <td class="mlabels-right">
128<span class="mlabels"><span class="mlabel">protected</span><span class="mlabel">default</span></span> </td>
129 </tr>
130</table>
131</div><div class="memdoc">
132
133<p class="reference">Referenced by <a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html#l00019">OptimizeConsecutiveReshapesImpl::Run()</a>.</p>
134
135</div>
136</div>
137<a id="accd18a7450d91508ab6e12379507e01d"></a>
138<h2 class="memtitle"><span class="permalink"><a href="#accd18a7450d91508ab6e12379507e01d">&#9670;&nbsp;</a></span>~OptimizeConsecutiveReshapesImpl()</h2>
139
140<div class="memitem">
141<div class="memproto">
142<table class="mlabels">
143 <tr>
144 <td class="mlabels-left">
145 <table class="memname">
146 <tr>
147 <td class="memname">~<a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html">OptimizeConsecutiveReshapesImpl</a> </td>
148 <td>(</td>
149 <td class="paramname"></td><td>)</td>
150 <td></td>
151 </tr>
152 </table>
153 </td>
154 <td class="mlabels-right">
155<span class="mlabels"><span class="mlabel">protected</span><span class="mlabel">default</span></span> </td>
156 </tr>
157</table>
158</div><div class="memdoc">
159
160<p class="reference">Referenced by <a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html#l00019">OptimizeConsecutiveReshapesImpl::Run()</a>.</p>
161
162</div>
163</div>
164<h2 class="groupheader">Member Function Documentation</h2>
165<a id="a5a8476ffc04ce7460bb09ad50d1d23de"></a>
166<h2 class="memtitle"><span class="permalink"><a href="#a5a8476ffc04ce7460bb09ad50d1d23de">&#9670;&nbsp;</a></span>Run()</h2>
167
168<div class="memitem">
169<div class="memproto">
170<table class="mlabels">
171 <tr>
172 <td class="mlabels-left">
173 <table class="memname">
174 <tr>
175 <td class="memname">void Run </td>
176 <td>(</td>
177 <td class="paramtype"><a class="el" href="classarmnn_1_1_graph.html">Graph</a> &amp;&#160;</td>
178 <td class="paramname"><em>graph</em>, </td>
179 </tr>
180 <tr>
181 <td class="paramkey"></td>
182 <td></td>
183 <td class="paramtype"><a class="el" href="classarmnn_1_1_input_slot.html">InputSlot</a> &amp;&#160;</td>
184 <td class="paramname"><em>connection</em>&#160;</td>
185 </tr>
186 <tr>
187 <td></td>
188 <td>)</td>
189 <td></td><td> const</td>
190 </tr>
191 </table>
192 </td>
193 <td class="mlabels-right">
194<span class="mlabels"><span class="mlabel">inline</span></span> </td>
195 </tr>
196</table>
197</div><div class="memdoc">
198<p>Run for every connection between a base <a class="el" href="classarmnn_1_1_reshape_layer.html" title="This layer represents a reshape operation. ">ReshapeLayer</a> and a child <a class="el" href="classarmnn_1_1_reshape_layer.html" title="This layer represents a reshape operation. ">ReshapeLayer</a>. Inserts an equivalent <a class="el" href="classarmnn_1_1_reshape_layer.html" title="This layer represents a reshape operation. ">ReshapeLayer</a> that bypasses both for that connection. </p>
199
200<p class="definition">Definition at line <a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html#l00019">19</a> of file <a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html">OptimizeConsecutiveReshapes.hpp</a>.</p>
201
202<p class="reference">References <a class="el" href="_layer_8hpp_source.html#l00055">InputSlot::GetConnectedOutputSlot()</a>, <a class="el" href="_layer_8hpp_source.html#l00310">Layer::GetInputSlot()</a>, <a class="el" href="_layer_8hpp_source.html#l00305">Layer::GetName()</a>, <a class="el" href="_layer_8hpp_source.html#l00138">OutputSlot::GetNumConnections()</a>, <a class="el" href="_layer_8hpp_source.html#l00221">Layer::GetOutputHandler()</a>, <a class="el" href="_layer_8hpp_source.html#l00312">Layer::GetOutputSlot()</a>, <a class="el" href="_layer_8hpp_source.html#l00052">InputSlot::GetOwningLayer()</a>, <a class="el" href="_layer_8hpp_source.html#l00115">OutputSlot::GetOwningLayer()</a>, <a class="el" href="_tensor_8hpp_source.html#l00088">TensorInfo::GetShape()</a>, <a class="el" href="_output_handler_8hpp_source.html#l00044">OutputHandler::GetTensorInfo()</a>, <a class="el" href="_layer_8hpp_source.html#l00259">Layer::GetType()</a>, <a class="el" href="_graph_8hpp_source.html#l00409">Graph::InsertNewLayer()</a>, <a class="el" href="_layer_8cpp_source.html#l00112">OutputSlot::MoveAllConnections()</a>, <a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html#ae344e9caa07713681429bd8eea772ebf">OptimizeConsecutiveReshapesImpl::OptimizeConsecutiveReshapesImpl()</a>, <a class="el" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">armnn::Reshape</a>, <a class="el" href="_output_handler_8cpp_source.html#l00017">OutputHandler::SetTensorInfo()</a>, and <a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html#accd18a7450d91508ab6e12379507e01d">OptimizeConsecutiveReshapesImpl::~OptimizeConsecutiveReshapesImpl()</a>.</p>
203<div class="fragment"><div class="line"><a name="l00020"></a><span class="lineno"> 20</span>&#160; {</div><div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160; Layer&amp; base = connection.GetConnectedOutputSlot()-&gt;GetOwningLayer();</div><div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160; Layer&amp; child = connection.GetOwningLayer();</div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160;</div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160; BOOST_ASSERT(base.GetType() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">LayerType::Reshape</a>);</div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160; BOOST_ASSERT(child.GetType() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">LayerType::Reshape</a>);</div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160;</div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; OutputSlot* parentOut = base.GetInputSlot(0).GetConnectedOutputSlot();</div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160;</div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160; <span class="keyword">const</span> TensorInfo&amp; inInfo = parentOut-&gt;GetTensorInfo();</div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160; <span class="keyword">const</span> TensorInfo&amp; outInfo = child.GetOutputHandler().GetTensorInfo();</div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160;</div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160; <span class="comment">// This Optimization is only appropriate when the base ReshapeLayer is connected to the child ReshapeLayer</span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160; <span class="comment">// and no other Layer.</span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160; <span class="keywordflow">if</span> (base.GetOutputSlot(0).GetNumConnections() &gt; 1)</div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; {</div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; <span class="keywordflow">return</span>;</div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160; }</div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160;</div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; <span class="keywordflow">if</span> (inInfo.GetShape() != outInfo.GetShape())</div><div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; {</div><div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; <span class="comment">// Inserts equivalent reshape before base layer.</span></div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; <span class="keyword">const</span> std::string name = std::string(<span class="stringliteral">&quot;merged-&quot;</span>) + base.GetName() + std::string(<span class="stringliteral">&quot;-with-&quot;</span>) + child.GetName();</div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; <span class="keyword">const</span> ReshapeDescriptor descriptor{outInfo.GetShape()};</div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; <span class="keyword">auto</span>&amp; newReshape = *graph.InsertNewLayer&lt;ReshapeLayer&gt;(base.GetInputSlot(0), descriptor, name.c_str());</div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160;</div><div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; <span class="comment">// Sets tensor info for new layer.</span></div><div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; newReshape.GetOutputHandler().SetTensorInfo(outInfo);</div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; <span class="comment">// Parent is now the new layer.</span></div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; parentOut = &amp;newReshape.GetOutputSlot();</div><div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160; }</div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160;</div><div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160; <span class="comment">// Moves connections in child output to parent layer.</span></div><div class="line"><a name="l00053"></a><span class="lineno"> 53</span>&#160; <span class="comment">// Child layer will be removed as it&#39;s left unconnected.</span></div><div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160; <span class="comment">// Base layer will be removed if left unconnected.</span></div><div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; child.GetOutputSlot().MoveAllConnections(*parentOut);</div><div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; }</div><div class="ttc" id="namespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">armnn::LayerType::Reshape</a></div></div>
204</div><!-- fragment -->
205</div>
206</div>
207<hr/>The documentation for this class was generated from the following file:<ul>
208<li>src/armnn/optimizations/<a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html">OptimizeConsecutiveReshapes.hpp</a></li>
209</ul>
210</div><!-- contents -->
211</div><!-- doc-content -->
212<!-- start footer part -->
213<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
214 <ul>
215 <li class="navelem"><a class="el" href="namespacearmnn.html">armnn</a></li><li class="navelem"><a class="el" href="namespacearmnn_1_1optimizations.html">optimizations</a></li><li class="navelem"><a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html">OptimizeConsecutiveReshapesImpl</a></li>
216 <li class="footer">Generated on Fri Mar 13 2020 16:07:05 for ArmNN by
217 <a href="http://www.doxygen.org/index.html">
218 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.13 </li>
219 </ul>
220</div>
221</body>
222</html>