Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / refs/heads/gh-pages / . / latest / _arm_compute_subgraph_utils_8hpp_source.html
blob: 112ec56504330a86a97617a724c9a7bdcd7fbfd1 [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/aclCommon/ArmComputeSubgraphUtils.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('_arm_compute_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">ArmComputeSubgraphUtils.hpp</div> </div>
97</div><!--header-->
98<div class="contents">
99<a href="_arm_compute_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>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]100<div class="line"><a name="l00002"></a><span class="lineno"> 2</span>&#160;<span class="comment">// Copyright © 2020-2024 Arm Ltd and Contributors. All rights reserved.</span></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]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="_optimization_views_8hpp.html">armnn/backends/OptimizationViews.hpp</a>&gt;</span></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]107<div class="line"><a name="l00009"></a><span class="lineno"> 9</span>&#160; </div>
108<div class="line"><a name="l00010"></a><span class="lineno"> 10</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="_arm_compute_utils_8hpp.html">aclCommon/ArmComputeUtils.hpp</a>&gt;</span></div>
109<div class="line"><a name="l00011"></a><span class="lineno"> 11</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="_subgraph_utils_8hpp.html">backendsCommon/SubgraphUtils.hpp</a>&gt;</span></div>
110<div class="line"><a name="l00012"></a><span class="lineno"> 12</span>&#160; </div>
111<div class="line"><a name="l00013"></a><span class="lineno"> 13</span>&#160;<span class="keyword">namespace </span><a class="code" href="namespacearmnn.html">armnn</a></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; </div>
114<div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160;<span class="keyword">namespace</span></div>
115<div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160;{</div>
116<div class="line"><a name="l00018"></a><span class="lineno"> 18</span>&#160; </div>
117<div class="line"><a name="l00019"></a><span class="lineno"> 19</span>&#160;<span class="comment">//</span></div>
118<div class="line"><a name="l00020"></a><span class="lineno"> 20</span>&#160;<span class="comment">// this helper only works if all layers where the inputs connect to are not selected</span></div>
119<div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160;<span class="comment">//</span></div>
120<div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160; </div>
121<div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160;<span class="keywordtype">bool</span> checkDataTypeInputandOutput(<span class="keyword">const</span> Layer&amp; layer)</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="keyword">auto</span> inputInfo = layer.GetInputSlot(0).GetTensorInfo();</div>
124<div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160; <span class="keyword">auto</span> outputInfo = layer.GetOutputSlot(0).GetTensorInfo();</div>
125<div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; <span class="keywordtype">bool</span> sameDataType = (inputInfo.GetDataType() == outputInfo.GetDataType());</div>
126<div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; </div>
127<div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160; <span class="comment">// Check is same quantization info (same scale and offset)</span></div>
128<div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160; <span class="keywordflow">if</span> (sameDataType)</div>
129<div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160; {</div>
130<div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160; <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn.html#ad44c007f21af2d0375e3ef9400a1b275">IsQuantizedType</a>(inputInfo.GetDataType()))</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="keywordtype">bool</span> sameScale = (inputInfo.GetQuantizationScale() == outputInfo.GetQuantizationScale());</div>
133<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; <span class="keywordtype">bool</span> sameOffset = (inputInfo.GetQuantizationOffset() == outputInfo.GetQuantizationOffset());</div>
134<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; </div>
135<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160; <span class="keywordflow">return</span> (sameScale &amp;&amp; sameOffset);</div>
136<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160; }</div>
137<div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; <span class="keywordflow">else</span></div>
138<div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; {</div>
139<div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
140<div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; }</div>
141<div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; }</div>
142<div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; <span class="keywordflow">else</span></div>
143<div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; {</div>
144<div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; <span class="keywordflow">return</span> <span class="keyword">false</span>;</div>
145<div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; }</div>
146<div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160;}</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="comment">// namespace</span></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;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
151<div class="line"><a name="l00053"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96"> 53</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
152<div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
153<div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer,</div>
154<div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
155<div class="line"><a name="l00057"></a><span class="lineno"> 57</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc)</div>
156<div class="line"><a name="l00058"></a><span class="lineno"> 58</span>&#160;{</div>
157<div class="line"><a name="l00059"></a><span class="lineno"> 59</span>&#160; replacementLayer-&gt;SetAdditionalInfoForObject(</div>
158<div class="line"><a name="l00060"></a><span class="lineno"> 60</span>&#160; std::make_shared&lt;ActivationDescriptor&gt;(activationDesc));</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; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> substitutionSubgraph({baseLayer, activationLayer},</div>
161<div class="line"><a name="l00063"></a><span class="lineno"> 63</span>&#160; CreateIInputsFrom({baseLayer}),</div>
162<div class="line"><a name="l00064"></a><span class="lineno"> 64</span>&#160; CreateIOutputsFrom({activationLayer}));</div>
163<div class="line"><a name="l00065"></a><span class="lineno"> 65</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> replacementSubgraph(replacementLayer);</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; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#a57340147dcb1c0ae05fdf14ff7af8a01">AddSubstitution</a>({substitutionSubgraph, replacementSubgraph});</div>
166<div class="line"><a name="l00068"></a><span class="lineno"> 68</span>&#160; </div>
167<div class="line"><a name="l00069"></a><span class="lineno"> 69</span>&#160; <span class="keywordflow">return</span> replacementLayer;</div>
168<div class="line"><a name="l00070"></a><span class="lineno"> 70</span>&#160;}</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;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
171<div class="line"><a name="l00073"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a13f537c12c25c936c152ccca78d902dd"> 73</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#a13f537c12c25c936c152ccca78d902dd">FuseAdditionLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
172<div class="line"><a name="l00074"></a><span class="lineno"> 74</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
173<div class="line"><a name="l00075"></a><span class="lineno"> 75</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
174<div class="line"><a name="l00076"></a><span class="lineno"> 76</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
175<div class="line"><a name="l00077"></a><span class="lineno"> 77</span>&#160; std::string name)</div>
176<div class="line"><a name="l00078"></a><span class="lineno"> 78</span>&#160;{</div>
177<div class="line"><a name="l00079"></a><span class="lineno"> 79</span>&#160; <a class="code" href="_deprecated_8hpp.html#ab66a241a0ed3ee89c866e777b035d0ed">ARMNN_NO_DEPRECATE_WARN_BEGIN</a></div>
178<div class="line"><a name="l00080"></a><span class="lineno"> 80</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement = 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#a39f1b38d89c4de186742eafcbb3b1319">AddAdditionLayer</a>(name.c_str());</div>
179<div class="line"><a name="l00081"></a><span class="lineno"> 81</span>&#160; <a class="code" href="_deprecated_8hpp.html#ad762b11b48e5c1d1c1743f529485728a">ARMNN_NO_DEPRECATE_WARN_END</a></div>
180<div class="line"><a name="l00082"></a><span class="lineno"> 82</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</div>
181<div class="line"><a name="l00083"></a><span class="lineno"> 83</span>&#160; </div>
182<div class="line"><a name="l00084"></a><span class="lineno"> 84</span>&#160; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
183<div class="line"><a name="l00085"></a><span class="lineno"> 85</span>&#160; baseLayer,</div>
184<div class="line"><a name="l00086"></a><span class="lineno"> 86</span>&#160; replacementLayer,</div>
185<div class="line"><a name="l00087"></a><span class="lineno"> 87</span>&#160; activationLayer,</div>
186<div class="line"><a name="l00088"></a><span class="lineno"> 88</span>&#160; activationDesc);</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">return</span> replacementLayer;</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; </div>
191<div class="line"><a name="l00093"></a><span class="lineno"> 93</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
192<div class="line"><a name="l00094"></a><span class="lineno"><a class="line" href="namespacearmnn.html#aeda4132cdab76b98c860690c80fdb7bd"> 94</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#aeda4132cdab76b98c860690c80fdb7bd">FuseSubtractionLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
193<div class="line"><a name="l00095"></a><span class="lineno"> 95</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
194<div class="line"><a name="l00096"></a><span class="lineno"> 96</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
195<div class="line"><a name="l00097"></a><span class="lineno"> 97</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
196<div class="line"><a name="l00098"></a><span class="lineno"> 98</span>&#160; std::string name)</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; <a class="code" href="_deprecated_8hpp.html#ab66a241a0ed3ee89c866e777b035d0ed">ARMNN_NO_DEPRECATE_WARN_BEGIN</a></div>
199<div class="line"><a name="l00101"></a><span class="lineno"> 101</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement = 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#af13795cdf49e63d8bc3cb409592cdb9d">AddSubtractionLayer</a>(name.c_str());</div>
200<div class="line"><a name="l00102"></a><span class="lineno"> 102</span>&#160; <a class="code" href="_deprecated_8hpp.html#ad762b11b48e5c1d1c1743f529485728a">ARMNN_NO_DEPRECATE_WARN_END</a></div>
201<div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</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; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
204<div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160; baseLayer,</div>
205<div class="line"><a name="l00107"></a><span class="lineno"> 107</span>&#160; replacementLayer,</div>
206<div class="line"><a name="l00108"></a><span class="lineno"> 108</span>&#160; activationLayer,</div>
207<div class="line"><a name="l00109"></a><span class="lineno"> 109</span>&#160; activationDesc);</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="keywordflow">return</span> replacementLayer;</div>
210<div class="line"><a name="l00112"></a><span class="lineno"> 112</span>&#160;}</div>
211<div class="line"><a name="l00113"></a><span class="lineno"> 113</span>&#160; </div>
212<div class="line"><a name="l00114"></a><span class="lineno"> 114</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
213<div class="line"><a name="l00115"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a90f36a629081255b526ac433fcedf8ba"> 115</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#a90f36a629081255b526ac433fcedf8ba">FuseDivisionLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
214<div class="line"><a name="l00116"></a><span class="lineno"> 116</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
215<div class="line"><a name="l00117"></a><span class="lineno"> 117</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
216<div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
217<div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160; std::string name)</div>
218<div class="line"><a name="l00120"></a><span class="lineno"> 120</span>&#160;{</div>
219<div class="line"><a name="l00121"></a><span class="lineno"> 121</span>&#160; <a class="code" href="_deprecated_8hpp.html#ab66a241a0ed3ee89c866e777b035d0ed">ARMNN_NO_DEPRECATE_WARN_BEGIN</a></div>
220<div class="line"><a name="l00122"></a><span class="lineno"> 122</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement = 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#a0aeb4e528cf6ba4b7caca14a94fbcafe">AddDivisionLayer</a>(name.c_str());</div>
221<div class="line"><a name="l00123"></a><span class="lineno"> 123</span>&#160; <a class="code" href="_deprecated_8hpp.html#ad762b11b48e5c1d1c1743f529485728a">ARMNN_NO_DEPRECATE_WARN_END</a></div>
222<div class="line"><a name="l00124"></a><span class="lineno"> 124</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</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; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
225<div class="line"><a name="l00127"></a><span class="lineno"> 127</span>&#160; baseLayer,</div>
226<div class="line"><a name="l00128"></a><span class="lineno"> 128</span>&#160; replacementLayer,</div>
227<div class="line"><a name="l00129"></a><span class="lineno"> 129</span>&#160; activationLayer,</div>
228<div class="line"><a name="l00130"></a><span class="lineno"> 130</span>&#160; activationDesc);</div>
229<div class="line"><a name="l00131"></a><span class="lineno"> 131</span>&#160; </div>
230<div class="line"><a name="l00132"></a><span class="lineno"> 132</span>&#160; <span class="keywordflow">return</span> replacementLayer;</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; </div>
233<div class="line"><a name="l00135"></a><span class="lineno"> 135</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
234<div class="line"><a name="l00136"></a><span class="lineno"><a class="line" href="namespacearmnn.html#af06b26928a460f9fcdef0327f448a34c"> 136</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#af06b26928a460f9fcdef0327f448a34c">FuseMultiplicationLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
235<div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
236<div class="line"><a name="l00138"></a><span class="lineno"> 138</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
237<div class="line"><a name="l00139"></a><span class="lineno"> 139</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
238<div class="line"><a name="l00140"></a><span class="lineno"> 140</span>&#160; std::string name)</div>
239<div class="line"><a name="l00141"></a><span class="lineno"> 141</span>&#160;{</div>
240<div class="line"><a name="l00142"></a><span class="lineno"> 142</span>&#160; <a class="code" href="_deprecated_8hpp.html#ab66a241a0ed3ee89c866e777b035d0ed">ARMNN_NO_DEPRECATE_WARN_BEGIN</a></div>
241<div class="line"><a name="l00143"></a><span class="lineno"> 143</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement = 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#abb59f6ba9988dae88e0f48e68d87fc32">AddMultiplicationLayer</a>(name.c_str());</div>
242<div class="line"><a name="l00144"></a><span class="lineno"> 144</span>&#160; <a class="code" href="_deprecated_8hpp.html#ad762b11b48e5c1d1c1743f529485728a">ARMNN_NO_DEPRECATE_WARN_END</a></div>
243<div class="line"><a name="l00145"></a><span class="lineno"> 145</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</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; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
246<div class="line"><a name="l00148"></a><span class="lineno"> 148</span>&#160; baseLayer,</div>
247<div class="line"><a name="l00149"></a><span class="lineno"> 149</span>&#160; replacementLayer,</div>
248<div class="line"><a name="l00150"></a><span class="lineno"> 150</span>&#160; activationLayer,</div>
249<div class="line"><a name="l00151"></a><span class="lineno"> 151</span>&#160; activationDesc);</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; <span class="keywordflow">return</span> replacementLayer;</div>
252<div class="line"><a name="l00154"></a><span class="lineno"> 154</span>&#160;}</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="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
255<div class="line"><a name="l00157"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a6a28e34d999aac5a0149d35b02a3cfe5"> 157</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#a6a28e34d999aac5a0149d35b02a3cfe5">FuseElementwiseBinaryLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
256<div class="line"><a name="l00158"></a><span class="lineno"> 158</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
257<div class="line"><a name="l00159"></a><span class="lineno"> 159</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
258<div class="line"><a name="l00160"></a><span class="lineno"> 160</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
259<div class="line"><a name="l00161"></a><span class="lineno"> 161</span>&#160; <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047">BinaryOperation</a> operation,</div>
260<div class="line"><a name="l00162"></a><span class="lineno"> 162</span>&#160; std::string name)</div>
261<div class="line"><a name="l00163"></a><span class="lineno"> 163</span>&#160;{</div>
262<div class="line"><a name="l00164"></a><span class="lineno"> 164</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement = 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#aab976306ba3424cbc0c7ea33417d718d">AddElementwiseBinaryLayer</a>(operation,</div>
263<div class="line"><a name="l00165"></a><span class="lineno"> 165</span>&#160; name.c_str());</div>
264<div class="line"><a name="l00166"></a><span class="lineno"> 166</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</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; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
267<div class="line"><a name="l00169"></a><span class="lineno"> 169</span>&#160; baseLayer,</div>
268<div class="line"><a name="l00170"></a><span class="lineno"> 170</span>&#160; replacementLayer,</div>
269<div class="line"><a name="l00171"></a><span class="lineno"> 171</span>&#160; activationLayer,</div>
270<div class="line"><a name="l00172"></a><span class="lineno"> 172</span>&#160; activationDesc);</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; <span class="keywordflow">return</span> replacementLayer;</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; </div>
275<div class="line"><a name="l00177"></a><span class="lineno"> 177</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
276<div class="line"><a name="l00178"></a><span class="lineno"><a class="line" href="namespacearmnn.html#aa4a4b141867313cddb82c96800db343e"> 178</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#aa4a4b141867313cddb82c96800db343e">FuseBatchNormalizationLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
277<div class="line"><a name="l00179"></a><span class="lineno"> 179</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
278<div class="line"><a name="l00180"></a><span class="lineno"> 180</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
279<div class="line"><a name="l00181"></a><span class="lineno"> 181</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
280<div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160; std::string name)</div>
281<div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160;{</div>
282<div class="line"><a name="l00184"></a><span class="lineno"> 184</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement =</div>
283<div class="line"><a name="l00185"></a><span class="lineno"> 185</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#a8f798e19187ac7ae6ae6153ee64ab645">AddBatchNormalizationLayer</a>(baseLayer-&gt;GetParameters(),</div>
284<div class="line"><a name="l00186"></a><span class="lineno"> 186</span>&#160; <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a>(),</div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]285<div class="line"><a name="l00187"></a><span class="lineno"> 187</span>&#160; <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a>(),</div>
286<div class="line"><a name="l00188"></a><span class="lineno"> 188</span>&#160; <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a>(),</div>
287<div class="line"><a name="l00189"></a><span class="lineno"> 189</span>&#160; <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a>(),</div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]288<div class="line"><a name="l00190"></a><span class="lineno"> 190</span>&#160; name.c_str());</div>
289<div class="line"><a name="l00191"></a><span class="lineno"> 191</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</div>
290<div class="line"><a name="l00192"></a><span class="lineno"> 192</span>&#160; </div>
291<div class="line"><a name="l00193"></a><span class="lineno"> 193</span>&#160; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
292<div class="line"><a name="l00194"></a><span class="lineno"> 194</span>&#160; baseLayer,</div>
293<div class="line"><a name="l00195"></a><span class="lineno"> 195</span>&#160; replacementLayer,</div>
294<div class="line"><a name="l00196"></a><span class="lineno"> 196</span>&#160; activationLayer,</div>
295<div class="line"><a name="l00197"></a><span class="lineno"> 197</span>&#160; activationDesc);</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; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> substitutionSubgraph({baseLayer, activationLayer},</div>
298<div class="line"><a name="l00200"></a><span class="lineno"> 200</span>&#160; CreateIInputsFrom({baseLayer}),</div>
299<div class="line"><a name="l00201"></a><span class="lineno"> 201</span>&#160; CreateIOutputsFrom({activationLayer}));</div>
300<div class="line"><a name="l00202"></a><span class="lineno"> 202</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> replacementSubgraph(replacementLayer);</div>
301<div class="line"><a name="l00203"></a><span class="lineno"> 203</span>&#160; </div>
302<div class="line"><a name="l00204"></a><span class="lineno"> 204</span>&#160; <span class="keywordflow">return</span> replacementLayer;</div>
303<div class="line"><a name="l00205"></a><span class="lineno"> 205</span>&#160;}</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;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
306<div class="line"><a name="l00208"></a><span class="lineno"><a class="line" href="namespacearmnn.html#afdb0a1b1703203354451e78e2fc43f3f"> 208</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#afdb0a1b1703203354451e78e2fc43f3f">FuseConvolution2dLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
307<div class="line"><a name="l00209"></a><span class="lineno"> 209</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
308<div class="line"><a name="l00210"></a><span class="lineno"> 210</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
309<div class="line"><a name="l00211"></a><span class="lineno"> 211</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
310<div class="line"><a name="l00212"></a><span class="lineno"> 212</span>&#160; std::string name)</div>
311<div class="line"><a name="l00213"></a><span class="lineno"> 213</span>&#160;{</div>
312<div class="line"><a name="l00214"></a><span class="lineno"> 214</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement = optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#ad04187fe81f68558b15b6049b2da9cf9">GetINetwork</a>()</div>
313<div class="line"><a name="l00215"></a><span class="lineno"> 215</span>&#160; -&gt;<a class="code" href="classarmnn_1_1_i_network.html#a383e74ef080d4a81b8b371be4b840248">AddConvolution2dLayer</a>(baseLayer-&gt;GetParameters(), name.c_str());</div>
314<div class="line"><a name="l00216"></a><span class="lineno"> 216</span>&#160; </div>
315<div class="line"><a name="l00217"></a><span class="lineno"> 217</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</div>
316<div class="line"><a name="l00218"></a><span class="lineno"> 218</span>&#160; </div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]317<div class="line"><a name="l00219"></a><span class="lineno"> 219</span>&#160; </div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]318<div class="line"><a name="l00220"></a><span class="lineno"> 220</span>&#160; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
319<div class="line"><a name="l00221"></a><span class="lineno"> 221</span>&#160; baseLayer,</div>
320<div class="line"><a name="l00222"></a><span class="lineno"> 222</span>&#160; replacementLayer,</div>
321<div class="line"><a name="l00223"></a><span class="lineno"> 223</span>&#160; activationLayer,</div>
322<div class="line"><a name="l00224"></a><span class="lineno"> 224</span>&#160; activationDesc);</div>
323<div class="line"><a name="l00225"></a><span class="lineno"> 225</span>&#160; </div>
324<div class="line"><a name="l00226"></a><span class="lineno"> 226</span>&#160; <span class="keywordflow">return</span> replacementLayer;</div>
325<div class="line"><a name="l00227"></a><span class="lineno"> 227</span>&#160;}</div>
326<div class="line"><a name="l00228"></a><span class="lineno"> 228</span>&#160; </div>
327<div class="line"><a name="l00229"></a><span class="lineno"> 229</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
328<div class="line"><a name="l00230"></a><span class="lineno"><a class="line" href="namespacearmnn.html#ae795a54643adfeb37ac95dee68807856"> 230</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#ae795a54643adfeb37ac95dee68807856">FuseDepthwiseConvolution2dLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
329<div class="line"><a name="l00231"></a><span class="lineno"> 231</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
330<div class="line"><a name="l00232"></a><span class="lineno"> 232</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
331<div class="line"><a name="l00233"></a><span class="lineno"> 233</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
332<div class="line"><a name="l00234"></a><span class="lineno"> 234</span>&#160; std::string name)</div>
333<div class="line"><a name="l00235"></a><span class="lineno"> 235</span>&#160;{</div>
334<div class="line"><a name="l00236"></a><span class="lineno"> 236</span>&#160; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement =</div>
335<div class="line"><a name="l00237"></a><span class="lineno"> 237</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#ac6771c8752811fcfa91b3f0dee748144">AddDepthwiseConvolution2dLayer</a>(baseLayer-&gt;GetParameters(), name.c_str());</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="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</div>
338<div class="line"><a name="l00240"></a><span class="lineno"> 240</span>&#160; </div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]339<div class="line"><a name="l00241"></a><span class="lineno"> 241</span>&#160; </div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]340<div class="line"><a name="l00242"></a><span class="lineno"> 242</span>&#160; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
341<div class="line"><a name="l00243"></a><span class="lineno"> 243</span>&#160; baseLayer,</div>
342<div class="line"><a name="l00244"></a><span class="lineno"> 244</span>&#160; replacementLayer,</div>
343<div class="line"><a name="l00245"></a><span class="lineno"> 245</span>&#160; activationLayer,</div>
344<div class="line"><a name="l00246"></a><span class="lineno"> 246</span>&#160; activationDesc);</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="keywordflow">return</span> replacementLayer;</div>
347<div class="line"><a name="l00249"></a><span class="lineno"> 249</span>&#160;}</div>
348<div class="line"><a name="l00250"></a><span class="lineno"> 250</span>&#160; </div>
349<div class="line"><a name="l00251"></a><span class="lineno"> 251</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
350<div class="line"><a name="l00252"></a><span class="lineno"><a class="line" href="namespacearmnn.html#ad2021f80998e436e27de7b7eeec67d54"> 252</a></span>&#160;<a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* <a class="code" href="namespacearmnn.html#ad2021f80998e436e27de7b7eeec67d54">FuseFullyConnectedLayer</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
351<div class="line"><a name="l00253"></a><span class="lineno"> 253</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
352<div class="line"><a name="l00254"></a><span class="lineno"> 254</span>&#160; <a class="code" href="classarmnn_1_1_activation_layer.html">ActivationLayer</a>* activationLayer,</div>
353<div class="line"><a name="l00255"></a><span class="lineno"> 255</span>&#160; <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a>&amp; activationDesc,</div>
354<div class="line"><a name="l00256"></a><span class="lineno"> 256</span>&#160; std::string name)</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; <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* replacement =</div>
357<div class="line"><a name="l00259"></a><span class="lineno"> 259</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#a471991a84030eb3ae601da2bee757870">AddFullyConnectedLayer</a>(baseLayer-&gt;GetParameters(),</div>
358<div class="line"><a name="l00260"></a><span class="lineno"> 260</span>&#160; name.c_str());</div>
359<div class="line"><a name="l00261"></a><span class="lineno"> 261</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* replacementLayer = PolymorphicDowncast&lt;LayerType*&gt;(replacement);</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; <a class="code" href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">FuseLayer</a>(optimizationViews,</div>
362<div class="line"><a name="l00264"></a><span class="lineno"> 264</span>&#160; baseLayer,</div>
363<div class="line"><a name="l00265"></a><span class="lineno"> 265</span>&#160; replacementLayer,</div>
364<div class="line"><a name="l00266"></a><span class="lineno"> 266</span>&#160; activationLayer,</div>
365<div class="line"><a name="l00267"></a><span class="lineno"> 267</span>&#160; activationDesc);</div>
366<div class="line"><a name="l00268"></a><span class="lineno"> 268</span>&#160; </div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]367<div class="line"><a name="l00269"></a><span class="lineno"> 269</span>&#160; </div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]368<div class="line"><a name="l00270"></a><span class="lineno"> 270</span>&#160; <span class="keywordflow">return</span> replacementLayer;</div>
369<div class="line"><a name="l00271"></a><span class="lineno"> 271</span>&#160;}</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;<span class="comment">//</span></div>
372<div class="line"><a name="l00274"></a><span class="lineno"> 274</span>&#160;<span class="comment">// If reduce layer has multiple axes, add new layer for each axis to simulate the same behaviour</span></div>
373<div class="line"><a name="l00275"></a><span class="lineno"> 275</span>&#160;<span class="comment">// as currently only one axis is supported.</span></div>
374<div class="line"><a name="l00276"></a><span class="lineno"> 276</span>&#160;<span class="comment">//</span></div>
375<div class="line"><a name="l00277"></a><span class="lineno"> 277</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
376<div class="line"><a name="l00278"></a><span class="lineno"><a class="line" href="namespacearmnn.html#a189b5dedbf7a3cbcbe7760ac72c7b07d"> 278</a></span>&#160;std::vector&lt;IConnectableLayer*&gt; <a class="code" href="namespacearmnn.html#a189b5dedbf7a3cbcbe7760ac72c7b07d">ChainReduceLayers</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
377<div class="line"><a name="l00279"></a><span class="lineno"> 279</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
378<div class="line"><a name="l00280"></a><span class="lineno"> 280</span>&#160; <a class="code" href="structarmnn_1_1_reduce_descriptor.html">ReduceDescriptor</a>&amp; desc)</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; <span class="comment">// Vector of new chained layers, used for substitution.</span></div>
381<div class="line"><a name="l00283"></a><span class="lineno"> 283</span>&#160; std::vector&lt;IConnectableLayer*&gt; layers;</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="comment">// Vector of axes so each layer is reshaped correctly.</span></div>
384<div class="line"><a name="l00286"></a><span class="lineno"> 286</span>&#160; std::vector&lt;uint32_t&gt; axes;</div>
385<div class="line"><a name="l00287"></a><span class="lineno"> 287</span>&#160; <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> recalulatedAxis = 0;</div>
386<div class="line"><a name="l00288"></a><span class="lineno"> 288</span>&#160; </div>
387<div class="line"><a name="l00289"></a><span class="lineno"> 289</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i != desc.<a class="code" href="structarmnn_1_1_reduce_descriptor.html#aa1c6fc8c96404252f1072632fc5acb59">m_vAxis</a>.size(); ++i)</div>
388<div class="line"><a name="l00290"></a><span class="lineno"> 290</span>&#160; {</div>
389<div class="line"><a name="l00291"></a><span class="lineno"> 291</span>&#160; <span class="comment">// Get TensorInfo from base layer and reduce shape using axis.</span></div>
390<div class="line"><a name="l00292"></a><span class="lineno"> 292</span>&#160; <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> layerInfo = baseLayer-&gt;GetInputSlot(0).GetConnectedOutputSlot()-&gt;GetTensorInfo();</div>
391<div class="line"><a name="l00293"></a><span class="lineno"> 293</span>&#160; </div>
392<div class="line"><a name="l00294"></a><span class="lineno"> 294</span>&#160; axes.emplace_back(desc.<a class="code" href="structarmnn_1_1_reduce_descriptor.html#aa1c6fc8c96404252f1072632fc5acb59">m_vAxis</a>[i]);</div>
393<div class="line"><a name="l00295"></a><span class="lineno"> 295</span>&#160; </div>
394<div class="line"><a name="l00296"></a><span class="lineno"> 296</span>&#160; <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>&amp; reducedTensorInfo = <a class="code" href="namespacearmnn.html#a9d43aa01b882d8776ab9403eea3cf779">ComputeReductionTensorShape</a>(layerInfo,</div>
395<div class="line"><a name="l00297"></a><span class="lineno"> 297</span>&#160; axes,</div>
396<div class="line"><a name="l00298"></a><span class="lineno"> 298</span>&#160; desc.<a class="code" href="structarmnn_1_1_reduce_descriptor.html#a28e0548abfc4e79c48f29a3d11a062e9">m_KeepDims</a>);</div>
397<div class="line"><a name="l00299"></a><span class="lineno"> 299</span>&#160; </div>
398<div class="line"><a name="l00300"></a><span class="lineno"> 300</span>&#160; <span class="comment">// Create a vector for the single axis to be assigned to the descriptor.</span></div>
399<div class="line"><a name="l00301"></a><span class="lineno"> 301</span>&#160; <span class="comment">// Update axis if keepDims is set reduce layers correctly.</span></div>
400<div class="line"><a name="l00302"></a><span class="lineno"> 302</span>&#160; std::vector&lt;uint32_t&gt; singleAxis(1, desc.<a class="code" href="structarmnn_1_1_reduce_descriptor.html#aa1c6fc8c96404252f1072632fc5acb59">m_vAxis</a>[i] - recalulatedAxis);</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">// Create a descriptor and assign single axis.</span></div>
403<div class="line"><a name="l00305"></a><span class="lineno"> 305</span>&#160; <a class="code" href="structarmnn_1_1_reduce_descriptor.html">ReduceDescriptor</a> newReduceDescriptor = baseLayer-&gt;GetParameters();</div>
404<div class="line"><a name="l00306"></a><span class="lineno"> 306</span>&#160; newReduceDescriptor.<a class="code" href="structarmnn_1_1_reduce_descriptor.html#aa1c6fc8c96404252f1072632fc5acb59">m_vAxis</a>.assign(singleAxis.begin(), singleAxis.end());</div>
405<div class="line"><a name="l00307"></a><span class="lineno"> 307</span>&#160; </div>
406<div class="line"><a name="l00308"></a><span class="lineno"> 308</span>&#160; <span class="comment">// Add new layer to graph.</span></div>
407<div class="line"><a name="l00309"></a><span class="lineno"> 309</span>&#160; std::string layerName = <span class="stringliteral">&quot;reduce_layer_&quot;</span> + std::to_string(i);</div>
408<div class="line"><a name="l00310"></a><span class="lineno"> 310</span>&#160; </div>
409<div class="line"><a name="l00311"></a><span class="lineno"> 311</span>&#160; <a class="code" href="classarmnn_1_1_layer.html">Layer</a>* replacementLayer = PolymorphicDowncast&lt;Layer*&gt;(</div>
410<div class="line"><a name="l00312"></a><span class="lineno"> 312</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#ae0cfae1ea51669892608a1a060d24fa0">AddReduceLayer</a>(newReduceDescriptor,</div>
411<div class="line"><a name="l00313"></a><span class="lineno"> 313</span>&#160; layerName.c_str()));</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="comment">// Connect previous layer with new layer.</span></div>
414<div class="line"><a name="l00316"></a><span class="lineno"> 316</span>&#160; <span class="comment">// The first and last layer will be connected when the subgraph is replaced.</span></div>
415<div class="line"><a name="l00317"></a><span class="lineno"> 317</span>&#160; <span class="keywordflow">if</span> (!layers.empty())</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; layers[i - 1]-&gt;GetOutputSlot(0).Connect(replacementLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#acf8b8e23bf647836592982f97088d375">GetInputSlot</a>(0));</div>
418<div class="line"><a name="l00320"></a><span class="lineno"> 320</span>&#160; }</div>
419<div class="line"><a name="l00321"></a><span class="lineno"> 321</span>&#160; </div>
420<div class="line"><a name="l00322"></a><span class="lineno"> 322</span>&#160; <span class="comment">// Set updated tensorInfo for new layer.</span></div>
421<div class="line"><a name="l00323"></a><span class="lineno"> 323</span>&#160; replacementLayer-&gt;<a class="code" href="classarmnn_1_1_layer.html#a0e36688a43c35668d8db5257274c68fe">GetOutputSlot</a>(0).<a class="code" href="classarmnn_1_1_output_slot.html#a7e5c5771d741dd5473989047a9314728">SetTensorInfo</a>(reducedTensorInfo);</div>
422<div class="line"><a name="l00324"></a><span class="lineno"> 324</span>&#160; </div>
423<div class="line"><a name="l00325"></a><span class="lineno"> 325</span>&#160; <span class="keywordflow">if</span> (!desc.<a class="code" href="structarmnn_1_1_reduce_descriptor.html#a28e0548abfc4e79c48f29a3d11a062e9">m_KeepDims</a>)</div>
424<div class="line"><a name="l00326"></a><span class="lineno"> 326</span>&#160; {</div>
425<div class="line"><a name="l00327"></a><span class="lineno"> 327</span>&#160; recalulatedAxis++;</div>
426<div class="line"><a name="l00328"></a><span class="lineno"> 328</span>&#160; }</div>
427<div class="line"><a name="l00329"></a><span class="lineno"> 329</span>&#160; </div>
428<div class="line"><a name="l00330"></a><span class="lineno"> 330</span>&#160; layers.emplace_back(replacementLayer);</div>
429<div class="line"><a name="l00331"></a><span class="lineno"> 331</span>&#160; }</div>
430<div class="line"><a name="l00332"></a><span class="lineno"> 332</span>&#160; <span class="keywordflow">return</span> layers;</div>
431<div class="line"><a name="l00333"></a><span class="lineno"> 333</span>&#160;}</div>
432<div class="line"><a name="l00334"></a><span class="lineno"> 334</span>&#160; </div>
433<div class="line"><a name="l00335"></a><span class="lineno"> 335</span>&#160;<span class="comment">//</span></div>
434<div class="line"><a name="l00336"></a><span class="lineno"> 336</span>&#160;<span class="comment">// Substitute baseLayer with new subgraph</span></div>
435<div class="line"><a name="l00337"></a><span class="lineno"> 337</span>&#160;<span class="comment">//</span></div>
436<div class="line"><a name="l00338"></a><span class="lineno"> 338</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
437<div class="line"><a name="l00339"></a><span class="lineno"><a class="line" href="namespacearmnn.html#abecdb9c231b9ced02bec27a1ca54184f"> 339</a></span>&#160;<span class="keywordtype">void</span> <a class="code" href="namespacearmnn.html#abecdb9c231b9ced02bec27a1ca54184f">ReplaceLayers</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
438<div class="line"><a name="l00340"></a><span class="lineno"> 340</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
439<div class="line"><a name="l00341"></a><span class="lineno"> 341</span>&#160; std::vector&lt;IConnectableLayer*&gt;&amp; layers)</div>
440<div class="line"><a name="l00342"></a><span class="lineno"> 342</span>&#160;{</div>
441<div class="line"><a name="l00343"></a><span class="lineno"> 343</span>&#160; std::list&lt;IConnectableLayer*&gt; replacementLayers(layers.begin(), layers.end());</div>
442<div class="line"><a name="l00344"></a><span class="lineno"> 344</span>&#160; </div>
443<div class="line"><a name="l00345"></a><span class="lineno"> 345</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> substitutionSubgraph(baseLayer);</div>
444<div class="line"><a name="l00346"></a><span class="lineno"> 346</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> replacementSubgraph(std::move(replacementLayers),</div>
445<div class="line"><a name="l00347"></a><span class="lineno"> 347</span>&#160; CreateIInputsFrom({replacementLayers.front()}),</div>
446<div class="line"><a name="l00348"></a><span class="lineno"> 348</span>&#160; CreateIOutputsFrom({replacementLayers.back()}));</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; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#a57340147dcb1c0ae05fdf14ff7af8a01">AddSubstitution</a>({substitutionSubgraph, replacementSubgraph});</div>
449<div class="line"><a name="l00351"></a><span class="lineno"> 351</span>&#160;}</div>
450<div class="line"><a name="l00352"></a><span class="lineno"> 352</span>&#160; </div>
451<div class="line"><a name="l00353"></a><span class="lineno"> 353</span>&#160;<span class="comment">//</span></div>
452<div class="line"><a name="l00354"></a><span class="lineno"> 354</span>&#160;<span class="comment">// Substitute a multi-layer subgraph with one new layer</span></div>
453<div class="line"><a name="l00355"></a><span class="lineno"> 355</span>&#160;<span class="comment">//</span></div>
454<div class="line"><a name="l00356"></a><span class="lineno"> 356</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> LayerType&gt;</div>
455<div class="line"><a name="l00357"></a><span class="lineno"><a class="line" href="namespacearmnn.html#af8b1723c2ceaf3aca2deddbf9b0e3e44"> 357</a></span>&#160;<span class="keywordtype">void</span> <a class="code" href="namespacearmnn.html#af8b1723c2ceaf3aca2deddbf9b0e3e44">ReplaceMultipleLayers</a>(<a class="code" href="classarmnn_1_1_optimization_views.html">OptimizationViews</a>&amp; optimizationViews,</div>
456<div class="line"><a name="l00358"></a><span class="lineno"> 358</span>&#160; std::vector&lt;IConnectableLayer*&gt;&amp; originalLayers,</div>
457<div class="line"><a name="l00359"></a><span class="lineno"> 359</span>&#160; <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4">LayerType</a>* baseLayer,</div>
458<div class="line"><a name="l00360"></a><span class="lineno"> 360</span>&#160; <span class="keyword">const</span> std::vector&lt;SlotList&gt; inputLayersSlotLists,</div>
459<div class="line"><a name="l00361"></a><span class="lineno"> 361</span>&#160; <span class="keyword">const</span> std::vector&lt;SlotList&gt; outputLayersSlotLists)</div>
460<div class="line"><a name="l00362"></a><span class="lineno"> 362</span>&#160;{</div>
461<div class="line"><a name="l00363"></a><span class="lineno"> 363</span>&#160; std::list&lt;IConnectableLayer*&gt; originalLayerList(originalLayers.begin(), originalLayers.end());</div>
462<div class="line"><a name="l00364"></a><span class="lineno"> 364</span>&#160; </div>
463<div class="line"><a name="l00365"></a><span class="lineno"> 365</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> substitutionSubgraph(</div>
464<div class="line"><a name="l00366"></a><span class="lineno"> 366</span>&#160; std::move(originalLayerList),</div>
465<div class="line"><a name="l00367"></a><span class="lineno"> 367</span>&#160; CreateIInputsFromSlotLists&lt;armnn::IConnectableLayer&gt;(originalLayers, inputLayersSlotLists),</div>
466<div class="line"><a name="l00368"></a><span class="lineno"> 368</span>&#160; CreateIOutputsFromSlotLists&lt;armnn::IConnectableLayer&gt;(originalLayers, outputLayersSlotLists));</div>
467<div class="line"><a name="l00369"></a><span class="lineno"> 369</span>&#160; <a class="code" href="classarmnn_1_1_subgraph_view.html">SubgraphView</a> replacementSubgraph(baseLayer);</div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]468<div class="line"><a name="l00370"></a><span class="lineno"> 370</span>&#160; </div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]469<div class="line"><a name="l00371"></a><span class="lineno"> 371</span>&#160; optimizationViews.<a class="code" href="classarmnn_1_1_optimization_views.html#a57340147dcb1c0ae05fdf14ff7af8a01">AddSubstitution</a>({substitutionSubgraph, replacementSubgraph});</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; </div>
472<div class="line"><a name="l00374"></a><span class="lineno"> 374</span>&#160;} <span class="comment">// namespace armnn</span></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]473</div><!-- fragment --></div><!-- contents -->
474</div><!-- doc-content -->
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]475<div class="ttc" id="anamespacearmnn_html_a189b5dedbf7a3cbcbe7760ac72c7b07d"><div class="ttname"><a href="namespacearmnn.html#a189b5dedbf7a3cbcbe7760ac72c7b07d">armnn::ChainReduceLayers</a></div><div class="ttdeci">std::vector&lt; IConnectableLayer * &gt; ChainReduceLayers(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ReduceDescriptor &amp;desc)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00278">ArmComputeSubgraphUtils.hpp:278</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]476<div class="ttc" id="astructarmnn_1_1_activation_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_activation_descriptor.html">armnn::ActivationDescriptor</a></div><div class="ttdoc">An ActivationDescriptor for the ActivationLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00036">Descriptors.hpp:36</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]477<div class="ttc" id="aclassarmnn_1_1_i_network_html_ae0cfae1ea51669892608a1a060d24fa0"><div class="ttname"><a href="classarmnn_1_1_i_network.html#ae0cfae1ea51669892608a1a060d24fa0">armnn::INetwork::AddReduceLayer</a></div><div class="ttdeci">IConnectableLayer * AddReduceLayer(const ReduceDescriptor &amp;reduceDescriptor, const char *name=nullptr)</div><div class="ttdoc">Adds a reduce layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00444">Network.cpp:444</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]478<div class="ttc" id="anamespacearmnn_html_ae795a54643adfeb37ac95dee68807856"><div class="ttname"><a href="namespacearmnn.html#ae795a54643adfeb37ac95dee68807856">armnn::FuseDepthwiseConvolution2dLayer</a></div><div class="ttdeci">LayerType * FuseDepthwiseConvolution2dLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00230">ArmComputeSubgraphUtils.hpp:230</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]479<div class="ttc" id="aclassarmnn_1_1_i_network_html_a39f1b38d89c4de186742eafcbb3b1319"><div class="ttname"><a href="classarmnn_1_1_i_network.html#a39f1b38d89c4de186742eafcbb3b1319">armnn::INetwork::AddAdditionLayer</a></div><div class="ttdeci">IConnectableLayer * AddAdditionLayer(const char *name=nullptr)</div><div class="ttdoc">Adds an addition layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00409">Network.cpp:409</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]480<div class="ttc" id="aclassarmnn_1_1_output_slot_html_a7e5c5771d741dd5473989047a9314728"><div class="ttname"><a href="classarmnn_1_1_output_slot.html#a7e5c5771d741dd5473989047a9314728">armnn::OutputSlot::SetTensorInfo</a></div><div class="ttdeci">void SetTensorInfo(const TensorInfo &amp;tensorInfo) override</div><div class="ttdef"><b>Definition:</b> <a href="_layer_8cpp_source.html#l00095">Layer.cpp:95</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]481<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>
482<div class="ttc" id="aclassarmnn_1_1_i_network_html_a471991a84030eb3ae601da2bee757870"><div class="ttname"><a href="classarmnn_1_1_i_network.html#a471991a84030eb3ae601da2bee757870">armnn::INetwork::AddFullyConnectedLayer</a></div><div class="ttdeci">IConnectableLayer * AddFullyConnectedLayer(const FullyConnectedDescriptor &amp;fullyConnectedDescriptor, const char *name=nullptr)</div><div class="ttdoc">Adds a fully connected layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00332">Network.cpp:332</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]483<div class="ttc" id="anamespacearmnn_html_afdb0a1b1703203354451e78e2fc43f3f"><div class="ttname"><a href="namespacearmnn.html#afdb0a1b1703203354451e78e2fc43f3f">armnn::FuseConvolution2dLayer</a></div><div class="ttdeci">LayerType * FuseConvolution2dLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00208">ArmComputeSubgraphUtils.hpp:208</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]484<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>
485<div class="ttc" id="a_deprecated_8hpp_html_ab66a241a0ed3ee89c866e777b035d0ed"><div class="ttname"><a href="_deprecated_8hpp.html#ab66a241a0ed3ee89c866e777b035d0ed">ARMNN_NO_DEPRECATE_WARN_BEGIN</a></div><div class="ttdeci">#define ARMNN_NO_DEPRECATE_WARN_BEGIN</div><div class="ttdef"><b>Definition:</b> <a href="_deprecated_8hpp_source.html#l00033">Deprecated.hpp:33</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]486<div class="ttc" id="anamespacearmnn_html_abecdb9c231b9ced02bec27a1ca54184f"><div class="ttname"><a href="namespacearmnn.html#abecdb9c231b9ced02bec27a1ca54184f">armnn::ReplaceLayers</a></div><div class="ttdeci">void ReplaceLayers(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, std::vector&lt; IConnectableLayer * &gt; &amp;layers)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00339">ArmComputeSubgraphUtils.hpp:339</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]487<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>
488<div class="ttc" id="a_optimization_views_8hpp_html"><div class="ttname"><a href="_optimization_views_8hpp.html">OptimizationViews.hpp</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]489<div class="ttc" id="anamespacearmnn_html_a13f537c12c25c936c152ccca78d902dd"><div class="ttname"><a href="namespacearmnn.html#a13f537c12c25c936c152ccca78d902dd">armnn::FuseAdditionLayer</a></div><div class="ttdeci">LayerType * FuseAdditionLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00073">ArmComputeSubgraphUtils.hpp:73</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]490<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]491<div class="ttc" id="anamespacearmnn_html_af06b26928a460f9fcdef0327f448a34c"><div class="ttname"><a href="namespacearmnn.html#af06b26928a460f9fcdef0327f448a34c">armnn::FuseMultiplicationLayer</a></div><div class="ttdeci">LayerType * FuseMultiplicationLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00136">ArmComputeSubgraphUtils.hpp:136</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]492<div class="ttc" id="aclassarmnn_1_1_i_network_html_a8f798e19187ac7ae6ae6153ee64ab645"><div class="ttname"><a href="classarmnn_1_1_i_network.html#a8f798e19187ac7ae6ae6153ee64ab645">armnn::INetwork::AddBatchNormalizationLayer</a></div><div class="ttdeci">IConnectableLayer * AddBatchNormalizationLayer(const BatchNormalizationDescriptor &amp;desc, const ConstTensor &amp;mean, const ConstTensor &amp;variance, const ConstTensor &amp;beta, const ConstTensor &amp;gamma, const char *name=nullptr)</div><div class="ttdoc">Adds a batch normalization layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00423">Network.cpp:423</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]493<div class="ttc" id="anamespacearmnn_html_aa4a4b141867313cddb82c96800db343e"><div class="ttname"><a href="namespacearmnn.html#aa4a4b141867313cddb82c96800db343e">armnn::FuseBatchNormalizationLayer</a></div><div class="ttdeci">LayerType * FuseBatchNormalizationLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00178">ArmComputeSubgraphUtils.hpp:178</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]494<div class="ttc" id="aclassarmnn_1_1_i_network_html_a383e74ef080d4a81b8b371be4b840248"><div class="ttname"><a href="classarmnn_1_1_i_network.html#a383e74ef080d4a81b8b371be4b840248">armnn::INetwork::AddConvolution2dLayer</a></div><div class="ttdeci">IConnectableLayer * AddConvolution2dLayer(const Convolution2dDescriptor &amp;convolution2dDescriptor, const char *name=nullptr)</div><div class="ttdoc">Adds a 2D convolution layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00272">Network.cpp:272</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]495<div class="ttc" id="anamespacearmnn_html_a90f36a629081255b526ac433fcedf8ba"><div class="ttname"><a href="namespacearmnn.html#a90f36a629081255b526ac433fcedf8ba">armnn::FuseDivisionLayer</a></div><div class="ttdeci">LayerType * FuseDivisionLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00115">ArmComputeSubgraphUtils.hpp:115</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]496<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>
497<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]498<div class="ttc" id="aclassarmnn_1_1_i_network_html_a0aeb4e528cf6ba4b7caca14a94fbcafe"><div class="ttname"><a href="classarmnn_1_1_i_network.html#a0aeb4e528cf6ba4b7caca14a94fbcafe">armnn::INetwork::AddDivisionLayer</a></div><div class="ttdeci">IConnectableLayer * AddDivisionLayer(const char *name=nullptr)</div><div class="ttdoc">Adds a division layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00508">Network.cpp:508</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]499<div class="ttc" id="a_arm_compute_utils_8hpp_html"><div class="ttname"><a href="_arm_compute_utils_8hpp.html">ArmComputeUtils.hpp</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]500<div class="ttc" id="anamespacearmnn_html_ad2021f80998e436e27de7b7eeec67d54"><div class="ttname"><a href="namespacearmnn.html#ad2021f80998e436e27de7b7eeec67d54">armnn::FuseFullyConnectedLayer</a></div><div class="ttdeci">LayerType * FuseFullyConnectedLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00252">ArmComputeSubgraphUtils.hpp:252</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]501<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 Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]502<div class="ttc" id="anamespacearmnn_html_a6a28e34d999aac5a0149d35b02a3cfe5"><div class="ttname"><a href="namespacearmnn.html#a6a28e34d999aac5a0149d35b02a3cfe5">armnn::FuseElementwiseBinaryLayer</a></div><div class="ttdeci">LayerType * FuseElementwiseBinaryLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, BinaryOperation operation, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00157">ArmComputeSubgraphUtils.hpp:157</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]503<div class="ttc" id="aclassarmnn_1_1_i_network_html_aab976306ba3424cbc0c7ea33417d718d"><div class="ttname"><a href="classarmnn_1_1_i_network.html#aab976306ba3424cbc0c7ea33417d718d">armnn::INetwork::AddElementwiseBinaryLayer</a></div><div class="ttdeci">IConnectableLayer * AddElementwiseBinaryLayer(const ElementwiseBinaryDescriptor &amp;elementwiseBinaryDescriptor, const char *name=nullptr)</div><div class="ttdoc">Add an ElementwiseBinary layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00314">Network.cpp:314</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]504<div class="ttc" id="astructarmnn_1_1_reduce_descriptor_html_a28e0548abfc4e79c48f29a3d11a062e9"><div class="ttname"><a href="structarmnn_1_1_reduce_descriptor.html#a28e0548abfc4e79c48f29a3d11a062e9">armnn::ReduceDescriptor::m_KeepDims</a></div><div class="ttdeci">bool m_KeepDims</div><div class="ttdoc">if true then output shape has no change.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01554">Descriptors.hpp:1554</a></div></div>
505<div class="ttc" id="aclassarmnn_1_1_i_network_html_af13795cdf49e63d8bc3cb409592cdb9d"><div class="ttname"><a href="classarmnn_1_1_i_network.html#af13795cdf49e63d8bc3cb409592cdb9d">armnn::INetwork::AddSubtractionLayer</a></div><div class="ttdeci">IConnectableLayer * AddSubtractionLayer(const char *name=nullptr)</div><div class="ttdoc">Adds a subtraction layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00515">Network.cpp:515</a></div></div>
506<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>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]507<div class="ttc" id="aclassarmnn_1_1_activation_layer_html"><div class="ttname"><a href="classarmnn_1_1_activation_layer.html">armnn::ActivationLayer</a></div><div class="ttdoc">This layer represents an activation operation with the specified activation function.</div><div class="ttdef"><b>Definition:</b> <a href="_activation_layer_8hpp_source.html#l00012">ActivationLayer.hpp:12</a></div></div>
508<div class="ttc" id="a_deprecated_8hpp_html_ad762b11b48e5c1d1c1743f529485728a"><div class="ttname"><a href="_deprecated_8hpp.html#ad762b11b48e5c1d1c1743f529485728a">ARMNN_NO_DEPRECATE_WARN_END</a></div><div class="ttdeci">#define ARMNN_NO_DEPRECATE_WARN_END</div><div class="ttdef"><b>Definition:</b> <a href="_deprecated_8hpp_source.html#l00034">Deprecated.hpp:34</a></div></div>
509<div class="ttc" id="aclassarmnn_1_1_i_network_html_ac6771c8752811fcfa91b3f0dee748144"><div class="ttname"><a href="classarmnn_1_1_i_network.html#ac6771c8752811fcfa91b3f0dee748144">armnn::INetwork::AddDepthwiseConvolution2dLayer</a></div><div class="ttdeci">IConnectableLayer * AddDepthwiseConvolution2dLayer(const DepthwiseConvolution2dDescriptor &amp;convolution2dDescriptor, const char *name=nullptr)</div><div class="ttdoc">Adds a 2D depthwise convolution layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00292">Network.cpp:292</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]510<div class="ttc" id="astructarmnn_1_1_reduce_descriptor_html_aa1c6fc8c96404252f1072632fc5acb59"><div class="ttname"><a href="structarmnn_1_1_reduce_descriptor.html#aa1c6fc8c96404252f1072632fc5acb59">armnn::ReduceDescriptor::m_vAxis</a></div><div class="ttdeci">std::vector&lt; uint32_t &gt; m_vAxis</div><div class="ttdoc">The indices of the dimensions to reduce.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01556">Descriptors.hpp:1556</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]511<div class="ttc" id="anamespacearmnn_html_aeda4132cdab76b98c860690c80fdb7bd"><div class="ttname"><a href="namespacearmnn.html#aeda4132cdab76b98c860690c80fdb7bd">armnn::FuseSubtractionLayer</a></div><div class="ttdeci">LayerType * FuseSubtractionLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc, std::string name)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00094">ArmComputeSubgraphUtils.hpp:94</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]512<div class="ttc" id="anamespacearmnn_html_ad44c007f21af2d0375e3ef9400a1b275"><div class="ttname"><a href="namespacearmnn.html#ad44c007f21af2d0375e3ef9400a1b275">armnn::IsQuantizedType</a></div><div class="ttdeci">constexpr bool IsQuantizedType()</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00311">TypesUtils.hpp:311</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]513<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>
514<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>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]515<div class="ttc" id="aclassarmnn_1_1_i_network_html_abb59f6ba9988dae88e0f48e68d87fc32"><div class="ttname"><a href="classarmnn_1_1_i_network.html#abb59f6ba9988dae88e0f48e68d87fc32">armnn::INetwork::AddMultiplicationLayer</a></div><div class="ttdeci">IConnectableLayer * AddMultiplicationLayer(const char *name=nullptr)</div><div class="ttdoc">Adds a multiplication layer to the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00416">Network.cpp:416</a></div></div>
Nikhil Raj38b600d2024-02-15 15:02:19 +0000[diff] [blame]516<div class="ttc" id="aclassarmnn_1_1_const_tensor_html"><div class="ttname"><a href="classarmnn_1_1_const_tensor.html">armnn::ConstTensor</a></div><div class="ttdoc">A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00329">Tensor.hpp:329</a></div></div>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]517<div class="ttc" id="anamespacearmnn_html_a9d43aa01b882d8776ab9403eea3cf779"><div class="ttname"><a href="namespacearmnn.html#a9d43aa01b882d8776ab9403eea3cf779">armnn::ComputeReductionTensorShape</a></div><div class="ttdeci">const TensorInfo ComputeReductionTensorShape(const armnn::TensorInfo &amp;input, const std::vector&lt; uint32_t &gt; &amp;vAxis, const bool keepDims)</div><div class="ttdoc">Function to compute the output tensor shape based on the axes and if keepDims is set.</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_utils_8hpp_source.html#l00320">ArmComputeUtils.hpp:320</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]518<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>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]519<div class="ttc" id="anamespacearmnn_html_a7e4811ef91f499424bca36e98a2fbd96"><div class="ttname"><a href="namespacearmnn.html#a7e4811ef91f499424bca36e98a2fbd96">armnn::FuseLayer</a></div><div class="ttdeci">LayerType * FuseLayer(OptimizationViews &amp;optimizationViews, LayerType *baseLayer, LayerType *replacementLayer, ActivationLayer *activationLayer, ActivationDescriptor &amp;activationDesc)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00053">ArmComputeSubgraphUtils.hpp:53</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]520<div class="ttc" id="astructarmnn_1_1_reduce_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_reduce_descriptor.html">armnn::ReduceDescriptor</a></div><div class="ttdoc">A ReduceDescriptor for the REDUCE operators.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01538">Descriptors.hpp:1538</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]521<div class="ttc" id="a_subgraph_utils_8hpp_html"><div class="ttname"><a href="_subgraph_utils_8hpp.html">SubgraphUtils.hpp</a></div></div>
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]522<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]523<div class="ttc" id="anamespacearmnn_html_af8b1723c2ceaf3aca2deddbf9b0e3e44"><div class="ttname"><a href="namespacearmnn.html#af8b1723c2ceaf3aca2deddbf9b0e3e44">armnn::ReplaceMultipleLayers</a></div><div class="ttdeci">void ReplaceMultipleLayers(OptimizationViews &amp;optimizationViews, std::vector&lt; IConnectableLayer * &gt; &amp;originalLayers, LayerType *baseLayer, const std::vector&lt; SlotList &gt; inputLayersSlotLists, const std::vector&lt; SlotList &gt; outputLayersSlotLists)</div><div class="ttdef"><b>Definition:</b> <a href="_arm_compute_subgraph_utils_8hpp_source.html#l00357">ArmComputeSubgraphUtils.hpp:357</a></div></div>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]524<!-- start footer part -->
525<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
526 <ul>
527 <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_c13beb47b846b3a63741c705c772cf8d.html">aclCommon</a></li><li class="navelem"><a class="el" href="_arm_compute_subgraph_utils_8hpp.html">ArmComputeSubgraphUtils.hpp</a></li>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]528 <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]529 <a href="http://www.doxygen.org/index.html">
530 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li>
531 </ul>
532</div>
533</body>
534</html>