Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / refs/heads/gh-pages / . / latest / structarmnn_1_1_broadcast_loop.html
blob: 6ecf5effaf88f28292ba490d38d4361a611a549f [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: BroadcastLoop Struct 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('structarmnn_1_1_broadcast_loop.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="#nested-classes">Classes</a> &#124;
97<a href="#pub-methods">Public Member Functions</a> &#124;
98<a href="structarmnn_1_1_broadcast_loop-members.html">List of all members</a> </div>
99 <div class="headertitle">
100<div class="title">BroadcastLoop Struct Reference</div> </div>
101</div><!--header-->
102<div class="contents">
103
104<p><code>#include &lt;<a class="el" href="_broadcast_8hpp_source.html">Broadcast.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:a9c28d135714489bab0f9c251ca66a845"><td class="memItemLeft" align="right" valign="top">&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="structarmnn_1_1_broadcast_loop.html#a9c28d135714489bab0f9c251ca66a845">BroadcastLoop</a> (const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;inShape0, const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;inShape1, const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;outShape)</td></tr>
109<tr class="separator:a9c28d135714489bab0f9c251ca66a845"><td class="memSeparator" colspan="2">&#160;</td></tr>
110<tr class="memitem:ac49ed66fded9e92273a7be4c54d6f6ef"><td class="memItemLeft" align="right" valign="top">&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="structarmnn_1_1_broadcast_loop.html#ac49ed66fded9e92273a7be4c54d6f6ef">BroadcastLoop</a> (const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;inShape, const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;outShape)</td></tr>
111<tr class="separator:ac49ed66fded9e92273a7be4c54d6f6ef"><td class="memSeparator" colspan="2">&#160;</td></tr>
112<tr class="memitem:a0ed4f8b0f36ff2a60cbc4faf3be5d2f7"><td class="memItemLeft" align="right" valign="top">unsigned int&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="structarmnn_1_1_broadcast_loop.html#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">GetNumDimensions</a> ()</td></tr>
113<tr class="separator:a0ed4f8b0f36ff2a60cbc4faf3be5d2f7"><td class="memSeparator" colspan="2">&#160;</td></tr>
114<tr class="memitem:ae9f47e12d6c2265d33fed54e88cf179f"><td class="memTemplParams" colspan="2">template&lt;typename Func , typename DecoderOp , typename EncoderOp &gt; </td></tr>
115<tr class="memitem:ae9f47e12d6c2265d33fed54e88cf179f"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="structarmnn_1_1_broadcast_loop.html#ae9f47e12d6c2265d33fed54e88cf179f">Unroll</a> (Func operationFunc, unsigned int dimension, DecoderOp &amp;inData0, DecoderOp &amp;inData1, EncoderOp &amp;outData)</td></tr>
116<tr class="separator:ae9f47e12d6c2265d33fed54e88cf179f"><td class="memSeparator" colspan="2">&#160;</td></tr>
117<tr class="memitem:a1b3b461824ae2769251ad66a09da5b52"><td class="memTemplParams" colspan="2">template&lt;typename Func , typename DecoderOp , typename EncoderOp &gt; </td></tr>
118<tr class="memitem:a1b3b461824ae2769251ad66a09da5b52"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="structarmnn_1_1_broadcast_loop.html#a1b3b461824ae2769251ad66a09da5b52">Unroll</a> (Func operationFunc, unsigned int dimension, DecoderOp &amp;inData, EncoderOp &amp;outData)</td></tr>
119<tr class="separator:a1b3b461824ae2769251ad66a09da5b52"><td class="memSeparator" colspan="2">&#160;</td></tr>
120</table>
121<a name="details" id="details"></a><h2 class="groupheader">Detailed Description</h2>
122<div class="textblock">
123<p class="definition">Definition at line <a class="el" href="_broadcast_8hpp_source.html#l00014">14</a> of file <a class="el" href="_broadcast_8hpp_source.html">Broadcast.hpp</a>.</p>
124</div><h2 class="groupheader">Constructor &amp; Destructor Documentation</h2>
125<a id="a9c28d135714489bab0f9c251ca66a845"></a>
126<h2 class="memtitle"><span class="permalink"><a href="#a9c28d135714489bab0f9c251ca66a845">&#9670;&nbsp;</a></span>BroadcastLoop() <span class="overload">[1/2]</span></h2>
127
128<div class="memitem">
129<div class="memproto">
130 <table class="memname">
131 <tr>
132 <td class="memname"><a class="el" href="structarmnn_1_1_broadcast_loop.html">BroadcastLoop</a> </td>
133 <td>(</td>
134 <td class="paramtype">const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;&#160;</td>
135 <td class="paramname"><em>inShape0</em>, </td>
136 </tr>
137 <tr>
138 <td class="paramkey"></td>
139 <td></td>
140 <td class="paramtype">const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;&#160;</td>
141 <td class="paramname"><em>inShape1</em>, </td>
142 </tr>
143 <tr>
144 <td class="paramkey"></td>
145 <td></td>
146 <td class="paramtype">const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;&#160;</td>
147 <td class="paramname"><em>outShape</em>&#160;</td>
148 </tr>
149 <tr>
150 <td></td>
151 <td>)</td>
152 <td></td><td></td>
153 </tr>
154 </table>
155</div><div class="memdoc">
156
157<p class="definition">Definition at line <a class="el" href="_broadcast_8cpp_source.html#l00011">11</a> of file <a class="el" href="_broadcast_8cpp_source.html">Broadcast.cpp</a>.</p>
158<div class="fragment"><div class="line"><a name="l00012"></a><span class="lineno"> 12</span>&#160;: m_DimData(outShape.GetNumDimensions())</div>
159<div class="line"><a name="l00013"></a><span class="lineno"> 13</span>&#160;{</div>
160<div class="line"><a name="l00014"></a><span class="lineno"> 14</span>&#160; <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> numDims = <a class="code" href="structarmnn_1_1_broadcast_loop.html#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">GetNumDimensions</a>();</div>
161<div class="line"><a name="l00015"></a><span class="lineno"> 15</span>&#160; </div>
162<div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> sIn0 = 1;</div>
163<div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> sIn1 = 1;</div>
164<div class="line"><a name="l00018"></a><span class="lineno"> 18</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> sOut = 1;</div>
165<div class="line"><a name="l00019"></a><span class="lineno"> 19</span>&#160; </div>
166<div class="line"><a name="l00020"></a><span class="lineno"> 20</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> j = numDims - 1, k = 0; k &lt; numDims ; k++, j--)</div>
167<div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160; {</div>
168<div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160; m_DimData[j].m_DimSize = outShape[j];</div>
169<div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160; m_DimData[j].m_Stride1 = (inShape0[j] &gt; 1) ? sIn0 : 0;</div>
170<div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160; m_DimData[j].m_Stride2 = (inShape1[j] &gt; 1) ? sIn1 : 0;</div>
171<div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160; m_DimData[j].m_StrideOut = sOut;</div>
172<div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160; </div>
173<div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; sIn0 *= inShape0[j];</div>
174<div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; sIn1 *= inShape1[j];</div>
175<div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160; sOut *= outShape[j];</div>
176<div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160; }</div>
177<div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160;}</div>
178</div><!-- fragment -->
179<p class="reference">References <a class="el" href="_broadcast_8hpp_source.html#l00020">BroadcastLoop::GetNumDimensions()</a>.</p>
180
181</div>
182</div>
183<a id="ac49ed66fded9e92273a7be4c54d6f6ef"></a>
184<h2 class="memtitle"><span class="permalink"><a href="#ac49ed66fded9e92273a7be4c54d6f6ef">&#9670;&nbsp;</a></span>BroadcastLoop() <span class="overload">[2/2]</span></h2>
185
186<div class="memitem">
187<div class="memproto">
188 <table class="memname">
189 <tr>
190 <td class="memname"><a class="el" href="structarmnn_1_1_broadcast_loop.html">BroadcastLoop</a> </td>
191 <td>(</td>
192 <td class="paramtype">const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;&#160;</td>
193 <td class="paramname"><em>inShape</em>, </td>
194 </tr>
195 <tr>
196 <td class="paramkey"></td>
197 <td></td>
198 <td class="paramtype">const <a class="el" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> &amp;&#160;</td>
199 <td class="paramname"><em>outShape</em>&#160;</td>
200 </tr>
201 <tr>
202 <td></td>
203 <td>)</td>
204 <td></td><td></td>
205 </tr>
206 </table>
207</div><div class="memdoc">
208
209<p class="definition">Definition at line <a class="el" href="_broadcast_8cpp_source.html#l00033">33</a> of file <a class="el" href="_broadcast_8cpp_source.html">Broadcast.cpp</a>.</p>
210<div class="fragment"><div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160;: m_DimData(outShape.GetNumDimensions())</div>
211<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160;{</div>
212<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> numDims = <a class="code" href="structarmnn_1_1_broadcast_loop.html#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">GetNumDimensions</a>();</div>
213<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160; </div>
214<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> sIn = 1;</div>
215<div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> sOut = 1;</div>
216<div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; </div>
217<div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> j = numDims - 1, k = 0; k &lt; numDims ; k++, j--)</div>
218<div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; {</div>
219<div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; m_DimData[j].m_DimSize = outShape[j];</div>
220<div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; m_DimData[j].m_Stride1 = (inShape[j] &gt; 1) ? sIn : 0;</div>
221<div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; m_DimData[j].m_StrideOut = sOut;</div>
222<div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; </div>
223<div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; sIn *= inShape[j];</div>
224<div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; sOut *= outShape[j];</div>
225<div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; }</div>
226<div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160;}</div>
227</div><!-- fragment -->
228<p class="reference">References <a class="el" href="_broadcast_8hpp_source.html#l00020">BroadcastLoop::GetNumDimensions()</a>.</p>
229
230</div>
231</div>
232<h2 class="groupheader">Member Function Documentation</h2>
233<a id="a0ed4f8b0f36ff2a60cbc4faf3be5d2f7"></a>
234<h2 class="memtitle"><span class="permalink"><a href="#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">&#9670;&nbsp;</a></span>GetNumDimensions()</h2>
235
236<div class="memitem">
237<div class="memproto">
238<table class="mlabels">
239 <tr>
240 <td class="mlabels-left">
241 <table class="memname">
242 <tr>
243 <td class="memname">unsigned int GetNumDimensions </td>
244 <td>(</td>
245 <td class="paramname"></td><td>)</td>
246 <td></td>
247 </tr>
248 </table>
249 </td>
250 <td class="mlabels-right">
251<span class="mlabels"><span class="mlabel">inline</span></span> </td>
252 </tr>
253</table>
254</div><div class="memdoc">
255
256<p class="definition">Definition at line <a class="el" href="_broadcast_8hpp_source.html#l00020">20</a> of file <a class="el" href="_broadcast_8hpp_source.html">Broadcast.hpp</a>.</p>
257<div class="fragment"><div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160; {</div>
258<div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160; <span class="keywordflow">return</span> <span class="keyword">static_cast&lt;</span><span class="keywordtype">unsigned</span> <span class="keywordtype">int</span><span class="keyword">&gt;</span>(m_DimData.size());</div>
259<div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160; }</div>
260</div><!-- fragment -->
261<p class="reference">Referenced by <a class="el" href="_broadcast_8cpp_source.html#l00011">BroadcastLoop::BroadcastLoop()</a>, and <a class="el" href="_broadcast_8hpp_source.html#l00026">BroadcastLoop::Unroll()</a>.</p>
262
263</div>
264</div>
265<a id="a1b3b461824ae2769251ad66a09da5b52"></a>
266<h2 class="memtitle"><span class="permalink"><a href="#a1b3b461824ae2769251ad66a09da5b52">&#9670;&nbsp;</a></span>Unroll() <span class="overload">[1/2]</span></h2>
267
268<div class="memitem">
269<div class="memproto">
270<table class="mlabels">
271 <tr>
272 <td class="mlabels-left">
273 <table class="memname">
274 <tr>
275 <td class="memname">void Unroll </td>
276 <td>(</td>
277 <td class="paramtype">Func&#160;</td>
278 <td class="paramname"><em>operationFunc</em>, </td>
279 </tr>
280 <tr>
281 <td class="paramkey"></td>
282 <td></td>
283 <td class="paramtype">unsigned int&#160;</td>
284 <td class="paramname"><em>dimension</em>, </td>
285 </tr>
286 <tr>
287 <td class="paramkey"></td>
288 <td></td>
289 <td class="paramtype">DecoderOp &amp;&#160;</td>
290 <td class="paramname"><em>inData</em>, </td>
291 </tr>
292 <tr>
293 <td class="paramkey"></td>
294 <td></td>
295 <td class="paramtype">EncoderOp &amp;&#160;</td>
296 <td class="paramname"><em>outData</em>&#160;</td>
297 </tr>
298 <tr>
299 <td></td>
300 <td>)</td>
301 <td></td><td></td>
302 </tr>
303 </table>
304 </td>
305 <td class="mlabels-right">
306<span class="mlabels"><span class="mlabel">inline</span></span> </td>
307 </tr>
308</table>
309</div><div class="memdoc">
310
311<p class="definition">Definition at line <a class="el" href="_broadcast_8hpp_source.html#l00062">62</a> of file <a class="el" href="_broadcast_8hpp_source.html">Broadcast.hpp</a>.</p>
312<div class="fragment"><div class="line"><a name="l00066"></a><span class="lineno"> 66</span>&#160; {</div>
313<div class="line"><a name="l00067"></a><span class="lineno"> 67</span>&#160; <span class="keywordflow">if</span> (dimension &gt;= <a class="code" href="structarmnn_1_1_broadcast_loop.html#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">GetNumDimensions</a>())</div>
314<div class="line"><a name="l00068"></a><span class="lineno"> 68</span>&#160; {</div>
315<div class="line"><a name="l00069"></a><span class="lineno"> 69</span>&#160; outData.Set(operationFunc(inData.Get()));</div>
316<div class="line"><a name="l00070"></a><span class="lineno"> 70</span>&#160; <span class="keywordflow">return</span>;</div>
317<div class="line"><a name="l00071"></a><span class="lineno"> 71</span>&#160; }</div>
318<div class="line"><a name="l00072"></a><span class="lineno"> 72</span>&#160; </div>
319<div class="line"><a name="l00073"></a><span class="lineno"> 73</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> inDataMovement = 0;</div>
320<div class="line"><a name="l00074"></a><span class="lineno"> 74</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> outDataMovement = 0;</div>
321<div class="line"><a name="l00075"></a><span class="lineno"> 75</span>&#160; </div>
322<div class="line"><a name="l00076"></a><span class="lineno"> 76</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; m_DimData[dimension].m_DimSize; i++)</div>
323<div class="line"><a name="l00077"></a><span class="lineno"> 77</span>&#160; {</div>
324<div class="line"><a name="l00078"></a><span class="lineno"> 78</span>&#160; <a class="code" href="structarmnn_1_1_broadcast_loop.html#ae9f47e12d6c2265d33fed54e88cf179f">Unroll</a>(operationFunc, dimension + 1, inData, outData);</div>
325<div class="line"><a name="l00079"></a><span class="lineno"> 79</span>&#160; </div>
326<div class="line"><a name="l00080"></a><span class="lineno"> 80</span>&#160; inData += m_DimData[dimension].m_Stride1;</div>
327<div class="line"><a name="l00081"></a><span class="lineno"> 81</span>&#160; outData += m_DimData[dimension].m_StrideOut;</div>
328<div class="line"><a name="l00082"></a><span class="lineno"> 82</span>&#160; </div>
329<div class="line"><a name="l00083"></a><span class="lineno"> 83</span>&#160; inDataMovement += m_DimData[dimension].m_Stride1;</div>
330<div class="line"><a name="l00084"></a><span class="lineno"> 84</span>&#160; outDataMovement += m_DimData[dimension].m_StrideOut;</div>
331<div class="line"><a name="l00085"></a><span class="lineno"> 85</span>&#160; }</div>
332<div class="line"><a name="l00086"></a><span class="lineno"> 86</span>&#160; </div>
333<div class="line"><a name="l00087"></a><span class="lineno"> 87</span>&#160; <span class="comment">// move iterator back to the start</span></div>
334<div class="line"><a name="l00088"></a><span class="lineno"> 88</span>&#160; inData -= inDataMovement;</div>
335<div class="line"><a name="l00089"></a><span class="lineno"> 89</span>&#160; outData -= outDataMovement;</div>
336<div class="line"><a name="l00090"></a><span class="lineno"> 90</span>&#160; }</div>
337</div><!-- fragment -->
338<p class="reference">References <a class="el" href="_broadcast_8hpp_source.html#l00020">BroadcastLoop::GetNumDimensions()</a>, and <a class="el" href="_broadcast_8hpp_source.html#l00026">BroadcastLoop::Unroll()</a>.</p>
339
340</div>
341</div>
342<a id="ae9f47e12d6c2265d33fed54e88cf179f"></a>
343<h2 class="memtitle"><span class="permalink"><a href="#ae9f47e12d6c2265d33fed54e88cf179f">&#9670;&nbsp;</a></span>Unroll() <span class="overload">[2/2]</span></h2>
344
345<div class="memitem">
346<div class="memproto">
347<table class="mlabels">
348 <tr>
349 <td class="mlabels-left">
350 <table class="memname">
351 <tr>
352 <td class="memname">void Unroll </td>
353 <td>(</td>
354 <td class="paramtype">Func&#160;</td>
355 <td class="paramname"><em>operationFunc</em>, </td>
356 </tr>
357 <tr>
358 <td class="paramkey"></td>
359 <td></td>
360 <td class="paramtype">unsigned int&#160;</td>
361 <td class="paramname"><em>dimension</em>, </td>
362 </tr>
363 <tr>
364 <td class="paramkey"></td>
365 <td></td>
366 <td class="paramtype">DecoderOp &amp;&#160;</td>
367 <td class="paramname"><em>inData0</em>, </td>
368 </tr>
369 <tr>
370 <td class="paramkey"></td>
371 <td></td>
372 <td class="paramtype">DecoderOp &amp;&#160;</td>
373 <td class="paramname"><em>inData1</em>, </td>
374 </tr>
375 <tr>
376 <td class="paramkey"></td>
377 <td></td>
378 <td class="paramtype">EncoderOp &amp;&#160;</td>
379 <td class="paramname"><em>outData</em>&#160;</td>
380 </tr>
381 <tr>
382 <td></td>
383 <td>)</td>
384 <td></td><td></td>
385 </tr>
386 </table>
387 </td>
388 <td class="mlabels-right">
389<span class="mlabels"><span class="mlabel">inline</span></span> </td>
390 </tr>
391</table>
392</div><div class="memdoc">
393
394<p class="definition">Definition at line <a class="el" href="_broadcast_8hpp_source.html#l00026">26</a> of file <a class="el" href="_broadcast_8hpp_source.html">Broadcast.hpp</a>.</p>
395<div class="fragment"><div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160; {</div>
396<div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160; <span class="keywordflow">if</span> (dimension &gt;= <a class="code" href="structarmnn_1_1_broadcast_loop.html#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">GetNumDimensions</a>())</div>
397<div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160; {</div>
398<div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160; outData.Set(operationFunc(inData0.Get(), inData1.Get()));</div>
399<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; <span class="keywordflow">return</span>;</div>
400<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; }</div>
401<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160; </div>
402<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> inData0Movement = 0;</div>
403<div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> inData1Movement = 0;</div>
404<div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> outDataMovement = 0;</div>
405<div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; </div>
406<div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; m_DimData[dimension].m_DimSize; i++)</div>
407<div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; {</div>
408<div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; <a class="code" href="structarmnn_1_1_broadcast_loop.html#ae9f47e12d6c2265d33fed54e88cf179f">Unroll</a>(operationFunc, dimension + 1, inData0, inData1, outData);</div>
409<div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; </div>
410<div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; inData0 += m_DimData[dimension].m_Stride1;</div>
411<div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; inData1 += m_DimData[dimension].m_Stride2;</div>
412<div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; outData += m_DimData[dimension].m_StrideOut;</div>
413<div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; </div>
414<div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160; inData0Movement += m_DimData[dimension].m_Stride1;</div>
415<div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160; inData1Movement += m_DimData[dimension].m_Stride2;</div>
416<div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160; outDataMovement += m_DimData[dimension].m_StrideOut;</div>
417<div class="line"><a name="l00053"></a><span class="lineno"> 53</span>&#160; }</div>
418<div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160; </div>
419<div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; <span class="comment">// move iterator back to the start</span></div>
420<div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; inData0 -= inData0Movement;</div>
421<div class="line"><a name="l00057"></a><span class="lineno"> 57</span>&#160; inData1 -= inData1Movement;</div>
422<div class="line"><a name="l00058"></a><span class="lineno"> 58</span>&#160; outData -= outDataMovement;</div>
423<div class="line"><a name="l00059"></a><span class="lineno"> 59</span>&#160; }</div>
424</div><!-- fragment -->
425<p class="reference">References <a class="el" href="_broadcast_8hpp_source.html#l00020">BroadcastLoop::GetNumDimensions()</a>.</p>
426
427<p class="reference">Referenced by <a class="el" href="_elementwise_function_8cpp_source.html#l00025">ElementwiseBinaryFunction&lt; Functor &gt;::ElementwiseBinaryFunction()</a>, <a class="el" href="_elementwise_function_8cpp_source.html#l00036">ElementwiseUnaryFunction&lt; Functor &gt;::ElementwiseUnaryFunction()</a>, <a class="el" href="_elementwise_function_8cpp_source.html#l00045">LogicalBinaryFunction&lt; Functor &gt;::LogicalBinaryFunction()</a>, <a class="el" href="_elementwise_function_8cpp_source.html#l00056">LogicalUnaryFunction&lt; Functor &gt;::LogicalUnaryFunction()</a>, <a class="el" href="_prelu_impl_8cpp_source.html#l00013">armnn::PreluImpl()</a>, and <a class="el" href="_broadcast_8hpp_source.html#l00062">BroadcastLoop::Unroll()</a>.</p>
428
429</div>
430</div>
431<hr/>The documentation for this struct was generated from the following files:<ul>
432<li>src/backends/reference/workloads/<a class="el" href="_broadcast_8hpp_source.html">Broadcast.hpp</a></li>
433<li>src/backends/reference/workloads/<a class="el" href="_broadcast_8cpp_source.html">Broadcast.cpp</a></li>
434</ul>
435</div><!-- contents -->
436</div><!-- doc-content -->
437<div class="ttc" id="astructarmnn_1_1_broadcast_loop_html_ae9f47e12d6c2265d33fed54e88cf179f"><div class="ttname"><a href="structarmnn_1_1_broadcast_loop.html#ae9f47e12d6c2265d33fed54e88cf179f">armnn::BroadcastLoop::Unroll</a></div><div class="ttdeci">void Unroll(Func operationFunc, unsigned int dimension, DecoderOp &amp;inData0, DecoderOp &amp;inData1, EncoderOp &amp;outData)</div><div class="ttdef"><b>Definition:</b> <a href="_broadcast_8hpp_source.html#l00026">Broadcast.hpp:26</a></div></div>
438<div class="ttc" id="astructarmnn_1_1_broadcast_loop_html_a0ed4f8b0f36ff2a60cbc4faf3be5d2f7"><div class="ttname"><a href="structarmnn_1_1_broadcast_loop.html#a0ed4f8b0f36ff2a60cbc4faf3be5d2f7">armnn::BroadcastLoop::GetNumDimensions</a></div><div class="ttdeci">unsigned int GetNumDimensions()</div><div class="ttdef"><b>Definition:</b> <a href="_broadcast_8hpp_source.html#l00020">Broadcast.hpp:20</a></div></div>
439<!-- start footer part -->
440<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
441 <ul>
442 <li class="navelem"><a class="el" href="namespacearmnn.html">armnn</a></li><li class="navelem"><a class="el" href="structarmnn_1_1_broadcast_loop.html">BroadcastLoop</a></li>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]443 <li class="footer">Generated on Thu May 16 2024 09:31:54 for Arm NN by
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]444 <a href="http://www.doxygen.org/index.html">
445 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li>
446 </ul>
447</div>
448</body>
449</html>