Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / refs/heads/gh-pages / . / latest / classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html
blob: 751a02d9c09f267c6b6d91a859ac406baeaaf333 [file] [log] [blame]
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]1<!-- HTML header for doxygen 1.8.17-->
2<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
3<html xmlns="http://www.w3.org/1999/xhtml">
4<head>
5<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
6<meta http-equiv="X-UA-Compatible" content="IE=9"/>
7<meta name="generator" content="Doxygen 1.8.17"/>
8<meta name="viewport" content="width=device-width, initial-scale=1"/>
9<title>Arm NN: OptimizeConsecutiveReshapesImpl Class Reference</title>
10<link href="tabs.css" rel="stylesheet" type="text/css"/>
11<script type="text/javascript" src="jquery.js"></script>
12<script type="text/javascript" src="dynsections.js"></script>
13<link href="navtree.css" rel="stylesheet" type="text/css"/>
14<script type="text/javascript" src="resize.js"></script>
15<script type="text/javascript" src="navtreedata.js"></script>
16<script type="text/javascript" src="navtree.js"></script>
17<link href="search/search.css" rel="stylesheet" type="text/css"/>
18<script type="text/javascript" src="search/searchdata.js"></script>
19<script type="text/javascript" src="search/search.js"></script>
20<script type="text/x-mathjax-config">
21 MathJax.Hub.Config({
22 extensions: ["tex2jax.js"],
23 jax: ["input/TeX","output/HTML-CSS"],
24});
25</script>
26<script type="text/javascript" async="async" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script>
27<link href="doxygen.css" rel="stylesheet" type="text/css" />
28<link href="customdoxygen.css" rel="stylesheet" type="text/css"/>
29</head>
30<body>
31<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
32<div id="titlearea">
33<table cellspacing="0" cellpadding="0">
34 <tbody>
35 <tr style="height: 56px;">
36 <img alt="ArmNN" src="Arm_NN_horizontal_blue.png" style="max-width: 15rem; margin-top: .5rem; margin-left 13px"/>
37 <td id="projectalign" style="padding-left: 0.9em;">
38 <div id="projectname">
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]39 &#160;<span id="projectnumber">24.05</span>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]40 </div>
41 </td>
42 </tr>
43 </tbody>
44</table>
45</div>
46<!-- end header part -->
47<!-- Generated by Doxygen 1.8.17 -->
48<script type="text/javascript">
49/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
50var searchBox = new SearchBox("searchBox", "search",false,'Search');
51/* @license-end */
52</script>
53<script type="text/javascript" src="menudata.js"></script>
54<script type="text/javascript" src="menu.js"></script>
55<script type="text/javascript">
56/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
57$(function() {
58 initMenu('',true,false,'search.php','Search');
59 $(document).ready(function() { init_search(); });
60});
61/* @license-end */</script>
62<div id="main-nav"></div>
63</div><!-- top -->
64<div id="side-nav" class="ui-resizable side-nav-resizable">
65 <div id="nav-tree">
66 <div id="nav-tree-contents">
67 <div id="nav-sync" class="sync"></div>
68 </div>
69 </div>
70 <div id="splitbar" style="-moz-user-select:none;"
71 class="ui-resizable-handle">
72 </div>
73</div>
74<script type="text/javascript">
75/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
76$(document).ready(function(){initNavTree('classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html',''); initResizable(); });
77/* @license-end */
78</script>
79<div id="doc-content">
80<!-- window showing the filter options -->
81<div id="MSearchSelectWindow"
82 onmouseover="return searchBox.OnSearchSelectShow()"
83 onmouseout="return searchBox.OnSearchSelectHide()"
84 onkeydown="return searchBox.OnSearchSelectKey(event)">
85</div>
86
87<!-- iframe showing the search results (closed by default) -->
88<div id="MSearchResultsWindow">
89<iframe src="javascript:void(0)" frameborder="0"
90 name="MSearchResults" id="MSearchResults">
91</iframe>
92</div>
93
94<div class="header">
95 <div class="summary">
96<a href="#pub-methods">Public Member Functions</a> &#124;
97<a href="#pro-methods">Protected Member Functions</a> &#124;
98<a href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl-members.html">List of all members</a> </div>
99 <div class="headertitle">
100<div class="title">OptimizeConsecutiveReshapesImpl Class Reference</div> </div>
101</div><!--header-->
102<div class="contents">
103
104<p><code>#include &lt;<a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html">OptimizeConsecutiveReshapes.hpp</a>&gt;</code></p>
105<table class="memberdecls">
106<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-methods"></a>
107Public Member Functions</h2></td></tr>
108<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>
109<tr class="memdesc:a5a8476ffc04ce7460bb09ad50d1d23de"><td class="mdescLeft">&#160;</td><td class="mdescRight">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>. <a href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html#a5a8476ffc04ce7460bb09ad50d1d23de">More...</a><br /></td></tr>
110<tr class="separator:a5a8476ffc04ce7460bb09ad50d1d23de"><td class="memSeparator" colspan="2">&#160;</td></tr>
111</table><table class="memberdecls">
112<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pro-methods"></a>
113Protected Member Functions</h2></td></tr>
114<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>
115<tr class="separator:ae344e9caa07713681429bd8eea772ebf"><td class="memSeparator" colspan="2">&#160;</td></tr>
116<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>
117<tr class="separator:accd18a7450d91508ab6e12379507e01d"><td class="memSeparator" colspan="2">&#160;</td></tr>
118</table>
119<a name="details" id="details"></a><h2 class="groupheader">Detailed Description</h2>
120<div class="textblock">
121<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>
122</div><h2 class="groupheader">Constructor &amp; Destructor Documentation</h2>
123<a id="ae344e9caa07713681429bd8eea772ebf"></a>
124<h2 class="memtitle"><span class="permalink"><a href="#ae344e9caa07713681429bd8eea772ebf">&#9670;&nbsp;</a></span>OptimizeConsecutiveReshapesImpl()</h2>
125
126<div class="memitem">
127<div class="memproto">
128<table class="mlabels">
129 <tr>
130 <td class="mlabels-left">
131 <table class="memname">
132 <tr>
133 <td class="memname"><a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html">OptimizeConsecutiveReshapesImpl</a> </td>
134 <td>(</td>
135 <td class="paramname"></td><td>)</td>
136 <td></td>
137 </tr>
138 </table>
139 </td>
140 <td class="mlabels-right">
141<span class="mlabels"><span class="mlabel">protected</span><span class="mlabel">default</span></span> </td>
142 </tr>
143</table>
144</div><div class="memdoc">
145
146</div>
147</div>
148<a id="accd18a7450d91508ab6e12379507e01d"></a>
149<h2 class="memtitle"><span class="permalink"><a href="#accd18a7450d91508ab6e12379507e01d">&#9670;&nbsp;</a></span>~OptimizeConsecutiveReshapesImpl()</h2>
150
151<div class="memitem">
152<div class="memproto">
153<table class="mlabels">
154 <tr>
155 <td class="mlabels-left">
156 <table class="memname">
157 <tr>
158 <td class="memname">~<a class="el" href="classarmnn_1_1optimizations_1_1_optimize_consecutive_reshapes_impl.html">OptimizeConsecutiveReshapesImpl</a> </td>
159 <td>(</td>
160 <td class="paramname"></td><td>)</td>
161 <td></td>
162 </tr>
163 </table>
164 </td>
165 <td class="mlabels-right">
166<span class="mlabels"><span class="mlabel">protected</span><span class="mlabel">default</span></span> </td>
167 </tr>
168</table>
169</div><div class="memdoc">
170
171</div>
172</div>
173<h2 class="groupheader">Member Function Documentation</h2>
174<a id="a5a8476ffc04ce7460bb09ad50d1d23de"></a>
175<h2 class="memtitle"><span class="permalink"><a href="#a5a8476ffc04ce7460bb09ad50d1d23de">&#9670;&nbsp;</a></span>Run()</h2>
176
177<div class="memitem">
178<div class="memproto">
179<table class="mlabels">
180 <tr>
181 <td class="mlabels-left">
182 <table class="memname">
183 <tr>
184 <td class="memname">void Run </td>
185 <td>(</td>
186 <td class="paramtype"><a class="el" href="classarmnn_1_1_graph.html">Graph</a> &amp;&#160;</td>
187 <td class="paramname"><em>graph</em>, </td>
188 </tr>
189 <tr>
190 <td class="paramkey"></td>
191 <td></td>
192 <td class="paramtype"><a class="el" href="classarmnn_1_1_input_slot.html">InputSlot</a> &amp;&#160;</td>
193 <td class="paramname"><em>connection</em>&#160;</td>
194 </tr>
195 <tr>
196 <td></td>
197 <td>)</td>
198 <td></td><td> const</td>
199 </tr>
200 </table>
201 </td>
202 <td class="mlabels-right">
203<span class="mlabels"><span class="mlabel">inline</span></span> </td>
204 </tr>
205</table>
206</div><div class="memdoc">
207
208<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>. </p>
209<p>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>
210
211<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>
212<div class="fragment"><div class="line"><a name="l00020"></a><span class="lineno"> 20</span>&#160; {</div>
213<div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160; Layer&amp; base = connection.GetConnectedOutputSlot()-&gt;GetOwningLayer();</div>
214<div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160; Layer&amp; child = connection.GetOwningLayer();</div>
215<div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160; </div>
216<div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160; <a class="code" href="_assert_8hpp.html#a5698be69cbd5dfe6c28fcd9867e8cbed">ARMNN_ASSERT</a>(base.GetType() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">LayerType::Reshape</a>);</div>
217<div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160; <a class="code" href="_assert_8hpp.html#a5698be69cbd5dfe6c28fcd9867e8cbed">ARMNN_ASSERT</a>(child.GetType() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">LayerType::Reshape</a>);</div>
218<div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160; </div>
219<div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; OutputSlot* parentOut = base.GetInputSlot(0).GetConnectedOutputSlot();</div>
220<div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; </div>
221<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>
222<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>
223<div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160; </div>
224<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>
225<div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160; <span class="comment">// and no other Layer.</span></div>
226<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>
227<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; {</div>
228<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; <span class="keywordflow">return</span>;</div>
229<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160; }</div>
230<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160; </div>
231<div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; <span class="keywordflow">if</span> (inInfo.GetShape() != outInfo.GetShape())</div>
232<div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; {</div>
233<div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; <span class="comment">// Inserts equivalent reshape before base layer.</span></div>
234<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>
235<div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; <span class="keyword">const</span> ReshapeDescriptor descriptor{outInfo.GetShape()};</div>
236<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>
237<div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; </div>
238<div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; <span class="comment">// Parent is now the new layer.</span></div>
239<div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; parentOut = &amp;newReshape.GetOutputSlot();</div>
240<div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; }</div>
241<div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; </div>
242<div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160; <span class="comment">// Moves connections in child output to parent layer.</span></div>
243<div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160; <span class="comment">// Child layer will be removed as it&#39;s left unconnected.</span></div>
244<div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160; <span class="comment">// Base layer will be removed if left unconnected.</span></div>
245<div class="line"><a name="l00053"></a><span class="lineno"> 53</span>&#160; child.GetOutputSlot().MoveAllConnections(*parentOut);</div>
246<div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160; }</div>
247</div><!-- fragment -->
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]248<p class="reference">References <a class="el" href="_assert_8hpp_source.html#l00014">ARMNN_ASSERT</a>, <a class="el" href="_layer_8hpp_source.html#l00056">InputSlot::GetConnectedOutputSlot()</a>, <a class="el" href="_layer_8hpp_source.html#l00337">Layer::GetInputSlot()</a>, <a class="el" href="_layer_8hpp_source.html#l00332">Layer::GetName()</a>, <a class="el" href="_layer_8hpp_source.html#l00158">OutputSlot::GetNumConnections()</a>, <a class="el" href="_layer_8hpp_source.html#l00245">Layer::GetOutputHandler()</a>, <a class="el" href="_layer_8hpp_source.html#l00339">Layer::GetOutputSlot()</a>, <a class="el" href="_layer_8hpp_source.html#l00053">InputSlot::GetOwningLayer()</a>, <a class="el" href="_layer_8hpp_source.html#l00132">OutputSlot::GetOwningLayer()</a>, <a class="el" href="_tensor_8hpp_source.html#l00193">TensorInfo::GetShape()</a>, <a class="el" href="_output_handler_8hpp_source.html#l00042">OutputHandler::GetTensorInfo()</a>, <a class="el" href="_layer_8cpp_source.html#l00100">OutputSlot::GetTensorInfo()</a>, <a class="el" href="_layer_8hpp_source.html#l00286">Layer::GetType()</a>, <a class="el" href="_graph_8hpp_source.html#l00481">Graph::InsertNewLayer()</a>, <a class="el" href="_layer_8cpp_source.html#l00156">OutputSlot::MoveAllConnections()</a>, and <a class="el" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">armnn::Reshape</a>.</p>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]249
250</div>
251</div>
252<hr/>The documentation for this class was generated from the following file:<ul>
253<li>src/armnn/optimizations/<a class="el" href="_optimize_consecutive_reshapes_8hpp_source.html">OptimizeConsecutiveReshapes.hpp</a></li>
254</ul>
255</div><!-- contents -->
256</div><!-- doc-content -->
257<div class="ttc" id="a_assert_8hpp_html_a5698be69cbd5dfe6c28fcd9867e8cbed"><div class="ttname"><a href="_assert_8hpp.html#a5698be69cbd5dfe6c28fcd9867e8cbed">ARMNN_ASSERT</a></div><div class="ttdeci">#define ARMNN_ASSERT(COND)</div><div class="ttdef"><b>Definition:</b> <a href="_assert_8hpp_source.html#l00014">Assert.hpp:14</a></div></div>
258<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa7c59ccedc6a3bd90c17f3b990afefad">armnn::LayerType::Reshape</a></div><div class="ttdeci">@ Reshape</div></div>
259<!-- start footer part -->
260<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
261 <ul>
262 <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>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]263 <li class="footer">Generated on Thu May 16 2024 09:32:00 for Arm NN by
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]264 <a href="http://www.doxygen.org/index.html">
265 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li>
266 </ul>
267</div>
268</body>
269</html>