Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / refs/heads/gh-pages / . / latest / _subgraph_utils_8hpp_source.html
blob: d2ca036b6e326e1228660147a842e5d9eb27380b [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: src/backends/backendsCommon/SubgraphUtils.hpp Source File</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('_subgraph_utils_8hpp_source.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="headertitle">
96<div class="title">SubgraphUtils.hpp</div> </div>
97</div><!--header-->
98<div class="contents">
99<a href="_subgraph_utils_8hpp.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span>&#160;<span class="comment">//</span></div>
100<div class="line"><a name="l00002"></a><span class="lineno"> 2</span>&#160;<span class="comment">// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved.</span></div>
101<div class="line"><a name="l00003"></a><span class="lineno"> 3</span>&#160;<span class="comment">// SPDX-License-Identifier: MIT</span></div>
102<div class="line"><a name="l00004"></a><span class="lineno"> 4</span>&#160;<span class="comment">//</span></div>
103<div class="line"><a name="l00005"></a><span class="lineno"> 5</span>&#160; </div>
104<div class="line"><a name="l00006"></a><span class="lineno"> 6</span>&#160;<span class="preprocessor">#pragma once</span></div>
105<div class="line"><a name="l00007"></a><span class="lineno"> 7</span>&#160; </div>
106<div class="line"><a name="l00008"></a><span class="lineno"> 8</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="_strategy_base_8hpp.html">armnn/StrategyBase.hpp</a>&gt;</span></div>
107<div class="line"><a name="l00009"></a><span class="lineno"> 9</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="_descriptors_8hpp.html">armnn/Descriptors.hpp</a>&gt;</span></div>
108<div class="line"><a name="l00010"></a><span class="lineno"> 10</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="_fold_pad_into_layer2d_8hpp.html">optimizations/FoldPadIntoLayer2d.hpp</a>&gt;</span></div>
109<div class="line"><a name="l00011"></a><span class="lineno"> 11</span>&#160; </div>
110<div class="line"><a name="l00012"></a><span class="lineno"> 12</span>&#160;<span class="keyword">namespace </span><a class="code" href="namespacearmnn.html">armnn</a></div>
111<div class="line"><a name="l00013"></a><span class="lineno"> 13</span>&#160;{</div>
112<div class="line"><a name="l00014"></a><span class="lineno"> 14</span>&#160; </div>
113<div class="line"><a name="l00015"></a><span class="lineno"> 15</span>&#160;<span class="keyword">namespace</span></div>
114<div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160;{</div>
115<div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160;<span class="comment"></span> </div>
116<div class="line"><a name="l00018"></a><span class="lineno"> 18</span>&#160;<span class="comment">/// Checks if a Layer has a DataLayout that is either NCHW or NCDHW.</span></div>
117<div class="line"><a name="l00019"></a><span class="lineno"> 19</span>&#160;<span class="comment"></span><span class="keyword">class </span>CheckForNCHW : <span class="keyword">public</span> StrategyBase&lt;NoThrowStrategy&gt;</div>
118<div class="line"><a name="l00020"></a><span class="lineno"> 20</span>&#160;{</div>
119<div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160;<span class="keyword">public</span>:</div>
120<div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160; CheckForNCHW()</div>
121<div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160; {}</div>
122<div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160; </div>
123<div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160; <span class="keywordtype">void</span> ExecuteStrategy(<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer,</div>
124<div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160; <span class="keyword">const</span> <a class="code" href="structarmnn_1_1_base_descriptor.html">armnn::BaseDescriptor</a>&amp; descriptor,</div>
125<div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; <span class="keyword">const</span> std::vector&lt;armnn::ConstTensor&gt;&amp; constants,</div>
126<div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; <span class="keyword">const</span> <span class="keywordtype">char</span>* name,</div>
127<div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160; <span class="keyword">const</span> <a class="code" href="namespacearmnn.html#ab8cf8f9fb6792e654c2d8d8382f6f01b">armnn::LayerBindingId</a> <span class="keywordtype">id</span> = 0)<span class="keyword"> override</span></div>
128<div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160;<span class="keyword"> </span>{</div>
129<div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160; <a class="code" href="namespacearmnn.html#a44affeeb090c3c6a3062830562672e84">armnn::IgnoreUnused</a>(layer, constants, <span class="keywordtype">id</span>, name);</div>
130<div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160; <span class="keywordflow">switch</span> (layer-&gt;<a class="code" href="classarmnn_1_1_i_connectable_layer.html#adceb04ae84c524e4d01881e3754a4d59">GetType</a>())</div>
131<div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160; {</div>
132<div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a9882ff3cfed27d6161c20a305e7a3484">armnn::LayerType::BatchMatMul</a>:</div>
133<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; {</div>
134<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; <span class="keyword">auto</span> desc = <span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_batch_mat_mul_descriptor.html">armnn::BatchMatMulDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor);</div>
135<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160; m_Result = desc.m_DataLayoutX == <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">DataLayout::NCHW</a> || desc.m_DataLayoutY == <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">DataLayout::NCHW</a>;</div>
136<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160; <span class="keywordflow">break</span>;</div>
137<div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; }</div>
138<div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4ae4743c3ec15d1d84169b17264634692e">armnn::LayerType::BatchNormalization</a>:</div>
139<div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; {</div>
140<div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_batch_normalization_descriptor.html">armnn::BatchNormalizationDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
141<div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; <span class="keywordflow">break</span>;</div>
142<div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; }</div>
143<div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a6ee06c6045d0c5b6565a247955ef0fc2">armnn::LayerType::BatchToSpaceNd</a>:</div>
144<div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; {</div>
145<div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_batch_to_space_nd_descriptor.html">armnn::BatchToSpaceNdDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
146<div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; <span class="keywordflow">break</span>;</div>
147<div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; }</div>
148<div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4adb033d2f81b68f9a17e8f62de69fed4a">armnn::LayerType::Convolution2d</a>:</div>
149<div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160; {</div>
150<div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_convolution2d_descriptor.html">armnn::Convolution2dDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
151<div class="line"><a name="l00053"></a><span class="lineno"> 53</span>&#160; <span class="keywordflow">break</span>;</div>
152<div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160; }</div>
153<div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a583550d0f265fd3756f7de0e42c51953">armnn::LayerType::Convolution3d</a>:</div>
154<div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; {</div>
155<div class="line"><a name="l00057"></a><span class="lineno"> 57</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_convolution3d_descriptor.html">armnn::Convolution3dDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
156<div class="line"><a name="l00058"></a><span class="lineno"> 58</span>&#160; <span class="keywordflow">break</span>;</div>
157<div class="line"><a name="l00059"></a><span class="lineno"> 59</span>&#160; }</div>
158<div class="line"><a name="l00060"></a><span class="lineno"> 60</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4af97adbfc88b7012a0243215b1076e7e7">armnn::LayerType::DepthwiseConvolution2d</a>:</div>
159<div class="line"><a name="l00061"></a><span class="lineno"> 61</span>&#160; {</div>
160<div class="line"><a name="l00062"></a><span class="lineno"> 62</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html">armnn::DepthwiseConvolution2dDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
161<div class="line"><a name="l00063"></a><span class="lineno"> 63</span>&#160; <span class="keywordflow">break</span>;</div>
162<div class="line"><a name="l00064"></a><span class="lineno"> 64</span>&#160; }</div>
163<div class="line"><a name="l00065"></a><span class="lineno"> 65</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a21baa4498161d195f5bb2e3627344ba4">armnn::LayerType::InstanceNormalization</a>:</div>
164<div class="line"><a name="l00066"></a><span class="lineno"> 66</span>&#160; {</div>
165<div class="line"><a name="l00067"></a><span class="lineno"> 67</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html">armnn::InstanceNormalizationDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
166<div class="line"><a name="l00068"></a><span class="lineno"> 68</span>&#160; <span class="keywordflow">break</span>;</div>
167<div class="line"><a name="l00069"></a><span class="lineno"> 69</span>&#160; }</div>
168<div class="line"><a name="l00070"></a><span class="lineno"> 70</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a1189467870bc421ae59277e750263eb2">armnn::LayerType::L2Normalization</a>:</div>
169<div class="line"><a name="l00071"></a><span class="lineno"> 71</span>&#160; {</div>
170<div class="line"><a name="l00072"></a><span class="lineno"> 72</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_l2_normalization_descriptor.html">armnn::L2NormalizationDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
171<div class="line"><a name="l00073"></a><span class="lineno"> 73</span>&#160; <span class="keywordflow">break</span>;</div>
172<div class="line"><a name="l00074"></a><span class="lineno"> 74</span>&#160; }</div>
173<div class="line"><a name="l00075"></a><span class="lineno"> 75</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aac61f2e17250a818dee4d12b112aa88f">armnn::LayerType::Normalization</a>:</div>
174<div class="line"><a name="l00076"></a><span class="lineno"> 76</span>&#160; {</div>
175<div class="line"><a name="l00077"></a><span class="lineno"> 77</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_normalization_descriptor.html">armnn::NormalizationDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
176<div class="line"><a name="l00078"></a><span class="lineno"> 78</span>&#160; <span class="keywordflow">break</span>;</div>
177<div class="line"><a name="l00079"></a><span class="lineno"> 79</span>&#160; }</div>
178<div class="line"><a name="l00080"></a><span class="lineno"> 80</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4ad662867a41bfb30b9f75dda2b5849001">armnn::LayerType::Pooling2d</a>:</div>
179<div class="line"><a name="l00081"></a><span class="lineno"> 81</span>&#160; {</div>
180<div class="line"><a name="l00082"></a><span class="lineno"> 82</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_pooling2d_descriptor.html">armnn::Pooling2dDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
181<div class="line"><a name="l00083"></a><span class="lineno"> 83</span>&#160; <span class="keywordflow">break</span>;</div>
182<div class="line"><a name="l00084"></a><span class="lineno"> 84</span>&#160; }</div>
183<div class="line"><a name="l00085"></a><span class="lineno"> 85</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a2b3140dc366b9fcd25ed786a79d1817c">armnn::LayerType::Pooling3d</a>:</div>
184<div class="line"><a name="l00086"></a><span class="lineno"> 86</span>&#160; {</div>
185<div class="line"><a name="l00087"></a><span class="lineno"> 87</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_pooling3d_descriptor.html">armnn::Pooling3dDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
186<div class="line"><a name="l00088"></a><span class="lineno"> 88</span>&#160; <span class="keywordflow">break</span>;</div>
187<div class="line"><a name="l00089"></a><span class="lineno"> 89</span>&#160; }</div>
188<div class="line"><a name="l00090"></a><span class="lineno"> 90</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a337c392144dca0d18290c6b4711a2279">armnn::LayerType::SpaceToBatchNd</a>:</div>
189<div class="line"><a name="l00091"></a><span class="lineno"> 91</span>&#160; {</div>
190<div class="line"><a name="l00092"></a><span class="lineno"> 92</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_space_to_batch_nd_descriptor.html">armnn::SpaceToBatchNdDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
191<div class="line"><a name="l00093"></a><span class="lineno"> 93</span>&#160; <span class="keywordflow">break</span>;</div>
192<div class="line"><a name="l00094"></a><span class="lineno"> 94</span>&#160; }</div>
193<div class="line"><a name="l00095"></a><span class="lineno"> 95</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a5e7ff12da912dc79e7e547281823fa4a">armnn::LayerType::SpaceToDepth</a>:</div>
194<div class="line"><a name="l00096"></a><span class="lineno"> 96</span>&#160; {</div>
195<div class="line"><a name="l00097"></a><span class="lineno"> 97</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html">armnn::SpaceToDepthDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
196<div class="line"><a name="l00098"></a><span class="lineno"> 98</span>&#160; <span class="keywordflow">break</span>;</div>
197<div class="line"><a name="l00099"></a><span class="lineno"> 99</span>&#160; }</div>
198<div class="line"><a name="l00100"></a><span class="lineno"> 100</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa31904f2b3479b5a00137fd985974b4d">armnn::LayerType::StridedSlice</a>:</div>
199<div class="line"><a name="l00101"></a><span class="lineno"> 101</span>&#160; {</div>
200<div class="line"><a name="l00102"></a><span class="lineno"> 102</span>&#160; CheckDescForNCHW(<span class="keyword">static_cast&lt;</span><span class="keyword">const </span><a class="code" href="structarmnn_1_1_strided_slice_descriptor.html">armnn::StridedSliceDescriptor</a>&amp;<span class="keyword">&gt;</span>(descriptor));</div>
201<div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160; <span class="keywordflow">break</span>;</div>
202<div class="line"><a name="l00104"></a><span class="lineno"> 104</span>&#160; }</div>
203<div class="line"><a name="l00105"></a><span class="lineno"> 105</span>&#160; <span class="keywordflow">default</span>:</div>
204<div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160; {</div>
205<div class="line"><a name="l00107"></a><span class="lineno"> 107</span>&#160; m_Result = <span class="keyword">false</span>;</div>
206<div class="line"><a name="l00108"></a><span class="lineno"> 108</span>&#160; }</div>
207<div class="line"><a name="l00109"></a><span class="lineno"> 109</span>&#160; }</div>
208<div class="line"><a name="l00110"></a><span class="lineno"> 110</span>&#160; }</div>
209<div class="line"><a name="l00111"></a><span class="lineno"> 111</span>&#160;<span class="comment"></span> </div>
210<div class="line"><a name="l00112"></a><span class="lineno"> 112</span>&#160;<span class="comment"> /// Returns true if the Layer had a DataLayout and it was NCHW or NCDHW.</span></div>
211<div class="line"><a name="l00113"></a><span class="lineno"> 113</span>&#160;<span class="comment"> /// Returns false if the Layer either doesn&#39;t have a DataLayout or if it</span></div>
212<div class="line"><a name="l00114"></a><span class="lineno"> 114</span>&#160;<span class="comment"> /// had a DataLayout that was neither NCHW nor NCDHW.</span></div>
213<div class="line"><a name="l00115"></a><span class="lineno"> 115</span>&#160;<span class="comment"></span> <span class="keywordtype">bool</span> Result()</div>
214<div class="line"><a name="l00116"></a><span class="lineno"> 116</span>&#160; {</div>
215<div class="line"><a name="l00117"></a><span class="lineno"> 117</span>&#160; <span class="keywordflow">return</span> m_Result;</div>
216<div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160; }</div>
217<div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160; </div>
218<div class="line"><a name="l00120"></a><span class="lineno"> 120</span>&#160;<span class="keyword">private</span>:</div>
219<div class="line"><a name="l00121"></a><span class="lineno"> 121</span>&#160; <span class="keyword">template</span>&lt;<span class="keyword">typename</span> Descriptor&gt;</div>
220<div class="line"><a name="l00122"></a><span class="lineno"> 122</span>&#160; <span class="keywordtype">void</span> CheckDescForNCHW(<span class="keyword">const</span> Descriptor&amp; descriptor)</div>
221<div class="line"><a name="l00123"></a><span class="lineno"> 123</span>&#160; {</div>
222<div class="line"><a name="l00124"></a><span class="lineno"> 124</span>&#160; m_Result = (descriptor.m_DataLayout == <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">DataLayout::NCHW</a>) || (descriptor.m_DataLayout == <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0a884e0167ebf9bbe6cfd6ca5ab97ab015">DataLayout::NCDHW</a>);</div>
223<div class="line"><a name="l00125"></a><span class="lineno"> 125</span>&#160; }</div>
224<div class="line"><a name="l00126"></a><span class="lineno"> 126</span>&#160; </div>
225<div class="line"><a name="l00127"></a><span class="lineno"> 127</span>&#160; <span class="keywordtype">bool</span> m_Result = <span class="keyword">false</span>;</div>
226<div class="line"><a name="l00128"></a><span class="lineno"> 128</span>&#160;};</div>
227<div class="line"><a name="l00129"></a><span class="lineno"> 129</span>&#160; </div>
228<div class="line"><a name="l00130"></a><span class="lineno"> 130</span>&#160;<span class="comment">//</span></div>
229<div class="line"><a name="l00131"></a><span class="lineno"> 131</span>&#160;<span class="comment">// this helper only works if all layers where the inputs connect to are not selected</span></div>
230<div class="line"><a name="l00132"></a><span class="lineno"> 132</span>&#160;<span class="comment">//</span></div>
231<div class="line"><a name="l00133"></a><span class="lineno"> 133</span>&#160; </div>
232<div class="line"><a name="l00134"></a><span class="lineno"> 134</span>&#160;<a class="code" href="classarmnn_1_1_subgraph_view.html#a4ad949634fd697549da161eb09d8b7fd">SubgraphView::IInputSlots</a> CreateIInputsFrom(<span class="keyword">const</span> std::vector&lt;armnn::IConnectableLayer*&gt;&amp; layers)</div>
233<div class="line"><a name="l00135"></a><span class="lineno"> 135</span>&#160;{</div>
234<div class="line"><a name="l00136"></a><span class="lineno"> 136</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html#a4ad949634fd697549da161eb09d8b7fd">SubgraphView::IInputSlots</a> result;</div>
235<div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160; <span class="keywordflow">for</span> (<span class="keyword">auto</span>&amp;&amp; layer : layers)</div>
236<div class="line"><a name="l00138"></a><span class="lineno"> 138</span>&#160; {</div>
237<div class="line"><a name="l00139"></a><span class="lineno"> 139</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0 ; i &lt; layer-&gt;<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a9c2cba04b6d7ace4fc2a2436b82a5a63">GetNumInputSlots</a>(); ++i)</div>
238<div class="line"><a name="l00140"></a><span class="lineno"> 140</span>&#160; {</div>
239<div class="line"><a name="l00141"></a><span class="lineno"> 141</span>&#160; result.push_back(&amp;(layer-&gt;<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(i)));</div>
240<div class="line"><a name="l00142"></a><span class="lineno"> 142</span>&#160; }</div>
241<div class="line"><a name="l00143"></a><span class="lineno"> 143</span>&#160; }</div>
242<div class="line"><a name="l00144"></a><span class="lineno"> 144</span>&#160; <span class="keywordflow">return</span> result;</div>
243<div class="line"><a name="l00145"></a><span class="lineno"> 145</span>&#160;}</div>
244<div class="line"><a name="l00146"></a><span class="lineno"> 146</span>&#160; </div>
245<div class="line"><a name="l00147"></a><span class="lineno"> 147</span>&#160;<span class="comment">//</span></div>
246<div class="line"><a name="l00148"></a><span class="lineno"> 148</span>&#160;<span class="comment">// this helper only works if all layers where the outputs connect to are not selected</span></div>
247<div class="line"><a name="l00149"></a><span class="lineno"> 149</span>&#160;<span class="comment">//</span></div>
248<div class="line"><a name="l00150"></a><span class="lineno"> 150</span>&#160; </div>
249<div class="line"><a name="l00151"></a><span class="lineno"> 151</span>&#160;<a class="code" href="classarmnn_1_1_subgraph_view.html#a3f717693851810942d3ae0180a97fb4c">SubgraphView::IOutputSlots</a> CreateIOutputsFrom(<span class="keyword">const</span> std::vector&lt;armnn::IConnectableLayer*&gt;&amp; layers)</div>
250<div class="line"><a name="l00152"></a><span class="lineno"> 152</span>&#160;{</div>
251<div class="line"><a name="l00153"></a><span class="lineno"> 153</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html#a3f717693851810942d3ae0180a97fb4c">SubgraphView::IOutputSlots</a> result;</div>
252<div class="line"><a name="l00154"></a><span class="lineno"> 154</span>&#160; <span class="keywordflow">for</span> (<span class="keyword">auto</span> &amp;&amp;layer: layers)</div>
253<div class="line"><a name="l00155"></a><span class="lineno"> 155</span>&#160; {</div>
254<div class="line"><a name="l00156"></a><span class="lineno"> 156</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; layer-&gt;<a class="code" href="classarmnn_1_1_i_connectable_layer.html#ac2dac3b61c94de52093616be4ab17f8d">GetNumOutputSlots</a>(); ++i)</div>
255<div class="line"><a name="l00157"></a><span class="lineno"> 157</span>&#160; {</div>
256<div class="line"><a name="l00158"></a><span class="lineno"> 158</span>&#160; result.push_back(&amp;(layer-&gt;<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(i)));</div>
257<div class="line"><a name="l00159"></a><span class="lineno"> 159</span>&#160; }</div>
258<div class="line"><a name="l00160"></a><span class="lineno"> 160</span>&#160; }</div>
259<div class="line"><a name="l00161"></a><span class="lineno"> 161</span>&#160; <span class="keywordflow">return</span> result;</div>
260<div class="line"><a name="l00162"></a><span class="lineno"> 162</span>&#160;}</div>
261<div class="line"><a name="l00163"></a><span class="lineno"> 163</span>&#160; </div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]262<div class="line"><a name="l00164"></a><span class="lineno"> 164</span>&#160;<span class="comment">// Type used to hold the slot numbers to create the lists from. There should</span></div>
263<div class="line"><a name="l00165"></a><span class="lineno"> 165</span>&#160;<span class="comment">// be a SlotList for each layer in the layers list</span></div>
264<div class="line"><a name="l00166"></a><span class="lineno"> 166</span>&#160;<span class="keyword">typedef</span> std::vector&lt;int&gt; SlotList;</div>
265<div class="line"><a name="l00167"></a><span class="lineno"> 167</span>&#160; </div>
266<div class="line"><a name="l00168"></a><span class="lineno"> 168</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> ILayerType&gt;</div>
267<div class="line"><a name="l00169"></a><span class="lineno"> 169</span>&#160;<a class="code" href="classarmnn_1_1_subgraph_view.html#a4ad949634fd697549da161eb09d8b7fd">SubgraphView::IInputSlots</a> CreateIInputsFromSlotLists(<span class="keyword">const</span> std::vector&lt;ILayerType*&gt;&amp; layers,</div>
268<div class="line"><a name="l00170"></a><span class="lineno"> 170</span>&#160; <span class="keyword">const</span> std::vector&lt;SlotList&gt;&amp; layersSlotLists)</div>
269<div class="line"><a name="l00171"></a><span class="lineno"> 171</span>&#160;{</div>
270<div class="line"><a name="l00172"></a><span class="lineno"> 172</span>&#160; <a class="code" href="_exceptions_8hpp.html#a31e4268013316d114e8927e4c0035ec5">ARMNN_THROW_INVALIDARG_IF_FALSE</a>(layersSlotLists.size() == layers.size());</div>
271<div class="line"><a name="l00173"></a><span class="lineno"> 173</span>&#160; </div>
272<div class="line"><a name="l00174"></a><span class="lineno"> 174</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html#a4ad949634fd697549da161eb09d8b7fd">SubgraphView::IInputSlots</a> result;</div>
273<div class="line"><a name="l00175"></a><span class="lineno"> 175</span>&#160; </div>
274<div class="line"><a name="l00176"></a><span class="lineno"> 176</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> layerIdx = 0; layerIdx &lt; layers.size(); ++layerIdx)</div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]275<div class="line"><a name="l00177"></a><span class="lineno"> 177</span>&#160; {</div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]276<div class="line"><a name="l00178"></a><span class="lineno"> 178</span>&#160; <span class="keyword">const</span> SlotList&amp; slotList = layersSlotLists[layerIdx];</div>
277<div class="line"><a name="l00179"></a><span class="lineno"> 179</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> slotIdx = 0 ; slotIdx &lt; layers[layerIdx]-&gt;GetNumInputSlots(); ++slotIdx)</div>
278<div class="line"><a name="l00180"></a><span class="lineno"> 180</span>&#160; {</div>
279<div class="line"><a name="l00181"></a><span class="lineno"> 181</span>&#160; <span class="keywordflow">if</span> (std::find(slotList.begin(), slotList.end(), slotIdx) != slotList.end())</div>
280<div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160; {</div>
281<div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160; result.push_back(&amp;(layers[layerIdx]-&gt;GetInputSlot(slotIdx)));</div>
282<div class="line"><a name="l00184"></a><span class="lineno"> 184</span>&#160; }</div>
283<div class="line"><a name="l00185"></a><span class="lineno"> 185</span>&#160; }</div>
284<div class="line"><a name="l00186"></a><span class="lineno"> 186</span>&#160; }</div>
285<div class="line"><a name="l00187"></a><span class="lineno"> 187</span>&#160; <span class="keywordflow">return</span> result;</div>
286<div class="line"><a name="l00188"></a><span class="lineno"> 188</span>&#160;}</div>
287<div class="line"><a name="l00189"></a><span class="lineno"> 189</span>&#160; </div>
288<div class="line"><a name="l00190"></a><span class="lineno"> 190</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> ILayerType&gt;</div>
289<div class="line"><a name="l00191"></a><span class="lineno"> 191</span>&#160;<a class="code" href="classarmnn_1_1_subgraph_view.html#a3f717693851810942d3ae0180a97fb4c">SubgraphView::IOutputSlots</a> CreateIOutputsFromSlotLists(<span class="keyword">const</span> std::vector&lt;ILayerType*&gt;&amp; layers,</div>
290<div class="line"><a name="l00192"></a><span class="lineno"> 192</span>&#160; <span class="keyword">const</span> std::vector&lt;SlotList&gt;&amp; layersSlotLists)</div>
291<div class="line"><a name="l00193"></a><span class="lineno"> 193</span>&#160;{</div>
292<div class="line"><a name="l00194"></a><span class="lineno"> 194</span>&#160; <a class="code" href="_exceptions_8hpp.html#a31e4268013316d114e8927e4c0035ec5">ARMNN_THROW_INVALIDARG_IF_FALSE</a>(layersSlotLists.size() == layers.size());</div>
293<div class="line"><a name="l00195"></a><span class="lineno"> 195</span>&#160; </div>
294<div class="line"><a name="l00196"></a><span class="lineno"> 196</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html#a3f717693851810942d3ae0180a97fb4c">SubgraphView::IOutputSlots</a> result;</div>
295<div class="line"><a name="l00197"></a><span class="lineno"> 197</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> layerIdx = 0; layerIdx &lt; layers.size(); ++layerIdx)</div>
296<div class="line"><a name="l00198"></a><span class="lineno"> 198</span>&#160; {</div>
297<div class="line"><a name="l00199"></a><span class="lineno"> 199</span>&#160; <span class="keyword">const</span> SlotList&amp; slotList = layersSlotLists[layerIdx];</div>
298<div class="line"><a name="l00200"></a><span class="lineno"> 200</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> slotIdx = 0; slotIdx &lt; layers[layerIdx]-&gt;GetNumOutputSlots(); ++slotIdx)</div>
299<div class="line"><a name="l00201"></a><span class="lineno"> 201</span>&#160; {</div>
300<div class="line"><a name="l00202"></a><span class="lineno"> 202</span>&#160; <span class="keywordtype">bool</span> foundIt = std::find(slotList.begin(), slotList.end(), slotIdx) != slotList.end();</div>
301<div class="line"><a name="l00203"></a><span class="lineno"> 203</span>&#160; <span class="keywordflow">if</span> (foundIt)</div>
302<div class="line"><a name="l00204"></a><span class="lineno"> 204</span>&#160; {</div>
303<div class="line"><a name="l00205"></a><span class="lineno"> 205</span>&#160; result.push_back(&amp;(layers[layerIdx]-&gt;GetOutputSlot(slotIdx)));</div>
304<div class="line"><a name="l00206"></a><span class="lineno"> 206</span>&#160; }</div>
305<div class="line"><a name="l00207"></a><span class="lineno"> 207</span>&#160; }</div>
306<div class="line"><a name="l00208"></a><span class="lineno"> 208</span>&#160; }</div>
307<div class="line"><a name="l00209"></a><span class="lineno"> 209</span>&#160; <span class="keywordflow">return</span> result;</div>
308<div class="line"><a name="l00210"></a><span class="lineno"> 210</span>&#160;}</div>
309<div class="line"><a name="l00211"></a><span class="lineno"> 211</span>&#160;}</div>
310<div class="line"><a name="l00212"></a><span class="lineno"> 212</span>&#160; </div>
311<div class="line"><a name="l00213"></a><span class="lineno"><a class="line" href="namespacearmnn.html#ab6b4f5f3473b0ce4b241d454f8cbdd26"> 213</a></span>&#160;<span class="keyword">inline</span> <span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.html#ab6b4f5f3473b0ce4b241d454f8cbdd26">IsNCHW</a>(<a class="code" href="classarmnn_1_1_layer.html">armnn::Layer</a>&amp; layer)</div>
312<div class="line"><a name="l00214"></a><span class="lineno"> 214</span>&#160;{</div>
313<div class="line"><a name="l00215"></a><span class="lineno"> 215</span>&#160; CheckForNCHW check;</div>
314<div class="line"><a name="l00216"></a><span class="lineno"> 216</span>&#160; layer.<a class="code" href="classarmnn_1_1_layer.html#a46fc3fdd4b2a5dd6d184e57983cf20bc">ExecuteStrategy</a>(check);</div>
315<div class="line"><a name="l00217"></a><span class="lineno"> 217</span>&#160; <span class="keywordflow">return</span> check.Result();</div>
316<div class="line"><a name="l00218"></a><span class="lineno"> 218</span>&#160;}</div>
317<div class="line"><a name="l00219"></a><span class="lineno"> 219</span>&#160; </div>
318<div class="line"><a name="l00220"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a5afd10e5e84ebf5cb8cfc8707492eda7"> 220</a></span>&#160;<span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code" href="namespacearmnn.html#a5afd10e5e84ebf5cb8cfc8707492eda7">ReportUntouchedLayers</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews, std::map&lt;LayerGuid, Layer*&gt; untouched)</div>
319<div class="line"><a name="l00221"></a><span class="lineno"> 221</span>&#160;{</div>
320<div class="line"><a name="l00222"></a><span class="lineno"> 222</span>&#160; std::vector&lt;Layer*&gt; untouchedVector;</div>
321<div class="line"><a name="l00223"></a><span class="lineno"> 223</span>&#160; <span class="keywordflow">for</span> (<span class="keyword">const</span> <span class="keyword">auto</span>&amp; pair : untouched)</div>
322<div class="line"><a name="l00224"></a><span class="lineno"> 224</span>&#160; {</div>
323<div class="line"><a name="l00225"></a><span class="lineno"> 225</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>* layer = pair.second;</div>
324<div class="line"><a name="l00226"></a><span class="lineno"> 226</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> subgraphView({layer},</div>
325<div class="line"><a name="l00227"></a><span class="lineno"> 227</span>&#160; CreateIInputsFrom({layer}),</div>
326<div class="line"><a name="l00228"></a><span class="lineno"> 228</span>&#160; CreateIOutputsFrom({layer}));</div>
327<div class="line"><a name="l00229"></a><span class="lineno"> 229</span>&#160; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#a28e41bdd6b719a3d60a1a0de2e1ebc95">AddUntouchedSubgraph</a>(std::move(subgraphView));</div>
328<div class="line"><a name="l00230"></a><span class="lineno"> 230</span>&#160; }</div>
329<div class="line"><a name="l00231"></a><span class="lineno"> 231</span>&#160;}</div>
330<div class="line"><a name="l00232"></a><span class="lineno"> 232</span>&#160; </div>
331<div class="line"><a name="l00233"></a><span class="lineno"> 233</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
332<div class="line"><a name="l00234"></a><span class="lineno"><a class="line" href="namespacearmnn.html#ae4b69405416c09ee6878e01d27c9450b"> 234</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#ae4b69405416c09ee6878e01d27c9450b">FoldPadLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
333<div class="line"><a name="l00235"></a><span class="lineno"> 235</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
334<div class="line"><a name="l00236"></a><span class="lineno"> 236</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer,</div>
335<div class="line"><a name="l00237"></a><span class="lineno"> 237</span>&#160; <a class="code" href="classarmnn_1_1_pad_layer.html">PadLayer</a>* padLayer)</div>
336<div class="line"><a name="l00238"></a><span class="lineno"> 238</span>&#160;{</div>
337<div class="line"><a name="l00239"></a><span class="lineno"> 239</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> substitutionSubgraph({padLayer, baseLayer},</div>
338<div class="line"><a name="l00240"></a><span class="lineno"> 240</span>&#160; CreateIInputsFrom({padLayer}),</div>
339<div class="line"><a name="l00241"></a><span class="lineno"> 241</span>&#160; CreateIOutputsFrom({baseLayer}));</div>
340<div class="line"><a name="l00242"></a><span class="lineno"> 242</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> replacementSubgraph(replacementLayer);</div>
341<div class="line"><a name="l00243"></a><span class="lineno"> 243</span>&#160; </div>
342<div class="line"><a name="l00244"></a><span class="lineno"> 244</span>&#160; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#a57340147dcb1c0ae05fdf14ff7af8a01">AddSubstitution</a>({substitutionSubgraph, replacementSubgraph});</div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]343<div class="line"><a name="l00245"></a><span class="lineno"> 245</span>&#160; </div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]344<div class="line"><a name="l00246"></a><span class="lineno"> 246</span>&#160; <span class="keywordflow">return</span> replacementLayer;</div>
345<div class="line"><a name="l00247"></a><span class="lineno"> 247</span>&#160;}</div>
346<div class="line"><a name="l00248"></a><span class="lineno"> 248</span>&#160;<span class="comment"></span> </div>
347<div class="line"><a name="l00249"></a><span class="lineno"> 249</span>&#160;<span class="comment">/// Checks if the Layer is connected to any Layer that has an NCHW layout.</span></div>
348<div class="line"><a name="l00250"></a><span class="lineno"><a class="line" href="namespacearmnn.html#ac5089a2ee61f22570982e7a56588365a"> 250</a></span>&#160;<span class="comment"></span><span class="keyword">inline</span> <span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.html#ac5089a2ee61f22570982e7a56588365a">ConnectedToLayerWithNCHW</a>(<a class="code" href="classarmnn_1_1_layer.html">Layer</a>* baseLayer)</div>
349<div class="line"><a name="l00251"></a><span class="lineno"> 251</span>&#160;{</div>
350<div class="line"><a name="l00252"></a><span class="lineno"> 252</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; parentLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">GetInputSlot</a>(0).<a class="code" href="classarmnn_1_1_input_slot.html#a9effd325a6d512a3f8ff4bd207d53255">GetConnectedOutputSlot</a>()-&gt;<a class="code" href="classarmnn_1_1_output_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
351<div class="line"><a name="l00253"></a><span class="lineno"> 253</span>&#160; </div>
352<div class="line"><a name="l00254"></a><span class="lineno"> 254</span>&#160; <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn.html#ab6b4f5f3473b0ce4b241d454f8cbdd26">IsNCHW</a>(parentLayer))</div>
353<div class="line"><a name="l00255"></a><span class="lineno"> 255</span>&#160; {</div>
354<div class="line"><a name="l00256"></a><span class="lineno"> 256</span>&#160; <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
355<div class="line"><a name="l00257"></a><span class="lineno"> 257</span>&#160; }</div>
356<div class="line"><a name="l00258"></a><span class="lineno"> 258</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).GetNumConnections(); ++i)</div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]357<div class="line"><a name="l00259"></a><span class="lineno"> 259</span>&#160; {</div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]358<div class="line"><a name="l00260"></a><span class="lineno"> 260</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; nextLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">GetConnection</a>(i)-&gt;<a class="code" href="classarmnn_1_1_input_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
359<div class="line"><a name="l00261"></a><span class="lineno"> 261</span>&#160; <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn.html#ab6b4f5f3473b0ce4b241d454f8cbdd26">IsNCHW</a>(nextLayer))</div>
360<div class="line"><a name="l00262"></a><span class="lineno"> 262</span>&#160; {</div>
361<div class="line"><a name="l00263"></a><span class="lineno"> 263</span>&#160; <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
362<div class="line"><a name="l00264"></a><span class="lineno"> 264</span>&#160; }</div>
363<div class="line"><a name="l00265"></a><span class="lineno"> 265</span>&#160; }</div>
364<div class="line"><a name="l00266"></a><span class="lineno"> 266</span>&#160; <span class="keywordflow">return</span> <span class="keyword">false</span>;</div>
365<div class="line"><a name="l00267"></a><span class="lineno"> 267</span>&#160;}</div>
366<div class="line"><a name="l00268"></a><span class="lineno"> 268</span>&#160;<span class="comment"></span> </div>
367<div class="line"><a name="l00269"></a><span class="lineno"> 269</span>&#160;<span class="comment">/// Checks the Layer&#39;s Connections to see if it&#39;s connected to a Layer with the provided layerType. If dimSize is</span></div>
368<div class="line"><a name="l00270"></a><span class="lineno"> 270</span>&#160;<span class="comment">/// provided will also check if the connecting Tensor has more than that number of dimensions</span></div>
369<div class="line"><a name="l00271"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a4f89a6ce848b8d9787761cdd1dbb751a"> 271</a></span>&#160;<span class="comment"></span><span class="keyword">inline</span> <span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.html#a4f89a6ce848b8d9787761cdd1dbb751a">ConnectedToLayerType</a>(<a class="code" href="classarmnn_1_1_layer.html">Layer</a>* baseLayer, <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a> layerType, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> dimSize = 0)</div>
370<div class="line"><a name="l00272"></a><span class="lineno"> 272</span>&#160;{</div>
371<div class="line"><a name="l00273"></a><span class="lineno"> 273</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; parentLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">GetInputSlot</a>(0).<a class="code" href="classarmnn_1_1_input_slot.html#a9effd325a6d512a3f8ff4bd207d53255">GetConnectedOutputSlot</a>()-&gt;<a class="code" href="classarmnn_1_1_output_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
372<div class="line"><a name="l00274"></a><span class="lineno"> 274</span>&#160; <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> parentTensorInfo = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">GetInputSlot</a>(0).<a class="code" href="classarmnn_1_1_input_slot.html#ada2ad7d1caeeb4ef6195c8925fad6a65">GetTensorInfo</a>();</div>
373<div class="line"><a name="l00275"></a><span class="lineno"> 275</span>&#160; </div>
374<div class="line"><a name="l00276"></a><span class="lineno"> 276</span>&#160; <span class="keywordflow">if</span> (parentTensorInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() &gt; dimSize &amp;&amp; parentLayer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == layerType)</div>
375<div class="line"><a name="l00277"></a><span class="lineno"> 277</span>&#160; {</div>
376<div class="line"><a name="l00278"></a><span class="lineno"> 278</span>&#160; <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
377<div class="line"><a name="l00279"></a><span class="lineno"> 279</span>&#160; }</div>
378<div class="line"><a name="l00280"></a><span class="lineno"> 280</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).GetNumConnections(); ++i)</div>
379<div class="line"><a name="l00281"></a><span class="lineno"> 281</span>&#160; {</div>
380<div class="line"><a name="l00282"></a><span class="lineno"> 282</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; nextLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">GetConnection</a>(i)-&gt;<a class="code" href="classarmnn_1_1_input_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
381<div class="line"><a name="l00283"></a><span class="lineno"> 283</span>&#160; <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> nextTensorInfo = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">GetConnection</a>(i)-&gt;<a class="code" href="classarmnn_1_1_input_slot.html#ada2ad7d1caeeb4ef6195c8925fad6a65">GetTensorInfo</a>();</div>
382<div class="line"><a name="l00284"></a><span class="lineno"> 284</span>&#160; </div>
383<div class="line"><a name="l00285"></a><span class="lineno"> 285</span>&#160; <span class="keywordflow">if</span> (nextTensorInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() &gt; dimSize &amp;&amp; nextLayer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == layerType)</div>
384<div class="line"><a name="l00286"></a><span class="lineno"> 286</span>&#160; {</div>
385<div class="line"><a name="l00287"></a><span class="lineno"> 287</span>&#160; <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
386<div class="line"><a name="l00288"></a><span class="lineno"> 288</span>&#160; }</div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]387<div class="line"><a name="l00289"></a><span class="lineno"> 289</span>&#160; }</div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]388<div class="line"><a name="l00290"></a><span class="lineno"> 290</span>&#160; <span class="keywordflow">return</span> <span class="keyword">false</span>;</div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]389<div class="line"><a name="l00291"></a><span class="lineno"> 291</span>&#160;}</div>
390<div class="line"><a name="l00292"></a><span class="lineno"> 292</span>&#160; </div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]391<div class="line"><a name="l00293"></a><span class="lineno"><a class="line" href="namespacearmnn.html#ae756568a7f2c58778bb6d759d2bd38db"> 293</a></span>&#160;<span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code" href="namespacearmnn.html#ae756568a7f2c58778bb6d759d2bd38db">RemoveReshapeLayer</a>(<a class="code" href="classarmnn_1_1_reshape_layer.html">ReshapeLayer</a>* baseLayer,</div>
392<div class="line"><a name="l00294"></a><span class="lineno"> 294</span>&#160; std::map&lt;LayerGuid, Layer*&gt;&amp; untouched,</div>
393<div class="line"><a name="l00295"></a><span class="lineno"> 295</span>&#160; <a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews)</div>
394<div class="line"><a name="l00296"></a><span class="lineno"> 296</span>&#160;{</div>
395<div class="line"><a name="l00297"></a><span class="lineno"> 297</span>&#160; <span class="keywordflow">if</span> (baseLayer == <span class="keyword">nullptr</span>)</div>
396<div class="line"><a name="l00298"></a><span class="lineno"> 298</span>&#160; {</div>
397<div class="line"><a name="l00299"></a><span class="lineno"> 299</span>&#160; <span class="keywordflow">return</span>;</div>
398<div class="line"><a name="l00300"></a><span class="lineno"> 300</span>&#160; }</div>
399<div class="line"><a name="l00301"></a><span class="lineno"> 301</span>&#160; <a class="code" href="structarmnn_1_1_reshape_descriptor.html">ReshapeDescriptor</a> reshapeDescriptor = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer_with_parameters.html#afa3e8a8f23589b1eaddbe203825bbdcf">GetParameters</a>();</div>
400<div class="line"><a name="l00302"></a><span class="lineno"> 302</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; parentLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">GetInputSlot</a>(0).<a class="code" href="classarmnn_1_1_input_slot.html#a9effd325a6d512a3f8ff4bd207d53255">GetConnectedOutputSlot</a>()-&gt;<a class="code" href="classarmnn_1_1_output_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
401<div class="line"><a name="l00303"></a><span class="lineno"> 303</span>&#160; </div>
402<div class="line"><a name="l00304"></a><span class="lineno"> 304</span>&#160; <span class="comment">// Cannot currently remove the Reshape if it&#39;s connected to an Input, Constant or Splitter</span></div>
403<div class="line"><a name="l00305"></a><span class="lineno"> 305</span>&#160; <span class="keywordflow">if</span> (parentLayer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a324118a6721dd6b8a9b9f4e327df2bf5">LayerType::Input</a> || parentLayer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4acb17869fe51048b5a5c4c6106551a255">LayerType::Constant</a>)</div>
404<div class="line"><a name="l00306"></a><span class="lineno"> 306</span>&#160; {</div>
405<div class="line"><a name="l00307"></a><span class="lineno"> 307</span>&#160; <span class="keywordflow">return</span>;</div>
406<div class="line"><a name="l00308"></a><span class="lineno"> 308</span>&#160; }</div>
407<div class="line"><a name="l00309"></a><span class="lineno"> 309</span>&#160; </div>
408<div class="line"><a name="l00310"></a><span class="lineno"> 310</span>&#160; <span class="comment">// Cannot currently remove the Reshape if it&#39;s connected to an OutputSlot or Concat</span></div>
409<div class="line"><a name="l00311"></a><span class="lineno"> 311</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).GetNumConnections(); ++i)</div>
410<div class="line"><a name="l00312"></a><span class="lineno"> 312</span>&#160; {</div>
411<div class="line"><a name="l00313"></a><span class="lineno"> 313</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; nextLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">GetConnection</a>(i)-&gt;<a class="code" href="classarmnn_1_1_input_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
412<div class="line"><a name="l00314"></a><span class="lineno"> 314</span>&#160; </div>
413<div class="line"><a name="l00315"></a><span class="lineno"> 315</span>&#160; <span class="keywordflow">if</span> (nextLayer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a29c2c02a361c9d7028472e5d92cd4a54">LayerType::Output</a>)</div>
414<div class="line"><a name="l00316"></a><span class="lineno"> 316</span>&#160; {</div>
415<div class="line"><a name="l00317"></a><span class="lineno"> 317</span>&#160; <span class="keywordflow">return</span>;</div>
416<div class="line"><a name="l00318"></a><span class="lineno"> 318</span>&#160; }</div>
417<div class="line"><a name="l00319"></a><span class="lineno"> 319</span>&#160; }</div>
418<div class="line"><a name="l00320"></a><span class="lineno"> 320</span>&#160; <span class="keyword">auto</span> it = untouched.find(baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a8dc12f0ee5b232d397bd18ced1a72a64">GetGuid</a>());</div>
419<div class="line"><a name="l00321"></a><span class="lineno"> 321</span>&#160; <span class="keywordflow">if</span> (it == untouched.end())</div>
420<div class="line"><a name="l00322"></a><span class="lineno"> 322</span>&#160; {</div>
421<div class="line"><a name="l00323"></a><span class="lineno"> 323</span>&#160; <span class="comment">// Already removed from map</span></div>
422<div class="line"><a name="l00324"></a><span class="lineno"> 324</span>&#160; <span class="keywordflow">return</span>;</div>
423<div class="line"><a name="l00325"></a><span class="lineno"> 325</span>&#160; }</div>
424<div class="line"><a name="l00326"></a><span class="lineno"> 326</span>&#160; untouched.erase(it);</div>
425<div class="line"><a name="l00327"></a><span class="lineno"> 327</span>&#160; </div>
426<div class="line"><a name="l00328"></a><span class="lineno"> 328</span>&#160; <span class="comment">// Override the InputSlot TensorInfos for all the layers connected to the Reshape&#39;s OutputSlot</span></div>
427<div class="line"><a name="l00329"></a><span class="lineno"> 329</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i &lt; baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).GetNumConnections(); ++i)</div>
428<div class="line"><a name="l00330"></a><span class="lineno"> 330</span>&#160; {</div>
429<div class="line"><a name="l00331"></a><span class="lineno"> 331</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; nextLayer = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">GetConnection</a>(i)-&gt;<a class="code" href="classarmnn_1_1_input_slot.html#a7ddaf04177053a536f0e7be83a642bc6">GetOwningLayer</a>();</div>
430<div class="line"><a name="l00332"></a><span class="lineno"> 332</span>&#160; <span class="keyword">auto</span> inputIndex = baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">GetConnection</a>(i)-&gt;<a class="code" href="classarmnn_1_1_input_slot.html#ae3f69b193433448bee0f3925da4b01a3">GetSlotIndex</a>();</div>
431<div class="line"><a name="l00333"></a><span class="lineno"> 333</span>&#160; <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> reshapeInfo(baseLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#ada2ad7d1caeeb4ef6195c8925fad6a65">GetTensorInfo</a>());</div>
432<div class="line"><a name="l00334"></a><span class="lineno"> 334</span>&#160; reshapeInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(reshapeDescriptor.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a>);</div>
433<div class="line"><a name="l00335"></a><span class="lineno"> 335</span>&#160; nextLayer.<a class="code" href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">GetInputSlot</a>(inputIndex).<a class="code" href="classarmnn_1_1_input_slot.html#a7f4ecb20ab4968de368166d8cbb2b1a3">SetTensorInfo</a>(reshapeInfo);</div>
434<div class="line"><a name="l00336"></a><span class="lineno"> 336</span>&#160; }</div>
435<div class="line"><a name="l00337"></a><span class="lineno"> 337</span>&#160; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#a4438f2ce297390e50f48dc426ba1f4f9">AddDeletedSubgraph</a>(baseLayer);</div>
436<div class="line"><a name="l00338"></a><span class="lineno"> 338</span>&#160;}</div>
437<div class="line"><a name="l00339"></a><span class="lineno"> 339</span>&#160; </div>
438<div class="line"><a name="l00340"></a><span class="lineno"> 340</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
439<div class="line"><a name="l00341"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a0a49066b56fc3bfe44478601c31e2fc0"> 341</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#a0a49066b56fc3bfe44478601c31e2fc0">FoldPadIntoAveragePool2d</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
440<div class="line"><a name="l00342"></a><span class="lineno"> 342</span>&#160; <a class="code" href="classarmnn_1_1_pooling2d_layer.html">Pooling2dLayer</a>* baseLayer,</div>
441<div class="line"><a name="l00343"></a><span class="lineno"> 343</span>&#160; <a class="code" href="structarmnn_1_1_pooling2d_descriptor.html">Pooling2dDescriptor</a>&amp; poolDescriptor,</div>
442<div class="line"><a name="l00344"></a><span class="lineno"> 344</span>&#160; <a class="code" href="classarmnn_1_1_pad_layer.html">PadLayer</a>* padLayer)</div>
443<div class="line"><a name="l00345"></a><span class="lineno"> 345</span>&#160;{</div>
444<div class="line"><a name="l00346"></a><span class="lineno"> 346</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement =</div>
445<div class="line"><a name="l00347"></a><span class="lineno"> 347</span>&#160; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#ad04187fe81f68558b15b6049b2da9cf9">GetINetwork</a>()-&gt;<a class="code" href="classarmnn_1_1_i_network.html#ae913b4351b7027f37eb5657dd7867733">AddPooling2dLayer</a>(poolDescriptor, <span class="stringliteral">&quot;folded-pad-into-pool2d&quot;</span>);</div>
446<div class="line"><a name="l00348"></a><span class="lineno"> 348</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</div>
447<div class="line"><a name="l00349"></a><span class="lineno"> 349</span>&#160; </div>
448<div class="line"><a name="l00350"></a><span class="lineno"> 350</span>&#160; <a class="code" href="namespacearmnn.html#ae4b69405416c09ee6878e01d27c9450b">FoldPadLayer</a>(optimizationViews,</div>
449<div class="line"><a name="l00351"></a><span class="lineno"> 351</span>&#160; baseLayer,</div>
450<div class="line"><a name="l00352"></a><span class="lineno"> 352</span>&#160; replacementLayer,</div>
451<div class="line"><a name="l00353"></a><span class="lineno"> 353</span>&#160; padLayer);</div>
452<div class="line"><a name="l00354"></a><span class="lineno"> 354</span>&#160; </div>
453<div class="line"><a name="l00355"></a><span class="lineno"> 355</span>&#160; <span class="keywordflow">return</span> replacementLayer;</div>
454<div class="line"><a name="l00356"></a><span class="lineno"> 356</span>&#160;}</div>
455<div class="line"><a name="l00357"></a><span class="lineno"> 357</span>&#160; </div>
456<div class="line"><a name="l00358"></a><span class="lineno"> 358</span>&#160;<span class="comment">//</span></div>
457<div class="line"><a name="l00359"></a><span class="lineno"> 359</span>&#160;<span class="comment">// Layer sequence detection such as add + mul + add ( + optional activation )</span></div>
458<div class="line"><a name="l00360"></a><span class="lineno"> 360</span>&#160;<span class="comment">//</span></div>
459<div class="line"><a name="l00361"></a><span class="lineno"> 361</span>&#160; </div>
460<div class="line"><a name="l00362"></a><span class="lineno"><a class="line" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb"> 362</a></span>&#160;<span class="keyword">inline</span> <span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">IsSequenceLayerType</a>(<a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; layer, <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a> type)</div>
461<div class="line"><a name="l00363"></a><span class="lineno"> 363</span>&#160;{</div>
462<div class="line"><a name="l00364"></a><span class="lineno"> 364</span>&#160; <span class="keywordflow">return</span> layer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == type;</div>
463<div class="line"><a name="l00365"></a><span class="lineno"> 365</span>&#160;}</div>
464<div class="line"><a name="l00366"></a><span class="lineno"> 366</span>&#160; </div>
465<div class="line"><a name="l00367"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a336dd839b81ad6a1d803c306b40f9af2"> 367</a></span>&#160;<span class="keyword">inline</span> <span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">IsSequenceLayerType</a>(<a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; layer, <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047">BinaryOperation</a> type)</div>
466<div class="line"><a name="l00368"></a><span class="lineno"> 368</span>&#160;{</div>
467<div class="line"><a name="l00369"></a><span class="lineno"> 369</span>&#160; <span class="keywordflow">return</span> (layer.<a class="code" href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">GetType</a>() == <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a699bcffd93aff3022014b9efc9eaefd1">LayerType::ElementwiseBinary</a>) &amp;&amp;</div>
468<div class="line"><a name="l00370"></a><span class="lineno"> 370</span>&#160; (PolymorphicDowncast&lt;ElementwiseBinaryLayer*&gt;(&amp;layer)-&gt;GetParameters().m_Operation == type);</div>
469<div class="line"><a name="l00371"></a><span class="lineno"> 371</span>&#160;}</div>
470<div class="line"><a name="l00372"></a><span class="lineno"> 372</span>&#160; </div>
471<div class="line"><a name="l00373"></a><span class="lineno"> 373</span>&#160;<span class="comment">// Detect a layer sequence and activation if specified. The activation must be at the end of the sequence.</span></div>
472<div class="line"><a name="l00374"></a><span class="lineno"> 374</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> TYPE&gt;</div>
473<div class="line"><a name="l00375"></a><span class="lineno"><a class="line" href="namespacearmnn.html#aa4452464859a025825afea5672640186"> 375</a></span>&#160;<span class="keywordtype">bool</span> <a class="code" href="namespacearmnn.html#aa4452464859a025825afea5672640186">IsLayerSequence</a>(<a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; currentLayer,</div>
474<div class="line"><a name="l00376"></a><span class="lineno"> 376</span>&#160; TYPE first,</div>
475<div class="line"><a name="l00377"></a><span class="lineno"> 377</span>&#160; TYPE second,</div>
476<div class="line"><a name="l00378"></a><span class="lineno"> 378</span>&#160; TYPE third,</div>
477<div class="line"><a name="l00379"></a><span class="lineno"> 379</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>* layerList[4],</div>
478<div class="line"><a name="l00380"></a><span class="lineno"> 380</span>&#160; <span class="keywordtype">bool</span> handleValidActivates,</div>
479<div class="line"><a name="l00381"></a><span class="lineno"> 381</span>&#160; <span class="keyword">const</span> std::vector&lt;ActivationFunction&gt;&amp; validActivates)</div>
480<div class="line"><a name="l00382"></a><span class="lineno"> 382</span>&#160;{</div>
481<div class="line"><a name="l00383"></a><span class="lineno"> 383</span>&#160; <span class="keyword">auto</span> PreviousLayer = [](<a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; layer)</div>
482<div class="line"><a name="l00384"></a><span class="lineno"> 384</span>&#160; {</div>
483<div class="line"><a name="l00385"></a><span class="lineno"> 385</span>&#160; <span class="keywordflow">return</span> &amp;layer.<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0).GetConnectedOutputSlot()-&gt;GetOwningLayer();</div>
484<div class="line"><a name="l00386"></a><span class="lineno"> 386</span>&#160; };</div>
485<div class="line"><a name="l00387"></a><span class="lineno"> 387</span>&#160; </div>
486<div class="line"><a name="l00388"></a><span class="lineno"> 388</span>&#160; <span class="keyword">auto</span> NextLayer = [](<a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; layer)</div>
487<div class="line"><a name="l00389"></a><span class="lineno"> 389</span>&#160; {</div>
488<div class="line"><a name="l00390"></a><span class="lineno"> 390</span>&#160; <span class="keywordflow">return</span> &amp;layer.<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_i_output_slot.html#a9fffb92dc6d2edd5258d574a5862fc83">GetConnection</a>(0)-&gt;GetOwningLayer();</div>
489<div class="line"><a name="l00391"></a><span class="lineno"> 391</span>&#160; };</div>
490<div class="line"><a name="l00392"></a><span class="lineno"> 392</span>&#160; </div>
491<div class="line"><a name="l00393"></a><span class="lineno"> 393</span>&#160; <span class="keyword">auto</span> LayerIncomingConnectionDataType = [](<a class="code" href="classarmnn_1_1_layer.html">Layer</a>&amp; layer)</div>
492<div class="line"><a name="l00394"></a><span class="lineno"> 394</span>&#160; {</div>
493<div class="line"><a name="l00395"></a><span class="lineno"> 395</span>&#160; <span class="keywordflow">return</span> layer.<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0).<a class="code" href="classarmnn_1_1_i_input_slot.html#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>().<a class="code" href="classarmnn_1_1_tensor_info.html#aea909c7327109228ef618d459015def3">GetDataType</a>();</div>
494<div class="line"><a name="l00396"></a><span class="lineno"> 396</span>&#160; };</div>
495<div class="line"><a name="l00397"></a><span class="lineno"> 397</span>&#160; </div>
496<div class="line"><a name="l00398"></a><span class="lineno"> 398</span>&#160; <span class="keywordtype">bool</span> result = <span class="keyword">false</span>;</div>
497<div class="line"><a name="l00399"></a><span class="lineno"> 399</span>&#160; </div>
498<div class="line"><a name="l00400"></a><span class="lineno"> 400</span>&#160; <span class="comment">// Match in reverse so there is only 1 connection to check</span></div>
499<div class="line"><a name="l00401"></a><span class="lineno"> 401</span>&#160; <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">IsSequenceLayerType</a>(currentLayer, third))</div>
500<div class="line"><a name="l00402"></a><span class="lineno"> 402</span>&#160; {</div>
501<div class="line"><a name="l00403"></a><span class="lineno"> 403</span>&#160; <span class="comment">// Save DataType of third layer</span></div>
502<div class="line"><a name="l00404"></a><span class="lineno"> 404</span>&#160; <a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6">DataType</a> dataType = LayerIncomingConnectionDataType(currentLayer);</div>
503<div class="line"><a name="l00405"></a><span class="lineno"> 405</span>&#160; </div>
504<div class="line"><a name="l00406"></a><span class="lineno"> 406</span>&#160; <span class="comment">// Save third layer</span></div>
505<div class="line"><a name="l00407"></a><span class="lineno"> 407</span>&#160; layerList[2] = &amp;currentLayer;</div>
506<div class="line"><a name="l00408"></a><span class="lineno"> 408</span>&#160; </div>
507<div class="line"><a name="l00409"></a><span class="lineno"> 409</span>&#160; <span class="comment">// Check the layers that proceed this one for the requested grouping</span></div>
508<div class="line"><a name="l00410"></a><span class="lineno"> 410</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a> *prevLayer = PreviousLayer(currentLayer);</div>
509<div class="line"><a name="l00411"></a><span class="lineno"> 411</span>&#160; <span class="keywordflow">if</span> (prevLayer &amp;&amp; <a class="code" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">IsSequenceLayerType</a>(*prevLayer, second))</div>
510<div class="line"><a name="l00412"></a><span class="lineno"> 412</span>&#160; {</div>
511<div class="line"><a name="l00413"></a><span class="lineno"> 413</span>&#160; <span class="keywordtype">bool</span> dataTypesMatch = (dataType == LayerIncomingConnectionDataType(*prevLayer));</div>
512<div class="line"><a name="l00414"></a><span class="lineno"> 414</span>&#160; <span class="keywordflow">if</span> (! dataTypesMatch)</div>
513<div class="line"><a name="l00415"></a><span class="lineno"> 415</span>&#160; {</div>
514<div class="line"><a name="l00416"></a><span class="lineno"> 416</span>&#160; <span class="keywordflow">return</span> result;</div>
515<div class="line"><a name="l00417"></a><span class="lineno"> 417</span>&#160; }</div>
516<div class="line"><a name="l00418"></a><span class="lineno"> 418</span>&#160; </div>
517<div class="line"><a name="l00419"></a><span class="lineno"> 419</span>&#160; layerList[1] = prevLayer;</div>
518<div class="line"><a name="l00420"></a><span class="lineno"> 420</span>&#160; prevLayer = PreviousLayer(*prevLayer);</div>
519<div class="line"><a name="l00421"></a><span class="lineno"> 421</span>&#160; <span class="keywordflow">if</span> (prevLayer &amp;&amp; <a class="code" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">IsSequenceLayerType</a>(*prevLayer, first))</div>
520<div class="line"><a name="l00422"></a><span class="lineno"> 422</span>&#160; {</div>
521<div class="line"><a name="l00423"></a><span class="lineno"> 423</span>&#160; dataTypesMatch = (dataType == LayerIncomingConnectionDataType(*prevLayer));</div>
522<div class="line"><a name="l00424"></a><span class="lineno"> 424</span>&#160; <span class="keywordflow">if</span> (! dataTypesMatch)</div>
523<div class="line"><a name="l00425"></a><span class="lineno"> 425</span>&#160; {</div>
524<div class="line"><a name="l00426"></a><span class="lineno"> 426</span>&#160; <span class="keywordflow">return</span> result;</div>
525<div class="line"><a name="l00427"></a><span class="lineno"> 427</span>&#160; }</div>
526<div class="line"><a name="l00428"></a><span class="lineno"> 428</span>&#160; </div>
527<div class="line"><a name="l00429"></a><span class="lineno"> 429</span>&#160; layerList[0] = prevLayer;</div>
528<div class="line"><a name="l00430"></a><span class="lineno"> 430</span>&#160; </div>
529<div class="line"><a name="l00431"></a><span class="lineno"> 431</span>&#160; <span class="comment">// Detected the first 3 layers if we get to this point so now</span></div>
530<div class="line"><a name="l00432"></a><span class="lineno"> 432</span>&#160; <span class="comment">// check to see if we have a valid activation. If there is no activation</span></div>
531<div class="line"><a name="l00433"></a><span class="lineno"> 433</span>&#160; <span class="comment">// then the sequence still matches.</span></div>
532<div class="line"><a name="l00434"></a><span class="lineno"> 434</span>&#160; <span class="keywordflow">if</span> (handleValidActivates)</div>
533<div class="line"><a name="l00435"></a><span class="lineno"> 435</span>&#160; {</div>
534<div class="line"><a name="l00436"></a><span class="lineno"> 436</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a> *nextLayer = NextLayer(currentLayer);</div>
535<div class="line"><a name="l00437"></a><span class="lineno"> 437</span>&#160; <span class="keywordflow">if</span> (nextLayer)</div>
536<div class="line"><a name="l00438"></a><span class="lineno"> 438</span>&#160; {</div>
537<div class="line"><a name="l00439"></a><span class="lineno"> 439</span>&#160; <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">IsSequenceLayerType</a>(*nextLayer, <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa9a62e70841c4d06dd16306a85700d36">LayerType::Activation</a>))</div>
538<div class="line"><a name="l00440"></a><span class="lineno"> 440</span>&#160; {</div>
539<div class="line"><a name="l00441"></a><span class="lineno"> 441</span>&#160; <span class="comment">// This layer is an activation, so it must be a valid type for the sequence</span></div>
540<div class="line"><a name="l00442"></a><span class="lineno"> 442</span>&#160; <a class="code" href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9ea">ActivationFunction</a> activationFunction =</div>
541<div class="line"><a name="l00443"></a><span class="lineno"> 443</span>&#160; PolymorphicDowncast&lt;ActivationLayer*&gt;(nextLayer)-&gt;GetParameters().m_Function;</div>
542<div class="line"><a name="l00444"></a><span class="lineno"> 444</span>&#160; <span class="keywordtype">long</span> count = std::count(validActivates.cbegin(),</div>
543<div class="line"><a name="l00445"></a><span class="lineno"> 445</span>&#160; validActivates.cend(),</div>
544<div class="line"><a name="l00446"></a><span class="lineno"> 446</span>&#160; activationFunction);</div>
545<div class="line"><a name="l00447"></a><span class="lineno"> 447</span>&#160; <span class="keywordflow">if</span> (count &gt; 0)</div>
546<div class="line"><a name="l00448"></a><span class="lineno"> 448</span>&#160; {</div>
547<div class="line"><a name="l00449"></a><span class="lineno"> 449</span>&#160; layerList[3] = nextLayer;</div>
548<div class="line"><a name="l00450"></a><span class="lineno"> 450</span>&#160; result = <span class="keyword">true</span>;</div>
549<div class="line"><a name="l00451"></a><span class="lineno"> 451</span>&#160; }</div>
550<div class="line"><a name="l00452"></a><span class="lineno"> 452</span>&#160; }</div>
551<div class="line"><a name="l00453"></a><span class="lineno"> 453</span>&#160; <span class="keywordflow">else</span></div>
552<div class="line"><a name="l00454"></a><span class="lineno"> 454</span>&#160; {</div>
553<div class="line"><a name="l00455"></a><span class="lineno"> 455</span>&#160; <span class="comment">// Next layer is not an activation so sequence still matches</span></div>
554<div class="line"><a name="l00456"></a><span class="lineno"> 456</span>&#160; result = <span class="keyword">true</span>;</div>
555<div class="line"><a name="l00457"></a><span class="lineno"> 457</span>&#160; }</div>
556<div class="line"><a name="l00458"></a><span class="lineno"> 458</span>&#160; }</div>
557<div class="line"><a name="l00459"></a><span class="lineno"> 459</span>&#160; }</div>
558<div class="line"><a name="l00460"></a><span class="lineno"> 460</span>&#160; <span class="keywordflow">else</span></div>
559<div class="line"><a name="l00461"></a><span class="lineno"> 461</span>&#160; {</div>
560<div class="line"><a name="l00462"></a><span class="lineno"> 462</span>&#160; result = <span class="keyword">true</span>;</div>
561<div class="line"><a name="l00463"></a><span class="lineno"> 463</span>&#160; }</div>
562<div class="line"><a name="l00464"></a><span class="lineno"> 464</span>&#160; }</div>
563<div class="line"><a name="l00465"></a><span class="lineno"> 465</span>&#160; }</div>
564<div class="line"><a name="l00466"></a><span class="lineno"> 466</span>&#160; }</div>
565<div class="line"><a name="l00467"></a><span class="lineno"> 467</span>&#160; </div>
566<div class="line"><a name="l00468"></a><span class="lineno"> 468</span>&#160; <span class="keywordflow">return</span> result;</div>
567<div class="line"><a name="l00469"></a><span class="lineno"> 469</span>&#160;}</div>
568<div class="line"><a name="l00470"></a><span class="lineno"> 470</span>&#160; </div>
569<div class="line"><a name="l00471"></a><span class="lineno"> 471</span>&#160;} <span class="comment">// namespace armnn</span></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]570</div><!-- fragment --></div><!-- contents -->
571</div><!-- doc-content -->
572<div class="ttc" id="astructarmnn_1_1_batch_normalization_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_batch_normalization_descriptor.html">armnn::BatchNormalizationDescriptor</a></div><div class="ttdoc">A BatchNormalizationDescriptor for the BatchNormalizationLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00828">Descriptors.hpp:828</a></div></div>
573<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a5e7ff12da912dc79e7e547281823fa4a"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a5e7ff12da912dc79e7e547281823fa4a">armnn::LayerType::SpaceToDepth</a></div><div class="ttdeci">@ SpaceToDepth</div></div>
574<div class="ttc" id="aclassarmnn_1_1_optimization_views_html_a28e41bdd6b719a3d60a1a0de2e1ebc95"><div class="ttname"><a href="classarmnn_1_1_optimization_views.html#a28e41bdd6b719a3d60a1a0de2e1ebc95">armnn::OptimizationViews::AddUntouchedSubgraph</a></div><div class="ttdeci">void AddUntouchedSubgraph(SubgraphView &amp;&amp;subgraph)</div><div class="ttdef"><b>Definition:</b> <a href="_optimization_views_8hpp_source.html#l00048">OptimizationViews.hpp:48</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]575<div class="ttc" id="aclassarmnn_1_1_input_slot_html_a7f4ecb20ab4968de368166d8cbb2b1a3"><div class="ttname"><a href="classarmnn_1_1_input_slot.html#a7f4ecb20ab4968de368166d8cbb2b1a3">armnn::InputSlot::SetTensorInfo</a></div><div class="ttdeci">void SetTensorInfo(const TensorInfo tensorInfo) override</div><div class="ttdoc">Sets the TensorInfo for this InputSlot.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8cpp_source.html#l00609">Layer.cpp:609</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]576<div class="ttc" id="aclassarmnn_1_1_subgraph_view_html_a3f717693851810942d3ae0180a97fb4c"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.html#a3f717693851810942d3ae0180a97fb4c">armnn::SubgraphView::IOutputSlots</a></div><div class="ttdeci">std::vector&lt; IOutputSlot * &gt; IOutputSlots</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_8hpp_source.html#l00060">SubgraphView.hpp:60</a></div></div>
577<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4ae4743c3ec15d1d84169b17264634692e"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4ae4743c3ec15d1d84169b17264634692e">armnn::LayerType::BatchNormalization</a></div><div class="ttdeci">@ BatchNormalization</div></div>
578<div class="ttc" id="anamespacearmnn_html_ad1d5cce2d9e9a5d61c243e5c989112e0a884e0167ebf9bbe6cfd6ca5ab97ab015"><div class="ttname"><a href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0a884e0167ebf9bbe6cfd6ca5ab97ab015">armnn::DataLayout::NCDHW</a></div><div class="ttdeci">@ NCDHW</div></div>
579<div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html_a9c2cba04b6d7ace4fc2a2436b82a5a63"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html#a9c2cba04b6d7ace4fc2a2436b82a5a63">armnn::IConnectableLayer::GetNumInputSlots</a></div><div class="ttdeci">virtual unsigned int GetNumInputSlots() const =0</div><div class="ttdoc">Returns the number of connectable input slots.</div></div>
580<div class="ttc" id="aclassarmnn_1_1_input_slot_html_a7ddaf04177053a536f0e7be83a642bc6"><div class="ttname"><a href="classarmnn_1_1_input_slot.html#a7ddaf04177053a536f0e7be83a642bc6">armnn::InputSlot::GetOwningLayer</a></div><div class="ttdeci">Layer &amp; GetOwningLayer() const</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00053">Layer.hpp:53</a></div></div>
581<div class="ttc" id="a_descriptors_8hpp_html"><div class="ttname"><a href="_descriptors_8hpp.html">Descriptors.hpp</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]582<div class="ttc" id="aclassarmnn_1_1_output_slot_html_ada2ad7d1caeeb4ef6195c8925fad6a65"><div class="ttname"><a href="classarmnn_1_1_output_slot.html#ada2ad7d1caeeb4ef6195c8925fad6a65">armnn::OutputSlot::GetTensorInfo</a></div><div class="ttdeci">const TensorInfo &amp; GetTensorInfo() const override</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8cpp_source.html#l00100">Layer.cpp:100</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]583<div class="ttc" id="astructarmnn_1_1_pooling3d_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_pooling3d_descriptor.html">armnn::Pooling3dDescriptor</a></div><div class="ttdoc">A Pooling3dDescriptor for the Pooling3dLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00431">Descriptors.hpp:431</a></div></div>
584<div class="ttc" id="astructarmnn_1_1_instance_normalization_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_instance_normalization_descriptor.html">armnn::InstanceNormalizationDescriptor</a></div><div class="ttdoc">An InstanceNormalizationDescriptor for InstanceNormalizationLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00847">Descriptors.hpp:847</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]585<div class="ttc" id="anamespacearmnn_html_a4f89a6ce848b8d9787761cdd1dbb751a"><div class="ttname"><a href="namespacearmnn.html#a4f89a6ce848b8d9787761cdd1dbb751a">armnn::ConnectedToLayerType</a></div><div class="ttdeci">bool ConnectedToLayerType(Layer *baseLayer, LayerType layerType, unsigned int dimSize=0)</div><div class="ttdoc">Checks the Layer's Connections to see if it's connected to a Layer with the provided layerType.</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00271">SubgraphUtils.hpp:271</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]586<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a21baa4498161d195f5bb2e3627344ba4"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a21baa4498161d195f5bb2e3627344ba4">armnn::LayerType::InstanceNormalization</a></div><div class="ttdeci">@ InstanceNormalization</div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]587<div class="ttc" id="anamespacearmnn_html_aa4452464859a025825afea5672640186"><div class="ttname"><a href="namespacearmnn.html#aa4452464859a025825afea5672640186">armnn::IsLayerSequence</a></div><div class="ttdeci">bool IsLayerSequence(Layer &amp;currentLayer, TYPE first, TYPE second, TYPE third, Layer *layerList[4], bool handleValidActivates, const std::vector&lt; ActivationFunction &gt; &amp;validActivates)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00375">SubgraphUtils.hpp:375</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]588<div class="ttc" id="aclassarmnn_1_1_tensor_info_html"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a></div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00152">Tensor.hpp:152</a></div></div>
589<div class="ttc" id="astructarmnn_1_1_l2_normalization_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_l2_normalization_descriptor.html">armnn::L2NormalizationDescriptor</a></div><div class="ttdoc">A L2NormalizationDescriptor for the L2NormalizationLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00809">Descriptors.hpp:809</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]590<div class="ttc" id="anamespacearmnn_html_aefc7bff008047b9f763c2cf82c1a0fcb"><div class="ttname"><a href="namespacearmnn.html#aefc7bff008047b9f763c2cf82c1a0fcb">armnn::IsSequenceLayerType</a></div><div class="ttdeci">bool IsSequenceLayerType(Layer &amp;layer, LayerType type)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00362">SubgraphUtils.hpp:362</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]591<div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html">armnn::NormalizationDescriptor</a></div><div class="ttdoc">A NormalizationDescriptor for the NormalizationLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00769">Descriptors.hpp:769</a></div></div>
Nikhil Raj38b600d2024-02-15 15:02:19 +0000[diff] [blame]592<div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a157e27d41e9f6b21f0d3c025fa47dc24"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">armnn::TensorInfo::GetNumDimensions</a></div><div class="ttdeci">unsigned int GetNumDimensions() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00197">Tensor.hpp:197</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]593<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4aa31904f2b3479b5a00137fd985974b4d"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa31904f2b3479b5a00137fd985974b4d">armnn::LayerType::StridedSlice</a></div><div class="ttdeci">@ StridedSlice</div></div>
594<div class="ttc" id="aclassarmnn_1_1_layer_html_a0e36688a43c35668d8db5257274c68fe"><div class="ttname"><a href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">armnn::Layer::GetOutputSlot</a></div><div class="ttdeci">const OutputSlot &amp; GetOutputSlot(unsigned int index=0) const override</div><div class="ttdoc">Get the const output slot handle by slot index.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00339">Layer.hpp:339</a></div></div>
595<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4aac61f2e17250a818dee4d12b112aa88f"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aac61f2e17250a818dee4d12b112aa88f">armnn::LayerType::Normalization</a></div><div class="ttdeci">@ Normalization</div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]596<div class="ttc" id="anamespacearmnn_html_ae4b69405416c09ee6878e01d27c9450b"><div class="ttname"><a href="namespacearmnn.html#ae4b69405416c09ee6878e01d27c9450b">armnn::FoldPadLayer</a></div><div class="ttdeci">LayerType * FoldPadLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, LayerType *replacementLayer, PadLayer *padLayer)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00234">SubgraphUtils.hpp:234</a></div></div>
597<div class="ttc" id="anamespacearmnn_html_ab6b4f5f3473b0ce4b241d454f8cbdd26"><div class="ttname"><a href="namespacearmnn.html#ab6b4f5f3473b0ce4b241d454f8cbdd26">armnn::IsNCHW</a></div><div class="ttdeci">bool IsNCHW(armnn::Layer &amp;layer)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00213">SubgraphUtils.hpp:213</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]598<div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html_ac2dac3b61c94de52093616be4ab17f8d"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html#ac2dac3b61c94de52093616be4ab17f8d">armnn::IConnectableLayer::GetNumOutputSlots</a></div><div class="ttdeci">virtual unsigned int GetNumOutputSlots() const =0</div><div class="ttdoc">Returns the number of connectable output slots.</div></div>
599<div class="ttc" id="aclassarmnn_1_1_layer_html_acf8b8e23bf647836592982f97088d375"><div class="ttname"><a href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">armnn::Layer::GetInputSlot</a></div><div class="ttdeci">const InputSlot &amp; GetInputSlot(unsigned int index) const override</div><div class="ttdoc">Get a const input slot handle by slot index.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00337">Layer.hpp:337</a></div></div>
600<div class="ttc" id="aclassarmnn_1_1_layer_with_parameters_html_afa3e8a8f23589b1eaddbe203825bbdcf"><div class="ttname"><a href="classarmnn_1_1_layer_with_parameters.html#afa3e8a8f23589b1eaddbe203825bbdcf">armnn::LayerWithParameters::GetParameters</a></div><div class="ttdeci">const Parameters &amp; GetParameters() const override</div><div class="ttdoc">If the layer has a descriptor return it.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_with_parameters_8hpp_source.html#l00019">LayerWithParameters.hpp:19</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]601<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a699bcffd93aff3022014b9efc9eaefd1"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a699bcffd93aff3022014b9efc9eaefd1">armnn::LayerType::ElementwiseBinary</a></div><div class="ttdeci">@ ElementwiseBinary</div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]602<div class="ttc" id="aclassarmnn_1_1_layer_html"><div class="ttname"><a href="classarmnn_1_1_layer.html">armnn::Layer</a></div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00230">Layer.hpp:230</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]603<div class="ttc" id="aclassarmnn_1_1_input_slot_html_ada2ad7d1caeeb4ef6195c8925fad6a65"><div class="ttname"><a href="classarmnn_1_1_input_slot.html#ada2ad7d1caeeb4ef6195c8925fad6a65">armnn::InputSlot::GetTensorInfo</a></div><div class="ttdeci">const TensorInfo &amp; GetTensorInfo() const override</div><div class="ttdoc">Gets the TensorInfo for this InputSlot.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8cpp_source.html#l00614">Layer.cpp:614</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]604<div class="ttc" id="aclassarmnn_1_1_input_slot_html_ae3f69b193433448bee0f3925da4b01a3"><div class="ttname"><a href="classarmnn_1_1_input_slot.html#ae3f69b193433448bee0f3925da4b01a3">armnn::InputSlot::GetSlotIndex</a></div><div class="ttdeci">unsigned int GetSlotIndex() const override</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00054">Layer.hpp:54</a></div></div>
605<div class="ttc" id="a_fold_pad_into_layer2d_8hpp_html"><div class="ttname"><a href="_fold_pad_into_layer2d_8hpp.html">FoldPadIntoLayer2d.hpp</a></div></div>
606<div class="ttc" id="aclassarmnn_1_1_output_slot_html_a7ddaf04177053a536f0e7be83a642bc6"><div class="ttname"><a href="classarmnn_1_1_output_slot.html#a7ddaf04177053a536f0e7be83a642bc6">armnn::OutputSlot::GetOwningLayer</a></div><div class="ttdeci">Layer &amp; GetOwningLayer() const</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00132">Layer.hpp:132</a></div></div>
607<div class="ttc" id="aclassarmnn_1_1_reshape_layer_html"><div class="ttname"><a href="classarmnn_1_1_reshape_layer.html">armnn::ReshapeLayer</a></div><div class="ttdoc">This layer represents a reshape operation.</div><div class="ttdef"><b>Definition:</b> <a href="_reshape_layer_8hpp_source.html#l00015">ReshapeLayer.hpp:15</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]608<div class="ttc" id="aclassarmnn_1_1_i_output_slot_html_a9fffb92dc6d2edd5258d574a5862fc83"><div class="ttname"><a href="classarmnn_1_1_i_output_slot.html#a9fffb92dc6d2edd5258d574a5862fc83">armnn::IOutputSlot::GetConnection</a></div><div class="ttdeci">virtual const IInputSlot * GetConnection(unsigned int index) const =0</div></div>
609<div class="ttc" id="anamespacearmnn_html_ae756568a7f2c58778bb6d759d2bd38db"><div class="ttname"><a href="namespacearmnn.html#ae756568a7f2c58778bb6d759d2bd38db">armnn::RemoveReshapeLayer</a></div><div class="ttdeci">void RemoveReshapeLayer(ReshapeLayer *baseLayer, std::map&lt; LayerGuid, Layer * &gt; &amp;untouched, OptimizationViews &amp;optimizationViews)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00293">SubgraphUtils.hpp:293</a></div></div>
610<div class="ttc" id="a_exceptions_8hpp_html_a31e4268013316d114e8927e4c0035ec5"><div class="ttname"><a href="_exceptions_8hpp.html#a31e4268013316d114e8927e4c0035ec5">ARMNN_THROW_INVALIDARG_IF_FALSE</a></div><div class="ttdeci">#define ARMNN_THROW_INVALIDARG_IF_FALSE(_cond)</div><div class="ttdef"><b>Definition:</b> <a href="_exceptions_8hpp_source.html#l00212">Exceptions.hpp:212</a></div></div>
611<div class="ttc" id="anamespacearmnn_html_ad8ed01ff3ff33333d8e19db4d2818bb6"><div class="ttname"><a href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6">armnn::DataType</a></div><div class="ttdeci">DataType</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00048">Types.hpp:48</a></div></div>
612<div class="ttc" id="anamespacearmnn_html_a0a49066b56fc3bfe44478601c31e2fc0"><div class="ttname"><a href="namespacearmnn.html#a0a49066b56fc3bfe44478601c31e2fc0">armnn::FoldPadIntoAveragePool2d</a></div><div class="ttdeci">LayerType * FoldPadIntoAveragePool2d(OptimizationViews &amp;optimizationViews, Pooling2dLayer *baseLayer, Pooling2dDescriptor &amp;poolDescriptor, PadLayer *padLayer)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00341">SubgraphUtils.hpp:341</a></div></div>
613<div class="ttc" id="astructarmnn_1_1_reshape_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_reshape_descriptor.html">armnn::ReshapeDescriptor</a></div><div class="ttdoc">A ReshapeDescriptor for the ReshapeLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01023">Descriptors.hpp:1023</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]614<div class="ttc" id="aclassarmnn_1_1_subgraph_view_html_a4ad949634fd697549da161eb09d8b7fd"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.html#a4ad949634fd697549da161eb09d8b7fd">armnn::SubgraphView::IInputSlots</a></div><div class="ttdeci">std::vector&lt; IInputSlot * &gt; IInputSlots</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_view_8hpp_source.html#l00058">SubgraphView.hpp:58</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]615<div class="ttc" id="anamespacearmnn_html_ab8cf8f9fb6792e654c2d8d8382f6f01b"><div class="ttname"><a href="namespacearmnn.html#ab8cf8f9fb6792e654c2d8d8382f6f01b">armnn::LayerBindingId</a></div><div class="ttdeci">int LayerBindingId</div><div class="ttdoc">Type of identifiers for bindable layers (inputs, outputs).</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00309">Types.hpp:309</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]616<div class="ttc" id="aclassarmnn_1_1_layer_html_a8dc12f0ee5b232d397bd18ced1a72a64"><div class="ttname"><a href="classarmnn_1_1_layer.html#a8dc12f0ee5b232d397bd18ced1a72a64">armnn::Layer::GetGuid</a></div><div class="ttdeci">LayerGuid GetGuid() const final</div><div class="ttdoc">Returns the unique id of the layer.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00343">Layer.hpp:343</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]617<div class="ttc" id="astructarmnn_1_1_batch_mat_mul_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_batch_mat_mul_descriptor.html">armnn::BatchMatMulDescriptor</a></div><div class="ttdoc">A BatchMatMulDescriptor for the BatchMatMul operator.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01584">Descriptors.hpp:1584</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]618<div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html_adceb04ae84c524e4d01881e3754a4d59"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html#adceb04ae84c524e4d01881e3754a4d59">armnn::IConnectableLayer::GetType</a></div><div class="ttdeci">virtual LayerType GetType() const =0</div><div class="ttdoc">Returns the armnn::LayerType of this layer.</div></div>
619<div class="ttc" id="aclassarmnn_1_1_subgraph_view_html"><div class="ttname"><a href="classarmnn_1_1_subgraph_view.html">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.html#l00031">SubgraphView.hpp:31</a></div></div>
620<div class="ttc" id="aclassarmnn_1_1_optimization_views_html"><div class="ttname"><a href="classarmnn_1_1_optimization_views.html">armnn::OptimizationViews</a></div><div class="ttdef"><b>Definition:</b> <a href="_optimization_views_8hpp_source.html#l00017">OptimizationViews.hpp:17</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]621<div class="ttc" id="astructarmnn_1_1_space_to_batch_nd_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_space_to_batch_nd_descriptor.html">armnn::SpaceToBatchNdDescriptor</a></div><div class="ttdoc">A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01043">Descriptors.hpp:1043</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]622<div class="ttc" id="astructarmnn_1_1_convolution3d_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_convolution3d_descriptor.html">armnn::Convolution3dDescriptor</a></div><div class="ttdoc">A Convolution3dDescriptor for the Convolution3dLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00588">Descriptors.hpp:588</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]623<div class="ttc" id="astructarmnn_1_1_reshape_descriptor_html_a1178f4dafdda81f59c15145ec327f7d9"><div class="ttname"><a href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">armnn::ReshapeDescriptor::m_TargetShape</a></div><div class="ttdeci">TensorShape m_TargetShape</div><div class="ttdoc">Target shape value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01039">Descriptors.hpp:1039</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]624<div class="ttc" id="aclassarmnn_1_1_pooling2d_layer_html"><div class="ttname"><a href="classarmnn_1_1_pooling2d_layer.html">armnn::Pooling2dLayer</a></div><div class="ttdoc">This layer represents a pooling 2d operation.</div><div class="ttdef"><b>Definition:</b> <a href="_pooling2d_layer_8hpp_source.html#l00013">Pooling2dLayer.hpp:13</a></div></div>
625<div class="ttc" id="astructarmnn_1_1_base_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_base_descriptor.html">armnn::BaseDescriptor</a></div><div class="ttdoc">Base class for all descriptors.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00022">Descriptors.hpp:22</a></div></div>
626<div class="ttc" id="aclassarmnn_1_1_optimization_views_html_a57340147dcb1c0ae05fdf14ff7af8a01"><div class="ttname"><a href="classarmnn_1_1_optimization_views.html#a57340147dcb1c0ae05fdf14ff7af8a01">armnn::OptimizationViews::AddSubstitution</a></div><div class="ttdeci">void AddSubstitution(SubstitutionPair &amp;&amp;substitution)</div><div class="ttdef"><b>Definition:</b> <a href="_optimization_views_8hpp_source.html#l00038">OptimizationViews.hpp:38</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]627<div class="ttc" id="anamespacearmnn_html_a56297e0f7b215eea46c818cb7528d9ea"><div class="ttname"><a href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9ea">armnn::ActivationFunction</a></div><div class="ttdeci">ActivationFunction</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00086">Types.hpp:86</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]628<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4ad662867a41bfb30b9f75dda2b5849001"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4ad662867a41bfb30b9f75dda2b5849001">armnn::LayerType::Pooling2d</a></div><div class="ttdeci">@ Pooling2d</div></div>
Nikhil Raj38b600d2024-02-15 15:02:19 +0000[diff] [blame]629<div class="ttc" id="aclassarmnn_1_1_tensor_info_html_aea909c7327109228ef618d459015def3"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#aea909c7327109228ef618d459015def3">armnn::TensorInfo::GetDataType</a></div><div class="ttdeci">DataType GetDataType() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00200">Tensor.hpp:200</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]630<div class="ttc" id="astructarmnn_1_1_batch_to_space_nd_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_batch_to_space_nd_descriptor.html">armnn::BatchToSpaceNdDescriptor</a></div><div class="ttdoc">A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00875">Descriptors.hpp:875</a></div></div>
631<div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html">armnn::Convolution2dDescriptor</a></div><div class="ttdoc">A Convolution2dDescriptor for the Convolution2dLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00534">Descriptors.hpp:534</a></div></div>
632<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a2b3140dc366b9fcd25ed786a79d1817c"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a2b3140dc366b9fcd25ed786a79d1817c">armnn::LayerType::Pooling3d</a></div><div class="ttdeci">@ Pooling3d</div></div>
633<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a9882ff3cfed27d6161c20a305e7a3484"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a9882ff3cfed27d6161c20a305e7a3484">armnn::LayerType::BatchMatMul</a></div><div class="ttdeci">@ BatchMatMul</div></div>
634<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4af97adbfc88b7012a0243215b1076e7e7"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4af97adbfc88b7012a0243215b1076e7e7">armnn::LayerType::DepthwiseConvolution2d</a></div><div class="ttdeci">@ DepthwiseConvolution2d</div></div>
635<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a6ee06c6045d0c5b6565a247955ef0fc2"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a6ee06c6045d0c5b6565a247955ef0fc2">armnn::LayerType::BatchToSpaceNd</a></div><div class="ttdeci">@ BatchToSpaceNd</div></div>
636<div class="ttc" id="aclassarmnn_1_1_layer_html_ad8e15c530c929ab823d89ae9fd2d3f11"><div class="ttname"><a href="classarmnn_1_1_layer.html#ad8e15c530c929ab823d89ae9fd2d3f11">armnn::Layer::GetType</a></div><div class="ttdeci">LayerType GetType() const override</div><div class="ttdoc">Returns the armnn::LayerType of this layer.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00286">Layer.hpp:286</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]637<div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html">armnn::StridedSliceDescriptor</a></div><div class="ttdoc">A StridedSliceDescriptor for the StridedSliceLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01303">Descriptors.hpp:1303</a></div></div>
638<div class="ttc" id="anamespacearmnn_html_ac4f8557279754ed7b3f749d55b0e3047"><div class="ttname"><a href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047">armnn::BinaryOperation</a></div><div class="ttdeci">BinaryOperation</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00138">Types.hpp:138</a></div></div>
639<div class="ttc" id="aclassarmnn_1_1_i_input_slot_html_a9943775a364fc4ab53b85ac88f311886"><div class="ttname"><a href="classarmnn_1_1_i_input_slot.html#a9943775a364fc4ab53b85ac88f311886">armnn::IInputSlot::GetTensorInfo</a></div><div class="ttdeci">virtual const TensorInfo &amp; GetTensorInfo() const =0</div><div class="ttdoc">Gets the TensorInfo for this InputSlot.</div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]640<div class="ttc" id="aclassarmnn_1_1_layer_html_a46fc3fdd4b2a5dd6d184e57983cf20bc"><div class="ttname"><a href="classarmnn_1_1_layer.html#a46fc3fdd4b2a5dd6d184e57983cf20bc">armnn::Layer::ExecuteStrategy</a></div><div class="ttdeci">void ExecuteStrategy(IStrategy &amp;strategy) const override</div><div class="ttdoc">Apply a visitor to this layer.</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8cpp_source.html#l00571">Layer.cpp:571</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]641<div class="ttc" id="aclassarmnn_1_1_i_network_html_ae913b4351b7027f37eb5657dd7867733"><div class="ttname"><a href="classarmnn_1_1_i_network.html#ae913b4351b7027f37eb5657dd7867733">armnn::INetwork::AddPooling2dLayer</a></div><div class="ttdeci">IConnectableLayer * AddPooling2dLayer(const Pooling2dDescriptor &amp;pooling2dDescriptor, const char *name=nullptr)</div><div class="ttdoc">Adds a 2D pooling layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00356">Network.cpp:356</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]642<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a337c392144dca0d18290c6b4711a2279"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a337c392144dca0d18290c6b4711a2279">armnn::LayerType::SpaceToBatchNd</a></div><div class="ttdeci">@ SpaceToBatchNd</div></div>
643<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a1189467870bc421ae59277e750263eb2"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a1189467870bc421ae59277e750263eb2">armnn::LayerType::L2Normalization</a></div><div class="ttdeci">@ L2Normalization</div></div>
644<div class="ttc" id="aclassarmnn_1_1_optimization_views_html_a4438f2ce297390e50f48dc426ba1f4f9"><div class="ttname"><a href="classarmnn_1_1_optimization_views.html#a4438f2ce297390e50f48dc426ba1f4f9">armnn::OptimizationViews::AddDeletedSubgraph</a></div><div class="ttdeci">void AddDeletedSubgraph(SubgraphView &amp;&amp;subgraph)</div><div class="ttdef"><b>Definition:</b> <a href="_optimization_views_8hpp_source.html#l00053">OptimizationViews.hpp:53</a></div></div>
645<div class="ttc" id="anamespacearmnn_html_a44affeeb090c3c6a3062830562672e84"><div class="ttname"><a href="namespacearmnn.html#a44affeeb090c3c6a3062830562672e84">armnn::IgnoreUnused</a></div><div class="ttdeci">void IgnoreUnused(Ts &amp;&amp;...)</div><div class="ttdef"><b>Definition:</b> <a href="_ignore_unused_8hpp_source.html#l00014">IgnoreUnused.hpp:14</a></div></div>
646<div class="ttc" id="aclassarmnn_1_1_input_slot_html_a9effd325a6d512a3f8ff4bd207d53255"><div class="ttname"><a href="classarmnn_1_1_input_slot.html#a9effd325a6d512a3f8ff4bd207d53255">armnn::InputSlot::GetConnectedOutputSlot</a></div><div class="ttdeci">const OutputSlot * GetConnectedOutputSlot() const</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8hpp_source.html#l00056">Layer.hpp:56</a></div></div>
Nikhil Raj38b600d2024-02-15 15:02:19 +0000[diff] [blame]647<div class="ttc" id="aclassarmnn_1_1_tensor_info_html_abe8889e8150beef5fd204b2d87b49298"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">armnn::TensorInfo::SetShape</a></div><div class="ttdeci">void SetShape(const TensorShape &amp;newShape)</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00195">Tensor.hpp:195</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]648<div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html_a80ac4eda2e7f2757ec9dd96fc96dbd16"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">armnn::IConnectableLayer::GetOutputSlot</a></div><div class="ttdeci">virtual const IOutputSlot &amp; GetOutputSlot(unsigned int index) const =0</div><div class="ttdoc">Get the const output slot handle by slot index.</div></div>
649<div class="ttc" id="anamespacearmnn_html"><div class="ttname"><a href="namespacearmnn.html">armnn</a></div><div class="ttdoc">Copyright (c) 2021 ARM Limited and Contributors.</div><div class="ttdef"><b>Definition:</b> <a href="01__00__quick__start_8dox_source.html#l00006">01_00_quick_start.dox:6</a></div></div>
650<div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html_a6ec9e0eb66d7d6a01240492a0b18104c"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">armnn::IConnectableLayer::GetInputSlot</a></div><div class="ttdeci">virtual const IInputSlot &amp; GetInputSlot(unsigned int index) const =0</div><div class="ttdoc">Get a const input slot handle by slot index.</div></div>
651<div class="ttc" id="aclassarmnn_1_1_optimization_views_html_ad04187fe81f68558b15b6049b2da9cf9"><div class="ttname"><a href="classarmnn_1_1_optimization_views.html#ad04187fe81f68558b15b6049b2da9cf9">armnn::OptimizationViews::GetINetwork</a></div><div class="ttdeci">INetwork * GetINetwork()</div><div class="ttdef"><b>Definition:</b> <a href="_optimization_views_8hpp_source.html#l00069">OptimizationViews.hpp:69</a></div></div>
652<div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a></div><div class="ttdoc">Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.</div><div class="ttdef"><b>Definition:</b> <a href="_i_network_8hpp_source.html#l00080">INetwork.hpp:80</a></div></div>
653<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a324118a6721dd6b8a9b9f4e327df2bf5"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a324118a6721dd6b8a9b9f4e327df2bf5">armnn::LayerType::Input</a></div><div class="ttdeci">@ Input</div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]654<div class="ttc" id="anamespacearmnn_html_a5afd10e5e84ebf5cb8cfc8707492eda7"><div class="ttname"><a href="namespacearmnn.html#a5afd10e5e84ebf5cb8cfc8707492eda7">armnn::ReportUntouchedLayers</a></div><div class="ttdeci">void ReportUntouchedLayers(OptimizationViews &amp;optimizationViews, std::map&lt; LayerGuid, Layer * &gt; untouched)</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00220">SubgraphUtils.hpp:220</a></div></div>
655<div class="ttc" id="anamespacearmnn_html_ac5089a2ee61f22570982e7a56588365a"><div class="ttname"><a href="namespacearmnn.html#ac5089a2ee61f22570982e7a56588365a">armnn::ConnectedToLayerWithNCHW</a></div><div class="ttdeci">bool ConnectedToLayerWithNCHW(Layer *baseLayer)</div><div class="ttdoc">Checks if the Layer is connected to any Layer that has an NCHW layout.</div><div class="ttdef"><b>Definition:</b> <a href="_subgraph_utils_8hpp_source.html#l00250">SubgraphUtils.hpp:250</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]656<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4adb033d2f81b68f9a17e8f62de69fed4a"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4adb033d2f81b68f9a17e8f62de69fed4a">armnn::LayerType::Convolution2d</a></div><div class="ttdeci">@ Convolution2d</div></div>
657<div class="ttc" id="astructarmnn_1_1_pooling2d_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_pooling2d_descriptor.html">armnn::Pooling2dDescriptor</a></div><div class="ttdoc">A Pooling2dDescriptor for the Pooling2dLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00371">Descriptors.hpp:371</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]658<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4aa9a62e70841c4d06dd16306a85700d36"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4aa9a62e70841c4d06dd16306a85700d36">armnn::LayerType::Activation</a></div><div class="ttdeci">@ Activation</div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]659<div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html">armnn::DepthwiseConvolution2dDescriptor</a></div><div class="ttdoc">A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00659">Descriptors.hpp:659</a></div></div>
660<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a583550d0f265fd3756f7de0e42c51953"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a583550d0f265fd3756f7de0e42c51953">armnn::LayerType::Convolution3d</a></div><div class="ttdeci">@ Convolution3d</div></div>
661<div class="ttc" id="a_strategy_base_8hpp_html"><div class="ttname"><a href="_strategy_base_8hpp.html">StrategyBase.hpp</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]662<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">armnn::LayerType</a></div><div class="ttdeci">LayerType</div><div class="ttdoc">When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below.</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00491">Types.hpp:491</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]663<div class="ttc" id="aclassarmnn_1_1_output_slot_html_a048e8c1536cf6b8871b093a73a4a3a85"><div class="ttname"><a href="classarmnn_1_1_output_slot.html#a048e8c1536cf6b8871b093a73a4a3a85">armnn::OutputSlot::GetConnection</a></div><div class="ttdeci">const InputSlot * GetConnection(unsigned int index) const override</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8cpp_source.html#l00083">Layer.cpp:83</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]664<div class="ttc" id="aclassarmnn_1_1_pad_layer_html"><div class="ttname"><a href="classarmnn_1_1_pad_layer.html">armnn::PadLayer</a></div><div class="ttdoc">This layer represents a pad operation.</div><div class="ttdef"><b>Definition:</b> <a href="_pad_layer_8hpp_source.html#l00014">PadLayer.hpp:14</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]665<div class="ttc" id="astructarmnn_1_1_space_to_depth_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_space_to_depth_descriptor.html">armnn::SpaceToDepthDescriptor</a></div><div class="ttdoc">A SpaceToDepthDescriptor for the SpaceToDepthLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01075">Descriptors.hpp:1075</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]666<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a29c2c02a361c9d7028472e5d92cd4a54"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a29c2c02a361c9d7028472e5d92cd4a54">armnn::LayerType::Output</a></div><div class="ttdeci">@ Output</div></div>
667<div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4acb17869fe51048b5a5c4c6106551a255"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4acb17869fe51048b5a5c4c6106551a255">armnn::LayerType::Constant</a></div><div class="ttdeci">@ Constant</div></div>
668<div class="ttc" id="anamespacearmnn_html_ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f"><div class="ttname"><a href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">armnn::DataLayout::NCHW</a></div><div class="ttdeci">@ NCHW</div></div>
669<!-- start footer part -->
670<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
671 <ul>
672 <li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.html">src</a></li><li class="navelem"><a class="el" href="dir_0f3cdec46afbc61a1ded8e1687c9c9a0.html">backends</a></li><li class="navelem"><a class="el" href="dir_797a213d7d01b98ef12d53b0820ea64e.html">backendsCommon</a></li><li class="navelem"><a class="el" href="_subgraph_utils_8hpp.html">SubgraphUtils.hpp</a></li>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]673 <li class="footer">Generated on Thu May 16 2024 09:31:46 for Arm NN by
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]674 <a href="http://www.doxygen.org/index.html">
675 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li>
676 </ul>
677</div>
678</body>
679</html>