Nikhil Raj | 1dc83fe | 2024-05-16 09:47:51 +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: shim/sl/canonical/Converter.cpp 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"> |
| 39 |  <span id="projectnumber">24.05</span> |
| 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&dn=gpl-2.0.txt GPL-v2 */ |
| 50 | var 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&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&dn=gpl-2.0.txt GPL-v2 */ |
| 76 | $(document).ready(function(){initNavTree('_converter_8cpp_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">Converter.cpp</div> </div> |
| 97 | </div><!--header--> |
| 98 | <div class="contents"> |
| 99 | <a href="_converter_8cpp.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">//</span></div> |
| 100 | <div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <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> <span class="comment">// SPDX-License-Identifier: MIT</span></div> |
| 102 | <div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment">//</span></div> |
| 103 | <div class="line"><a name="l00005"></a><span class="lineno"> 5</span>  </div> |
| 104 | <div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="preprocessor">#include "<a class="code" href="_converter_8hpp.html">Converter.hpp</a>"</span></div> |
| 105 | <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="preprocessor">#include <half/half.hpp></span></div> |
| 106 | <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="preprocessor">#include <<a class="code" href="_exceptions_8hpp.html">armnn/Exceptions.hpp</a>></span></div> |
| 107 | <div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="preprocessor">#include <<a class="code" href="_tensor_utils_8hpp.html">armnnUtils/TensorUtils.hpp</a>></span></div> |
| 108 | <div class="line"><a name="l00010"></a><span class="lineno"> 10</span>  </div> |
| 109 | <div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="keyword">namespace </span><a class="code" href="namespacearmnn__driver.html">armnn_driver</a></div> |
| 110 | <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> {</div> |
| 111 | <div class="line"><a name="l00013"></a><span class="lineno"> 13</span>  </div> |
| 112 | <div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="keyword">using namespace </span><a class="code" href="namespaceandroid_1_1nn.html">android::nn</a>;</div> |
| 113 | <div class="line"><a name="l00015"></a><span class="lineno"><a class="line" href="namespacearmnn__driver.html#a0f38fa92b2468d5378258a2b074c1a31"> 15</a></span> <span class="keyword">using</span> <a class="code" href="namespacearmnn__driver.html#a0f38fa92b2468d5378258a2b074c1a31">Half</a> = half_float::half;</div> |
| 114 | <div class="line"><a name="l00016"></a><span class="lineno"> 16</span>  </div> |
| 115 | <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="keyword">namespace</span></div> |
| 116 | <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> {</div> |
| 117 | <div class="line"><a name="l00019"></a><span class="lineno"> 19</span>  </div> |
| 118 | <div class="line"><a name="l00020"></a><span class="lineno"> 20</span> } <span class="comment">// anonymouse namespace</span></div> |
| 119 | <div class="line"><a name="l00021"></a><span class="lineno"> 21</span>  </div> |
| 120 | <div class="line"><a name="l00022"></a><span class="lineno"><a class="line" href="classarmnn__driver_1_1_converter.html#a066acd78f531b709b725a730bfbee887"> 22</a></span> <span class="keywordtype">bool</span> <a class="code" href="classarmnn__driver_1_1_converter.html#a066acd78f531b709b725a730bfbee887">Converter::ConvertOperation</a>(<span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_converter.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_converter.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 121 | <div class="line"><a name="l00023"></a><span class="lineno"> 23</span> {</div> |
| 122 | <div class="line"><a name="l00024"></a><span class="lineno"> 24</span>  <span class="keywordflow">switch</span> (operation.type)</div> |
| 123 | <div class="line"><a name="l00025"></a><span class="lineno"> 25</span>  {</div> |
| 124 | <div class="line"><a name="l00026"></a><span class="lineno"> 26</span>  <span class="keywordflow">case</span> OperationType::ABS:</div> |
| 125 | <div class="line"><a name="l00027"></a><span class="lineno"> 27</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Abs);</div> |
| 126 | <div class="line"><a name="l00028"></a><span class="lineno"> 28</span>  <span class="keywordflow">case</span> OperationType::ADD:</div> |
| 127 | <div class="line"><a name="l00029"></a><span class="lineno"> 29</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047aec211f7c20af43e742bf2570c3cb84f9">armnn::BinaryOperation::Add</a>);</div> |
| 128 | <div class="line"><a name="l00030"></a><span class="lineno"> 30</span>  <span class="keywordflow">case</span> OperationType::ARGMAX:</div> |
| 129 | <div class="line"><a name="l00031"></a><span class="lineno"> 31</span>  <span class="keywordflow">return</span> ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Max);</div> |
| 130 | <div class="line"><a name="l00032"></a><span class="lineno"> 32</span>  <span class="keywordflow">case</span> OperationType::ARGMIN:</div> |
| 131 | <div class="line"><a name="l00033"></a><span class="lineno"> 33</span>  <span class="keywordflow">return</span> ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Min);</div> |
| 132 | <div class="line"><a name="l00034"></a><span class="lineno"> 34</span>  <span class="keywordflow">case</span> OperationType::AVERAGE_POOL_2D:</div> |
| 133 | <div class="line"><a name="l00035"></a><span class="lineno"> 35</span>  <span class="keywordflow">return</span> ConvertAveragePool2d(operation, model, data);</div> |
| 134 | <div class="line"><a name="l00036"></a><span class="lineno"> 36</span>  <span class="keywordflow">case</span> OperationType::BATCH_MATMUL:</div> |
| 135 | <div class="line"><a name="l00037"></a><span class="lineno"> 37</span>  <span class="keywordflow">return</span> ConvertBatchMatMul(operation, model, data);</div> |
| 136 | <div class="line"><a name="l00038"></a><span class="lineno"> 38</span>  <span class="keywordflow">case</span> OperationType::BATCH_TO_SPACE_ND:</div> |
| 137 | <div class="line"><a name="l00039"></a><span class="lineno"> 39</span>  <span class="keywordflow">return</span> ConvertBatchToSpaceNd(operation, model, data);</div> |
| 138 | <div class="line"><a name="l00040"></a><span class="lineno"> 40</span>  <span class="keywordflow">case</span> OperationType::CAST:</div> |
| 139 | <div class="line"><a name="l00041"></a><span class="lineno"> 41</span>  <span class="keywordflow">return</span> ConvertCast(operation, model, data);</div> |
| 140 | <div class="line"><a name="l00042"></a><span class="lineno"> 42</span>  <span class="keywordflow">case</span> OperationType::CONCATENATION:</div> |
| 141 | <div class="line"><a name="l00043"></a><span class="lineno"> 43</span>  <span class="keywordflow">return</span> ConvertConcatenation(operation, model, data);</div> |
| 142 | <div class="line"><a name="l00044"></a><span class="lineno"> 44</span>  <span class="keywordflow">case</span> OperationType::CONV_2D:</div> |
| 143 | <div class="line"><a name="l00045"></a><span class="lineno"> 45</span>  <span class="keywordflow">return</span> ConvertConv2d(operation, model, data);</div> |
| 144 | <div class="line"><a name="l00046"></a><span class="lineno"> 46</span>  <span class="keywordflow">case</span> OperationType::DEPTH_TO_SPACE:</div> |
| 145 | <div class="line"><a name="l00047"></a><span class="lineno"> 47</span>  <span class="keywordflow">return</span> ConvertDepthToSpace(operation, model, data);</div> |
| 146 | <div class="line"><a name="l00048"></a><span class="lineno"> 48</span>  <span class="keywordflow">case</span> OperationType::DEPTHWISE_CONV_2D:</div> |
| 147 | <div class="line"><a name="l00049"></a><span class="lineno"> 49</span>  <span class="keywordflow">return</span> ConvertDepthwiseConv2d(operation, model, data);</div> |
| 148 | <div class="line"><a name="l00050"></a><span class="lineno"> 50</span>  <span class="keywordflow">case</span> OperationType::DEQUANTIZE:</div> |
| 149 | <div class="line"><a name="l00051"></a><span class="lineno"> 51</span>  <span class="keywordflow">return</span> ConvertDequantize(operation, model, data);</div> |
| 150 | <div class="line"><a name="l00052"></a><span class="lineno"> 52</span>  <span class="keywordflow">case</span> OperationType::DIV:</div> |
| 151 | <div class="line"><a name="l00053"></a><span class="lineno"> 53</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047a43d11850893d2fe84a1e618121c1cc0a">armnn::BinaryOperation::Div</a>);</div> |
| 152 | <div class="line"><a name="l00054"></a><span class="lineno"> 54</span>  <span class="keywordflow">case</span> OperationType::ELU:</div> |
| 153 | <div class="line"><a name="l00055"></a><span class="lineno"> 55</span>  <span class="keywordflow">return</span> ConvertElu(operation, model, data);</div> |
| 154 | <div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  <span class="keywordflow">case</span> OperationType::EQUAL:</div> |
| 155 | <div class="line"><a name="l00057"></a><span class="lineno"> 57</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::Equal);</div> |
| 156 | <div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  <span class="keywordflow">case</span> OperationType::EXP:</div> |
| 157 | <div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Exp);</div> |
| 158 | <div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  <span class="keywordflow">case</span> OperationType::EXPAND_DIMS:</div> |
| 159 | <div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  <span class="keywordflow">return</span> ConvertExpandDims(operation, model, data);</div> |
| 160 | <div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  <span class="keywordflow">case</span> OperationType::FILL:</div> |
| 161 | <div class="line"><a name="l00063"></a><span class="lineno"> 63</span>  <span class="keywordflow">return</span> ConvertFill(operation, model, data);</div> |
| 162 | <div class="line"><a name="l00064"></a><span class="lineno"> 64</span>  <span class="keywordflow">case</span> OperationType::FLOOR:</div> |
| 163 | <div class="line"><a name="l00065"></a><span class="lineno"> 65</span>  <span class="keywordflow">return</span> ConvertFloor(operation, model, data);</div> |
| 164 | <div class="line"><a name="l00066"></a><span class="lineno"> 66</span>  <span class="keywordflow">case</span> OperationType::FULLY_CONNECTED:</div> |
| 165 | <div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  <span class="keywordflow">return</span> ConvertFullyConnected(operation, model, data);</div> |
| 166 | <div class="line"><a name="l00068"></a><span class="lineno"> 68</span>  <span class="keywordflow">case</span> OperationType::GATHER:</div> |
| 167 | <div class="line"><a name="l00069"></a><span class="lineno"> 69</span>  <span class="keywordflow">return</span> ConvertGather(operation, model, data);</div> |
| 168 | <div class="line"><a name="l00070"></a><span class="lineno"> 70</span>  <span class="keywordflow">case</span> OperationType::GREATER:</div> |
| 169 | <div class="line"><a name="l00071"></a><span class="lineno"> 71</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::Greater);</div> |
| 170 | <div class="line"><a name="l00072"></a><span class="lineno"> 72</span>  <span class="keywordflow">case</span> OperationType::GREATER_EQUAL:</div> |
| 171 | <div class="line"><a name="l00073"></a><span class="lineno"> 73</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::GreaterOrEqual);</div> |
| 172 | <div class="line"><a name="l00074"></a><span class="lineno"> 74</span>  <span class="keywordflow">case</span> OperationType::GROUPED_CONV_2D:</div> |
| 173 | <div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  <span class="keywordflow">return</span> ConvertGroupedConv2d(operation, model, data);</div> |
| 174 | <div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  <span class="keywordflow">case</span> OperationType::HARD_SWISH:</div> |
| 175 | <div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  <span class="keywordflow">return</span> ConvertHardSwish(operation, model, data);</div> |
| 176 | <div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  <span class="keywordflow">case</span> OperationType::INSTANCE_NORMALIZATION:</div> |
| 177 | <div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  <span class="keywordflow">return</span> ConvertInstanceNormalization(operation, model, data);</div> |
| 178 | <div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  <span class="keywordflow">case</span> OperationType::L2_NORMALIZATION:</div> |
| 179 | <div class="line"><a name="l00081"></a><span class="lineno"> 81</span>  <span class="keywordflow">return</span> ConvertL2Normalization(operation, model, data);</div> |
| 180 | <div class="line"><a name="l00082"></a><span class="lineno"> 82</span>  <span class="keywordflow">case</span> OperationType::L2_POOL_2D:</div> |
| 181 | <div class="line"><a name="l00083"></a><span class="lineno"> 83</span>  <span class="keywordflow">return</span> ConvertL2Pool2d(operation, model, data);</div> |
| 182 | <div class="line"><a name="l00084"></a><span class="lineno"> 84</span>  <span class="keywordflow">case</span> OperationType::LESS:</div> |
| 183 | <div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::Less);</div> |
| 184 | <div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  <span class="keywordflow">case</span> OperationType::LESS_EQUAL:</div> |
| 185 | <div class="line"><a name="l00087"></a><span class="lineno"> 87</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::LessOrEqual);</div> |
| 186 | <div class="line"><a name="l00088"></a><span class="lineno"> 88</span>  <span class="keywordflow">case</span> OperationType::LOCAL_RESPONSE_NORMALIZATION:</div> |
| 187 | <div class="line"><a name="l00089"></a><span class="lineno"> 89</span>  <span class="keywordflow">return</span> ConvertLocalResponseNormalization(operation, model, data);</div> |
| 188 | <div class="line"><a name="l00090"></a><span class="lineno"> 90</span>  <span class="keywordflow">case</span> OperationType::LOG:</div> |
| 189 | <div class="line"><a name="l00091"></a><span class="lineno"> 91</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Log);</div> |
| 190 | <div class="line"><a name="l00092"></a><span class="lineno"> 92</span>  <span class="keywordflow">case</span> OperationType::LOGICAL_AND:</div> |
| 191 | <div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  <span class="keywordflow">return</span> ConvertLogicalBinary(operation, model, data, LogicalBinaryOperation::LogicalAnd);</div> |
| 192 | <div class="line"><a name="l00094"></a><span class="lineno"> 94</span>  <span class="keywordflow">case</span> OperationType::LOGICAL_NOT:</div> |
| 193 | <div class="line"><a name="l00095"></a><span class="lineno"> 95</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::LogicalNot);</div> |
| 194 | <div class="line"><a name="l00096"></a><span class="lineno"> 96</span>  <span class="keywordflow">case</span> OperationType::LOGICAL_OR:</div> |
| 195 | <div class="line"><a name="l00097"></a><span class="lineno"> 97</span>  <span class="keywordflow">return</span> ConvertLogicalBinary(operation, model, data, LogicalBinaryOperation::LogicalOr);</div> |
| 196 | <div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  <span class="keywordflow">case</span> OperationType::LOGISTIC:</div> |
| 197 | <div class="line"><a name="l00099"></a><span class="lineno"> 99</span>  <span class="keywordflow">return</span> ConvertLogistic(operation, model, data);</div> |
| 198 | <div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  <span class="keywordflow">case</span> OperationType::LOG_SOFTMAX:</div> |
| 199 | <div class="line"><a name="l00101"></a><span class="lineno"> 101</span>  <span class="keywordflow">return</span> ConvertLogSoftmax(operation, model, data);</div> |
| 200 | <div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  <span class="keywordflow">case</span> OperationType::LSTM:</div> |
| 201 | <div class="line"><a name="l00103"></a><span class="lineno"> 103</span>  <span class="keywordflow">return</span> ConvertLstm(operation, model, data);</div> |
| 202 | <div class="line"><a name="l00104"></a><span class="lineno"> 104</span>  <span class="keywordflow">case</span> OperationType::MAX_POOL_2D:</div> |
| 203 | <div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  <span class="keywordflow">return</span> ConvertMaxPool2d(operation, model, data);</div> |
| 204 | <div class="line"><a name="l00106"></a><span class="lineno"> 106</span>  <span class="keywordflow">case</span> OperationType::MAXIMUM:</div> |
| 205 | <div class="line"><a name="l00107"></a><span class="lineno"> 107</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047a8321e79c278ec510f63675c040594892">armnn::BinaryOperation::Maximum</a>);</div> |
| 206 | <div class="line"><a name="l00108"></a><span class="lineno"> 108</span>  <span class="keywordflow">case</span> OperationType::MEAN:</div> |
| 207 | <div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  <span class="keywordflow">return</span> ConvertMean(operation, model, data);</div> |
| 208 | <div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  <span class="keywordflow">case</span> OperationType::MINIMUM:</div> |
| 209 | <div class="line"><a name="l00111"></a><span class="lineno"> 111</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047aa1d0ec6d56f8833a078b5a7ac4caf2d4">armnn::BinaryOperation::Minimum</a>);</div> |
| 210 | <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  <span class="keywordflow">case</span> OperationType::MUL:</div> |
| 211 | <div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047a62b6d55816cf737bfc6f42e60df1a3f2">armnn::BinaryOperation::Mul</a>);</div> |
| 212 | <div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <span class="keywordflow">case</span> OperationType::NEG:</div> |
| 213 | <div class="line"><a name="l00115"></a><span class="lineno"> 115</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Neg);</div> |
| 214 | <div class="line"><a name="l00116"></a><span class="lineno"> 116</span>  <span class="keywordflow">case</span> OperationType::NOT_EQUAL:</div> |
| 215 | <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::NotEqual);</div> |
| 216 | <div class="line"><a name="l00118"></a><span class="lineno"> 118</span>  <span class="keywordflow">case</span> OperationType::PAD:</div> |
| 217 | <div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <span class="keywordflow">return</span> ConvertPad(operation, model, data);</div> |
| 218 | <div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  <span class="keywordflow">case</span> OperationType::PAD_V2:</div> |
| 219 | <div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  <span class="keywordflow">return</span> ConvertPadV2(operation, model, data);</div> |
| 220 | <div class="line"><a name="l00122"></a><span class="lineno"> 122</span>  <span class="keywordflow">case</span> OperationType::PRELU:</div> |
| 221 | <div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  <span class="keywordflow">return</span> ConvertPrelu(operation, model, data);</div> |
| 222 | <div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  <span class="keywordflow">case</span> OperationType::POW:</div> |
| 223 | <div class="line"><a name="l00125"></a><span class="lineno"> 125</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, BinaryOperation::Power);</div> |
| 224 | <div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  <span class="keywordflow">case</span> OperationType::QUANTIZE:</div> |
| 225 | <div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  <span class="keywordflow">return</span> ConvertQuantize(operation, model, data);</div> |
| 226 | <div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  <span class="keywordflow">case</span> OperationType::QUANTIZED_LSTM:</div> |
| 227 | <div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <span class="keywordflow">return</span> ConvertQuantizedLstm(operation, model, data);</div> |
| 228 | <div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  <span class="keywordflow">case</span> OperationType::QUANTIZED_16BIT_LSTM:</div> |
| 229 | <div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  <span class="keywordflow">return</span> ConvertQuantized16BitLstm(operation, model, data);</div> |
| 230 | <div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  <span class="keywordflow">case</span> OperationType::RANK:</div> |
| 231 | <div class="line"><a name="l00133"></a><span class="lineno"> 133</span>  <span class="keywordflow">return</span> ConvertRank(operation, model, data);</div> |
| 232 | <div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  <span class="keywordflow">case</span> OperationType::REDUCE_MAX:</div> |
| 233 | <div class="line"><a name="l00135"></a><span class="lineno"> 135</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a3f6b313fd199d1552844f3f95fd25cd2">ConvertReduce</a>(operation, model, data, <a class="code" href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0a6a061313d22e51e0f25b7cd4dc065233">armnn::ReduceOperation::Max</a>);</div> |
| 234 | <div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  <span class="keywordflow">case</span> OperationType::REDUCE_MIN:</div> |
| 235 | <div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a3f6b313fd199d1552844f3f95fd25cd2">ConvertReduce</a>(operation, model, data, <a class="code" href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0a78d811e98514cd165dda532286610fd2">armnn::ReduceOperation::Min</a>);</div> |
| 236 | <div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  <span class="keywordflow">case</span> OperationType::REDUCE_PROD:</div> |
| 237 | <div class="line"><a name="l00139"></a><span class="lineno"> 139</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a3f6b313fd199d1552844f3f95fd25cd2">ConvertReduce</a>(operation, model, data, <a class="code" href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0a362a33c23b08e4a32a4ec53fbb82cccd">armnn::ReduceOperation::Prod</a>);</div> |
| 238 | <div class="line"><a name="l00140"></a><span class="lineno"> 140</span>  <span class="keywordflow">case</span> OperationType::REDUCE_SUM:</div> |
| 239 | <div class="line"><a name="l00141"></a><span class="lineno"> 141</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a3f6b313fd199d1552844f3f95fd25cd2">ConvertReduce</a>(operation, model, data, <a class="code" href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0aa0ec87054b5e5b7847d0d8780a01a3d5">armnn::ReduceOperation::Sum</a>);</div> |
| 240 | <div class="line"><a name="l00142"></a><span class="lineno"> 142</span>  <span class="keywordflow">case</span> OperationType::RELU:</div> |
| 241 | <div class="line"><a name="l00143"></a><span class="lineno"> 143</span>  <span class="keywordflow">return</span> ConvertReLu(operation, model, data);</div> |
| 242 | <div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keywordflow">case</span> OperationType::RELU1:</div> |
| 243 | <div class="line"><a name="l00145"></a><span class="lineno"> 145</span>  <span class="keywordflow">return</span> ConvertReLu1(operation, model, data);</div> |
| 244 | <div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  <span class="keywordflow">case</span> OperationType::RELU6:</div> |
| 245 | <div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  <span class="keywordflow">return</span> ConvertReLu6(operation, model, data);</div> |
| 246 | <div class="line"><a name="l00148"></a><span class="lineno"> 148</span>  <span class="keywordflow">case</span> OperationType::RESHAPE:</div> |
| 247 | <div class="line"><a name="l00149"></a><span class="lineno"> 149</span>  <span class="keywordflow">return</span> ConvertReshape(operation, model, data);</div> |
| 248 | <div class="line"><a name="l00150"></a><span class="lineno"> 150</span>  <span class="keywordflow">case</span> OperationType::RESIZE_BILINEAR:</div> |
| 249 | <div class="line"><a name="l00151"></a><span class="lineno"> 151</span>  <span class="keywordflow">return</span> ConvertResize(operation, model, data, ResizeMethod::Bilinear);</div> |
| 250 | <div class="line"><a name="l00152"></a><span class="lineno"> 152</span>  <span class="keywordflow">case</span> OperationType::RESIZE_NEAREST_NEIGHBOR:</div> |
| 251 | <div class="line"><a name="l00153"></a><span class="lineno"> 153</span>  <span class="keywordflow">return</span> ConvertResize(operation, model, data, ResizeMethod::NearestNeighbor);</div> |
| 252 | <div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  <span class="keywordflow">case</span> OperationType::REVERSE:</div> |
| 253 | <div class="line"><a name="l00155"></a><span class="lineno"> 155</span>  <span class="keywordflow">return</span> ConvertReverseV2(operation, model, data);</div> |
| 254 | <div class="line"><a name="l00156"></a><span class="lineno"> 156</span>  <span class="keywordflow">case</span> OperationType::RSQRT:</div> |
| 255 | <div class="line"><a name="l00157"></a><span class="lineno"> 157</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Rsqrt);</div> |
| 256 | <div class="line"><a name="l00158"></a><span class="lineno"> 158</span>  <span class="keywordflow">case</span> OperationType::SIN:</div> |
| 257 | <div class="line"><a name="l00159"></a><span class="lineno"> 159</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Sin);</div> |
| 258 | <div class="line"><a name="l00160"></a><span class="lineno"> 160</span>  <span class="keywordflow">case</span> OperationType::SOFTMAX:</div> |
| 259 | <div class="line"><a name="l00161"></a><span class="lineno"> 161</span>  <span class="keywordflow">return</span> ConvertSoftmax(operation, model, data);</div> |
| 260 | <div class="line"><a name="l00162"></a><span class="lineno"> 162</span>  <span class="keywordflow">case</span> OperationType::SPACE_TO_BATCH_ND :</div> |
| 261 | <div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  <span class="keywordflow">return</span> ConvertSpaceToBatchNd(operation, model, data);</div> |
| 262 | <div class="line"><a name="l00164"></a><span class="lineno"> 164</span>  <span class="keywordflow">case</span> OperationType::SPACE_TO_DEPTH:</div> |
| 263 | <div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  <span class="keywordflow">return</span> ConvertSpaceToDepth(operation, model, data);</div> |
| 264 | <div class="line"><a name="l00166"></a><span class="lineno"> 166</span>  <span class="keywordflow">case</span> OperationType::SQRT:</div> |
| 265 | <div class="line"><a name="l00167"></a><span class="lineno"> 167</span>  <span class="keywordflow">return</span> ConvertSqrt(operation, model, data);</div> |
| 266 | <div class="line"><a name="l00168"></a><span class="lineno"> 168</span>  <span class="keywordflow">case</span> OperationType::SQUEEZE:</div> |
| 267 | <div class="line"><a name="l00169"></a><span class="lineno"> 169</span>  <span class="keywordflow">return</span> ConvertSqueeze(operation, model, data);</div> |
| 268 | <div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  <span class="keywordflow">case</span> OperationType::STRIDED_SLICE:</div> |
| 269 | <div class="line"><a name="l00171"></a><span class="lineno"> 171</span>  <span class="keywordflow">return</span> ConvertStridedSlice(operation, model, data);</div> |
| 270 | <div class="line"><a name="l00172"></a><span class="lineno"> 172</span>  <span class="keywordflow">case</span> OperationType::SUB:</div> |
| 271 | <div class="line"><a name="l00173"></a><span class="lineno"> 173</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, BinaryOperation::Sub);</div> |
| 272 | <div class="line"><a name="l00174"></a><span class="lineno"> 174</span>  <span class="keywordflow">case</span> OperationType::TILE:</div> |
| 273 | <div class="line"><a name="l00175"></a><span class="lineno"> 175</span>  <span class="keywordflow">return</span> ConvertTile(operation, model, data);</div> |
| 274 | <div class="line"><a name="l00176"></a><span class="lineno"> 176</span>  <span class="keywordflow">case</span> OperationType::TRANSPOSE:</div> |
| 275 | <div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  <span class="keywordflow">return</span> ConvertTranspose(operation, model, data);</div> |
| 276 | <div class="line"><a name="l00178"></a><span class="lineno"> 178</span>  <span class="keywordflow">case</span> OperationType::TRANSPOSE_CONV_2D:</div> |
| 277 | <div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  <span class="keywordflow">return</span> ConvertTransposeConv2d(operation, model, data);</div> |
| 278 | <div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  <span class="keywordflow">case</span> OperationType::TANH:</div> |
| 279 | <div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  <span class="keywordflow">return</span> ConvertTanH(operation, model, data);</div> |
| 280 | <div class="line"><a name="l00182"></a><span class="lineno"> 182</span>  <span class="keywordflow">default</span>:</div> |
| 281 | <div class="line"><a name="l00183"></a><span class="lineno"> 183</span>  VLOG(DRIVER) << <span class="stringliteral">"Operation type: "</span> << operation.type << <span class="stringliteral">"is not supported in ArmnnDriver"</span>;</div> |
| 282 | <div class="line"><a name="l00184"></a><span class="lineno"> 184</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 283 | <div class="line"><a name="l00185"></a><span class="lineno"> 185</span>  }</div> |
| 284 | <div class="line"><a name="l00186"></a><span class="lineno"> 186</span> }</div> |
| 285 | <div class="line"><a name="l00187"></a><span class="lineno"> 187</span>  </div> |
| 286 | <div class="line"><a name="l00188"></a><span class="lineno"> 188</span> <span class="keywordtype">bool</span> Converter::ConvertArgMinMax(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 287 | <div class="line"><a name="l00189"></a><span class="lineno"> 189</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 288 | <div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data,</div> |
| 289 | <div class="line"><a name="l00191"></a><span class="lineno"> 191</span>  <a class="code" href="namespacearmnn.html#ae7e8cbf71db6a490789ca6dcaa8deeae">armnn::ArgMinMaxFunction</a> argMinMaxFunction)</div> |
| 290 | <div class="line"><a name="l00192"></a><span class="lineno"> 192</span> {</div> |
| 291 | <div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertArgMinMax()"</span>;</div> |
| 292 | <div class="line"><a name="l00194"></a><span class="lineno"> 194</span>  VLOG(DRIVER) << <span class="stringliteral">"argMinMaxFunction = "</span> << <a class="code" href="namespacearmnn.html#a5cda3502382f06a64c3cbeb1829bd850">GetArgMinMaxFunctionAsCString</a>(argMinMaxFunction);</div> |
| 293 | <div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  </div> |
| 294 | <div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 295 | <div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  </div> |
| 296 | <div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  <span class="keywordflow">if</span> (!input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 297 | <div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  {</div> |
| 298 | <div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 299 | <div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  }</div> |
| 300 | <div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  </div> |
| 301 | <div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  int32_t axis;</div> |
| 302 | <div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, axis, model, data))</div> |
| 303 | <div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  {</div> |
| 304 | <div class="line"><a name="l00206"></a><span class="lineno"> 206</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs. Failed to read axis."</span>, __func__);</div> |
| 305 | <div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  }</div> |
| 306 | <div class="line"><a name="l00208"></a><span class="lineno"> 208</span>  </div> |
| 307 | <div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 308 | <div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  <span class="keywordtype">int</span> rank = <span class="keyword">static_cast<</span><span class="keywordtype">int</span><span class="keyword">></span>(inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>());</div> |
| 309 | <div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  </div> |
| 310 | <div class="line"><a name="l00212"></a><span class="lineno"> 212</span>  <span class="keywordflow">if</span> (((axis < -rank) && (axis < 0)) || ((axis >= rank) && (axis > 0)))</div> |
| 311 | <div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  {</div> |
| 312 | <div class="line"><a name="l00214"></a><span class="lineno"> 214</span>  <span class="comment">// Square bracket denotes inclusive n while parenthesis denotes exclusive n</span></div> |
| 313 | <div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  <span class="comment">// E.g. Rank 4 tensor can have axis in range [-4, 3)</span></div> |
| 314 | <div class="line"><a name="l00216"></a><span class="lineno"> 216</span>  <span class="comment">// -1 == 3, -2 == 2, -3 == 1, -4 == 0</span></div> |
| 315 | <div class="line"><a name="l00217"></a><span class="lineno"> 217</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Axis must be in range [-n, n)"</span>, __func__);</div> |
| 316 | <div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  }</div> |
| 317 | <div class="line"><a name="l00219"></a><span class="lineno"> 219</span>  </div> |
| 318 | <div class="line"><a name="l00220"></a><span class="lineno"> 220</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 319 | <div class="line"><a name="l00221"></a><span class="lineno"> 221</span>  <span class="keywordflow">if</span> (!output)</div> |
| 320 | <div class="line"><a name="l00222"></a><span class="lineno"> 222</span>  {</div> |
| 321 | <div class="line"><a name="l00223"></a><span class="lineno"> 223</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 322 | <div class="line"><a name="l00224"></a><span class="lineno"> 224</span>  }</div> |
| 323 | <div class="line"><a name="l00225"></a><span class="lineno"> 225</span>  </div> |
| 324 | <div class="line"><a name="l00226"></a><span class="lineno"> 226</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo0 = input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 325 | <div class="line"><a name="l00227"></a><span class="lineno"> 227</span>  </div> |
| 326 | <div class="line"><a name="l00228"></a><span class="lineno"> 228</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 327 | <div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  </div> |
| 328 | <div class="line"><a name="l00230"></a><span class="lineno"> 230</span>  <a class="code" href="structarmnn_1_1_arg_min_max_descriptor.html">armnn::ArgMinMaxDescriptor</a> descriptor;</div> |
| 329 | <div class="line"><a name="l00231"></a><span class="lineno"> 231</span>  descriptor.<a class="code" href="structarmnn_1_1_arg_min_max_descriptor.html#ab1ae6f520bb1a4da191a0ae907477f23">m_Function</a> = argMinMaxFunction;</div> |
| 330 | <div class="line"><a name="l00232"></a><span class="lineno"> 232</span>  descriptor.<a class="code" href="structarmnn_1_1_arg_min_max_descriptor.html#a214c3636fdf0ea5bac8edb42d0e6c7f0">m_Axis</a> = axis;</div> |
| 331 | <div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  </div> |
| 332 | <div class="line"><a name="l00234"></a><span class="lineno"> 234</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 333 | <div class="line"><a name="l00235"></a><span class="lineno"> 235</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 334 | <div class="line"><a name="l00236"></a><span class="lineno"> 236</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 335 | <div class="line"><a name="l00237"></a><span class="lineno"> 237</span>  {</div> |
| 336 | <div class="line"><a name="l00238"></a><span class="lineno"> 238</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 337 | <div class="line"><a name="l00239"></a><span class="lineno"> 239</span>  IsArgMinMaxSupported,</div> |
| 338 | <div class="line"><a name="l00240"></a><span class="lineno"> 240</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 339 | <div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  isSupported,</div> |
| 340 | <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  setBackend,</div> |
| 341 | <div class="line"><a name="l00243"></a><span class="lineno"> 243</span>  inputInfo0,</div> |
| 342 | <div class="line"><a name="l00244"></a><span class="lineno"> 244</span>  outputInfo,</div> |
| 343 | <div class="line"><a name="l00245"></a><span class="lineno"> 245</span>  descriptor);</div> |
| 344 | <div class="line"><a name="l00246"></a><span class="lineno"> 246</span>  };</div> |
| 345 | <div class="line"><a name="l00247"></a><span class="lineno"> 247</span>  </div> |
| 346 | <div class="line"><a name="l00248"></a><span class="lineno"> 248</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 347 | <div class="line"><a name="l00249"></a><span class="lineno"> 249</span>  {</div> |
| 348 | <div class="line"><a name="l00250"></a><span class="lineno"> 250</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 349 | <div class="line"><a name="l00251"></a><span class="lineno"> 251</span>  }</div> |
| 350 | <div class="line"><a name="l00252"></a><span class="lineno"> 252</span>  <span class="keywordflow">else</span></div> |
| 351 | <div class="line"><a name="l00253"></a><span class="lineno"> 253</span>  {</div> |
| 352 | <div class="line"><a name="l00254"></a><span class="lineno"> 254</span>  validateFunc(outputInfo, isSupported);</div> |
| 353 | <div class="line"><a name="l00255"></a><span class="lineno"> 255</span>  }</div> |
| 354 | <div class="line"><a name="l00256"></a><span class="lineno"> 256</span>  </div> |
| 355 | <div class="line"><a name="l00257"></a><span class="lineno"> 257</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 356 | <div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  {</div> |
| 357 | <div class="line"><a name="l00259"></a><span class="lineno"> 259</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 358 | <div class="line"><a name="l00260"></a><span class="lineno"> 260</span>  }</div> |
| 359 | <div class="line"><a name="l00261"></a><span class="lineno"> 261</span>  </div> |
| 360 | <div class="line"><a name="l00262"></a><span class="lineno"> 262</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddArgMinMaxLayer(descriptor);</div> |
| 361 | <div class="line"><a name="l00263"></a><span class="lineno"> 263</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 362 | <div class="line"><a name="l00264"></a><span class="lineno"> 264</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 363 | <div class="line"><a name="l00265"></a><span class="lineno"> 265</span>  </div> |
| 364 | <div class="line"><a name="l00266"></a><span class="lineno"> 266</span>  input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 365 | <div class="line"><a name="l00267"></a><span class="lineno"> 267</span>  </div> |
| 366 | <div class="line"><a name="l00268"></a><span class="lineno"> 268</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 367 | <div class="line"><a name="l00269"></a><span class="lineno"> 269</span> }</div> |
| 368 | <div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  </div> |
| 369 | <div class="line"><a name="l00271"></a><span class="lineno"> 271</span> <span class="keywordtype">bool</span> Converter::ConvertAveragePool2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 370 | <div class="line"><a name="l00272"></a><span class="lineno"> 272</span> {</div> |
| 371 | <div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertAveragePool2d()"</span>;</div> |
| 372 | <div class="line"><a name="l00274"></a><span class="lineno"> 274</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a2eb75cb48464cf87c290503d7bea79a4">ConvertPooling2d</a>(operation, __func__, PoolingAlgorithm::Average, model, data);</div> |
| 373 | <div class="line"><a name="l00275"></a><span class="lineno"> 275</span> }</div> |
| 374 | <div class="line"><a name="l00276"></a><span class="lineno"> 276</span>  </div> |
| 375 | <div class="line"><a name="l00277"></a><span class="lineno"> 277</span> <span class="keywordtype">bool</span> Converter::ConvertBatchMatMul(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 376 | <div class="line"><a name="l00278"></a><span class="lineno"> 278</span> {</div> |
| 377 | <div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertBatchMatMul()"</span>;</div> |
| 378 | <div class="line"><a name="l00280"></a><span class="lineno"> 280</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 379 | <div class="line"><a name="l00281"></a><span class="lineno"> 281</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input1 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 380 | <div class="line"><a name="l00282"></a><span class="lineno"> 282</span>  </div> |
| 381 | <div class="line"><a name="l00283"></a><span class="lineno"> 283</span>  <span class="keywordflow">if</span> (!input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() || !input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 382 | <div class="line"><a name="l00284"></a><span class="lineno"> 284</span>  {</div> |
| 383 | <div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 384 | <div class="line"><a name="l00286"></a><span class="lineno"> 286</span>  }</div> |
| 385 | <div class="line"><a name="l00287"></a><span class="lineno"> 287</span>  </div> |
| 386 | <div class="line"><a name="l00288"></a><span class="lineno"> 288</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo0 = input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 387 | <div class="line"><a name="l00289"></a><span class="lineno"> 289</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo1 = input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 388 | <div class="line"><a name="l00290"></a><span class="lineno"> 290</span>  </div> |
| 389 | <div class="line"><a name="l00291"></a><span class="lineno"> 291</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rankInput0 = inputInfo0.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 390 | <div class="line"><a name="l00292"></a><span class="lineno"> 292</span>  <span class="keywordflow">if</span> (rankInput0 > 4 || rankInput0 < 2)</div> |
| 391 | <div class="line"><a name="l00293"></a><span class="lineno"> 293</span>  {</div> |
| 392 | <div class="line"><a name="l00294"></a><span class="lineno"> 294</span>  Fail(<span class="stringliteral">"%s: Only inputs with rank at least 2 and up to 4 are supported"</span>, __func__);</div> |
| 393 | <div class="line"><a name="l00295"></a><span class="lineno"> 295</span>  }</div> |
| 394 | <div class="line"><a name="l00296"></a><span class="lineno"> 296</span>  </div> |
| 395 | <div class="line"><a name="l00297"></a><span class="lineno"> 297</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rankInput1 = inputInfo1.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 396 | <div class="line"><a name="l00298"></a><span class="lineno"> 298</span>  <span class="keywordflow">if</span> (rankInput1 > 4 || rankInput1 < 2)</div> |
| 397 | <div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  {</div> |
| 398 | <div class="line"><a name="l00300"></a><span class="lineno"> 300</span>  Fail(<span class="stringliteral">"%s: Only inputs with rank at least 2 and up to 4 are supported"</span>, __func__);</div> |
| 399 | <div class="line"><a name="l00301"></a><span class="lineno"> 301</span>  }</div> |
| 400 | <div class="line"><a name="l00302"></a><span class="lineno"> 302</span>  </div> |
| 401 | <div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  <span class="comment">// Determine data type of input tensor 0</span></div> |
| 402 | <div class="line"><a name="l00304"></a><span class="lineno"> 304</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> input0Type;</div> |
| 403 | <div class="line"><a name="l00305"></a><span class="lineno"> 305</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 0, model, input0Type))</div> |
| 404 | <div class="line"><a name="l00306"></a><span class="lineno"> 306</span>  {</div> |
| 405 | <div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 406 | <div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  }</div> |
| 407 | <div class="line"><a name="l00309"></a><span class="lineno"> 309</span>  </div> |
| 408 | <div class="line"><a name="l00310"></a><span class="lineno"> 310</span>  <span class="comment">// Determine data type of input tensor 0</span></div> |
| 409 | <div class="line"><a name="l00311"></a><span class="lineno"> 311</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> input1Type;</div> |
| 410 | <div class="line"><a name="l00312"></a><span class="lineno"> 312</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 0, model, input1Type))</div> |
| 411 | <div class="line"><a name="l00313"></a><span class="lineno"> 313</span>  {</div> |
| 412 | <div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 413 | <div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  }</div> |
| 414 | <div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  </div> |
| 415 | <div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  <span class="keywordflow">if</span> (input0Type != input1Type)</div> |
| 416 | <div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  {</div> |
| 417 | <div class="line"><a name="l00319"></a><span class="lineno"> 319</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (Inputs must have same OperandCode)"</span>, __func__);</div> |
| 418 | <div class="line"><a name="l00320"></a><span class="lineno"> 320</span>  }</div> |
| 419 | <div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  </div> |
| 420 | <div class="line"><a name="l00322"></a><span class="lineno"> 322</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 421 | <div class="line"><a name="l00323"></a><span class="lineno"> 323</span>  <span class="keywordflow">if</span> (!output)</div> |
| 422 | <div class="line"><a name="l00324"></a><span class="lineno"> 324</span>  {</div> |
| 423 | <div class="line"><a name="l00325"></a><span class="lineno"> 325</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 424 | <div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  }</div> |
| 425 | <div class="line"><a name="l00327"></a><span class="lineno"> 327</span>  </div> |
| 426 | <div class="line"><a name="l00328"></a><span class="lineno"> 328</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 427 | <div class="line"><a name="l00329"></a><span class="lineno"> 329</span>  </div> |
| 428 | <div class="line"><a name="l00330"></a><span class="lineno"> 330</span>  <a class="code" href="structarmnn_1_1_batch_mat_mul_descriptor.html">armnn::BatchMatMulDescriptor</a> batchMatMulDesc;</div> |
| 429 | <div class="line"><a name="l00331"></a><span class="lineno"> 331</span>  </div> |
| 430 | <div class="line"><a name="l00332"></a><span class="lineno"> 332</span>  <span class="comment">// Inputs 2 and 3 are adjoint in Android NeuralNetworks, but they perform transpose.</span></div> |
| 431 | <div class="line"><a name="l00333"></a><span class="lineno"> 333</span>  <span class="comment">// This is why we are linking them with transpose parameters in the descriptor</span></div> |
| 432 | <div class="line"><a name="l00334"></a><span class="lineno"> 334</span>  batchMatMulDesc.<a class="code" href="structarmnn_1_1_batch_mat_mul_descriptor.html#acb441bb8db19bcce78d15cdd8ceb5ea0">m_TransposeX</a> = <a class="code" href="namespacearmnn__driver.html#ae28f9841f6cb1775849b7921b54da32e">GetOptionalBool</a>(operation, 2, model, data);</div> |
| 433 | <div class="line"><a name="l00335"></a><span class="lineno"> 335</span>  batchMatMulDesc.<a class="code" href="structarmnn_1_1_batch_mat_mul_descriptor.html#a112b466e5d2ab9d1887178adbe3afa1c">m_TransposeY</a> = <a class="code" href="namespacearmnn__driver.html#ae28f9841f6cb1775849b7921b54da32e">GetOptionalBool</a>(operation, 3, model, data);</div> |
| 434 | <div class="line"><a name="l00336"></a><span class="lineno"> 336</span>  </div> |
| 435 | <div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 436 | <div class="line"><a name="l00338"></a><span class="lineno"> 338</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 437 | <div class="line"><a name="l00339"></a><span class="lineno"> 339</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 438 | <div class="line"><a name="l00340"></a><span class="lineno"> 340</span>  {</div> |
| 439 | <div class="line"><a name="l00341"></a><span class="lineno"> 341</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 440 | <div class="line"><a name="l00342"></a><span class="lineno"> 342</span>  IsBatchMatMulSupported,</div> |
| 441 | <div class="line"><a name="l00343"></a><span class="lineno"> 343</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 442 | <div class="line"><a name="l00344"></a><span class="lineno"> 344</span>  isSupported,</div> |
| 443 | <div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  setBackend,</div> |
| 444 | <div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  inputInfo0,</div> |
| 445 | <div class="line"><a name="l00347"></a><span class="lineno"> 347</span>  inputInfo1,</div> |
| 446 | <div class="line"><a name="l00348"></a><span class="lineno"> 348</span>  outputInfo,</div> |
| 447 | <div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  batchMatMulDesc);</div> |
| 448 | <div class="line"><a name="l00350"></a><span class="lineno"> 350</span>  };</div> |
| 449 | <div class="line"><a name="l00351"></a><span class="lineno"> 351</span>  </div> |
| 450 | <div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 451 | <div class="line"><a name="l00353"></a><span class="lineno"> 353</span>  {</div> |
| 452 | <div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  validateFunc(outputInfo, isSupported);</div> |
| 453 | <div class="line"><a name="l00355"></a><span class="lineno"> 355</span>  }</div> |
| 454 | <div class="line"><a name="l00356"></a><span class="lineno"> 356</span>  <span class="keywordflow">else</span></div> |
| 455 | <div class="line"><a name="l00357"></a><span class="lineno"> 357</span>  {</div> |
| 456 | <div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 457 | <div class="line"><a name="l00359"></a><span class="lineno"> 359</span>  }</div> |
| 458 | <div class="line"><a name="l00360"></a><span class="lineno"> 360</span>  </div> |
| 459 | <div class="line"><a name="l00361"></a><span class="lineno"> 361</span>  </div> |
| 460 | <div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 461 | <div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  {</div> |
| 462 | <div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 463 | <div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  }</div> |
| 464 | <div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  </div> |
| 465 | <div class="line"><a name="l00367"></a><span class="lineno"> 367</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddBatchMatMulLayer(batchMatMulDesc);</div> |
| 466 | <div class="line"><a name="l00368"></a><span class="lineno"> 368</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 467 | <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 468 | <div class="line"><a name="l00370"></a><span class="lineno"> 370</span>  input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 469 | <div class="line"><a name="l00371"></a><span class="lineno"> 371</span>  input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 470 | <div class="line"><a name="l00372"></a><span class="lineno"> 372</span>  </div> |
| 471 | <div class="line"><a name="l00373"></a><span class="lineno"> 373</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 472 | <div class="line"><a name="l00374"></a><span class="lineno"> 374</span> }</div> |
| 473 | <div class="line"><a name="l00375"></a><span class="lineno"> 375</span>  </div> |
| 474 | <div class="line"><a name="l00376"></a><span class="lineno"> 376</span> <span class="keywordtype">bool</span> Converter::ConvertBatchToSpaceNd(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 475 | <div class="line"><a name="l00377"></a><span class="lineno"> 377</span> {</div> |
| 476 | <div class="line"><a name="l00378"></a><span class="lineno"> 378</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertBatchToSpaceNd()"</span>;</div> |
| 477 | <div class="line"><a name="l00379"></a><span class="lineno"> 379</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 478 | <div class="line"><a name="l00380"></a><span class="lineno"> 380</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 479 | <div class="line"><a name="l00381"></a><span class="lineno"> 381</span>  {</div> |
| 480 | <div class="line"><a name="l00382"></a><span class="lineno"> 382</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 481 | <div class="line"><a name="l00383"></a><span class="lineno"> 383</span>  }</div> |
| 482 | <div class="line"><a name="l00384"></a><span class="lineno"> 384</span>  </div> |
| 483 | <div class="line"><a name="l00385"></a><span class="lineno"> 385</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 484 | <div class="line"><a name="l00386"></a><span class="lineno"> 386</span>  <span class="keywordflow">if</span> (!output)</div> |
| 485 | <div class="line"><a name="l00387"></a><span class="lineno"> 387</span>  {</div> |
| 486 | <div class="line"><a name="l00388"></a><span class="lineno"> 388</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 487 | <div class="line"><a name="l00389"></a><span class="lineno"> 389</span>  }</div> |
| 488 | <div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  </div> |
| 489 | <div class="line"><a name="l00391"></a><span class="lineno"> 391</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 490 | <div class="line"><a name="l00392"></a><span class="lineno"> 392</span>  </div> |
| 491 | <div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* blockOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 492 | <div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  <span class="keywordflow">if</span> (!blockOperand)</div> |
| 493 | <div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  {</div> |
| 494 | <div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 1"</span>, __func__);</div> |
| 495 | <div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  }</div> |
| 496 | <div class="line"><a name="l00398"></a><span class="lineno"> 398</span>  </div> |
| 497 | <div class="line"><a name="l00399"></a><span class="lineno"> 399</span>  <span class="comment">// Convert the block operand to int32</span></div> |
| 498 | <div class="line"><a name="l00400"></a><span class="lineno"> 400</span>  std::vector<int32_t> block;</div> |
| 499 | <div class="line"><a name="l00401"></a><span class="lineno"> 401</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*blockOperand, block, model, data))</div> |
| 500 | <div class="line"><a name="l00402"></a><span class="lineno"> 402</span>  {</div> |
| 501 | <div class="line"><a name="l00403"></a><span class="lineno"> 403</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Input 1 has invalid values"</span>, __func__);</div> |
| 502 | <div class="line"><a name="l00404"></a><span class="lineno"> 404</span>  }</div> |
| 503 | <div class="line"><a name="l00405"></a><span class="lineno"> 405</span>  </div> |
| 504 | <div class="line"><a name="l00406"></a><span class="lineno"> 406</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 505 | <div class="line"><a name="l00407"></a><span class="lineno"> 407</span>  </div> |
| 506 | <div class="line"><a name="l00408"></a><span class="lineno"> 408</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 507 | <div class="line"><a name="l00409"></a><span class="lineno"> 409</span>  <span class="keywordflow">if</span> (rank != 4)</div> |
| 508 | <div class="line"><a name="l00410"></a><span class="lineno"> 410</span>  {</div> |
| 509 | <div class="line"><a name="l00411"></a><span class="lineno"> 411</span>  Fail(<span class="stringliteral">"%s: Only inputs with rank equal to 4 are supported"</span>, __func__);</div> |
| 510 | <div class="line"><a name="l00412"></a><span class="lineno"> 412</span>  }</div> |
| 511 | <div class="line"><a name="l00413"></a><span class="lineno"> 413</span>  </div> |
| 512 | <div class="line"><a name="l00414"></a><span class="lineno"> 414</span>  <span class="keywordflow">if</span> (std::any_of(block.cbegin(), block.cend(), [](int32_t i){ return i < 1; }))</div> |
| 513 | <div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  {</div> |
| 514 | <div class="line"><a name="l00416"></a><span class="lineno"> 416</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Block sizes for each spatial dimension of the input tensor must be"</span></div> |
| 515 | <div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  <span class="stringliteral">" greater than or equal to 1"</span>, __func__);</div> |
| 516 | <div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  }</div> |
| 517 | <div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  </div> |
| 518 | <div class="line"><a name="l00420"></a><span class="lineno"> 420</span>  <a class="code" href="structarmnn_1_1_batch_to_space_nd_descriptor.html">armnn::BatchToSpaceNdDescriptor</a> batchToSpaceNdDesc;</div> |
| 519 | <div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  batchToSpaceNdDesc.<a class="code" href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a02e143524aefddd40b485fcf7dea6696">m_BlockShape</a>.assign(block.cbegin(), block.cend());</div> |
| 520 | <div class="line"><a name="l00422"></a><span class="lineno"> 422</span>  batchToSpaceNdDesc.<a class="code" href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 521 | <div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  </div> |
| 522 | <div class="line"><a name="l00424"></a><span class="lineno"> 424</span>  <span class="keywordflow">if</span> (Is12OrLaterOperand(*output))</div> |
| 523 | <div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  {</div> |
| 524 | <div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  batchToSpaceNdDesc.<a class="code" href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 2, model, data);</div> |
| 525 | <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  }</div> |
| 526 | <div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="comment">// Setting crops to 0,0 0,0 as it is not supported in Android NN API</span></div> |
| 527 | <div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  batchToSpaceNdDesc.<a class="code" href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a3941f674c071c9503e00d2b59e92e454">m_Crops</a> = {{0, 0}, {0, 0}};</div> |
| 528 | <div class="line"><a name="l00430"></a><span class="lineno"> 430</span>  </div> |
| 529 | <div class="line"><a name="l00431"></a><span class="lineno"> 431</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 530 | <div class="line"><a name="l00432"></a><span class="lineno"> 432</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 531 | <div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 532 | <div class="line"><a name="l00434"></a><span class="lineno"> 434</span>  {</div> |
| 533 | <div class="line"><a name="l00435"></a><span class="lineno"> 435</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 534 | <div class="line"><a name="l00436"></a><span class="lineno"> 436</span>  IsBatchToSpaceNdSupported,</div> |
| 535 | <div class="line"><a name="l00437"></a><span class="lineno"> 437</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 536 | <div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  isSupported,</div> |
| 537 | <div class="line"><a name="l00439"></a><span class="lineno"> 439</span>  setBackend,</div> |
| 538 | <div class="line"><a name="l00440"></a><span class="lineno"> 440</span>  inputInfo,</div> |
| 539 | <div class="line"><a name="l00441"></a><span class="lineno"> 441</span>  outputInfo,</div> |
| 540 | <div class="line"><a name="l00442"></a><span class="lineno"> 442</span>  batchToSpaceNdDesc);</div> |
| 541 | <div class="line"><a name="l00443"></a><span class="lineno"> 443</span>  };</div> |
| 542 | <div class="line"><a name="l00444"></a><span class="lineno"> 444</span>  </div> |
| 543 | <div class="line"><a name="l00445"></a><span class="lineno"> 445</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 544 | <div class="line"><a name="l00446"></a><span class="lineno"> 446</span>  {</div> |
| 545 | <div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  validateFunc(outputInfo, isSupported);</div> |
| 546 | <div class="line"><a name="l00448"></a><span class="lineno"> 448</span>  }</div> |
| 547 | <div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  <span class="keywordflow">else</span></div> |
| 548 | <div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  {</div> |
| 549 | <div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 550 | <div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  }</div> |
| 551 | <div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  </div> |
| 552 | <div class="line"><a name="l00454"></a><span class="lineno"> 454</span>  </div> |
| 553 | <div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 554 | <div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  {</div> |
| 555 | <div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 556 | <div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  }</div> |
| 557 | <div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  </div> |
| 558 | <div class="line"><a name="l00460"></a><span class="lineno"> 460</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddBatchToSpaceNdLayer(batchToSpaceNdDesc);</div> |
| 559 | <div class="line"><a name="l00461"></a><span class="lineno"> 461</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 560 | <div class="line"><a name="l00462"></a><span class="lineno"> 462</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 561 | <div class="line"><a name="l00463"></a><span class="lineno"> 463</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 562 | <div class="line"><a name="l00464"></a><span class="lineno"> 464</span>  </div> |
| 563 | <div class="line"><a name="l00465"></a><span class="lineno"> 465</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 564 | <div class="line"><a name="l00466"></a><span class="lineno"> 466</span> }</div> |
| 565 | <div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  </div> |
| 566 | <div class="line"><a name="l00468"></a><span class="lineno"> 468</span> <span class="keywordtype">bool</span> Converter::ConvertCast(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 567 | <div class="line"><a name="l00469"></a><span class="lineno"> 469</span> {</div> |
| 568 | <div class="line"><a name="l00470"></a><span class="lineno"> 470</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertCast()"</span>;</div> |
| 569 | <div class="line"><a name="l00471"></a><span class="lineno"> 471</span>  </div> |
| 570 | <div class="line"><a name="l00472"></a><span class="lineno"> 472</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 571 | <div class="line"><a name="l00473"></a><span class="lineno"> 473</span>  </div> |
| 572 | <div class="line"><a name="l00474"></a><span class="lineno"> 474</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 573 | <div class="line"><a name="l00475"></a><span class="lineno"> 475</span>  {</div> |
| 574 | <div class="line"><a name="l00476"></a><span class="lineno"> 476</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 575 | <div class="line"><a name="l00477"></a><span class="lineno"> 477</span>  }</div> |
| 576 | <div class="line"><a name="l00478"></a><span class="lineno"> 478</span>  </div> |
| 577 | <div class="line"><a name="l00479"></a><span class="lineno"> 479</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 578 | <div class="line"><a name="l00480"></a><span class="lineno"> 480</span>  <span class="keywordflow">if</span> (!output)</div> |
| 579 | <div class="line"><a name="l00481"></a><span class="lineno"> 481</span>  {</div> |
| 580 | <div class="line"><a name="l00482"></a><span class="lineno"> 482</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 581 | <div class="line"><a name="l00483"></a><span class="lineno"> 483</span>  }</div> |
| 582 | <div class="line"><a name="l00484"></a><span class="lineno"> 484</span>  </div> |
| 583 | <div class="line"><a name="l00485"></a><span class="lineno"> 485</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 584 | <div class="line"><a name="l00486"></a><span class="lineno"> 486</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 585 | <div class="line"><a name="l00487"></a><span class="lineno"> 487</span>  </div> |
| 586 | <div class="line"><a name="l00488"></a><span class="lineno"> 488</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 587 | <div class="line"><a name="l00489"></a><span class="lineno"> 489</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 588 | <div class="line"><a name="l00490"></a><span class="lineno"> 490</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 589 | <div class="line"><a name="l00491"></a><span class="lineno"> 491</span>  {</div> |
| 590 | <div class="line"><a name="l00492"></a><span class="lineno"> 492</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 591 | <div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  IsCastSupported,</div> |
| 592 | <div class="line"><a name="l00494"></a><span class="lineno"> 494</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 593 | <div class="line"><a name="l00495"></a><span class="lineno"> 495</span>  isSupported,</div> |
| 594 | <div class="line"><a name="l00496"></a><span class="lineno"> 496</span>  setBackend,</div> |
| 595 | <div class="line"><a name="l00497"></a><span class="lineno"> 497</span>  inputInfo,</div> |
| 596 | <div class="line"><a name="l00498"></a><span class="lineno"> 498</span>  outputInfo);</div> |
| 597 | <div class="line"><a name="l00499"></a><span class="lineno"> 499</span>  };</div> |
| 598 | <div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  </div> |
| 599 | <div class="line"><a name="l00501"></a><span class="lineno"> 501</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 600 | <div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  {</div> |
| 601 | <div class="line"><a name="l00503"></a><span class="lineno"> 503</span>  validateFunc(outputInfo, isSupported);</div> |
| 602 | <div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  }</div> |
| 603 | <div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  <span class="keywordflow">else</span></div> |
| 604 | <div class="line"><a name="l00506"></a><span class="lineno"> 506</span>  {</div> |
| 605 | <div class="line"><a name="l00507"></a><span class="lineno"> 507</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 606 | <div class="line"><a name="l00508"></a><span class="lineno"> 508</span>  }</div> |
| 607 | <div class="line"><a name="l00509"></a><span class="lineno"> 509</span>  </div> |
| 608 | <div class="line"><a name="l00510"></a><span class="lineno"> 510</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 609 | <div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  {</div> |
| 610 | <div class="line"><a name="l00512"></a><span class="lineno"> 512</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 611 | <div class="line"><a name="l00513"></a><span class="lineno"> 513</span>  }</div> |
| 612 | <div class="line"><a name="l00514"></a><span class="lineno"> 514</span>  </div> |
| 613 | <div class="line"><a name="l00515"></a><span class="lineno"> 515</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddCastLayer();</div> |
| 614 | <div class="line"><a name="l00516"></a><span class="lineno"> 516</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 615 | <div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 616 | <div class="line"><a name="l00518"></a><span class="lineno"> 518</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 617 | <div class="line"><a name="l00519"></a><span class="lineno"> 519</span>  </div> |
| 618 | <div class="line"><a name="l00520"></a><span class="lineno"> 520</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 619 | <div class="line"><a name="l00521"></a><span class="lineno"> 521</span> }</div> |
| 620 | <div class="line"><a name="l00522"></a><span class="lineno"> 522</span>  </div> |
| 621 | <div class="line"><a name="l00523"></a><span class="lineno"> 523</span> <span class="keywordtype">bool</span> Converter::ConvertComparison(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 622 | <div class="line"><a name="l00524"></a><span class="lineno"> 524</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 623 | <div class="line"><a name="l00525"></a><span class="lineno"> 525</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data,</div> |
| 624 | <div class="line"><a name="l00526"></a><span class="lineno"> 526</span>  <a class="code" href="namespacearmnn.html#a2d299363c9fc33334c571fa29ca4f58c">ComparisonOperation</a> comparisonOperation)</div> |
| 625 | <div class="line"><a name="l00527"></a><span class="lineno"> 527</span> {</div> |
| 626 | <div class="line"><a name="l00528"></a><span class="lineno"> 528</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertComparison()"</span>;</div> |
| 627 | <div class="line"><a name="l00529"></a><span class="lineno"> 529</span>  VLOG(DRIVER) << <span class="stringliteral">"comparisonOperation = "</span> << <a class="code" href="namespacearmnn.html#aabb76a77e95921785f576bb29b495cd8">GetComparisonOperationAsCString</a>(comparisonOperation);</div> |
| 628 | <div class="line"><a name="l00530"></a><span class="lineno"> 530</span>  </div> |
| 629 | <div class="line"><a name="l00531"></a><span class="lineno"> 531</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 630 | <div class="line"><a name="l00532"></a><span class="lineno"> 532</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input1 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 631 | <div class="line"><a name="l00533"></a><span class="lineno"> 533</span>  </div> |
| 632 | <div class="line"><a name="l00534"></a><span class="lineno"> 534</span>  <span class="keywordflow">if</span> (!(input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>()))</div> |
| 633 | <div class="line"><a name="l00535"></a><span class="lineno"> 535</span>  {</div> |
| 634 | <div class="line"><a name="l00536"></a><span class="lineno"> 536</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 635 | <div class="line"><a name="l00537"></a><span class="lineno"> 537</span>  }</div> |
| 636 | <div class="line"><a name="l00538"></a><span class="lineno"> 538</span>  </div> |
| 637 | <div class="line"><a name="l00539"></a><span class="lineno"> 539</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 638 | <div class="line"><a name="l00540"></a><span class="lineno"> 540</span>  <span class="keywordflow">if</span> (!output)</div> |
| 639 | <div class="line"><a name="l00541"></a><span class="lineno"> 541</span>  {</div> |
| 640 | <div class="line"><a name="l00542"></a><span class="lineno"> 542</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 641 | <div class="line"><a name="l00543"></a><span class="lineno"> 543</span>  }</div> |
| 642 | <div class="line"><a name="l00544"></a><span class="lineno"> 544</span>  </div> |
| 643 | <div class="line"><a name="l00545"></a><span class="lineno"> 545</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo0 = input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 644 | <div class="line"><a name="l00546"></a><span class="lineno"> 546</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo1 = input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 645 | <div class="line"><a name="l00547"></a><span class="lineno"> 547</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 646 | <div class="line"><a name="l00548"></a><span class="lineno"> 548</span>  </div> |
| 647 | <div class="line"><a name="l00549"></a><span class="lineno"> 549</span>  <a class="code" href="structarmnn_1_1_comparison_descriptor.html">ComparisonDescriptor</a> descriptor(comparisonOperation);</div> |
| 648 | <div class="line"><a name="l00550"></a><span class="lineno"> 550</span>  </div> |
| 649 | <div class="line"><a name="l00551"></a><span class="lineno"> 551</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 650 | <div class="line"><a name="l00552"></a><span class="lineno"> 552</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 651 | <div class="line"><a name="l00553"></a><span class="lineno"> 553</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 652 | <div class="line"><a name="l00554"></a><span class="lineno"> 554</span>  {</div> |
| 653 | <div class="line"><a name="l00555"></a><span class="lineno"> 555</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 654 | <div class="line"><a name="l00556"></a><span class="lineno"> 556</span>  IsComparisonSupported,</div> |
| 655 | <div class="line"><a name="l00557"></a><span class="lineno"> 557</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 656 | <div class="line"><a name="l00558"></a><span class="lineno"> 558</span>  isSupported,</div> |
| 657 | <div class="line"><a name="l00559"></a><span class="lineno"> 559</span>  setBackend,</div> |
| 658 | <div class="line"><a name="l00560"></a><span class="lineno"> 560</span>  inputInfo0,</div> |
| 659 | <div class="line"><a name="l00561"></a><span class="lineno"> 561</span>  inputInfo1,</div> |
| 660 | <div class="line"><a name="l00562"></a><span class="lineno"> 562</span>  outputInfo,</div> |
| 661 | <div class="line"><a name="l00563"></a><span class="lineno"> 563</span>  descriptor);</div> |
| 662 | <div class="line"><a name="l00564"></a><span class="lineno"> 564</span>  };</div> |
| 663 | <div class="line"><a name="l00565"></a><span class="lineno"> 565</span>  </div> |
| 664 | <div class="line"><a name="l00566"></a><span class="lineno"> 566</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 665 | <div class="line"><a name="l00567"></a><span class="lineno"> 567</span>  {</div> |
| 666 | <div class="line"><a name="l00568"></a><span class="lineno"> 568</span>  validateFunc(outputInfo, isSupported);</div> |
| 667 | <div class="line"><a name="l00569"></a><span class="lineno"> 569</span>  }</div> |
| 668 | <div class="line"><a name="l00570"></a><span class="lineno"> 570</span>  <span class="keywordflow">else</span></div> |
| 669 | <div class="line"><a name="l00571"></a><span class="lineno"> 571</span>  {</div> |
| 670 | <div class="line"><a name="l00572"></a><span class="lineno"> 572</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 671 | <div class="line"><a name="l00573"></a><span class="lineno"> 573</span>  }</div> |
| 672 | <div class="line"><a name="l00574"></a><span class="lineno"> 574</span>  </div> |
| 673 | <div class="line"><a name="l00575"></a><span class="lineno"> 575</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 674 | <div class="line"><a name="l00576"></a><span class="lineno"> 576</span>  {</div> |
| 675 | <div class="line"><a name="l00577"></a><span class="lineno"> 577</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 676 | <div class="line"><a name="l00578"></a><span class="lineno"> 578</span>  }</div> |
| 677 | <div class="line"><a name="l00579"></a><span class="lineno"> 579</span>  </div> |
| 678 | <div class="line"><a name="l00580"></a><span class="lineno"> 580</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddComparisonLayer(descriptor);</div> |
| 679 | <div class="line"><a name="l00581"></a><span class="lineno"> 581</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 680 | <div class="line"><a name="l00582"></a><span class="lineno"> 582</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 681 | <div class="line"><a name="l00583"></a><span class="lineno"> 583</span>  </div> |
| 682 | <div class="line"><a name="l00584"></a><span class="lineno"> 584</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input0, input1, layer, data);</div> |
| 683 | <div class="line"><a name="l00585"></a><span class="lineno"> 585</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| 684 | <div class="line"><a name="l00586"></a><span class="lineno"> 586</span>  {</div> |
| 685 | <div class="line"><a name="l00587"></a><span class="lineno"> 587</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 686 | <div class="line"><a name="l00588"></a><span class="lineno"> 588</span>  }</div> |
| 687 | <div class="line"><a name="l00589"></a><span class="lineno"> 589</span>  </div> |
| 688 | <div class="line"><a name="l00590"></a><span class="lineno"> 590</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 689 | <div class="line"><a name="l00591"></a><span class="lineno"> 591</span>  {</div> |
| 690 | <div class="line"><a name="l00592"></a><span class="lineno"> 592</span>  input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 691 | <div class="line"><a name="l00593"></a><span class="lineno"> 593</span>  input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 692 | <div class="line"><a name="l00594"></a><span class="lineno"> 594</span>  }</div> |
| 693 | <div class="line"><a name="l00595"></a><span class="lineno"> 595</span>  </div> |
| 694 | <div class="line"><a name="l00596"></a><span class="lineno"> 596</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 695 | <div class="line"><a name="l00597"></a><span class="lineno"> 597</span> }</div> |
| 696 | <div class="line"><a name="l00598"></a><span class="lineno"> 598</span>  </div> |
| 697 | <div class="line"><a name="l00599"></a><span class="lineno"> 599</span>  </div> |
| 698 | <div class="line"><a name="l00600"></a><span class="lineno"> 600</span> <span class="keywordtype">bool</span> Converter::ConvertConcatenation(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 699 | <div class="line"><a name="l00601"></a><span class="lineno"> 601</span> {</div> |
| 700 | <div class="line"><a name="l00602"></a><span class="lineno"> 602</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConcatenation()"</span>;</div> |
| 701 | <div class="line"><a name="l00603"></a><span class="lineno"> 603</span>  </div> |
| 702 | <div class="line"><a name="l00604"></a><span class="lineno"> 604</span>  <span class="comment">// The first N (0..N-1) inputs are tensors. The Nth input is the concatenation axis.</span></div> |
| 703 | <div class="line"><a name="l00605"></a><span class="lineno"> 605</span>  <span class="keywordflow">if</span> (operation.inputs.size() <= 1)</div> |
| 704 | <div class="line"><a name="l00606"></a><span class="lineno"> 606</span>  {</div> |
| 705 | <div class="line"><a name="l00607"></a><span class="lineno"> 607</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has insufficient arguments"</span>, __func__);</div> |
| 706 | <div class="line"><a name="l00608"></a><span class="lineno"> 608</span>  }</div> |
| 707 | <div class="line"><a name="l00609"></a><span class="lineno"> 609</span>  </div> |
| 708 | <div class="line"><a name="l00610"></a><span class="lineno"> 610</span>  <span class="comment">// Get inputs and outputs</span></div> |
| 709 | <div class="line"><a name="l00611"></a><span class="lineno"> 611</span>  <span class="keyword">const</span> std::size_t numInputTensors = operation.inputs.size() - 1;</div> |
| 710 | <div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  </div> |
| 711 | <div class="line"><a name="l00613"></a><span class="lineno"> 613</span>  int32_t concatDim;</div> |
| 712 | <div class="line"><a name="l00614"></a><span class="lineno"> 614</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, numInputTensors, OperandType::INT32, concatDim, model, data))</div> |
| 713 | <div class="line"><a name="l00615"></a><span class="lineno"> 615</span>  {</div> |
| 714 | <div class="line"><a name="l00616"></a><span class="lineno"> 616</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 715 | <div class="line"><a name="l00617"></a><span class="lineno"> 617</span>  }</div> |
| 716 | <div class="line"><a name="l00618"></a><span class="lineno"> 618</span>  </div> |
| 717 | <div class="line"><a name="l00619"></a><span class="lineno"> 619</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 718 | <div class="line"><a name="l00620"></a><span class="lineno"> 620</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 719 | <div class="line"><a name="l00621"></a><span class="lineno"> 621</span>  {</div> |
| 720 | <div class="line"><a name="l00622"></a><span class="lineno"> 622</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has no outputs"</span>, __func__);</div> |
| 721 | <div class="line"><a name="l00623"></a><span class="lineno"> 623</span>  }</div> |
| 722 | <div class="line"><a name="l00624"></a><span class="lineno"> 624</span>  </div> |
| 723 | <div class="line"><a name="l00625"></a><span class="lineno"> 625</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 724 | <div class="line"><a name="l00626"></a><span class="lineno"> 626</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> outputShape = outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 725 | <div class="line"><a name="l00627"></a><span class="lineno"> 627</span>  <span class="keyword">const</span> <span class="keywordtype">bool</span> isDynamicTensor = <a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo);</div> |
| 726 | <div class="line"><a name="l00628"></a><span class="lineno"> 628</span>  <span class="comment">//</span></div> |
| 727 | <div class="line"><a name="l00629"></a><span class="lineno"> 629</span>  <span class="comment">// handle negative concat dims along the lines of tensorflow as described here:</span></div> |
| 728 | <div class="line"><a name="l00630"></a><span class="lineno"> 630</span>  <span class="comment">// https://www.tensorflow.org/api_docs/python/tf/concat</span></div> |
| 729 | <div class="line"><a name="l00631"></a><span class="lineno"> 631</span>  <span class="comment">// "negative axis refers to axis + rank(values)-th dimension"</span></div> |
| 730 | <div class="line"><a name="l00632"></a><span class="lineno"> 632</span>  <span class="comment">//</span></div> |
| 731 | <div class="line"><a name="l00633"></a><span class="lineno"> 633</span>  <span class="keywordflow">if</span> (concatDim < 0)</div> |
| 732 | <div class="line"><a name="l00634"></a><span class="lineno"> 634</span>  {</div> |
| 733 | <div class="line"><a name="l00635"></a><span class="lineno"> 635</span>  concatDim += outputShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 734 | <div class="line"><a name="l00636"></a><span class="lineno"> 636</span>  }</div> |
| 735 | <div class="line"><a name="l00637"></a><span class="lineno"> 637</span>  </div> |
| 736 | <div class="line"><a name="l00638"></a><span class="lineno"> 638</span>  <span class="keywordflow">if</span> (concatDim >= <span class="keyword">static_cast<</span>int32_t<span class="keyword">></span>(outputShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>()) || concatDim < 0)</div> |
| 737 | <div class="line"><a name="l00639"></a><span class="lineno"> 639</span>  {</div> |
| 738 | <div class="line"><a name="l00640"></a><span class="lineno"> 640</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid concat axis: %d"</span>, __func__, concatDim);</div> |
| 739 | <div class="line"><a name="l00641"></a><span class="lineno"> 641</span>  }</div> |
| 740 | <div class="line"><a name="l00642"></a><span class="lineno"> 642</span>  </div> |
| 741 | <div class="line"><a name="l00643"></a><span class="lineno"> 643</span>  std::vector<LayerInputHandle> inputHandles;</div> |
| 742 | <div class="line"><a name="l00644"></a><span class="lineno"> 644</span>  std::vector<armnn::TensorShape> inputShapes;</div> |
| 743 | <div class="line"><a name="l00645"></a><span class="lineno"> 645</span>  </div> |
| 744 | <div class="line"><a name="l00646"></a><span class="lineno"> 646</span>  inputHandles.reserve(numInputTensors);</div> |
| 745 | <div class="line"><a name="l00647"></a><span class="lineno"> 647</span>  inputShapes.reserve(numInputTensors);</div> |
| 746 | <div class="line"><a name="l00648"></a><span class="lineno"> 648</span>  </div> |
| 747 | <div class="line"><a name="l00649"></a><span class="lineno"> 649</span>  <span class="keywordtype">bool</span> inputsHaveBeenReshaped = <span class="keyword">false</span>;</div> |
| 748 | <div class="line"><a name="l00650"></a><span class="lineno"> 650</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> tensorDimensionsAdded = 0;</div> |
| 749 | <div class="line"><a name="l00651"></a><span class="lineno"> 651</span>  <span class="keywordflow">for</span> (uint32_t i = 0; i < numInputTensors; ++i)</div> |
| 750 | <div class="line"><a name="l00652"></a><span class="lineno"> 652</span>  {</div> |
| 751 | <div class="line"><a name="l00653"></a><span class="lineno"> 653</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* operand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, i, model);</div> |
| 752 | <div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  <span class="keywordflow">if</span> (!operand)</div> |
| 753 | <div class="line"><a name="l00655"></a><span class="lineno"> 655</span>  {</div> |
| 754 | <div class="line"><a name="l00656"></a><span class="lineno"> 656</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 755 | <div class="line"><a name="l00657"></a><span class="lineno"> 657</span>  }</div> |
| 756 | <div class="line"><a name="l00658"></a><span class="lineno"> 658</span>  </div> |
| 757 | <div class="line"><a name="l00659"></a><span class="lineno"> 659</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> operandInputHandle = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, i, model, data);</div> |
| 758 | <div class="line"><a name="l00660"></a><span class="lineno"> 660</span>  <span class="keywordflow">if</span> (!operandInputHandle.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 759 | <div class="line"><a name="l00661"></a><span class="lineno"> 661</span>  {</div> |
| 760 | <div class="line"><a name="l00662"></a><span class="lineno"> 662</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 761 | <div class="line"><a name="l00663"></a><span class="lineno"> 663</span>  }</div> |
| 762 | <div class="line"><a name="l00664"></a><span class="lineno"> 664</span>  </div> |
| 763 | <div class="line"><a name="l00665"></a><span class="lineno"> 665</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> operandShape = GetTensorShapeForOperand(*operand);</div> |
| 764 | <div class="line"><a name="l00666"></a><span class="lineno"> 666</span>  <span class="keywordflow">if</span> (operandShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() == 0)</div> |
| 765 | <div class="line"><a name="l00667"></a><span class="lineno"> 667</span>  {</div> |
| 766 | <div class="line"><a name="l00668"></a><span class="lineno"> 668</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operands with rank 0 are not supported"</span>, __func__);</div> |
| 767 | <div class="line"><a name="l00669"></a><span class="lineno"> 669</span>  }</div> |
| 768 | <div class="line"><a name="l00670"></a><span class="lineno"> 670</span>  </div> |
| 769 | <div class="line"><a name="l00671"></a><span class="lineno"> 671</span>  <span class="keywordflow">if</span> (RequiresReshape(operandShape))</div> |
| 770 | <div class="line"><a name="l00672"></a><span class="lineno"> 672</span>  {</div> |
| 771 | <div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  inputsHaveBeenReshaped = <span class="keyword">true</span>;</div> |
| 772 | <div class="line"><a name="l00674"></a><span class="lineno"> 674</span>  </div> |
| 773 | <div class="line"><a name="l00675"></a><span class="lineno"> 675</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> reshapeInfo = operandInputHandle.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 774 | <div class="line"><a name="l00676"></a><span class="lineno"> 676</span>  </div> |
| 775 | <div class="line"><a name="l00677"></a><span class="lineno"> 677</span>  <span class="comment">// Expand the tensor to three dimensions</span></div> |
| 776 | <div class="line"><a name="l00678"></a><span class="lineno"> 678</span>  <span class="keywordflow">if</span> (operandShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() == 2)</div> |
| 777 | <div class="line"><a name="l00679"></a><span class="lineno"> 679</span>  {</div> |
| 778 | <div class="line"><a name="l00680"></a><span class="lineno"> 680</span>  reshapeInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(<a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({1, operandShape[0], operandShape[1]}));</div> |
| 779 | <div class="line"><a name="l00681"></a><span class="lineno"> 681</span>  tensorDimensionsAdded = 1;</div> |
| 780 | <div class="line"><a name="l00682"></a><span class="lineno"> 682</span>  }</div> |
| 781 | <div class="line"><a name="l00683"></a><span class="lineno"> 683</span>  <span class="keywordflow">else</span></div> |
| 782 | <div class="line"><a name="l00684"></a><span class="lineno"> 684</span>  {</div> |
| 783 | <div class="line"><a name="l00685"></a><span class="lineno"> 685</span>  reshapeInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(<a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({1, 1, operandShape[0]}));</div> |
| 784 | <div class="line"><a name="l00686"></a><span class="lineno"> 686</span>  tensorDimensionsAdded = 2;</div> |
| 785 | <div class="line"><a name="l00687"></a><span class="lineno"> 687</span>  }</div> |
| 786 | <div class="line"><a name="l00688"></a><span class="lineno"> 688</span>  </div> |
| 787 | <div class="line"><a name="l00689"></a><span class="lineno"> 689</span>  <a class="code" href="structarmnn_1_1_reshape_descriptor.html">armnn::ReshapeDescriptor</a> reshapeDescriptor;</div> |
| 788 | <div class="line"><a name="l00690"></a><span class="lineno"> 690</span>  reshapeDescriptor.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a> = reshapeInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 789 | <div class="line"><a name="l00691"></a><span class="lineno"> 691</span>  </div> |
| 790 | <div class="line"><a name="l00692"></a><span class="lineno"> 692</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 791 | <div class="line"><a name="l00693"></a><span class="lineno"> 693</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendReshape;</div> |
| 792 | <div class="line"><a name="l00694"></a><span class="lineno"> 694</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 793 | <div class="line"><a name="l00695"></a><span class="lineno"> 695</span>  IsReshapeSupported,</div> |
| 794 | <div class="line"><a name="l00696"></a><span class="lineno"> 696</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 795 | <div class="line"><a name="l00697"></a><span class="lineno"> 697</span>  isSupported,</div> |
| 796 | <div class="line"><a name="l00698"></a><span class="lineno"> 698</span>  setBackendReshape,</div> |
| 797 | <div class="line"><a name="l00699"></a><span class="lineno"> 699</span>  operandInputHandle.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 798 | <div class="line"><a name="l00700"></a><span class="lineno"> 700</span>  reshapeInfo,</div> |
| 799 | <div class="line"><a name="l00701"></a><span class="lineno"> 701</span>  reshapeDescriptor);</div> |
| 800 | <div class="line"><a name="l00702"></a><span class="lineno"> 702</span>  </div> |
| 801 | <div class="line"><a name="l00703"></a><span class="lineno"> 703</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 802 | <div class="line"><a name="l00704"></a><span class="lineno"> 704</span>  {</div> |
| 803 | <div class="line"><a name="l00705"></a><span class="lineno"> 705</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 804 | <div class="line"><a name="l00706"></a><span class="lineno"> 706</span>  }</div> |
| 805 | <div class="line"><a name="l00707"></a><span class="lineno"> 707</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>& newReshape = AddReshapeLayer(*data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>, operandInputHandle, reshapeInfo);</div> |
| 806 | <div class="line"><a name="l00708"></a><span class="lineno"> 708</span>  newReshape.<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendReshape);</div> |
| 807 | <div class="line"><a name="l00709"></a><span class="lineno"> 709</span>  </div> |
| 808 | <div class="line"><a name="l00710"></a><span class="lineno"> 710</span>  <span class="comment">// Point to the reshape operation rather then the input operation</span></div> |
| 809 | <div class="line"><a name="l00711"></a><span class="lineno"> 711</span>  operandShape = reshapeInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 810 | <div class="line"><a name="l00712"></a><span class="lineno"> 712</span>  operandInputHandle = <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a>(<span class="keyword">true</span>, &newReshape.<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(0), reshapeInfo);</div> |
| 811 | <div class="line"><a name="l00713"></a><span class="lineno"> 713</span>  }</div> |
| 812 | <div class="line"><a name="l00714"></a><span class="lineno"> 714</span>  </div> |
| 813 | <div class="line"><a name="l00715"></a><span class="lineno"> 715</span>  inputShapes.emplace_back(operandShape);</div> |
| 814 | <div class="line"><a name="l00716"></a><span class="lineno"> 716</span>  inputHandles.emplace_back(operandInputHandle);</div> |
| 815 | <div class="line"><a name="l00717"></a><span class="lineno"> 717</span>  </div> |
| 816 | <div class="line"><a name="l00718"></a><span class="lineno"> 718</span>  <span class="keywordflow">if</span> (!inputHandles.back().IsValid())</div> |
| 817 | <div class="line"><a name="l00719"></a><span class="lineno"> 719</span>  {</div> |
| 818 | <div class="line"><a name="l00720"></a><span class="lineno"> 720</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 819 | <div class="line"><a name="l00721"></a><span class="lineno"> 721</span>  }</div> |
| 820 | <div class="line"><a name="l00722"></a><span class="lineno"> 722</span>  }</div> |
| 821 | <div class="line"><a name="l00723"></a><span class="lineno"> 723</span>  </div> |
| 822 | <div class="line"><a name="l00724"></a><span class="lineno"> 724</span>  <span class="keywordflow">if</span> (inputShapes.size() != inputHandles.size())</div> |
| 823 | <div class="line"><a name="l00725"></a><span class="lineno"> 725</span>  {</div> |
| 824 | <div class="line"><a name="l00726"></a><span class="lineno"> 726</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: invalid model input shapes size doesn't match input handles size: %i != %i"</span>, __func__,</div> |
| 825 | <div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  inputShapes.size(), inputHandles.size());</div> |
| 826 | <div class="line"><a name="l00728"></a><span class="lineno"> 728</span>  }</div> |
| 827 | <div class="line"><a name="l00729"></a><span class="lineno"> 729</span>  </div> |
| 828 | <div class="line"><a name="l00730"></a><span class="lineno"> 730</span>  <span class="keywordflow">if</span> (inputsHaveBeenReshaped)</div> |
| 829 | <div class="line"><a name="l00731"></a><span class="lineno"> 731</span>  {</div> |
| 830 | <div class="line"><a name="l00732"></a><span class="lineno"> 732</span>  <span class="comment">// Adjust the concatenation dimension by the amount of dimensions added (if any)</span></div> |
| 831 | <div class="line"><a name="l00733"></a><span class="lineno"> 733</span>  concatDim += tensorDimensionsAdded;</div> |
| 832 | <div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  </div> |
| 833 | <div class="line"><a name="l00735"></a><span class="lineno"> 735</span>  <span class="comment">// Add extra dimensions to the output shape to reflect the addition of the reshape layers</span></div> |
| 834 | <div class="line"><a name="l00736"></a><span class="lineno"> 736</span>  <span class="keywordflow">if</span> (tensorDimensionsAdded == 1)</div> |
| 835 | <div class="line"><a name="l00737"></a><span class="lineno"> 737</span>  {</div> |
| 836 | <div class="line"><a name="l00738"></a><span class="lineno"> 738</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 837 | <div class="line"><a name="l00739"></a><span class="lineno"> 739</span>  {</div> |
| 838 | <div class="line"><a name="l00740"></a><span class="lineno"> 740</span>  outputShape = <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({1, 0, 0}, {<span class="keyword">true</span>, <span class="keyword">false</span>, <span class="keyword">false</span>});</div> |
| 839 | <div class="line"><a name="l00741"></a><span class="lineno"> 741</span>  }</div> |
| 840 | <div class="line"><a name="l00742"></a><span class="lineno"> 742</span>  <span class="keywordflow">else</span></div> |
| 841 | <div class="line"><a name="l00743"></a><span class="lineno"> 743</span>  {</div> |
| 842 | <div class="line"><a name="l00744"></a><span class="lineno"> 744</span>  outputShape = <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({1, outputShape[0], outputShape[1]});</div> |
| 843 | <div class="line"><a name="l00745"></a><span class="lineno"> 745</span>  }</div> |
| 844 | <div class="line"><a name="l00746"></a><span class="lineno"> 746</span>  }</div> |
| 845 | <div class="line"><a name="l00747"></a><span class="lineno"> 747</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (tensorDimensionsAdded == 2)</div> |
| 846 | <div class="line"><a name="l00748"></a><span class="lineno"> 748</span>  {</div> |
| 847 | <div class="line"><a name="l00749"></a><span class="lineno"> 749</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 848 | <div class="line"><a name="l00750"></a><span class="lineno"> 750</span>  {</div> |
| 849 | <div class="line"><a name="l00751"></a><span class="lineno"> 751</span>  outputShape = <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({1, 1, 0}, {<span class="keyword">true</span>, <span class="keyword">true</span>, <span class="keyword">false</span>});</div> |
| 850 | <div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  }</div> |
| 851 | <div class="line"><a name="l00753"></a><span class="lineno"> 753</span>  <span class="keywordflow">else</span></div> |
| 852 | <div class="line"><a name="l00754"></a><span class="lineno"> 754</span>  {</div> |
| 853 | <div class="line"><a name="l00755"></a><span class="lineno"> 755</span>  outputShape = <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({1, 1, outputShape[0]});</div> |
| 854 | <div class="line"><a name="l00756"></a><span class="lineno"> 756</span>  }</div> |
| 855 | <div class="line"><a name="l00757"></a><span class="lineno"> 757</span>  }</div> |
| 856 | <div class="line"><a name="l00758"></a><span class="lineno"> 758</span>  }</div> |
| 857 | <div class="line"><a name="l00759"></a><span class="lineno"> 759</span>  </div> |
| 858 | <div class="line"><a name="l00760"></a><span class="lineno"> 760</span>  <span class="comment">// Check if permutations is required and get the pair of permutations required for the concatenation.</span></div> |
| 859 | <div class="line"><a name="l00761"></a><span class="lineno"> 761</span>  <span class="comment">// Permutation is required when the concat dimension is 2 for a 4D tensor or 1 for a 3D tensor.</span></div> |
| 860 | <div class="line"><a name="l00762"></a><span class="lineno"> 762</span>  std::pair<armnn::PermutationVector, armnn::PermutationVector> permutationPair =</div> |
| 861 | <div class="line"><a name="l00763"></a><span class="lineno"> 763</span>  std::make_pair(IdentityPermutation4D, IdentityPermutation4D);</div> |
| 862 | <div class="line"><a name="l00764"></a><span class="lineno"> 764</span>  <span class="keywordtype">bool</span> needPermute = CreateConcatPermutationParameters(inputShapes[0].GetNumDimensions(),</div> |
| 863 | <div class="line"><a name="l00765"></a><span class="lineno"> 765</span>  concatDim,</div> |
| 864 | <div class="line"><a name="l00766"></a><span class="lineno"> 766</span>  permutationPair);</div> |
| 865 | <div class="line"><a name="l00767"></a><span class="lineno"> 767</span>  </div> |
| 866 | <div class="line"><a name="l00768"></a><span class="lineno"> 768</span>  <span class="comment">// Only relevant to static tensors as dynamic output tensors will be transposed as a result of inferring from input</span></div> |
| 867 | <div class="line"><a name="l00769"></a><span class="lineno"> 769</span>  <span class="keywordflow">if</span> (!isDynamicTensor)</div> |
| 868 | <div class="line"><a name="l00770"></a><span class="lineno"> 770</span>  {</div> |
| 869 | <div class="line"><a name="l00771"></a><span class="lineno"> 771</span>  <span class="keywordflow">if</span> (needPermute)</div> |
| 870 | <div class="line"><a name="l00772"></a><span class="lineno"> 772</span>  {</div> |
| 871 | <div class="line"><a name="l00773"></a><span class="lineno"> 773</span>  outputShape = <a class="code" href="namespacearmnn_utils.html#a428a9a6ffdf0e8d723b50c038c56c336">armnnUtils::TransposeTensorShape</a>(outputShape, permutationPair.first);</div> |
| 872 | <div class="line"><a name="l00774"></a><span class="lineno"> 774</span>  }</div> |
| 873 | <div class="line"><a name="l00775"></a><span class="lineno"> 775</span>  </div> |
| 874 | <div class="line"><a name="l00776"></a><span class="lineno"> 776</span>  outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(outputShape);</div> |
| 875 | <div class="line"><a name="l00777"></a><span class="lineno"> 777</span>  }</div> |
| 876 | <div class="line"><a name="l00778"></a><span class="lineno"> 778</span>  <span class="comment">// this is no-op for identity swizzles, otherwise it replaces both</span></div> |
| 877 | <div class="line"><a name="l00779"></a><span class="lineno"> 779</span>  <span class="comment">// the handles and shapes with the swizzled layer output handles and shapes</span></div> |
| 878 | <div class="line"><a name="l00780"></a><span class="lineno"> 780</span>  <span class="keywordflow">if</span> (!TransposeInputTensors(data, inputHandles, inputShapes, permutationPair.first))</div> |
| 879 | <div class="line"><a name="l00781"></a><span class="lineno"> 781</span>  {</div> |
| 880 | <div class="line"><a name="l00782"></a><span class="lineno"> 782</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 881 | <div class="line"><a name="l00783"></a><span class="lineno"> 783</span>  }</div> |
| 882 | <div class="line"><a name="l00784"></a><span class="lineno"> 784</span>  </div> |
| 883 | <div class="line"><a name="l00785"></a><span class="lineno"> 785</span>  <span class="comment">// Create an armnn concat layer descriptor - this will also perform validation on the input shapes</span></div> |
| 884 | <div class="line"><a name="l00786"></a><span class="lineno"> 786</span>  <a class="code" href="structarmnn_1_1_origins_descriptor.html">armnn::OriginsDescriptor</a> concatDescriptor;</div> |
| 885 | <div class="line"><a name="l00787"></a><span class="lineno"> 787</span>  </div> |
| 886 | <div class="line"><a name="l00788"></a><span class="lineno"> 788</span>  <span class="keywordflow">try</span></div> |
| 887 | <div class="line"><a name="l00789"></a><span class="lineno"> 789</span>  {</div> |
| 888 | <div class="line"><a name="l00790"></a><span class="lineno"> 790</span>  <span class="comment">// The concat descriptor is always created across the only supported concat dimension</span></div> |
| 889 | <div class="line"><a name="l00791"></a><span class="lineno"> 791</span>  <span class="comment">// which is 0, 1 or 3 for a 4-D tensor, or 0 or 2 for a 3-D tensor.</span></div> |
| 890 | <div class="line"><a name="l00792"></a><span class="lineno"> 792</span>  concatDescriptor = <a class="code" href="namespacearmnn.html#a733ae6b70d0bfa43433c3e7606992328">armnn::CreateDescriptorForConcatenation</a>(inputShapes.begin(),</div> |
| 891 | <div class="line"><a name="l00793"></a><span class="lineno"> 793</span>  inputShapes.end(),</div> |
| 892 | <div class="line"><a name="l00794"></a><span class="lineno"> 794</span>  concatDim);</div> |
| 893 | <div class="line"><a name="l00795"></a><span class="lineno"> 795</span>  } <span class="keywordflow">catch</span> (std::exception& error)</div> |
| 894 | <div class="line"><a name="l00796"></a><span class="lineno"> 796</span>  {</div> |
| 895 | <div class="line"><a name="l00797"></a><span class="lineno"> 797</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Error preparing concat descriptor. %s"</span>, __func__, <a class="code" href="namespacearmnn.html#a4dc0adc6737b5944e7671bee71788407acb5e100e5a9a3e7f6d1fd97512215282">error</a>.what());</div> |
| 896 | <div class="line"><a name="l00798"></a><span class="lineno"> 798</span>  }</div> |
| 897 | <div class="line"><a name="l00799"></a><span class="lineno"> 799</span>  </div> |
| 898 | <div class="line"><a name="l00800"></a><span class="lineno"> 800</span>  <span class="comment">// Validate the output shape is correct given the input shapes based on the</span></div> |
| 899 | <div class="line"><a name="l00801"></a><span class="lineno"> 801</span>  <span class="comment">// only valid concat dimension which is 0, 1 or 3 for a 4-D tensor, or 0 or 2 for a 3-D tensor.</span></div> |
| 900 | <div class="line"><a name="l00802"></a><span class="lineno"> 802</span>  <span class="keywordflow">if</span> (!isDynamicTensor)</div> |
| 901 | <div class="line"><a name="l00803"></a><span class="lineno"> 803</span>  {</div> |
| 902 | <div class="line"><a name="l00804"></a><span class="lineno"> 804</span>  <span class="keywordflow">if</span> (!ValidateConcatOutputShape(inputShapes, outputShape, concatDim))</div> |
| 903 | <div class="line"><a name="l00805"></a><span class="lineno"> 805</span>  {</div> |
| 904 | <div class="line"><a name="l00806"></a><span class="lineno"> 806</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Error validating the output shape for concat"</span>, __func__);</div> |
| 905 | <div class="line"><a name="l00807"></a><span class="lineno"> 807</span>  }</div> |
| 906 | <div class="line"><a name="l00808"></a><span class="lineno"> 808</span>  }</div> |
| 907 | <div class="line"><a name="l00809"></a><span class="lineno"> 809</span>  </div> |
| 908 | <div class="line"><a name="l00810"></a><span class="lineno"> 810</span>  std::vector<const armnn::TensorInfo*> inputTensorInfos;</div> |
| 909 | <div class="line"><a name="l00811"></a><span class="lineno"> 811</span>  std::transform(inputHandles.begin(), inputHandles.end(), std::back_inserter(inputTensorInfos),</div> |
| 910 | <div class="line"><a name="l00812"></a><span class="lineno"> 812</span>  [](<span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a>& h)->const <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>*{ return &h.GetTensorInfo(); });</div> |
| 911 | <div class="line"><a name="l00813"></a><span class="lineno"> 813</span>  </div> |
| 912 | <div class="line"><a name="l00814"></a><span class="lineno"> 814</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 913 | <div class="line"><a name="l00815"></a><span class="lineno"> 815</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendConcat;</div> |
| 914 | <div class="line"><a name="l00816"></a><span class="lineno"> 816</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported){</div> |
| 915 | <div class="line"><a name="l00817"></a><span class="lineno"> 817</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 916 | <div class="line"><a name="l00818"></a><span class="lineno"> 818</span>  IsConcatSupported,</div> |
| 917 | <div class="line"><a name="l00819"></a><span class="lineno"> 819</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 918 | <div class="line"><a name="l00820"></a><span class="lineno"> 820</span>  isSupported,</div> |
| 919 | <div class="line"><a name="l00821"></a><span class="lineno"> 821</span>  setBackendConcat,</div> |
| 920 | <div class="line"><a name="l00822"></a><span class="lineno"> 822</span>  inputTensorInfos,</div> |
| 921 | <div class="line"><a name="l00823"></a><span class="lineno"> 823</span>  outputInfo,</div> |
| 922 | <div class="line"><a name="l00824"></a><span class="lineno"> 824</span>  concatDescriptor);</div> |
| 923 | <div class="line"><a name="l00825"></a><span class="lineno"> 825</span>  };</div> |
| 924 | <div class="line"><a name="l00826"></a><span class="lineno"> 826</span>  </div> |
| 925 | <div class="line"><a name="l00827"></a><span class="lineno"> 827</span>  <span class="keywordflow">if</span> (!isDynamicTensor)</div> |
| 926 | <div class="line"><a name="l00828"></a><span class="lineno"> 828</span>  {</div> |
| 927 | <div class="line"><a name="l00829"></a><span class="lineno"> 829</span>  validateFunc(outputInfo, isSupported);</div> |
| 928 | <div class="line"><a name="l00830"></a><span class="lineno"> 830</span>  }</div> |
| 929 | <div class="line"><a name="l00831"></a><span class="lineno"> 831</span>  <span class="keywordflow">else</span></div> |
| 930 | <div class="line"><a name="l00832"></a><span class="lineno"> 832</span>  {</div> |
| 931 | <div class="line"><a name="l00833"></a><span class="lineno"> 833</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 932 | <div class="line"><a name="l00834"></a><span class="lineno"> 834</span>  }</div> |
| 933 | <div class="line"><a name="l00835"></a><span class="lineno"> 835</span>  </div> |
| 934 | <div class="line"><a name="l00836"></a><span class="lineno"> 836</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 935 | <div class="line"><a name="l00837"></a><span class="lineno"> 837</span>  {</div> |
| 936 | <div class="line"><a name="l00838"></a><span class="lineno"> 838</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 937 | <div class="line"><a name="l00839"></a><span class="lineno"> 839</span>  }</div> |
| 938 | <div class="line"><a name="l00840"></a><span class="lineno"> 840</span>  </div> |
| 939 | <div class="line"><a name="l00841"></a><span class="lineno"> 841</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddConcatLayer(concatDescriptor);</div> |
| 940 | <div class="line"><a name="l00842"></a><span class="lineno"> 842</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendConcat);</div> |
| 941 | <div class="line"><a name="l00843"></a><span class="lineno"> 843</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 942 | <div class="line"><a name="l00844"></a><span class="lineno"> 844</span>  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#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(outputInfo);</div> |
| 943 | <div class="line"><a name="l00845"></a><span class="lineno"> 845</span>  <span class="comment">// Connect inputs to the layer</span></div> |
| 944 | <div class="line"><a name="l00846"></a><span class="lineno"> 846</span>  <span class="keyword">const</span> <span class="keywordtype">int</span> numInputSlots = layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a9c2cba04b6d7ace4fc2a2436b82a5a63">GetNumInputSlots</a>();</div> |
| 945 | <div class="line"><a name="l00847"></a><span class="lineno"> 847</span>  assert(<span class="keyword">static_cast<</span>std::size_t<span class="keyword">></span>(numInputSlots) == inputHandles.size());</div> |
| 946 | <div class="line"><a name="l00848"></a><span class="lineno"> 848</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < numInputSlots; ++i)</div> |
| 947 | <div class="line"><a name="l00849"></a><span class="lineno"> 849</span>  {</div> |
| 948 | <div class="line"><a name="l00850"></a><span class="lineno"> 850</span>  <span class="comment">// connect the input directly to the merge (concat) layer</span></div> |
| 949 | <div class="line"><a name="l00851"></a><span class="lineno"> 851</span>  inputHandles[<span class="keyword">static_cast<</span><span class="keywordtype">unsigned</span> <span class="keywordtype">int</span><span class="keyword">></span>(i)].<a class="code" href="_test_utils_8cpp.html#a0b295acb179f15eb3fb862b32008f782">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(i));</div> |
| 950 | <div class="line"><a name="l00852"></a><span class="lineno"> 852</span>  }</div> |
| 951 | <div class="line"><a name="l00853"></a><span class="lineno"> 853</span>  </div> |
| 952 | <div class="line"><a name="l00854"></a><span class="lineno"> 854</span>  <span class="comment">// Transpose the output shape</span></div> |
| 953 | <div class="line"><a name="l00855"></a><span class="lineno"> 855</span>  <span class="keyword">auto</span> transposeOutputShape = [&](){</div> |
| 954 | <div class="line"><a name="l00856"></a><span class="lineno"> 856</span>  <a class="code" href="structarmnn_1_1_transpose_descriptor.html">armnn::TransposeDescriptor</a> transposeDesc;</div> |
| 955 | <div class="line"><a name="l00857"></a><span class="lineno"> 857</span>  transposeDesc.<a class="code" href="structarmnn_1_1_transpose_descriptor.html#a14433af2b223695b40d8c8f8ba2ebb8f">m_DimMappings</a> = permutationPair.second;</div> |
| 956 | <div class="line"><a name="l00858"></a><span class="lineno"> 858</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> inputTransposeInfo = 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#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>();</div> |
| 957 | <div class="line"><a name="l00859"></a><span class="lineno"> 859</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> outputTransposeInfo = <a class="code" href="namespacearmnn_utils.html#a428a9a6ffdf0e8d723b50c038c56c336">armnnUtils::TransposeTensorShape</a>(inputTransposeInfo,</div> |
| 958 | <div class="line"><a name="l00860"></a><span class="lineno"> 860</span>  permutationPair.second);</div> |
| 959 | <div class="line"><a name="l00861"></a><span class="lineno"> 861</span>  isSupported = <span class="keyword">false</span>;</div> |
| 960 | <div class="line"><a name="l00862"></a><span class="lineno"> 862</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendTranspose;</div> |
| 961 | <div class="line"><a name="l00863"></a><span class="lineno"> 863</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 962 | <div class="line"><a name="l00864"></a><span class="lineno"> 864</span>  IsTransposeSupported,</div> |
| 963 | <div class="line"><a name="l00865"></a><span class="lineno"> 865</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 964 | <div class="line"><a name="l00866"></a><span class="lineno"> 866</span>  isSupported,</div> |
| 965 | <div class="line"><a name="l00867"></a><span class="lineno"> 867</span>  setBackendTranspose,</div> |
| 966 | <div class="line"><a name="l00868"></a><span class="lineno"> 868</span>  inputTransposeInfo,</div> |
| 967 | <div class="line"><a name="l00869"></a><span class="lineno"> 869</span>  outputTransposeInfo,</div> |
| 968 | <div class="line"><a name="l00870"></a><span class="lineno"> 870</span>  transposeDesc);</div> |
| 969 | <div class="line"><a name="l00871"></a><span class="lineno"> 871</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 970 | <div class="line"><a name="l00872"></a><span class="lineno"> 872</span>  {</div> |
| 971 | <div class="line"><a name="l00873"></a><span class="lineno"> 873</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 972 | <div class="line"><a name="l00874"></a><span class="lineno"> 874</span>  }</div> |
| 973 | <div class="line"><a name="l00875"></a><span class="lineno"> 875</span>  <span class="comment">// Add permutation layer and connect the output to it, the permutation becomes the output layer</span></div> |
| 974 | <div class="line"><a name="l00876"></a><span class="lineno"> 876</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>& deswizzleLayer = AddTransposeLayer(*data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>, layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(0),</div> |
| 975 | <div class="line"><a name="l00877"></a><span class="lineno"> 877</span>  permutationPair.second);</div> |
| 976 | <div class="line"><a name="l00878"></a><span class="lineno"> 878</span>  deswizzleLayer.<a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendTranspose);</div> |
| 977 | <div class="line"><a name="l00879"></a><span class="lineno"> 879</span>  layer = &deswizzleLayer;</div> |
| 978 | <div class="line"><a name="l00880"></a><span class="lineno"> 880</span>  </div> |
| 979 | <div class="line"><a name="l00881"></a><span class="lineno"> 881</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| 980 | <div class="line"><a name="l00882"></a><span class="lineno"> 882</span>  };</div> |
| 981 | <div class="line"><a name="l00883"></a><span class="lineno"> 883</span>  </div> |
| 982 | <div class="line"><a name="l00884"></a><span class="lineno"> 884</span>  <span class="keywordflow">if</span> (needPermute && !isDynamicTensor)</div> |
| 983 | <div class="line"><a name="l00885"></a><span class="lineno"> 885</span>  {</div> |
| 984 | <div class="line"><a name="l00886"></a><span class="lineno"> 886</span>  transposeOutputShape();</div> |
| 985 | <div class="line"><a name="l00887"></a><span class="lineno"> 887</span>  }</div> |
| 986 | <div class="line"><a name="l00888"></a><span class="lineno"> 888</span>  </div> |
| 987 | <div class="line"><a name="l00889"></a><span class="lineno"> 889</span>  <span class="keywordflow">if</span> (inputsHaveBeenReshaped)</div> |
| 988 | <div class="line"><a name="l00890"></a><span class="lineno"> 890</span>  {</div> |
| 989 | <div class="line"><a name="l00891"></a><span class="lineno"> 891</span>  <span class="keywordflow">if</span> (isDynamicTensor)</div> |
| 990 | <div class="line"><a name="l00892"></a><span class="lineno"> 892</span>  {</div> |
| 991 | <div class="line"><a name="l00893"></a><span class="lineno"> 893</span>  <span class="comment">// Infer the output shapes of concat if outputs are type 1 dynamic</span></div> |
| 992 | <div class="line"><a name="l00894"></a><span class="lineno"> 894</span>  <span class="keywordflow">if</span> (!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#ab4e57caf059794897b17266090ebcb39">IsTensorInfoSet</a>())</div> |
| 993 | <div class="line"><a name="l00895"></a><span class="lineno"> 895</span>  {</div> |
| 994 | <div class="line"><a name="l00896"></a><span class="lineno"> 896</span>  <span class="keywordflow">throw</span> <a class="code" href="classarmnn_1_1_exception.html">armnn::Exception</a>(</div> |
| 995 | <div class="line"><a name="l00897"></a><span class="lineno"> 897</span>  <span class="stringliteral">"tensor info is not set on output slot, cannot process dynamic tensor after input reshape"</span>);</div> |
| 996 | <div class="line"><a name="l00898"></a><span class="lineno"> 898</span>  }</div> |
| 997 | <div class="line"><a name="l00899"></a><span class="lineno"> 899</span>  <span class="keywordflow">if</span> (!ValidateConcatOutputShape(inputShapes,</div> |
| 998 | <div class="line"><a name="l00900"></a><span class="lineno"> 900</span>  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#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>().<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>(),</div> |
| 999 | <div class="line"><a name="l00901"></a><span class="lineno"> 901</span>  concatDim))</div> |
| 1000 | <div class="line"><a name="l00902"></a><span class="lineno"> 902</span>  {</div> |
| 1001 | <div class="line"><a name="l00903"></a><span class="lineno"> 903</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Error validating the output shape for concat"</span>, __func__);</div> |
| 1002 | <div class="line"><a name="l00904"></a><span class="lineno"> 904</span>  }</div> |
| 1003 | <div class="line"><a name="l00905"></a><span class="lineno"> 905</span>  transposeOutputShape();</div> |
| 1004 | <div class="line"><a name="l00906"></a><span class="lineno"> 906</span>  }</div> |
| 1005 | <div class="line"><a name="l00907"></a><span class="lineno"> 907</span>  </div> |
| 1006 | <div class="line"><a name="l00908"></a><span class="lineno"> 908</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> afterConcatInfo = 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#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>();</div> |
| 1007 | <div class="line"><a name="l00909"></a><span class="lineno"> 909</span>  <span class="comment">// Undo the reshape knowing the amount of dimensions added</span></div> |
| 1008 | <div class="line"><a name="l00910"></a><span class="lineno"> 910</span>  <span class="keywordflow">if</span> (tensorDimensionsAdded == 1)</div> |
| 1009 | <div class="line"><a name="l00911"></a><span class="lineno"> 911</span>  {</div> |
| 1010 | <div class="line"><a name="l00912"></a><span class="lineno"> 912</span>  afterConcatInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(</div> |
| 1011 | <div class="line"><a name="l00913"></a><span class="lineno"> 913</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({afterConcatInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[1], afterConcatInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[2]}));</div> |
| 1012 | <div class="line"><a name="l00914"></a><span class="lineno"> 914</span>  }</div> |
| 1013 | <div class="line"><a name="l00915"></a><span class="lineno"> 915</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (tensorDimensionsAdded == 2)</div> |
| 1014 | <div class="line"><a name="l00916"></a><span class="lineno"> 916</span>  {</div> |
| 1015 | <div class="line"><a name="l00917"></a><span class="lineno"> 917</span>  afterConcatInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(<a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>({afterConcatInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[2]}));</div> |
| 1016 | <div class="line"><a name="l00918"></a><span class="lineno"> 918</span>  }</div> |
| 1017 | <div class="line"><a name="l00919"></a><span class="lineno"> 919</span>  </div> |
| 1018 | <div class="line"><a name="l00920"></a><span class="lineno"> 920</span>  <a class="code" href="structarmnn_1_1_reshape_descriptor.html">armnn::ReshapeDescriptor</a> reshapeDescriptor;</div> |
| 1019 | <div class="line"><a name="l00921"></a><span class="lineno"> 921</span>  reshapeDescriptor.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a> = afterConcatInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 1020 | <div class="line"><a name="l00922"></a><span class="lineno"> 922</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> concatInfo = 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#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>();</div> |
| 1021 | <div class="line"><a name="l00923"></a><span class="lineno"> 923</span>  </div> |
| 1022 | <div class="line"><a name="l00924"></a><span class="lineno"> 924</span>  isSupported = <span class="keyword">false</span>;</div> |
| 1023 | <div class="line"><a name="l00925"></a><span class="lineno"> 925</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendReshape2;</div> |
| 1024 | <div class="line"><a name="l00926"></a><span class="lineno"> 926</span>  <span class="keyword">auto</span> validateReshapeFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& afterConcatInfo, <span class="keywordtype">bool</span>& isSupported){</div> |
| 1025 | <div class="line"><a name="l00927"></a><span class="lineno"> 927</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1026 | <div class="line"><a name="l00928"></a><span class="lineno"> 928</span>  IsReshapeSupported,</div> |
| 1027 | <div class="line"><a name="l00929"></a><span class="lineno"> 929</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1028 | <div class="line"><a name="l00930"></a><span class="lineno"> 930</span>  isSupported,</div> |
| 1029 | <div class="line"><a name="l00931"></a><span class="lineno"> 931</span>  setBackendReshape2,</div> |
| 1030 | <div class="line"><a name="l00932"></a><span class="lineno"> 932</span>  concatInfo,</div> |
| 1031 | <div class="line"><a name="l00933"></a><span class="lineno"> 933</span>  afterConcatInfo,</div> |
| 1032 | <div class="line"><a name="l00934"></a><span class="lineno"> 934</span>  reshapeDescriptor);</div> |
| 1033 | <div class="line"><a name="l00935"></a><span class="lineno"> 935</span>  };</div> |
| 1034 | <div class="line"><a name="l00936"></a><span class="lineno"> 936</span>  </div> |
| 1035 | <div class="line"><a name="l00937"></a><span class="lineno"> 937</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(afterConcatInfo))</div> |
| 1036 | <div class="line"><a name="l00938"></a><span class="lineno"> 938</span>  {</div> |
| 1037 | <div class="line"><a name="l00939"></a><span class="lineno"> 939</span>  validateReshapeFunc(afterConcatInfo, isSupported);</div> |
| 1038 | <div class="line"><a name="l00940"></a><span class="lineno"> 940</span>  }</div> |
| 1039 | <div class="line"><a name="l00941"></a><span class="lineno"> 941</span>  <span class="keywordflow">else</span></div> |
| 1040 | <div class="line"><a name="l00942"></a><span class="lineno"> 942</span>  {</div> |
| 1041 | <div class="line"><a name="l00943"></a><span class="lineno"> 943</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1042 | <div class="line"><a name="l00944"></a><span class="lineno"> 944</span>  }</div> |
| 1043 | <div class="line"><a name="l00945"></a><span class="lineno"> 945</span>  </div> |
| 1044 | <div class="line"><a name="l00946"></a><span class="lineno"> 946</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1045 | <div class="line"><a name="l00947"></a><span class="lineno"> 947</span>  {</div> |
| 1046 | <div class="line"><a name="l00948"></a><span class="lineno"> 948</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1047 | <div class="line"><a name="l00949"></a><span class="lineno"> 949</span>  }</div> |
| 1048 | <div class="line"><a name="l00950"></a><span class="lineno"> 950</span>  layer = &AddReshapeLayer(*data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>, layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(0), afterConcatInfo);</div> |
| 1049 | <div class="line"><a name="l00951"></a><span class="lineno"> 951</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendReshape2);</div> |
| 1050 | <div class="line"><a name="l00952"></a><span class="lineno"> 952</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation,</div> |
| 1051 | <div class="line"><a name="l00953"></a><span class="lineno"> 953</span>  0,</div> |
| 1052 | <div class="line"><a name="l00954"></a><span class="lineno"> 954</span>  *layer,</div> |
| 1053 | <div class="line"><a name="l00955"></a><span class="lineno"> 955</span>  model,</div> |
| 1054 | <div class="line"><a name="l00956"></a><span class="lineno"> 956</span>  data,</div> |
| 1055 | <div class="line"><a name="l00957"></a><span class="lineno"> 957</span>  <span class="keyword">nullptr</span>,</div> |
| 1056 | <div class="line"><a name="l00958"></a><span class="lineno"> 958</span>  validateReshapeFunc);</div> |
| 1057 | <div class="line"><a name="l00959"></a><span class="lineno"> 959</span>  }</div> |
| 1058 | <div class="line"><a name="l00960"></a><span class="lineno"> 960</span>  </div> |
| 1059 | <div class="line"><a name="l00961"></a><span class="lineno"> 961</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 1060 | <div class="line"><a name="l00962"></a><span class="lineno"> 962</span> }</div> |
| 1061 | <div class="line"><a name="l00963"></a><span class="lineno"> 963</span>  </div> |
| 1062 | <div class="line"><a name="l00964"></a><span class="lineno"> 964</span> <span class="keywordtype">bool</span> Converter::ConvertConv2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1063 | <div class="line"><a name="l00965"></a><span class="lineno"> 965</span> {</div> |
| 1064 | <div class="line"><a name="l00966"></a><span class="lineno"> 966</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConv2d()"</span>;</div> |
| 1065 | <div class="line"><a name="l00967"></a><span class="lineno"> 967</span>  </div> |
| 1066 | <div class="line"><a name="l00968"></a><span class="lineno"> 968</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1067 | <div class="line"><a name="l00969"></a><span class="lineno"> 969</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1068 | <div class="line"><a name="l00970"></a><span class="lineno"> 970</span>  {</div> |
| 1069 | <div class="line"><a name="l00971"></a><span class="lineno"> 971</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1070 | <div class="line"><a name="l00972"></a><span class="lineno"> 972</span>  }</div> |
| 1071 | <div class="line"><a name="l00973"></a><span class="lineno"> 973</span>  </div> |
| 1072 | <div class="line"><a name="l00974"></a><span class="lineno"> 974</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1073 | <div class="line"><a name="l00975"></a><span class="lineno"> 975</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1074 | <div class="line"><a name="l00976"></a><span class="lineno"> 976</span>  {</div> |
| 1075 | <div class="line"><a name="l00977"></a><span class="lineno"> 977</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 1076 | <div class="line"><a name="l00978"></a><span class="lineno"> 978</span>  }</div> |
| 1077 | <div class="line"><a name="l00979"></a><span class="lineno"> 979</span>  </div> |
| 1078 | <div class="line"><a name="l00980"></a><span class="lineno"> 980</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1079 | <div class="line"><a name="l00981"></a><span class="lineno"> 981</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1080 | <div class="line"><a name="l00982"></a><span class="lineno"> 982</span>  </div> |
| 1081 | <div class="line"><a name="l00983"></a><span class="lineno"> 983</span>  <a class="code" href="structarmnn_1_1_convolution2d_descriptor.html">Convolution2dDescriptor</a> desc;</div> |
| 1082 | <div class="line"><a name="l00984"></a><span class="lineno"> 984</span>  desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = DataLayout::NHWC;</div> |
| 1083 | <div class="line"><a name="l00985"></a><span class="lineno"> 985</span>  </div> |
| 1084 | <div class="line"><a name="l00986"></a><span class="lineno"> 986</span>  <span class="comment">// Determine whether padding is implicit or explicit</span></div> |
| 1085 | <div class="line"><a name="l00987"></a><span class="lineno"> 987</span>  <span class="keywordtype">bool</span> implicitPadding = operation.inputs.size() == 7</div> |
| 1086 | <div class="line"><a name="l00988"></a><span class="lineno"> 988</span>  || (operation.inputs.size() >= 8</div> |
| 1087 | <div class="line"><a name="l00989"></a><span class="lineno"> 989</span>  && <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 7, model)->type == OperandType::BOOL);</div> |
| 1088 | <div class="line"><a name="l00990"></a><span class="lineno"> 990</span>  </div> |
| 1089 | <div class="line"><a name="l00991"></a><span class="lineno"> 991</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| 1090 | <div class="line"><a name="l00992"></a><span class="lineno"> 992</span>  {</div> |
| 1091 | <div class="line"><a name="l00993"></a><span class="lineno"> 993</span>  desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 7, model, data);</div> |
| 1092 | <div class="line"><a name="l00994"></a><span class="lineno"> 994</span>  }</div> |
| 1093 | <div class="line"><a name="l00995"></a><span class="lineno"> 995</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operation.inputs.size() >= 10)</div> |
| 1094 | <div class="line"><a name="l00996"></a><span class="lineno"> 996</span>  {</div> |
| 1095 | <div class="line"><a name="l00997"></a><span class="lineno"> 997</span>  desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 10, model, data);</div> |
| 1096 | <div class="line"><a name="l00998"></a><span class="lineno"> 998</span>  }</div> |
| 1097 | <div class="line"><a name="l00999"></a><span class="lineno"> 999</span>  </div> |
| 1098 | <div class="line"><a name="l01000"></a><span class="lineno"> 1000</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_permutation_vector.html">PermutationVector</a> OHWIToOIHW = {0, 2, 3, 1};</div> |
| 1099 | <div class="line"><a name="l01001"></a><span class="lineno"> 1001</span>  </div> |
| 1100 | <div class="line"><a name="l01002"></a><span class="lineno"> 1002</span>  <span class="comment">// ArmNN does not currently support non-fixed weights or bias</span></div> |
| 1101 | <div class="line"><a name="l01003"></a><span class="lineno"> 1003</span>  <span class="comment">// The NNAPI filter is always OHWI [depth_out, filter_height, filter_width, depth_in] but ArmNN expects the</span></div> |
| 1102 | <div class="line"><a name="l01004"></a><span class="lineno"> 1004</span>  <span class="comment">// filter's height and width indices to match the input's height and width indices so we permute it to OIHW if</span></div> |
| 1103 | <div class="line"><a name="l01005"></a><span class="lineno"> 1005</span>  <span class="comment">// the DataLayout is NCHW</span></div> |
| 1104 | <div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>  </div> |
| 1105 | <div class="line"><a name="l01007"></a><span class="lineno"> 1007</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ae780aad3075e52e802f9c0ed5d88e570">IsWeightsValid</a>(operation, 1, model, <span class="keyword">false</span>) && desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> == DataLayout::NCHW)</div> |
| 1106 | <div class="line"><a name="l01008"></a><span class="lineno"> 1008</span>  {</div> |
| 1107 | <div class="line"><a name="l01009"></a><span class="lineno"> 1009</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has unsupported weights OperandLifeTime"</span>, __func__);</div> |
| 1108 | <div class="line"><a name="l01010"></a><span class="lineno"> 1010</span>  }</div> |
| 1109 | <div class="line"><a name="l01011"></a><span class="lineno"> 1011</span>  </div> |
| 1110 | <div class="line"><a name="l01012"></a><span class="lineno"> 1012</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> weightsInput = (desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> == DataLayout::NCHW)</div> |
| 1111 | <div class="line"><a name="l01013"></a><span class="lineno"> 1013</span>  ? <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data, OHWIToOIHW, &input)</div> |
| 1112 | <div class="line"><a name="l01014"></a><span class="lineno"> 1014</span>  : <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data, <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>, &input);</div> |
| 1113 | <div class="line"><a name="l01015"></a><span class="lineno"> 1015</span>  </div> |
| 1114 | <div class="line"><a name="l01016"></a><span class="lineno"> 1016</span>  <span class="keywordflow">if</span> (!weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1115 | <div class="line"><a name="l01017"></a><span class="lineno"> 1017</span>  {</div> |
| 1116 | <div class="line"><a name="l01018"></a><span class="lineno"> 1018</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1117 | <div class="line"><a name="l01019"></a><span class="lineno"> 1019</span>  }</div> |
| 1118 | <div class="line"><a name="l01020"></a><span class="lineno"> 1020</span>  </div> |
| 1119 | <div class="line"><a name="l01021"></a><span class="lineno"> 1021</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> biasInput = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 2, model, data, <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>, &input); <span class="comment">// 1D</span></div> |
| 1120 | <div class="line"><a name="l01022"></a><span class="lineno"> 1022</span>  <span class="keywordflow">if</span> (!biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1121 | <div class="line"><a name="l01023"></a><span class="lineno"> 1023</span>  {</div> |
| 1122 | <div class="line"><a name="l01024"></a><span class="lineno"> 1024</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1123 | <div class="line"><a name="l01025"></a><span class="lineno"> 1025</span>  }</div> |
| 1124 | <div class="line"><a name="l01026"></a><span class="lineno"> 1026</span>  </div> |
| 1125 | <div class="line"><a name="l01027"></a><span class="lineno"> 1027</span>  biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a594af6896fa4bf3de9ee961fa8e2aade">SanitizeQuantizationScale</a>(weightsInput, input);</div> |
| 1126 | <div class="line"><a name="l01028"></a><span class="lineno"> 1028</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> weightsInfo = weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1127 | <div class="line"><a name="l01029"></a><span class="lineno"> 1029</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> biasInfo = biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1128 | <div class="line"><a name="l01030"></a><span class="lineno"> 1030</span>  </div> |
| 1129 | <div class="line"><a name="l01031"></a><span class="lineno"> 1031</span>  ActivationFn activation;</div> |
| 1130 | <div class="line"><a name="l01032"></a><span class="lineno"> 1032</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| 1131 | <div class="line"><a name="l01033"></a><span class="lineno"> 1033</span>  {</div> |
| 1132 | <div class="line"><a name="l01034"></a><span class="lineno"> 1034</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| 1133 | <div class="line"><a name="l01035"></a><span class="lineno"> 1035</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa9ef723ace8562b9636e8387a120fbde">GetInputPaddingScheme</a>(operation, 3, paddingScheme, model, data)</div> |
| 1134 | <div class="line"><a name="l01036"></a><span class="lineno"> 1036</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data)</div> |
| 1135 | <div class="line"><a name="l01037"></a><span class="lineno"> 1037</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data)</div> |
| 1136 | <div class="line"><a name="l01038"></a><span class="lineno"> 1038</span>  || !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 6, activation, model, data)</div> |
| 1137 | <div class="line"><a name="l01039"></a><span class="lineno"> 1039</span>  || !<a class="code" href="namespacearmnn__driver.html#acfe37581bc8296b5d4cdb311b1e3f142">GetOptionalConvolutionDilationParams</a>(operation, 8, desc, model, data))</div> |
| 1138 | <div class="line"><a name="l01040"></a><span class="lineno"> 1040</span>  {</div> |
| 1139 | <div class="line"><a name="l01041"></a><span class="lineno"> 1041</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (implicit padding)"</span>, __func__);</div> |
| 1140 | <div class="line"><a name="l01042"></a><span class="lineno"> 1042</span>  }</div> |
| 1141 | <div class="line"><a name="l01043"></a><span class="lineno"> 1043</span>  </div> |
| 1142 | <div class="line"><a name="l01044"></a><span class="lineno"> 1044</span>  <a class="code" href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a> dataLayoutIndexed(desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a>);</div> |
| 1143 | <div class="line"><a name="l01045"></a><span class="lineno"> 1045</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> widthIndex = dataLayoutIndexed.GetWidthIndex();</div> |
| 1144 | <div class="line"><a name="l01046"></a><span class="lineno"> 1046</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> heightIndex = dataLayoutIndexed.GetHeightIndex();</div> |
| 1145 | <div class="line"><a name="l01047"></a><span class="lineno"> 1047</span>  <span class="keyword">const</span> uint32_t kernelX = weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[widthIndex];</div> |
| 1146 | <div class="line"><a name="l01048"></a><span class="lineno"> 1048</span>  <span class="keyword">const</span> uint32_t kernelY = weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[heightIndex];</div> |
| 1147 | <div class="line"><a name="l01049"></a><span class="lineno"> 1049</span>  <span class="keyword">const</span> uint32_t inputX = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[widthIndex];</div> |
| 1148 | <div class="line"><a name="l01050"></a><span class="lineno"> 1050</span>  <span class="keyword">const</span> uint32_t inputY = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[heightIndex];</div> |
| 1149 | <div class="line"><a name="l01051"></a><span class="lineno"> 1051</span>  </div> |
| 1150 | <div class="line"><a name="l01052"></a><span class="lineno"> 1052</span>  CalcPadding(inputX, kernelX, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aa3c6a77a963a98ccb8ea7b8fd008a8c1">m_DilationX</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, paddingScheme);</div> |
| 1151 | <div class="line"><a name="l01053"></a><span class="lineno"> 1053</span>  CalcPadding(inputY, kernelY, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a302b688d88dd73cde0fb1faef6679907">m_DilationY</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, paddingScheme);</div> |
| 1152 | <div class="line"><a name="l01054"></a><span class="lineno"> 1054</span>  </div> |
| 1153 | <div class="line"><a name="l01055"></a><span class="lineno"> 1055</span>  }</div> |
| 1154 | <div class="line"><a name="l01056"></a><span class="lineno"> 1056</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operation.inputs.size() >= 10)</div> |
| 1155 | <div class="line"><a name="l01057"></a><span class="lineno"> 1057</span>  {</div> |
| 1156 | <div class="line"><a name="l01058"></a><span class="lineno"> 1058</span>  <span class="comment">// explicit padding</span></div> |
| 1157 | <div class="line"><a name="l01059"></a><span class="lineno"> 1059</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 3, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, model, data)</div> |
| 1158 | <div class="line"><a name="l01060"></a><span class="lineno"> 1060</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, model, data)</div> |
| 1159 | <div class="line"><a name="l01061"></a><span class="lineno"> 1061</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, model, data)</div> |
| 1160 | <div class="line"><a name="l01062"></a><span class="lineno"> 1062</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 6, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, model, data)</div> |
| 1161 | <div class="line"><a name="l01063"></a><span class="lineno"> 1063</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 7, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data)</div> |
| 1162 | <div class="line"><a name="l01064"></a><span class="lineno"> 1064</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 8, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data)</div> |
| 1163 | <div class="line"><a name="l01065"></a><span class="lineno"> 1065</span>  || !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 9, activation, model, data)</div> |
| 1164 | <div class="line"><a name="l01066"></a><span class="lineno"> 1066</span>  || !<a class="code" href="namespacearmnn__driver.html#acfe37581bc8296b5d4cdb311b1e3f142">GetOptionalConvolutionDilationParams</a>(operation, 11, desc, model, data))</div> |
| 1165 | <div class="line"><a name="l01067"></a><span class="lineno"> 1067</span>  {</div> |
| 1166 | <div class="line"><a name="l01068"></a><span class="lineno"> 1068</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (explicit padding)"</span>, __func__);</div> |
| 1167 | <div class="line"><a name="l01069"></a><span class="lineno"> 1069</span>  }</div> |
| 1168 | <div class="line"><a name="l01070"></a><span class="lineno"> 1070</span>  }</div> |
| 1169 | <div class="line"><a name="l01071"></a><span class="lineno"> 1071</span>  <span class="keywordflow">else</span></div> |
| 1170 | <div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>  {</div> |
| 1171 | <div class="line"><a name="l01073"></a><span class="lineno"> 1073</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported number of operation inputs"</span>, __func__);</div> |
| 1172 | <div class="line"><a name="l01074"></a><span class="lineno"> 1074</span>  }</div> |
| 1173 | <div class="line"><a name="l01075"></a><span class="lineno"> 1075</span>  </div> |
| 1174 | <div class="line"><a name="l01076"></a><span class="lineno"> 1076</span>  desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">m_BiasEnabled</a> = <span class="keyword">true</span>;</div> |
| 1175 | <div class="line"><a name="l01077"></a><span class="lineno"> 1077</span>  <a class="code" href="classarmnn_1_1_optional.html">Optional<TensorInfo></a> biases(biasInfo);</div> |
| 1176 | <div class="line"><a name="l01078"></a><span class="lineno"> 1078</span>  </div> |
| 1177 | <div class="line"><a name="l01079"></a><span class="lineno"> 1079</span>  <span class="keywordtype">bool</span> requiresValidation = <span class="keyword">true</span>;</div> |
| 1178 | <div class="line"><a name="l01080"></a><span class="lineno"> 1080</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* weightsOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 1179 | <div class="line"><a name="l01081"></a><span class="lineno"> 1081</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* biasOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 2, model);</div> |
| 1180 | <div class="line"><a name="l01082"></a><span class="lineno"> 1082</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a7bf2bd59bc8ca21e0dd906044bfc2faf">IsConnectedToDequantize</a>(weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a7a886c1735beae3f31f2062e1b327ac4">GetOutputSlot</a>())</div> |
| 1181 | <div class="line"><a name="l01083"></a><span class="lineno"> 1083</span>  || <a class="code" href="namespacearmnn__driver.html#a7bf2bd59bc8ca21e0dd906044bfc2faf">IsConnectedToDequantize</a>(biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a7a886c1735beae3f31f2062e1b327ac4">GetOutputSlot</a>()))</div> |
| 1182 | <div class="line"><a name="l01084"></a><span class="lineno"> 1084</span>  {</div> |
| 1183 | <div class="line"><a name="l01085"></a><span class="lineno"> 1085</span>  <span class="comment">// Do not require validation for now. There will be an optimization step</span></div> |
| 1184 | <div class="line"><a name="l01086"></a><span class="lineno"> 1086</span>  <span class="comment">// [ConvertConstDequantisationLayersToConstLayers] will convert layers to Constant layers</span></div> |
| 1185 | <div class="line"><a name="l01087"></a><span class="lineno"> 1087</span>  <span class="comment">// then at the end of the optimization there will be layer supported validation.</span></div> |
| 1186 | <div class="line"><a name="l01088"></a><span class="lineno"> 1088</span>  requiresValidation = <span class="keyword">false</span>;</div> |
| 1187 | <div class="line"><a name="l01089"></a><span class="lineno"> 1089</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConv2d(): Weights and Biases are as INPUTS."</span>;</div> |
| 1188 | <div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>  }</div> |
| 1189 | <div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>  </div> |
| 1190 | <div class="line"><a name="l01092"></a><span class="lineno"> 1092</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1191 | <div class="line"><a name="l01093"></a><span class="lineno"> 1093</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported) {</div> |
| 1192 | <div class="line"><a name="l01094"></a><span class="lineno"> 1094</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1193 | <div class="line"><a name="l01095"></a><span class="lineno"> 1095</span>  IsConvolution2dSupported,</div> |
| 1194 | <div class="line"><a name="l01096"></a><span class="lineno"> 1096</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1195 | <div class="line"><a name="l01097"></a><span class="lineno"> 1097</span>  isSupported,</div> |
| 1196 | <div class="line"><a name="l01098"></a><span class="lineno"> 1098</span>  setBackend,</div> |
| 1197 | <div class="line"><a name="l01099"></a><span class="lineno"> 1099</span>  inputInfo,</div> |
| 1198 | <div class="line"><a name="l01100"></a><span class="lineno"> 1100</span>  outputInfo,</div> |
| 1199 | <div class="line"><a name="l01101"></a><span class="lineno"> 1101</span>  desc,</div> |
| 1200 | <div class="line"><a name="l01102"></a><span class="lineno"> 1102</span>  weightsInfo,</div> |
| 1201 | <div class="line"><a name="l01103"></a><span class="lineno"> 1103</span>  biases);</div> |
| 1202 | <div class="line"><a name="l01104"></a><span class="lineno"> 1104</span>  };</div> |
| 1203 | <div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>  </div> |
| 1204 | <div class="line"><a name="l01106"></a><span class="lineno"> 1106</span>  <span class="keywordflow">if</span> (requiresValidation)</div> |
| 1205 | <div class="line"><a name="l01107"></a><span class="lineno"> 1107</span>  {</div> |
| 1206 | <div class="line"><a name="l01108"></a><span class="lineno"> 1108</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConv2d(): Requires Validation!"</span>;</div> |
| 1207 | <div class="line"><a name="l01109"></a><span class="lineno"> 1109</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1208 | <div class="line"><a name="l01110"></a><span class="lineno"> 1110</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1209 | <div class="line"><a name="l01111"></a><span class="lineno"> 1111</span>  {</div> |
| 1210 | <div class="line"><a name="l01112"></a><span class="lineno"> 1112</span>  validateFunc(outputInfo, isSupported);</div> |
| 1211 | <div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>  }</div> |
| 1212 | <div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>  <span class="keywordflow">else</span></div> |
| 1213 | <div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>  {</div> |
| 1214 | <div class="line"><a name="l01116"></a><span class="lineno"> 1116</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1215 | <div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>  }</div> |
| 1216 | <div class="line"><a name="l01118"></a><span class="lineno"> 1118</span>  </div> |
| 1217 | <div class="line"><a name="l01119"></a><span class="lineno"> 1119</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1218 | <div class="line"><a name="l01120"></a><span class="lineno"> 1120</span>  {</div> |
| 1219 | <div class="line"><a name="l01121"></a><span class="lineno"> 1121</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1220 | <div class="line"><a name="l01122"></a><span class="lineno"> 1122</span>  }</div> |
| 1221 | <div class="line"><a name="l01123"></a><span class="lineno"> 1123</span>  }</div> |
| 1222 | <div class="line"><a name="l01124"></a><span class="lineno"> 1124</span>  </div> |
| 1223 | <div class="line"><a name="l01125"></a><span class="lineno"> 1125</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* startLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddConvolution2dLayer(desc);</div> |
| 1224 | <div class="line"><a name="l01126"></a><span class="lineno"> 1126</span>  startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1225 | <div class="line"><a name="l01127"></a><span class="lineno"> 1127</span>  </div> |
| 1226 | <div class="line"><a name="l01128"></a><span class="lineno"> 1128</span>  <span class="keywordflow">if</span> (!startLayer)</div> |
| 1227 | <div class="line"><a name="l01129"></a><span class="lineno"> 1129</span>  {</div> |
| 1228 | <div class="line"><a name="l01130"></a><span class="lineno"> 1130</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddConvolution2dLayer failed"</span>, __func__);</div> |
| 1229 | <div class="line"><a name="l01131"></a><span class="lineno"> 1131</span>  }</div> |
| 1230 | <div class="line"><a name="l01132"></a><span class="lineno"> 1132</span>  </div> |
| 1231 | <div class="line"><a name="l01133"></a><span class="lineno"> 1133</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1232 | <div class="line"><a name="l01134"></a><span class="lineno"> 1134</span>  weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 1233 | <div class="line"><a name="l01135"></a><span class="lineno"> 1135</span>  biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 1234 | <div class="line"><a name="l01136"></a><span class="lineno"> 1136</span>  </div> |
| 1235 | <div class="line"><a name="l01137"></a><span class="lineno"> 1137</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *startLayer, model, data, <span class="keyword">nullptr</span>, validateFunc, activation);</div> |
| 1236 | <div class="line"><a name="l01138"></a><span class="lineno"> 1138</span> }</div> |
| 1237 | <div class="line"><a name="l01139"></a><span class="lineno"> 1139</span>  </div> |
| 1238 | <div class="line"><a name="l01140"></a><span class="lineno"> 1140</span> <span class="keywordtype">bool</span> Converter::ConvertDepthToSpace(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1239 | <div class="line"><a name="l01141"></a><span class="lineno"> 1141</span> {</div> |
| 1240 | <div class="line"><a name="l01142"></a><span class="lineno"> 1142</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthToSpace()"</span>;</div> |
| 1241 | <div class="line"><a name="l01143"></a><span class="lineno"> 1143</span>  </div> |
| 1242 | <div class="line"><a name="l01144"></a><span class="lineno"> 1144</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1243 | <div class="line"><a name="l01145"></a><span class="lineno"> 1145</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() )</div> |
| 1244 | <div class="line"><a name="l01146"></a><span class="lineno"> 1146</span>  {</div> |
| 1245 | <div class="line"><a name="l01147"></a><span class="lineno"> 1147</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1246 | <div class="line"><a name="l01148"></a><span class="lineno"> 1148</span>  }</div> |
| 1247 | <div class="line"><a name="l01149"></a><span class="lineno"> 1149</span>  </div> |
| 1248 | <div class="line"><a name="l01150"></a><span class="lineno"> 1150</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1249 | <div class="line"><a name="l01151"></a><span class="lineno"> 1151</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 1250 | <div class="line"><a name="l01152"></a><span class="lineno"> 1152</span>  <span class="keywordflow">if</span> (rank != 4)</div> |
| 1251 | <div class="line"><a name="l01153"></a><span class="lineno"> 1153</span>  {</div> |
| 1252 | <div class="line"><a name="l01154"></a><span class="lineno"> 1154</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Only inputs with rank 4 are supported"</span>, __func__);</div> |
| 1253 | <div class="line"><a name="l01155"></a><span class="lineno"> 1155</span>  }</div> |
| 1254 | <div class="line"><a name="l01156"></a><span class="lineno"> 1156</span>  </div> |
| 1255 | <div class="line"><a name="l01157"></a><span class="lineno"> 1157</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1256 | <div class="line"><a name="l01158"></a><span class="lineno"> 1158</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1257 | <div class="line"><a name="l01159"></a><span class="lineno"> 1159</span>  {</div> |
| 1258 | <div class="line"><a name="l01160"></a><span class="lineno"> 1160</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 1259 | <div class="line"><a name="l01161"></a><span class="lineno"> 1161</span>  }</div> |
| 1260 | <div class="line"><a name="l01162"></a><span class="lineno"> 1162</span>  </div> |
| 1261 | <div class="line"><a name="l01163"></a><span class="lineno"> 1163</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1262 | <div class="line"><a name="l01164"></a><span class="lineno"> 1164</span>  </div> |
| 1263 | <div class="line"><a name="l01165"></a><span class="lineno"> 1165</span>  <a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html">armnn::DepthToSpaceDescriptor</a> descriptor;</div> |
| 1264 | <div class="line"><a name="l01166"></a><span class="lineno"> 1166</span>  </div> |
| 1265 | <div class="line"><a name="l01167"></a><span class="lineno"> 1167</span>  <a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, descriptor.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6c6b8957f1e176867e5fb05b1a1a1486">m_BlockSize</a>, model, data);</div> |
| 1266 | <div class="line"><a name="l01168"></a><span class="lineno"> 1168</span>  <span class="keywordflow">if</span> (descriptor.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6c6b8957f1e176867e5fb05b1a1a1486">m_BlockSize</a> <= 1)</div> |
| 1267 | <div class="line"><a name="l01169"></a><span class="lineno"> 1169</span>  {</div> |
| 1268 | <div class="line"><a name="l01170"></a><span class="lineno"> 1170</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Block size must be at least 1 in all dimensions"</span>, __func__);</div> |
| 1269 | <div class="line"><a name="l01171"></a><span class="lineno"> 1171</span>  }</div> |
| 1270 | <div class="line"><a name="l01172"></a><span class="lineno"> 1172</span>  </div> |
| 1271 | <div class="line"><a name="l01173"></a><span class="lineno"> 1173</span>  descriptor.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 1272 | <div class="line"><a name="l01174"></a><span class="lineno"> 1174</span>  <span class="keywordflow">if</span> (Is12OrLaterOperand(*output))</div> |
| 1273 | <div class="line"><a name="l01175"></a><span class="lineno"> 1175</span>  {</div> |
| 1274 | <div class="line"><a name="l01176"></a><span class="lineno"> 1176</span>  descriptor.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 2, model, data);</div> |
| 1275 | <div class="line"><a name="l01177"></a><span class="lineno"> 1177</span>  }</div> |
| 1276 | <div class="line"><a name="l01178"></a><span class="lineno"> 1178</span>  </div> |
| 1277 | <div class="line"><a name="l01179"></a><span class="lineno"> 1179</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1278 | <div class="line"><a name="l01180"></a><span class="lineno"> 1180</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1279 | <div class="line"><a name="l01181"></a><span class="lineno"> 1181</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 1280 | <div class="line"><a name="l01182"></a><span class="lineno"> 1182</span>  {</div> |
| 1281 | <div class="line"><a name="l01183"></a><span class="lineno"> 1183</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1282 | <div class="line"><a name="l01184"></a><span class="lineno"> 1184</span>  IsDepthToSpaceSupported,</div> |
| 1283 | <div class="line"><a name="l01185"></a><span class="lineno"> 1185</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1284 | <div class="line"><a name="l01186"></a><span class="lineno"> 1186</span>  isSupported,</div> |
| 1285 | <div class="line"><a name="l01187"></a><span class="lineno"> 1187</span>  setBackend,</div> |
| 1286 | <div class="line"><a name="l01188"></a><span class="lineno"> 1188</span>  inputInfo,</div> |
| 1287 | <div class="line"><a name="l01189"></a><span class="lineno"> 1189</span>  outputInfo,</div> |
| 1288 | <div class="line"><a name="l01190"></a><span class="lineno"> 1190</span>  descriptor);</div> |
| 1289 | <div class="line"><a name="l01191"></a><span class="lineno"> 1191</span>  };</div> |
| 1290 | <div class="line"><a name="l01192"></a><span class="lineno"> 1192</span>  </div> |
| 1291 | <div class="line"><a name="l01193"></a><span class="lineno"> 1193</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1292 | <div class="line"><a name="l01194"></a><span class="lineno"> 1194</span>  {</div> |
| 1293 | <div class="line"><a name="l01195"></a><span class="lineno"> 1195</span>  validateFunc(outputInfo, isSupported);</div> |
| 1294 | <div class="line"><a name="l01196"></a><span class="lineno"> 1196</span>  }</div> |
| 1295 | <div class="line"><a name="l01197"></a><span class="lineno"> 1197</span>  <span class="keywordflow">else</span></div> |
| 1296 | <div class="line"><a name="l01198"></a><span class="lineno"> 1198</span>  {</div> |
| 1297 | <div class="line"><a name="l01199"></a><span class="lineno"> 1199</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1298 | <div class="line"><a name="l01200"></a><span class="lineno"> 1200</span>  }</div> |
| 1299 | <div class="line"><a name="l01201"></a><span class="lineno"> 1201</span>  </div> |
| 1300 | <div class="line"><a name="l01202"></a><span class="lineno"> 1202</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1301 | <div class="line"><a name="l01203"></a><span class="lineno"> 1203</span>  {</div> |
| 1302 | <div class="line"><a name="l01204"></a><span class="lineno"> 1204</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1303 | <div class="line"><a name="l01205"></a><span class="lineno"> 1205</span>  }</div> |
| 1304 | <div class="line"><a name="l01206"></a><span class="lineno"> 1206</span>  </div> |
| 1305 | <div class="line"><a name="l01207"></a><span class="lineno"> 1207</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddDepthToSpaceLayer(descriptor);</div> |
| 1306 | <div class="line"><a name="l01208"></a><span class="lineno"> 1208</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1307 | <div class="line"><a name="l01209"></a><span class="lineno"> 1209</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 1308 | <div class="line"><a name="l01210"></a><span class="lineno"> 1210</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1309 | <div class="line"><a name="l01211"></a><span class="lineno"> 1211</span>  </div> |
| 1310 | <div class="line"><a name="l01212"></a><span class="lineno"> 1212</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 1311 | <div class="line"><a name="l01213"></a><span class="lineno"> 1213</span> }</div> |
| 1312 | <div class="line"><a name="l01214"></a><span class="lineno"> 1214</span>  </div> |
| 1313 | <div class="line"><a name="l01215"></a><span class="lineno"> 1215</span> <span class="keywordtype">bool</span> Converter::ConvertDepthwiseConv2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1314 | <div class="line"><a name="l01216"></a><span class="lineno"> 1216</span> {</div> |
| 1315 | <div class="line"><a name="l01217"></a><span class="lineno"> 1217</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthwiseConv2d()"</span>;</div> |
| 1316 | <div class="line"><a name="l01218"></a><span class="lineno"> 1218</span>  </div> |
| 1317 | <div class="line"><a name="l01219"></a><span class="lineno"> 1219</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1318 | <div class="line"><a name="l01220"></a><span class="lineno"> 1220</span>  </div> |
| 1319 | <div class="line"><a name="l01221"></a><span class="lineno"> 1221</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1320 | <div class="line"><a name="l01222"></a><span class="lineno"> 1222</span>  {</div> |
| 1321 | <div class="line"><a name="l01223"></a><span class="lineno"> 1223</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1322 | <div class="line"><a name="l01224"></a><span class="lineno"> 1224</span>  }</div> |
| 1323 | <div class="line"><a name="l01225"></a><span class="lineno"> 1225</span>  </div> |
| 1324 | <div class="line"><a name="l01226"></a><span class="lineno"> 1226</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1325 | <div class="line"><a name="l01227"></a><span class="lineno"> 1227</span>  </div> |
| 1326 | <div class="line"><a name="l01228"></a><span class="lineno"> 1228</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1327 | <div class="line"><a name="l01229"></a><span class="lineno"> 1229</span>  {</div> |
| 1328 | <div class="line"><a name="l01230"></a><span class="lineno"> 1230</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 1329 | <div class="line"><a name="l01231"></a><span class="lineno"> 1231</span>  }</div> |
| 1330 | <div class="line"><a name="l01232"></a><span class="lineno"> 1232</span>  </div> |
| 1331 | <div class="line"><a name="l01233"></a><span class="lineno"> 1233</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1332 | <div class="line"><a name="l01234"></a><span class="lineno"> 1234</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1333 | <div class="line"><a name="l01235"></a><span class="lineno"> 1235</span>  </div> |
| 1334 | <div class="line"><a name="l01236"></a><span class="lineno"> 1236</span>  <span class="comment">// ArmNN does not currently support non-fixed weights or bias</span></div> |
| 1335 | <div class="line"><a name="l01237"></a><span class="lineno"> 1237</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ae780aad3075e52e802f9c0ed5d88e570">IsWeightsValid</a>(operation, 1, model, <span class="keyword">false</span>))</div> |
| 1336 | <div class="line"><a name="l01238"></a><span class="lineno"> 1238</span>  {</div> |
| 1337 | <div class="line"><a name="l01239"></a><span class="lineno"> 1239</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: This Operation has unsupported weights OperandLifeTime"</span>, __func__);</div> |
| 1338 | <div class="line"><a name="l01240"></a><span class="lineno"> 1240</span>  }</div> |
| 1339 | <div class="line"><a name="l01241"></a><span class="lineno"> 1241</span>  </div> |
| 1340 | <div class="line"><a name="l01242"></a><span class="lineno"> 1242</span>  <span class="comment">// Find the shape of the weights tensor. In AndroidNN this will be [ 1, H, W, I * M ]</span></div> |
| 1341 | <div class="line"><a name="l01243"></a><span class="lineno"> 1243</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* weightsOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 1342 | <div class="line"><a name="l01244"></a><span class="lineno"> 1244</span>  </div> |
| 1343 | <div class="line"><a name="l01245"></a><span class="lineno"> 1245</span>  <span class="comment">// Basic sanity check on the weights shape.</span></div> |
| 1344 | <div class="line"><a name="l01246"></a><span class="lineno"> 1246</span>  <span class="comment">// ANEURALNETWORKS_DEPTHWISE_CONV_2D specifies a 4-D tensor, of shape</span></div> |
| 1345 | <div class="line"><a name="l01247"></a><span class="lineno"> 1247</span>  <span class="comment">// [1, filter_height, filter_width, depth_out]</span></div> |
| 1346 | <div class="line"><a name="l01248"></a><span class="lineno"> 1248</span>  <span class="keywordflow">if</span> (weightsOperand->dimensions[0] != 1)</div> |
| 1347 | <div class="line"><a name="l01249"></a><span class="lineno"> 1249</span>  {</div> |
| 1348 | <div class="line"><a name="l01250"></a><span class="lineno"> 1250</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Filter operand dimension 0 is invalid, should be 1"</span>, __func__);</div> |
| 1349 | <div class="line"><a name="l01251"></a><span class="lineno"> 1251</span>  }</div> |
| 1350 | <div class="line"><a name="l01252"></a><span class="lineno"> 1252</span>  </div> |
| 1351 | <div class="line"><a name="l01253"></a><span class="lineno"> 1253</span>  <a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html">armnn::DepthwiseConvolution2dDescriptor</a> desc;</div> |
| 1352 | <div class="line"><a name="l01254"></a><span class="lineno"> 1254</span>  desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 1353 | <div class="line"><a name="l01255"></a><span class="lineno"> 1255</span>  </div> |
| 1354 | <div class="line"><a name="l01256"></a><span class="lineno"> 1256</span>  <span class="comment">// Determine whether padding is implicit or explicit</span></div> |
| 1355 | <div class="line"><a name="l01257"></a><span class="lineno"> 1257</span>  <span class="keywordtype">bool</span> implicitPadding = operation.inputs.size() == 8</div> |
| 1356 | <div class="line"><a name="l01258"></a><span class="lineno"> 1258</span>  || (operation.inputs.size() >= 9</div> |
| 1357 | <div class="line"><a name="l01259"></a><span class="lineno"> 1259</span>  && <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 8, model)->type == OperandType::BOOL);</div> |
| 1358 | <div class="line"><a name="l01260"></a><span class="lineno"> 1260</span>  </div> |
| 1359 | <div class="line"><a name="l01261"></a><span class="lineno"> 1261</span>  <span class="comment">// Look ahead to find the optional DataLayout, if present</span></div> |
| 1360 | <div class="line"><a name="l01262"></a><span class="lineno"> 1262</span>  <span class="keyword">const</span> uint32_t dataLayoutFlagIndex = implicitPadding ? 8 : 11;</div> |
| 1361 | <div class="line"><a name="l01263"></a><span class="lineno"> 1263</span>  desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, dataLayoutFlagIndex, model, data);</div> |
| 1362 | <div class="line"><a name="l01264"></a><span class="lineno"> 1264</span>  </div> |
| 1363 | <div class="line"><a name="l01265"></a><span class="lineno"> 1265</span>  <a class="code" href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a> dataLayoutIndexed(desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a>);</div> |
| 1364 | <div class="line"><a name="l01266"></a><span class="lineno"> 1266</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> widthIndex = dataLayoutIndexed.GetWidthIndex();</div> |
| 1365 | <div class="line"><a name="l01267"></a><span class="lineno"> 1267</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> heightIndex = dataLayoutIndexed.GetHeightIndex();</div> |
| 1366 | <div class="line"><a name="l01268"></a><span class="lineno"> 1268</span>  </div> |
| 1367 | <div class="line"><a name="l01269"></a><span class="lineno"> 1269</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> weightsInput = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data, <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>, &input);</div> |
| 1368 | <div class="line"><a name="l01270"></a><span class="lineno"> 1270</span>  <span class="keywordflow">if</span> (!weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1369 | <div class="line"><a name="l01271"></a><span class="lineno"> 1271</span>  {</div> |
| 1370 | <div class="line"><a name="l01272"></a><span class="lineno"> 1272</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1371 | <div class="line"><a name="l01273"></a><span class="lineno"> 1273</span>  }</div> |
| 1372 | <div class="line"><a name="l01274"></a><span class="lineno"> 1274</span>  </div> |
| 1373 | <div class="line"><a name="l01275"></a><span class="lineno"> 1275</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* biasOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 2, model);</div> |
| 1374 | <div class="line"><a name="l01276"></a><span class="lineno"> 1276</span>  <span class="keywordflow">if</span> (!biasOperand)</div> |
| 1375 | <div class="line"><a name="l01277"></a><span class="lineno"> 1277</span>  {</div> |
| 1376 | <div class="line"><a name="l01278"></a><span class="lineno"> 1278</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read bias"</span>, __func__);</div> |
| 1377 | <div class="line"><a name="l01279"></a><span class="lineno"> 1279</span>  }</div> |
| 1378 | <div class="line"><a name="l01280"></a><span class="lineno"> 1280</span>  </div> |
| 1379 | <div class="line"><a name="l01281"></a><span class="lineno"> 1281</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> biasInput = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 2, model, data, <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>, &input); <span class="comment">// 1D</span></div> |
| 1380 | <div class="line"><a name="l01282"></a><span class="lineno"> 1282</span>  <span class="keywordflow">if</span> (!biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1381 | <div class="line"><a name="l01283"></a><span class="lineno"> 1283</span>  {</div> |
| 1382 | <div class="line"><a name="l01284"></a><span class="lineno"> 1284</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1383 | <div class="line"><a name="l01285"></a><span class="lineno"> 1285</span>  }</div> |
| 1384 | <div class="line"><a name="l01286"></a><span class="lineno"> 1286</span>  </div> |
| 1385 | <div class="line"><a name="l01287"></a><span class="lineno"> 1287</span>  biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a594af6896fa4bf3de9ee961fa8e2aade">SanitizeQuantizationScale</a>(weightsInput, input);</div> |
| 1386 | <div class="line"><a name="l01288"></a><span class="lineno"> 1288</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> weightsInfo = weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1387 | <div class="line"><a name="l01289"></a><span class="lineno"> 1289</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> biasInfo = biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1388 | <div class="line"><a name="l01290"></a><span class="lineno"> 1290</span>  </div> |
| 1389 | <div class="line"><a name="l01291"></a><span class="lineno"> 1291</span>  ActivationFn activation;</div> |
| 1390 | <div class="line"><a name="l01292"></a><span class="lineno"> 1292</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| 1391 | <div class="line"><a name="l01293"></a><span class="lineno"> 1293</span>  {</div> |
| 1392 | <div class="line"><a name="l01294"></a><span class="lineno"> 1294</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| 1393 | <div class="line"><a name="l01295"></a><span class="lineno"> 1295</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa9ef723ace8562b9636e8387a120fbde">GetInputPaddingScheme</a>(operation, 3, paddingScheme, model, data)</div> |
| 1394 | <div class="line"><a name="l01296"></a><span class="lineno"> 1296</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data)</div> |
| 1395 | <div class="line"><a name="l01297"></a><span class="lineno"> 1297</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data)</div> |
| 1396 | <div class="line"><a name="l01298"></a><span class="lineno"> 1298</span>  || !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 7, activation, model, data)</div> |
| 1397 | <div class="line"><a name="l01299"></a><span class="lineno"> 1299</span>  || !<a class="code" href="namespacearmnn__driver.html#acfe37581bc8296b5d4cdb311b1e3f142">GetOptionalConvolutionDilationParams</a>(operation, 9, desc, model, data))</div> |
| 1398 | <div class="line"><a name="l01300"></a><span class="lineno"> 1300</span>  {</div> |
| 1399 | <div class="line"><a name="l01301"></a><span class="lineno"> 1301</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (implicit padding)"</span>, __func__);</div> |
| 1400 | <div class="line"><a name="l01302"></a><span class="lineno"> 1302</span>  }</div> |
| 1401 | <div class="line"><a name="l01303"></a><span class="lineno"> 1303</span>  </div> |
| 1402 | <div class="line"><a name="l01304"></a><span class="lineno"> 1304</span>  <span class="keyword">const</span> uint32_t kernelX = weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[2];</div> |
| 1403 | <div class="line"><a name="l01305"></a><span class="lineno"> 1305</span>  <span class="keyword">const</span> uint32_t kernelY = weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[1];</div> |
| 1404 | <div class="line"><a name="l01306"></a><span class="lineno"> 1306</span>  <span class="keyword">const</span> uint32_t inputX = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[widthIndex];</div> |
| 1405 | <div class="line"><a name="l01307"></a><span class="lineno"> 1307</span>  <span class="keyword">const</span> uint32_t inputY = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[heightIndex];</div> |
| 1406 | <div class="line"><a name="l01308"></a><span class="lineno"> 1308</span>  </div> |
| 1407 | <div class="line"><a name="l01309"></a><span class="lineno"> 1309</span>  CalcPadding(inputX, kernelX, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aa3c6a77a963a98ccb8ea7b8fd008a8c1">m_DilationX</a>, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, paddingScheme);</div> |
| 1408 | <div class="line"><a name="l01310"></a><span class="lineno"> 1310</span>  CalcPadding(inputY, kernelY, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a302b688d88dd73cde0fb1faef6679907">m_DilationY</a>, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, paddingScheme);</div> |
| 1409 | <div class="line"><a name="l01311"></a><span class="lineno"> 1311</span>  }</div> |
| 1410 | <div class="line"><a name="l01312"></a><span class="lineno"> 1312</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operation.inputs.size() >= 11)</div> |
| 1411 | <div class="line"><a name="l01313"></a><span class="lineno"> 1313</span>  {</div> |
| 1412 | <div class="line"><a name="l01314"></a><span class="lineno"> 1314</span>  <span class="comment">// explicit padding</span></div> |
| 1413 | <div class="line"><a name="l01315"></a><span class="lineno"> 1315</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 3, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, model, data)</div> |
| 1414 | <div class="line"><a name="l01316"></a><span class="lineno"> 1316</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, model, data)</div> |
| 1415 | <div class="line"><a name="l01317"></a><span class="lineno"> 1317</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, model, data)</div> |
| 1416 | <div class="line"><a name="l01318"></a><span class="lineno"> 1318</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 6, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, model, data)</div> |
| 1417 | <div class="line"><a name="l01319"></a><span class="lineno"> 1319</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 7, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data)</div> |
| 1418 | <div class="line"><a name="l01320"></a><span class="lineno"> 1320</span>  || !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 8, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data)</div> |
| 1419 | <div class="line"><a name="l01321"></a><span class="lineno"> 1321</span>  || !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 10, activation, model, data)</div> |
| 1420 | <div class="line"><a name="l01322"></a><span class="lineno"> 1322</span>  || !<a class="code" href="namespacearmnn__driver.html#acfe37581bc8296b5d4cdb311b1e3f142">GetOptionalConvolutionDilationParams</a>(operation, 12, desc, model, data))</div> |
| 1421 | <div class="line"><a name="l01323"></a><span class="lineno"> 1323</span>  {</div> |
| 1422 | <div class="line"><a name="l01324"></a><span class="lineno"> 1324</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (explicit padding)"</span>, __func__);</div> |
| 1423 | <div class="line"><a name="l01325"></a><span class="lineno"> 1325</span>  }</div> |
| 1424 | <div class="line"><a name="l01326"></a><span class="lineno"> 1326</span>  }</div> |
| 1425 | <div class="line"><a name="l01327"></a><span class="lineno"> 1327</span>  <span class="keywordflow">else</span></div> |
| 1426 | <div class="line"><a name="l01328"></a><span class="lineno"> 1328</span>  {</div> |
| 1427 | <div class="line"><a name="l01329"></a><span class="lineno"> 1329</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported number of operation inputs"</span>, __func__);</div> |
| 1428 | <div class="line"><a name="l01330"></a><span class="lineno"> 1330</span>  }</div> |
| 1429 | <div class="line"><a name="l01331"></a><span class="lineno"> 1331</span>  </div> |
| 1430 | <div class="line"><a name="l01332"></a><span class="lineno"> 1332</span>  desc.<a class="code" href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">m_BiasEnabled</a> = <span class="keyword">true</span>;</div> |
| 1431 | <div class="line"><a name="l01333"></a><span class="lineno"> 1333</span>  <a class="code" href="classarmnn_1_1_optional.html">Optional<TensorInfo></a> biases(biasInfo);</div> |
| 1432 | <div class="line"><a name="l01334"></a><span class="lineno"> 1334</span>  </div> |
| 1433 | <div class="line"><a name="l01335"></a><span class="lineno"> 1335</span>  <span class="keywordtype">bool</span> requiresValidation = <span class="keyword">true</span>;</div> |
| 1434 | <div class="line"><a name="l01336"></a><span class="lineno"> 1336</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a7bf2bd59bc8ca21e0dd906044bfc2faf">IsConnectedToDequantize</a>(weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a7a886c1735beae3f31f2062e1b327ac4">GetOutputSlot</a>()) || <a class="code" href="namespacearmnn__driver.html#a7bf2bd59bc8ca21e0dd906044bfc2faf">IsConnectedToDequantize</a>(biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a7a886c1735beae3f31f2062e1b327ac4">GetOutputSlot</a>()))</div> |
| 1435 | <div class="line"><a name="l01337"></a><span class="lineno"> 1337</span>  {</div> |
| 1436 | <div class="line"><a name="l01338"></a><span class="lineno"> 1338</span>  <span class="comment">// Do not require validation for now. There will be an optimization step</span></div> |
| 1437 | <div class="line"><a name="l01339"></a><span class="lineno"> 1339</span>  <span class="comment">// [ConvertConstDequantisationLayersToConstLayers] will convert layers to Constant layers</span></div> |
| 1438 | <div class="line"><a name="l01340"></a><span class="lineno"> 1340</span>  <span class="comment">// then at the end of the optimization there will be layer supported validation.</span></div> |
| 1439 | <div class="line"><a name="l01341"></a><span class="lineno"> 1341</span>  requiresValidation = <span class="keyword">false</span>;</div> |
| 1440 | <div class="line"><a name="l01342"></a><span class="lineno"> 1342</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthwiseConv2d(): Weights and Biases are as INPUTS."</span>;</div> |
| 1441 | <div class="line"><a name="l01343"></a><span class="lineno"> 1343</span>  }</div> |
| 1442 | <div class="line"><a name="l01344"></a><span class="lineno"> 1344</span>  </div> |
| 1443 | <div class="line"><a name="l01345"></a><span class="lineno"> 1345</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1444 | <div class="line"><a name="l01346"></a><span class="lineno"> 1346</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported) {</div> |
| 1445 | <div class="line"><a name="l01347"></a><span class="lineno"> 1347</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1446 | <div class="line"><a name="l01348"></a><span class="lineno"> 1348</span>  IsDepthwiseConvolutionSupported,</div> |
| 1447 | <div class="line"><a name="l01349"></a><span class="lineno"> 1349</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1448 | <div class="line"><a name="l01350"></a><span class="lineno"> 1350</span>  isSupported,</div> |
| 1449 | <div class="line"><a name="l01351"></a><span class="lineno"> 1351</span>  setBackend,</div> |
| 1450 | <div class="line"><a name="l01352"></a><span class="lineno"> 1352</span>  inputInfo,</div> |
| 1451 | <div class="line"><a name="l01353"></a><span class="lineno"> 1353</span>  outputInfo,</div> |
| 1452 | <div class="line"><a name="l01354"></a><span class="lineno"> 1354</span>  desc,</div> |
| 1453 | <div class="line"><a name="l01355"></a><span class="lineno"> 1355</span>  weightsInfo,</div> |
| 1454 | <div class="line"><a name="l01356"></a><span class="lineno"> 1356</span>  biases);</div> |
| 1455 | <div class="line"><a name="l01357"></a><span class="lineno"> 1357</span>  };</div> |
| 1456 | <div class="line"><a name="l01358"></a><span class="lineno"> 1358</span>  </div> |
| 1457 | <div class="line"><a name="l01359"></a><span class="lineno"> 1359</span>  <span class="keywordflow">if</span> (requiresValidation)</div> |
| 1458 | <div class="line"><a name="l01360"></a><span class="lineno"> 1360</span>  {</div> |
| 1459 | <div class="line"><a name="l01361"></a><span class="lineno"> 1361</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthwiseConv2d(): Requires Validation!"</span>;</div> |
| 1460 | <div class="line"><a name="l01362"></a><span class="lineno"> 1362</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1461 | <div class="line"><a name="l01363"></a><span class="lineno"> 1363</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1462 | <div class="line"><a name="l01364"></a><span class="lineno"> 1364</span>  {</div> |
| 1463 | <div class="line"><a name="l01365"></a><span class="lineno"> 1365</span>  validateFunc(outputInfo, isSupported);</div> |
| 1464 | <div class="line"><a name="l01366"></a><span class="lineno"> 1366</span>  }</div> |
| 1465 | <div class="line"><a name="l01367"></a><span class="lineno"> 1367</span>  <span class="keywordflow">else</span></div> |
| 1466 | <div class="line"><a name="l01368"></a><span class="lineno"> 1368</span>  {</div> |
| 1467 | <div class="line"><a name="l01369"></a><span class="lineno"> 1369</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1468 | <div class="line"><a name="l01370"></a><span class="lineno"> 1370</span>  }</div> |
| 1469 | <div class="line"><a name="l01371"></a><span class="lineno"> 1371</span>  </div> |
| 1470 | <div class="line"><a name="l01372"></a><span class="lineno"> 1372</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1471 | <div class="line"><a name="l01373"></a><span class="lineno"> 1373</span>  {</div> |
| 1472 | <div class="line"><a name="l01374"></a><span class="lineno"> 1374</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1473 | <div class="line"><a name="l01375"></a><span class="lineno"> 1375</span>  }</div> |
| 1474 | <div class="line"><a name="l01376"></a><span class="lineno"> 1376</span>  }</div> |
| 1475 | <div class="line"><a name="l01377"></a><span class="lineno"> 1377</span>  </div> |
| 1476 | <div class="line"><a name="l01378"></a><span class="lineno"> 1378</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* startLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddDepthwiseConvolution2dLayer(desc);</div> |
| 1477 | <div class="line"><a name="l01379"></a><span class="lineno"> 1379</span>  startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1478 | <div class="line"><a name="l01380"></a><span class="lineno"> 1380</span>  </div> |
| 1479 | <div class="line"><a name="l01381"></a><span class="lineno"> 1381</span>  <span class="keywordflow">if</span> (!startLayer)</div> |
| 1480 | <div class="line"><a name="l01382"></a><span class="lineno"> 1382</span>  {</div> |
| 1481 | <div class="line"><a name="l01383"></a><span class="lineno"> 1383</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddDepthwiseConvolution2dLayer failed"</span>, __func__);</div> |
| 1482 | <div class="line"><a name="l01384"></a><span class="lineno"> 1384</span>  }</div> |
| 1483 | <div class="line"><a name="l01385"></a><span class="lineno"> 1385</span>  </div> |
| 1484 | <div class="line"><a name="l01386"></a><span class="lineno"> 1386</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1485 | <div class="line"><a name="l01387"></a><span class="lineno"> 1387</span>  </div> |
| 1486 | <div class="line"><a name="l01388"></a><span class="lineno"> 1388</span>  <span class="comment">// Connect weights and bias inputs</span></div> |
| 1487 | <div class="line"><a name="l01389"></a><span class="lineno"> 1389</span>  weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 1488 | <div class="line"><a name="l01390"></a><span class="lineno"> 1390</span>  biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 1489 | <div class="line"><a name="l01391"></a><span class="lineno"> 1391</span>  </div> |
| 1490 | <div class="line"><a name="l01392"></a><span class="lineno"> 1392</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *startLayer, model, data, <span class="keyword">nullptr</span>, validateFunc, activation);</div> |
| 1491 | <div class="line"><a name="l01393"></a><span class="lineno"> 1393</span> }</div> |
| 1492 | <div class="line"><a name="l01394"></a><span class="lineno"> 1394</span>  </div> |
| 1493 | <div class="line"><a name="l01395"></a><span class="lineno"> 1395</span> <span class="keywordtype">bool</span> Converter::ConvertDequantize(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1494 | <div class="line"><a name="l01396"></a><span class="lineno"> 1396</span> {</div> |
| 1495 | <div class="line"><a name="l01397"></a><span class="lineno"> 1397</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDequantize()"</span>;</div> |
| 1496 | <div class="line"><a name="l01398"></a><span class="lineno"> 1398</span>  </div> |
| 1497 | <div class="line"><a name="l01399"></a><span class="lineno"> 1399</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1498 | <div class="line"><a name="l01400"></a><span class="lineno"> 1400</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1499 | <div class="line"><a name="l01401"></a><span class="lineno"> 1401</span>  {</div> |
| 1500 | <div class="line"><a name="l01402"></a><span class="lineno"> 1402</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid input"</span>, __func__);</div> |
| 1501 | <div class="line"><a name="l01403"></a><span class="lineno"> 1403</span>  }</div> |
| 1502 | <div class="line"><a name="l01404"></a><span class="lineno"> 1404</span>  </div> |
| 1503 | <div class="line"><a name="l01405"></a><span class="lineno"> 1405</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1504 | <div class="line"><a name="l01406"></a><span class="lineno"> 1406</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_optional.html">armnn::Optional<unsigned int></a>& quantizationDim = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b8fc85ce966c035d789cf22db5088a1">GetQuantizationDim</a>();</div> |
| 1505 | <div class="line"><a name="l01407"></a><span class="lineno"> 1407</span>  <span class="keywordflow">if</span> (quantizationDim.<a class="code" href="classarmnn_1_1_optional_base.html#a86b749ce2c4bc627fa8a1fcfaf0e314f">has_value</a>() && quantizationDim.<a class="code" href="classarmnn_1_1_optional_reference_switch.html#a77c7d528ac063d870b8c8426ec81c1c3">value</a>() != 0)</div> |
| 1506 | <div class="line"><a name="l01408"></a><span class="lineno"> 1408</span>  {</div> |
| 1507 | <div class="line"><a name="l01409"></a><span class="lineno"> 1409</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has quantization dimension different than 0"</span>, __func__);</div> |
| 1508 | <div class="line"><a name="l01410"></a><span class="lineno"> 1410</span>  }</div> |
| 1509 | <div class="line"><a name="l01411"></a><span class="lineno"> 1411</span>  </div> |
| 1510 | <div class="line"><a name="l01412"></a><span class="lineno"> 1412</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* <span class="keyword">const</span> outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1511 | <div class="line"><a name="l01413"></a><span class="lineno"> 1413</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 1512 | <div class="line"><a name="l01414"></a><span class="lineno"> 1414</span>  {</div> |
| 1513 | <div class="line"><a name="l01415"></a><span class="lineno"> 1415</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid outputs"</span>, __func__);</div> |
| 1514 | <div class="line"><a name="l01416"></a><span class="lineno"> 1416</span>  }</div> |
| 1515 | <div class="line"><a name="l01417"></a><span class="lineno"> 1417</span>  </div> |
| 1516 | <div class="line"><a name="l01418"></a><span class="lineno"> 1418</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 1517 | <div class="line"><a name="l01419"></a><span class="lineno"> 1419</span>  </div> |
| 1518 | <div class="line"><a name="l01420"></a><span class="lineno"> 1420</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1519 | <div class="line"><a name="l01421"></a><span class="lineno"> 1421</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1520 | <div class="line"><a name="l01422"></a><span class="lineno"> 1422</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 1521 | <div class="line"><a name="l01423"></a><span class="lineno"> 1423</span>  {</div> |
| 1522 | <div class="line"><a name="l01424"></a><span class="lineno"> 1424</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1523 | <div class="line"><a name="l01425"></a><span class="lineno"> 1425</span>  IsDequantizeSupported,</div> |
| 1524 | <div class="line"><a name="l01426"></a><span class="lineno"> 1426</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1525 | <div class="line"><a name="l01427"></a><span class="lineno"> 1427</span>  isSupported,</div> |
| 1526 | <div class="line"><a name="l01428"></a><span class="lineno"> 1428</span>  setBackend,</div> |
| 1527 | <div class="line"><a name="l01429"></a><span class="lineno"> 1429</span>  inputInfo,</div> |
| 1528 | <div class="line"><a name="l01430"></a><span class="lineno"> 1430</span>  outputInfo);</div> |
| 1529 | <div class="line"><a name="l01431"></a><span class="lineno"> 1431</span>  };</div> |
| 1530 | <div class="line"><a name="l01432"></a><span class="lineno"> 1432</span>  </div> |
| 1531 | <div class="line"><a name="l01433"></a><span class="lineno"> 1433</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1532 | <div class="line"><a name="l01434"></a><span class="lineno"> 1434</span>  {</div> |
| 1533 | <div class="line"><a name="l01435"></a><span class="lineno"> 1435</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1534 | <div class="line"><a name="l01436"></a><span class="lineno"> 1436</span>  }</div> |
| 1535 | <div class="line"><a name="l01437"></a><span class="lineno"> 1437</span>  <span class="keywordflow">else</span></div> |
| 1536 | <div class="line"><a name="l01438"></a><span class="lineno"> 1438</span>  {</div> |
| 1537 | <div class="line"><a name="l01439"></a><span class="lineno"> 1439</span>  validateFunc(outputInfo, isSupported);</div> |
| 1538 | <div class="line"><a name="l01440"></a><span class="lineno"> 1440</span>  }</div> |
| 1539 | <div class="line"><a name="l01441"></a><span class="lineno"> 1441</span>  </div> |
| 1540 | <div class="line"><a name="l01442"></a><span class="lineno"> 1442</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1541 | <div class="line"><a name="l01443"></a><span class="lineno"> 1443</span>  {</div> |
| 1542 | <div class="line"><a name="l01444"></a><span class="lineno"> 1444</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1543 | <div class="line"><a name="l01445"></a><span class="lineno"> 1445</span>  }</div> |
| 1544 | <div class="line"><a name="l01446"></a><span class="lineno"> 1446</span>  </div> |
| 1545 | <div class="line"><a name="l01447"></a><span class="lineno"> 1447</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddDequantizeLayer();</div> |
| 1546 | <div class="line"><a name="l01448"></a><span class="lineno"> 1448</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1547 | <div class="line"><a name="l01449"></a><span class="lineno"> 1449</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 1548 | <div class="line"><a name="l01450"></a><span class="lineno"> 1450</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1549 | <div class="line"><a name="l01451"></a><span class="lineno"> 1451</span>  </div> |
| 1550 | <div class="line"><a name="l01452"></a><span class="lineno"> 1452</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 1551 | <div class="line"><a name="l01453"></a><span class="lineno"> 1453</span> }</div> |
| 1552 | <div class="line"><a name="l01454"></a><span class="lineno"> 1454</span>  </div> |
| 1553 | <div class="line"><a name="l01455"></a><span class="lineno"> 1455</span> <span class="keywordtype">bool</span> Converter::ConvertElementwiseBinary(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 1554 | <div class="line"><a name="l01456"></a><span class="lineno"> 1456</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 1555 | <div class="line"><a name="l01457"></a><span class="lineno"> 1457</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data,</div> |
| 1556 | <div class="line"><a name="l01458"></a><span class="lineno"> 1458</span>  <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047">armnn::BinaryOperation</a> binaryOperation)</div> |
| 1557 | <div class="line"><a name="l01459"></a><span class="lineno"> 1459</span> {</div> |
| 1558 | <div class="line"><a name="l01460"></a><span class="lineno"> 1460</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertElementwiseBinary()"</span>;</div> |
| 1559 | <div class="line"><a name="l01461"></a><span class="lineno"> 1461</span>  VLOG(DRIVER) << <span class="stringliteral">"binaryOperation = "</span> << <a class="code" href="namespacearmnn.html#aed346b74795fd613f727ca8369778704">GetBinaryOperationAsCString</a>(binaryOperation);</div> |
| 1560 | <div class="line"><a name="l01462"></a><span class="lineno"> 1462</span>  </div> |
| 1561 | <div class="line"><a name="l01463"></a><span class="lineno"> 1463</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1562 | <div class="line"><a name="l01464"></a><span class="lineno"> 1464</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input1 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 1563 | <div class="line"><a name="l01465"></a><span class="lineno"> 1465</span>  </div> |
| 1564 | <div class="line"><a name="l01466"></a><span class="lineno"> 1466</span>  <span class="keywordflow">if</span> (!input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() || !input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1565 | <div class="line"><a name="l01467"></a><span class="lineno"> 1467</span>  {</div> |
| 1566 | <div class="line"><a name="l01468"></a><span class="lineno"> 1468</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1567 | <div class="line"><a name="l01469"></a><span class="lineno"> 1469</span>  }</div> |
| 1568 | <div class="line"><a name="l01470"></a><span class="lineno"> 1470</span>  </div> |
| 1569 | <div class="line"><a name="l01471"></a><span class="lineno"> 1471</span>  <span class="comment">// The FuseActivation parameter is always the input index 2, and it should be optional</span></div> |
| 1570 | <div class="line"><a name="l01472"></a><span class="lineno"> 1472</span>  ActivationFn activationFunction;</div> |
| 1571 | <div class="line"><a name="l01473"></a><span class="lineno"> 1473</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#acd62ec697055015572a6ba61eaba4f1a">GetOptionalInputActivation</a>(operation, 2, activationFunction, model, data))</div> |
| 1572 | <div class="line"><a name="l01474"></a><span class="lineno"> 1474</span>  {</div> |
| 1573 | <div class="line"><a name="l01475"></a><span class="lineno"> 1475</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid optional input: activation function"</span>, __func__);</div> |
| 1574 | <div class="line"><a name="l01476"></a><span class="lineno"> 1476</span>  }</div> |
| 1575 | <div class="line"><a name="l01477"></a><span class="lineno"> 1477</span>  </div> |
| 1576 | <div class="line"><a name="l01478"></a><span class="lineno"> 1478</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1577 | <div class="line"><a name="l01479"></a><span class="lineno"> 1479</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1578 | <div class="line"><a name="l01480"></a><span class="lineno"> 1480</span>  {</div> |
| 1579 | <div class="line"><a name="l01481"></a><span class="lineno"> 1481</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output"</span>, __func__);</div> |
| 1580 | <div class="line"><a name="l01482"></a><span class="lineno"> 1482</span>  }</div> |
| 1581 | <div class="line"><a name="l01483"></a><span class="lineno"> 1483</span>  </div> |
| 1582 | <div class="line"><a name="l01484"></a><span class="lineno"> 1484</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1583 | <div class="line"><a name="l01485"></a><span class="lineno"> 1485</span>  </div> |
| 1584 | <div class="line"><a name="l01486"></a><span class="lineno"> 1486</span>  <a class="code" href="structarmnn_1_1_elementwise_binary_descriptor.html">armnn::ElementwiseBinaryDescriptor</a> descriptor(binaryOperation);</div> |
| 1585 | <div class="line"><a name="l01487"></a><span class="lineno"> 1487</span>  </div> |
| 1586 | <div class="line"><a name="l01488"></a><span class="lineno"> 1488</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1587 | <div class="line"><a name="l01489"></a><span class="lineno"> 1489</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 1588 | <div class="line"><a name="l01490"></a><span class="lineno"> 1490</span>  {</div> |
| 1589 | <div class="line"><a name="l01491"></a><span class="lineno"> 1491</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1590 | <div class="line"><a name="l01492"></a><span class="lineno"> 1492</span>  IsElementwiseBinarySupported,</div> |
| 1591 | <div class="line"><a name="l01493"></a><span class="lineno"> 1493</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1592 | <div class="line"><a name="l01494"></a><span class="lineno"> 1494</span>  isSupported,</div> |
| 1593 | <div class="line"><a name="l01495"></a><span class="lineno"> 1495</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a>(),</div> |
| 1594 | <div class="line"><a name="l01496"></a><span class="lineno"> 1496</span>  input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 1595 | <div class="line"><a name="l01497"></a><span class="lineno"> 1497</span>  input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 1596 | <div class="line"><a name="l01498"></a><span class="lineno"> 1498</span>  outputInfo,</div> |
| 1597 | <div class="line"><a name="l01499"></a><span class="lineno"> 1499</span>  binaryOperation);</div> |
| 1598 | <div class="line"><a name="l01500"></a><span class="lineno"> 1500</span>  };</div> |
| 1599 | <div class="line"><a name="l01501"></a><span class="lineno"> 1501</span>  </div> |
| 1600 | <div class="line"><a name="l01502"></a><span class="lineno"> 1502</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1601 | <div class="line"><a name="l01503"></a><span class="lineno"> 1503</span>  {</div> |
| 1602 | <div class="line"><a name="l01504"></a><span class="lineno"> 1504</span>  validateFunc(outputInfo, isSupported);</div> |
| 1603 | <div class="line"><a name="l01505"></a><span class="lineno"> 1505</span>  }</div> |
| 1604 | <div class="line"><a name="l01506"></a><span class="lineno"> 1506</span>  <span class="keywordflow">else</span></div> |
| 1605 | <div class="line"><a name="l01507"></a><span class="lineno"> 1507</span>  {</div> |
| 1606 | <div class="line"><a name="l01508"></a><span class="lineno"> 1508</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1607 | <div class="line"><a name="l01509"></a><span class="lineno"> 1509</span>  }</div> |
| 1608 | <div class="line"><a name="l01510"></a><span class="lineno"> 1510</span>  </div> |
| 1609 | <div class="line"><a name="l01511"></a><span class="lineno"> 1511</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1610 | <div class="line"><a name="l01512"></a><span class="lineno"> 1512</span>  {</div> |
| 1611 | <div class="line"><a name="l01513"></a><span class="lineno"> 1513</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1612 | <div class="line"><a name="l01514"></a><span class="lineno"> 1514</span>  }</div> |
| 1613 | <div class="line"><a name="l01515"></a><span class="lineno"> 1515</span>  </div> |
| 1614 | <div class="line"><a name="l01516"></a><span class="lineno"> 1516</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddElementwiseBinaryLayer(descriptor);</div> |
| 1615 | <div class="line"><a name="l01517"></a><span class="lineno"> 1517</span>  <span class="keywordflow">if</span> (!layer)</div> |
| 1616 | <div class="line"><a name="l01518"></a><span class="lineno"> 1518</span>  {</div> |
| 1617 | <div class="line"><a name="l01519"></a><span class="lineno"> 1519</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not add the ElementwiseBinaryLayer"</span>, __func__);</div> |
| 1618 | <div class="line"><a name="l01520"></a><span class="lineno"> 1520</span>  }</div> |
| 1619 | <div class="line"><a name="l01521"></a><span class="lineno"> 1521</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input0, input1, layer, data);</div> |
| 1620 | <div class="line"><a name="l01522"></a><span class="lineno"> 1522</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| 1621 | <div class="line"><a name="l01523"></a><span class="lineno"> 1523</span>  {</div> |
| 1622 | <div class="line"><a name="l01524"></a><span class="lineno"> 1524</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1623 | <div class="line"><a name="l01525"></a><span class="lineno"> 1525</span>  }</div> |
| 1624 | <div class="line"><a name="l01526"></a><span class="lineno"> 1526</span>  </div> |
| 1625 | <div class="line"><a name="l01527"></a><span class="lineno"> 1527</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model,</div> |
| 1626 | <div class="line"><a name="l01528"></a><span class="lineno"> 1528</span>  data, <span class="keyword">nullptr</span>, validateFunc, activationFunction);</div> |
| 1627 | <div class="line"><a name="l01529"></a><span class="lineno"> 1529</span> }</div> |
| 1628 | <div class="line"><a name="l01530"></a><span class="lineno"> 1530</span>  </div> |
| 1629 | <div class="line"><a name="l01531"></a><span class="lineno"> 1531</span> <span class="keywordtype">bool</span> Converter::ConvertElementwiseUnary(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 1630 | <div class="line"><a name="l01532"></a><span class="lineno"> 1532</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 1631 | <div class="line"><a name="l01533"></a><span class="lineno"> 1533</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data,</div> |
| 1632 | <div class="line"><a name="l01534"></a><span class="lineno"> 1534</span>  <a class="code" href="namespacearmnn.html#a1cfaa710db2a54673b21d2ea2da757c8">UnaryOperation</a> unaryOperation)</div> |
| 1633 | <div class="line"><a name="l01535"></a><span class="lineno"> 1535</span> {</div> |
| 1634 | <div class="line"><a name="l01536"></a><span class="lineno"> 1536</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertElementwiseUnary()"</span>;</div> |
| 1635 | <div class="line"><a name="l01537"></a><span class="lineno"> 1537</span>  VLOG(DRIVER) << <span class="stringliteral">"unaryOperation = "</span> << <a class="code" href="namespacearmnn.html#a6dac966f265381903c8ef4f392becced">GetUnaryOperationAsCString</a>(unaryOperation);</div> |
| 1636 | <div class="line"><a name="l01538"></a><span class="lineno"> 1538</span>  </div> |
| 1637 | <div class="line"><a name="l01539"></a><span class="lineno"> 1539</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1638 | <div class="line"><a name="l01540"></a><span class="lineno"> 1540</span>  </div> |
| 1639 | <div class="line"><a name="l01541"></a><span class="lineno"> 1541</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1640 | <div class="line"><a name="l01542"></a><span class="lineno"> 1542</span>  {</div> |
| 1641 | <div class="line"><a name="l01543"></a><span class="lineno"> 1543</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid input"</span>, __func__);</div> |
| 1642 | <div class="line"><a name="l01544"></a><span class="lineno"> 1544</span>  }</div> |
| 1643 | <div class="line"><a name="l01545"></a><span class="lineno"> 1545</span>  </div> |
| 1644 | <div class="line"><a name="l01546"></a><span class="lineno"> 1546</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1645 | <div class="line"><a name="l01547"></a><span class="lineno"> 1547</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1646 | <div class="line"><a name="l01548"></a><span class="lineno"> 1548</span>  {</div> |
| 1647 | <div class="line"><a name="l01549"></a><span class="lineno"> 1549</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 1648 | <div class="line"><a name="l01550"></a><span class="lineno"> 1550</span>  }</div> |
| 1649 | <div class="line"><a name="l01551"></a><span class="lineno"> 1551</span>  </div> |
| 1650 | <div class="line"><a name="l01552"></a><span class="lineno"> 1552</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1651 | <div class="line"><a name="l01553"></a><span class="lineno"> 1553</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1652 | <div class="line"><a name="l01554"></a><span class="lineno"> 1554</span>  </div> |
| 1653 | <div class="line"><a name="l01555"></a><span class="lineno"> 1555</span>  <a class="code" href="structarmnn_1_1_elementwise_unary_descriptor.html">ElementwiseUnaryDescriptor</a> descriptor(unaryOperation);</div> |
| 1654 | <div class="line"><a name="l01556"></a><span class="lineno"> 1556</span>  </div> |
| 1655 | <div class="line"><a name="l01557"></a><span class="lineno"> 1557</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1656 | <div class="line"><a name="l01558"></a><span class="lineno"> 1558</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1657 | <div class="line"><a name="l01559"></a><span class="lineno"> 1559</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 1658 | <div class="line"><a name="l01560"></a><span class="lineno"> 1560</span>  {</div> |
| 1659 | <div class="line"><a name="l01561"></a><span class="lineno"> 1561</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1660 | <div class="line"><a name="l01562"></a><span class="lineno"> 1562</span>  IsElementwiseUnarySupported,</div> |
| 1661 | <div class="line"><a name="l01563"></a><span class="lineno"> 1563</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1662 | <div class="line"><a name="l01564"></a><span class="lineno"> 1564</span>  isSupported,</div> |
| 1663 | <div class="line"><a name="l01565"></a><span class="lineno"> 1565</span>  setBackend,</div> |
| 1664 | <div class="line"><a name="l01566"></a><span class="lineno"> 1566</span>  inputInfo,</div> |
| 1665 | <div class="line"><a name="l01567"></a><span class="lineno"> 1567</span>  outputInfo,</div> |
| 1666 | <div class="line"><a name="l01568"></a><span class="lineno"> 1568</span>  descriptor);</div> |
| 1667 | <div class="line"><a name="l01569"></a><span class="lineno"> 1569</span>  };</div> |
| 1668 | <div class="line"><a name="l01570"></a><span class="lineno"> 1570</span>  </div> |
| 1669 | <div class="line"><a name="l01571"></a><span class="lineno"> 1571</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1670 | <div class="line"><a name="l01572"></a><span class="lineno"> 1572</span>  {</div> |
| 1671 | <div class="line"><a name="l01573"></a><span class="lineno"> 1573</span>  validateFunc(outputInfo, isSupported);</div> |
| 1672 | <div class="line"><a name="l01574"></a><span class="lineno"> 1574</span>  }</div> |
| 1673 | <div class="line"><a name="l01575"></a><span class="lineno"> 1575</span>  <span class="keywordflow">else</span></div> |
| 1674 | <div class="line"><a name="l01576"></a><span class="lineno"> 1576</span>  {</div> |
| 1675 | <div class="line"><a name="l01577"></a><span class="lineno"> 1577</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1676 | <div class="line"><a name="l01578"></a><span class="lineno"> 1578</span>  }</div> |
| 1677 | <div class="line"><a name="l01579"></a><span class="lineno"> 1579</span>  </div> |
| 1678 | <div class="line"><a name="l01580"></a><span class="lineno"> 1580</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1679 | <div class="line"><a name="l01581"></a><span class="lineno"> 1581</span>  {</div> |
| 1680 | <div class="line"><a name="l01582"></a><span class="lineno"> 1582</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1681 | <div class="line"><a name="l01583"></a><span class="lineno"> 1583</span>  }</div> |
| 1682 | <div class="line"><a name="l01584"></a><span class="lineno"> 1584</span>  </div> |
| 1683 | <div class="line"><a name="l01585"></a><span class="lineno"> 1585</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddElementwiseUnaryLayer(descriptor);</div> |
| 1684 | <div class="line"><a name="l01586"></a><span class="lineno"> 1586</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1685 | <div class="line"><a name="l01587"></a><span class="lineno"> 1587</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 1686 | <div class="line"><a name="l01588"></a><span class="lineno"> 1588</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1687 | <div class="line"><a name="l01589"></a><span class="lineno"> 1589</span>  </div> |
| 1688 | <div class="line"><a name="l01590"></a><span class="lineno"> 1590</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 1689 | <div class="line"><a name="l01591"></a><span class="lineno"> 1591</span> }</div> |
| 1690 | <div class="line"><a name="l01592"></a><span class="lineno"> 1592</span>  </div> |
| 1691 | <div class="line"><a name="l01593"></a><span class="lineno"> 1593</span> <span class="keywordtype">bool</span> Converter::ConvertElu(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1692 | <div class="line"><a name="l01594"></a><span class="lineno"> 1594</span> {</div> |
| 1693 | <div class="line"><a name="l01595"></a><span class="lineno"> 1595</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertElu()"</span>;</div> |
| 1694 | <div class="line"><a name="l01596"></a><span class="lineno"> 1596</span>  </div> |
| 1695 | <div class="line"><a name="l01597"></a><span class="lineno"> 1597</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1696 | <div class="line"><a name="l01598"></a><span class="lineno"> 1598</span>  <span class="keywordflow">if</span> (!input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1697 | <div class="line"><a name="l01599"></a><span class="lineno"> 1599</span>  {</div> |
| 1698 | <div class="line"><a name="l01600"></a><span class="lineno"> 1600</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1699 | <div class="line"><a name="l01601"></a><span class="lineno"> 1601</span>  }</div> |
| 1700 | <div class="line"><a name="l01602"></a><span class="lineno"> 1602</span>  </div> |
| 1701 | <div class="line"><a name="l01603"></a><span class="lineno"> 1603</span>  <span class="comment">// Determine data type of input tensor</span></div> |
| 1702 | <div class="line"><a name="l01604"></a><span class="lineno"> 1604</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> inputType;</div> |
| 1703 | <div class="line"><a name="l01605"></a><span class="lineno"> 1605</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 0, model, inputType))</div> |
| 1704 | <div class="line"><a name="l01606"></a><span class="lineno"> 1606</span>  {</div> |
| 1705 | <div class="line"><a name="l01607"></a><span class="lineno"> 1607</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1706 | <div class="line"><a name="l01608"></a><span class="lineno"> 1608</span>  }</div> |
| 1707 | <div class="line"><a name="l01609"></a><span class="lineno"> 1609</span>  </div> |
| 1708 | <div class="line"><a name="l01610"></a><span class="lineno"> 1610</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a> desc;</div> |
| 1709 | <div class="line"><a name="l01611"></a><span class="lineno"> 1611</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = ActivationFunction::Elu;</div> |
| 1710 | <div class="line"><a name="l01612"></a><span class="lineno"> 1612</span>  </div> |
| 1711 | <div class="line"><a name="l01613"></a><span class="lineno"> 1613</span>  <span class="comment">// Read alpha</span></div> |
| 1712 | <div class="line"><a name="l01614"></a><span class="lineno"> 1614</span>  <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT16)</div> |
| 1713 | <div class="line"><a name="l01615"></a><span class="lineno"> 1615</span>  {</div> |
| 1714 | <div class="line"><a name="l01616"></a><span class="lineno"> 1616</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> alpha;</div> |
| 1715 | <div class="line"><a name="l01617"></a><span class="lineno"> 1617</span>  </div> |
| 1716 | <div class="line"><a name="l01618"></a><span class="lineno"> 1618</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT16, alpha, model, data))</div> |
| 1717 | <div class="line"><a name="l01619"></a><span class="lineno"> 1619</span>  {</div> |
| 1718 | <div class="line"><a name="l01620"></a><span class="lineno"> 1620</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (FLOAT16)"</span>, __func__);</div> |
| 1719 | <div class="line"><a name="l01621"></a><span class="lineno"> 1621</span>  }</div> |
| 1720 | <div class="line"><a name="l01622"></a><span class="lineno"> 1622</span>  </div> |
| 1721 | <div class="line"><a name="l01623"></a><span class="lineno"> 1623</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a017b2990003a014234f13e999dc7c689">m_A</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(alpha);</div> |
| 1722 | <div class="line"><a name="l01624"></a><span class="lineno"> 1624</span>  }</div> |
| 1723 | <div class="line"><a name="l01625"></a><span class="lineno"> 1625</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT32)</div> |
| 1724 | <div class="line"><a name="l01626"></a><span class="lineno"> 1626</span>  {</div> |
| 1725 | <div class="line"><a name="l01627"></a><span class="lineno"> 1627</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT32, desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a017b2990003a014234f13e999dc7c689">m_A</a>, model, data))</div> |
| 1726 | <div class="line"><a name="l01628"></a><span class="lineno"> 1628</span>  {</div> |
| 1727 | <div class="line"><a name="l01629"></a><span class="lineno"> 1629</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (FLOAT32)"</span>, __func__);</div> |
| 1728 | <div class="line"><a name="l01630"></a><span class="lineno"> 1630</span>  }</div> |
| 1729 | <div class="line"><a name="l01631"></a><span class="lineno"> 1631</span>  }</div> |
| 1730 | <div class="line"><a name="l01632"></a><span class="lineno"> 1632</span>  <span class="keywordflow">else</span></div> |
| 1731 | <div class="line"><a name="l01633"></a><span class="lineno"> 1633</span>  {</div> |
| 1732 | <div class="line"><a name="l01634"></a><span class="lineno"> 1634</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported input tensor type: %d"</span>, __func__, inputType);</div> |
| 1733 | <div class="line"><a name="l01635"></a><span class="lineno"> 1635</span>  }</div> |
| 1734 | <div class="line"><a name="l01636"></a><span class="lineno"> 1636</span>  </div> |
| 1735 | <div class="line"><a name="l01637"></a><span class="lineno"> 1637</span>  <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">return ::ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 1736 | <div class="line"><a name="l01638"></a><span class="lineno"> 1638</span> }</div> |
| 1737 | <div class="line"><a name="l01639"></a><span class="lineno"> 1639</span>  </div> |
| 1738 | <div class="line"><a name="l01640"></a><span class="lineno"> 1640</span> <span class="keywordtype">bool</span> Converter::ConvertExpandDims(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1739 | <div class="line"><a name="l01641"></a><span class="lineno"> 1641</span> {</div> |
| 1740 | <div class="line"><a name="l01642"></a><span class="lineno"> 1642</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertExpandDims()"</span>;</div> |
| 1741 | <div class="line"><a name="l01643"></a><span class="lineno"> 1643</span>  </div> |
| 1742 | <div class="line"><a name="l01644"></a><span class="lineno"> 1644</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1743 | <div class="line"><a name="l01645"></a><span class="lineno"> 1645</span>  </div> |
| 1744 | <div class="line"><a name="l01646"></a><span class="lineno"> 1646</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1745 | <div class="line"><a name="l01647"></a><span class="lineno"> 1647</span>  {</div> |
| 1746 | <div class="line"><a name="l01648"></a><span class="lineno"> 1648</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid input"</span>, __func__);</div> |
| 1747 | <div class="line"><a name="l01649"></a><span class="lineno"> 1649</span>  }</div> |
| 1748 | <div class="line"><a name="l01650"></a><span class="lineno"> 1650</span>  </div> |
| 1749 | <div class="line"><a name="l01651"></a><span class="lineno"> 1651</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1750 | <div class="line"><a name="l01652"></a><span class="lineno"> 1652</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1751 | <div class="line"><a name="l01653"></a><span class="lineno"> 1653</span>  {</div> |
| 1752 | <div class="line"><a name="l01654"></a><span class="lineno"> 1654</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid output"</span>, __func__);</div> |
| 1753 | <div class="line"><a name="l01655"></a><span class="lineno"> 1655</span>  }</div> |
| 1754 | <div class="line"><a name="l01656"></a><span class="lineno"> 1656</span>  </div> |
| 1755 | <div class="line"><a name="l01657"></a><span class="lineno"> 1657</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1756 | <div class="line"><a name="l01658"></a><span class="lineno"> 1658</span>  </div> |
| 1757 | <div class="line"><a name="l01659"></a><span class="lineno"> 1659</span>  int32_t axis;</div> |
| 1758 | <div class="line"><a name="l01660"></a><span class="lineno"> 1660</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, axis, model, data))</div> |
| 1759 | <div class="line"><a name="l01661"></a><span class="lineno"> 1661</span>  {</div> |
| 1760 | <div class="line"><a name="l01662"></a><span class="lineno"> 1662</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: failed to get axis input value"</span>, __func__);</div> |
| 1761 | <div class="line"><a name="l01663"></a><span class="lineno"> 1663</span>  }</div> |
| 1762 | <div class="line"><a name="l01664"></a><span class="lineno"> 1664</span>  </div> |
| 1763 | <div class="line"><a name="l01665"></a><span class="lineno"> 1665</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> targetShape;</div> |
| 1764 | <div class="line"><a name="l01666"></a><span class="lineno"> 1666</span>  </div> |
| 1765 | <div class="line"><a name="l01667"></a><span class="lineno"> 1667</span>  <span class="keywordflow">try</span></div> |
| 1766 | <div class="line"><a name="l01668"></a><span class="lineno"> 1668</span>  {</div> |
| 1767 | <div class="line"><a name="l01669"></a><span class="lineno"> 1669</span>  targetShape = <a class="code" href="namespacearmnn_utils.html#a0d3b1be320610515e0cac8d745d9f8c2">armnnUtils::ExpandDims</a>(input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>().<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>(), axis);</div> |
| 1768 | <div class="line"><a name="l01670"></a><span class="lineno"> 1670</span>  }</div> |
| 1769 | <div class="line"><a name="l01671"></a><span class="lineno"> 1671</span>  <span class="keywordflow">catch</span> (<span class="keyword">const</span> std::exception& e)</div> |
| 1770 | <div class="line"><a name="l01672"></a><span class="lineno"> 1672</span>  {</div> |
| 1771 | <div class="line"><a name="l01673"></a><span class="lineno"> 1673</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: %s"</span>, __func__, e.what());</div> |
| 1772 | <div class="line"><a name="l01674"></a><span class="lineno"> 1674</span>  }</div> |
| 1773 | <div class="line"><a name="l01675"></a><span class="lineno"> 1675</span>  </div> |
| 1774 | <div class="line"><a name="l01676"></a><span class="lineno"> 1676</span>  <a class="code" href="structarmnn_1_1_reshape_descriptor.html">ReshapeDescriptor</a> reshapeDescriptor;</div> |
| 1775 | <div class="line"><a name="l01677"></a><span class="lineno"> 1677</span>  reshapeDescriptor.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a> = targetShape;</div> |
| 1776 | <div class="line"><a name="l01678"></a><span class="lineno"> 1678</span>  </div> |
| 1777 | <div class="line"><a name="l01679"></a><span class="lineno"> 1679</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1778 | <div class="line"><a name="l01680"></a><span class="lineno"> 1680</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1779 | <div class="line"><a name="l01681"></a><span class="lineno"> 1681</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 1780 | <div class="line"><a name="l01682"></a><span class="lineno"> 1682</span>  {</div> |
| 1781 | <div class="line"><a name="l01683"></a><span class="lineno"> 1683</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1782 | <div class="line"><a name="l01684"></a><span class="lineno"> 1684</span>  IsReshapeSupported,</div> |
| 1783 | <div class="line"><a name="l01685"></a><span class="lineno"> 1685</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1784 | <div class="line"><a name="l01686"></a><span class="lineno"> 1686</span>  isSupported,</div> |
| 1785 | <div class="line"><a name="l01687"></a><span class="lineno"> 1687</span>  setBackend,</div> |
| 1786 | <div class="line"><a name="l01688"></a><span class="lineno"> 1688</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 1787 | <div class="line"><a name="l01689"></a><span class="lineno"> 1689</span>  outputInfo,</div> |
| 1788 | <div class="line"><a name="l01690"></a><span class="lineno"> 1690</span>  reshapeDescriptor);</div> |
| 1789 | <div class="line"><a name="l01691"></a><span class="lineno"> 1691</span>  };</div> |
| 1790 | <div class="line"><a name="l01692"></a><span class="lineno"> 1692</span>  </div> |
| 1791 | <div class="line"><a name="l01693"></a><span class="lineno"> 1693</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1792 | <div class="line"><a name="l01694"></a><span class="lineno"> 1694</span>  {</div> |
| 1793 | <div class="line"><a name="l01695"></a><span class="lineno"> 1695</span>  <span class="keywordflow">if</span> (targetShape != outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>())</div> |
| 1794 | <div class="line"><a name="l01696"></a><span class="lineno"> 1696</span>  {</div> |
| 1795 | <div class="line"><a name="l01697"></a><span class="lineno"> 1697</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Shape of the output operand does not match the resolved expanded shape"</span>, __func__);</div> |
| 1796 | <div class="line"><a name="l01698"></a><span class="lineno"> 1698</span>  }</div> |
| 1797 | <div class="line"><a name="l01699"></a><span class="lineno"> 1699</span>  validateFunc(outputInfo, isSupported);</div> |
| 1798 | <div class="line"><a name="l01700"></a><span class="lineno"> 1700</span>  }</div> |
| 1799 | <div class="line"><a name="l01701"></a><span class="lineno"> 1701</span>  <span class="keywordflow">else</span></div> |
| 1800 | <div class="line"><a name="l01702"></a><span class="lineno"> 1702</span>  {</div> |
| 1801 | <div class="line"><a name="l01703"></a><span class="lineno"> 1703</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1802 | <div class="line"><a name="l01704"></a><span class="lineno"> 1704</span>  }</div> |
| 1803 | <div class="line"><a name="l01705"></a><span class="lineno"> 1705</span>  </div> |
| 1804 | <div class="line"><a name="l01706"></a><span class="lineno"> 1706</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1805 | <div class="line"><a name="l01707"></a><span class="lineno"> 1707</span>  {</div> |
| 1806 | <div class="line"><a name="l01708"></a><span class="lineno"> 1708</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1807 | <div class="line"><a name="l01709"></a><span class="lineno"> 1709</span>  }</div> |
| 1808 | <div class="line"><a name="l01710"></a><span class="lineno"> 1710</span>  </div> |
| 1809 | <div class="line"><a name="l01711"></a><span class="lineno"> 1711</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddReshapeLayer(reshapeDescriptor);</div> |
| 1810 | <div class="line"><a name="l01712"></a><span class="lineno"> 1712</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1811 | <div class="line"><a name="l01713"></a><span class="lineno"> 1713</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 1812 | <div class="line"><a name="l01714"></a><span class="lineno"> 1714</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1813 | <div class="line"><a name="l01715"></a><span class="lineno"> 1715</span>  </div> |
| 1814 | <div class="line"><a name="l01716"></a><span class="lineno"> 1716</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 1815 | <div class="line"><a name="l01717"></a><span class="lineno"> 1717</span> }</div> |
| 1816 | <div class="line"><a name="l01718"></a><span class="lineno"> 1718</span>  </div> |
| 1817 | <div class="line"><a name="l01719"></a><span class="lineno"> 1719</span> <span class="keywordtype">bool</span> Converter::ConvertFill(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1818 | <div class="line"><a name="l01720"></a><span class="lineno"> 1720</span> {</div> |
| 1819 | <div class="line"><a name="l01721"></a><span class="lineno"> 1721</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertFill()"</span>;</div> |
| 1820 | <div class="line"><a name="l01722"></a><span class="lineno"> 1722</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1821 | <div class="line"><a name="l01723"></a><span class="lineno"> 1723</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1822 | <div class="line"><a name="l01724"></a><span class="lineno"> 1724</span>  {</div> |
| 1823 | <div class="line"><a name="l01725"></a><span class="lineno"> 1725</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1824 | <div class="line"><a name="l01726"></a><span class="lineno"> 1726</span>  }</div> |
| 1825 | <div class="line"><a name="l01727"></a><span class="lineno"> 1727</span>  </div> |
| 1826 | <div class="line"><a name="l01728"></a><span class="lineno"> 1728</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1827 | <div class="line"><a name="l01729"></a><span class="lineno"> 1729</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1828 | <div class="line"><a name="l01730"></a><span class="lineno"> 1730</span>  {</div> |
| 1829 | <div class="line"><a name="l01731"></a><span class="lineno"> 1731</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output"</span>, __func__);</div> |
| 1830 | <div class="line"><a name="l01732"></a><span class="lineno"> 1732</span>  }</div> |
| 1831 | <div class="line"><a name="l01733"></a><span class="lineno"> 1733</span>  </div> |
| 1832 | <div class="line"><a name="l01734"></a><span class="lineno"> 1734</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1833 | <div class="line"><a name="l01735"></a><span class="lineno"> 1735</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1834 | <div class="line"><a name="l01736"></a><span class="lineno"> 1736</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1835 | <div class="line"><a name="l01737"></a><span class="lineno"> 1737</span>  {</div> |
| 1836 | <div class="line"><a name="l01738"></a><span class="lineno"> 1738</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Dynamic output tensors are not supported"</span>, __func__);</div> |
| 1837 | <div class="line"><a name="l01739"></a><span class="lineno"> 1739</span>  }</div> |
| 1838 | <div class="line"><a name="l01740"></a><span class="lineno"> 1740</span>  </div> |
| 1839 | <div class="line"><a name="l01741"></a><span class="lineno"> 1741</span>  <span class="comment">// Determine data type of output tensor</span></div> |
| 1840 | <div class="line"><a name="l01742"></a><span class="lineno"> 1742</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> outputType = output->type;</div> |
| 1841 | <div class="line"><a name="l01743"></a><span class="lineno"> 1743</span>  <a class="code" href="structarmnn_1_1_fill_descriptor.html">FillDescriptor</a> descriptor;</div> |
| 1842 | <div class="line"><a name="l01744"></a><span class="lineno"> 1744</span>  <span class="comment">// Read the scalar fill value</span></div> |
| 1843 | <div class="line"><a name="l01745"></a><span class="lineno"> 1745</span>  <span class="keywordflow">if</span> (outputType == OperandType::TENSOR_FLOAT16)</div> |
| 1844 | <div class="line"><a name="l01746"></a><span class="lineno"> 1746</span>  {</div> |
| 1845 | <div class="line"><a name="l01747"></a><span class="lineno"> 1747</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> value;</div> |
| 1846 | <div class="line"><a name="l01748"></a><span class="lineno"> 1748</span>  </div> |
| 1847 | <div class="line"><a name="l01749"></a><span class="lineno"> 1749</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT16, value, model, data))</div> |
| 1848 | <div class="line"><a name="l01750"></a><span class="lineno"> 1750</span>  {</div> |
| 1849 | <div class="line"><a name="l01751"></a><span class="lineno"> 1751</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs %d"</span>, __func__, outputType);</div> |
| 1850 | <div class="line"><a name="l01752"></a><span class="lineno"> 1752</span>  }</div> |
| 1851 | <div class="line"><a name="l01753"></a><span class="lineno"> 1753</span>  </div> |
| 1852 | <div class="line"><a name="l01754"></a><span class="lineno"> 1754</span>  descriptor.<a class="code" href="structarmnn_1_1_fill_descriptor.html#ab3ebc5cf4a617d43371a4cb7fecdeb32">m_Value</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(value);</div> |
| 1853 | <div class="line"><a name="l01755"></a><span class="lineno"> 1755</span>  }</div> |
| 1854 | <div class="line"><a name="l01756"></a><span class="lineno"> 1756</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (outputType == OperandType::TENSOR_FLOAT32)</div> |
| 1855 | <div class="line"><a name="l01757"></a><span class="lineno"> 1757</span>  {</div> |
| 1856 | <div class="line"><a name="l01758"></a><span class="lineno"> 1758</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT32, descriptor.<a class="code" href="structarmnn_1_1_fill_descriptor.html#ab3ebc5cf4a617d43371a4cb7fecdeb32">m_Value</a>, model, data))</div> |
| 1857 | <div class="line"><a name="l01759"></a><span class="lineno"> 1759</span>  {</div> |
| 1858 | <div class="line"><a name="l01760"></a><span class="lineno"> 1760</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs %d"</span>, __func__, outputType);</div> |
| 1859 | <div class="line"><a name="l01761"></a><span class="lineno"> 1761</span>  }</div> |
| 1860 | <div class="line"><a name="l01762"></a><span class="lineno"> 1762</span>  }</div> |
| 1861 | <div class="line"><a name="l01763"></a><span class="lineno"> 1763</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (outputType == OperandType::TENSOR_INT32)</div> |
| 1862 | <div class="line"><a name="l01764"></a><span class="lineno"> 1764</span>  {</div> |
| 1863 | <div class="line"><a name="l01765"></a><span class="lineno"> 1765</span>  int32_t value;</div> |
| 1864 | <div class="line"><a name="l01766"></a><span class="lineno"> 1766</span>  </div> |
| 1865 | <div class="line"><a name="l01767"></a><span class="lineno"> 1767</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, value, model, data))</div> |
| 1866 | <div class="line"><a name="l01768"></a><span class="lineno"> 1768</span>  {</div> |
| 1867 | <div class="line"><a name="l01769"></a><span class="lineno"> 1769</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs %d"</span>, __func__, outputType);</div> |
| 1868 | <div class="line"><a name="l01770"></a><span class="lineno"> 1770</span>  }</div> |
| 1869 | <div class="line"><a name="l01771"></a><span class="lineno"> 1771</span>  </div> |
| 1870 | <div class="line"><a name="l01772"></a><span class="lineno"> 1772</span>  descriptor.<a class="code" href="structarmnn_1_1_fill_descriptor.html#ab3ebc5cf4a617d43371a4cb7fecdeb32">m_Value</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(value);</div> |
| 1871 | <div class="line"><a name="l01773"></a><span class="lineno"> 1773</span>  }</div> |
| 1872 | <div class="line"><a name="l01774"></a><span class="lineno"> 1774</span>  <span class="keywordflow">else</span></div> |
| 1873 | <div class="line"><a name="l01775"></a><span class="lineno"> 1775</span>  {</div> |
| 1874 | <div class="line"><a name="l01776"></a><span class="lineno"> 1776</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported input tensor type: %d"</span>, __func__, outputType);</div> |
| 1875 | <div class="line"><a name="l01777"></a><span class="lineno"> 1777</span>  }</div> |
| 1876 | <div class="line"><a name="l01778"></a><span class="lineno"> 1778</span>  </div> |
| 1877 | <div class="line"><a name="l01779"></a><span class="lineno"> 1779</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1878 | <div class="line"><a name="l01780"></a><span class="lineno"> 1780</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1879 | <div class="line"><a name="l01781"></a><span class="lineno"> 1781</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1880 | <div class="line"><a name="l01782"></a><span class="lineno"> 1782</span>  IsFillSupported,</div> |
| 1881 | <div class="line"><a name="l01783"></a><span class="lineno"> 1783</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1882 | <div class="line"><a name="l01784"></a><span class="lineno"> 1784</span>  isSupported,</div> |
| 1883 | <div class="line"><a name="l01785"></a><span class="lineno"> 1785</span>  setBackend,</div> |
| 1884 | <div class="line"><a name="l01786"></a><span class="lineno"> 1786</span>  inputInfo,</div> |
| 1885 | <div class="line"><a name="l01787"></a><span class="lineno"> 1787</span>  outputInfo,</div> |
| 1886 | <div class="line"><a name="l01788"></a><span class="lineno"> 1788</span>  descriptor);</div> |
| 1887 | <div class="line"><a name="l01789"></a><span class="lineno"> 1789</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1888 | <div class="line"><a name="l01790"></a><span class="lineno"> 1790</span>  {</div> |
| 1889 | <div class="line"><a name="l01791"></a><span class="lineno"> 1791</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1890 | <div class="line"><a name="l01792"></a><span class="lineno"> 1792</span>  }</div> |
| 1891 | <div class="line"><a name="l01793"></a><span class="lineno"> 1793</span>  </div> |
| 1892 | <div class="line"><a name="l01794"></a><span class="lineno"> 1794</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddFillLayer(descriptor);</div> |
| 1893 | <div class="line"><a name="l01795"></a><span class="lineno"> 1795</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1894 | <div class="line"><a name="l01796"></a><span class="lineno"> 1796</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 1895 | <div class="line"><a name="l01797"></a><span class="lineno"> 1797</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1896 | <div class="line"><a name="l01798"></a><span class="lineno"> 1798</span>  </div> |
| 1897 | <div class="line"><a name="l01799"></a><span class="lineno"> 1799</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data);</div> |
| 1898 | <div class="line"><a name="l01800"></a><span class="lineno"> 1800</span> }</div> |
| 1899 | <div class="line"><a name="l01801"></a><span class="lineno"> 1801</span>  </div> |
| 1900 | <div class="line"><a name="l01802"></a><span class="lineno"> 1802</span> <span class="keywordtype">bool</span> Converter::ConvertFloor(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1901 | <div class="line"><a name="l01803"></a><span class="lineno"> 1803</span> {</div> |
| 1902 | <div class="line"><a name="l01804"></a><span class="lineno"> 1804</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertFloor()"</span>;</div> |
| 1903 | <div class="line"><a name="l01805"></a><span class="lineno"> 1805</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1904 | <div class="line"><a name="l01806"></a><span class="lineno"> 1806</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1905 | <div class="line"><a name="l01807"></a><span class="lineno"> 1807</span>  {</div> |
| 1906 | <div class="line"><a name="l01808"></a><span class="lineno"> 1808</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1907 | <div class="line"><a name="l01809"></a><span class="lineno"> 1809</span>  }</div> |
| 1908 | <div class="line"><a name="l01810"></a><span class="lineno"> 1810</span>  </div> |
| 1909 | <div class="line"><a name="l01811"></a><span class="lineno"> 1811</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* <span class="keyword">const</span> outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1910 | <div class="line"><a name="l01812"></a><span class="lineno"> 1812</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 1911 | <div class="line"><a name="l01813"></a><span class="lineno"> 1813</span>  {</div> |
| 1912 | <div class="line"><a name="l01814"></a><span class="lineno"> 1814</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid outputs"</span>, __func__);</div> |
| 1913 | <div class="line"><a name="l01815"></a><span class="lineno"> 1815</span>  }</div> |
| 1914 | <div class="line"><a name="l01816"></a><span class="lineno"> 1816</span>  </div> |
| 1915 | <div class="line"><a name="l01817"></a><span class="lineno"> 1817</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 1916 | <div class="line"><a name="l01818"></a><span class="lineno"> 1818</span>  </div> |
| 1917 | <div class="line"><a name="l01819"></a><span class="lineno"> 1819</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 1918 | <div class="line"><a name="l01820"></a><span class="lineno"> 1820</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 1919 | <div class="line"><a name="l01821"></a><span class="lineno"> 1821</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 1920 | <div class="line"><a name="l01822"></a><span class="lineno"> 1822</span>  {</div> |
| 1921 | <div class="line"><a name="l01823"></a><span class="lineno"> 1823</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 1922 | <div class="line"><a name="l01824"></a><span class="lineno"> 1824</span>  IsFloorSupported,</div> |
| 1923 | <div class="line"><a name="l01825"></a><span class="lineno"> 1825</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 1924 | <div class="line"><a name="l01826"></a><span class="lineno"> 1826</span>  isSupported,</div> |
| 1925 | <div class="line"><a name="l01827"></a><span class="lineno"> 1827</span>  setBackend,</div> |
| 1926 | <div class="line"><a name="l01828"></a><span class="lineno"> 1828</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 1927 | <div class="line"><a name="l01829"></a><span class="lineno"> 1829</span>  outputInfo);</div> |
| 1928 | <div class="line"><a name="l01830"></a><span class="lineno"> 1830</span>  };</div> |
| 1929 | <div class="line"><a name="l01831"></a><span class="lineno"> 1831</span>  </div> |
| 1930 | <div class="line"><a name="l01832"></a><span class="lineno"> 1832</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 1931 | <div class="line"><a name="l01833"></a><span class="lineno"> 1833</span>  {</div> |
| 1932 | <div class="line"><a name="l01834"></a><span class="lineno"> 1834</span>  validateFunc(outputInfo, isSupported);</div> |
| 1933 | <div class="line"><a name="l01835"></a><span class="lineno"> 1835</span>  }</div> |
| 1934 | <div class="line"><a name="l01836"></a><span class="lineno"> 1836</span>  <span class="keywordflow">else</span></div> |
| 1935 | <div class="line"><a name="l01837"></a><span class="lineno"> 1837</span>  {</div> |
| 1936 | <div class="line"><a name="l01838"></a><span class="lineno"> 1838</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 1937 | <div class="line"><a name="l01839"></a><span class="lineno"> 1839</span>  }</div> |
| 1938 | <div class="line"><a name="l01840"></a><span class="lineno"> 1840</span>  </div> |
| 1939 | <div class="line"><a name="l01841"></a><span class="lineno"> 1841</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 1940 | <div class="line"><a name="l01842"></a><span class="lineno"> 1842</span>  {</div> |
| 1941 | <div class="line"><a name="l01843"></a><span class="lineno"> 1843</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 1942 | <div class="line"><a name="l01844"></a><span class="lineno"> 1844</span>  }</div> |
| 1943 | <div class="line"><a name="l01845"></a><span class="lineno"> 1845</span>  </div> |
| 1944 | <div class="line"><a name="l01846"></a><span class="lineno"> 1846</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddFloorLayer();</div> |
| 1945 | <div class="line"><a name="l01847"></a><span class="lineno"> 1847</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 1946 | <div class="line"><a name="l01848"></a><span class="lineno"> 1848</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 1947 | <div class="line"><a name="l01849"></a><span class="lineno"> 1849</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 1948 | <div class="line"><a name="l01850"></a><span class="lineno"> 1850</span>  </div> |
| 1949 | <div class="line"><a name="l01851"></a><span class="lineno"> 1851</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 1950 | <div class="line"><a name="l01852"></a><span class="lineno"> 1852</span> }</div> |
| 1951 | <div class="line"><a name="l01853"></a><span class="lineno"> 1853</span>  </div> |
| 1952 | <div class="line"><a name="l01854"></a><span class="lineno"> 1854</span> <span class="keywordtype">bool</span> Converter::ConvertFullyConnected(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 1953 | <div class="line"><a name="l01855"></a><span class="lineno"> 1855</span> {</div> |
| 1954 | <div class="line"><a name="l01856"></a><span class="lineno"> 1856</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertFullyConnected()"</span>;</div> |
| 1955 | <div class="line"><a name="l01857"></a><span class="lineno"> 1857</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 1956 | <div class="line"><a name="l01858"></a><span class="lineno"> 1858</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1957 | <div class="line"><a name="l01859"></a><span class="lineno"> 1859</span>  {</div> |
| 1958 | <div class="line"><a name="l01860"></a><span class="lineno"> 1860</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1959 | <div class="line"><a name="l01861"></a><span class="lineno"> 1861</span>  }</div> |
| 1960 | <div class="line"><a name="l01862"></a><span class="lineno"> 1862</span>  </div> |
| 1961 | <div class="line"><a name="l01863"></a><span class="lineno"> 1863</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 1962 | <div class="line"><a name="l01864"></a><span class="lineno"> 1864</span>  <span class="keywordflow">if</span> (!output)</div> |
| 1963 | <div class="line"><a name="l01865"></a><span class="lineno"> 1865</span>  {</div> |
| 1964 | <div class="line"><a name="l01866"></a><span class="lineno"> 1866</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 1965 | <div class="line"><a name="l01867"></a><span class="lineno"> 1867</span>  }</div> |
| 1966 | <div class="line"><a name="l01868"></a><span class="lineno"> 1868</span>  </div> |
| 1967 | <div class="line"><a name="l01869"></a><span class="lineno"> 1869</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 1968 | <div class="line"><a name="l01870"></a><span class="lineno"> 1870</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 1969 | <div class="line"><a name="l01871"></a><span class="lineno"> 1871</span>  </div> |
| 1970 | <div class="line"><a name="l01872"></a><span class="lineno"> 1872</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> weightsInput = <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a>();</div> |
| 1971 | <div class="line"><a name="l01873"></a><span class="lineno"> 1873</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* weightsOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 1972 | <div class="line"><a name="l01874"></a><span class="lineno"> 1874</span>  <span class="keywordflow">if</span> (!weightsOperand)</div> |
| 1973 | <div class="line"><a name="l01875"></a><span class="lineno"> 1875</span>  {</div> |
| 1974 | <div class="line"><a name="l01876"></a><span class="lineno"> 1876</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read weights"</span>, __func__);</div> |
| 1975 | <div class="line"><a name="l01877"></a><span class="lineno"> 1877</span>  }</div> |
| 1976 | <div class="line"><a name="l01878"></a><span class="lineno"> 1878</span>  </div> |
| 1977 | <div class="line"><a name="l01879"></a><span class="lineno"> 1879</span>  <span class="comment">// If weights are constant a separate constant layer will be created to store data.</span></div> |
| 1978 | <div class="line"><a name="l01880"></a><span class="lineno"> 1880</span>  <span class="comment">// Otherwise handle non const weights as inputs.</span></div> |
| 1979 | <div class="line"><a name="l01881"></a><span class="lineno"> 1881</span>  weightsInput = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 1980 | <div class="line"><a name="l01882"></a><span class="lineno"> 1882</span>  <span class="keywordflow">if</span> (!weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1981 | <div class="line"><a name="l01883"></a><span class="lineno"> 1883</span>  {</div> |
| 1982 | <div class="line"><a name="l01884"></a><span class="lineno"> 1884</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1983 | <div class="line"><a name="l01885"></a><span class="lineno"> 1885</span>  }</div> |
| 1984 | <div class="line"><a name="l01886"></a><span class="lineno"> 1886</span>  </div> |
| 1985 | <div class="line"><a name="l01887"></a><span class="lineno"> 1887</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> biasInput = <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a>();</div> |
| 1986 | <div class="line"><a name="l01888"></a><span class="lineno"> 1888</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* biasOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 2, model);</div> |
| 1987 | <div class="line"><a name="l01889"></a><span class="lineno"> 1889</span>  <span class="keywordflow">if</span> (!biasOperand)</div> |
| 1988 | <div class="line"><a name="l01890"></a><span class="lineno"> 1890</span>  {</div> |
| 1989 | <div class="line"><a name="l01891"></a><span class="lineno"> 1891</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read bias"</span>, __func__);</div> |
| 1990 | <div class="line"><a name="l01892"></a><span class="lineno"> 1892</span>  }</div> |
| 1991 | <div class="line"><a name="l01893"></a><span class="lineno"> 1893</span>  </div> |
| 1992 | <div class="line"><a name="l01894"></a><span class="lineno"> 1894</span>  <span class="comment">// If bias are constant a separate constant layer will be created to store data.</span></div> |
| 1993 | <div class="line"><a name="l01895"></a><span class="lineno"> 1895</span>  <span class="comment">// Otherwise handle non const bias as inputs.</span></div> |
| 1994 | <div class="line"><a name="l01896"></a><span class="lineno"> 1896</span>  biasInput = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 2, model, data); <span class="comment">// 1D</span></div> |
| 1995 | <div class="line"><a name="l01897"></a><span class="lineno"> 1897</span>  <span class="keywordflow">if</span> (!biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 1996 | <div class="line"><a name="l01898"></a><span class="lineno"> 1898</span>  {</div> |
| 1997 | <div class="line"><a name="l01899"></a><span class="lineno"> 1899</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 1998 | <div class="line"><a name="l01900"></a><span class="lineno"> 1900</span>  }</div> |
| 1999 | <div class="line"><a name="l01901"></a><span class="lineno"> 1901</span>  </div> |
| 2000 | <div class="line"><a name="l01902"></a><span class="lineno"> 1902</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> weightsInfo = weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2001 | <div class="line"><a name="l01903"></a><span class="lineno"> 1903</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> reshapedInfo = inputInfo;</div> |
| 2002 | <div class="line"><a name="l01904"></a><span class="lineno"> 1904</span>  <span class="keywordflow">try</span></div> |
| 2003 | <div class="line"><a name="l01905"></a><span class="lineno"> 1905</span>  {</div> |
| 2004 | <div class="line"><a name="l01906"></a><span class="lineno"> 1906</span>  reshapedInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(FlattenFullyConnectedInput(inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>(), weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()));</div> |
| 2005 | <div class="line"><a name="l01907"></a><span class="lineno"> 1907</span>  }</div> |
| 2006 | <div class="line"><a name="l01908"></a><span class="lineno"> 1908</span>  <span class="keywordflow">catch</span> (<span class="keyword">const</span> std::exception& e)</div> |
| 2007 | <div class="line"><a name="l01909"></a><span class="lineno"> 1909</span>  {</div> |
| 2008 | <div class="line"><a name="l01910"></a><span class="lineno"> 1910</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: %s"</span>, __func__, e.what());</div> |
| 2009 | <div class="line"><a name="l01911"></a><span class="lineno"> 1911</span>  }</div> |
| 2010 | <div class="line"><a name="l01912"></a><span class="lineno"> 1912</span>  </div> |
| 2011 | <div class="line"><a name="l01913"></a><span class="lineno"> 1913</span>  <span class="comment">// Ensuring that the bias value is within 1% of the weights input (small float differences can exist)</span></div> |
| 2012 | <div class="line"><a name="l01914"></a><span class="lineno"> 1914</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> biasInfo = biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2013 | <div class="line"><a name="l01915"></a><span class="lineno"> 1915</span>  SanitizeBiasQuantizationScale(biasInfo, weightsInfo, reshapedInfo);</div> |
| 2014 | <div class="line"><a name="l01916"></a><span class="lineno"> 1916</span>  </div> |
| 2015 | <div class="line"><a name="l01917"></a><span class="lineno"> 1917</span>  ActivationFn activationFunction;</div> |
| 2016 | <div class="line"><a name="l01918"></a><span class="lineno"> 1918</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 3, activationFunction, model, data))</div> |
| 2017 | <div class="line"><a name="l01919"></a><span class="lineno"> 1919</span>  {</div> |
| 2018 | <div class="line"><a name="l01920"></a><span class="lineno"> 1920</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2019 | <div class="line"><a name="l01921"></a><span class="lineno"> 1921</span>  }</div> |
| 2020 | <div class="line"><a name="l01922"></a><span class="lineno"> 1922</span>  </div> |
| 2021 | <div class="line"><a name="l01923"></a><span class="lineno"> 1923</span>  <a class="code" href="structarmnn_1_1_fully_connected_descriptor.html">armnn::FullyConnectedDescriptor</a> desc;</div> |
| 2022 | <div class="line"><a name="l01924"></a><span class="lineno"> 1924</span>  desc.<a class="code" href="structarmnn_1_1_fully_connected_descriptor.html#a281fcaec86e17c97f7b8402633f6b55a">m_TransposeWeightMatrix</a> = <span class="keyword">true</span>;</div> |
| 2023 | <div class="line"><a name="l01925"></a><span class="lineno"> 1925</span>  desc.<a class="code" href="structarmnn_1_1_fully_connected_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">m_BiasEnabled</a> = <span class="keyword">true</span>;</div> |
| 2024 | <div class="line"><a name="l01926"></a><span class="lineno"> 1926</span>  desc.<a class="code" href="structarmnn_1_1_fully_connected_descriptor.html#a2d3dcfc10f90adedc995b64211dab6e9">m_ConstantWeights</a> = <a class="code" href="namespacearmnn__driver.html#a9ddfceb499ef7b3d01053399bc79b852">IsOperandConstant</a>(*weightsOperand);</div> |
| 2025 | <div class="line"><a name="l01927"></a><span class="lineno"> 1927</span>  </div> |
| 2026 | <div class="line"><a name="l01928"></a><span class="lineno"> 1928</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2027 | <div class="line"><a name="l01929"></a><span class="lineno"> 1929</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2028 | <div class="line"><a name="l01930"></a><span class="lineno"> 1930</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2029 | <div class="line"><a name="l01931"></a><span class="lineno"> 1931</span>  {</div> |
| 2030 | <div class="line"><a name="l01932"></a><span class="lineno"> 1932</span>  <span class="keywordflow">if</span> (!VerifyFullyConnectedShapes(reshapedInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>(),</div> |
| 2031 | <div class="line"><a name="l01933"></a><span class="lineno"> 1933</span>  weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>(),</div> |
| 2032 | <div class="line"><a name="l01934"></a><span class="lineno"> 1934</span>  outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>(),</div> |
| 2033 | <div class="line"><a name="l01935"></a><span class="lineno"> 1935</span>  desc.<a class="code" href="structarmnn_1_1_fully_connected_descriptor.html#a281fcaec86e17c97f7b8402633f6b55a">m_TransposeWeightMatrix</a>))</div> |
| 2034 | <div class="line"><a name="l01936"></a><span class="lineno"> 1936</span>  {</div> |
| 2035 | <div class="line"><a name="l01937"></a><span class="lineno"> 1937</span>  isSupported = <span class="keyword">false</span>;</div> |
| 2036 | <div class="line"><a name="l01938"></a><span class="lineno"> 1938</span>  Fail(<span class="stringliteral">"%s: Expected outputShape does not match actual outputShape"</span>, __func__);</div> |
| 2037 | <div class="line"><a name="l01939"></a><span class="lineno"> 1939</span>  <span class="keywordflow">return</span>;</div> |
| 2038 | <div class="line"><a name="l01940"></a><span class="lineno"> 1940</span>  }</div> |
| 2039 | <div class="line"><a name="l01941"></a><span class="lineno"> 1941</span>  </div> |
| 2040 | <div class="line"><a name="l01942"></a><span class="lineno"> 1942</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2041 | <div class="line"><a name="l01943"></a><span class="lineno"> 1943</span>  IsFullyConnectedSupported,</div> |
| 2042 | <div class="line"><a name="l01944"></a><span class="lineno"> 1944</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2043 | <div class="line"><a name="l01945"></a><span class="lineno"> 1945</span>  isSupported,</div> |
| 2044 | <div class="line"><a name="l01946"></a><span class="lineno"> 1946</span>  setBackend,</div> |
| 2045 | <div class="line"><a name="l01947"></a><span class="lineno"> 1947</span>  reshapedInfo,</div> |
| 2046 | <div class="line"><a name="l01948"></a><span class="lineno"> 1948</span>  outputInfo,</div> |
| 2047 | <div class="line"><a name="l01949"></a><span class="lineno"> 1949</span>  weightsInfo,</div> |
| 2048 | <div class="line"><a name="l01950"></a><span class="lineno"> 1950</span>  biasInfo,</div> |
| 2049 | <div class="line"><a name="l01951"></a><span class="lineno"> 1951</span>  desc);</div> |
| 2050 | <div class="line"><a name="l01952"></a><span class="lineno"> 1952</span>  };</div> |
| 2051 | <div class="line"><a name="l01953"></a><span class="lineno"> 1953</span>  </div> |
| 2052 | <div class="line"><a name="l01954"></a><span class="lineno"> 1954</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2053 | <div class="line"><a name="l01955"></a><span class="lineno"> 1955</span>  {</div> |
| 2054 | <div class="line"><a name="l01956"></a><span class="lineno"> 1956</span>  validateFunc(outputInfo, isSupported);</div> |
| 2055 | <div class="line"><a name="l01957"></a><span class="lineno"> 1957</span>  }</div> |
| 2056 | <div class="line"><a name="l01958"></a><span class="lineno"> 1958</span>  <span class="keywordflow">else</span></div> |
| 2057 | <div class="line"><a name="l01959"></a><span class="lineno"> 1959</span>  {</div> |
| 2058 | <div class="line"><a name="l01960"></a><span class="lineno"> 1960</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2059 | <div class="line"><a name="l01961"></a><span class="lineno"> 1961</span>  }</div> |
| 2060 | <div class="line"><a name="l01962"></a><span class="lineno"> 1962</span>  </div> |
| 2061 | <div class="line"><a name="l01963"></a><span class="lineno"> 1963</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2062 | <div class="line"><a name="l01964"></a><span class="lineno"> 1964</span>  {</div> |
| 2063 | <div class="line"><a name="l01965"></a><span class="lineno"> 1965</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2064 | <div class="line"><a name="l01966"></a><span class="lineno"> 1966</span>  }</div> |
| 2065 | <div class="line"><a name="l01967"></a><span class="lineno"> 1967</span>  </div> |
| 2066 | <div class="line"><a name="l01968"></a><span class="lineno"> 1968</span>  <span class="comment">// Add FullyConnected layer. Weights and bias will be connected as constant layers or non const inputs.</span></div> |
| 2067 | <div class="line"><a name="l01969"></a><span class="lineno"> 1969</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* startLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddFullyConnectedLayer(desc);</div> |
| 2068 | <div class="line"><a name="l01970"></a><span class="lineno"> 1970</span>  startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2069 | <div class="line"><a name="l01971"></a><span class="lineno"> 1971</span>  </div> |
| 2070 | <div class="line"><a name="l01972"></a><span class="lineno"> 1972</span>  <span class="keywordflow">if</span> (inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() > 2U)</div> |
| 2071 | <div class="line"><a name="l01973"></a><span class="lineno"> 1973</span>  {</div> |
| 2072 | <div class="line"><a name="l01974"></a><span class="lineno"> 1974</span>  <a class="code" href="structarmnn_1_1_reshape_descriptor.html">armnn::ReshapeDescriptor</a> reshapeDescriptor;</div> |
| 2073 | <div class="line"><a name="l01975"></a><span class="lineno"> 1975</span>  reshapeDescriptor.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a> = reshapedInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 2074 | <div class="line"><a name="l01976"></a><span class="lineno"> 1976</span>  </div> |
| 2075 | <div class="line"><a name="l01977"></a><span class="lineno"> 1977</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* reshapeLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddReshapeLayer(reshapeDescriptor);</div> |
| 2076 | <div class="line"><a name="l01978"></a><span class="lineno"> 1978</span>  assert(reshapeLayer != <span class="keyword">nullptr</span>);</div> |
| 2077 | <div class="line"><a name="l01979"></a><span class="lineno"> 1979</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(reshapeLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2078 | <div class="line"><a name="l01980"></a><span class="lineno"> 1980</span>  reshapeLayer-><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#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(reshapedInfo);</div> |
| 2079 | <div class="line"><a name="l01981"></a><span class="lineno"> 1981</span>  reshapeLayer-><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#ac1835f8756a9f03c02fcf9664e3a0fce">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2080 | <div class="line"><a name="l01982"></a><span class="lineno"> 1982</span>  }</div> |
| 2081 | <div class="line"><a name="l01983"></a><span class="lineno"> 1983</span>  <span class="keywordflow">else</span></div> |
| 2082 | <div class="line"><a name="l01984"></a><span class="lineno"> 1984</span>  {</div> |
| 2083 | <div class="line"><a name="l01985"></a><span class="lineno"> 1985</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2084 | <div class="line"><a name="l01986"></a><span class="lineno"> 1986</span>  }</div> |
| 2085 | <div class="line"><a name="l01987"></a><span class="lineno"> 1987</span>  </div> |
| 2086 | <div class="line"><a name="l01988"></a><span class="lineno"> 1988</span>  <span class="comment">// Connect weights and bias inputs</span></div> |
| 2087 | <div class="line"><a name="l01989"></a><span class="lineno"> 1989</span>  weightsInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 2088 | <div class="line"><a name="l01990"></a><span class="lineno"> 1990</span>  biasInput.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 2089 | <div class="line"><a name="l01991"></a><span class="lineno"> 1991</span>  </div> |
| 2090 | <div class="line"><a name="l01992"></a><span class="lineno"> 1992</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *startLayer, model,</div> |
| 2091 | <div class="line"><a name="l01993"></a><span class="lineno"> 1993</span>  data, <span class="keyword">nullptr</span>, validateFunc, activationFunction);</div> |
| 2092 | <div class="line"><a name="l01994"></a><span class="lineno"> 1994</span> }</div> |
| 2093 | <div class="line"><a name="l01995"></a><span class="lineno"> 1995</span>  </div> |
| 2094 | <div class="line"><a name="l01996"></a><span class="lineno"> 1996</span> <span class="keywordtype">bool</span> Converter::ConvertGather(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2095 | <div class="line"><a name="l01997"></a><span class="lineno"> 1997</span> {</div> |
| 2096 | <div class="line"><a name="l01998"></a><span class="lineno"> 1998</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertGather()"</span>;</div> |
| 2097 | <div class="line"><a name="l01999"></a><span class="lineno"> 1999</span>  </div> |
| 2098 | <div class="line"><a name="l02000"></a><span class="lineno"> 2000</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2099 | <div class="line"><a name="l02001"></a><span class="lineno"> 2001</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2100 | <div class="line"><a name="l02002"></a><span class="lineno"> 2002</span>  {</div> |
| 2101 | <div class="line"><a name="l02003"></a><span class="lineno"> 2003</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid input"</span>, __func__);</div> |
| 2102 | <div class="line"><a name="l02004"></a><span class="lineno"> 2004</span>  }</div> |
| 2103 | <div class="line"><a name="l02005"></a><span class="lineno"> 2005</span>  <span class="keyword">auto</span> inputDimensions = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>().<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 2104 | <div class="line"><a name="l02006"></a><span class="lineno"> 2006</span>  </div> |
| 2105 | <div class="line"><a name="l02007"></a><span class="lineno"> 2007</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> indices = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 2, model, data);</div> |
| 2106 | <div class="line"><a name="l02008"></a><span class="lineno"> 2008</span>  <span class="keywordflow">if</span> (!indices.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2107 | <div class="line"><a name="l02009"></a><span class="lineno"> 2009</span>  {</div> |
| 2108 | <div class="line"><a name="l02010"></a><span class="lineno"> 2010</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid indices"</span>, __func__);</div> |
| 2109 | <div class="line"><a name="l02011"></a><span class="lineno"> 2011</span>  }</div> |
| 2110 | <div class="line"><a name="l02012"></a><span class="lineno"> 2012</span>  <span class="keyword">auto</span> indicesDimensions = indices.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>().<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 2111 | <div class="line"><a name="l02013"></a><span class="lineno"> 2013</span>  </div> |
| 2112 | <div class="line"><a name="l02014"></a><span class="lineno"> 2014</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2113 | <div class="line"><a name="l02015"></a><span class="lineno"> 2015</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2114 | <div class="line"><a name="l02016"></a><span class="lineno"> 2016</span>  {</div> |
| 2115 | <div class="line"><a name="l02017"></a><span class="lineno"> 2017</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid output"</span>, __func__);</div> |
| 2116 | <div class="line"><a name="l02018"></a><span class="lineno"> 2018</span>  }</div> |
| 2117 | <div class="line"><a name="l02019"></a><span class="lineno"> 2019</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2118 | <div class="line"><a name="l02020"></a><span class="lineno"> 2020</span>  <span class="keyword">auto</span> outputDimensions = outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 2119 | <div class="line"><a name="l02021"></a><span class="lineno"> 2021</span>  <span class="keywordflow">if</span> (outputDimensions != inputDimensions + indicesDimensions - 1)</div> |
| 2120 | <div class="line"><a name="l02022"></a><span class="lineno"> 2022</span>  {</div> |
| 2121 | <div class="line"><a name="l02023"></a><span class="lineno"> 2023</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid output dimensions: %d. Output must be an (%d + %d - 1)-D tensor"</span>,</div> |
| 2122 | <div class="line"><a name="l02024"></a><span class="lineno"> 2024</span>  __func__, outputDimensions, inputDimensions, indicesDimensions);</div> |
| 2123 | <div class="line"><a name="l02025"></a><span class="lineno"> 2025</span>  }</div> |
| 2124 | <div class="line"><a name="l02026"></a><span class="lineno"> 2026</span>  </div> |
| 2125 | <div class="line"><a name="l02027"></a><span class="lineno"> 2027</span>  int32_t axis;</div> |
| 2126 | <div class="line"><a name="l02028"></a><span class="lineno"> 2028</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, axis, model, data))</div> |
| 2127 | <div class="line"><a name="l02029"></a><span class="lineno"> 2029</span>  {</div> |
| 2128 | <div class="line"><a name="l02030"></a><span class="lineno"> 2030</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid or unsupported axis operand"</span>, __func__);</div> |
| 2129 | <div class="line"><a name="l02031"></a><span class="lineno"> 2031</span>  }</div> |
| 2130 | <div class="line"><a name="l02032"></a><span class="lineno"> 2032</span>  int32_t inputDimensions_int = <span class="keyword">static_cast<</span>int32_t<span class="keyword">></span>(inputDimensions);</div> |
| 2131 | <div class="line"><a name="l02033"></a><span class="lineno"> 2033</span>  <span class="keywordflow">if</span> ((axis < -inputDimensions_int) || (inputDimensions_int <= axis))</div> |
| 2132 | <div class="line"><a name="l02034"></a><span class="lineno"> 2034</span>  {</div> |
| 2133 | <div class="line"><a name="l02035"></a><span class="lineno"> 2035</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid axis: %d. It is out of bounds [-%d, %d))"</span>, __func__, axis,</div> |
| 2134 | <div class="line"><a name="l02036"></a><span class="lineno"> 2036</span>  inputDimensions, inputDimensions);</div> |
| 2135 | <div class="line"><a name="l02037"></a><span class="lineno"> 2037</span>  }</div> |
| 2136 | <div class="line"><a name="l02038"></a><span class="lineno"> 2038</span>  </div> |
| 2137 | <div class="line"><a name="l02039"></a><span class="lineno"> 2039</span>  <a class="code" href="structarmnn_1_1_gather_descriptor.html">GatherDescriptor</a> desc;</div> |
| 2138 | <div class="line"><a name="l02040"></a><span class="lineno"> 2040</span>  desc.<a class="code" href="structarmnn_1_1_gather_descriptor.html#a35d11c7d509d1adbae1ae01c58394a7f">m_Axis</a> = axis;</div> |
| 2139 | <div class="line"><a name="l02041"></a><span class="lineno"> 2041</span>  </div> |
| 2140 | <div class="line"><a name="l02042"></a><span class="lineno"> 2042</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2141 | <div class="line"><a name="l02043"></a><span class="lineno"> 2043</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2142 | <div class="line"><a name="l02044"></a><span class="lineno"> 2044</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2143 | <div class="line"><a name="l02045"></a><span class="lineno"> 2045</span>  {</div> |
| 2144 | <div class="line"><a name="l02046"></a><span class="lineno"> 2046</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2145 | <div class="line"><a name="l02047"></a><span class="lineno"> 2047</span>  IsGatherSupported,</div> |
| 2146 | <div class="line"><a name="l02048"></a><span class="lineno"> 2048</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2147 | <div class="line"><a name="l02049"></a><span class="lineno"> 2049</span>  isSupported,</div> |
| 2148 | <div class="line"><a name="l02050"></a><span class="lineno"> 2050</span>  setBackend,</div> |
| 2149 | <div class="line"><a name="l02051"></a><span class="lineno"> 2051</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 2150 | <div class="line"><a name="l02052"></a><span class="lineno"> 2052</span>  indices.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 2151 | <div class="line"><a name="l02053"></a><span class="lineno"> 2053</span>  outputInfo,</div> |
| 2152 | <div class="line"><a name="l02054"></a><span class="lineno"> 2054</span>  desc);</div> |
| 2153 | <div class="line"><a name="l02055"></a><span class="lineno"> 2055</span>  };</div> |
| 2154 | <div class="line"><a name="l02056"></a><span class="lineno"> 2056</span>  </div> |
| 2155 | <div class="line"><a name="l02057"></a><span class="lineno"> 2057</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2156 | <div class="line"><a name="l02058"></a><span class="lineno"> 2058</span>  {</div> |
| 2157 | <div class="line"><a name="l02059"></a><span class="lineno"> 2059</span>  validateFunc(outputInfo, isSupported);</div> |
| 2158 | <div class="line"><a name="l02060"></a><span class="lineno"> 2060</span>  }</div> |
| 2159 | <div class="line"><a name="l02061"></a><span class="lineno"> 2061</span>  <span class="keywordflow">else</span></div> |
| 2160 | <div class="line"><a name="l02062"></a><span class="lineno"> 2062</span>  {</div> |
| 2161 | <div class="line"><a name="l02063"></a><span class="lineno"> 2063</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2162 | <div class="line"><a name="l02064"></a><span class="lineno"> 2064</span>  }</div> |
| 2163 | <div class="line"><a name="l02065"></a><span class="lineno"> 2065</span>  </div> |
| 2164 | <div class="line"><a name="l02066"></a><span class="lineno"> 2066</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2165 | <div class="line"><a name="l02067"></a><span class="lineno"> 2067</span>  {</div> |
| 2166 | <div class="line"><a name="l02068"></a><span class="lineno"> 2068</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2167 | <div class="line"><a name="l02069"></a><span class="lineno"> 2069</span>  }</div> |
| 2168 | <div class="line"><a name="l02070"></a><span class="lineno"> 2070</span>  </div> |
| 2169 | <div class="line"><a name="l02071"></a><span class="lineno"> 2071</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddGatherLayer(desc);</div> |
| 2170 | <div class="line"><a name="l02072"></a><span class="lineno"> 2072</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2171 | <div class="line"><a name="l02073"></a><span class="lineno"> 2073</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 2172 | <div class="line"><a name="l02074"></a><span class="lineno"> 2074</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2173 | <div class="line"><a name="l02075"></a><span class="lineno"> 2075</span>  indices.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 2174 | <div class="line"><a name="l02076"></a><span class="lineno"> 2076</span>  </div> |
| 2175 | <div class="line"><a name="l02077"></a><span class="lineno"> 2077</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 2176 | <div class="line"><a name="l02078"></a><span class="lineno"> 2078</span> }</div> |
| 2177 | <div class="line"><a name="l02079"></a><span class="lineno"> 2079</span>  </div> |
| 2178 | <div class="line"><a name="l02080"></a><span class="lineno"> 2080</span> <span class="keywordtype">bool</span> Converter::ConvertGroupedConv2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2179 | <div class="line"><a name="l02081"></a><span class="lineno"> 2081</span> {</div> |
| 2180 | <div class="line"><a name="l02082"></a><span class="lineno"> 2082</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertGroupedConv2d()"</span>;</div> |
| 2181 | <div class="line"><a name="l02083"></a><span class="lineno"> 2083</span>  <span class="comment">//</span></div> |
| 2182 | <div class="line"><a name="l02084"></a><span class="lineno"> 2084</span>  <span class="comment">// Parse data</span></div> |
| 2183 | <div class="line"><a name="l02085"></a><span class="lineno"> 2085</span>  <span class="comment">//</span></div> |
| 2184 | <div class="line"><a name="l02086"></a><span class="lineno"> 2086</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2185 | <div class="line"><a name="l02087"></a><span class="lineno"> 2087</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2186 | <div class="line"><a name="l02088"></a><span class="lineno"> 2088</span>  {</div> |
| 2187 | <div class="line"><a name="l02089"></a><span class="lineno"> 2089</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2188 | <div class="line"><a name="l02090"></a><span class="lineno"> 2090</span>  }</div> |
| 2189 | <div class="line"><a name="l02091"></a><span class="lineno"> 2091</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2190 | <div class="line"><a name="l02092"></a><span class="lineno"> 2092</span>  </div> |
| 2191 | <div class="line"><a name="l02093"></a><span class="lineno"> 2093</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2192 | <div class="line"><a name="l02094"></a><span class="lineno"> 2094</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2193 | <div class="line"><a name="l02095"></a><span class="lineno"> 2095</span>  {</div> |
| 2194 | <div class="line"><a name="l02096"></a><span class="lineno"> 2096</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 2195 | <div class="line"><a name="l02097"></a><span class="lineno"> 2097</span>  }</div> |
| 2196 | <div class="line"><a name="l02098"></a><span class="lineno"> 2098</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2197 | <div class="line"><a name="l02099"></a><span class="lineno"> 2099</span>  </div> |
| 2198 | <div class="line"><a name="l02100"></a><span class="lineno"> 2100</span>  <span class="comment">// Look ahead to determine data layout</span></div> |
| 2199 | <div class="line"><a name="l02101"></a><span class="lineno"> 2101</span>  <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0">DataLayout</a> dataLayout = DataLayout::NHWC;</div> |
| 2200 | <div class="line"><a name="l02102"></a><span class="lineno"> 2102</span>  <span class="keywordflow">if</span> (operation.inputs.size() == 12)</div> |
| 2201 | <div class="line"><a name="l02103"></a><span class="lineno"> 2103</span>  {</div> |
| 2202 | <div class="line"><a name="l02104"></a><span class="lineno"> 2104</span>  dataLayout = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 11, model, data);</div> |
| 2203 | <div class="line"><a name="l02105"></a><span class="lineno"> 2105</span>  }</div> |
| 2204 | <div class="line"><a name="l02106"></a><span class="lineno"> 2106</span>  <span class="keywordflow">else</span></div> |
| 2205 | <div class="line"><a name="l02107"></a><span class="lineno"> 2107</span>  {</div> |
| 2206 | <div class="line"><a name="l02108"></a><span class="lineno"> 2108</span>  dataLayout = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 8, model, data);</div> |
| 2207 | <div class="line"><a name="l02109"></a><span class="lineno"> 2109</span>  }</div> |
| 2208 | <div class="line"><a name="l02110"></a><span class="lineno"> 2110</span>  </div> |
| 2209 | <div class="line"><a name="l02111"></a><span class="lineno"> 2111</span>  <span class="comment">// NOTE:</span></div> |
| 2210 | <div class="line"><a name="l02112"></a><span class="lineno"> 2112</span>  <span class="comment">// NNAPI weights are always OHWI, i.e. [depth_out, filter_height, filter_width, depth_group],</span></div> |
| 2211 | <div class="line"><a name="l02113"></a><span class="lineno"> 2113</span>  <span class="comment">// but Arm NN expects the filter's height and width indices to match the input's height and</span></div> |
| 2212 | <div class="line"><a name="l02114"></a><span class="lineno"> 2114</span>  <span class="comment">// width indices so when the DataLayout is NCHW, we need to permute the weights to OIHW</span></div> |
| 2213 | <div class="line"><a name="l02115"></a><span class="lineno"> 2115</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_permutation_vector.html">PermutationVector</a> ohwiToOihw = { 0u, 2u, 3u, 1u };</div> |
| 2214 | <div class="line"><a name="l02116"></a><span class="lineno"> 2116</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> weightsPin = (dataLayout == DataLayout::NCHW) ?</div> |
| 2215 | <div class="line"><a name="l02117"></a><span class="lineno"> 2117</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 1,</div> |
| 2216 | <div class="line"><a name="l02118"></a><span class="lineno"> 2118</span>  model, data, ohwiToOihw) :</div> |
| 2217 | <div class="line"><a name="l02119"></a><span class="lineno"> 2119</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 1, model, data);</div> |
| 2218 | <div class="line"><a name="l02120"></a><span class="lineno"> 2120</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> biasesPin =</div> |
| 2219 | <div class="line"><a name="l02121"></a><span class="lineno"> 2121</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 2, model, data);</div> |
| 2220 | <div class="line"><a name="l02122"></a><span class="lineno"> 2122</span>  <span class="keywordflow">if</span> (!weightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() || !biasesPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2221 | <div class="line"><a name="l02123"></a><span class="lineno"> 2123</span>  {</div> |
| 2222 | <div class="line"><a name="l02124"></a><span class="lineno"> 2124</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2223 | <div class="line"><a name="l02125"></a><span class="lineno"> 2125</span>  }</div> |
| 2224 | <div class="line"><a name="l02126"></a><span class="lineno"> 2126</span>  </div> |
| 2225 | <div class="line"><a name="l02127"></a><span class="lineno"> 2127</span>  <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> weights = weightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a971023b9874b0d773e8aa4c6d6744a2b">GetConstTensor</a>();</div> |
| 2226 | <div class="line"><a name="l02128"></a><span class="lineno"> 2128</span>  <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> biases = biasesPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a971023b9874b0d773e8aa4c6d6744a2b">GetConstTensor</a>();</div> |
| 2227 | <div class="line"><a name="l02129"></a><span class="lineno"> 2129</span>  SanitizeBiasQuantizationScale(biases.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>(), weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>(), inputInfo);</div> |
| 2228 | <div class="line"><a name="l02130"></a><span class="lineno"> 2130</span>  </div> |
| 2229 | <div class="line"><a name="l02131"></a><span class="lineno"> 2131</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>& inputShape = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 2230 | <div class="line"><a name="l02132"></a><span class="lineno"> 2132</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>& outputShape = outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 2231 | <div class="line"><a name="l02133"></a><span class="lineno"> 2133</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>& weightsShape = weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 2232 | <div class="line"><a name="l02134"></a><span class="lineno"> 2134</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>& biasesShape = biases.<a class="code" href="classarmnn_1_1_base_tensor.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 2233 | <div class="line"><a name="l02135"></a><span class="lineno"> 2135</span>  </div> |
| 2234 | <div class="line"><a name="l02136"></a><span class="lineno"> 2136</span>  <a class="code" href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a> dataLayoutIndexed(dataLayout);</div> |
| 2235 | <div class="line"><a name="l02137"></a><span class="lineno"> 2137</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> channelsIndex = dataLayoutIndexed.GetChannelsIndex();</div> |
| 2236 | <div class="line"><a name="l02138"></a><span class="lineno"> 2138</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> heightIndex = dataLayoutIndexed.GetHeightIndex();</div> |
| 2237 | <div class="line"><a name="l02139"></a><span class="lineno"> 2139</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> widthIndex = dataLayoutIndexed.GetWidthIndex();</div> |
| 2238 | <div class="line"><a name="l02140"></a><span class="lineno"> 2140</span>  </div> |
| 2239 | <div class="line"><a name="l02141"></a><span class="lineno"> 2141</span>  <a class="code" href="structarmnn_1_1_convolution2d_descriptor.html">Convolution2dDescriptor</a> desc;</div> |
| 2240 | <div class="line"><a name="l02142"></a><span class="lineno"> 2142</span>  desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = dataLayout;</div> |
| 2241 | <div class="line"><a name="l02143"></a><span class="lineno"> 2143</span>  desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">m_BiasEnabled</a> = <span class="keyword">true</span>;</div> |
| 2242 | <div class="line"><a name="l02144"></a><span class="lineno"> 2144</span>  </div> |
| 2243 | <div class="line"><a name="l02145"></a><span class="lineno"> 2145</span>  <span class="keywordtype">int</span> numGroups;</div> |
| 2244 | <div class="line"><a name="l02146"></a><span class="lineno"> 2146</span>  ActivationFn activation;</div> |
| 2245 | <div class="line"><a name="l02147"></a><span class="lineno"> 2147</span>  </div> |
| 2246 | <div class="line"><a name="l02148"></a><span class="lineno"> 2148</span>  <span class="keywordflow">if</span> (operation.inputs.size() == 12)</div> |
| 2247 | <div class="line"><a name="l02149"></a><span class="lineno"> 2149</span>  {</div> |
| 2248 | <div class="line"><a name="l02150"></a><span class="lineno"> 2150</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 3, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, model, data) ||</div> |
| 2249 | <div class="line"><a name="l02151"></a><span class="lineno"> 2151</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, model, data) ||</div> |
| 2250 | <div class="line"><a name="l02152"></a><span class="lineno"> 2152</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, model, data) ||</div> |
| 2251 | <div class="line"><a name="l02153"></a><span class="lineno"> 2153</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 6, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, model, data) ||</div> |
| 2252 | <div class="line"><a name="l02154"></a><span class="lineno"> 2154</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 7, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data) ||</div> |
| 2253 | <div class="line"><a name="l02155"></a><span class="lineno"> 2155</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 8, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data) ||</div> |
| 2254 | <div class="line"><a name="l02156"></a><span class="lineno"> 2156</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 9, OperandType::INT32, numGroups, model, data) ||</div> |
| 2255 | <div class="line"><a name="l02157"></a><span class="lineno"> 2157</span>  !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 10, activation, model, data))</div> |
| 2256 | <div class="line"><a name="l02158"></a><span class="lineno"> 2158</span>  {</div> |
| 2257 | <div class="line"><a name="l02159"></a><span class="lineno"> 2159</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (explicit padding)"</span>, __func__);</div> |
| 2258 | <div class="line"><a name="l02160"></a><span class="lineno"> 2160</span>  }</div> |
| 2259 | <div class="line"><a name="l02161"></a><span class="lineno"> 2161</span>  </div> |
| 2260 | <div class="line"><a name="l02162"></a><span class="lineno"> 2162</span>  }</div> |
| 2261 | <div class="line"><a name="l02163"></a><span class="lineno"> 2163</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operation.inputs.size() == 9)</div> |
| 2262 | <div class="line"><a name="l02164"></a><span class="lineno"> 2164</span>  {</div> |
| 2263 | <div class="line"><a name="l02165"></a><span class="lineno"> 2165</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| 2264 | <div class="line"><a name="l02166"></a><span class="lineno"> 2166</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa9ef723ace8562b9636e8387a120fbde">GetInputPaddingScheme</a>(operation, 3, paddingScheme, model, data) ||</div> |
| 2265 | <div class="line"><a name="l02167"></a><span class="lineno"> 2167</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data) ||</div> |
| 2266 | <div class="line"><a name="l02168"></a><span class="lineno"> 2168</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data) ||</div> |
| 2267 | <div class="line"><a name="l02169"></a><span class="lineno"> 2169</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 6, OperandType::INT32, numGroups, model, data) ||</div> |
| 2268 | <div class="line"><a name="l02170"></a><span class="lineno"> 2170</span>  !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 7, activation, model, data))</div> |
| 2269 | <div class="line"><a name="l02171"></a><span class="lineno"> 2171</span>  {</div> |
| 2270 | <div class="line"><a name="l02172"></a><span class="lineno"> 2172</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (implicit padding)"</span>, __func__);</div> |
| 2271 | <div class="line"><a name="l02173"></a><span class="lineno"> 2173</span>  }</div> |
| 2272 | <div class="line"><a name="l02174"></a><span class="lineno"> 2174</span>  </div> |
| 2273 | <div class="line"><a name="l02175"></a><span class="lineno"> 2175</span>  <span class="keyword">const</span> uint32_t inputX = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[widthIndex];</div> |
| 2274 | <div class="line"><a name="l02176"></a><span class="lineno"> 2176</span>  <span class="keyword">const</span> uint32_t inputY = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[heightIndex];</div> |
| 2275 | <div class="line"><a name="l02177"></a><span class="lineno"> 2177</span>  </div> |
| 2276 | <div class="line"><a name="l02178"></a><span class="lineno"> 2178</span>  <span class="keyword">const</span> uint32_t kernelX = weightsShape[widthIndex];</div> |
| 2277 | <div class="line"><a name="l02179"></a><span class="lineno"> 2179</span>  <span class="keyword">const</span> uint32_t kernelY = weightsShape[heightIndex];</div> |
| 2278 | <div class="line"><a name="l02180"></a><span class="lineno"> 2180</span>  </div> |
| 2279 | <div class="line"><a name="l02181"></a><span class="lineno"> 2181</span>  CalcPadding(inputX, kernelX, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, paddingScheme);</div> |
| 2280 | <div class="line"><a name="l02182"></a><span class="lineno"> 2182</span>  CalcPadding(inputY, kernelY, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, desc.<a class="code" href="structarmnn_1_1_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, paddingScheme);</div> |
| 2281 | <div class="line"><a name="l02183"></a><span class="lineno"> 2183</span>  }</div> |
| 2282 | <div class="line"><a name="l02184"></a><span class="lineno"> 2184</span>  <span class="keywordflow">else</span></div> |
| 2283 | <div class="line"><a name="l02185"></a><span class="lineno"> 2185</span>  {</div> |
| 2284 | <div class="line"><a name="l02186"></a><span class="lineno"> 2186</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported number of operation inputs"</span>, __func__);</div> |
| 2285 | <div class="line"><a name="l02187"></a><span class="lineno"> 2187</span>  }</div> |
| 2286 | <div class="line"><a name="l02188"></a><span class="lineno"> 2188</span>  </div> |
| 2287 | <div class="line"><a name="l02189"></a><span class="lineno"> 2189</span>  <span class="comment">// Equivalent to outputShape[channelsIndex], but we can't know the outputShape in the case of dynamic tensors</span></div> |
| 2288 | <div class="line"><a name="l02190"></a><span class="lineno"> 2190</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> outputChannels = weightsShape[0];</div> |
| 2289 | <div class="line"><a name="l02191"></a><span class="lineno"> 2191</span>  </div> |
| 2290 | <div class="line"><a name="l02192"></a><span class="lineno"> 2192</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> channelsPerGroup = weightsShape[channelsIndex];</div> |
| 2291 | <div class="line"><a name="l02193"></a><span class="lineno"> 2193</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> channelMultiplier = outputChannels / numGroups;</div> |
| 2292 | <div class="line"><a name="l02194"></a><span class="lineno"> 2194</span>  </div> |
| 2293 | <div class="line"><a name="l02195"></a><span class="lineno"> 2195</span>  <span class="comment">//</span></div> |
| 2294 | <div class="line"><a name="l02196"></a><span class="lineno"> 2196</span>  <span class="comment">// Validate all relevant inputs</span></div> |
| 2295 | <div class="line"><a name="l02197"></a><span class="lineno"> 2197</span>  <span class="comment">//</span></div> |
| 2296 | <div class="line"><a name="l02198"></a><span class="lineno"> 2198</span>  <span class="keywordflow">if</span> (numGroups <= 0)</div> |
| 2297 | <div class="line"><a name="l02199"></a><span class="lineno"> 2199</span>  {</div> |
| 2298 | <div class="line"><a name="l02200"></a><span class="lineno"> 2200</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Number of groups must be greater than 0. Got: %d"</span>, __func__, numGroups);</div> |
| 2299 | <div class="line"><a name="l02201"></a><span class="lineno"> 2201</span>  }</div> |
| 2300 | <div class="line"><a name="l02202"></a><span class="lineno"> 2202</span>  </div> |
| 2301 | <div class="line"><a name="l02203"></a><span class="lineno"> 2203</span>  <span class="keywordflow">if</span> (outputChannels % numGroups != 0u)</div> |
| 2302 | <div class="line"><a name="l02204"></a><span class="lineno"> 2204</span>  {</div> |
| 2303 | <div class="line"><a name="l02205"></a><span class="lineno"> 2205</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Output channels must be divisible by the number of groups"</span>, __func__);</div> |
| 2304 | <div class="line"><a name="l02206"></a><span class="lineno"> 2206</span>  }</div> |
| 2305 | <div class="line"><a name="l02207"></a><span class="lineno"> 2207</span>  </div> |
| 2306 | <div class="line"><a name="l02208"></a><span class="lineno"> 2208</span>  <span class="comment">//</span></div> |
| 2307 | <div class="line"><a name="l02209"></a><span class="lineno"> 2209</span>  <span class="comment">// Set up Splitter layer</span></div> |
| 2308 | <div class="line"><a name="l02210"></a><span class="lineno"> 2210</span>  <span class="comment">//</span></div> |
| 2309 | <div class="line"><a name="l02211"></a><span class="lineno"> 2211</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> splitterDimSizes[4] = { inputShape[0], inputShape[1], inputShape[2], inputShape[3] };</div> |
| 2310 | <div class="line"><a name="l02212"></a><span class="lineno"> 2212</span>  splitterDimSizes[channelsIndex] /= numGroups; <span class="comment">// split in depth</span></div> |
| 2311 | <div class="line"><a name="l02213"></a><span class="lineno"> 2213</span>  </div> |
| 2312 | <div class="line"><a name="l02214"></a><span class="lineno"> 2214</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> splitterOutputInfo(4,</div> |
| 2313 | <div class="line"><a name="l02215"></a><span class="lineno"> 2215</span>  splitterDimSizes,</div> |
| 2314 | <div class="line"><a name="l02216"></a><span class="lineno"> 2216</span>  inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#aea909c7327109228ef618d459015def3">GetDataType</a>(),</div> |
| 2315 | <div class="line"><a name="l02217"></a><span class="lineno"> 2217</span>  inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a047ca888c43bd7fb5702853bf72410d0">GetQuantizationScale</a>(),</div> |
| 2316 | <div class="line"><a name="l02218"></a><span class="lineno"> 2218</span>  inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a770b51078da02f44a819e9f95d8058b5">GetQuantizationOffset</a>());</div> |
| 2317 | <div class="line"><a name="l02219"></a><span class="lineno"> 2219</span>  </div> |
| 2318 | <div class="line"><a name="l02220"></a><span class="lineno"> 2220</span>  std::vector<std::reference_wrapper<TensorInfo>> splitterOutputInfos(numGroups, std::ref(splitterOutputInfo));</div> |
| 2319 | <div class="line"><a name="l02221"></a><span class="lineno"> 2221</span>  </div> |
| 2320 | <div class="line"><a name="l02222"></a><span class="lineno"> 2222</span>  <a class="code" href="structarmnn_1_1_views_descriptor.html">ViewsDescriptor</a> splitterDesc(numGroups);</div> |
| 2321 | <div class="line"><a name="l02223"></a><span class="lineno"> 2223</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> group = 0u; group < numGroups; ++group)</div> |
| 2322 | <div class="line"><a name="l02224"></a><span class="lineno"> 2224</span>  {</div> |
| 2323 | <div class="line"><a name="l02225"></a><span class="lineno"> 2225</span>  splitterDesc.SetViewOriginCoord(group, channelsIndex, splitterDimSizes[channelsIndex] * group);</div> |
| 2324 | <div class="line"><a name="l02226"></a><span class="lineno"> 2226</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> dimIdx = 0u; dimIdx < 4u; dimIdx++)</div> |
| 2325 | <div class="line"><a name="l02227"></a><span class="lineno"> 2227</span>  {</div> |
| 2326 | <div class="line"><a name="l02228"></a><span class="lineno"> 2228</span>  splitterDesc.SetViewSize(group, dimIdx, splitterDimSizes[dimIdx]);</div> |
| 2327 | <div class="line"><a name="l02229"></a><span class="lineno"> 2229</span>  }</div> |
| 2328 | <div class="line"><a name="l02230"></a><span class="lineno"> 2230</span>  }</div> |
| 2329 | <div class="line"><a name="l02231"></a><span class="lineno"> 2231</span>  </div> |
| 2330 | <div class="line"><a name="l02232"></a><span class="lineno"> 2232</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2331 | <div class="line"><a name="l02233"></a><span class="lineno"> 2233</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendSplit;</div> |
| 2332 | <div class="line"><a name="l02234"></a><span class="lineno"> 2234</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2333 | <div class="line"><a name="l02235"></a><span class="lineno"> 2235</span>  IsSplitterSupported,</div> |
| 2334 | <div class="line"><a name="l02236"></a><span class="lineno"> 2236</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2335 | <div class="line"><a name="l02237"></a><span class="lineno"> 2237</span>  isSupported,</div> |
| 2336 | <div class="line"><a name="l02238"></a><span class="lineno"> 2238</span>  setBackendSplit,</div> |
| 2337 | <div class="line"><a name="l02239"></a><span class="lineno"> 2239</span>  inputInfo,</div> |
| 2338 | <div class="line"><a name="l02240"></a><span class="lineno"> 2240</span>  splitterOutputInfos,</div> |
| 2339 | <div class="line"><a name="l02241"></a><span class="lineno"> 2241</span>  splitterDesc);</div> |
| 2340 | <div class="line"><a name="l02242"></a><span class="lineno"> 2242</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2341 | <div class="line"><a name="l02243"></a><span class="lineno"> 2243</span>  {</div> |
| 2342 | <div class="line"><a name="l02244"></a><span class="lineno"> 2244</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2343 | <div class="line"><a name="l02245"></a><span class="lineno"> 2245</span>  }</div> |
| 2344 | <div class="line"><a name="l02246"></a><span class="lineno"> 2246</span>  </div> |
| 2345 | <div class="line"><a name="l02247"></a><span class="lineno"> 2247</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* splitterLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddSplitterLayer(splitterDesc);</div> |
| 2346 | <div class="line"><a name="l02248"></a><span class="lineno"> 2248</span>  splitterLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendSplit);</div> |
| 2347 | <div class="line"><a name="l02249"></a><span class="lineno"> 2249</span>  <span class="keywordflow">if</span> (!splitterLayer)</div> |
| 2348 | <div class="line"><a name="l02250"></a><span class="lineno"> 2250</span>  {</div> |
| 2349 | <div class="line"><a name="l02251"></a><span class="lineno"> 2251</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to add SplitterLayer"</span>, __func__);</div> |
| 2350 | <div class="line"><a name="l02252"></a><span class="lineno"> 2252</span>  }</div> |
| 2351 | <div class="line"><a name="l02253"></a><span class="lineno"> 2253</span>  </div> |
| 2352 | <div class="line"><a name="l02254"></a><span class="lineno"> 2254</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(splitterLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2353 | <div class="line"><a name="l02255"></a><span class="lineno"> 2255</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> group = 0u; group < splitterLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#ac2dac3b61c94de52093616be4ab17f8d">GetNumOutputSlots</a>(); ++group)</div> |
| 2354 | <div class="line"><a name="l02256"></a><span class="lineno"> 2256</span>  {</div> |
| 2355 | <div class="line"><a name="l02257"></a><span class="lineno"> 2257</span>  splitterLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(group).<a class="code" href="classarmnn_1_1_i_output_slot.html#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(splitterOutputInfo);</div> |
| 2356 | <div class="line"><a name="l02258"></a><span class="lineno"> 2258</span>  }</div> |
| 2357 | <div class="line"><a name="l02259"></a><span class="lineno"> 2259</span>  </div> |
| 2358 | <div class="line"><a name="l02260"></a><span class="lineno"> 2260</span>  <span class="comment">//</span></div> |
| 2359 | <div class="line"><a name="l02261"></a><span class="lineno"> 2261</span>  <span class="comment">// Set up Convolution2d layers for each group</span></div> |
| 2360 | <div class="line"><a name="l02262"></a><span class="lineno"> 2262</span>  <span class="comment">//</span></div> |
| 2361 | <div class="line"><a name="l02263"></a><span class="lineno"> 2263</span>  </div> |
| 2362 | <div class="line"><a name="l02264"></a><span class="lineno"> 2264</span>  <span class="comment">// Set up group tensor shapes</span></div> |
| 2363 | <div class="line"><a name="l02265"></a><span class="lineno"> 2265</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> groupInputShape(inputShape);</div> |
| 2364 | <div class="line"><a name="l02266"></a><span class="lineno"> 2266</span>  groupInputShape[channelsIndex] = channelsPerGroup;</div> |
| 2365 | <div class="line"><a name="l02267"></a><span class="lineno"> 2267</span>  </div> |
| 2366 | <div class="line"><a name="l02268"></a><span class="lineno"> 2268</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> groupWeightsShape(weightsShape);</div> |
| 2367 | <div class="line"><a name="l02269"></a><span class="lineno"> 2269</span>  groupWeightsShape[0] /= channelMultiplier * numGroups;</div> |
| 2368 | <div class="line"><a name="l02270"></a><span class="lineno"> 2270</span>  </div> |
| 2369 | <div class="line"><a name="l02271"></a><span class="lineno"> 2271</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> groupBiasesShape({ 1 });</div> |
| 2370 | <div class="line"><a name="l02272"></a><span class="lineno"> 2272</span>  </div> |
| 2371 | <div class="line"><a name="l02273"></a><span class="lineno"> 2273</span>  <span class="comment">// Set up group tensor infos</span></div> |
| 2372 | <div class="line"><a name="l02274"></a><span class="lineno"> 2274</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> groupInputInfo(inputInfo);</div> |
| 2373 | <div class="line"><a name="l02275"></a><span class="lineno"> 2275</span>  groupInputInfo.SetShape(groupInputShape);</div> |
| 2374 | <div class="line"><a name="l02276"></a><span class="lineno"> 2276</span>  </div> |
| 2375 | <div class="line"><a name="l02277"></a><span class="lineno"> 2277</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& weightsInfo = weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>();</div> |
| 2376 | <div class="line"><a name="l02278"></a><span class="lineno"> 2278</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> groupWeightsInfo(weightsInfo);</div> |
| 2377 | <div class="line"><a name="l02279"></a><span class="lineno"> 2279</span>  groupWeightsInfo.SetShape(groupWeightsShape);</div> |
| 2378 | <div class="line"><a name="l02280"></a><span class="lineno"> 2280</span>  </div> |
| 2379 | <div class="line"><a name="l02281"></a><span class="lineno"> 2281</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& biasesInfo = biases.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>();</div> |
| 2380 | <div class="line"><a name="l02282"></a><span class="lineno"> 2282</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> groupBiasesInfo(biasesInfo);</div> |
| 2381 | <div class="line"><a name="l02283"></a><span class="lineno"> 2283</span>  groupBiasesInfo.SetShape(groupBiasesShape);</div> |
| 2382 | <div class="line"><a name="l02284"></a><span class="lineno"> 2284</span>  </div> |
| 2383 | <div class="line"><a name="l02285"></a><span class="lineno"> 2285</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> groupOutputInfo(outputInfo);</div> |
| 2384 | <div class="line"><a name="l02286"></a><span class="lineno"> 2286</span>  </div> |
| 2385 | <div class="line"><a name="l02287"></a><span class="lineno"> 2287</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a> groupOutputShape(outputShape);</div> |
| 2386 | <div class="line"><a name="l02288"></a><span class="lineno"> 2288</span>  <span class="keyword">const</span> <span class="keywordtype">bool</span> isDynamic = <a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo);</div> |
| 2387 | <div class="line"><a name="l02289"></a><span class="lineno"> 2289</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| 2388 | <div class="line"><a name="l02290"></a><span class="lineno"> 2290</span>  {</div> |
| 2389 | <div class="line"><a name="l02291"></a><span class="lineno"> 2291</span>  groupOutputShape[channelsIndex] = 1;</div> |
| 2390 | <div class="line"><a name="l02292"></a><span class="lineno"> 2292</span>  }</div> |
| 2391 | <div class="line"><a name="l02293"></a><span class="lineno"> 2293</span>  groupOutputInfo.SetShape(groupOutputShape);</div> |
| 2392 | <div class="line"><a name="l02294"></a><span class="lineno"> 2294</span>  </div> |
| 2393 | <div class="line"><a name="l02295"></a><span class="lineno"> 2295</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> weightsDataTypeSize = <a class="code" href="namespacearmnn.html#aa02b9e06fb20fa3c13da0427e6ee5ab2">GetDataTypeSize</a>(groupWeightsInfo.GetDataType());</div> |
| 2394 | <div class="line"><a name="l02296"></a><span class="lineno"> 2296</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> biasesDataTypeSize = <a class="code" href="namespacearmnn.html#aa02b9e06fb20fa3c13da0427e6ee5ab2">GetDataTypeSize</a>(groupBiasesInfo.GetDataType());</div> |
| 2395 | <div class="line"><a name="l02297"></a><span class="lineno"> 2297</span>  </div> |
| 2396 | <div class="line"><a name="l02298"></a><span class="lineno"> 2298</span>  std::vector<IConnectableLayer*> convLayers(numGroups * channelMultiplier, <span class="keyword">nullptr</span>);</div> |
| 2397 | <div class="line"><a name="l02299"></a><span class="lineno"> 2299</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> group = 0u; group < numGroups; ++group)</div> |
| 2398 | <div class="line"><a name="l02300"></a><span class="lineno"> 2300</span>  {</div> |
| 2399 | <div class="line"><a name="l02301"></a><span class="lineno"> 2301</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> m = 0u; m < channelMultiplier; ++m)</div> |
| 2400 | <div class="line"><a name="l02302"></a><span class="lineno"> 2302</span>  {</div> |
| 2401 | <div class="line"><a name="l02303"></a><span class="lineno"> 2303</span>  <span class="keyword">auto</span> index = group * channelMultiplier + m;</div> |
| 2402 | <div class="line"><a name="l02304"></a><span class="lineno"> 2304</span>  </div> |
| 2403 | <div class="line"><a name="l02305"></a><span class="lineno"> 2305</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> weightsDataOffset = groupWeightsShape.GetNumElements() * index * weightsDataTypeSize;</div> |
| 2404 | <div class="line"><a name="l02306"></a><span class="lineno"> 2306</span>  <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> biasesDataOffset = groupBiasesShape.GetNumElements() * index * biasesDataTypeSize;</div> |
| 2405 | <div class="line"><a name="l02307"></a><span class="lineno"> 2307</span>  </div> |
| 2406 | <div class="line"><a name="l02308"></a><span class="lineno"> 2308</span>  <span class="keywordflow">if</span> (weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#ab85cd8cc10c96a7c99c14042c251fc48">HasPerAxisQuantization</a>())</div> |
| 2407 | <div class="line"><a name="l02309"></a><span class="lineno"> 2309</span>  {</div> |
| 2408 | <div class="line"><a name="l02310"></a><span class="lineno"> 2310</span>  <span class="comment">// Extract per-axis quantization scales for group weights</span></div> |
| 2409 | <div class="line"><a name="l02311"></a><span class="lineno"> 2311</span>  <span class="keyword">const</span> std::vector<float>& weightsQuantScales = weightsInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8bc11f1fa23ef42532f9fdd04d355270">GetQuantizationScales</a>();</div> |
| 2410 | <div class="line"><a name="l02312"></a><span class="lineno"> 2312</span>  groupWeightsInfo.SetQuantizationScales(</div> |
| 2411 | <div class="line"><a name="l02313"></a><span class="lineno"> 2313</span>  std::vector<float>(weightsQuantScales.begin() + index,</div> |
| 2412 | <div class="line"><a name="l02314"></a><span class="lineno"> 2314</span>  weightsQuantScales.begin() + index + groupWeightsShape[0]));</div> |
| 2413 | <div class="line"><a name="l02315"></a><span class="lineno"> 2315</span>  </div> |
| 2414 | <div class="line"><a name="l02316"></a><span class="lineno"> 2316</span>  <span class="comment">// Extract per-axis quantization scales for group biases</span></div> |
| 2415 | <div class="line"><a name="l02317"></a><span class="lineno"> 2317</span>  <span class="keyword">const</span> std::vector<float>& biasesQuantScales = biasesInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8bc11f1fa23ef42532f9fdd04d355270">GetQuantizationScales</a>();</div> |
| 2416 | <div class="line"><a name="l02318"></a><span class="lineno"> 2318</span>  groupBiasesInfo.SetQuantizationScales(</div> |
| 2417 | <div class="line"><a name="l02319"></a><span class="lineno"> 2319</span>  std::vector<float>(biasesQuantScales.begin() + index,</div> |
| 2418 | <div class="line"><a name="l02320"></a><span class="lineno"> 2320</span>  biasesQuantScales.begin() + index + groupWeightsShape[0]));</div> |
| 2419 | <div class="line"><a name="l02321"></a><span class="lineno"> 2321</span>  }</div> |
| 2420 | <div class="line"><a name="l02322"></a><span class="lineno"> 2322</span>  </div> |
| 2421 | <div class="line"><a name="l02323"></a><span class="lineno"> 2323</span>  <span class="comment">// Extract weights and biases data for current group convolution</span></div> |
| 2422 | <div class="line"><a name="l02324"></a><span class="lineno"> 2324</span>  <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> groupWeights(groupWeightsInfo,</div> |
| 2423 | <div class="line"><a name="l02325"></a><span class="lineno"> 2325</span>  <span class="keyword">static_cast<</span><span class="keyword">const </span><span class="keywordtype">void</span> *<span class="keyword">></span>(<span class="keyword">reinterpret_cast<</span><span class="keyword">const </span><span class="keywordtype">char</span> *<span class="keyword">></span>(weights.<a class="code" href="classarmnn_1_1_base_tensor.html#aa81f67ac64f0c249e26499600c45d996">GetMemoryArea</a>()) +</div> |
| 2424 | <div class="line"><a name="l02326"></a><span class="lineno"> 2326</span>  weightsDataOffset));</div> |
| 2425 | <div class="line"><a name="l02327"></a><span class="lineno"> 2327</span>  <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> groupBiases(groupBiasesInfo,</div> |
| 2426 | <div class="line"><a name="l02328"></a><span class="lineno"> 2328</span>  <span class="keyword">static_cast<</span><span class="keyword">const </span><span class="keywordtype">void</span> *<span class="keyword">></span>(<span class="keyword">reinterpret_cast<</span><span class="keyword">const </span><span class="keywordtype">char</span> *<span class="keyword">></span>(biases.<a class="code" href="classarmnn_1_1_base_tensor.html#aa81f67ac64f0c249e26499600c45d996">GetMemoryArea</a>()) +</div> |
| 2427 | <div class="line"><a name="l02329"></a><span class="lineno"> 2329</span>  biasesDataOffset));</div> |
| 2428 | <div class="line"><a name="l02330"></a><span class="lineno"> 2330</span>  </div> |
| 2429 | <div class="line"><a name="l02331"></a><span class="lineno"> 2331</span>  isSupported = <span class="keyword">false</span>;</div> |
| 2430 | <div class="line"><a name="l02332"></a><span class="lineno"> 2332</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendConv;</div> |
| 2431 | <div class="line"><a name="l02333"></a><span class="lineno"> 2333</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2432 | <div class="line"><a name="l02334"></a><span class="lineno"> 2334</span>  {</div> |
| 2433 | <div class="line"><a name="l02335"></a><span class="lineno"> 2335</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2434 | <div class="line"><a name="l02336"></a><span class="lineno"> 2336</span>  IsConvolution2dSupported,</div> |
| 2435 | <div class="line"><a name="l02337"></a><span class="lineno"> 2337</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2436 | <div class="line"><a name="l02338"></a><span class="lineno"> 2338</span>  isSupported,</div> |
| 2437 | <div class="line"><a name="l02339"></a><span class="lineno"> 2339</span>  setBackendConv,</div> |
| 2438 | <div class="line"><a name="l02340"></a><span class="lineno"> 2340</span>  groupInputInfo,</div> |
| 2439 | <div class="line"><a name="l02341"></a><span class="lineno"> 2341</span>  outputInfo,</div> |
| 2440 | <div class="line"><a name="l02342"></a><span class="lineno"> 2342</span>  desc,</div> |
| 2441 | <div class="line"><a name="l02343"></a><span class="lineno"> 2343</span>  groupWeightsInfo,</div> |
| 2442 | <div class="line"><a name="l02344"></a><span class="lineno"> 2344</span>  <a class="code" href="classarmnn_1_1_optional.html">Optional<TensorInfo></a>(groupBiasesInfo));</div> |
| 2443 | <div class="line"><a name="l02345"></a><span class="lineno"> 2345</span>  };</div> |
| 2444 | <div class="line"><a name="l02346"></a><span class="lineno"> 2346</span>  </div> |
| 2445 | <div class="line"><a name="l02347"></a><span class="lineno"> 2347</span>  <span class="keywordflow">if</span>(!isDynamic)</div> |
| 2446 | <div class="line"><a name="l02348"></a><span class="lineno"> 2348</span>  {</div> |
| 2447 | <div class="line"><a name="l02349"></a><span class="lineno"> 2349</span>  validateFunc(groupOutputInfo, isSupported);</div> |
| 2448 | <div class="line"><a name="l02350"></a><span class="lineno"> 2350</span>  }</div> |
| 2449 | <div class="line"><a name="l02351"></a><span class="lineno"> 2351</span>  <span class="keywordflow">else</span></div> |
| 2450 | <div class="line"><a name="l02352"></a><span class="lineno"> 2352</span>  {</div> |
| 2451 | <div class="line"><a name="l02353"></a><span class="lineno"> 2353</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2452 | <div class="line"><a name="l02354"></a><span class="lineno"> 2354</span>  }</div> |
| 2453 | <div class="line"><a name="l02355"></a><span class="lineno"> 2355</span>  </div> |
| 2454 | <div class="line"><a name="l02356"></a><span class="lineno"> 2356</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2455 | <div class="line"><a name="l02357"></a><span class="lineno"> 2357</span>  {</div> |
| 2456 | <div class="line"><a name="l02358"></a><span class="lineno"> 2358</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2457 | <div class="line"><a name="l02359"></a><span class="lineno"> 2359</span>  }</div> |
| 2458 | <div class="line"><a name="l02360"></a><span class="lineno"> 2360</span>  </div> |
| 2459 | <div class="line"><a name="l02361"></a><span class="lineno"> 2361</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* weightsLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddConstantLayer(groupWeights);</div> |
| 2460 | <div class="line"><a name="l02362"></a><span class="lineno"> 2362</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* biasLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddConstantLayer(groupBiases);</div> |
| 2461 | <div class="line"><a name="l02363"></a><span class="lineno"> 2363</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* convLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddConvolution2dLayer(desc);</div> |
| 2462 | <div class="line"><a name="l02364"></a><span class="lineno"> 2364</span>  </div> |
| 2463 | <div class="line"><a name="l02365"></a><span class="lineno"> 2365</span>  convLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendConv);</div> |
| 2464 | <div class="line"><a name="l02366"></a><span class="lineno"> 2366</span>  </div> |
| 2465 | <div class="line"><a name="l02367"></a><span class="lineno"> 2367</span>  <span class="keywordflow">if</span> (!convLayer)</div> |
| 2466 | <div class="line"><a name="l02368"></a><span class="lineno"> 2368</span>  {</div> |
| 2467 | <div class="line"><a name="l02369"></a><span class="lineno"> 2369</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddConvolution2dLayer failed"</span>, __func__);</div> |
| 2468 | <div class="line"><a name="l02370"></a><span class="lineno"> 2370</span>  }</div> |
| 2469 | <div class="line"><a name="l02371"></a><span class="lineno"> 2371</span>  </div> |
| 2470 | <div class="line"><a name="l02372"></a><span class="lineno"> 2372</span>  splitterLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(group).<a class="code" href="classarmnn_1_1_i_output_slot.html#ac1835f8756a9f03c02fcf9664e3a0fce">Connect</a>(convLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2471 | <div class="line"><a name="l02373"></a><span class="lineno"> 2373</span>  weightsLayer-><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#ac1835f8756a9f03c02fcf9664e3a0fce">Connect</a>(convLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 2472 | <div class="line"><a name="l02374"></a><span class="lineno"> 2374</span>  biasLayer-><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#ac1835f8756a9f03c02fcf9664e3a0fce">Connect</a>(convLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 2473 | <div class="line"><a name="l02375"></a><span class="lineno"> 2375</span>  </div> |
| 2474 | <div class="line"><a name="l02376"></a><span class="lineno"> 2376</span>  weightsLayer-><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#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(groupWeightsInfo);</div> |
| 2475 | <div class="line"><a name="l02377"></a><span class="lineno"> 2377</span>  biasLayer-><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#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(groupBiasesInfo);</div> |
| 2476 | <div class="line"><a name="l02378"></a><span class="lineno"> 2378</span>  convLayer-><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#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(groupOutputInfo);</div> |
| 2477 | <div class="line"><a name="l02379"></a><span class="lineno"> 2379</span>  </div> |
| 2478 | <div class="line"><a name="l02380"></a><span class="lineno"> 2380</span>  <span class="keywordflow">if</span>(isDynamic)</div> |
| 2479 | <div class="line"><a name="l02381"></a><span class="lineno"> 2381</span>  {</div> |
| 2480 | <div class="line"><a name="l02382"></a><span class="lineno"> 2382</span>  convLayer-><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#ab4e57caf059794897b17266090ebcb39">IsTensorInfoSet</a>();</div> |
| 2481 | <div class="line"><a name="l02383"></a><span class="lineno"> 2383</span>  </div> |
| 2482 | <div class="line"><a name="l02384"></a><span class="lineno"> 2384</span>  validateFunc(convLayer-><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#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>(), isSupported);</div> |
| 2483 | <div class="line"><a name="l02385"></a><span class="lineno"> 2385</span>  </div> |
| 2484 | <div class="line"><a name="l02386"></a><span class="lineno"> 2386</span>  outputInfo = convLayer-><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#a9943775a364fc4ab53b85ac88f311886">GetTensorInfo</a>();</div> |
| 2485 | <div class="line"><a name="l02387"></a><span class="lineno"> 2387</span>  </div> |
| 2486 | <div class="line"><a name="l02388"></a><span class="lineno"> 2388</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2487 | <div class="line"><a name="l02389"></a><span class="lineno"> 2389</span>  {</div> |
| 2488 | <div class="line"><a name="l02390"></a><span class="lineno"> 2390</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2489 | <div class="line"><a name="l02391"></a><span class="lineno"> 2391</span>  }</div> |
| 2490 | <div class="line"><a name="l02392"></a><span class="lineno"> 2392</span>  }</div> |
| 2491 | <div class="line"><a name="l02393"></a><span class="lineno"> 2393</span>  </div> |
| 2492 | <div class="line"><a name="l02394"></a><span class="lineno"> 2394</span>  convLayers[index] = convLayer;</div> |
| 2493 | <div class="line"><a name="l02395"></a><span class="lineno"> 2395</span>  }</div> |
| 2494 | <div class="line"><a name="l02396"></a><span class="lineno"> 2396</span>  }</div> |
| 2495 | <div class="line"><a name="l02397"></a><span class="lineno"> 2397</span>  </div> |
| 2496 | <div class="line"><a name="l02398"></a><span class="lineno"> 2398</span>  <span class="comment">//</span></div> |
| 2497 | <div class="line"><a name="l02399"></a><span class="lineno"> 2399</span>  <span class="comment">// Set up Concat layer</span></div> |
| 2498 | <div class="line"><a name="l02400"></a><span class="lineno"> 2400</span>  <span class="comment">//</span></div> |
| 2499 | <div class="line"><a name="l02401"></a><span class="lineno"> 2401</span>  <a class="code" href="structarmnn_1_1_origins_descriptor.html">ConcatDescriptor</a> concatDescriptor;</div> |
| 2500 | <div class="line"><a name="l02402"></a><span class="lineno"> 2402</span>  <span class="comment">// Equivalent to outputShape[channelsIndex], but we can't know the outputShape in the case of dynamic tensors</span></div> |
| 2501 | <div class="line"><a name="l02403"></a><span class="lineno"> 2403</span>  concatDescriptor = <a class="code" href="namespacearmnn.html#a7863c179ff92feec660c48ab7b95ae55">ConcatDescriptor</a>(weightsShape[0]);</div> |
| 2502 | <div class="line"><a name="l02404"></a><span class="lineno"> 2404</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> group = 0u; group < numGroups; ++group)</div> |
| 2503 | <div class="line"><a name="l02405"></a><span class="lineno"> 2405</span>  {</div> |
| 2504 | <div class="line"><a name="l02406"></a><span class="lineno"> 2406</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> m = 0u; m < channelMultiplier; ++m)</div> |
| 2505 | <div class="line"><a name="l02407"></a><span class="lineno"> 2407</span>  {</div> |
| 2506 | <div class="line"><a name="l02408"></a><span class="lineno"> 2408</span>  <span class="keyword">auto</span> index = group * channelMultiplier + m;</div> |
| 2507 | <div class="line"><a name="l02409"></a><span class="lineno"> 2409</span>  concatDescriptor.<a class="code" href="structarmnn_1_1_origins_descriptor.html#a2b125117aa61f9baf3a9cb8658aa61a2">SetViewOriginCoord</a>(index, channelsIndex, index);</div> |
| 2508 | <div class="line"><a name="l02410"></a><span class="lineno"> 2410</span>  concatDescriptor.<a class="code" href="structarmnn_1_1_origins_descriptor.html#a5b192c5fcd96a0f75542524cf646b355">SetConcatAxis</a>(channelsIndex);</div> |
| 2509 | <div class="line"><a name="l02411"></a><span class="lineno"> 2411</span>  }</div> |
| 2510 | <div class="line"><a name="l02412"></a><span class="lineno"> 2412</span>  }</div> |
| 2511 | <div class="line"><a name="l02413"></a><span class="lineno"> 2413</span>  </div> |
| 2512 | <div class="line"><a name="l02414"></a><span class="lineno"> 2414</span>  isSupported = <span class="keyword">false</span>;</div> |
| 2513 | <div class="line"><a name="l02415"></a><span class="lineno"> 2415</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackendConcat;</div> |
| 2514 | <div class="line"><a name="l02416"></a><span class="lineno"> 2416</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2515 | <div class="line"><a name="l02417"></a><span class="lineno"> 2417</span>  IsConcatSupported,</div> |
| 2516 | <div class="line"><a name="l02418"></a><span class="lineno"> 2418</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2517 | <div class="line"><a name="l02419"></a><span class="lineno"> 2419</span>  isSupported,</div> |
| 2518 | <div class="line"><a name="l02420"></a><span class="lineno"> 2420</span>  setBackendConcat,</div> |
| 2519 | <div class="line"><a name="l02421"></a><span class="lineno"> 2421</span>  std::vector<const TensorInfo*>(numGroups * channelMultiplier, &groupOutputInfo),</div> |
| 2520 | <div class="line"><a name="l02422"></a><span class="lineno"> 2422</span>  outputInfo,</div> |
| 2521 | <div class="line"><a name="l02423"></a><span class="lineno"> 2423</span>  concatDescriptor);</div> |
| 2522 | <div class="line"><a name="l02424"></a><span class="lineno"> 2424</span>  </div> |
| 2523 | <div class="line"><a name="l02425"></a><span class="lineno"> 2425</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2524 | <div class="line"><a name="l02426"></a><span class="lineno"> 2426</span>  {</div> |
| 2525 | <div class="line"><a name="l02427"></a><span class="lineno"> 2427</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2526 | <div class="line"><a name="l02428"></a><span class="lineno"> 2428</span>  }</div> |
| 2527 | <div class="line"><a name="l02429"></a><span class="lineno"> 2429</span>  </div> |
| 2528 | <div class="line"><a name="l02430"></a><span class="lineno"> 2430</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* concatLayer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddConcatLayer(concatDescriptor);</div> |
| 2529 | <div class="line"><a name="l02431"></a><span class="lineno"> 2431</span>  concatLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackendConcat);</div> |
| 2530 | <div class="line"><a name="l02432"></a><span class="lineno"> 2432</span>  <span class="keywordflow">if</span> (!concatLayer)</div> |
| 2531 | <div class="line"><a name="l02433"></a><span class="lineno"> 2433</span>  {</div> |
| 2532 | <div class="line"><a name="l02434"></a><span class="lineno"> 2434</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddConcatLayer failed"</span>, __func__);</div> |
| 2533 | <div class="line"><a name="l02435"></a><span class="lineno"> 2435</span>  }</div> |
| 2534 | <div class="line"><a name="l02436"></a><span class="lineno"> 2436</span>  </div> |
| 2535 | <div class="line"><a name="l02437"></a><span class="lineno"> 2437</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> group = 0u; group < numGroups; ++group)</div> |
| 2536 | <div class="line"><a name="l02438"></a><span class="lineno"> 2438</span>  {</div> |
| 2537 | <div class="line"><a name="l02439"></a><span class="lineno"> 2439</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> m = 0u; m < channelMultiplier; ++m)</div> |
| 2538 | <div class="line"><a name="l02440"></a><span class="lineno"> 2440</span>  {</div> |
| 2539 | <div class="line"><a name="l02441"></a><span class="lineno"> 2441</span>  <span class="keyword">auto</span> index = group * channelMultiplier + m;</div> |
| 2540 | <div class="line"><a name="l02442"></a><span class="lineno"> 2442</span>  convLayers[index]->GetOutputSlot(0).Connect(concatLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(index));</div> |
| 2541 | <div class="line"><a name="l02443"></a><span class="lineno"> 2443</span>  }</div> |
| 2542 | <div class="line"><a name="l02444"></a><span class="lineno"> 2444</span>  }</div> |
| 2543 | <div class="line"><a name="l02445"></a><span class="lineno"> 2445</span>  concatLayer-><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#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(outputInfo);</div> |
| 2544 | <div class="line"><a name="l02446"></a><span class="lineno"> 2446</span>  </div> |
| 2545 | <div class="line"><a name="l02447"></a><span class="lineno"> 2447</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *concatLayer, model,</div> |
| 2546 | <div class="line"><a name="l02448"></a><span class="lineno"> 2448</span>  data, <span class="keyword">nullptr</span>, <span class="keyword">nullptr</span>, activation);</div> |
| 2547 | <div class="line"><a name="l02449"></a><span class="lineno"> 2449</span> }</div> |
| 2548 | <div class="line"><a name="l02450"></a><span class="lineno"> 2450</span>  </div> |
| 2549 | <div class="line"><a name="l02451"></a><span class="lineno"> 2451</span> <span class="keywordtype">bool</span> Converter::ConvertHardSwish(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2550 | <div class="line"><a name="l02452"></a><span class="lineno"> 2452</span> {</div> |
| 2551 | <div class="line"><a name="l02453"></a><span class="lineno"> 2453</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertHardSwish()"</span>;</div> |
| 2552 | <div class="line"><a name="l02454"></a><span class="lineno"> 2454</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a> desc;</div> |
| 2553 | <div class="line"><a name="l02455"></a><span class="lineno"> 2455</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = ActivationFunction::HardSwish;</div> |
| 2554 | <div class="line"><a name="l02456"></a><span class="lineno"> 2456</span>  </div> |
| 2555 | <div class="line"><a name="l02457"></a><span class="lineno"> 2457</span>  <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">return ::ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 2556 | <div class="line"><a name="l02458"></a><span class="lineno"> 2458</span> }</div> |
| 2557 | <div class="line"><a name="l02459"></a><span class="lineno"> 2459</span>  </div> |
| 2558 | <div class="line"><a name="l02460"></a><span class="lineno"> 2460</span> <span class="keywordtype">bool</span> Converter::ConvertInstanceNormalization(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2559 | <div class="line"><a name="l02461"></a><span class="lineno"> 2461</span> {</div> |
| 2560 | <div class="line"><a name="l02462"></a><span class="lineno"> 2462</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertInstanceNormalization()"</span>;</div> |
| 2561 | <div class="line"><a name="l02463"></a><span class="lineno"> 2463</span>  </div> |
| 2562 | <div class="line"><a name="l02464"></a><span class="lineno"> 2464</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2563 | <div class="line"><a name="l02465"></a><span class="lineno"> 2465</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2564 | <div class="line"><a name="l02466"></a><span class="lineno"> 2466</span>  {</div> |
| 2565 | <div class="line"><a name="l02467"></a><span class="lineno"> 2467</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has an invalid input 0"</span>, __func__);</div> |
| 2566 | <div class="line"><a name="l02468"></a><span class="lineno"> 2468</span>  }</div> |
| 2567 | <div class="line"><a name="l02469"></a><span class="lineno"> 2469</span>  </div> |
| 2568 | <div class="line"><a name="l02470"></a><span class="lineno"> 2470</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2569 | <div class="line"><a name="l02471"></a><span class="lineno"> 2471</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2570 | <div class="line"><a name="l02472"></a><span class="lineno"> 2472</span>  {</div> |
| 2571 | <div class="line"><a name="l02473"></a><span class="lineno"> 2473</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has an invalid output"</span>, __func__);</div> |
| 2572 | <div class="line"><a name="l02474"></a><span class="lineno"> 2474</span>  }</div> |
| 2573 | <div class="line"><a name="l02475"></a><span class="lineno"> 2475</span>  </div> |
| 2574 | <div class="line"><a name="l02476"></a><span class="lineno"> 2476</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2575 | <div class="line"><a name="l02477"></a><span class="lineno"> 2477</span>  </div> |
| 2576 | <div class="line"><a name="l02478"></a><span class="lineno"> 2478</span>  <span class="comment">// Determine data type of input tensor</span></div> |
| 2577 | <div class="line"><a name="l02479"></a><span class="lineno"> 2479</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> inputType;</div> |
| 2578 | <div class="line"><a name="l02480"></a><span class="lineno"> 2480</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 0, model, inputType))</div> |
| 2579 | <div class="line"><a name="l02481"></a><span class="lineno"> 2481</span>  {</div> |
| 2580 | <div class="line"><a name="l02482"></a><span class="lineno"> 2482</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2581 | <div class="line"><a name="l02483"></a><span class="lineno"> 2483</span>  }</div> |
| 2582 | <div class="line"><a name="l02484"></a><span class="lineno"> 2484</span>  </div> |
| 2583 | <div class="line"><a name="l02485"></a><span class="lineno"> 2485</span>  <a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html">InstanceNormalizationDescriptor</a> desc;</div> |
| 2584 | <div class="line"><a name="l02486"></a><span class="lineno"> 2486</span>  </div> |
| 2585 | <div class="line"><a name="l02487"></a><span class="lineno"> 2487</span>  <span class="comment">// Read gamma, beta & epsilon</span></div> |
| 2586 | <div class="line"><a name="l02488"></a><span class="lineno"> 2488</span>  <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT16)</div> |
| 2587 | <div class="line"><a name="l02489"></a><span class="lineno"> 2489</span>  {</div> |
| 2588 | <div class="line"><a name="l02490"></a><span class="lineno"> 2490</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Gamma;</div> |
| 2589 | <div class="line"><a name="l02491"></a><span class="lineno"> 2491</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Beta;</div> |
| 2590 | <div class="line"><a name="l02492"></a><span class="lineno"> 2492</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Epsilon;</div> |
| 2591 | <div class="line"><a name="l02493"></a><span class="lineno"> 2493</span>  </div> |
| 2592 | <div class="line"><a name="l02494"></a><span class="lineno"> 2494</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT16, fp16Gamma, model, data) ||</div> |
| 2593 | <div class="line"><a name="l02495"></a><span class="lineno"> 2495</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 2, OperandType::FLOAT16, fp16Beta, model, data) ||</div> |
| 2594 | <div class="line"><a name="l02496"></a><span class="lineno"> 2496</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 3, OperandType::FLOAT16, fp16Epsilon, model, data))</div> |
| 2595 | <div class="line"><a name="l02497"></a><span class="lineno"> 2497</span>  {</div> |
| 2596 | <div class="line"><a name="l02498"></a><span class="lineno"> 2498</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (FLOAT16)"</span>, __func__);</div> |
| 2597 | <div class="line"><a name="l02499"></a><span class="lineno"> 2499</span>  }</div> |
| 2598 | <div class="line"><a name="l02500"></a><span class="lineno"> 2500</span>  </div> |
| 2599 | <div class="line"><a name="l02501"></a><span class="lineno"> 2501</span>  desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a5e078fd505aef7bccaa05c8058e096cc">m_Gamma</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(fp16Gamma);</div> |
| 2600 | <div class="line"><a name="l02502"></a><span class="lineno"> 2502</span>  desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(fp16Beta);</div> |
| 2601 | <div class="line"><a name="l02503"></a><span class="lineno"> 2503</span>  desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a11c821c7524251004a72ed13c510853c">m_Eps</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(fp16Epsilon);</div> |
| 2602 | <div class="line"><a name="l02504"></a><span class="lineno"> 2504</span>  }</div> |
| 2603 | <div class="line"><a name="l02505"></a><span class="lineno"> 2505</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT32)</div> |
| 2604 | <div class="line"><a name="l02506"></a><span class="lineno"> 2506</span>  {</div> |
| 2605 | <div class="line"><a name="l02507"></a><span class="lineno"> 2507</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT32, desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a5e078fd505aef7bccaa05c8058e096cc">m_Gamma</a>, model, data) ||</div> |
| 2606 | <div class="line"><a name="l02508"></a><span class="lineno"> 2508</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 2, OperandType::FLOAT32, desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a>, model, data) ||</div> |
| 2607 | <div class="line"><a name="l02509"></a><span class="lineno"> 2509</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 3, OperandType::FLOAT32, desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a11c821c7524251004a72ed13c510853c">m_Eps</a>, model, data))</div> |
| 2608 | <div class="line"><a name="l02510"></a><span class="lineno"> 2510</span>  {</div> |
| 2609 | <div class="line"><a name="l02511"></a><span class="lineno"> 2511</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (FLOAT32)"</span>, __func__);</div> |
| 2610 | <div class="line"><a name="l02512"></a><span class="lineno"> 2512</span>  }</div> |
| 2611 | <div class="line"><a name="l02513"></a><span class="lineno"> 2513</span>  }</div> |
| 2612 | <div class="line"><a name="l02514"></a><span class="lineno"> 2514</span>  <span class="keywordflow">else</span></div> |
| 2613 | <div class="line"><a name="l02515"></a><span class="lineno"> 2515</span>  {</div> |
| 2614 | <div class="line"><a name="l02516"></a><span class="lineno"> 2516</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported input tensor type: %d"</span>, __func__, inputType);</div> |
| 2615 | <div class="line"><a name="l02517"></a><span class="lineno"> 2517</span>  }</div> |
| 2616 | <div class="line"><a name="l02518"></a><span class="lineno"> 2518</span>  </div> |
| 2617 | <div class="line"><a name="l02519"></a><span class="lineno"> 2519</span>  desc.<a class="code" href="structarmnn_1_1_instance_normalization_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 4, model, data);</div> |
| 2618 | <div class="line"><a name="l02520"></a><span class="lineno"> 2520</span>  </div> |
| 2619 | <div class="line"><a name="l02521"></a><span class="lineno"> 2521</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2620 | <div class="line"><a name="l02522"></a><span class="lineno"> 2522</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2621 | <div class="line"><a name="l02523"></a><span class="lineno"> 2523</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2622 | <div class="line"><a name="l02524"></a><span class="lineno"> 2524</span>  {</div> |
| 2623 | <div class="line"><a name="l02525"></a><span class="lineno"> 2525</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2624 | <div class="line"><a name="l02526"></a><span class="lineno"> 2526</span>  IsInstanceNormalizationSupported,</div> |
| 2625 | <div class="line"><a name="l02527"></a><span class="lineno"> 2527</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2626 | <div class="line"><a name="l02528"></a><span class="lineno"> 2528</span>  isSupported,</div> |
| 2627 | <div class="line"><a name="l02529"></a><span class="lineno"> 2529</span>  setBackend,</div> |
| 2628 | <div class="line"><a name="l02530"></a><span class="lineno"> 2530</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 2629 | <div class="line"><a name="l02531"></a><span class="lineno"> 2531</span>  outputInfo,</div> |
| 2630 | <div class="line"><a name="l02532"></a><span class="lineno"> 2532</span>  desc);</div> |
| 2631 | <div class="line"><a name="l02533"></a><span class="lineno"> 2533</span>  };</div> |
| 2632 | <div class="line"><a name="l02534"></a><span class="lineno"> 2534</span>  </div> |
| 2633 | <div class="line"><a name="l02535"></a><span class="lineno"> 2535</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2634 | <div class="line"><a name="l02536"></a><span class="lineno"> 2536</span>  {</div> |
| 2635 | <div class="line"><a name="l02537"></a><span class="lineno"> 2537</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2636 | <div class="line"><a name="l02538"></a><span class="lineno"> 2538</span>  }</div> |
| 2637 | <div class="line"><a name="l02539"></a><span class="lineno"> 2539</span>  <span class="keywordflow">else</span></div> |
| 2638 | <div class="line"><a name="l02540"></a><span class="lineno"> 2540</span>  {</div> |
| 2639 | <div class="line"><a name="l02541"></a><span class="lineno"> 2541</span>  validateFunc(outputInfo, isSupported);</div> |
| 2640 | <div class="line"><a name="l02542"></a><span class="lineno"> 2542</span>  }</div> |
| 2641 | <div class="line"><a name="l02543"></a><span class="lineno"> 2543</span>  </div> |
| 2642 | <div class="line"><a name="l02544"></a><span class="lineno"> 2544</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2643 | <div class="line"><a name="l02545"></a><span class="lineno"> 2545</span>  {</div> |
| 2644 | <div class="line"><a name="l02546"></a><span class="lineno"> 2546</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2645 | <div class="line"><a name="l02547"></a><span class="lineno"> 2547</span>  }</div> |
| 2646 | <div class="line"><a name="l02548"></a><span class="lineno"> 2548</span>  </div> |
| 2647 | <div class="line"><a name="l02549"></a><span class="lineno"> 2549</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddInstanceNormalizationLayer(desc);</div> |
| 2648 | <div class="line"><a name="l02550"></a><span class="lineno"> 2550</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2649 | <div class="line"><a name="l02551"></a><span class="lineno"> 2551</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2650 | <div class="line"><a name="l02552"></a><span class="lineno"> 2552</span>  </div> |
| 2651 | <div class="line"><a name="l02553"></a><span class="lineno"> 2553</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 2652 | <div class="line"><a name="l02554"></a><span class="lineno"> 2554</span> }</div> |
| 2653 | <div class="line"><a name="l02555"></a><span class="lineno"> 2555</span>  </div> |
| 2654 | <div class="line"><a name="l02556"></a><span class="lineno"> 2556</span> <span class="keywordtype">bool</span> Converter::ConvertL2Normalization(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2655 | <div class="line"><a name="l02557"></a><span class="lineno"> 2557</span> {</div> |
| 2656 | <div class="line"><a name="l02558"></a><span class="lineno"> 2558</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertL2Normalization()"</span>;</div> |
| 2657 | <div class="line"><a name="l02559"></a><span class="lineno"> 2559</span>  </div> |
| 2658 | <div class="line"><a name="l02560"></a><span class="lineno"> 2560</span>  <span class="keywordflow">if</span> (operation.inputs.size() != 1)</div> |
| 2659 | <div class="line"><a name="l02561"></a><span class="lineno"> 2561</span>  {</div> |
| 2660 | <div class="line"><a name="l02562"></a><span class="lineno"> 2562</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Optional inputs are not supported"</span>, __func__);</div> |
| 2661 | <div class="line"><a name="l02563"></a><span class="lineno"> 2563</span>  }</div> |
| 2662 | <div class="line"><a name="l02564"></a><span class="lineno"> 2564</span>  </div> |
| 2663 | <div class="line"><a name="l02565"></a><span class="lineno"> 2565</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2664 | <div class="line"><a name="l02566"></a><span class="lineno"> 2566</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2665 | <div class="line"><a name="l02567"></a><span class="lineno"> 2567</span>  {</div> |
| 2666 | <div class="line"><a name="l02568"></a><span class="lineno"> 2568</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2667 | <div class="line"><a name="l02569"></a><span class="lineno"> 2569</span>  }</div> |
| 2668 | <div class="line"><a name="l02570"></a><span class="lineno"> 2570</span>  </div> |
| 2669 | <div class="line"><a name="l02571"></a><span class="lineno"> 2571</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2670 | <div class="line"><a name="l02572"></a><span class="lineno"> 2572</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2671 | <div class="line"><a name="l02573"></a><span class="lineno"> 2573</span>  {</div> |
| 2672 | <div class="line"><a name="l02574"></a><span class="lineno"> 2574</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 2673 | <div class="line"><a name="l02575"></a><span class="lineno"> 2575</span>  }</div> |
| 2674 | <div class="line"><a name="l02576"></a><span class="lineno"> 2576</span>  </div> |
| 2675 | <div class="line"><a name="l02577"></a><span class="lineno"> 2577</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2676 | <div class="line"><a name="l02578"></a><span class="lineno"> 2578</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2677 | <div class="line"><a name="l02579"></a><span class="lineno"> 2579</span>  </div> |
| 2678 | <div class="line"><a name="l02580"></a><span class="lineno"> 2580</span>  <span class="keywordflow">if</span> (outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() != 4u)</div> |
| 2679 | <div class="line"><a name="l02581"></a><span class="lineno"> 2581</span>  {</div> |
| 2680 | <div class="line"><a name="l02582"></a><span class="lineno"> 2582</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Tensor Rank other than 4 is not supported"</span>, __func__);</div> |
| 2681 | <div class="line"><a name="l02583"></a><span class="lineno"> 2583</span>  }</div> |
| 2682 | <div class="line"><a name="l02584"></a><span class="lineno"> 2584</span>  </div> |
| 2683 | <div class="line"><a name="l02585"></a><span class="lineno"> 2585</span>  <a class="code" href="structarmnn_1_1_l2_normalization_descriptor.html">armnn::L2NormalizationDescriptor</a> desc;</div> |
| 2684 | <div class="line"><a name="l02586"></a><span class="lineno"> 2586</span>  desc.<a class="code" href="structarmnn_1_1_l2_normalization_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 2685 | <div class="line"><a name="l02587"></a><span class="lineno"> 2587</span>  </div> |
| 2686 | <div class="line"><a name="l02588"></a><span class="lineno"> 2588</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2687 | <div class="line"><a name="l02589"></a><span class="lineno"> 2589</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2688 | <div class="line"><a name="l02590"></a><span class="lineno"> 2590</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2689 | <div class="line"><a name="l02591"></a><span class="lineno"> 2591</span>  {</div> |
| 2690 | <div class="line"><a name="l02592"></a><span class="lineno"> 2592</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2691 | <div class="line"><a name="l02593"></a><span class="lineno"> 2593</span>  IsL2NormalizationSupported,</div> |
| 2692 | <div class="line"><a name="l02594"></a><span class="lineno"> 2594</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2693 | <div class="line"><a name="l02595"></a><span class="lineno"> 2595</span>  isSupported,</div> |
| 2694 | <div class="line"><a name="l02596"></a><span class="lineno"> 2596</span>  setBackend,</div> |
| 2695 | <div class="line"><a name="l02597"></a><span class="lineno"> 2597</span>  inputInfo,</div> |
| 2696 | <div class="line"><a name="l02598"></a><span class="lineno"> 2598</span>  outputInfo,</div> |
| 2697 | <div class="line"><a name="l02599"></a><span class="lineno"> 2599</span>  desc);</div> |
| 2698 | <div class="line"><a name="l02600"></a><span class="lineno"> 2600</span>  };</div> |
| 2699 | <div class="line"><a name="l02601"></a><span class="lineno"> 2601</span>  </div> |
| 2700 | <div class="line"><a name="l02602"></a><span class="lineno"> 2602</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2701 | <div class="line"><a name="l02603"></a><span class="lineno"> 2603</span>  {</div> |
| 2702 | <div class="line"><a name="l02604"></a><span class="lineno"> 2604</span>  validateFunc(outputInfo, isSupported);</div> |
| 2703 | <div class="line"><a name="l02605"></a><span class="lineno"> 2605</span>  }</div> |
| 2704 | <div class="line"><a name="l02606"></a><span class="lineno"> 2606</span>  <span class="keywordflow">else</span></div> |
| 2705 | <div class="line"><a name="l02607"></a><span class="lineno"> 2607</span>  {</div> |
| 2706 | <div class="line"><a name="l02608"></a><span class="lineno"> 2608</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2707 | <div class="line"><a name="l02609"></a><span class="lineno"> 2609</span>  }</div> |
| 2708 | <div class="line"><a name="l02610"></a><span class="lineno"> 2610</span>  </div> |
| 2709 | <div class="line"><a name="l02611"></a><span class="lineno"> 2611</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2710 | <div class="line"><a name="l02612"></a><span class="lineno"> 2612</span>  {</div> |
| 2711 | <div class="line"><a name="l02613"></a><span class="lineno"> 2613</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2712 | <div class="line"><a name="l02614"></a><span class="lineno"> 2614</span>  }</div> |
| 2713 | <div class="line"><a name="l02615"></a><span class="lineno"> 2615</span>  </div> |
| 2714 | <div class="line"><a name="l02616"></a><span class="lineno"> 2616</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddL2NormalizationLayer(desc);</div> |
| 2715 | <div class="line"><a name="l02617"></a><span class="lineno"> 2617</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2716 | <div class="line"><a name="l02618"></a><span class="lineno"> 2618</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 2717 | <div class="line"><a name="l02619"></a><span class="lineno"> 2619</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2718 | <div class="line"><a name="l02620"></a><span class="lineno"> 2620</span>  </div> |
| 2719 | <div class="line"><a name="l02621"></a><span class="lineno"> 2621</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 2720 | <div class="line"><a name="l02622"></a><span class="lineno"> 2622</span> }</div> |
| 2721 | <div class="line"><a name="l02623"></a><span class="lineno"> 2623</span>  </div> |
| 2722 | <div class="line"><a name="l02624"></a><span class="lineno"> 2624</span> <span class="keywordtype">bool</span> Converter::ConvertL2Pool2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2723 | <div class="line"><a name="l02625"></a><span class="lineno"> 2625</span> {</div> |
| 2724 | <div class="line"><a name="l02626"></a><span class="lineno"> 2626</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertL2Pool2d()"</span>;</div> |
| 2725 | <div class="line"><a name="l02627"></a><span class="lineno"> 2627</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a2eb75cb48464cf87c290503d7bea79a4">ConvertPooling2d</a>(operation, __func__, PoolingAlgorithm::L2, model, data);</div> |
| 2726 | <div class="line"><a name="l02628"></a><span class="lineno"> 2628</span> }</div> |
| 2727 | <div class="line"><a name="l02629"></a><span class="lineno"> 2629</span>  </div> |
| 2728 | <div class="line"><a name="l02630"></a><span class="lineno"> 2630</span> <span class="keywordtype">bool</span> Converter::ConvertLocalResponseNormalization(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 2729 | <div class="line"><a name="l02631"></a><span class="lineno"> 2631</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 2730 | <div class="line"><a name="l02632"></a><span class="lineno"> 2632</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2731 | <div class="line"><a name="l02633"></a><span class="lineno"> 2633</span> {</div> |
| 2732 | <div class="line"><a name="l02634"></a><span class="lineno"> 2634</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLocalResponseNormalization()"</span>;</div> |
| 2733 | <div class="line"><a name="l02635"></a><span class="lineno"> 2635</span>  </div> |
| 2734 | <div class="line"><a name="l02636"></a><span class="lineno"> 2636</span>  <span class="keywordflow">if</span> (operation.inputs.size() != 5)</div> |
| 2735 | <div class="line"><a name="l02637"></a><span class="lineno"> 2637</span>  {</div> |
| 2736 | <div class="line"><a name="l02638"></a><span class="lineno"> 2638</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Optional inputs are not supported"</span>, __func__);</div> |
| 2737 | <div class="line"><a name="l02639"></a><span class="lineno"> 2639</span>  }</div> |
| 2738 | <div class="line"><a name="l02640"></a><span class="lineno"> 2640</span>  </div> |
| 2739 | <div class="line"><a name="l02641"></a><span class="lineno"> 2641</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2740 | <div class="line"><a name="l02642"></a><span class="lineno"> 2642</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2741 | <div class="line"><a name="l02643"></a><span class="lineno"> 2643</span>  {</div> |
| 2742 | <div class="line"><a name="l02644"></a><span class="lineno"> 2644</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2743 | <div class="line"><a name="l02645"></a><span class="lineno"> 2645</span>  }</div> |
| 2744 | <div class="line"><a name="l02646"></a><span class="lineno"> 2646</span>  </div> |
| 2745 | <div class="line"><a name="l02647"></a><span class="lineno"> 2647</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2746 | <div class="line"><a name="l02648"></a><span class="lineno"> 2648</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2747 | <div class="line"><a name="l02649"></a><span class="lineno"> 2649</span>  {</div> |
| 2748 | <div class="line"><a name="l02650"></a><span class="lineno"> 2650</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 2749 | <div class="line"><a name="l02651"></a><span class="lineno"> 2651</span>  }</div> |
| 2750 | <div class="line"><a name="l02652"></a><span class="lineno"> 2652</span>  </div> |
| 2751 | <div class="line"><a name="l02653"></a><span class="lineno"> 2653</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2752 | <div class="line"><a name="l02654"></a><span class="lineno"> 2654</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2753 | <div class="line"><a name="l02655"></a><span class="lineno"> 2655</span>  </div> |
| 2754 | <div class="line"><a name="l02656"></a><span class="lineno"> 2656</span>  <span class="keywordflow">if</span> (outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() != 4u)</div> |
| 2755 | <div class="line"><a name="l02657"></a><span class="lineno"> 2657</span>  {</div> |
| 2756 | <div class="line"><a name="l02658"></a><span class="lineno"> 2658</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Tensor Rank other than 4 is not supported"</span>, __func__);</div> |
| 2757 | <div class="line"><a name="l02659"></a><span class="lineno"> 2659</span>  }</div> |
| 2758 | <div class="line"><a name="l02660"></a><span class="lineno"> 2660</span>  </div> |
| 2759 | <div class="line"><a name="l02661"></a><span class="lineno"> 2661</span>  <a class="code" href="structarmnn_1_1_normalization_descriptor.html">armnn::NormalizationDescriptor</a> descriptor;</div> |
| 2760 | <div class="line"><a name="l02662"></a><span class="lineno"> 2662</span>  descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 2761 | <div class="line"><a name="l02663"></a><span class="lineno"> 2663</span>  descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#afe1f0f09d49ad2befc01f8789187b7dd">m_NormChannelType</a> = <a class="code" href="namespacearmnn.html#abe18a5033f2ab9c0de82c676b48f5437a810f43f3996922151c39b76143faeecc">armnn::NormalizationAlgorithmChannel::Across</a>;</div> |
| 2762 | <div class="line"><a name="l02664"></a><span class="lineno"> 2664</span>  descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#a05945f080edf694b631960728b87aadb">m_NormMethodType</a> = <a class="code" href="namespacearmnn.html#ad605d1661fa0d8c7fea651d82fbe11c9ac65d2e51c610dd3853a3c777aa8bfe9d">armnn::NormalizationAlgorithmMethod::LocalBrightness</a>;</div> |
| 2763 | <div class="line"><a name="l02665"></a><span class="lineno"> 2665</span>  </div> |
| 2764 | <div class="line"><a name="l02666"></a><span class="lineno"> 2666</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 2765 | <div class="line"><a name="l02667"></a><span class="lineno"> 2667</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#aa70c05f1aad12fbd9d9ec43ea4557b03">m_NormSize</a>, model, data) ||</div> |
| 2766 | <div class="line"><a name="l02668"></a><span class="lineno"> 2668</span>  !<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 2, descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#a8526ea7cf860d8e7f8340e9f9354f9f0">m_K</a>, model, data) ||</div> |
| 2767 | <div class="line"><a name="l02669"></a><span class="lineno"> 2669</span>  !<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 3, descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#a174279be57d7596eeb04c6b7f7510f99">m_Alpha</a>, model, data) ||</div> |
| 2768 | <div class="line"><a name="l02670"></a><span class="lineno"> 2670</span>  !<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 4, descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a>, model, data))</div> |
| 2769 | <div class="line"><a name="l02671"></a><span class="lineno"> 2671</span>  {</div> |
| 2770 | <div class="line"><a name="l02672"></a><span class="lineno"> 2672</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2771 | <div class="line"><a name="l02673"></a><span class="lineno"> 2673</span>  }</div> |
| 2772 | <div class="line"><a name="l02674"></a><span class="lineno"> 2674</span>  </div> |
| 2773 | <div class="line"><a name="l02675"></a><span class="lineno"> 2675</span>  <span class="comment">// ArmNN expects normSize to be the full size of the normalization</span></div> |
| 2774 | <div class="line"><a name="l02676"></a><span class="lineno"> 2676</span>  <span class="comment">// window rather than the radius as in AndroidNN.</span></div> |
| 2775 | <div class="line"><a name="l02677"></a><span class="lineno"> 2677</span>  descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#aa70c05f1aad12fbd9d9ec43ea4557b03">m_NormSize</a> = 1 + (2 * descriptor.<a class="code" href="structarmnn_1_1_normalization_descriptor.html#aa70c05f1aad12fbd9d9ec43ea4557b03">m_NormSize</a>);</div> |
| 2776 | <div class="line"><a name="l02678"></a><span class="lineno"> 2678</span>  </div> |
| 2777 | <div class="line"><a name="l02679"></a><span class="lineno"> 2679</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2778 | <div class="line"><a name="l02680"></a><span class="lineno"> 2680</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2779 | <div class="line"><a name="l02681"></a><span class="lineno"> 2681</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2780 | <div class="line"><a name="l02682"></a><span class="lineno"> 2682</span>  {</div> |
| 2781 | <div class="line"><a name="l02683"></a><span class="lineno"> 2683</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2782 | <div class="line"><a name="l02684"></a><span class="lineno"> 2684</span>  IsNormalizationSupported,</div> |
| 2783 | <div class="line"><a name="l02685"></a><span class="lineno"> 2685</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2784 | <div class="line"><a name="l02686"></a><span class="lineno"> 2686</span>  isSupported,</div> |
| 2785 | <div class="line"><a name="l02687"></a><span class="lineno"> 2687</span>  setBackend,</div> |
| 2786 | <div class="line"><a name="l02688"></a><span class="lineno"> 2688</span>  inputInfo,</div> |
| 2787 | <div class="line"><a name="l02689"></a><span class="lineno"> 2689</span>  outputInfo,</div> |
| 2788 | <div class="line"><a name="l02690"></a><span class="lineno"> 2690</span>  descriptor);</div> |
| 2789 | <div class="line"><a name="l02691"></a><span class="lineno"> 2691</span>  };</div> |
| 2790 | <div class="line"><a name="l02692"></a><span class="lineno"> 2692</span>  </div> |
| 2791 | <div class="line"><a name="l02693"></a><span class="lineno"> 2693</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2792 | <div class="line"><a name="l02694"></a><span class="lineno"> 2694</span>  {</div> |
| 2793 | <div class="line"><a name="l02695"></a><span class="lineno"> 2695</span>  validateFunc(outputInfo, isSupported);</div> |
| 2794 | <div class="line"><a name="l02696"></a><span class="lineno"> 2696</span>  }</div> |
| 2795 | <div class="line"><a name="l02697"></a><span class="lineno"> 2697</span>  <span class="keywordflow">else</span></div> |
| 2796 | <div class="line"><a name="l02698"></a><span class="lineno"> 2698</span>  {</div> |
| 2797 | <div class="line"><a name="l02699"></a><span class="lineno"> 2699</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2798 | <div class="line"><a name="l02700"></a><span class="lineno"> 2700</span>  }</div> |
| 2799 | <div class="line"><a name="l02701"></a><span class="lineno"> 2701</span>  </div> |
| 2800 | <div class="line"><a name="l02702"></a><span class="lineno"> 2702</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2801 | <div class="line"><a name="l02703"></a><span class="lineno"> 2703</span>  {</div> |
| 2802 | <div class="line"><a name="l02704"></a><span class="lineno"> 2704</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2803 | <div class="line"><a name="l02705"></a><span class="lineno"> 2705</span>  }</div> |
| 2804 | <div class="line"><a name="l02706"></a><span class="lineno"> 2706</span>  </div> |
| 2805 | <div class="line"><a name="l02707"></a><span class="lineno"> 2707</span>  </div> |
| 2806 | <div class="line"><a name="l02708"></a><span class="lineno"> 2708</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddNormalizationLayer(descriptor);</div> |
| 2807 | <div class="line"><a name="l02709"></a><span class="lineno"> 2709</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2808 | <div class="line"><a name="l02710"></a><span class="lineno"> 2710</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 2809 | <div class="line"><a name="l02711"></a><span class="lineno"> 2711</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2810 | <div class="line"><a name="l02712"></a><span class="lineno"> 2712</span>  </div> |
| 2811 | <div class="line"><a name="l02713"></a><span class="lineno"> 2713</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 2812 | <div class="line"><a name="l02714"></a><span class="lineno"> 2714</span> }</div> |
| 2813 | <div class="line"><a name="l02715"></a><span class="lineno"> 2715</span>  </div> |
| 2814 | <div class="line"><a name="l02716"></a><span class="lineno"> 2716</span> <span class="keywordtype">bool</span> Converter::ConvertLogicalBinary(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 2815 | <div class="line"><a name="l02717"></a><span class="lineno"> 2717</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 2816 | <div class="line"><a name="l02718"></a><span class="lineno"> 2718</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data,</div> |
| 2817 | <div class="line"><a name="l02719"></a><span class="lineno"> 2719</span>  <a class="code" href="namespacearmnn.html#a2da4db0140d1a6dc69c9c82e9ef5379e">armnn::LogicalBinaryOperation</a> logicalOperation)</div> |
| 2818 | <div class="line"><a name="l02720"></a><span class="lineno"> 2720</span> {</div> |
| 2819 | <div class="line"><a name="l02721"></a><span class="lineno"> 2721</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLogicalBinary()"</span>;</div> |
| 2820 | <div class="line"><a name="l02722"></a><span class="lineno"> 2722</span>  VLOG(DRIVER) << <span class="stringliteral">"ConvertLogicalBinary()"</span>;</div> |
| 2821 | <div class="line"><a name="l02723"></a><span class="lineno"> 2723</span>  VLOG(DRIVER) << <span class="stringliteral">"logicalOperation = "</span> << <a class="code" href="namespacearmnn.html#a0ad431ff1600c845fda078ac58a47809">GetLogicalBinaryOperationAsCString</a>(logicalOperation);</div> |
| 2822 | <div class="line"><a name="l02724"></a><span class="lineno"> 2724</span>  </div> |
| 2823 | <div class="line"><a name="l02725"></a><span class="lineno"> 2725</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2824 | <div class="line"><a name="l02726"></a><span class="lineno"> 2726</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input1 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 2825 | <div class="line"><a name="l02727"></a><span class="lineno"> 2727</span>  </div> |
| 2826 | <div class="line"><a name="l02728"></a><span class="lineno"> 2728</span>  <span class="keywordflow">if</span> (!(input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>()))</div> |
| 2827 | <div class="line"><a name="l02729"></a><span class="lineno"> 2729</span>  {</div> |
| 2828 | <div class="line"><a name="l02730"></a><span class="lineno"> 2730</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2829 | <div class="line"><a name="l02731"></a><span class="lineno"> 2731</span>  }</div> |
| 2830 | <div class="line"><a name="l02732"></a><span class="lineno"> 2732</span>  </div> |
| 2831 | <div class="line"><a name="l02733"></a><span class="lineno"> 2733</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2832 | <div class="line"><a name="l02734"></a><span class="lineno"> 2734</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2833 | <div class="line"><a name="l02735"></a><span class="lineno"> 2735</span>  {</div> |
| 2834 | <div class="line"><a name="l02736"></a><span class="lineno"> 2736</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 2835 | <div class="line"><a name="l02737"></a><span class="lineno"> 2737</span>  }</div> |
| 2836 | <div class="line"><a name="l02738"></a><span class="lineno"> 2738</span>  </div> |
| 2837 | <div class="line"><a name="l02739"></a><span class="lineno"> 2739</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo0 = input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2838 | <div class="line"><a name="l02740"></a><span class="lineno"> 2740</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo1 = input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 2839 | <div class="line"><a name="l02741"></a><span class="lineno"> 2741</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2840 | <div class="line"><a name="l02742"></a><span class="lineno"> 2742</span>  </div> |
| 2841 | <div class="line"><a name="l02743"></a><span class="lineno"> 2743</span>  <a class="code" href="structarmnn_1_1_logical_binary_descriptor.html">LogicalBinaryDescriptor</a> descriptor(logicalOperation);</div> |
| 2842 | <div class="line"><a name="l02744"></a><span class="lineno"> 2744</span>  </div> |
| 2843 | <div class="line"><a name="l02745"></a><span class="lineno"> 2745</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2844 | <div class="line"><a name="l02746"></a><span class="lineno"> 2746</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2845 | <div class="line"><a name="l02747"></a><span class="lineno"> 2747</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2846 | <div class="line"><a name="l02748"></a><span class="lineno"> 2748</span>  {</div> |
| 2847 | <div class="line"><a name="l02749"></a><span class="lineno"> 2749</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2848 | <div class="line"><a name="l02750"></a><span class="lineno"> 2750</span>  IsLogicalBinarySupported,</div> |
| 2849 | <div class="line"><a name="l02751"></a><span class="lineno"> 2751</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2850 | <div class="line"><a name="l02752"></a><span class="lineno"> 2752</span>  isSupported,</div> |
| 2851 | <div class="line"><a name="l02753"></a><span class="lineno"> 2753</span>  setBackend,</div> |
| 2852 | <div class="line"><a name="l02754"></a><span class="lineno"> 2754</span>  inputInfo0,</div> |
| 2853 | <div class="line"><a name="l02755"></a><span class="lineno"> 2755</span>  inputInfo1,</div> |
| 2854 | <div class="line"><a name="l02756"></a><span class="lineno"> 2756</span>  outputInfo,</div> |
| 2855 | <div class="line"><a name="l02757"></a><span class="lineno"> 2757</span>  descriptor);</div> |
| 2856 | <div class="line"><a name="l02758"></a><span class="lineno"> 2758</span>  };</div> |
| 2857 | <div class="line"><a name="l02759"></a><span class="lineno"> 2759</span>  </div> |
| 2858 | <div class="line"><a name="l02760"></a><span class="lineno"> 2760</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2859 | <div class="line"><a name="l02761"></a><span class="lineno"> 2761</span>  {</div> |
| 2860 | <div class="line"><a name="l02762"></a><span class="lineno"> 2762</span>  validateFunc(outputInfo, isSupported);</div> |
| 2861 | <div class="line"><a name="l02763"></a><span class="lineno"> 2763</span>  }</div> |
| 2862 | <div class="line"><a name="l02764"></a><span class="lineno"> 2764</span>  <span class="keywordflow">else</span></div> |
| 2863 | <div class="line"><a name="l02765"></a><span class="lineno"> 2765</span>  {</div> |
| 2864 | <div class="line"><a name="l02766"></a><span class="lineno"> 2766</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2865 | <div class="line"><a name="l02767"></a><span class="lineno"> 2767</span>  }</div> |
| 2866 | <div class="line"><a name="l02768"></a><span class="lineno"> 2768</span>  </div> |
| 2867 | <div class="line"><a name="l02769"></a><span class="lineno"> 2769</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2868 | <div class="line"><a name="l02770"></a><span class="lineno"> 2770</span>  {</div> |
| 2869 | <div class="line"><a name="l02771"></a><span class="lineno"> 2771</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2870 | <div class="line"><a name="l02772"></a><span class="lineno"> 2772</span>  }</div> |
| 2871 | <div class="line"><a name="l02773"></a><span class="lineno"> 2773</span>  </div> |
| 2872 | <div class="line"><a name="l02774"></a><span class="lineno"> 2774</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddLogicalBinaryLayer(descriptor);</div> |
| 2873 | <div class="line"><a name="l02775"></a><span class="lineno"> 2775</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2874 | <div class="line"><a name="l02776"></a><span class="lineno"> 2776</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 2875 | <div class="line"><a name="l02777"></a><span class="lineno"> 2777</span>  </div> |
| 2876 | <div class="line"><a name="l02778"></a><span class="lineno"> 2778</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input0, input1, layer, data);</div> |
| 2877 | <div class="line"><a name="l02779"></a><span class="lineno"> 2779</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| 2878 | <div class="line"><a name="l02780"></a><span class="lineno"> 2780</span>  {</div> |
| 2879 | <div class="line"><a name="l02781"></a><span class="lineno"> 2781</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2880 | <div class="line"><a name="l02782"></a><span class="lineno"> 2782</span>  }</div> |
| 2881 | <div class="line"><a name="l02783"></a><span class="lineno"> 2783</span>  </div> |
| 2882 | <div class="line"><a name="l02784"></a><span class="lineno"> 2784</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 2883 | <div class="line"><a name="l02785"></a><span class="lineno"> 2785</span> }</div> |
| 2884 | <div class="line"><a name="l02786"></a><span class="lineno"> 2786</span>  </div> |
| 2885 | <div class="line"><a name="l02787"></a><span class="lineno"> 2787</span> <span class="keywordtype">bool</span> Converter::ConvertLogistic(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2886 | <div class="line"><a name="l02788"></a><span class="lineno"> 2788</span> {</div> |
| 2887 | <div class="line"><a name="l02789"></a><span class="lineno"> 2789</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLogistic()"</span>;</div> |
| 2888 | <div class="line"><a name="l02790"></a><span class="lineno"> 2790</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">armnn::ActivationDescriptor</a> desc;</div> |
| 2889 | <div class="line"><a name="l02791"></a><span class="lineno"> 2791</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = <a class="code" href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaa21eebb164e4b8b9bcf64fdb4d8d5dff4">armnn::ActivationFunction::Sigmoid</a>;</div> |
| 2890 | <div class="line"><a name="l02792"></a><span class="lineno"> 2792</span>  </div> |
| 2891 | <div class="line"><a name="l02793"></a><span class="lineno"> 2793</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 2892 | <div class="line"><a name="l02794"></a><span class="lineno"> 2794</span> }</div> |
| 2893 | <div class="line"><a name="l02795"></a><span class="lineno"> 2795</span>  </div> |
| 2894 | <div class="line"><a name="l02796"></a><span class="lineno"> 2796</span> <span class="keywordtype">bool</span> Converter::ConvertLogSoftmax(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2895 | <div class="line"><a name="l02797"></a><span class="lineno"> 2797</span> {</div> |
| 2896 | <div class="line"><a name="l02798"></a><span class="lineno"> 2798</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLogSoftmax()"</span>;</div> |
| 2897 | <div class="line"><a name="l02799"></a><span class="lineno"> 2799</span>  </div> |
| 2898 | <div class="line"><a name="l02800"></a><span class="lineno"> 2800</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2899 | <div class="line"><a name="l02801"></a><span class="lineno"> 2801</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2900 | <div class="line"><a name="l02802"></a><span class="lineno"> 2802</span>  {</div> |
| 2901 | <div class="line"><a name="l02803"></a><span class="lineno"> 2803</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to read input 0"</span>, __func__);</div> |
| 2902 | <div class="line"><a name="l02804"></a><span class="lineno"> 2804</span>  }</div> |
| 2903 | <div class="line"><a name="l02805"></a><span class="lineno"> 2805</span>  </div> |
| 2904 | <div class="line"><a name="l02806"></a><span class="lineno"> 2806</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 2905 | <div class="line"><a name="l02807"></a><span class="lineno"> 2807</span>  <span class="keywordflow">if</span> (!output)</div> |
| 2906 | <div class="line"><a name="l02808"></a><span class="lineno"> 2808</span>  {</div> |
| 2907 | <div class="line"><a name="l02809"></a><span class="lineno"> 2809</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to read output"</span>, __func__);</div> |
| 2908 | <div class="line"><a name="l02810"></a><span class="lineno"> 2810</span>  }</div> |
| 2909 | <div class="line"><a name="l02811"></a><span class="lineno"> 2811</span>  </div> |
| 2910 | <div class="line"><a name="l02812"></a><span class="lineno"> 2812</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 2911 | <div class="line"><a name="l02813"></a><span class="lineno"> 2813</span>  </div> |
| 2912 | <div class="line"><a name="l02814"></a><span class="lineno"> 2814</span>  <span class="comment">// Determine data type of input tensor</span></div> |
| 2913 | <div class="line"><a name="l02815"></a><span class="lineno"> 2815</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> inputType;</div> |
| 2914 | <div class="line"><a name="l02816"></a><span class="lineno"> 2816</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 0, model, inputType))</div> |
| 2915 | <div class="line"><a name="l02817"></a><span class="lineno"> 2817</span>  {</div> |
| 2916 | <div class="line"><a name="l02818"></a><span class="lineno"> 2818</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 2917 | <div class="line"><a name="l02819"></a><span class="lineno"> 2819</span>  }</div> |
| 2918 | <div class="line"><a name="l02820"></a><span class="lineno"> 2820</span>  </div> |
| 2919 | <div class="line"><a name="l02821"></a><span class="lineno"> 2821</span>  <a class="code" href="structarmnn_1_1_softmax_descriptor.html">LogSoftmaxDescriptor</a> descriptor;</div> |
| 2920 | <div class="line"><a name="l02822"></a><span class="lineno"> 2822</span>  </div> |
| 2921 | <div class="line"><a name="l02823"></a><span class="lineno"> 2823</span>  <span class="comment">// Read beta</span></div> |
| 2922 | <div class="line"><a name="l02824"></a><span class="lineno"> 2824</span>  <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT16)</div> |
| 2923 | <div class="line"><a name="l02825"></a><span class="lineno"> 2825</span>  {</div> |
| 2924 | <div class="line"><a name="l02826"></a><span class="lineno"> 2826</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Beta;</div> |
| 2925 | <div class="line"><a name="l02827"></a><span class="lineno"> 2827</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT16, fp16Beta, model, data))</div> |
| 2926 | <div class="line"><a name="l02828"></a><span class="lineno"> 2828</span>  {</div> |
| 2927 | <div class="line"><a name="l02829"></a><span class="lineno"> 2829</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to read input 1 (FLOAT16)"</span>, __func__);</div> |
| 2928 | <div class="line"><a name="l02830"></a><span class="lineno"> 2830</span>  }</div> |
| 2929 | <div class="line"><a name="l02831"></a><span class="lineno"> 2831</span>  </div> |
| 2930 | <div class="line"><a name="l02832"></a><span class="lineno"> 2832</span>  descriptor.<a class="code" href="structarmnn_1_1_softmax_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(fp16Beta);</div> |
| 2931 | <div class="line"><a name="l02833"></a><span class="lineno"> 2833</span>  }</div> |
| 2932 | <div class="line"><a name="l02834"></a><span class="lineno"> 2834</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT32)</div> |
| 2933 | <div class="line"><a name="l02835"></a><span class="lineno"> 2835</span>  {</div> |
| 2934 | <div class="line"><a name="l02836"></a><span class="lineno"> 2836</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT32, descriptor.<a class="code" href="structarmnn_1_1_softmax_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a>, model, data))</div> |
| 2935 | <div class="line"><a name="l02837"></a><span class="lineno"> 2837</span>  {</div> |
| 2936 | <div class="line"><a name="l02838"></a><span class="lineno"> 2838</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to read input 1 (FLOAT32)"</span>, __func__);</div> |
| 2937 | <div class="line"><a name="l02839"></a><span class="lineno"> 2839</span>  }</div> |
| 2938 | <div class="line"><a name="l02840"></a><span class="lineno"> 2840</span>  }</div> |
| 2939 | <div class="line"><a name="l02841"></a><span class="lineno"> 2841</span>  <span class="keywordflow">else</span></div> |
| 2940 | <div class="line"><a name="l02842"></a><span class="lineno"> 2842</span>  {</div> |
| 2941 | <div class="line"><a name="l02843"></a><span class="lineno"> 2843</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported input tensor type: %d"</span>, __func__, inputType);</div> |
| 2942 | <div class="line"><a name="l02844"></a><span class="lineno"> 2844</span>  }</div> |
| 2943 | <div class="line"><a name="l02845"></a><span class="lineno"> 2845</span>  </div> |
| 2944 | <div class="line"><a name="l02846"></a><span class="lineno"> 2846</span>  <span class="comment">// Read axis</span></div> |
| 2945 | <div class="line"><a name="l02847"></a><span class="lineno"> 2847</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 2, descriptor.<a class="code" href="structarmnn_1_1_softmax_descriptor.html#a214c3636fdf0ea5bac8edb42d0e6c7f0">m_Axis</a>, model, data))</div> |
| 2946 | <div class="line"><a name="l02848"></a><span class="lineno"> 2848</span>  {</div> |
| 2947 | <div class="line"><a name="l02849"></a><span class="lineno"> 2849</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to read input 2"</span>, __func__);</div> |
| 2948 | <div class="line"><a name="l02850"></a><span class="lineno"> 2850</span>  }</div> |
| 2949 | <div class="line"><a name="l02851"></a><span class="lineno"> 2851</span>  </div> |
| 2950 | <div class="line"><a name="l02852"></a><span class="lineno"> 2852</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 2951 | <div class="line"><a name="l02853"></a><span class="lineno"> 2853</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 2952 | <div class="line"><a name="l02854"></a><span class="lineno"> 2854</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 2953 | <div class="line"><a name="l02855"></a><span class="lineno"> 2855</span>  {</div> |
| 2954 | <div class="line"><a name="l02856"></a><span class="lineno"> 2856</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 2955 | <div class="line"><a name="l02857"></a><span class="lineno"> 2857</span>  IsLogSoftmaxSupported,</div> |
| 2956 | <div class="line"><a name="l02858"></a><span class="lineno"> 2858</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 2957 | <div class="line"><a name="l02859"></a><span class="lineno"> 2859</span>  isSupported,</div> |
| 2958 | <div class="line"><a name="l02860"></a><span class="lineno"> 2860</span>  setBackend,</div> |
| 2959 | <div class="line"><a name="l02861"></a><span class="lineno"> 2861</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 2960 | <div class="line"><a name="l02862"></a><span class="lineno"> 2862</span>  outputInfo,</div> |
| 2961 | <div class="line"><a name="l02863"></a><span class="lineno"> 2863</span>  descriptor);</div> |
| 2962 | <div class="line"><a name="l02864"></a><span class="lineno"> 2864</span>  };</div> |
| 2963 | <div class="line"><a name="l02865"></a><span class="lineno"> 2865</span>  </div> |
| 2964 | <div class="line"><a name="l02866"></a><span class="lineno"> 2866</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 2965 | <div class="line"><a name="l02867"></a><span class="lineno"> 2867</span>  {</div> |
| 2966 | <div class="line"><a name="l02868"></a><span class="lineno"> 2868</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 2967 | <div class="line"><a name="l02869"></a><span class="lineno"> 2869</span>  }</div> |
| 2968 | <div class="line"><a name="l02870"></a><span class="lineno"> 2870</span>  <span class="keywordflow">else</span></div> |
| 2969 | <div class="line"><a name="l02871"></a><span class="lineno"> 2871</span>  {</div> |
| 2970 | <div class="line"><a name="l02872"></a><span class="lineno"> 2872</span>  validateFunc(outputInfo, isSupported);</div> |
| 2971 | <div class="line"><a name="l02873"></a><span class="lineno"> 2873</span>  }</div> |
| 2972 | <div class="line"><a name="l02874"></a><span class="lineno"> 2874</span>  </div> |
| 2973 | <div class="line"><a name="l02875"></a><span class="lineno"> 2875</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 2974 | <div class="line"><a name="l02876"></a><span class="lineno"> 2876</span>  {</div> |
| 2975 | <div class="line"><a name="l02877"></a><span class="lineno"> 2877</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 2976 | <div class="line"><a name="l02878"></a><span class="lineno"> 2878</span>  }</div> |
| 2977 | <div class="line"><a name="l02879"></a><span class="lineno"> 2879</span>  </div> |
| 2978 | <div class="line"><a name="l02880"></a><span class="lineno"> 2880</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddLogSoftmaxLayer(descriptor);</div> |
| 2979 | <div class="line"><a name="l02881"></a><span class="lineno"> 2881</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 2980 | <div class="line"><a name="l02882"></a><span class="lineno"> 2882</span>  <span class="keywordflow">if</span> (!layer)</div> |
| 2981 | <div class="line"><a name="l02883"></a><span class="lineno"> 2883</span>  {</div> |
| 2982 | <div class="line"><a name="l02884"></a><span class="lineno"> 2884</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddLogSoftmaxLayer() returned nullptr"</span>, __func__);</div> |
| 2983 | <div class="line"><a name="l02885"></a><span class="lineno"> 2885</span>  }</div> |
| 2984 | <div class="line"><a name="l02886"></a><span class="lineno"> 2886</span>  </div> |
| 2985 | <div class="line"><a name="l02887"></a><span class="lineno"> 2887</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 2986 | <div class="line"><a name="l02888"></a><span class="lineno"> 2888</span>  </div> |
| 2987 | <div class="line"><a name="l02889"></a><span class="lineno"> 2889</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 2988 | <div class="line"><a name="l02890"></a><span class="lineno"> 2890</span> }</div> |
| 2989 | <div class="line"><a name="l02891"></a><span class="lineno"> 2891</span>  </div> |
| 2990 | <div class="line"><a name="l02892"></a><span class="lineno"> 2892</span> <span class="keywordtype">bool</span> Converter::ConvertLstm(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 2991 | <div class="line"><a name="l02893"></a><span class="lineno"> 2893</span> {</div> |
| 2992 | <div class="line"><a name="l02894"></a><span class="lineno"> 2894</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLstm()"</span>;</div> |
| 2993 | <div class="line"><a name="l02895"></a><span class="lineno"> 2895</span>  </div> |
| 2994 | <div class="line"><a name="l02896"></a><span class="lineno"> 2896</span>  <span class="comment">// Inputs:</span></div> |
| 2995 | <div class="line"><a name="l02897"></a><span class="lineno"> 2897</span>  <span class="comment">// 00: The input: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, input_size], where</span></div> |
| 2996 | <div class="line"><a name="l02898"></a><span class="lineno"> 2898</span>  <span class="comment">// “batch_size” corresponds to the batching dimension, and “input_size” is the size of the input.</span></div> |
| 2997 | <div class="line"><a name="l02899"></a><span class="lineno"> 2899</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 2998 | <div class="line"><a name="l02900"></a><span class="lineno"> 2900</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 2999 | <div class="line"><a name="l02901"></a><span class="lineno"> 2901</span>  {</div> |
| 3000 | <div class="line"><a name="l02902"></a><span class="lineno"> 2902</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0: input"</span>, __func__);</div> |
| 3001 | <div class="line"><a name="l02903"></a><span class="lineno"> 2903</span>  }</div> |
| 3002 | <div class="line"><a name="l02904"></a><span class="lineno"> 2904</span>  <span class="comment">// 18: The output state: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, output_size].</span></div> |
| 3003 | <div class="line"><a name="l02905"></a><span class="lineno"> 2905</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> outputStateIn = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 18, model, data);</div> |
| 3004 | <div class="line"><a name="l02906"></a><span class="lineno"> 2906</span>  <span class="keywordflow">if</span> (!outputStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3005 | <div class="line"><a name="l02907"></a><span class="lineno"> 2907</span>  {</div> |
| 3006 | <div class="line"><a name="l02908"></a><span class="lineno"> 2908</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 18: outputStateIn"</span>, __func__);</div> |
| 3007 | <div class="line"><a name="l02909"></a><span class="lineno"> 2909</span>  }</div> |
| 3008 | <div class="line"><a name="l02910"></a><span class="lineno"> 2910</span>  <span class="comment">// 19: The cell state: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, num_units].</span></div> |
| 3009 | <div class="line"><a name="l02911"></a><span class="lineno"> 2911</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> cellStateIn = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 19, model, data);</div> |
| 3010 | <div class="line"><a name="l02912"></a><span class="lineno"> 2912</span>  <span class="keywordflow">if</span> (!cellStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3011 | <div class="line"><a name="l02913"></a><span class="lineno"> 2913</span>  {</div> |
| 3012 | <div class="line"><a name="l02914"></a><span class="lineno"> 2914</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 19: cellStateIn"</span>, __func__);</div> |
| 3013 | <div class="line"><a name="l02915"></a><span class="lineno"> 2915</span>  }</div> |
| 3014 | <div class="line"><a name="l02916"></a><span class="lineno"> 2916</span>  </div> |
| 3015 | <div class="line"><a name="l02917"></a><span class="lineno"> 2917</span>  <span class="comment">// Get the mandatory input tensors:</span></div> |
| 3016 | <div class="line"><a name="l02918"></a><span class="lineno"> 2918</span>  <span class="comment">// 02: The input-to-forget weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3017 | <div class="line"><a name="l02919"></a><span class="lineno"> 2919</span>  <span class="comment">// [num_units, input_size].</span></div> |
| 3018 | <div class="line"><a name="l02920"></a><span class="lineno"> 2920</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToForgetWeightsPin =</div> |
| 3019 | <div class="line"><a name="l02921"></a><span class="lineno"> 2921</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 2));</div> |
| 3020 | <div class="line"><a name="l02922"></a><span class="lineno"> 2922</span>  <span class="comment">// 03: The input-to-cell weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3021 | <div class="line"><a name="l02923"></a><span class="lineno"> 2923</span>  <span class="comment">// [num_units, input_size].</span></div> |
| 3022 | <div class="line"><a name="l02924"></a><span class="lineno"> 2924</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToCellWeightsPin =</div> |
| 3023 | <div class="line"><a name="l02925"></a><span class="lineno"> 2925</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 3));</div> |
| 3024 | <div class="line"><a name="l02926"></a><span class="lineno"> 2926</span>  <span class="comment">// 04: The input-to-output weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3025 | <div class="line"><a name="l02927"></a><span class="lineno"> 2927</span>  <span class="comment">// [num_units, input_size].</span></div> |
| 3026 | <div class="line"><a name="l02928"></a><span class="lineno"> 2928</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToOutputWeightsPin =</div> |
| 3027 | <div class="line"><a name="l02929"></a><span class="lineno"> 2929</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 4));</div> |
| 3028 | <div class="line"><a name="l02930"></a><span class="lineno"> 2930</span>  <span class="comment">// 06: The recurrent-to-forget weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3029 | <div class="line"><a name="l02931"></a><span class="lineno"> 2931</span>  <span class="comment">// [num_units, output_size].</span></div> |
| 3030 | <div class="line"><a name="l02932"></a><span class="lineno"> 2932</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToForgetWeightsPin =</div> |
| 3031 | <div class="line"><a name="l02933"></a><span class="lineno"> 2933</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 6));</div> |
| 3032 | <div class="line"><a name="l02934"></a><span class="lineno"> 2934</span>  <span class="comment">// 07: The recurrent-to-cell weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3033 | <div class="line"><a name="l02935"></a><span class="lineno"> 2935</span>  <span class="comment">// [num_units, output_size].</span></div> |
| 3034 | <div class="line"><a name="l02936"></a><span class="lineno"> 2936</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToCellWeightsPin =</div> |
| 3035 | <div class="line"><a name="l02937"></a><span class="lineno"> 2937</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 7));</div> |
| 3036 | <div class="line"><a name="l02938"></a><span class="lineno"> 2938</span>  <span class="comment">// 08: The recurrent-to-output weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3037 | <div class="line"><a name="l02939"></a><span class="lineno"> 2939</span>  <span class="comment">// [num_units, output_size].</span></div> |
| 3038 | <div class="line"><a name="l02940"></a><span class="lineno"> 2940</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToOutputWeightsPin =</div> |
| 3039 | <div class="line"><a name="l02941"></a><span class="lineno"> 2941</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 8));</div> |
| 3040 | <div class="line"><a name="l02942"></a><span class="lineno"> 2942</span>  <span class="comment">// 13: The forget gate bias: A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3041 | <div class="line"><a name="l02943"></a><span class="lineno"> 2943</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> forgetGateBiasPin =</div> |
| 3042 | <div class="line"><a name="l02944"></a><span class="lineno"> 2944</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 13, model, data);</div> |
| 3043 | <div class="line"><a name="l02945"></a><span class="lineno"> 2945</span>  <span class="comment">// 14: The cell bias: A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3044 | <div class="line"><a name="l02946"></a><span class="lineno"> 2946</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellBiasPin =</div> |
| 3045 | <div class="line"><a name="l02947"></a><span class="lineno"> 2947</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 14, model, data);</div> |
| 3046 | <div class="line"><a name="l02948"></a><span class="lineno"> 2948</span>  <span class="comment">// 15: The output gate bias: A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3047 | <div class="line"><a name="l02949"></a><span class="lineno"> 2949</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> outputGateBiasPin =</div> |
| 3048 | <div class="line"><a name="l02950"></a><span class="lineno"> 2950</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 15, model, data);</div> |
| 3049 | <div class="line"><a name="l02951"></a><span class="lineno"> 2951</span>  </div> |
| 3050 | <div class="line"><a name="l02952"></a><span class="lineno"> 2952</span>  <span class="keywordflow">if</span> (!inputToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3051 | <div class="line"><a name="l02953"></a><span class="lineno"> 2953</span>  !inputToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3052 | <div class="line"><a name="l02954"></a><span class="lineno"> 2954</span>  !inputToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3053 | <div class="line"><a name="l02955"></a><span class="lineno"> 2955</span>  !recurrentToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3054 | <div class="line"><a name="l02956"></a><span class="lineno"> 2956</span>  !recurrentToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3055 | <div class="line"><a name="l02957"></a><span class="lineno"> 2957</span>  !recurrentToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3056 | <div class="line"><a name="l02958"></a><span class="lineno"> 2958</span>  !forgetGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3057 | <div class="line"><a name="l02959"></a><span class="lineno"> 2959</span>  !cellBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3058 | <div class="line"><a name="l02960"></a><span class="lineno"> 2960</span>  !outputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3059 | <div class="line"><a name="l02961"></a><span class="lineno"> 2961</span>  {</div> |
| 3060 | <div class="line"><a name="l02962"></a><span class="lineno"> 2962</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid tensor inputs"</span>, __func__);</div> |
| 3061 | <div class="line"><a name="l02963"></a><span class="lineno"> 2963</span>  }</div> |
| 3062 | <div class="line"><a name="l02964"></a><span class="lineno"> 2964</span>  </div> |
| 3063 | <div class="line"><a name="l02965"></a><span class="lineno"> 2965</span>  <span class="comment">// Get the optional input tensors:</span></div> |
| 3064 | <div class="line"><a name="l02966"></a><span class="lineno"> 2966</span>  <span class="comment">// 01: The input-to-input weights: Optional. A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3065 | <div class="line"><a name="l02967"></a><span class="lineno"> 2967</span>  <span class="comment">// [num_units, input_size], where “num_units” corresponds to the number of cell units.</span></div> |
| 3066 | <div class="line"><a name="l02968"></a><span class="lineno"> 2968</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToInputWeightsPin =</div> |
| 3067 | <div class="line"><a name="l02969"></a><span class="lineno"> 2969</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 1, <span class="keyword">true</span>));</div> |
| 3068 | <div class="line"><a name="l02970"></a><span class="lineno"> 2970</span>  <span class="comment">// 05: The recurrent-to-input weights: Optional. A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3069 | <div class="line"><a name="l02971"></a><span class="lineno"> 2971</span>  <span class="comment">// [num_units, output_size], where “output_size” corresponds to either the number of cell units (i.e.,</span></div> |
| 3070 | <div class="line"><a name="l02972"></a><span class="lineno"> 2972</span>  <span class="comment">// “num_units”), or the second dimension of the “projection_weights”, if defined.</span></div> |
| 3071 | <div class="line"><a name="l02973"></a><span class="lineno"> 2973</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToInputWeightsPin =</div> |
| 3072 | <div class="line"><a name="l02974"></a><span class="lineno"> 2974</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 5, <span class="keyword">true</span>));</div> |
| 3073 | <div class="line"><a name="l02975"></a><span class="lineno"> 2975</span>  <span class="comment">// 09: The cell-to-input weights: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3074 | <div class="line"><a name="l02976"></a><span class="lineno"> 2976</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellToInputWeightsPin =</div> |
| 3075 | <div class="line"><a name="l02977"></a><span class="lineno"> 2977</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 9, <span class="keyword">true</span>));</div> |
| 3076 | <div class="line"><a name="l02978"></a><span class="lineno"> 2978</span>  <span class="comment">// 10: The cell-to-forget weights: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3077 | <div class="line"><a name="l02979"></a><span class="lineno"> 2979</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellToForgetWeightsPin =</div> |
| 3078 | <div class="line"><a name="l02980"></a><span class="lineno"> 2980</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 10, <span class="keyword">true</span>));</div> |
| 3079 | <div class="line"><a name="l02981"></a><span class="lineno"> 2981</span>  <span class="comment">// 11: The cell-to-output weights: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3080 | <div class="line"><a name="l02982"></a><span class="lineno"> 2982</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellToOutputWeightsPin =</div> |
| 3081 | <div class="line"><a name="l02983"></a><span class="lineno"> 2983</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 11, <span class="keyword">true</span>));</div> |
| 3082 | <div class="line"><a name="l02984"></a><span class="lineno"> 2984</span>  <span class="comment">// 12: The input gate bias: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [num_units].</span></div> |
| 3083 | <div class="line"><a name="l02985"></a><span class="lineno"> 2985</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputGateBiasPin =</div> |
| 3084 | <div class="line"><a name="l02986"></a><span class="lineno"> 2986</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3085 | <div class="line"><a name="l02987"></a><span class="lineno"> 2987</span>  12,</div> |
| 3086 | <div class="line"><a name="l02988"></a><span class="lineno"> 2988</span>  model,</div> |
| 3087 | <div class="line"><a name="l02989"></a><span class="lineno"> 2989</span>  data,</div> |
| 3088 | <div class="line"><a name="l02990"></a><span class="lineno"> 2990</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3089 | <div class="line"><a name="l02991"></a><span class="lineno"> 2991</span>  <span class="keyword">nullptr</span>,</div> |
| 3090 | <div class="line"><a name="l02992"></a><span class="lineno"> 2992</span>  <span class="keyword">true</span>);</div> |
| 3091 | <div class="line"><a name="l02993"></a><span class="lineno"> 2993</span>  </div> |
| 3092 | <div class="line"><a name="l02994"></a><span class="lineno"> 2994</span>  <span class="comment">// 16: The projection weights: Optional. A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape</span></div> |
| 3093 | <div class="line"><a name="l02995"></a><span class="lineno"> 2995</span>  <span class="comment">// [output_size, num_units].</span></div> |
| 3094 | <div class="line"><a name="l02996"></a><span class="lineno"> 2996</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> projectionWeightsPin =</div> |
| 3095 | <div class="line"><a name="l02997"></a><span class="lineno"> 2997</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 16, <span class="keyword">true</span>));</div> |
| 3096 | <div class="line"><a name="l02998"></a><span class="lineno"> 2998</span>  <span class="comment">// 17: The projection bias: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [output_size].</span></div> |
| 3097 | <div class="line"><a name="l02999"></a><span class="lineno"> 2999</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> projectionBiasPin =</div> |
| 3098 | <div class="line"><a name="l03000"></a><span class="lineno"> 3000</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3099 | <div class="line"><a name="l03001"></a><span class="lineno"> 3001</span>  17,</div> |
| 3100 | <div class="line"><a name="l03002"></a><span class="lineno"> 3002</span>  model,</div> |
| 3101 | <div class="line"><a name="l03003"></a><span class="lineno"> 3003</span>  data,</div> |
| 3102 | <div class="line"><a name="l03004"></a><span class="lineno"> 3004</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3103 | <div class="line"><a name="l03005"></a><span class="lineno"> 3005</span>  <span class="keyword">nullptr</span>,</div> |
| 3104 | <div class="line"><a name="l03006"></a><span class="lineno"> 3006</span>  <span class="keyword">true</span>);</div> |
| 3105 | <div class="line"><a name="l03007"></a><span class="lineno"> 3007</span>  </div> |
| 3106 | <div class="line"><a name="l03008"></a><span class="lineno"> 3008</span>  <span class="keywordflow">if</span> ((!inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3107 | <div class="line"><a name="l03009"></a><span class="lineno"> 3009</span>  (!recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3108 | <div class="line"><a name="l03010"></a><span class="lineno"> 3010</span>  (!cellToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3109 | <div class="line"><a name="l03011"></a><span class="lineno"> 3011</span>  (!cellToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3110 | <div class="line"><a name="l03012"></a><span class="lineno"> 3012</span>  (!cellToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3111 | <div class="line"><a name="l03013"></a><span class="lineno"> 3013</span>  (!inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3112 | <div class="line"><a name="l03014"></a><span class="lineno"> 3014</span>  (!projectionWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !projectionWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()) ||</div> |
| 3113 | <div class="line"><a name="l03015"></a><span class="lineno"> 3015</span>  (!projectionBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !projectionBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()))</div> |
| 3114 | <div class="line"><a name="l03016"></a><span class="lineno"> 3016</span>  {</div> |
| 3115 | <div class="line"><a name="l03017"></a><span class="lineno"> 3017</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid tensor inputs"</span>, __func__);</div> |
| 3116 | <div class="line"><a name="l03018"></a><span class="lineno"> 3018</span>  }</div> |
| 3117 | <div class="line"><a name="l03019"></a><span class="lineno"> 3019</span>  </div> |
| 3118 | <div class="line"><a name="l03020"></a><span class="lineno"> 3020</span>  <span class="comment">// Get the mandatory input scalars (actually 1-D tensors of size 1):</span></div> |
| 3119 | <div class="line"><a name="l03021"></a><span class="lineno"> 3021</span>  <span class="comment">// 20: The activation function: A value indicating the activation function:</span></div> |
| 3120 | <div class="line"><a name="l03022"></a><span class="lineno"> 3022</span>  <span class="comment">// 0: None; 1: Relu; 3: Relu6; 4: Tanh; 6: Sigmoid.</span></div> |
| 3121 | <div class="line"><a name="l03023"></a><span class="lineno"> 3023</span>  <span class="comment">// 21: The clipping threshold: for the cell state, such that values are bound within [-cell_clip, cell_clip].</span></div> |
| 3122 | <div class="line"><a name="l03024"></a><span class="lineno"> 3024</span>  <span class="comment">// If set to 0.0 then clipping is disabled.</span></div> |
| 3123 | <div class="line"><a name="l03025"></a><span class="lineno"> 3025</span>  <span class="comment">// 22: The clipping threshold: for the output from the projection layer, such that values are bound within</span></div> |
| 3124 | <div class="line"><a name="l03026"></a><span class="lineno"> 3026</span>  <span class="comment">// [-proj_clip, proj_clip]. If set to 0.0 then clipping is disabled.</span></div> |
| 3125 | <div class="line"><a name="l03027"></a><span class="lineno"> 3027</span>  ActivationFn activation = ActivationFn::kActivationNone;</div> |
| 3126 | <div class="line"><a name="l03028"></a><span class="lineno"> 3028</span>  <span class="keywordtype">float</span> cellClip;</div> |
| 3127 | <div class="line"><a name="l03029"></a><span class="lineno"> 3029</span>  <span class="keywordtype">float</span> projClip;</div> |
| 3128 | <div class="line"><a name="l03030"></a><span class="lineno"> 3030</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a4c6b55ca5080b58c2cc6d7c6c7347d96">GetInputActivationFunctionFromTensor</a>(operation, 20, activation, model, data) ||</div> |
| 3129 | <div class="line"><a name="l03031"></a><span class="lineno"> 3031</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 21, OperandType::FLOAT32, cellClip, model, data) ||</div> |
| 3130 | <div class="line"><a name="l03032"></a><span class="lineno"> 3032</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 22, OperandType::FLOAT32, projClip, model, data))</div> |
| 3131 | <div class="line"><a name="l03033"></a><span class="lineno"> 3033</span>  {</div> |
| 3132 | <div class="line"><a name="l03034"></a><span class="lineno"> 3034</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid scalar inputs"</span>, __func__);</div> |
| 3133 | <div class="line"><a name="l03035"></a><span class="lineno"> 3035</span>  }</div> |
| 3134 | <div class="line"><a name="l03036"></a><span class="lineno"> 3036</span>  </div> |
| 3135 | <div class="line"><a name="l03037"></a><span class="lineno"> 3037</span>  <span class="comment">// Get the normalization tensors</span></div> |
| 3136 | <div class="line"><a name="l03038"></a><span class="lineno"> 3038</span>  <span class="comment">// 23: The input layer normalization weights. A 1-D tensor of shape [num_units].</span></div> |
| 3137 | <div class="line"><a name="l03039"></a><span class="lineno"> 3039</span>  <span class="comment">// Used to rescale normalized inputs to activation at input gate.</span></div> |
| 3138 | <div class="line"><a name="l03040"></a><span class="lineno"> 3040</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputLayerNormWeightsPin</div> |
| 3139 | <div class="line"><a name="l03041"></a><span class="lineno"> 3041</span>  (<a class="code" href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">DequantizeAndMakeConstTensorPin</a>(operation, model, data, 23, <span class="keyword">true</span>));</div> |
| 3140 | <div class="line"><a name="l03042"></a><span class="lineno"> 3042</span>  </div> |
| 3141 | <div class="line"><a name="l03043"></a><span class="lineno"> 3043</span>  <span class="comment">// 24: The forget layer normalization weights. A 1-D tensor of shape [num_units].</span></div> |
| 3142 | <div class="line"><a name="l03044"></a><span class="lineno"> 3044</span>  <span class="comment">// Used to rescale normalized inputs to activation at forget gate.</span></div> |
| 3143 | <div class="line"><a name="l03045"></a><span class="lineno"> 3045</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> forgetLayerNormWeightsPin =</div> |
| 3144 | <div class="line"><a name="l03046"></a><span class="lineno"> 3046</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3145 | <div class="line"><a name="l03047"></a><span class="lineno"> 3047</span>  24,</div> |
| 3146 | <div class="line"><a name="l03048"></a><span class="lineno"> 3048</span>  model,</div> |
| 3147 | <div class="line"><a name="l03049"></a><span class="lineno"> 3049</span>  data,</div> |
| 3148 | <div class="line"><a name="l03050"></a><span class="lineno"> 3050</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3149 | <div class="line"><a name="l03051"></a><span class="lineno"> 3051</span>  <span class="keyword">nullptr</span>,</div> |
| 3150 | <div class="line"><a name="l03052"></a><span class="lineno"> 3052</span>  <span class="keyword">true</span>);</div> |
| 3151 | <div class="line"><a name="l03053"></a><span class="lineno"> 3053</span>  </div> |
| 3152 | <div class="line"><a name="l03054"></a><span class="lineno"> 3054</span>  <span class="comment">// 25: The cell layer normalization weights. A 1-D tensor of shape [num_units].</span></div> |
| 3153 | <div class="line"><a name="l03055"></a><span class="lineno"> 3055</span>  <span class="comment">// Used to rescale normalized inputs to activation at cell gate.</span></div> |
| 3154 | <div class="line"><a name="l03056"></a><span class="lineno"> 3056</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellLayerNormWeightsPin =</div> |
| 3155 | <div class="line"><a name="l03057"></a><span class="lineno"> 3057</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3156 | <div class="line"><a name="l03058"></a><span class="lineno"> 3058</span>  25,</div> |
| 3157 | <div class="line"><a name="l03059"></a><span class="lineno"> 3059</span>  model,</div> |
| 3158 | <div class="line"><a name="l03060"></a><span class="lineno"> 3060</span>  data,</div> |
| 3159 | <div class="line"><a name="l03061"></a><span class="lineno"> 3061</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3160 | <div class="line"><a name="l03062"></a><span class="lineno"> 3062</span>  <span class="keyword">nullptr</span>,</div> |
| 3161 | <div class="line"><a name="l03063"></a><span class="lineno"> 3063</span>  <span class="keyword">true</span>);</div> |
| 3162 | <div class="line"><a name="l03064"></a><span class="lineno"> 3064</span>  </div> |
| 3163 | <div class="line"><a name="l03065"></a><span class="lineno"> 3065</span>  <span class="comment">// 26: The output layer normalization weights. A 1-D tensor of shape [num_units].</span></div> |
| 3164 | <div class="line"><a name="l03066"></a><span class="lineno"> 3066</span>  <span class="comment">// Used to rescale normalized inputs to activation at output gate.</span></div> |
| 3165 | <div class="line"><a name="l03067"></a><span class="lineno"> 3067</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> outputLayerNormWeightsPin =</div> |
| 3166 | <div class="line"><a name="l03068"></a><span class="lineno"> 3068</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3167 | <div class="line"><a name="l03069"></a><span class="lineno"> 3069</span>  26,</div> |
| 3168 | <div class="line"><a name="l03070"></a><span class="lineno"> 3070</span>  model,</div> |
| 3169 | <div class="line"><a name="l03071"></a><span class="lineno"> 3071</span>  data,</div> |
| 3170 | <div class="line"><a name="l03072"></a><span class="lineno"> 3072</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3171 | <div class="line"><a name="l03073"></a><span class="lineno"> 3073</span>  <span class="keyword">nullptr</span>,</div> |
| 3172 | <div class="line"><a name="l03074"></a><span class="lineno"> 3074</span>  <span class="keyword">true</span>);</div> |
| 3173 | <div class="line"><a name="l03075"></a><span class="lineno"> 3075</span>  </div> |
| 3174 | <div class="line"><a name="l03076"></a><span class="lineno"> 3076</span>  <span class="comment">// Outputs:</span></div> |
| 3175 | <div class="line"><a name="l03077"></a><span class="lineno"> 3077</span>  <span class="comment">// 00: The scratch buffer: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, num_units * 4]</span></div> |
| 3176 | <div class="line"><a name="l03078"></a><span class="lineno"> 3078</span>  <span class="comment">// with CIFG, or [batch_size, num_units * 3] without CIFG.</span></div> |
| 3177 | <div class="line"><a name="l03079"></a><span class="lineno"> 3079</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* scratchBuffer = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 3178 | <div class="line"><a name="l03080"></a><span class="lineno"> 3080</span>  <span class="keywordflow">if</span> (!scratchBuffer)</div> |
| 3179 | <div class="line"><a name="l03081"></a><span class="lineno"> 3081</span>  {</div> |
| 3180 | <div class="line"><a name="l03082"></a><span class="lineno"> 3082</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0: scratchBuffer"</span>, __func__);</div> |
| 3181 | <div class="line"><a name="l03083"></a><span class="lineno"> 3083</span>  }</div> |
| 3182 | <div class="line"><a name="l03084"></a><span class="lineno"> 3084</span>  <span class="comment">// 01: The output state (out): A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, output_size].</span></div> |
| 3183 | <div class="line"><a name="l03085"></a><span class="lineno"> 3085</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputStateOut = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 1, model);</div> |
| 3184 | <div class="line"><a name="l03086"></a><span class="lineno"> 3086</span>  <span class="keywordflow">if</span> (!outputStateOut)</div> |
| 3185 | <div class="line"><a name="l03087"></a><span class="lineno"> 3087</span>  {</div> |
| 3186 | <div class="line"><a name="l03088"></a><span class="lineno"> 3088</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 1: outputStateOut"</span>, __func__);</div> |
| 3187 | <div class="line"><a name="l03089"></a><span class="lineno"> 3089</span>  }</div> |
| 3188 | <div class="line"><a name="l03090"></a><span class="lineno"> 3090</span>  <span class="comment">// 02: The cell state (out): A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, num_units].</span></div> |
| 3189 | <div class="line"><a name="l03091"></a><span class="lineno"> 3091</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* cellStateOut = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 2, model);</div> |
| 3190 | <div class="line"><a name="l03092"></a><span class="lineno"> 3092</span>  <span class="keywordflow">if</span> (!cellStateOut)</div> |
| 3191 | <div class="line"><a name="l03093"></a><span class="lineno"> 3093</span>  {</div> |
| 3192 | <div class="line"><a name="l03094"></a><span class="lineno"> 3094</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 2: cellStateOut"</span>, __func__);</div> |
| 3193 | <div class="line"><a name="l03095"></a><span class="lineno"> 3095</span>  }</div> |
| 3194 | <div class="line"><a name="l03096"></a><span class="lineno"> 3096</span>  <span class="comment">// 03: The output: A 2-D tensor of ANEURALNETWORKS_TENSOR_FLOAT32, of shape [batch_size, output_size]. This is</span></div> |
| 3195 | <div class="line"><a name="l03097"></a><span class="lineno"> 3097</span>  <span class="comment">// effectively the same as the current “output state (out)” value.</span></div> |
| 3196 | <div class="line"><a name="l03098"></a><span class="lineno"> 3098</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 3, model);</div> |
| 3197 | <div class="line"><a name="l03099"></a><span class="lineno"> 3099</span>  <span class="keywordflow">if</span> (!output)</div> |
| 3198 | <div class="line"><a name="l03100"></a><span class="lineno"> 3100</span>  {</div> |
| 3199 | <div class="line"><a name="l03101"></a><span class="lineno"> 3101</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 3: output"</span>, __func__);</div> |
| 3200 | <div class="line"><a name="l03102"></a><span class="lineno"> 3102</span>  }</div> |
| 3201 | <div class="line"><a name="l03103"></a><span class="lineno"> 3103</span>  </div> |
| 3202 | <div class="line"><a name="l03104"></a><span class="lineno"> 3104</span>  <span class="comment">// set the params structure for the AddLstmLayer call</span></div> |
| 3203 | <div class="line"><a name="l03105"></a><span class="lineno"> 3105</span>  <a class="code" href="structarmnn_1_1_lstm_input_params.html">LstmInputParams</a> params;</div> |
| 3204 | <div class="line"><a name="l03106"></a><span class="lineno"> 3106</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> = inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3205 | <div class="line"><a name="l03107"></a><span class="lineno"> 3107</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">m_InputToForgetWeights</a> = inputToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3206 | <div class="line"><a name="l03108"></a><span class="lineno"> 3108</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">m_InputToCellWeights</a> = inputToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3207 | <div class="line"><a name="l03109"></a><span class="lineno"> 3109</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">m_InputToOutputWeights</a> = inputToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3208 | <div class="line"><a name="l03110"></a><span class="lineno"> 3110</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> = recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3209 | <div class="line"><a name="l03111"></a><span class="lineno"> 3111</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">m_RecurrentToForgetWeights</a> = recurrentToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3210 | <div class="line"><a name="l03112"></a><span class="lineno"> 3112</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">m_RecurrentToCellWeights</a> = recurrentToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3211 | <div class="line"><a name="l03113"></a><span class="lineno"> 3113</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">m_RecurrentToOutputWeights</a> = recurrentToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3212 | <div class="line"><a name="l03114"></a><span class="lineno"> 3114</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a> = cellToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3213 | <div class="line"><a name="l03115"></a><span class="lineno"> 3115</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a> = cellToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3214 | <div class="line"><a name="l03116"></a><span class="lineno"> 3116</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a> = cellToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3215 | <div class="line"><a name="l03117"></a><span class="lineno"> 3117</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> = inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3216 | <div class="line"><a name="l03118"></a><span class="lineno"> 3118</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">m_ForgetGateBias</a> = forgetGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3217 | <div class="line"><a name="l03119"></a><span class="lineno"> 3119</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">m_CellBias</a> = cellBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3218 | <div class="line"><a name="l03120"></a><span class="lineno"> 3120</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">m_OutputGateBias</a> = outputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3219 | <div class="line"><a name="l03121"></a><span class="lineno"> 3121</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">m_ProjectionWeights</a> = projectionWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3220 | <div class="line"><a name="l03122"></a><span class="lineno"> 3122</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a> = projectionBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3221 | <div class="line"><a name="l03123"></a><span class="lineno"> 3123</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a> = inputLayerNormWeightsPin.GetConstTensorPtr();</div> |
| 3222 | <div class="line"><a name="l03124"></a><span class="lineno"> 3124</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a> = forgetLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3223 | <div class="line"><a name="l03125"></a><span class="lineno"> 3125</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a> = cellLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3224 | <div class="line"><a name="l03126"></a><span class="lineno"> 3126</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a> = outputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 3225 | <div class="line"><a name="l03127"></a><span class="lineno"> 3127</span>  </div> |
| 3226 | <div class="line"><a name="l03128"></a><span class="lineno"> 3128</span>  <span class="comment">// set the layer descriptor</span></div> |
| 3227 | <div class="line"><a name="l03129"></a><span class="lineno"> 3129</span>  <a class="code" href="structarmnn_1_1_lstm_descriptor.html">LstmDescriptor</a> desc;</div> |
| 3228 | <div class="line"><a name="l03130"></a><span class="lineno"> 3130</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ae1b07ed928036004bd257169e5aeeef4">m_ActivationFunc</a> = activation;</div> |
| 3229 | <div class="line"><a name="l03131"></a><span class="lineno"> 3131</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a531a3907ec13d3772370da88030191a5">m_ClippingThresCell</a> = cellClip;</div> |
| 3230 | <div class="line"><a name="l03132"></a><span class="lineno"> 3132</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a86e88bef0df4df96df752b4b8955a3af">m_ClippingThresProj</a> = projClip;</div> |
| 3231 | <div class="line"><a name="l03133"></a><span class="lineno"> 3133</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3232 | <div class="line"><a name="l03134"></a><span class="lineno"> 3134</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3233 | <div class="line"><a name="l03135"></a><span class="lineno"> 3135</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> == <span class="keyword">nullptr</span>);</div> |
| 3234 | <div class="line"><a name="l03136"></a><span class="lineno"> 3136</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 3235 | <div class="line"><a name="l03137"></a><span class="lineno"> 3137</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a> != <span class="keyword">nullptr</span>);</div> |
| 3236 | <div class="line"><a name="l03138"></a><span class="lineno"> 3138</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">m_ProjectionEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">m_ProjectionWeights</a> != <span class="keyword">nullptr</span>);</div> |
| 3237 | <div class="line"><a name="l03139"></a><span class="lineno"> 3139</span>  desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">m_LayerNormEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 3238 | <div class="line"><a name="l03140"></a><span class="lineno"> 3140</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 3239 | <div class="line"><a name="l03141"></a><span class="lineno"> 3141</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 3240 | <div class="line"><a name="l03142"></a><span class="lineno"> 3142</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a> != <span class="keyword">nullptr</span>);</div> |
| 3241 | <div class="line"><a name="l03143"></a><span class="lineno"> 3143</span>  </div> |
| 3242 | <div class="line"><a name="l03144"></a><span class="lineno"> 3144</span>  <span class="comment">// validate the optional input groups</span></div> |
| 3243 | <div class="line"><a name="l03145"></a><span class="lineno"> 3145</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> &&</div> |
| 3244 | <div class="line"><a name="l03146"></a><span class="lineno"> 3146</span>  (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 3245 | <div class="line"><a name="l03147"></a><span class="lineno"> 3147</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 3246 | <div class="line"><a name="l03148"></a><span class="lineno"> 3148</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> != <span class="keyword">nullptr</span>))</div> |
| 3247 | <div class="line"><a name="l03149"></a><span class="lineno"> 3149</span>  {</div> |
| 3248 | <div class="line"><a name="l03150"></a><span class="lineno"> 3150</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: All, or none, of input-to-input weights, recurrent-to-input weights,"</span></div> |
| 3249 | <div class="line"><a name="l03151"></a><span class="lineno"> 3151</span>  <span class="stringliteral">" and input gate bias must be provided"</span>, __func__);</div> |
| 3250 | <div class="line"><a name="l03152"></a><span class="lineno"> 3152</span>  }</div> |
| 3251 | <div class="line"><a name="l03153"></a><span class="lineno"> 3153</span>  </div> |
| 3252 | <div class="line"><a name="l03154"></a><span class="lineno"> 3154</span>  <span class="keywordflow">if</span> (!desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">m_ProjectionEnabled</a> && params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a> != <span class="keyword">nullptr</span>)</div> |
| 3253 | <div class="line"><a name="l03155"></a><span class="lineno"> 3155</span>  {</div> |
| 3254 | <div class="line"><a name="l03156"></a><span class="lineno"> 3156</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: projection bias should not be provided without projection weights"</span>, __func__);</div> |
| 3255 | <div class="line"><a name="l03157"></a><span class="lineno"> 3157</span>  }</div> |
| 3256 | <div class="line"><a name="l03158"></a><span class="lineno"> 3158</span>  </div> |
| 3257 | <div class="line"><a name="l03159"></a><span class="lineno"> 3159</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a> &&</div> |
| 3258 | <div class="line"><a name="l03160"></a><span class="lineno"> 3160</span>  (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3259 | <div class="line"><a name="l03161"></a><span class="lineno"> 3161</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3260 | <div class="line"><a name="l03162"></a><span class="lineno"> 3162</span>  (!desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> && params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a> == <span class="keyword">nullptr</span>)))</div> |
| 3261 | <div class="line"><a name="l03163"></a><span class="lineno"> 3163</span>  {</div> |
| 3262 | <div class="line"><a name="l03164"></a><span class="lineno"> 3164</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: All, or none, of cell-to-forget weights and cell-to-output weights must be provided"</span></div> |
| 3263 | <div class="line"><a name="l03165"></a><span class="lineno"> 3165</span>  <span class="stringliteral">" and, if CIFG is not enabled, cell-to-input weights must also be provided"</span>, __func__);</div> |
| 3264 | <div class="line"><a name="l03166"></a><span class="lineno"> 3166</span>  }</div> |
| 3265 | <div class="line"><a name="l03167"></a><span class="lineno"> 3167</span>  </div> |
| 3266 | <div class="line"><a name="l03168"></a><span class="lineno"> 3168</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">m_LayerNormEnabled</a> &&</div> |
| 3267 | <div class="line"><a name="l03169"></a><span class="lineno"> 3169</span>  (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3268 | <div class="line"><a name="l03170"></a><span class="lineno"> 3170</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3269 | <div class="line"><a name="l03171"></a><span class="lineno"> 3171</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 3270 | <div class="line"><a name="l03172"></a><span class="lineno"> 3172</span>  (!desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> && params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a> == <span class="keyword">nullptr</span>)))</div> |
| 3271 | <div class="line"><a name="l03173"></a><span class="lineno"> 3173</span>  {</div> |
| 3272 | <div class="line"><a name="l03174"></a><span class="lineno"> 3174</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: All, or none, of forget-norm weights, cell-norm weights and output-norm weights must be"</span></div> |
| 3273 | <div class="line"><a name="l03175"></a><span class="lineno"> 3175</span>  <span class="stringliteral">" provided and, if CIFG is not enabled, input-norm weights must also be provided"</span>, __func__);</div> |
| 3274 | <div class="line"><a name="l03176"></a><span class="lineno"> 3176</span>  }</div> |
| 3275 | <div class="line"><a name="l03177"></a><span class="lineno"> 3177</span>  </div> |
| 3276 | <div class="line"><a name="l03178"></a><span class="lineno"> 3178</span>  <span class="comment">// Check if the layer is supported</span></div> |
| 3277 | <div class="line"><a name="l03179"></a><span class="lineno"> 3179</span>  <span class="comment">// Inputs</span></div> |
| 3278 | <div class="line"><a name="l03180"></a><span class="lineno"> 3180</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3279 | <div class="line"><a name="l03181"></a><span class="lineno"> 3181</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputStateInInfo = outputStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3280 | <div class="line"><a name="l03182"></a><span class="lineno"> 3182</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& cellStateInInfo = cellStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3281 | <div class="line"><a name="l03183"></a><span class="lineno"> 3183</span>  </div> |
| 3282 | <div class="line"><a name="l03184"></a><span class="lineno"> 3184</span>  <span class="comment">// Outputs</span></div> |
| 3283 | <div class="line"><a name="l03185"></a><span class="lineno"> 3185</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& scratchBufferInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*scratchBuffer);</div> |
| 3284 | <div class="line"><a name="l03186"></a><span class="lineno"> 3186</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputStateOutInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputStateOut);</div> |
| 3285 | <div class="line"><a name="l03187"></a><span class="lineno"> 3187</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& cellStateOutInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*cellStateOut);</div> |
| 3286 | <div class="line"><a name="l03188"></a><span class="lineno"> 3188</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 3287 | <div class="line"><a name="l03189"></a><span class="lineno"> 3189</span>  </div> |
| 3288 | <div class="line"><a name="l03190"></a><span class="lineno"> 3190</span>  <span class="comment">// Basic parameters</span></div> |
| 3289 | <div class="line"><a name="l03191"></a><span class="lineno"> 3191</span>  <a class="code" href="structarmnn_1_1_lstm_input_params_info.html">LstmInputParamsInfo</a> paramsInfo;</div> |
| 3290 | <div class="line"><a name="l03192"></a><span class="lineno"> 3192</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a4b3f6e6f2268416ffd7a34fda95ffd0b">m_InputToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">m_InputToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3291 | <div class="line"><a name="l03193"></a><span class="lineno"> 3193</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#af07c65d3b7886a00f10c69093e76a341">m_InputToCellWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">m_InputToCellWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3292 | <div class="line"><a name="l03194"></a><span class="lineno"> 3194</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a33242cb43250f04f1736161ca8ad2db9">m_InputToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">m_InputToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3293 | <div class="line"><a name="l03195"></a><span class="lineno"> 3195</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a11b90bd2694aa7f4b434aadd75fbb1c2">m_RecurrentToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">m_RecurrentToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3294 | <div class="line"><a name="l03196"></a><span class="lineno"> 3196</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a6c7873a3320a59a63aa26c006db905c0">m_RecurrentToCellWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">m_RecurrentToCellWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3295 | <div class="line"><a name="l03197"></a><span class="lineno"> 3197</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#af434b509cd5232ef762c4b21b2dfaae0">m_RecurrentToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">m_RecurrentToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3296 | <div class="line"><a name="l03198"></a><span class="lineno"> 3198</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a21626d0b91f8942c3ef8e48e17db9f21">m_ForgetGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">m_ForgetGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3297 | <div class="line"><a name="l03199"></a><span class="lineno"> 3199</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ab173a067eeb7295d84f5327bcc05a6c1">m_CellBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">m_CellBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3298 | <div class="line"><a name="l03200"></a><span class="lineno"> 3200</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a48cb83dcb58c786ea36f5d37695e75b1">m_OutputGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">m_OutputGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3299 | <div class="line"><a name="l03201"></a><span class="lineno"> 3201</span>  </div> |
| 3300 | <div class="line"><a name="l03202"></a><span class="lineno"> 3202</span>  <span class="comment">// Optional parameters</span></div> |
| 3301 | <div class="line"><a name="l03203"></a><span class="lineno"> 3203</span>  <span class="keywordflow">if</span> (!desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a>)</div> |
| 3302 | <div class="line"><a name="l03204"></a><span class="lineno"> 3204</span>  {</div> |
| 3303 | <div class="line"><a name="l03205"></a><span class="lineno"> 3205</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a15f9f65126ad3d4d82d6c19d0662ae01">m_InputToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3304 | <div class="line"><a name="l03206"></a><span class="lineno"> 3206</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a9ab864961ded72f5ce7ea4eb9921f826">m_RecurrentToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3305 | <div class="line"><a name="l03207"></a><span class="lineno"> 3207</span>  <span class="keywordflow">if</span> (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a> != <span class="keyword">nullptr</span>)</div> |
| 3306 | <div class="line"><a name="l03208"></a><span class="lineno"> 3208</span>  {</div> |
| 3307 | <div class="line"><a name="l03209"></a><span class="lineno"> 3209</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ab99b2297c5a17114926b8b47c693fe43">m_CellToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3308 | <div class="line"><a name="l03210"></a><span class="lineno"> 3210</span>  }</div> |
| 3309 | <div class="line"><a name="l03211"></a><span class="lineno"> 3211</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a042ea9d6d3842b87f112db126e806a93">m_InputGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3310 | <div class="line"><a name="l03212"></a><span class="lineno"> 3212</span>  }</div> |
| 3311 | <div class="line"><a name="l03213"></a><span class="lineno"> 3213</span>  </div> |
| 3312 | <div class="line"><a name="l03214"></a><span class="lineno"> 3214</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">m_ProjectionEnabled</a>)</div> |
| 3313 | <div class="line"><a name="l03215"></a><span class="lineno"> 3215</span>  {</div> |
| 3314 | <div class="line"><a name="l03216"></a><span class="lineno"> 3216</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ac6fb4a3a8e7e8a359ac34badc8178b1a">m_ProjectionWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">m_ProjectionWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3315 | <div class="line"><a name="l03217"></a><span class="lineno"> 3217</span>  <span class="keywordflow">if</span> (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a> != <span class="keyword">nullptr</span>)</div> |
| 3316 | <div class="line"><a name="l03218"></a><span class="lineno"> 3218</span>  {</div> |
| 3317 | <div class="line"><a name="l03219"></a><span class="lineno"> 3219</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ae22fc962c59e7c24986718f5af0020db">m_ProjectionBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3318 | <div class="line"><a name="l03220"></a><span class="lineno"> 3220</span>  }</div> |
| 3319 | <div class="line"><a name="l03221"></a><span class="lineno"> 3221</span>  }</div> |
| 3320 | <div class="line"><a name="l03222"></a><span class="lineno"> 3222</span>  </div> |
| 3321 | <div class="line"><a name="l03223"></a><span class="lineno"> 3223</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a>)</div> |
| 3322 | <div class="line"><a name="l03224"></a><span class="lineno"> 3224</span>  {</div> |
| 3323 | <div class="line"><a name="l03225"></a><span class="lineno"> 3225</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a1ddc4634491d95d37fe15049decadb8c">m_CellToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3324 | <div class="line"><a name="l03226"></a><span class="lineno"> 3226</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a16b4d4acd2a0d6d533ff3fe8b8e71765">m_CellToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3325 | <div class="line"><a name="l03227"></a><span class="lineno"> 3227</span>  }</div> |
| 3326 | <div class="line"><a name="l03228"></a><span class="lineno"> 3228</span>  </div> |
| 3327 | <div class="line"><a name="l03229"></a><span class="lineno"> 3229</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">m_LayerNormEnabled</a>)</div> |
| 3328 | <div class="line"><a name="l03230"></a><span class="lineno"> 3230</span>  {</div> |
| 3329 | <div class="line"><a name="l03231"></a><span class="lineno"> 3231</span>  <span class="keywordflow">if</span>(!desc.<a class="code" href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a>)</div> |
| 3330 | <div class="line"><a name="l03232"></a><span class="lineno"> 3232</span>  {</div> |
| 3331 | <div class="line"><a name="l03233"></a><span class="lineno"> 3233</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ab8c7712c0c6bc427dede8c76945e8f89">m_InputLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3332 | <div class="line"><a name="l03234"></a><span class="lineno"> 3234</span>  }</div> |
| 3333 | <div class="line"><a name="l03235"></a><span class="lineno"> 3235</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a513ad7774e0f8969ab68d91a04ecca65">m_ForgetLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3334 | <div class="line"><a name="l03236"></a><span class="lineno"> 3236</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a567622530e51eaa2e82c9790a3a43f02">m_CellLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3335 | <div class="line"><a name="l03237"></a><span class="lineno"> 3237</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a9935ad8aff43cf1e0622bf6d82710513">m_OutputLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 3336 | <div class="line"><a name="l03238"></a><span class="lineno"> 3238</span>  }</div> |
| 3337 | <div class="line"><a name="l03239"></a><span class="lineno"> 3239</span>  </div> |
| 3338 | <div class="line"><a name="l03240"></a><span class="lineno"> 3240</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 3339 | <div class="line"><a name="l03241"></a><span class="lineno"> 3241</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 3340 | <div class="line"><a name="l03242"></a><span class="lineno"> 3242</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 3341 | <div class="line"><a name="l03243"></a><span class="lineno"> 3243</span>  {</div> |
| 3342 | <div class="line"><a name="l03244"></a><span class="lineno"> 3244</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 3343 | <div class="line"><a name="l03245"></a><span class="lineno"> 3245</span>  IsLstmSupported,</div> |
| 3344 | <div class="line"><a name="l03246"></a><span class="lineno"> 3246</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 3345 | <div class="line"><a name="l03247"></a><span class="lineno"> 3247</span>  isSupported,</div> |
| 3346 | <div class="line"><a name="l03248"></a><span class="lineno"> 3248</span>  setBackend,</div> |
| 3347 | <div class="line"><a name="l03249"></a><span class="lineno"> 3249</span>  inputInfo,</div> |
| 3348 | <div class="line"><a name="l03250"></a><span class="lineno"> 3250</span>  outputStateInInfo,</div> |
| 3349 | <div class="line"><a name="l03251"></a><span class="lineno"> 3251</span>  cellStateInInfo,</div> |
| 3350 | <div class="line"><a name="l03252"></a><span class="lineno"> 3252</span>  scratchBufferInfo,</div> |
| 3351 | <div class="line"><a name="l03253"></a><span class="lineno"> 3253</span>  outputStateOutInfo,</div> |
| 3352 | <div class="line"><a name="l03254"></a><span class="lineno"> 3254</span>  cellStateOutInfo,</div> |
| 3353 | <div class="line"><a name="l03255"></a><span class="lineno"> 3255</span>  outputInfo,</div> |
| 3354 | <div class="line"><a name="l03256"></a><span class="lineno"> 3256</span>  desc,</div> |
| 3355 | <div class="line"><a name="l03257"></a><span class="lineno"> 3257</span>  paramsInfo);</div> |
| 3356 | <div class="line"><a name="l03258"></a><span class="lineno"> 3258</span>  };</div> |
| 3357 | <div class="line"><a name="l03259"></a><span class="lineno"> 3259</span>  </div> |
| 3358 | <div class="line"><a name="l03260"></a><span class="lineno"> 3260</span>  <span class="keywordtype">bool</span> isDynamic = <span class="keyword">false</span>;</div> |
| 3359 | <div class="line"><a name="l03261"></a><span class="lineno"> 3261</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputStateOutInfo) &&</div> |
| 3360 | <div class="line"><a name="l03262"></a><span class="lineno"> 3262</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(scratchBufferInfo) &&</div> |
| 3361 | <div class="line"><a name="l03263"></a><span class="lineno"> 3263</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(cellStateOutInfo) &&</div> |
| 3362 | <div class="line"><a name="l03264"></a><span class="lineno"> 3264</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 3363 | <div class="line"><a name="l03265"></a><span class="lineno"> 3265</span>  {</div> |
| 3364 | <div class="line"><a name="l03266"></a><span class="lineno"> 3266</span>  validateFunc(outputInfo, isSupported);</div> |
| 3365 | <div class="line"><a name="l03267"></a><span class="lineno"> 3267</span>  }</div> |
| 3366 | <div class="line"><a name="l03268"></a><span class="lineno"> 3268</span>  <span class="keywordflow">else</span></div> |
| 3367 | <div class="line"><a name="l03269"></a><span class="lineno"> 3269</span>  {</div> |
| 3368 | <div class="line"><a name="l03270"></a><span class="lineno"> 3270</span>  isDynamic = <span class="keyword">true</span>;</div> |
| 3369 | <div class="line"><a name="l03271"></a><span class="lineno"> 3271</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 3370 | <div class="line"><a name="l03272"></a><span class="lineno"> 3272</span>  }</div> |
| 3371 | <div class="line"><a name="l03273"></a><span class="lineno"> 3273</span>  </div> |
| 3372 | <div class="line"><a name="l03274"></a><span class="lineno"> 3274</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 3373 | <div class="line"><a name="l03275"></a><span class="lineno"> 3275</span>  {</div> |
| 3374 | <div class="line"><a name="l03276"></a><span class="lineno"> 3276</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3375 | <div class="line"><a name="l03277"></a><span class="lineno"> 3277</span>  }</div> |
| 3376 | <div class="line"><a name="l03278"></a><span class="lineno"> 3278</span>  </div> |
| 3377 | <div class="line"><a name="l03279"></a><span class="lineno"> 3279</span>  <span class="comment">// Add the layer</span></div> |
| 3378 | <div class="line"><a name="l03280"></a><span class="lineno"> 3280</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddLstmLayer(desc, params, <span class="stringliteral">"Lstm"</span>);</div> |
| 3379 | <div class="line"><a name="l03281"></a><span class="lineno"> 3281</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 3380 | <div class="line"><a name="l03282"></a><span class="lineno"> 3282</span>  </div> |
| 3381 | <div class="line"><a name="l03283"></a><span class="lineno"> 3283</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 3382 | <div class="line"><a name="l03284"></a><span class="lineno"> 3284</span>  outputStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 3383 | <div class="line"><a name="l03285"></a><span class="lineno"> 3285</span>  cellStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 3384 | <div class="line"><a name="l03286"></a><span class="lineno"> 3286</span>  </div> |
| 3385 | <div class="line"><a name="l03287"></a><span class="lineno"> 3287</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| 3386 | <div class="line"><a name="l03288"></a><span class="lineno"> 3288</span>  {</div> |
| 3387 | <div class="line"><a name="l03289"></a><span class="lineno"> 3289</span>  <span class="keywordflow">return</span> (</div> |
| 3388 | <div class="line"><a name="l03290"></a><span class="lineno"> 3290</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, 0, model, data) &&</div> |
| 3389 | <div class="line"><a name="l03291"></a><span class="lineno"> 3291</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 1, *layer, 1, model, data) &&</div> |
| 3390 | <div class="line"><a name="l03292"></a><span class="lineno"> 3292</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 2, *layer, 2, model, data) &&</div> |
| 3391 | <div class="line"><a name="l03293"></a><span class="lineno"> 3293</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 3, *layer, 3, model, data));</div> |
| 3392 | <div class="line"><a name="l03294"></a><span class="lineno"> 3294</span>  }</div> |
| 3393 | <div class="line"><a name="l03295"></a><span class="lineno"> 3295</span>  <span class="keywordflow">else</span></div> |
| 3394 | <div class="line"><a name="l03296"></a><span class="lineno"> 3296</span>  {</div> |
| 3395 | <div class="line"><a name="l03297"></a><span class="lineno"> 3297</span>  <span class="keywordflow">return</span> (</div> |
| 3396 | <div class="line"><a name="l03298"></a><span class="lineno"> 3298</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, 0, model, data) &&</div> |
| 3397 | <div class="line"><a name="l03299"></a><span class="lineno"> 3299</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 1, *layer, 1, model, data) &&</div> |
| 3398 | <div class="line"><a name="l03300"></a><span class="lineno"> 3300</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 2, *layer, 2, model, data) &&</div> |
| 3399 | <div class="line"><a name="l03301"></a><span class="lineno"> 3301</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| 3400 | <div class="line"><a name="l03302"></a><span class="lineno"> 3302</span>  operation, 3, *layer, 3, model, data, <span class="keyword">nullptr</span>, validateFunc, ActivationFn::kActivationNone, <span class="keyword">true</span>));</div> |
| 3401 | <div class="line"><a name="l03303"></a><span class="lineno"> 3303</span>  }</div> |
| 3402 | <div class="line"><a name="l03304"></a><span class="lineno"> 3304</span>  </div> |
| 3403 | <div class="line"><a name="l03305"></a><span class="lineno"> 3305</span> }</div> |
| 3404 | <div class="line"><a name="l03306"></a><span class="lineno"> 3306</span>  </div> |
| 3405 | <div class="line"><a name="l03307"></a><span class="lineno"> 3307</span> <span class="keywordtype">bool</span> Converter::ConvertMaxPool2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3406 | <div class="line"><a name="l03308"></a><span class="lineno"> 3308</span> {</div> |
| 3407 | <div class="line"><a name="l03309"></a><span class="lineno"> 3309</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertMaxPool2d()"</span>;</div> |
| 3408 | <div class="line"><a name="l03310"></a><span class="lineno"> 3310</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a2eb75cb48464cf87c290503d7bea79a4">ConvertPooling2d</a>(operation, __func__, PoolingAlgorithm::Max, model, data);</div> |
| 3409 | <div class="line"><a name="l03311"></a><span class="lineno"> 3311</span> }</div> |
| 3410 | <div class="line"><a name="l03312"></a><span class="lineno"> 3312</span>  </div> |
| 3411 | <div class="line"><a name="l03313"></a><span class="lineno"> 3313</span> <span class="keywordtype">bool</span> Converter::ConvertMean(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3412 | <div class="line"><a name="l03314"></a><span class="lineno"> 3314</span> {</div> |
| 3413 | <div class="line"><a name="l03315"></a><span class="lineno"> 3315</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertMean()"</span>;</div> |
| 3414 | <div class="line"><a name="l03316"></a><span class="lineno"> 3316</span>  </div> |
| 3415 | <div class="line"><a name="l03317"></a><span class="lineno"> 3317</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 3416 | <div class="line"><a name="l03318"></a><span class="lineno"> 3318</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3417 | <div class="line"><a name="l03319"></a><span class="lineno"> 3319</span>  {</div> |
| 3418 | <div class="line"><a name="l03320"></a><span class="lineno"> 3320</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 3419 | <div class="line"><a name="l03321"></a><span class="lineno"> 3321</span>  }</div> |
| 3420 | <div class="line"><a name="l03322"></a><span class="lineno"> 3322</span>  </div> |
| 3421 | <div class="line"><a name="l03323"></a><span class="lineno"> 3323</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 3422 | <div class="line"><a name="l03324"></a><span class="lineno"> 3324</span>  <span class="keywordflow">if</span> (!output)</div> |
| 3423 | <div class="line"><a name="l03325"></a><span class="lineno"> 3325</span>  {</div> |
| 3424 | <div class="line"><a name="l03326"></a><span class="lineno"> 3326</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 3425 | <div class="line"><a name="l03327"></a><span class="lineno"> 3327</span>  }</div> |
| 3426 | <div class="line"><a name="l03328"></a><span class="lineno"> 3328</span>  </div> |
| 3427 | <div class="line"><a name="l03329"></a><span class="lineno"> 3329</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 3428 | <div class="line"><a name="l03330"></a><span class="lineno"> 3330</span>  </div> |
| 3429 | <div class="line"><a name="l03331"></a><span class="lineno"> 3331</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* axisOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 3430 | <div class="line"><a name="l03332"></a><span class="lineno"> 3332</span>  <span class="keywordflow">if</span> (!axisOperand)</div> |
| 3431 | <div class="line"><a name="l03333"></a><span class="lineno"> 3333</span>  {</div> |
| 3432 | <div class="line"><a name="l03334"></a><span class="lineno"> 3334</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 1"</span>, __func__);</div> |
| 3433 | <div class="line"><a name="l03335"></a><span class="lineno"> 3335</span>  }</div> |
| 3434 | <div class="line"><a name="l03336"></a><span class="lineno"> 3336</span>  </div> |
| 3435 | <div class="line"><a name="l03337"></a><span class="lineno"> 3337</span>  std::vector<int32_t> axis;</div> |
| 3436 | <div class="line"><a name="l03338"></a><span class="lineno"> 3338</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*axisOperand, axis, model, data))</div> |
| 3437 | <div class="line"><a name="l03339"></a><span class="lineno"> 3339</span>  {</div> |
| 3438 | <div class="line"><a name="l03340"></a><span class="lineno"> 3340</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Input 1 has invalid values"</span>, __func__);</div> |
| 3439 | <div class="line"><a name="l03341"></a><span class="lineno"> 3341</span>  }</div> |
| 3440 | <div class="line"><a name="l03342"></a><span class="lineno"> 3342</span>  </div> |
| 3441 | <div class="line"><a name="l03343"></a><span class="lineno"> 3343</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3442 | <div class="line"><a name="l03344"></a><span class="lineno"> 3344</span>  </div> |
| 3443 | <div class="line"><a name="l03345"></a><span class="lineno"> 3345</span>  <span class="comment">// Convert the axis to unsigned int and remove duplicates.</span></div> |
| 3444 | <div class="line"><a name="l03346"></a><span class="lineno"> 3346</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 3445 | <div class="line"><a name="l03347"></a><span class="lineno"> 3347</span>  std::set<unsigned int> uniqueAxis;</div> |
| 3446 | <div class="line"><a name="l03348"></a><span class="lineno"> 3348</span>  std::transform(axis.begin(), axis.end(),</div> |
| 3447 | <div class="line"><a name="l03349"></a><span class="lineno"> 3349</span>  std::inserter(uniqueAxis, uniqueAxis.begin()),</div> |
| 3448 | <div class="line"><a name="l03350"></a><span class="lineno"> 3350</span>  [rank](<span class="keywordtype">int</span> i) -> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> { return (i + rank) % rank; });</div> |
| 3449 | <div class="line"><a name="l03351"></a><span class="lineno"> 3351</span>  </div> |
| 3450 | <div class="line"><a name="l03352"></a><span class="lineno"> 3352</span>  <span class="comment">// Get the "keep dims" flag.</span></div> |
| 3451 | <div class="line"><a name="l03353"></a><span class="lineno"> 3353</span>  int32_t keepDims = 0;</div> |
| 3452 | <div class="line"><a name="l03354"></a><span class="lineno"> 3354</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 2, keepDims, model, data))</div> |
| 3453 | <div class="line"><a name="l03355"></a><span class="lineno"> 3355</span>  {</div> |
| 3454 | <div class="line"><a name="l03356"></a><span class="lineno"> 3356</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 2"</span>, __func__);</div> |
| 3455 | <div class="line"><a name="l03357"></a><span class="lineno"> 3357</span>  }</div> |
| 3456 | <div class="line"><a name="l03358"></a><span class="lineno"> 3358</span>  </div> |
| 3457 | <div class="line"><a name="l03359"></a><span class="lineno"> 3359</span>  <a class="code" href="structarmnn_1_1_mean_descriptor.html">armnn::MeanDescriptor</a> descriptor;</div> |
| 3458 | <div class="line"><a name="l03360"></a><span class="lineno"> 3360</span>  descriptor.<a class="code" href="structarmnn_1_1_mean_descriptor.html#a1f0d67b087c491248bd1cde3ff995a95">m_Axis</a>.assign(uniqueAxis.begin(), uniqueAxis.end());</div> |
| 3459 | <div class="line"><a name="l03361"></a><span class="lineno"> 3361</span>  descriptor.<a class="code" href="structarmnn_1_1_mean_descriptor.html#a28e0548abfc4e79c48f29a3d11a062e9">m_KeepDims</a> = keepDims > 0;</div> |
| 3460 | <div class="line"><a name="l03362"></a><span class="lineno"> 3362</span>  </div> |
| 3461 | <div class="line"><a name="l03363"></a><span class="lineno"> 3363</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 3462 | <div class="line"><a name="l03364"></a><span class="lineno"> 3364</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 3463 | <div class="line"><a name="l03365"></a><span class="lineno"> 3365</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 3464 | <div class="line"><a name="l03366"></a><span class="lineno"> 3366</span>  {</div> |
| 3465 | <div class="line"><a name="l03367"></a><span class="lineno"> 3367</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 3466 | <div class="line"><a name="l03368"></a><span class="lineno"> 3368</span>  IsMeanSupported,</div> |
| 3467 | <div class="line"><a name="l03369"></a><span class="lineno"> 3369</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 3468 | <div class="line"><a name="l03370"></a><span class="lineno"> 3370</span>  isSupported,</div> |
| 3469 | <div class="line"><a name="l03371"></a><span class="lineno"> 3371</span>  setBackend,</div> |
| 3470 | <div class="line"><a name="l03372"></a><span class="lineno"> 3372</span>  inputInfo,</div> |
| 3471 | <div class="line"><a name="l03373"></a><span class="lineno"> 3373</span>  outputInfo,</div> |
| 3472 | <div class="line"><a name="l03374"></a><span class="lineno"> 3374</span>  descriptor);</div> |
| 3473 | <div class="line"><a name="l03375"></a><span class="lineno"> 3375</span>  };</div> |
| 3474 | <div class="line"><a name="l03376"></a><span class="lineno"> 3376</span>  </div> |
| 3475 | <div class="line"><a name="l03377"></a><span class="lineno"> 3377</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 3476 | <div class="line"><a name="l03378"></a><span class="lineno"> 3378</span>  {</div> |
| 3477 | <div class="line"><a name="l03379"></a><span class="lineno"> 3379</span>  validateFunc(outputInfo, isSupported);</div> |
| 3478 | <div class="line"><a name="l03380"></a><span class="lineno"> 3380</span>  }</div> |
| 3479 | <div class="line"><a name="l03381"></a><span class="lineno"> 3381</span>  <span class="keywordflow">else</span></div> |
| 3480 | <div class="line"><a name="l03382"></a><span class="lineno"> 3382</span>  {</div> |
| 3481 | <div class="line"><a name="l03383"></a><span class="lineno"> 3383</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 3482 | <div class="line"><a name="l03384"></a><span class="lineno"> 3384</span>  }</div> |
| 3483 | <div class="line"><a name="l03385"></a><span class="lineno"> 3385</span>  </div> |
| 3484 | <div class="line"><a name="l03386"></a><span class="lineno"> 3386</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 3485 | <div class="line"><a name="l03387"></a><span class="lineno"> 3387</span>  {</div> |
| 3486 | <div class="line"><a name="l03388"></a><span class="lineno"> 3388</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3487 | <div class="line"><a name="l03389"></a><span class="lineno"> 3389</span>  }</div> |
| 3488 | <div class="line"><a name="l03390"></a><span class="lineno"> 3390</span>  </div> |
| 3489 | <div class="line"><a name="l03391"></a><span class="lineno"> 3391</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddMeanLayer(descriptor);</div> |
| 3490 | <div class="line"><a name="l03392"></a><span class="lineno"> 3392</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 3491 | <div class="line"><a name="l03393"></a><span class="lineno"> 3393</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 3492 | <div class="line"><a name="l03394"></a><span class="lineno"> 3394</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 3493 | <div class="line"><a name="l03395"></a><span class="lineno"> 3395</span>  </div> |
| 3494 | <div class="line"><a name="l03396"></a><span class="lineno"> 3396</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 3495 | <div class="line"><a name="l03397"></a><span class="lineno"> 3397</span> }</div> |
| 3496 | <div class="line"><a name="l03398"></a><span class="lineno"> 3398</span>  </div> |
| 3497 | <div class="line"><a name="l03399"></a><span class="lineno"> 3399</span> <span class="keywordtype">bool</span> Converter::ConvertPad(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3498 | <div class="line"><a name="l03400"></a><span class="lineno"> 3400</span> {</div> |
| 3499 | <div class="line"><a name="l03401"></a><span class="lineno"> 3401</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertPad()"</span>;</div> |
| 3500 | <div class="line"><a name="l03402"></a><span class="lineno"> 3402</span>  </div> |
| 3501 | <div class="line"><a name="l03403"></a><span class="lineno"> 3403</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 3502 | <div class="line"><a name="l03404"></a><span class="lineno"> 3404</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3503 | <div class="line"><a name="l03405"></a><span class="lineno"> 3405</span>  {</div> |
| 3504 | <div class="line"><a name="l03406"></a><span class="lineno"> 3406</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 3505 | <div class="line"><a name="l03407"></a><span class="lineno"> 3407</span>  }</div> |
| 3506 | <div class="line"><a name="l03408"></a><span class="lineno"> 3408</span>  </div> |
| 3507 | <div class="line"><a name="l03409"></a><span class="lineno"> 3409</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3508 | <div class="line"><a name="l03410"></a><span class="lineno"> 3410</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 3509 | <div class="line"><a name="l03411"></a><span class="lineno"> 3411</span>  </div> |
| 3510 | <div class="line"><a name="l03412"></a><span class="lineno"> 3412</span>  <a class="code" href="structarmnn_1_1_pad_descriptor.html">armnn::PadDescriptor</a> descriptor;</div> |
| 3511 | <div class="line"><a name="l03413"></a><span class="lineno"> 3413</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a8d4f662a4fb5b9272724bdaeacbd370a">ConvertPaddings</a>(operation, model, data, rank, descriptor))</div> |
| 3512 | <div class="line"><a name="l03414"></a><span class="lineno"> 3414</span>  {</div> |
| 3513 | <div class="line"><a name="l03415"></a><span class="lineno"> 3415</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not convert paddings"</span>, __func__);</div> |
| 3514 | <div class="line"><a name="l03416"></a><span class="lineno"> 3416</span>  }</div> |
| 3515 | <div class="line"><a name="l03417"></a><span class="lineno"> 3417</span>  </div> |
| 3516 | <div class="line"><a name="l03418"></a><span class="lineno"> 3418</span>  <span class="comment">// For a ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and ANEURALNETWORKS_TENSOR_QUANT8_ASYMM_SIGNED tensor,</span></div> |
| 3517 | <div class="line"><a name="l03419"></a><span class="lineno"> 3419</span>  <span class="comment">// the scale and zeroPoint must be the same as input0</span></div> |
| 3518 | <div class="line"><a name="l03420"></a><span class="lineno"> 3420</span>  <span class="comment">// Before Android Q, the pad value for ANEURALNETWORKS_TENSOR_QUANT8_ASYMM was undefined. Since Android Q the pad</span></div> |
| 3519 | <div class="line"><a name="l03421"></a><span class="lineno"> 3421</span>  <span class="comment">// value must be "logical zero" we set it to be equal to the QuantizationOffset so effectively it ends up as</span></div> |
| 3520 | <div class="line"><a name="l03422"></a><span class="lineno"> 3422</span>  <span class="comment">// (QuantizationOffset - QuantizationOffset) * scale = 0.</span></div> |
| 3521 | <div class="line"><a name="l03423"></a><span class="lineno"> 3423</span>  <span class="keywordflow">if</span> (inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#aea909c7327109228ef618d459015def3">GetDataType</a>() == <a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e">armnn::DataType::QAsymmU8</a> || inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#aea909c7327109228ef618d459015def3">GetDataType</a>() == <a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a9d02ea768c081d4bdb2b7cab0b3f510d">armnn::DataType::QAsymmS8</a>)</div> |
| 3522 | <div class="line"><a name="l03424"></a><span class="lineno"> 3424</span>  {</div> |
| 3523 | <div class="line"><a name="l03425"></a><span class="lineno"> 3425</span>  descriptor.<a class="code" href="structarmnn_1_1_pad_descriptor.html#a410fa919f78af0f0f100bd1594eca4ab">m_PadValue</a> = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a770b51078da02f44a819e9f95d8058b5">GetQuantizationOffset</a>();</div> |
| 3524 | <div class="line"><a name="l03426"></a><span class="lineno"> 3426</span>  }</div> |
| 3525 | <div class="line"><a name="l03427"></a><span class="lineno"> 3427</span>  </div> |
| 3526 | <div class="line"><a name="l03428"></a><span class="lineno"> 3428</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 3527 | <div class="line"><a name="l03429"></a><span class="lineno"> 3429</span>  <span class="keywordflow">if</span> (!output)</div> |
| 3528 | <div class="line"><a name="l03430"></a><span class="lineno"> 3430</span>  {</div> |
| 3529 | <div class="line"><a name="l03431"></a><span class="lineno"> 3431</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output"</span>, __func__);</div> |
| 3530 | <div class="line"><a name="l03432"></a><span class="lineno"> 3432</span>  }</div> |
| 3531 | <div class="line"><a name="l03433"></a><span class="lineno"> 3433</span>  </div> |
| 3532 | <div class="line"><a name="l03434"></a><span class="lineno"> 3434</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 3533 | <div class="line"><a name="l03435"></a><span class="lineno"> 3435</span>  </div> |
| 3534 | <div class="line"><a name="l03436"></a><span class="lineno"> 3436</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 3535 | <div class="line"><a name="l03437"></a><span class="lineno"> 3437</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 3536 | <div class="line"><a name="l03438"></a><span class="lineno"> 3438</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 3537 | <div class="line"><a name="l03439"></a><span class="lineno"> 3439</span>  {</div> |
| 3538 | <div class="line"><a name="l03440"></a><span class="lineno"> 3440</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 3539 | <div class="line"><a name="l03441"></a><span class="lineno"> 3441</span>  IsPadSupported,</div> |
| 3540 | <div class="line"><a name="l03442"></a><span class="lineno"> 3442</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 3541 | <div class="line"><a name="l03443"></a><span class="lineno"> 3443</span>  isSupported,</div> |
| 3542 | <div class="line"><a name="l03444"></a><span class="lineno"> 3444</span>  setBackend,</div> |
| 3543 | <div class="line"><a name="l03445"></a><span class="lineno"> 3445</span>  inputInfo,</div> |
| 3544 | <div class="line"><a name="l03446"></a><span class="lineno"> 3446</span>  outputInfo,</div> |
| 3545 | <div class="line"><a name="l03447"></a><span class="lineno"> 3447</span>  descriptor);</div> |
| 3546 | <div class="line"><a name="l03448"></a><span class="lineno"> 3448</span>  };</div> |
| 3547 | <div class="line"><a name="l03449"></a><span class="lineno"> 3449</span>  </div> |
| 3548 | <div class="line"><a name="l03450"></a><span class="lineno"> 3450</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 3549 | <div class="line"><a name="l03451"></a><span class="lineno"> 3451</span>  {</div> |
| 3550 | <div class="line"><a name="l03452"></a><span class="lineno"> 3452</span>  validateFunc(outputInfo, isSupported);</div> |
| 3551 | <div class="line"><a name="l03453"></a><span class="lineno"> 3453</span>  }</div> |
| 3552 | <div class="line"><a name="l03454"></a><span class="lineno"> 3454</span>  <span class="keywordflow">else</span></div> |
| 3553 | <div class="line"><a name="l03455"></a><span class="lineno"> 3455</span>  {</div> |
| 3554 | <div class="line"><a name="l03456"></a><span class="lineno"> 3456</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 3555 | <div class="line"><a name="l03457"></a><span class="lineno"> 3457</span>  }</div> |
| 3556 | <div class="line"><a name="l03458"></a><span class="lineno"> 3458</span>  </div> |
| 3557 | <div class="line"><a name="l03459"></a><span class="lineno"> 3459</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 3558 | <div class="line"><a name="l03460"></a><span class="lineno"> 3460</span>  {</div> |
| 3559 | <div class="line"><a name="l03461"></a><span class="lineno"> 3461</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3560 | <div class="line"><a name="l03462"></a><span class="lineno"> 3462</span>  }</div> |
| 3561 | <div class="line"><a name="l03463"></a><span class="lineno"> 3463</span>  </div> |
| 3562 | <div class="line"><a name="l03464"></a><span class="lineno"> 3464</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddPadLayer(descriptor);</div> |
| 3563 | <div class="line"><a name="l03465"></a><span class="lineno"> 3465</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 3564 | <div class="line"><a name="l03466"></a><span class="lineno"> 3466</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 3565 | <div class="line"><a name="l03467"></a><span class="lineno"> 3467</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 3566 | <div class="line"><a name="l03468"></a><span class="lineno"> 3468</span>  </div> |
| 3567 | <div class="line"><a name="l03469"></a><span class="lineno"> 3469</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 3568 | <div class="line"><a name="l03470"></a><span class="lineno"> 3470</span> }</div> |
| 3569 | <div class="line"><a name="l03471"></a><span class="lineno"> 3471</span>  </div> |
| 3570 | <div class="line"><a name="l03472"></a><span class="lineno"> 3472</span> <span class="keywordtype">bool</span> Converter::ConvertPadV2(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3571 | <div class="line"><a name="l03473"></a><span class="lineno"> 3473</span> {</div> |
| 3572 | <div class="line"><a name="l03474"></a><span class="lineno"> 3474</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertPadV2()"</span>;</div> |
| 3573 | <div class="line"><a name="l03475"></a><span class="lineno"> 3475</span>  </div> |
| 3574 | <div class="line"><a name="l03476"></a><span class="lineno"> 3476</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 3575 | <div class="line"><a name="l03477"></a><span class="lineno"> 3477</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3576 | <div class="line"><a name="l03478"></a><span class="lineno"> 3478</span>  {</div> |
| 3577 | <div class="line"><a name="l03479"></a><span class="lineno"> 3479</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0"</span>, __func__);</div> |
| 3578 | <div class="line"><a name="l03480"></a><span class="lineno"> 3480</span>  }</div> |
| 3579 | <div class="line"><a name="l03481"></a><span class="lineno"> 3481</span>  </div> |
| 3580 | <div class="line"><a name="l03482"></a><span class="lineno"> 3482</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 3581 | <div class="line"><a name="l03483"></a><span class="lineno"> 3483</span>  <span class="keywordflow">if</span> (!output)</div> |
| 3582 | <div class="line"><a name="l03484"></a><span class="lineno"> 3484</span>  {</div> |
| 3583 | <div class="line"><a name="l03485"></a><span class="lineno"> 3485</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output"</span>, __func__);</div> |
| 3584 | <div class="line"><a name="l03486"></a><span class="lineno"> 3486</span>  }</div> |
| 3585 | <div class="line"><a name="l03487"></a><span class="lineno"> 3487</span>  </div> |
| 3586 | <div class="line"><a name="l03488"></a><span class="lineno"> 3488</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3587 | <div class="line"><a name="l03489"></a><span class="lineno"> 3489</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 3588 | <div class="line"><a name="l03490"></a><span class="lineno"> 3490</span>  </div> |
| 3589 | <div class="line"><a name="l03491"></a><span class="lineno"> 3491</span>  <a class="code" href="structarmnn_1_1_pad_descriptor.html">PadDescriptor</a> descriptor;</div> |
| 3590 | <div class="line"><a name="l03492"></a><span class="lineno"> 3492</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a8d4f662a4fb5b9272724bdaeacbd370a">ConvertPaddings</a>(operation, model, data, rank, descriptor))</div> |
| 3591 | <div class="line"><a name="l03493"></a><span class="lineno"> 3493</span>  {</div> |
| 3592 | <div class="line"><a name="l03494"></a><span class="lineno"> 3494</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not convert paddings"</span>, __func__);</div> |
| 3593 | <div class="line"><a name="l03495"></a><span class="lineno"> 3495</span>  }</div> |
| 3594 | <div class="line"><a name="l03496"></a><span class="lineno"> 3496</span>  </div> |
| 3595 | <div class="line"><a name="l03497"></a><span class="lineno"> 3497</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 3596 | <div class="line"><a name="l03498"></a><span class="lineno"> 3498</span>  </div> |
| 3597 | <div class="line"><a name="l03499"></a><span class="lineno"> 3499</span>  <span class="comment">// Determine type of padding value</span></div> |
| 3598 | <div class="line"><a name="l03500"></a><span class="lineno"> 3500</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType0;</div> |
| 3599 | <div class="line"><a name="l03501"></a><span class="lineno"> 3501</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType2;</div> |
| 3600 | <div class="line"><a name="l03502"></a><span class="lineno"> 3502</span>  </div> |
| 3601 | <div class="line"><a name="l03503"></a><span class="lineno"> 3503</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 0, model, operandType0) ||</div> |
| 3602 | <div class="line"><a name="l03504"></a><span class="lineno"> 3504</span>  !<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 2, model, operandType2))</div> |
| 3603 | <div class="line"><a name="l03505"></a><span class="lineno"> 3505</span>  {</div> |
| 3604 | <div class="line"><a name="l03506"></a><span class="lineno"> 3506</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 3605 | <div class="line"><a name="l03507"></a><span class="lineno"> 3507</span>  }</div> |
| 3606 | <div class="line"><a name="l03508"></a><span class="lineno"> 3508</span>  </div> |
| 3607 | <div class="line"><a name="l03509"></a><span class="lineno"> 3509</span>  <span class="comment">// Read value to use for padding</span></div> |
| 3608 | <div class="line"><a name="l03510"></a><span class="lineno"> 3510</span>  <span class="keywordflow">if</span> (operandType0 == OperandType::TENSOR_FLOAT16 && operandType2 == OperandType::FLOAT16)</div> |
| 3609 | <div class="line"><a name="l03511"></a><span class="lineno"> 3511</span>  {</div> |
| 3610 | <div class="line"><a name="l03512"></a><span class="lineno"> 3512</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> f16PadValue;</div> |
| 3611 | <div class="line"><a name="l03513"></a><span class="lineno"> 3513</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 2, operandType2, f16PadValue, model, data))</div> |
| 3612 | <div class="line"><a name="l03514"></a><span class="lineno"> 3514</span>  {</div> |
| 3613 | <div class="line"><a name="l03515"></a><span class="lineno"> 3515</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 2 (FLOAT16)"</span>, __func__);</div> |
| 3614 | <div class="line"><a name="l03516"></a><span class="lineno"> 3516</span>  }</div> |
| 3615 | <div class="line"><a name="l03517"></a><span class="lineno"> 3517</span>  </div> |
| 3616 | <div class="line"><a name="l03518"></a><span class="lineno"> 3518</span>  descriptor.<a class="code" href="structarmnn_1_1_pad_descriptor.html#a410fa919f78af0f0f100bd1594eca4ab">m_PadValue</a> = f16PadValue;</div> |
| 3617 | <div class="line"><a name="l03519"></a><span class="lineno"> 3519</span>  }</div> |
| 3618 | <div class="line"><a name="l03520"></a><span class="lineno"> 3520</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operandType0 == OperandType::TENSOR_FLOAT32 && operandType2 == OperandType::FLOAT32)</div> |
| 3619 | <div class="line"><a name="l03521"></a><span class="lineno"> 3521</span>  {</div> |
| 3620 | <div class="line"><a name="l03522"></a><span class="lineno"> 3522</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 2, descriptor.<a class="code" href="structarmnn_1_1_pad_descriptor.html#a410fa919f78af0f0f100bd1594eca4ab">m_PadValue</a>, model, data))</div> |
| 3621 | <div class="line"><a name="l03523"></a><span class="lineno"> 3523</span>  {</div> |
| 3622 | <div class="line"><a name="l03524"></a><span class="lineno"> 3524</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 2 (FLOAT32)"</span>, __func__);</div> |
| 3623 | <div class="line"><a name="l03525"></a><span class="lineno"> 3525</span>  }</div> |
| 3624 | <div class="line"><a name="l03526"></a><span class="lineno"> 3526</span>  }</div> |
| 3625 | <div class="line"><a name="l03527"></a><span class="lineno"> 3527</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#ab3c135a0d4b1152f6b989ea6afddd6bc">isQuantizedOperand</a>(operandType0) && operandType2 == OperandType::INT32)</div> |
| 3626 | <div class="line"><a name="l03528"></a><span class="lineno"> 3528</span>  {</div> |
| 3627 | <div class="line"><a name="l03529"></a><span class="lineno"> 3529</span>  int32_t intPadValue = 0;</div> |
| 3628 | <div class="line"><a name="l03530"></a><span class="lineno"> 3530</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 2, intPadValue, model, data))</div> |
| 3629 | <div class="line"><a name="l03531"></a><span class="lineno"> 3531</span>  {</div> |
| 3630 | <div class="line"><a name="l03532"></a><span class="lineno"> 3532</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 2 (INT32)"</span>, __func__);</div> |
| 3631 | <div class="line"><a name="l03533"></a><span class="lineno"> 3533</span>  }</div> |
| 3632 | <div class="line"><a name="l03534"></a><span class="lineno"> 3534</span>  descriptor.<a class="code" href="structarmnn_1_1_pad_descriptor.html#a410fa919f78af0f0f100bd1594eca4ab">m_PadValue</a> = intPadValue;</div> |
| 3633 | <div class="line"><a name="l03535"></a><span class="lineno"> 3535</span>  }</div> |
| 3634 | <div class="line"><a name="l03536"></a><span class="lineno"> 3536</span>  <span class="keywordflow">else</span></div> |
| 3635 | <div class="line"><a name="l03537"></a><span class="lineno"> 3537</span>  {</div> |
| 3636 | <div class="line"><a name="l03538"></a><span class="lineno"> 3538</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs: type mismatch"</span>, __func__);</div> |
| 3637 | <div class="line"><a name="l03539"></a><span class="lineno"> 3539</span>  }</div> |
| 3638 | <div class="line"><a name="l03540"></a><span class="lineno"> 3540</span>  </div> |
| 3639 | <div class="line"><a name="l03541"></a><span class="lineno"> 3541</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 3640 | <div class="line"><a name="l03542"></a><span class="lineno"> 3542</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 3641 | <div class="line"><a name="l03543"></a><span class="lineno"> 3543</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 3642 | <div class="line"><a name="l03544"></a><span class="lineno"> 3544</span>  {</div> |
| 3643 | <div class="line"><a name="l03545"></a><span class="lineno"> 3545</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 3644 | <div class="line"><a name="l03546"></a><span class="lineno"> 3546</span>  IsPadSupported,</div> |
| 3645 | <div class="line"><a name="l03547"></a><span class="lineno"> 3547</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 3646 | <div class="line"><a name="l03548"></a><span class="lineno"> 3548</span>  isSupported,</div> |
| 3647 | <div class="line"><a name="l03549"></a><span class="lineno"> 3549</span>  setBackend,</div> |
| 3648 | <div class="line"><a name="l03550"></a><span class="lineno"> 3550</span>  inputInfo,</div> |
| 3649 | <div class="line"><a name="l03551"></a><span class="lineno"> 3551</span>  outputInfo,</div> |
| 3650 | <div class="line"><a name="l03552"></a><span class="lineno"> 3552</span>  descriptor);</div> |
| 3651 | <div class="line"><a name="l03553"></a><span class="lineno"> 3553</span>  };</div> |
| 3652 | <div class="line"><a name="l03554"></a><span class="lineno"> 3554</span>  </div> |
| 3653 | <div class="line"><a name="l03555"></a><span class="lineno"> 3555</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 3654 | <div class="line"><a name="l03556"></a><span class="lineno"> 3556</span>  {</div> |
| 3655 | <div class="line"><a name="l03557"></a><span class="lineno"> 3557</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 3656 | <div class="line"><a name="l03558"></a><span class="lineno"> 3558</span>  }</div> |
| 3657 | <div class="line"><a name="l03559"></a><span class="lineno"> 3559</span>  <span class="keywordflow">else</span></div> |
| 3658 | <div class="line"><a name="l03560"></a><span class="lineno"> 3560</span>  {</div> |
| 3659 | <div class="line"><a name="l03561"></a><span class="lineno"> 3561</span>  validateFunc(outputInfo, isSupported);</div> |
| 3660 | <div class="line"><a name="l03562"></a><span class="lineno"> 3562</span>  }</div> |
| 3661 | <div class="line"><a name="l03563"></a><span class="lineno"> 3563</span>  </div> |
| 3662 | <div class="line"><a name="l03564"></a><span class="lineno"> 3564</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 3663 | <div class="line"><a name="l03565"></a><span class="lineno"> 3565</span>  {</div> |
| 3664 | <div class="line"><a name="l03566"></a><span class="lineno"> 3566</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3665 | <div class="line"><a name="l03567"></a><span class="lineno"> 3567</span>  }</div> |
| 3666 | <div class="line"><a name="l03568"></a><span class="lineno"> 3568</span>  </div> |
| 3667 | <div class="line"><a name="l03569"></a><span class="lineno"> 3569</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddPadLayer(descriptor);</div> |
| 3668 | <div class="line"><a name="l03570"></a><span class="lineno"> 3570</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 3669 | <div class="line"><a name="l03571"></a><span class="lineno"> 3571</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 3670 | <div class="line"><a name="l03572"></a><span class="lineno"> 3572</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 3671 | <div class="line"><a name="l03573"></a><span class="lineno"> 3573</span>  </div> |
| 3672 | <div class="line"><a name="l03574"></a><span class="lineno"> 3574</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 3673 | <div class="line"><a name="l03575"></a><span class="lineno"> 3575</span> }</div> |
| 3674 | <div class="line"><a name="l03576"></a><span class="lineno"> 3576</span>  </div> |
| 3675 | <div class="line"><a name="l03577"></a><span class="lineno"> 3577</span> <span class="keywordtype">bool</span> Converter::ConvertPrelu(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3676 | <div class="line"><a name="l03578"></a><span class="lineno"> 3578</span> {</div> |
| 3677 | <div class="line"><a name="l03579"></a><span class="lineno"> 3579</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertPrelu()"</span>;</div> |
| 3678 | <div class="line"><a name="l03580"></a><span class="lineno"> 3580</span>  </div> |
| 3679 | <div class="line"><a name="l03581"></a><span class="lineno"> 3581</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 3680 | <div class="line"><a name="l03582"></a><span class="lineno"> 3582</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> alpha = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 3681 | <div class="line"><a name="l03583"></a><span class="lineno"> 3583</span>  </div> |
| 3682 | <div class="line"><a name="l03584"></a><span class="lineno"> 3584</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() || !alpha.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3683 | <div class="line"><a name="l03585"></a><span class="lineno"> 3585</span>  {</div> |
| 3684 | <div class="line"><a name="l03586"></a><span class="lineno"> 3586</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 3685 | <div class="line"><a name="l03587"></a><span class="lineno"> 3587</span>  }</div> |
| 3686 | <div class="line"><a name="l03588"></a><span class="lineno"> 3588</span>  </div> |
| 3687 | <div class="line"><a name="l03589"></a><span class="lineno"> 3589</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 3688 | <div class="line"><a name="l03590"></a><span class="lineno"> 3590</span>  </div> |
| 3689 | <div class="line"><a name="l03591"></a><span class="lineno"> 3591</span>  <span class="keywordflow">if</span> (!output)</div> |
| 3690 | <div class="line"><a name="l03592"></a><span class="lineno"> 3592</span>  {</div> |
| 3691 | <div class="line"><a name="l03593"></a><span class="lineno"> 3593</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output"</span>, __func__);</div> |
| 3692 | <div class="line"><a name="l03594"></a><span class="lineno"> 3594</span>  }</div> |
| 3693 | <div class="line"><a name="l03595"></a><span class="lineno"> 3595</span>  </div> |
| 3694 | <div class="line"><a name="l03596"></a><span class="lineno"> 3596</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3695 | <div class="line"><a name="l03597"></a><span class="lineno"> 3597</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& alphaInfo = alpha.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 3696 | <div class="line"><a name="l03598"></a><span class="lineno"> 3598</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 3697 | <div class="line"><a name="l03599"></a><span class="lineno"> 3599</span>  </div> |
| 3698 | <div class="line"><a name="l03600"></a><span class="lineno"> 3600</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 3699 | <div class="line"><a name="l03601"></a><span class="lineno"> 3601</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 3700 | <div class="line"><a name="l03602"></a><span class="lineno"> 3602</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 3701 | <div class="line"><a name="l03603"></a><span class="lineno"> 3603</span>  {</div> |
| 3702 | <div class="line"><a name="l03604"></a><span class="lineno"> 3604</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 3703 | <div class="line"><a name="l03605"></a><span class="lineno"> 3605</span>  IsPreluSupported,</div> |
| 3704 | <div class="line"><a name="l03606"></a><span class="lineno"> 3606</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 3705 | <div class="line"><a name="l03607"></a><span class="lineno"> 3607</span>  isSupported,</div> |
| 3706 | <div class="line"><a name="l03608"></a><span class="lineno"> 3608</span>  setBackend,</div> |
| 3707 | <div class="line"><a name="l03609"></a><span class="lineno"> 3609</span>  inputInfo,</div> |
| 3708 | <div class="line"><a name="l03610"></a><span class="lineno"> 3610</span>  alphaInfo,</div> |
| 3709 | <div class="line"><a name="l03611"></a><span class="lineno"> 3611</span>  outputInfo);</div> |
| 3710 | <div class="line"><a name="l03612"></a><span class="lineno"> 3612</span>  };</div> |
| 3711 | <div class="line"><a name="l03613"></a><span class="lineno"> 3613</span>  </div> |
| 3712 | <div class="line"><a name="l03614"></a><span class="lineno"> 3614</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 3713 | <div class="line"><a name="l03615"></a><span class="lineno"> 3615</span>  {</div> |
| 3714 | <div class="line"><a name="l03616"></a><span class="lineno"> 3616</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 3715 | <div class="line"><a name="l03617"></a><span class="lineno"> 3617</span>  }</div> |
| 3716 | <div class="line"><a name="l03618"></a><span class="lineno"> 3618</span>  <span class="keywordflow">else</span></div> |
| 3717 | <div class="line"><a name="l03619"></a><span class="lineno"> 3619</span>  {</div> |
| 3718 | <div class="line"><a name="l03620"></a><span class="lineno"> 3620</span>  validateFunc(outputInfo, isSupported);</div> |
| 3719 | <div class="line"><a name="l03621"></a><span class="lineno"> 3621</span>  }</div> |
| 3720 | <div class="line"><a name="l03622"></a><span class="lineno"> 3622</span>  </div> |
| 3721 | <div class="line"><a name="l03623"></a><span class="lineno"> 3623</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 3722 | <div class="line"><a name="l03624"></a><span class="lineno"> 3624</span>  {</div> |
| 3723 | <div class="line"><a name="l03625"></a><span class="lineno"> 3625</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3724 | <div class="line"><a name="l03626"></a><span class="lineno"> 3626</span>  }</div> |
| 3725 | <div class="line"><a name="l03627"></a><span class="lineno"> 3627</span>  </div> |
| 3726 | <div class="line"><a name="l03628"></a><span class="lineno"> 3628</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddPreluLayer();</div> |
| 3727 | <div class="line"><a name="l03629"></a><span class="lineno"> 3629</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 3728 | <div class="line"><a name="l03630"></a><span class="lineno"> 3630</span>  </div> |
| 3729 | <div class="line"><a name="l03631"></a><span class="lineno"> 3631</span>  <span class="keywordflow">if</span> (!layer)</div> |
| 3730 | <div class="line"><a name="l03632"></a><span class="lineno"> 3632</span>  {</div> |
| 3731 | <div class="line"><a name="l03633"></a><span class="lineno"> 3633</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddPreluLayer failed"</span>, __func__);</div> |
| 3732 | <div class="line"><a name="l03634"></a><span class="lineno"> 3634</span>  }</div> |
| 3733 | <div class="line"><a name="l03635"></a><span class="lineno"> 3635</span>  </div> |
| 3734 | <div class="line"><a name="l03636"></a><span class="lineno"> 3636</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input, alpha, layer, data);</div> |
| 3735 | <div class="line"><a name="l03637"></a><span class="lineno"> 3637</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| 3736 | <div class="line"><a name="l03638"></a><span class="lineno"> 3638</span>  {</div> |
| 3737 | <div class="line"><a name="l03639"></a><span class="lineno"> 3639</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3738 | <div class="line"><a name="l03640"></a><span class="lineno"> 3640</span>  }</div> |
| 3739 | <div class="line"><a name="l03641"></a><span class="lineno"> 3641</span>  </div> |
| 3740 | <div class="line"><a name="l03642"></a><span class="lineno"> 3642</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 3741 | <div class="line"><a name="l03643"></a><span class="lineno"> 3643</span> }</div> |
| 3742 | <div class="line"><a name="l03644"></a><span class="lineno"> 3644</span>  </div> |
| 3743 | <div class="line"><a name="l03645"></a><span class="lineno"> 3645</span> <span class="keywordtype">bool</span> Converter::ConvertQuantize(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3744 | <div class="line"><a name="l03646"></a><span class="lineno"> 3646</span> {</div> |
| 3745 | <div class="line"><a name="l03647"></a><span class="lineno"> 3647</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertQuantize()"</span>;</div> |
| 3746 | <div class="line"><a name="l03648"></a><span class="lineno"> 3648</span>  </div> |
| 3747 | <div class="line"><a name="l03649"></a><span class="lineno"> 3649</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 3748 | <div class="line"><a name="l03650"></a><span class="lineno"> 3650</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3749 | <div class="line"><a name="l03651"></a><span class="lineno"> 3651</span>  {</div> |
| 3750 | <div class="line"><a name="l03652"></a><span class="lineno"> 3652</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid input"</span>, __func__);</div> |
| 3751 | <div class="line"><a name="l03653"></a><span class="lineno"> 3653</span>  }</div> |
| 3752 | <div class="line"><a name="l03654"></a><span class="lineno"> 3654</span>  </div> |
| 3753 | <div class="line"><a name="l03655"></a><span class="lineno"> 3655</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* <span class="keyword">const</span> outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 3754 | <div class="line"><a name="l03656"></a><span class="lineno"> 3656</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 3755 | <div class="line"><a name="l03657"></a><span class="lineno"> 3657</span>  {</div> |
| 3756 | <div class="line"><a name="l03658"></a><span class="lineno"> 3658</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid outputs"</span>, __func__);</div> |
| 3757 | <div class="line"><a name="l03659"></a><span class="lineno"> 3659</span>  }</div> |
| 3758 | <div class="line"><a name="l03660"></a><span class="lineno"> 3660</span>  </div> |
| 3759 | <div class="line"><a name="l03661"></a><span class="lineno"> 3661</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 3760 | <div class="line"><a name="l03662"></a><span class="lineno"> 3662</span>  </div> |
| 3761 | <div class="line"><a name="l03663"></a><span class="lineno"> 3663</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 3762 | <div class="line"><a name="l03664"></a><span class="lineno"> 3664</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 3763 | <div class="line"><a name="l03665"></a><span class="lineno"> 3665</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 3764 | <div class="line"><a name="l03666"></a><span class="lineno"> 3666</span>  {</div> |
| 3765 | <div class="line"><a name="l03667"></a><span class="lineno"> 3667</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 3766 | <div class="line"><a name="l03668"></a><span class="lineno"> 3668</span>  IsQuantizeSupported,</div> |
| 3767 | <div class="line"><a name="l03669"></a><span class="lineno"> 3669</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 3768 | <div class="line"><a name="l03670"></a><span class="lineno"> 3670</span>  isSupported,</div> |
| 3769 | <div class="line"><a name="l03671"></a><span class="lineno"> 3671</span>  setBackend,</div> |
| 3770 | <div class="line"><a name="l03672"></a><span class="lineno"> 3672</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 3771 | <div class="line"><a name="l03673"></a><span class="lineno"> 3673</span>  outputInfo);</div> |
| 3772 | <div class="line"><a name="l03674"></a><span class="lineno"> 3674</span>  };</div> |
| 3773 | <div class="line"><a name="l03675"></a><span class="lineno"> 3675</span>  </div> |
| 3774 | <div class="line"><a name="l03676"></a><span class="lineno"> 3676</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 3775 | <div class="line"><a name="l03677"></a><span class="lineno"> 3677</span>  {</div> |
| 3776 | <div class="line"><a name="l03678"></a><span class="lineno"> 3678</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 3777 | <div class="line"><a name="l03679"></a><span class="lineno"> 3679</span>  }</div> |
| 3778 | <div class="line"><a name="l03680"></a><span class="lineno"> 3680</span>  <span class="keywordflow">else</span></div> |
| 3779 | <div class="line"><a name="l03681"></a><span class="lineno"> 3681</span>  {</div> |
| 3780 | <div class="line"><a name="l03682"></a><span class="lineno"> 3682</span>  validateFunc(outputInfo, isSupported);</div> |
| 3781 | <div class="line"><a name="l03683"></a><span class="lineno"> 3683</span>  }</div> |
| 3782 | <div class="line"><a name="l03684"></a><span class="lineno"> 3684</span>  </div> |
| 3783 | <div class="line"><a name="l03685"></a><span class="lineno"> 3685</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 3784 | <div class="line"><a name="l03686"></a><span class="lineno"> 3686</span>  {</div> |
| 3785 | <div class="line"><a name="l03687"></a><span class="lineno"> 3687</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 3786 | <div class="line"><a name="l03688"></a><span class="lineno"> 3688</span>  }</div> |
| 3787 | <div class="line"><a name="l03689"></a><span class="lineno"> 3689</span>  </div> |
| 3788 | <div class="line"><a name="l03690"></a><span class="lineno"> 3690</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddQuantizeLayer();</div> |
| 3789 | <div class="line"><a name="l03691"></a><span class="lineno"> 3691</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 3790 | <div class="line"><a name="l03692"></a><span class="lineno"> 3692</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 3791 | <div class="line"><a name="l03693"></a><span class="lineno"> 3693</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 3792 | <div class="line"><a name="l03694"></a><span class="lineno"> 3694</span>  </div> |
| 3793 | <div class="line"><a name="l03695"></a><span class="lineno"> 3695</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 3794 | <div class="line"><a name="l03696"></a><span class="lineno"> 3696</span> }</div> |
| 3795 | <div class="line"><a name="l03697"></a><span class="lineno"> 3697</span>  </div> |
| 3796 | <div class="line"><a name="l03698"></a><span class="lineno"> 3698</span> <span class="keywordtype">bool</span> Converter::ConvertQuantizedLstm(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 3797 | <div class="line"><a name="l03699"></a><span class="lineno"> 3699</span> {</div> |
| 3798 | <div class="line"><a name="l03700"></a><span class="lineno"> 3700</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertQuantizedLstm()"</span>;</div> |
| 3799 | <div class="line"><a name="l03701"></a><span class="lineno"> 3701</span>  </div> |
| 3800 | <div class="line"><a name="l03702"></a><span class="lineno"> 3702</span>  VLOG(DRIVER) << <span class="stringliteral">"ConvertQuantizedLstm()"</span>;</div> |
| 3801 | <div class="line"><a name="l03703"></a><span class="lineno"> 3703</span>  </div> |
| 3802 | <div class="line"><a name="l03704"></a><span class="lineno"> 3704</span>  <span class="comment">//Inputs:</span></div> |
| 3803 | <div class="line"><a name="l03705"></a><span class="lineno"> 3705</span>  <span class="comment">// 0: The input: A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape [numBatches, inputSize]</span></div> |
| 3804 | <div class="line"><a name="l03706"></a><span class="lineno"> 3706</span>  <span class="comment">// specifying the input to the LSTM cell. Tensor is quantized with a fixed quantization range of -1, 127/128.</span></div> |
| 3805 | <div class="line"><a name="l03707"></a><span class="lineno"> 3707</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 3806 | <div class="line"><a name="l03708"></a><span class="lineno"> 3708</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3807 | <div class="line"><a name="l03709"></a><span class="lineno"> 3709</span>  {</div> |
| 3808 | <div class="line"><a name="l03710"></a><span class="lineno"> 3710</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0: input"</span>, __func__);</div> |
| 3809 | <div class="line"><a name="l03711"></a><span class="lineno"> 3711</span>  }</div> |
| 3810 | <div class="line"><a name="l03712"></a><span class="lineno"> 3712</span>  </div> |
| 3811 | <div class="line"><a name="l03713"></a><span class="lineno"> 3713</span>  <span class="comment">// 18: The output state: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_ASYMM, of shape [batch_size, output_size].</span></div> |
| 3812 | <div class="line"><a name="l03714"></a><span class="lineno"> 3714</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> outputStatePrevTimeStep = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 18, model, data);</div> |
| 3813 | <div class="line"><a name="l03715"></a><span class="lineno"> 3715</span>  <span class="keywordflow">if</span> (!outputStatePrevTimeStep.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3814 | <div class="line"><a name="l03716"></a><span class="lineno"> 3716</span>  {</div> |
| 3815 | <div class="line"><a name="l03717"></a><span class="lineno"> 3717</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 18: outputStatePrevTimeStep"</span>, __func__);</div> |
| 3816 | <div class="line"><a name="l03718"></a><span class="lineno"> 3718</span>  }</div> |
| 3817 | <div class="line"><a name="l03719"></a><span class="lineno"> 3719</span>  </div> |
| 3818 | <div class="line"><a name="l03720"></a><span class="lineno"> 3720</span>  <span class="comment">// 19: The cell state: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT16_SYMM, of shape [batch_size, num_units].</span></div> |
| 3819 | <div class="line"><a name="l03721"></a><span class="lineno"> 3721</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> cellStatePrevTimeStep = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 19, model, data);</div> |
| 3820 | <div class="line"><a name="l03722"></a><span class="lineno"> 3722</span>  <span class="keywordflow">if</span> (!cellStatePrevTimeStep.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3821 | <div class="line"><a name="l03723"></a><span class="lineno"> 3723</span>  {</div> |
| 3822 | <div class="line"><a name="l03724"></a><span class="lineno"> 3724</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 19: cellStatePrevTimeStep"</span>, __func__);</div> |
| 3823 | <div class="line"><a name="l03725"></a><span class="lineno"> 3725</span>  }</div> |
| 3824 | <div class="line"><a name="l03726"></a><span class="lineno"> 3726</span>  </div> |
| 3825 | <div class="line"><a name="l03727"></a><span class="lineno"> 3727</span>  <span class="comment">// Get the mandatory input tensors:</span></div> |
| 3826 | <div class="line"><a name="l03728"></a><span class="lineno"> 3728</span>  </div> |
| 3827 | <div class="line"><a name="l03729"></a><span class="lineno"> 3729</span>  <span class="comment">// 02: The input-to-forget weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3828 | <div class="line"><a name="l03730"></a><span class="lineno"> 3730</span>  <span class="comment">// [num_units, input_size].</span></div> |
| 3829 | <div class="line"><a name="l03731"></a><span class="lineno"> 3731</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToForgetWeightsPin =</div> |
| 3830 | <div class="line"><a name="l03732"></a><span class="lineno"> 3732</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 2, model, data);</div> |
| 3831 | <div class="line"><a name="l03733"></a><span class="lineno"> 3733</span>  </div> |
| 3832 | <div class="line"><a name="l03734"></a><span class="lineno"> 3734</span>  <span class="comment">// 03: The input-to-cell weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3833 | <div class="line"><a name="l03735"></a><span class="lineno"> 3735</span>  <span class="comment">// [num_units, input_size].</span></div> |
| 3834 | <div class="line"><a name="l03736"></a><span class="lineno"> 3736</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToCellWeightsPin =</div> |
| 3835 | <div class="line"><a name="l03737"></a><span class="lineno"> 3737</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 3, model, data);</div> |
| 3836 | <div class="line"><a name="l03738"></a><span class="lineno"> 3738</span>  </div> |
| 3837 | <div class="line"><a name="l03739"></a><span class="lineno"> 3739</span>  <span class="comment">// 04: The input-to-output weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3838 | <div class="line"><a name="l03740"></a><span class="lineno"> 3740</span>  <span class="comment">// [num_units, input_size].</span></div> |
| 3839 | <div class="line"><a name="l03741"></a><span class="lineno"> 3741</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToOutputWeightsPin =</div> |
| 3840 | <div class="line"><a name="l03742"></a><span class="lineno"> 3742</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 4, model, data);</div> |
| 3841 | <div class="line"><a name="l03743"></a><span class="lineno"> 3743</span>  </div> |
| 3842 | <div class="line"><a name="l03744"></a><span class="lineno"> 3744</span>  <span class="comment">// 06: The recurrent-to-forget weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3843 | <div class="line"><a name="l03745"></a><span class="lineno"> 3745</span>  <span class="comment">// [num_units, output_size].</span></div> |
| 3844 | <div class="line"><a name="l03746"></a><span class="lineno"> 3746</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToForgetWeightsPin =</div> |
| 3845 | <div class="line"><a name="l03747"></a><span class="lineno"> 3747</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 6, model, data);</div> |
| 3846 | <div class="line"><a name="l03748"></a><span class="lineno"> 3748</span>  </div> |
| 3847 | <div class="line"><a name="l03749"></a><span class="lineno"> 3749</span>  <span class="comment">// 07: The recurrent-to-cell weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3848 | <div class="line"><a name="l03750"></a><span class="lineno"> 3750</span>  <span class="comment">// [num_units, output_size].</span></div> |
| 3849 | <div class="line"><a name="l03751"></a><span class="lineno"> 3751</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToCellWeightsPin =</div> |
| 3850 | <div class="line"><a name="l03752"></a><span class="lineno"> 3752</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 7, model, data);</div> |
| 3851 | <div class="line"><a name="l03753"></a><span class="lineno"> 3753</span>  </div> |
| 3852 | <div class="line"><a name="l03754"></a><span class="lineno"> 3754</span>  <span class="comment">// 08: The recurrent-to-output weights: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3853 | <div class="line"><a name="l03755"></a><span class="lineno"> 3755</span>  <span class="comment">// [num_units, output_size].</span></div> |
| 3854 | <div class="line"><a name="l03756"></a><span class="lineno"> 3756</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToOutputWeightsPin =</div> |
| 3855 | <div class="line"><a name="l03757"></a><span class="lineno"> 3757</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 8, model, data);</div> |
| 3856 | <div class="line"><a name="l03758"></a><span class="lineno"> 3758</span>  </div> |
| 3857 | <div class="line"><a name="l03759"></a><span class="lineno"> 3759</span>  <span class="comment">// 13: The forget gate bias: A 1-D tensor of ANEURALNETWORKS_TENSOR_INT32, of shape [num_units].</span></div> |
| 3858 | <div class="line"><a name="l03760"></a><span class="lineno"> 3760</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> forgetGateBiasPin =</div> |
| 3859 | <div class="line"><a name="l03761"></a><span class="lineno"> 3761</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 13, model, data);</div> |
| 3860 | <div class="line"><a name="l03762"></a><span class="lineno"> 3762</span>  </div> |
| 3861 | <div class="line"><a name="l03763"></a><span class="lineno"> 3763</span>  <span class="comment">// 14: The cell bias: A 1-D tensor of ANEURALNETWORKS_TENSOR_INT32, of shape [num_units].</span></div> |
| 3862 | <div class="line"><a name="l03764"></a><span class="lineno"> 3764</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellBiasPin =</div> |
| 3863 | <div class="line"><a name="l03765"></a><span class="lineno"> 3765</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 14, model, data);</div> |
| 3864 | <div class="line"><a name="l03766"></a><span class="lineno"> 3766</span>  </div> |
| 3865 | <div class="line"><a name="l03767"></a><span class="lineno"> 3767</span>  <span class="comment">// 15: The output gate bias: A 1-D tensor of ANEURALNETWORKS_TENSOR_INT32, of shape [num_units].</span></div> |
| 3866 | <div class="line"><a name="l03768"></a><span class="lineno"> 3768</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> outputGateBiasPin =</div> |
| 3867 | <div class="line"><a name="l03769"></a><span class="lineno"> 3769</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 15, model, data);</div> |
| 3868 | <div class="line"><a name="l03770"></a><span class="lineno"> 3770</span>  </div> |
| 3869 | <div class="line"><a name="l03771"></a><span class="lineno"> 3771</span>  <span class="keywordflow">if</span> (!inputToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3870 | <div class="line"><a name="l03772"></a><span class="lineno"> 3772</span>  !inputToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3871 | <div class="line"><a name="l03773"></a><span class="lineno"> 3773</span>  !inputToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3872 | <div class="line"><a name="l03774"></a><span class="lineno"> 3774</span>  !recurrentToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3873 | <div class="line"><a name="l03775"></a><span class="lineno"> 3775</span>  !recurrentToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3874 | <div class="line"><a name="l03776"></a><span class="lineno"> 3776</span>  !recurrentToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3875 | <div class="line"><a name="l03777"></a><span class="lineno"> 3777</span>  !forgetGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3876 | <div class="line"><a name="l03778"></a><span class="lineno"> 3778</span>  !cellBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 3877 | <div class="line"><a name="l03779"></a><span class="lineno"> 3779</span>  !outputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 3878 | <div class="line"><a name="l03780"></a><span class="lineno"> 3780</span>  {</div> |
| 3879 | <div class="line"><a name="l03781"></a><span class="lineno"> 3781</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid tensor inputs"</span>, __func__);</div> |
| 3880 | <div class="line"><a name="l03782"></a><span class="lineno"> 3782</span>  }</div> |
| 3881 | <div class="line"><a name="l03783"></a><span class="lineno"> 3783</span>  </div> |
| 3882 | <div class="line"><a name="l03784"></a><span class="lineno"> 3784</span>  <span class="comment">// Get the optional input tensors:</span></div> |
| 3883 | <div class="line"><a name="l03785"></a><span class="lineno"> 3785</span>  </div> |
| 3884 | <div class="line"><a name="l03786"></a><span class="lineno"> 3786</span>  <span class="comment">// 01: The input-to-input weights: Optional. A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3885 | <div class="line"><a name="l03787"></a><span class="lineno"> 3787</span>  <span class="comment">// [num_units, input_size], where “num_units” corresponds to the number of cell units.</span></div> |
| 3886 | <div class="line"><a name="l03788"></a><span class="lineno"> 3788</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToInputWeightsPin =</div> |
| 3887 | <div class="line"><a name="l03789"></a><span class="lineno"> 3789</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3888 | <div class="line"><a name="l03790"></a><span class="lineno"> 3790</span>  1,</div> |
| 3889 | <div class="line"><a name="l03791"></a><span class="lineno"> 3791</span>  model,</div> |
| 3890 | <div class="line"><a name="l03792"></a><span class="lineno"> 3792</span>  data,</div> |
| 3891 | <div class="line"><a name="l03793"></a><span class="lineno"> 3793</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3892 | <div class="line"><a name="l03794"></a><span class="lineno"> 3794</span>  <span class="keyword">nullptr</span>,</div> |
| 3893 | <div class="line"><a name="l03795"></a><span class="lineno"> 3795</span>  <span class="keyword">true</span>);</div> |
| 3894 | <div class="line"><a name="l03796"></a><span class="lineno"> 3796</span>  </div> |
| 3895 | <div class="line"><a name="l03797"></a><span class="lineno"> 3797</span>  <span class="comment">// 05: The recurrent-to-input weights: Optional. A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3896 | <div class="line"><a name="l03798"></a><span class="lineno"> 3798</span>  <span class="comment">// [num_units, output_size], where “output_size” corresponds to either the number of cell units (i.e.,</span></div> |
| 3897 | <div class="line"><a name="l03799"></a><span class="lineno"> 3799</span>  <span class="comment">// “num_units”), or the second dimension of the “projection_weights”, if defined.</span></div> |
| 3898 | <div class="line"><a name="l03800"></a><span class="lineno"> 3800</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToInputWeightsPin =</div> |
| 3899 | <div class="line"><a name="l03801"></a><span class="lineno"> 3801</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3900 | <div class="line"><a name="l03802"></a><span class="lineno"> 3802</span>  5,</div> |
| 3901 | <div class="line"><a name="l03803"></a><span class="lineno"> 3803</span>  model,</div> |
| 3902 | <div class="line"><a name="l03804"></a><span class="lineno"> 3804</span>  data,</div> |
| 3903 | <div class="line"><a name="l03805"></a><span class="lineno"> 3805</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3904 | <div class="line"><a name="l03806"></a><span class="lineno"> 3806</span>  <span class="keyword">nullptr</span>,</div> |
| 3905 | <div class="line"><a name="l03807"></a><span class="lineno"> 3807</span>  <span class="keyword">true</span>);</div> |
| 3906 | <div class="line"><a name="l03808"></a><span class="lineno"> 3808</span>  </div> |
| 3907 | <div class="line"><a name="l03809"></a><span class="lineno"> 3809</span>  <span class="comment">// 09: The cell-to-input weights: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_QUANT16_SYMM, of shape</span></div> |
| 3908 | <div class="line"><a name="l03810"></a><span class="lineno"> 3810</span>  <span class="comment">// [num_units].</span></div> |
| 3909 | <div class="line"><a name="l03811"></a><span class="lineno"> 3811</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellToInputWeightsPin =</div> |
| 3910 | <div class="line"><a name="l03812"></a><span class="lineno"> 3812</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3911 | <div class="line"><a name="l03813"></a><span class="lineno"> 3813</span>  9,</div> |
| 3912 | <div class="line"><a name="l03814"></a><span class="lineno"> 3814</span>  model,</div> |
| 3913 | <div class="line"><a name="l03815"></a><span class="lineno"> 3815</span>  data,</div> |
| 3914 | <div class="line"><a name="l03816"></a><span class="lineno"> 3816</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3915 | <div class="line"><a name="l03817"></a><span class="lineno"> 3817</span>  <span class="keyword">nullptr</span>,</div> |
| 3916 | <div class="line"><a name="l03818"></a><span class="lineno"> 3818</span>  <span class="keyword">true</span>);</div> |
| 3917 | <div class="line"><a name="l03819"></a><span class="lineno"> 3819</span>  </div> |
| 3918 | <div class="line"><a name="l03820"></a><span class="lineno"> 3820</span>  <span class="comment">// 10: The cell-to-forget weights: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_QUANT16_SYMM, of shape</span></div> |
| 3919 | <div class="line"><a name="l03821"></a><span class="lineno"> 3821</span>  <span class="comment">// [num_units].</span></div> |
| 3920 | <div class="line"><a name="l03822"></a><span class="lineno"> 3822</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellToForgetWeightsPin =</div> |
| 3921 | <div class="line"><a name="l03823"></a><span class="lineno"> 3823</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3922 | <div class="line"><a name="l03824"></a><span class="lineno"> 3824</span>  10,</div> |
| 3923 | <div class="line"><a name="l03825"></a><span class="lineno"> 3825</span>  model,</div> |
| 3924 | <div class="line"><a name="l03826"></a><span class="lineno"> 3826</span>  data,</div> |
| 3925 | <div class="line"><a name="l03827"></a><span class="lineno"> 3827</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3926 | <div class="line"><a name="l03828"></a><span class="lineno"> 3828</span>  <span class="keyword">nullptr</span>,</div> |
| 3927 | <div class="line"><a name="l03829"></a><span class="lineno"> 3829</span>  <span class="keyword">true</span>);</div> |
| 3928 | <div class="line"><a name="l03830"></a><span class="lineno"> 3830</span>  </div> |
| 3929 | <div class="line"><a name="l03831"></a><span class="lineno"> 3831</span>  <span class="comment">// 11: The cell-to-output weights: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_QUANT16_SYMM, of shape</span></div> |
| 3930 | <div class="line"><a name="l03832"></a><span class="lineno"> 3832</span>  <span class="comment">// [num_units].</span></div> |
| 3931 | <div class="line"><a name="l03833"></a><span class="lineno"> 3833</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellToOutputWeightsPin =</div> |
| 3932 | <div class="line"><a name="l03834"></a><span class="lineno"> 3834</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3933 | <div class="line"><a name="l03835"></a><span class="lineno"> 3835</span>  11,</div> |
| 3934 | <div class="line"><a name="l03836"></a><span class="lineno"> 3836</span>  model,</div> |
| 3935 | <div class="line"><a name="l03837"></a><span class="lineno"> 3837</span>  data,</div> |
| 3936 | <div class="line"><a name="l03838"></a><span class="lineno"> 3838</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3937 | <div class="line"><a name="l03839"></a><span class="lineno"> 3839</span>  <span class="keyword">nullptr</span>,</div> |
| 3938 | <div class="line"><a name="l03840"></a><span class="lineno"> 3840</span>  <span class="keyword">true</span>);</div> |
| 3939 | <div class="line"><a name="l03841"></a><span class="lineno"> 3841</span>  </div> |
| 3940 | <div class="line"><a name="l03842"></a><span class="lineno"> 3842</span>  <span class="comment">// 12: The input gate bias: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_INT32, of shape [num_units].</span></div> |
| 3941 | <div class="line"><a name="l03843"></a><span class="lineno"> 3843</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputGateBiasPin =</div> |
| 3942 | <div class="line"><a name="l03844"></a><span class="lineno"> 3844</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3943 | <div class="line"><a name="l03845"></a><span class="lineno"> 3845</span>  12,</div> |
| 3944 | <div class="line"><a name="l03846"></a><span class="lineno"> 3846</span>  model,</div> |
| 3945 | <div class="line"><a name="l03847"></a><span class="lineno"> 3847</span>  data,</div> |
| 3946 | <div class="line"><a name="l03848"></a><span class="lineno"> 3848</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3947 | <div class="line"><a name="l03849"></a><span class="lineno"> 3849</span>  <span class="keyword">nullptr</span>,</div> |
| 3948 | <div class="line"><a name="l03850"></a><span class="lineno"> 3850</span>  <span class="keyword">true</span>);</div> |
| 3949 | <div class="line"><a name="l03851"></a><span class="lineno"> 3851</span>  </div> |
| 3950 | <div class="line"><a name="l03852"></a><span class="lineno"> 3852</span>  <span class="comment">// 16: The projection weights: Optional. A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_SYMM, of shape</span></div> |
| 3951 | <div class="line"><a name="l03853"></a><span class="lineno"> 3853</span>  <span class="comment">// [output_size, num_units].</span></div> |
| 3952 | <div class="line"><a name="l03854"></a><span class="lineno"> 3854</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> projectionWeightsPin =</div> |
| 3953 | <div class="line"><a name="l03855"></a><span class="lineno"> 3855</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3954 | <div class="line"><a name="l03856"></a><span class="lineno"> 3856</span>  16,</div> |
| 3955 | <div class="line"><a name="l03857"></a><span class="lineno"> 3857</span>  model,</div> |
| 3956 | <div class="line"><a name="l03858"></a><span class="lineno"> 3858</span>  data,</div> |
| 3957 | <div class="line"><a name="l03859"></a><span class="lineno"> 3859</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3958 | <div class="line"><a name="l03860"></a><span class="lineno"> 3860</span>  <span class="keyword">nullptr</span>,</div> |
| 3959 | <div class="line"><a name="l03861"></a><span class="lineno"> 3861</span>  <span class="keyword">true</span>);</div> |
| 3960 | <div class="line"><a name="l03862"></a><span class="lineno"> 3862</span>  </div> |
| 3961 | <div class="line"><a name="l03863"></a><span class="lineno"> 3863</span>  <span class="comment">// 17: The projection bias: Optional. A 1-D tensor of ANEURALNETWORKS_TENSOR_INT32, of shape [output_size].</span></div> |
| 3962 | <div class="line"><a name="l03864"></a><span class="lineno"> 3864</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> projectionBiasPin =</div> |
| 3963 | <div class="line"><a name="l03865"></a><span class="lineno"> 3865</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3964 | <div class="line"><a name="l03866"></a><span class="lineno"> 3866</span>  17,</div> |
| 3965 | <div class="line"><a name="l03867"></a><span class="lineno"> 3867</span>  model,</div> |
| 3966 | <div class="line"><a name="l03868"></a><span class="lineno"> 3868</span>  data,</div> |
| 3967 | <div class="line"><a name="l03869"></a><span class="lineno"> 3869</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3968 | <div class="line"><a name="l03870"></a><span class="lineno"> 3870</span>  <span class="keyword">nullptr</span>,</div> |
| 3969 | <div class="line"><a name="l03871"></a><span class="lineno"> 3871</span>  <span class="keyword">true</span>);</div> |
| 3970 | <div class="line"><a name="l03872"></a><span class="lineno"> 3872</span>  </div> |
| 3971 | <div class="line"><a name="l03873"></a><span class="lineno"> 3873</span>  <span class="keywordflow">if</span> ((!inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3972 | <div class="line"><a name="l03874"></a><span class="lineno"> 3874</span>  || (!recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3973 | <div class="line"><a name="l03875"></a><span class="lineno"> 3875</span>  || (!cellToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3974 | <div class="line"><a name="l03876"></a><span class="lineno"> 3876</span>  || (!cellToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3975 | <div class="line"><a name="l03877"></a><span class="lineno"> 3877</span>  || (!cellToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3976 | <div class="line"><a name="l03878"></a><span class="lineno"> 3878</span>  || (!inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3977 | <div class="line"><a name="l03879"></a><span class="lineno"> 3879</span>  || (!projectionWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !projectionWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 3978 | <div class="line"><a name="l03880"></a><span class="lineno"> 3880</span>  || (!projectionBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !projectionBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()))</div> |
| 3979 | <div class="line"><a name="l03881"></a><span class="lineno"> 3881</span>  {</div> |
| 3980 | <div class="line"><a name="l03882"></a><span class="lineno"> 3882</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid tensor inputs"</span>, __func__);</div> |
| 3981 | <div class="line"><a name="l03883"></a><span class="lineno"> 3883</span>  }</div> |
| 3982 | <div class="line"><a name="l03884"></a><span class="lineno"> 3884</span>  </div> |
| 3983 | <div class="line"><a name="l03885"></a><span class="lineno"> 3885</span>  </div> |
| 3984 | <div class="line"><a name="l03886"></a><span class="lineno"> 3886</span>  <span class="comment">// Get the optional normalization tensors</span></div> |
| 3985 | <div class="line"><a name="l03887"></a><span class="lineno"> 3887</span>  </div> |
| 3986 | <div class="line"><a name="l03888"></a><span class="lineno"> 3888</span>  <span class="comment">// 20: The input layer normalization weights. A 1-D tensor of shape [num_units] ANEURALNETWORKS_TENSOR_QUANT16_SYMM.</span></div> |
| 3987 | <div class="line"><a name="l03889"></a><span class="lineno"> 3889</span>  <span class="comment">// Used to rescale normalized inputs to activation at input gate.</span></div> |
| 3988 | <div class="line"><a name="l03890"></a><span class="lineno"> 3890</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputLayerNormWeightsPin =</div> |
| 3989 | <div class="line"><a name="l03891"></a><span class="lineno"> 3891</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 3990 | <div class="line"><a name="l03892"></a><span class="lineno"> 3892</span>  20,</div> |
| 3991 | <div class="line"><a name="l03893"></a><span class="lineno"> 3893</span>  model,</div> |
| 3992 | <div class="line"><a name="l03894"></a><span class="lineno"> 3894</span>  data,</div> |
| 3993 | <div class="line"><a name="l03895"></a><span class="lineno"> 3895</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 3994 | <div class="line"><a name="l03896"></a><span class="lineno"> 3896</span>  <span class="keyword">nullptr</span>,</div> |
| 3995 | <div class="line"><a name="l03897"></a><span class="lineno"> 3897</span>  <span class="keyword">true</span>);</div> |
| 3996 | <div class="line"><a name="l03898"></a><span class="lineno"> 3898</span>  </div> |
| 3997 | <div class="line"><a name="l03899"></a><span class="lineno"> 3899</span>  <span class="comment">// 21: The forget layer normalization weights. A 1-D tensor of shape [num_units] ANEURALNETWORKS_TENSOR_QUANT16_SYMM</span></div> |
| 3998 | <div class="line"><a name="l03900"></a><span class="lineno"> 3900</span>  <span class="comment">// Used to rescale normalized inputs to activation at forget gate.</span></div> |
| 3999 | <div class="line"><a name="l03901"></a><span class="lineno"> 3901</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> forgetLayerNormWeightsPin =</div> |
| 4000 | <div class="line"><a name="l03902"></a><span class="lineno"> 3902</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 4001 | <div class="line"><a name="l03903"></a><span class="lineno"> 3903</span>  21,</div> |
| 4002 | <div class="line"><a name="l03904"></a><span class="lineno"> 3904</span>  model,</div> |
| 4003 | <div class="line"><a name="l03905"></a><span class="lineno"> 3905</span>  data,</div> |
| 4004 | <div class="line"><a name="l03906"></a><span class="lineno"> 3906</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 4005 | <div class="line"><a name="l03907"></a><span class="lineno"> 3907</span>  <span class="keyword">nullptr</span>,</div> |
| 4006 | <div class="line"><a name="l03908"></a><span class="lineno"> 3908</span>  <span class="keyword">true</span>);</div> |
| 4007 | <div class="line"><a name="l03909"></a><span class="lineno"> 3909</span>  </div> |
| 4008 | <div class="line"><a name="l03910"></a><span class="lineno"> 3910</span>  <span class="comment">// 22: The cell layer normalization weights. A 1-D tensor of shape [num_units] ANEURALNETWORKS_TENSOR_QUANT16_SYMM.</span></div> |
| 4009 | <div class="line"><a name="l03911"></a><span class="lineno"> 3911</span>  <span class="comment">// Used to rescale normalized inputs to activation at cell gate.</span></div> |
| 4010 | <div class="line"><a name="l03912"></a><span class="lineno"> 3912</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellLayerNormWeightsPin =</div> |
| 4011 | <div class="line"><a name="l03913"></a><span class="lineno"> 3913</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 4012 | <div class="line"><a name="l03914"></a><span class="lineno"> 3914</span>  22,</div> |
| 4013 | <div class="line"><a name="l03915"></a><span class="lineno"> 3915</span>  model,</div> |
| 4014 | <div class="line"><a name="l03916"></a><span class="lineno"> 3916</span>  data,</div> |
| 4015 | <div class="line"><a name="l03917"></a><span class="lineno"> 3917</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 4016 | <div class="line"><a name="l03918"></a><span class="lineno"> 3918</span>  <span class="keyword">nullptr</span>,</div> |
| 4017 | <div class="line"><a name="l03919"></a><span class="lineno"> 3919</span>  <span class="keyword">true</span>);</div> |
| 4018 | <div class="line"><a name="l03920"></a><span class="lineno"> 3920</span>  </div> |
| 4019 | <div class="line"><a name="l03921"></a><span class="lineno"> 3921</span>  <span class="comment">// 23: The output layer normalization weights. A 1-D tensor of shape [num_units].</span></div> |
| 4020 | <div class="line"><a name="l03922"></a><span class="lineno"> 3922</span>  <span class="comment">// Used to rescale normalized inputs to activation at output gate.</span></div> |
| 4021 | <div class="line"><a name="l03923"></a><span class="lineno"> 3923</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> outputLayerNormWeightsPin =</div> |
| 4022 | <div class="line"><a name="l03924"></a><span class="lineno"> 3924</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| 4023 | <div class="line"><a name="l03925"></a><span class="lineno"> 3925</span>  23,</div> |
| 4024 | <div class="line"><a name="l03926"></a><span class="lineno"> 3926</span>  model,</div> |
| 4025 | <div class="line"><a name="l03927"></a><span class="lineno"> 3927</span>  data,</div> |
| 4026 | <div class="line"><a name="l03928"></a><span class="lineno"> 3928</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| 4027 | <div class="line"><a name="l03929"></a><span class="lineno"> 3929</span>  <span class="keyword">nullptr</span>,</div> |
| 4028 | <div class="line"><a name="l03930"></a><span class="lineno"> 3930</span>  <span class="keyword">true</span>);</div> |
| 4029 | <div class="line"><a name="l03931"></a><span class="lineno"> 3931</span>  </div> |
| 4030 | <div class="line"><a name="l03932"></a><span class="lineno"> 3932</span>  <span class="keywordflow">if</span> ((!inputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !inputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 4031 | <div class="line"><a name="l03933"></a><span class="lineno"> 3933</span>  || (!forgetLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !forgetLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 4032 | <div class="line"><a name="l03934"></a><span class="lineno"> 3934</span>  || (!cellLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !cellLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>())</div> |
| 4033 | <div class="line"><a name="l03935"></a><span class="lineno"> 3935</span>  || (!outputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() && !outputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">IsOptional</a>()))</div> |
| 4034 | <div class="line"><a name="l03936"></a><span class="lineno"> 3936</span>  {</div> |
| 4035 | <div class="line"><a name="l03937"></a><span class="lineno"> 3937</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid tensor inputs"</span>, __func__);</div> |
| 4036 | <div class="line"><a name="l03938"></a><span class="lineno"> 3938</span>  }</div> |
| 4037 | <div class="line"><a name="l03939"></a><span class="lineno"> 3939</span>  </div> |
| 4038 | <div class="line"><a name="l03940"></a><span class="lineno"> 3940</span>  <span class="comment">// Get the optional input scalars:</span></div> |
| 4039 | <div class="line"><a name="l03941"></a><span class="lineno"> 3941</span>  <span class="comment">// 24: The cell clip: If provided the cell state is clipped by this value prior to the cell output activation.</span></div> |
| 4040 | <div class="line"><a name="l03942"></a><span class="lineno"> 3942</span>  <span class="comment">// 25: The projection clip: If provided and projection is enabled, this is used for clipping the projected values.</span></div> |
| 4041 | <div class="line"><a name="l03943"></a><span class="lineno"> 3943</span>  </div> |
| 4042 | <div class="line"><a name="l03944"></a><span class="lineno"> 3944</span>  <span class="comment">// Get the mandatory input scalars:</span></div> |
| 4043 | <div class="line"><a name="l03945"></a><span class="lineno"> 3945</span>  <span class="comment">// 26: The scale of the intermediate result of matmul, i.e. input to layer normalization, at input gate.</span></div> |
| 4044 | <div class="line"><a name="l03946"></a><span class="lineno"> 3946</span>  <span class="comment">// 27: The scale of the intermediate result of matmul, i.e. input to layer normalization, at forget gate.</span></div> |
| 4045 | <div class="line"><a name="l03947"></a><span class="lineno"> 3947</span>  <span class="comment">// 28: The scale of the intermediate result of matmul, i.e. input to layer normalization, at cell gate.</span></div> |
| 4046 | <div class="line"><a name="l03948"></a><span class="lineno"> 3948</span>  <span class="comment">// 29: The scale of the intermediate result of matmul, i.e. input to layer normalization, at output gate.</span></div> |
| 4047 | <div class="line"><a name="l03949"></a><span class="lineno"> 3949</span>  <span class="comment">// 30: The zero point of the hidden state, i.e. input to projection.</span></div> |
| 4048 | <div class="line"><a name="l03950"></a><span class="lineno"> 3950</span>  <span class="comment">// 31: The scale of the hidden state, i.e. input to projection.</span></div> |
| 4049 | <div class="line"><a name="l03951"></a><span class="lineno"> 3951</span>  <span class="keywordtype">float</span> cellClip, projClip, matMulInputGate, matMulForgetGate, matMulCellGate, matMulOutputGate, projInputScale;</div> |
| 4050 | <div class="line"><a name="l03952"></a><span class="lineno"> 3952</span>  <span class="keywordtype">int</span> projInputZeroPoint;</div> |
| 4051 | <div class="line"><a name="l03953"></a><span class="lineno"> 3953</span>  </div> |
| 4052 | <div class="line"><a name="l03954"></a><span class="lineno"> 3954</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 24, OperandType::FLOAT32, cellClip, model, data, <span class="keyword">true</span>) ||</div> |
| 4053 | <div class="line"><a name="l03955"></a><span class="lineno"> 3955</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 25, OperandType::FLOAT32, projClip, model, data, <span class="keyword">true</span>) ||</div> |
| 4054 | <div class="line"><a name="l03956"></a><span class="lineno"> 3956</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 26, OperandType::FLOAT32, matMulInputGate, model, data) ||</div> |
| 4055 | <div class="line"><a name="l03957"></a><span class="lineno"> 3957</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 27, OperandType::FLOAT32, matMulForgetGate, model, data) ||</div> |
| 4056 | <div class="line"><a name="l03958"></a><span class="lineno"> 3958</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 28, OperandType::FLOAT32, matMulCellGate, model, data) ||</div> |
| 4057 | <div class="line"><a name="l03959"></a><span class="lineno"> 3959</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 29, OperandType::FLOAT32, matMulOutputGate, model, data) ||</div> |
| 4058 | <div class="line"><a name="l03960"></a><span class="lineno"> 3960</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 30, OperandType::INT32, projInputZeroPoint, model, data) ||</div> |
| 4059 | <div class="line"><a name="l03961"></a><span class="lineno"> 3961</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 31, OperandType::FLOAT32, projInputScale, model, data))</div> |
| 4060 | <div class="line"><a name="l03962"></a><span class="lineno"> 3962</span>  {</div> |
| 4061 | <div class="line"><a name="l03963"></a><span class="lineno"> 3963</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid scalar inputs"</span>, __func__);</div> |
| 4062 | <div class="line"><a name="l03964"></a><span class="lineno"> 3964</span>  }</div> |
| 4063 | <div class="line"><a name="l03965"></a><span class="lineno"> 3965</span>  </div> |
| 4064 | <div class="line"><a name="l03966"></a><span class="lineno"> 3966</span>  <span class="comment">// Outputs:</span></div> |
| 4065 | <div class="line"><a name="l03967"></a><span class="lineno"> 3967</span>  <span class="comment">// 0: The output state (out): A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_ASYMM_SIGNED, of shape [batch_size,</span></div> |
| 4066 | <div class="line"><a name="l03968"></a><span class="lineno"> 3968</span>  <span class="comment">// output_size].</span></div> |
| 4067 | <div class="line"><a name="l03969"></a><span class="lineno"> 3969</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputStateOut = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4068 | <div class="line"><a name="l03970"></a><span class="lineno"> 3970</span>  <span class="keywordflow">if</span> (!outputStateOut)</div> |
| 4069 | <div class="line"><a name="l03971"></a><span class="lineno"> 3971</span>  {</div> |
| 4070 | <div class="line"><a name="l03972"></a><span class="lineno"> 3972</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0: outputStateOut"</span>, __func__);</div> |
| 4071 | <div class="line"><a name="l03973"></a><span class="lineno"> 3973</span>  }</div> |
| 4072 | <div class="line"><a name="l03974"></a><span class="lineno"> 3974</span>  </div> |
| 4073 | <div class="line"><a name="l03975"></a><span class="lineno"> 3975</span>  <span class="comment">// 1: The cell state (out): A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT16_SYMM, of shape [batch_size, num_units].</span></div> |
| 4074 | <div class="line"><a name="l03976"></a><span class="lineno"> 3976</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* cellStateOut = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 1, model);</div> |
| 4075 | <div class="line"><a name="l03977"></a><span class="lineno"> 3977</span>  <span class="keywordflow">if</span> (!cellStateOut)</div> |
| 4076 | <div class="line"><a name="l03978"></a><span class="lineno"> 3978</span>  {</div> |
| 4077 | <div class="line"><a name="l03979"></a><span class="lineno"> 3979</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 1: cellStateOut"</span>, __func__);</div> |
| 4078 | <div class="line"><a name="l03980"></a><span class="lineno"> 3980</span>  }</div> |
| 4079 | <div class="line"><a name="l03981"></a><span class="lineno"> 3981</span>  </div> |
| 4080 | <div class="line"><a name="l03982"></a><span class="lineno"> 3982</span>  <span class="comment">// 2: The output: A 2-D tensor of ANEURALNETWORKS_TENSOR_QUANT8_ASYMM_SIGNED, of shape [batch_size, output_size].</span></div> |
| 4081 | <div class="line"><a name="l03983"></a><span class="lineno"> 3983</span>  <span class="comment">// This is effectively the same as the current “output state (out)” value.</span></div> |
| 4082 | <div class="line"><a name="l03984"></a><span class="lineno"> 3984</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 2, model);</div> |
| 4083 | <div class="line"><a name="l03985"></a><span class="lineno"> 3985</span>  <span class="keywordflow">if</span> (!output)</div> |
| 4084 | <div class="line"><a name="l03986"></a><span class="lineno"> 3986</span>  {</div> |
| 4085 | <div class="line"><a name="l03987"></a><span class="lineno"> 3987</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 2: output"</span>, __func__);</div> |
| 4086 | <div class="line"><a name="l03988"></a><span class="lineno"> 3988</span>  }</div> |
| 4087 | <div class="line"><a name="l03989"></a><span class="lineno"> 3989</span>  </div> |
| 4088 | <div class="line"><a name="l03990"></a><span class="lineno"> 3990</span>  <span class="comment">// set the params structure for the AddLstmLayer call</span></div> |
| 4089 | <div class="line"><a name="l03991"></a><span class="lineno"> 3991</span>  <a class="code" href="structarmnn_1_1_lstm_input_params.html">LstmInputParams</a> params;</div> |
| 4090 | <div class="line"><a name="l03992"></a><span class="lineno"> 3992</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> = inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4091 | <div class="line"><a name="l03993"></a><span class="lineno"> 3993</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">m_InputToForgetWeights</a> = inputToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4092 | <div class="line"><a name="l03994"></a><span class="lineno"> 3994</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">m_InputToCellWeights</a> = inputToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4093 | <div class="line"><a name="l03995"></a><span class="lineno"> 3995</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">m_InputToOutputWeights</a> = inputToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4094 | <div class="line"><a name="l03996"></a><span class="lineno"> 3996</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> = recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4095 | <div class="line"><a name="l03997"></a><span class="lineno"> 3997</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">m_RecurrentToForgetWeights</a> = recurrentToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4096 | <div class="line"><a name="l03998"></a><span class="lineno"> 3998</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">m_RecurrentToCellWeights</a> = recurrentToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4097 | <div class="line"><a name="l03999"></a><span class="lineno"> 3999</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">m_RecurrentToOutputWeights</a> = recurrentToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4098 | <div class="line"><a name="l04000"></a><span class="lineno"> 4000</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a> = cellToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4099 | <div class="line"><a name="l04001"></a><span class="lineno"> 4001</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a> = cellToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4100 | <div class="line"><a name="l04002"></a><span class="lineno"> 4002</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a> = cellToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4101 | <div class="line"><a name="l04003"></a><span class="lineno"> 4003</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> = inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4102 | <div class="line"><a name="l04004"></a><span class="lineno"> 4004</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">m_ForgetGateBias</a> = forgetGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4103 | <div class="line"><a name="l04005"></a><span class="lineno"> 4005</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">m_CellBias</a> = cellBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4104 | <div class="line"><a name="l04006"></a><span class="lineno"> 4006</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">m_OutputGateBias</a> = outputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4105 | <div class="line"><a name="l04007"></a><span class="lineno"> 4007</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">m_ProjectionWeights</a> = projectionWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4106 | <div class="line"><a name="l04008"></a><span class="lineno"> 4008</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a> = projectionBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4107 | <div class="line"><a name="l04009"></a><span class="lineno"> 4009</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a> = inputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4108 | <div class="line"><a name="l04010"></a><span class="lineno"> 4010</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a> = forgetLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4109 | <div class="line"><a name="l04011"></a><span class="lineno"> 4011</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a> = cellLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4110 | <div class="line"><a name="l04012"></a><span class="lineno"> 4012</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a> = outputLayerNormWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4111 | <div class="line"><a name="l04013"></a><span class="lineno"> 4013</span>  </div> |
| 4112 | <div class="line"><a name="l04014"></a><span class="lineno"> 4014</span>  <span class="comment">// set the layer descriptor</span></div> |
| 4113 | <div class="line"><a name="l04015"></a><span class="lineno"> 4015</span>  <a class="code" href="structarmnn_1_1_q_lstm_descriptor.html">QLstmDescriptor</a> desc;</div> |
| 4114 | <div class="line"><a name="l04016"></a><span class="lineno"> 4016</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ac81fb0e66dc623dc37c77f219f53a6d3">m_CellClip</a> = cellClip;</div> |
| 4115 | <div class="line"><a name="l04017"></a><span class="lineno"> 4017</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#aa6a518b65088f34803b3214334bdff61">m_ProjectionClip</a> = projClip;</div> |
| 4116 | <div class="line"><a name="l04018"></a><span class="lineno"> 4018</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4117 | <div class="line"><a name="l04019"></a><span class="lineno"> 4019</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4118 | <div class="line"><a name="l04020"></a><span class="lineno"> 4020</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> == <span class="keyword">nullptr</span>);</div> |
| 4119 | <div class="line"><a name="l04021"></a><span class="lineno"> 4021</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 4120 | <div class="line"><a name="l04022"></a><span class="lineno"> 4022</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a> != <span class="keyword">nullptr</span>);</div> |
| 4121 | <div class="line"><a name="l04023"></a><span class="lineno"> 4023</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">m_ProjectionEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">m_ProjectionWeights</a> != <span class="keyword">nullptr</span>);</div> |
| 4122 | <div class="line"><a name="l04024"></a><span class="lineno"> 4024</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">m_LayerNormEnabled</a> = (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 4123 | <div class="line"><a name="l04025"></a><span class="lineno"> 4025</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 4124 | <div class="line"><a name="l04026"></a><span class="lineno"> 4026</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 4125 | <div class="line"><a name="l04027"></a><span class="lineno"> 4027</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a> != <span class="keyword">nullptr</span>);</div> |
| 4126 | <div class="line"><a name="l04028"></a><span class="lineno"> 4028</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a09e1f097944f61cc901240f9300364cf">m_InputIntermediateScale</a> = matMulInputGate;</div> |
| 4127 | <div class="line"><a name="l04029"></a><span class="lineno"> 4029</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#afec7f36158448f723b426a9527acb189">m_ForgetIntermediateScale</a> = matMulForgetGate;</div> |
| 4128 | <div class="line"><a name="l04030"></a><span class="lineno"> 4030</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a0477ee1b44ace6090119178eea78cb0b">m_CellIntermediateScale</a> = matMulCellGate;</div> |
| 4129 | <div class="line"><a name="l04031"></a><span class="lineno"> 4031</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#aa43409f9b457352c95c89f20ce5d844d">m_OutputIntermediateScale</a> = matMulOutputGate;</div> |
| 4130 | <div class="line"><a name="l04032"></a><span class="lineno"> 4032</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#af8f724af7210b52529216feefa993c98">m_HiddenStateScale</a> = projInputScale;</div> |
| 4131 | <div class="line"><a name="l04033"></a><span class="lineno"> 4033</span>  desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a4556cbd764d4848d8ad0637a9eed580d">m_HiddenStateZeroPoint</a> = projInputZeroPoint;</div> |
| 4132 | <div class="line"><a name="l04034"></a><span class="lineno"> 4034</span>  </div> |
| 4133 | <div class="line"><a name="l04035"></a><span class="lineno"> 4035</span>  <span class="comment">// validate the optional input groups</span></div> |
| 4134 | <div class="line"><a name="l04036"></a><span class="lineno"> 4036</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> &&</div> |
| 4135 | <div class="line"><a name="l04037"></a><span class="lineno"> 4037</span>  (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 4136 | <div class="line"><a name="l04038"></a><span class="lineno"> 4038</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> != <span class="keyword">nullptr</span> ||</div> |
| 4137 | <div class="line"><a name="l04039"></a><span class="lineno"> 4039</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> != <span class="keyword">nullptr</span>))</div> |
| 4138 | <div class="line"><a name="l04040"></a><span class="lineno"> 4040</span>  {</div> |
| 4139 | <div class="line"><a name="l04041"></a><span class="lineno"> 4041</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: All, or none, of input-to-input weights, recurrent-to-input weights,"</span></div> |
| 4140 | <div class="line"><a name="l04042"></a><span class="lineno"> 4042</span>  <span class="stringliteral">" and input gate bias must be provided"</span>, __func__);</div> |
| 4141 | <div class="line"><a name="l04043"></a><span class="lineno"> 4043</span>  }</div> |
| 4142 | <div class="line"><a name="l04044"></a><span class="lineno"> 4044</span>  </div> |
| 4143 | <div class="line"><a name="l04045"></a><span class="lineno"> 4045</span>  <span class="keywordflow">if</span> (!desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">m_ProjectionEnabled</a> && params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a> != <span class="keyword">nullptr</span>)</div> |
| 4144 | <div class="line"><a name="l04046"></a><span class="lineno"> 4046</span>  {</div> |
| 4145 | <div class="line"><a name="l04047"></a><span class="lineno"> 4047</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: projection bias should not be provided without projection weights"</span>, __func__);</div> |
| 4146 | <div class="line"><a name="l04048"></a><span class="lineno"> 4048</span>  }</div> |
| 4147 | <div class="line"><a name="l04049"></a><span class="lineno"> 4049</span>  </div> |
| 4148 | <div class="line"><a name="l04050"></a><span class="lineno"> 4050</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a> &&</div> |
| 4149 | <div class="line"><a name="l04051"></a><span class="lineno"> 4051</span>  (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4150 | <div class="line"><a name="l04052"></a><span class="lineno"> 4052</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4151 | <div class="line"><a name="l04053"></a><span class="lineno"> 4053</span>  (!desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> && params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a> == <span class="keyword">nullptr</span>)))</div> |
| 4152 | <div class="line"><a name="l04054"></a><span class="lineno"> 4054</span>  {</div> |
| 4153 | <div class="line"><a name="l04055"></a><span class="lineno"> 4055</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: All, or none, of cell-to-forget weights and cell-to-output weights must be provided"</span></div> |
| 4154 | <div class="line"><a name="l04056"></a><span class="lineno"> 4056</span>  <span class="stringliteral">" and, if CIFG is not enabled, cell-to-input weights must also be provided"</span>, __func__);</div> |
| 4155 | <div class="line"><a name="l04057"></a><span class="lineno"> 4057</span>  }</div> |
| 4156 | <div class="line"><a name="l04058"></a><span class="lineno"> 4058</span>  </div> |
| 4157 | <div class="line"><a name="l04059"></a><span class="lineno"> 4059</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">m_LayerNormEnabled</a> &&</div> |
| 4158 | <div class="line"><a name="l04060"></a><span class="lineno"> 4060</span>  (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4159 | <div class="line"><a name="l04061"></a><span class="lineno"> 4061</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4160 | <div class="line"><a name="l04062"></a><span class="lineno"> 4062</span>  params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a> == <span class="keyword">nullptr</span> ||</div> |
| 4161 | <div class="line"><a name="l04063"></a><span class="lineno"> 4063</span>  (!desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a> && params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a> == <span class="keyword">nullptr</span>)))</div> |
| 4162 | <div class="line"><a name="l04064"></a><span class="lineno"> 4064</span>  {</div> |
| 4163 | <div class="line"><a name="l04065"></a><span class="lineno"> 4065</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: All, or none, of forget-norm weights, cell-norm weights and output-norm weights must be"</span></div> |
| 4164 | <div class="line"><a name="l04066"></a><span class="lineno"> 4066</span>  <span class="stringliteral">" provided and, if CIFG is not enabled, input-norm weights must also be provided"</span>, __func__);</div> |
| 4165 | <div class="line"><a name="l04067"></a><span class="lineno"> 4067</span>  }</div> |
| 4166 | <div class="line"><a name="l04068"></a><span class="lineno"> 4068</span>  </div> |
| 4167 | <div class="line"><a name="l04069"></a><span class="lineno"> 4069</span>  <span class="comment">// Basic parameters</span></div> |
| 4168 | <div class="line"><a name="l04070"></a><span class="lineno"> 4070</span>  <a class="code" href="structarmnn_1_1_lstm_input_params_info.html">LstmInputParamsInfo</a> paramsInfo;</div> |
| 4169 | <div class="line"><a name="l04071"></a><span class="lineno"> 4071</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a4b3f6e6f2268416ffd7a34fda95ffd0b">m_InputToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">m_InputToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4170 | <div class="line"><a name="l04072"></a><span class="lineno"> 4072</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#af07c65d3b7886a00f10c69093e76a341">m_InputToCellWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">m_InputToCellWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4171 | <div class="line"><a name="l04073"></a><span class="lineno"> 4073</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a33242cb43250f04f1736161ca8ad2db9">m_InputToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">m_InputToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4172 | <div class="line"><a name="l04074"></a><span class="lineno"> 4074</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a11b90bd2694aa7f4b434aadd75fbb1c2">m_RecurrentToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">m_RecurrentToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4173 | <div class="line"><a name="l04075"></a><span class="lineno"> 4075</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a6c7873a3320a59a63aa26c006db905c0">m_RecurrentToCellWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">m_RecurrentToCellWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4174 | <div class="line"><a name="l04076"></a><span class="lineno"> 4076</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#af434b509cd5232ef762c4b21b2dfaae0">m_RecurrentToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">m_RecurrentToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4175 | <div class="line"><a name="l04077"></a><span class="lineno"> 4077</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a21626d0b91f8942c3ef8e48e17db9f21">m_ForgetGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">m_ForgetGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4176 | <div class="line"><a name="l04078"></a><span class="lineno"> 4078</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ab173a067eeb7295d84f5327bcc05a6c1">m_CellBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">m_CellBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4177 | <div class="line"><a name="l04079"></a><span class="lineno"> 4079</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a48cb83dcb58c786ea36f5d37695e75b1">m_OutputGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">m_OutputGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4178 | <div class="line"><a name="l04080"></a><span class="lineno"> 4080</span>  </div> |
| 4179 | <div class="line"><a name="l04081"></a><span class="lineno"> 4081</span>  <span class="comment">// Inputs</span></div> |
| 4180 | <div class="line"><a name="l04082"></a><span class="lineno"> 4082</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4181 | <div class="line"><a name="l04083"></a><span class="lineno"> 4083</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputStatePrevTimeStepInfo = outputStatePrevTimeStep.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4182 | <div class="line"><a name="l04084"></a><span class="lineno"> 4084</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& cellStatePrevTimeStepInfo = cellStatePrevTimeStep.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4183 | <div class="line"><a name="l04085"></a><span class="lineno"> 4085</span>  </div> |
| 4184 | <div class="line"><a name="l04086"></a><span class="lineno"> 4086</span>  <span class="comment">// Outputs</span></div> |
| 4185 | <div class="line"><a name="l04087"></a><span class="lineno"> 4087</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> outputStateOutInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputStateOut);</div> |
| 4186 | <div class="line"><a name="l04088"></a><span class="lineno"> 4088</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 4187 | <div class="line"><a name="l04089"></a><span class="lineno"> 4089</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& cellStateOutInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*cellStateOut);</div> |
| 4188 | <div class="line"><a name="l04090"></a><span class="lineno"> 4090</span>  </div> |
| 4189 | <div class="line"><a name="l04091"></a><span class="lineno"> 4091</span>  <span class="comment">// Optional parameters</span></div> |
| 4190 | <div class="line"><a name="l04092"></a><span class="lineno"> 4092</span>  <span class="keywordflow">if</span> (!desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a>)</div> |
| 4191 | <div class="line"><a name="l04093"></a><span class="lineno"> 4093</span>  {</div> |
| 4192 | <div class="line"><a name="l04094"></a><span class="lineno"> 4094</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a15f9f65126ad3d4d82d6c19d0662ae01">m_InputToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4193 | <div class="line"><a name="l04095"></a><span class="lineno"> 4095</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a9ab864961ded72f5ce7ea4eb9921f826">m_RecurrentToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4194 | <div class="line"><a name="l04096"></a><span class="lineno"> 4096</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a>)</div> |
| 4195 | <div class="line"><a name="l04097"></a><span class="lineno"> 4097</span>  {</div> |
| 4196 | <div class="line"><a name="l04098"></a><span class="lineno"> 4098</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ab99b2297c5a17114926b8b47c693fe43">m_CellToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">m_CellToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4197 | <div class="line"><a name="l04099"></a><span class="lineno"> 4099</span>  }</div> |
| 4198 | <div class="line"><a name="l04100"></a><span class="lineno"> 4100</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a042ea9d6d3842b87f112db126e806a93">m_InputGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4199 | <div class="line"><a name="l04101"></a><span class="lineno"> 4101</span>  }</div> |
| 4200 | <div class="line"><a name="l04102"></a><span class="lineno"> 4102</span>  </div> |
| 4201 | <div class="line"><a name="l04103"></a><span class="lineno"> 4103</span>  </div> |
| 4202 | <div class="line"><a name="l04104"></a><span class="lineno"> 4104</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">m_ProjectionEnabled</a>)</div> |
| 4203 | <div class="line"><a name="l04105"></a><span class="lineno"> 4105</span>  {</div> |
| 4204 | <div class="line"><a name="l04106"></a><span class="lineno"> 4106</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ac6fb4a3a8e7e8a359ac34badc8178b1a">m_ProjectionWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">m_ProjectionWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4205 | <div class="line"><a name="l04107"></a><span class="lineno"> 4107</span>  <span class="keywordflow">if</span> (params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a> != <span class="keyword">nullptr</span>)</div> |
| 4206 | <div class="line"><a name="l04108"></a><span class="lineno"> 4108</span>  {</div> |
| 4207 | <div class="line"><a name="l04109"></a><span class="lineno"> 4109</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ae22fc962c59e7c24986718f5af0020db">m_ProjectionBias</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">m_ProjectionBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4208 | <div class="line"><a name="l04110"></a><span class="lineno"> 4110</span>  }</div> |
| 4209 | <div class="line"><a name="l04111"></a><span class="lineno"> 4111</span>  }</div> |
| 4210 | <div class="line"><a name="l04112"></a><span class="lineno"> 4112</span>  <span class="keywordflow">else</span></div> |
| 4211 | <div class="line"><a name="l04113"></a><span class="lineno"> 4113</span>  {</div> |
| 4212 | <div class="line"><a name="l04114"></a><span class="lineno"> 4114</span>  <span class="comment">// If Projection is disabled, override non-const outputs to change the quant info with hidden params, then</span></div> |
| 4213 | <div class="line"><a name="l04115"></a><span class="lineno"> 4115</span>  <span class="comment">// create a new const TensorInfo based on this</span></div> |
| 4214 | <div class="line"><a name="l04116"></a><span class="lineno"> 4116</span>  outputStateOutInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a685739c4eb65a580e075282cfe6787d6">SetQuantizationScale</a>(projInputScale);</div> |
| 4215 | <div class="line"><a name="l04117"></a><span class="lineno"> 4117</span>  outputStateOutInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a63cbc581012c957f9d68d224ddc3e43c">SetQuantizationOffset</a>(projInputZeroPoint);</div> |
| 4216 | <div class="line"><a name="l04118"></a><span class="lineno"> 4118</span>  outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a685739c4eb65a580e075282cfe6787d6">SetQuantizationScale</a>(projInputScale);</div> |
| 4217 | <div class="line"><a name="l04119"></a><span class="lineno"> 4119</span>  outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a63cbc581012c957f9d68d224ddc3e43c">SetQuantizationOffset</a>(projInputZeroPoint);</div> |
| 4218 | <div class="line"><a name="l04120"></a><span class="lineno"> 4120</span>  }</div> |
| 4219 | <div class="line"><a name="l04121"></a><span class="lineno"> 4121</span>  </div> |
| 4220 | <div class="line"><a name="l04122"></a><span class="lineno"> 4122</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> constOutputStateOutInfo(outputStateOutInfo);</div> |
| 4221 | <div class="line"><a name="l04123"></a><span class="lineno"> 4123</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> constOutputInfo(outputInfo);</div> |
| 4222 | <div class="line"><a name="l04124"></a><span class="lineno"> 4124</span>  </div> |
| 4223 | <div class="line"><a name="l04125"></a><span class="lineno"> 4125</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">m_PeepholeEnabled</a>)</div> |
| 4224 | <div class="line"><a name="l04126"></a><span class="lineno"> 4126</span>  {</div> |
| 4225 | <div class="line"><a name="l04127"></a><span class="lineno"> 4127</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a1ddc4634491d95d37fe15049decadb8c">m_CellToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">m_CellToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4226 | <div class="line"><a name="l04128"></a><span class="lineno"> 4128</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a16b4d4acd2a0d6d533ff3fe8b8e71765">m_CellToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">m_CellToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4227 | <div class="line"><a name="l04129"></a><span class="lineno"> 4129</span>  }</div> |
| 4228 | <div class="line"><a name="l04130"></a><span class="lineno"> 4130</span>  </div> |
| 4229 | <div class="line"><a name="l04131"></a><span class="lineno"> 4131</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">m_LayerNormEnabled</a>)</div> |
| 4230 | <div class="line"><a name="l04132"></a><span class="lineno"> 4132</span>  {</div> |
| 4231 | <div class="line"><a name="l04133"></a><span class="lineno"> 4133</span>  <span class="keywordflow">if</span>(!desc.<a class="code" href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">m_CifgEnabled</a>)</div> |
| 4232 | <div class="line"><a name="l04134"></a><span class="lineno"> 4134</span>  {</div> |
| 4233 | <div class="line"><a name="l04135"></a><span class="lineno"> 4135</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#ab8c7712c0c6bc427dede8c76945e8f89">m_InputLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">m_InputLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4234 | <div class="line"><a name="l04136"></a><span class="lineno"> 4136</span>  }</div> |
| 4235 | <div class="line"><a name="l04137"></a><span class="lineno"> 4137</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a513ad7774e0f8969ab68d91a04ecca65">m_ForgetLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">m_ForgetLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4236 | <div class="line"><a name="l04138"></a><span class="lineno"> 4138</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a567622530e51eaa2e82c9790a3a43f02">m_CellLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">m_CellLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4237 | <div class="line"><a name="l04139"></a><span class="lineno"> 4139</span>  paramsInfo.<a class="code" href="structarmnn_1_1_lstm_input_params_info.html#a9935ad8aff43cf1e0622bf6d82710513">m_OutputLayerNormWeights</a> = &(params.<a class="code" href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">m_OutputLayerNormWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4238 | <div class="line"><a name="l04140"></a><span class="lineno"> 4140</span>  }</div> |
| 4239 | <div class="line"><a name="l04141"></a><span class="lineno"> 4141</span>  </div> |
| 4240 | <div class="line"><a name="l04142"></a><span class="lineno"> 4142</span>  <span class="comment">// Check if the layer is supported</span></div> |
| 4241 | <div class="line"><a name="l04143"></a><span class="lineno"> 4143</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4242 | <div class="line"><a name="l04144"></a><span class="lineno"> 4144</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4243 | <div class="line"><a name="l04145"></a><span class="lineno"> 4145</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& cellStateOutInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 4244 | <div class="line"><a name="l04146"></a><span class="lineno"> 4146</span>  {</div> |
| 4245 | <div class="line"><a name="l04147"></a><span class="lineno"> 4147</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4246 | <div class="line"><a name="l04148"></a><span class="lineno"> 4148</span>  IsQLstmSupported,</div> |
| 4247 | <div class="line"><a name="l04149"></a><span class="lineno"> 4149</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4248 | <div class="line"><a name="l04150"></a><span class="lineno"> 4150</span>  isSupported,</div> |
| 4249 | <div class="line"><a name="l04151"></a><span class="lineno"> 4151</span>  setBackend,</div> |
| 4250 | <div class="line"><a name="l04152"></a><span class="lineno"> 4152</span>  inputInfo,</div> |
| 4251 | <div class="line"><a name="l04153"></a><span class="lineno"> 4153</span>  outputStatePrevTimeStepInfo,</div> |
| 4252 | <div class="line"><a name="l04154"></a><span class="lineno"> 4154</span>  cellStatePrevTimeStepInfo,</div> |
| 4253 | <div class="line"><a name="l04155"></a><span class="lineno"> 4155</span>  constOutputStateOutInfo,</div> |
| 4254 | <div class="line"><a name="l04156"></a><span class="lineno"> 4156</span>  cellStateOutInfo,</div> |
| 4255 | <div class="line"><a name="l04157"></a><span class="lineno"> 4157</span>  constOutputInfo,</div> |
| 4256 | <div class="line"><a name="l04158"></a><span class="lineno"> 4158</span>  desc,</div> |
| 4257 | <div class="line"><a name="l04159"></a><span class="lineno"> 4159</span>  paramsInfo);</div> |
| 4258 | <div class="line"><a name="l04160"></a><span class="lineno"> 4160</span>  };</div> |
| 4259 | <div class="line"><a name="l04161"></a><span class="lineno"> 4161</span>  </div> |
| 4260 | <div class="line"><a name="l04162"></a><span class="lineno"> 4162</span>  <span class="keywordtype">bool</span> isDynamic = <span class="keyword">false</span>;</div> |
| 4261 | <div class="line"><a name="l04163"></a><span class="lineno"> 4163</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(constOutputStateOutInfo) &&</div> |
| 4262 | <div class="line"><a name="l04164"></a><span class="lineno"> 4164</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(cellStateOutInfo) &&</div> |
| 4263 | <div class="line"><a name="l04165"></a><span class="lineno"> 4165</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(constOutputInfo))</div> |
| 4264 | <div class="line"><a name="l04166"></a><span class="lineno"> 4166</span>  {</div> |
| 4265 | <div class="line"><a name="l04167"></a><span class="lineno"> 4167</span>  validateFunc(outputInfo, isSupported);</div> |
| 4266 | <div class="line"><a name="l04168"></a><span class="lineno"> 4168</span>  }</div> |
| 4267 | <div class="line"><a name="l04169"></a><span class="lineno"> 4169</span>  <span class="keywordflow">else</span></div> |
| 4268 | <div class="line"><a name="l04170"></a><span class="lineno"> 4170</span>  {</div> |
| 4269 | <div class="line"><a name="l04171"></a><span class="lineno"> 4171</span>  isDynamic = <span class="keyword">true</span>;</div> |
| 4270 | <div class="line"><a name="l04172"></a><span class="lineno"> 4172</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 4271 | <div class="line"><a name="l04173"></a><span class="lineno"> 4173</span>  }</div> |
| 4272 | <div class="line"><a name="l04174"></a><span class="lineno"> 4174</span>  </div> |
| 4273 | <div class="line"><a name="l04175"></a><span class="lineno"> 4175</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4274 | <div class="line"><a name="l04176"></a><span class="lineno"> 4176</span>  {</div> |
| 4275 | <div class="line"><a name="l04177"></a><span class="lineno"> 4177</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4276 | <div class="line"><a name="l04178"></a><span class="lineno"> 4178</span>  }</div> |
| 4277 | <div class="line"><a name="l04179"></a><span class="lineno"> 4179</span>  </div> |
| 4278 | <div class="line"><a name="l04180"></a><span class="lineno"> 4180</span>  <span class="comment">// Add the layer</span></div> |
| 4279 | <div class="line"><a name="l04181"></a><span class="lineno"> 4181</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddQLstmLayer(desc, params, <span class="stringliteral">"QLstm"</span>);</div> |
| 4280 | <div class="line"><a name="l04182"></a><span class="lineno"> 4182</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4281 | <div class="line"><a name="l04183"></a><span class="lineno"> 4183</span>  </div> |
| 4282 | <div class="line"><a name="l04184"></a><span class="lineno"> 4184</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4283 | <div class="line"><a name="l04185"></a><span class="lineno"> 4185</span>  outputStatePrevTimeStep.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 4284 | <div class="line"><a name="l04186"></a><span class="lineno"> 4186</span>  cellStatePrevTimeStep.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 4285 | <div class="line"><a name="l04187"></a><span class="lineno"> 4187</span>  </div> |
| 4286 | <div class="line"><a name="l04188"></a><span class="lineno"> 4188</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| 4287 | <div class="line"><a name="l04189"></a><span class="lineno"> 4189</span>  {</div> |
| 4288 | <div class="line"><a name="l04190"></a><span class="lineno"> 4190</span>  <span class="keywordflow">return</span> ( <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| 4289 | <div class="line"><a name="l04191"></a><span class="lineno"> 4191</span>  operation, 0, *layer, 0, model, data, &constOutputStateOutInfo) &&</div> |
| 4290 | <div class="line"><a name="l04192"></a><span class="lineno"> 4192</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 1, *layer, 1, model, data) &&</div> |
| 4291 | <div class="line"><a name="l04193"></a><span class="lineno"> 4193</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 2, *layer, 2, model, data, &constOutputInfo));</div> |
| 4292 | <div class="line"><a name="l04194"></a><span class="lineno"> 4194</span>  }</div> |
| 4293 | <div class="line"><a name="l04195"></a><span class="lineno"> 4195</span>  <span class="keywordflow">else</span></div> |
| 4294 | <div class="line"><a name="l04196"></a><span class="lineno"> 4196</span>  {</div> |
| 4295 | <div class="line"><a name="l04197"></a><span class="lineno"> 4197</span>  <span class="keywordflow">return</span> ( <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| 4296 | <div class="line"><a name="l04198"></a><span class="lineno"> 4198</span>  operation, 0, *layer, 0, model, data, &constOutputStateOutInfo) &&</div> |
| 4297 | <div class="line"><a name="l04199"></a><span class="lineno"> 4199</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| 4298 | <div class="line"><a name="l04200"></a><span class="lineno"> 4200</span>  operation, 1, *layer, 1, model, data, <span class="keyword">nullptr</span>, validateFunc,</div> |
| 4299 | <div class="line"><a name="l04201"></a><span class="lineno"> 4201</span>  ActivationFn::kActivationNone, <span class="keyword">true</span>) &&</div> |
| 4300 | <div class="line"><a name="l04202"></a><span class="lineno"> 4202</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 2, *layer, 2, model, data, &constOutputInfo));</div> |
| 4301 | <div class="line"><a name="l04203"></a><span class="lineno"> 4203</span>  }</div> |
| 4302 | <div class="line"><a name="l04204"></a><span class="lineno"> 4204</span> }</div> |
| 4303 | <div class="line"><a name="l04205"></a><span class="lineno"> 4205</span>  </div> |
| 4304 | <div class="line"><a name="l04206"></a><span class="lineno"> 4206</span> <span class="keywordtype">bool</span> Converter::ConvertQuantized16BitLstm(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4305 | <div class="line"><a name="l04207"></a><span class="lineno"> 4207</span> {</div> |
| 4306 | <div class="line"><a name="l04208"></a><span class="lineno"> 4208</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertQuantized16BitLstm()"</span>;</div> |
| 4307 | <div class="line"><a name="l04209"></a><span class="lineno"> 4209</span>  VLOG(DRIVER) << <span class="stringliteral">"Policy::ConvertQuantized16BitLstm()"</span>;</div> |
| 4308 | <div class="line"><a name="l04210"></a><span class="lineno"> 4210</span>  </div> |
| 4309 | <div class="line"><a name="l04211"></a><span class="lineno"> 4211</span>  <span class="comment">//Inputs:</span></div> |
| 4310 | <div class="line"><a name="l04212"></a><span class="lineno"> 4212</span>  <span class="comment">// 0: The input: A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape [numBatches, inputSize]</span></div> |
| 4311 | <div class="line"><a name="l04213"></a><span class="lineno"> 4213</span>  <span class="comment">// specifying the input to the LSTM cell. Tensor is quantized with a fixed quantization range of -1, 127/128.</span></div> |
| 4312 | <div class="line"><a name="l04214"></a><span class="lineno"> 4214</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4313 | <div class="line"><a name="l04215"></a><span class="lineno"> 4215</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4314 | <div class="line"><a name="l04216"></a><span class="lineno"> 4216</span>  {</div> |
| 4315 | <div class="line"><a name="l04217"></a><span class="lineno"> 4217</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0: input"</span>, __func__);</div> |
| 4316 | <div class="line"><a name="l04218"></a><span class="lineno"> 4218</span>  }</div> |
| 4317 | <div class="line"><a name="l04219"></a><span class="lineno"> 4219</span>  </div> |
| 4318 | <div class="line"><a name="l04220"></a><span class="lineno"> 4220</span>  <span class="comment">//13: The previous cell state: A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT16_SYMM and shape</span></div> |
| 4319 | <div class="line"><a name="l04221"></a><span class="lineno"> 4221</span>  <span class="comment">// [numBatches, outputSize] specifying the cell state from the previous time step of the LSTM cell.</span></div> |
| 4320 | <div class="line"><a name="l04222"></a><span class="lineno"> 4222</span>  <span class="comment">// It is quantized using a quantization range of -2^4, 2^4 * 32767/32768.</span></div> |
| 4321 | <div class="line"><a name="l04223"></a><span class="lineno"> 4223</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> previousCellStateIn = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 13, model, data);</div> |
| 4322 | <div class="line"><a name="l04224"></a><span class="lineno"> 4224</span>  <span class="keywordflow">if</span> (!previousCellStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4323 | <div class="line"><a name="l04225"></a><span class="lineno"> 4225</span>  {</div> |
| 4324 | <div class="line"><a name="l04226"></a><span class="lineno"> 4226</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 13: previousCellStateIn"</span>, __func__);</div> |
| 4325 | <div class="line"><a name="l04227"></a><span class="lineno"> 4227</span>  }</div> |
| 4326 | <div class="line"><a name="l04228"></a><span class="lineno"> 4228</span>  </div> |
| 4327 | <div class="line"><a name="l04229"></a><span class="lineno"> 4229</span>  <span class="comment">// 14: The previous output state: A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4328 | <div class="line"><a name="l04230"></a><span class="lineno"> 4230</span>  <span class="comment">// [numBathes, outputSize] specifying the output of the LSTM cell from previous time-step. Tensor</span></div> |
| 4329 | <div class="line"><a name="l04231"></a><span class="lineno"> 4231</span>  <span class="comment">// is quantized with a fixed quantization range of -1, 127/128.</span></div> |
| 4330 | <div class="line"><a name="l04232"></a><span class="lineno"> 4232</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> previousOutputIn = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 14, model, data);</div> |
| 4331 | <div class="line"><a name="l04233"></a><span class="lineno"> 4233</span>  <span class="keywordflow">if</span> (!previousOutputIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4332 | <div class="line"><a name="l04234"></a><span class="lineno"> 4234</span>  {</div> |
| 4333 | <div class="line"><a name="l04235"></a><span class="lineno"> 4235</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 14: previousOutputIn"</span>, __func__);</div> |
| 4334 | <div class="line"><a name="l04236"></a><span class="lineno"> 4236</span>  }</div> |
| 4335 | <div class="line"><a name="l04237"></a><span class="lineno"> 4237</span>  </div> |
| 4336 | <div class="line"><a name="l04238"></a><span class="lineno"> 4238</span>  <span class="comment">// Get the input tensors:</span></div> |
| 4337 | <div class="line"><a name="l04239"></a><span class="lineno"> 4239</span>  <span class="comment">// 1: The input-to-input weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4338 | <div class="line"><a name="l04240"></a><span class="lineno"> 4240</span>  <span class="comment">// [outputSize, inputSize] specifying input-to-input part of weights for fully-connected layer inside the</span></div> |
| 4339 | <div class="line"><a name="l04241"></a><span class="lineno"> 4241</span>  <span class="comment">// LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4340 | <div class="line"><a name="l04242"></a><span class="lineno"> 4242</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToInputWeightsPin =</div> |
| 4341 | <div class="line"><a name="l04243"></a><span class="lineno"> 4243</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 1, model, data);</div> |
| 4342 | <div class="line"><a name="l04244"></a><span class="lineno"> 4244</span>  </div> |
| 4343 | <div class="line"><a name="l04245"></a><span class="lineno"> 4245</span>  <span class="comment">// 2: The input-to-forget weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4344 | <div class="line"><a name="l04246"></a><span class="lineno"> 4246</span>  <span class="comment">// [outputSize, inputSize] specifying input-to-forget part of weights for fully-connected layer inside the</span></div> |
| 4345 | <div class="line"><a name="l04247"></a><span class="lineno"> 4247</span>  <span class="comment">// LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4346 | <div class="line"><a name="l04248"></a><span class="lineno"> 4248</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToForgetWeightsPin =</div> |
| 4347 | <div class="line"><a name="l04249"></a><span class="lineno"> 4249</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 2, model, data);</div> |
| 4348 | <div class="line"><a name="l04250"></a><span class="lineno"> 4250</span>  </div> |
| 4349 | <div class="line"><a name="l04251"></a><span class="lineno"> 4251</span>  <span class="comment">// 3: The input-to-cell weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4350 | <div class="line"><a name="l04252"></a><span class="lineno"> 4252</span>  <span class="comment">// [outputSize, inputSize] specifying input-to-cell part of weights for fully-connected layer inside the</span></div> |
| 4351 | <div class="line"><a name="l04253"></a><span class="lineno"> 4253</span>  <span class="comment">// LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4352 | <div class="line"><a name="l04254"></a><span class="lineno"> 4254</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToCellWeightsPin =</div> |
| 4353 | <div class="line"><a name="l04255"></a><span class="lineno"> 4255</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 3, model, data);</div> |
| 4354 | <div class="line"><a name="l04256"></a><span class="lineno"> 4256</span>  </div> |
| 4355 | <div class="line"><a name="l04257"></a><span class="lineno"> 4257</span>  <span class="comment">// 4: The input-to-output weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4356 | <div class="line"><a name="l04258"></a><span class="lineno"> 4258</span>  <span class="comment">// [outputSize, inputSize] specifying input-to-output part of weights for fully-connected layer inside the</span></div> |
| 4357 | <div class="line"><a name="l04259"></a><span class="lineno"> 4259</span>  <span class="comment">// LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4358 | <div class="line"><a name="l04260"></a><span class="lineno"> 4260</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputToOutputWeightsPin =</div> |
| 4359 | <div class="line"><a name="l04261"></a><span class="lineno"> 4261</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 4, model, data);</div> |
| 4360 | <div class="line"><a name="l04262"></a><span class="lineno"> 4262</span>  </div> |
| 4361 | <div class="line"><a name="l04263"></a><span class="lineno"> 4263</span>  <span class="comment">// 5: The recurrent-to-input weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4362 | <div class="line"><a name="l04264"></a><span class="lineno"> 4264</span>  <span class="comment">// [outputSize, outputSize] specifying recurrent-to-input part of weights for fully-connected layer inside</span></div> |
| 4363 | <div class="line"><a name="l04265"></a><span class="lineno"> 4265</span>  <span class="comment">// the LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4364 | <div class="line"><a name="l04266"></a><span class="lineno"> 4266</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToInputWeightsPin =</div> |
| 4365 | <div class="line"><a name="l04267"></a><span class="lineno"> 4267</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 5, model, data);</div> |
| 4366 | <div class="line"><a name="l04268"></a><span class="lineno"> 4268</span>  </div> |
| 4367 | <div class="line"><a name="l04269"></a><span class="lineno"> 4269</span>  <span class="comment">// 6: The recurrent-to-forget weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4368 | <div class="line"><a name="l04270"></a><span class="lineno"> 4270</span>  <span class="comment">// [outputSize, outputSize] specifying recurrent-to-forget part of weights for fully-connected layer inside</span></div> |
| 4369 | <div class="line"><a name="l04271"></a><span class="lineno"> 4271</span>  <span class="comment">// the LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4370 | <div class="line"><a name="l04272"></a><span class="lineno"> 4272</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToForgetWeightsPin =</div> |
| 4371 | <div class="line"><a name="l04273"></a><span class="lineno"> 4273</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 6, model, data);</div> |
| 4372 | <div class="line"><a name="l04274"></a><span class="lineno"> 4274</span>  </div> |
| 4373 | <div class="line"><a name="l04275"></a><span class="lineno"> 4275</span>  <span class="comment">// 7: The recurrent-to-cell weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4374 | <div class="line"><a name="l04276"></a><span class="lineno"> 4276</span>  <span class="comment">// [outputSize, outputSize] specifying recurrent-to-cell part of weights for fully-connected layer inside</span></div> |
| 4375 | <div class="line"><a name="l04277"></a><span class="lineno"> 4277</span>  <span class="comment">// the LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4376 | <div class="line"><a name="l04278"></a><span class="lineno"> 4278</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToCellWeightsPin =</div> |
| 4377 | <div class="line"><a name="l04279"></a><span class="lineno"> 4279</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 7, model, data);</div> |
| 4378 | <div class="line"><a name="l04280"></a><span class="lineno"> 4280</span>  </div> |
| 4379 | <div class="line"><a name="l04281"></a><span class="lineno"> 4281</span>  <span class="comment">// 8: The recurrent-to-output weights. A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape</span></div> |
| 4380 | <div class="line"><a name="l04282"></a><span class="lineno"> 4282</span>  <span class="comment">// [outputSize, outputSize] specifying recurrent-to-output part of weights for fully-connected layer inside</span></div> |
| 4381 | <div class="line"><a name="l04283"></a><span class="lineno"> 4283</span>  <span class="comment">// the LSTM cell. Quantization zero point and scale must be the same across all the weights.</span></div> |
| 4382 | <div class="line"><a name="l04284"></a><span class="lineno"> 4284</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> recurrentToOutputWeightsPin =</div> |
| 4383 | <div class="line"><a name="l04285"></a><span class="lineno"> 4285</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 8, model, data);</div> |
| 4384 | <div class="line"><a name="l04286"></a><span class="lineno"> 4286</span>  </div> |
| 4385 | <div class="line"><a name="l04287"></a><span class="lineno"> 4287</span>  <span class="comment">// 9: The input gate bias. A 1-D tensor of type ANEURALNETWORKS_TENSOR_INT32 and shape [outputSize] specifying the</span></div> |
| 4386 | <div class="line"><a name="l04288"></a><span class="lineno"> 4288</span>  <span class="comment">// bias for the fully-connected layer inside the LSTM cell. Bias is quantized with scale being a product</span></div> |
| 4387 | <div class="line"><a name="l04289"></a><span class="lineno"> 4289</span>  <span class="comment">// of input and weights scales and zeroPoint equal to 0.</span></div> |
| 4388 | <div class="line"><a name="l04290"></a><span class="lineno"> 4290</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> inputGateBiasPin =</div> |
| 4389 | <div class="line"><a name="l04291"></a><span class="lineno"> 4291</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 9, model, data);</div> |
| 4390 | <div class="line"><a name="l04292"></a><span class="lineno"> 4292</span>  </div> |
| 4391 | <div class="line"><a name="l04293"></a><span class="lineno"> 4293</span>  <span class="comment">// 10: The forget gate bias. A 1-D tensor of type ANEURALNETWORKS_TENSOR_INT32 and shape [outputSize] specifying</span></div> |
| 4392 | <div class="line"><a name="l04294"></a><span class="lineno"> 4294</span>  <span class="comment">// the bias for the fully-connected layer inside the LSTM cell. Bias is quantized with scale being a product</span></div> |
| 4393 | <div class="line"><a name="l04295"></a><span class="lineno"> 4295</span>  <span class="comment">// of input and weights scales and zeroPoint equal to 0.</span></div> |
| 4394 | <div class="line"><a name="l04296"></a><span class="lineno"> 4296</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> forgetGateBiasPin =</div> |
| 4395 | <div class="line"><a name="l04297"></a><span class="lineno"> 4297</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 10, model, data);</div> |
| 4396 | <div class="line"><a name="l04298"></a><span class="lineno"> 4298</span>  </div> |
| 4397 | <div class="line"><a name="l04299"></a><span class="lineno"> 4299</span>  <span class="comment">// 11:The cell bias. A 1-D tensor of type ANEURALNETWORKS_TENSOR_INT32 and shape [outputSize] specifying the bias</span></div> |
| 4398 | <div class="line"><a name="l04300"></a><span class="lineno"> 4300</span>  <span class="comment">// for the fully-connected layer inside the LSTM cell. Bias is quantized with scale being a product of input</span></div> |
| 4399 | <div class="line"><a name="l04301"></a><span class="lineno"> 4301</span>  <span class="comment">// and weights scales and zeroPoint equal to 0.</span></div> |
| 4400 | <div class="line"><a name="l04302"></a><span class="lineno"> 4302</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> cellBiasPin =</div> |
| 4401 | <div class="line"><a name="l04303"></a><span class="lineno"> 4303</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 11, model, data);</div> |
| 4402 | <div class="line"><a name="l04304"></a><span class="lineno"> 4304</span>  </div> |
| 4403 | <div class="line"><a name="l04305"></a><span class="lineno"> 4305</span>  <span class="comment">// 12:The output gate bias. A 1-D tensor of type ANEURALNETWORKS_TENSOR_INT32 and shape [outputSize] specifying</span></div> |
| 4404 | <div class="line"><a name="l04306"></a><span class="lineno"> 4306</span>  <span class="comment">// the bias for the fully-connected layer inside the LSTM cell. Bias is quantized with scale being a product</span></div> |
| 4405 | <div class="line"><a name="l04307"></a><span class="lineno"> 4307</span>  <span class="comment">// of input and weights scales and zeroPoint equal to 0.</span></div> |
| 4406 | <div class="line"><a name="l04308"></a><span class="lineno"> 4308</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> outputGateBiasPin =</div> |
| 4407 | <div class="line"><a name="l04309"></a><span class="lineno"> 4309</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 12, model, data);</div> |
| 4408 | <div class="line"><a name="l04310"></a><span class="lineno"> 4310</span>  </div> |
| 4409 | <div class="line"><a name="l04311"></a><span class="lineno"> 4311</span>  <span class="keywordflow">if</span> (!inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4410 | <div class="line"><a name="l04312"></a><span class="lineno"> 4312</span>  !inputToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4411 | <div class="line"><a name="l04313"></a><span class="lineno"> 4313</span>  !inputToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4412 | <div class="line"><a name="l04314"></a><span class="lineno"> 4314</span>  !inputToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4413 | <div class="line"><a name="l04315"></a><span class="lineno"> 4315</span>  !recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4414 | <div class="line"><a name="l04316"></a><span class="lineno"> 4316</span>  !recurrentToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4415 | <div class="line"><a name="l04317"></a><span class="lineno"> 4317</span>  !recurrentToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4416 | <div class="line"><a name="l04318"></a><span class="lineno"> 4318</span>  !recurrentToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4417 | <div class="line"><a name="l04319"></a><span class="lineno"> 4319</span>  !inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4418 | <div class="line"><a name="l04320"></a><span class="lineno"> 4320</span>  !forgetGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4419 | <div class="line"><a name="l04321"></a><span class="lineno"> 4321</span>  !cellBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() ||</div> |
| 4420 | <div class="line"><a name="l04322"></a><span class="lineno"> 4322</span>  !outputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4421 | <div class="line"><a name="l04323"></a><span class="lineno"> 4323</span>  {</div> |
| 4422 | <div class="line"><a name="l04324"></a><span class="lineno"> 4324</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid tensor inputs"</span>, __func__);</div> |
| 4423 | <div class="line"><a name="l04325"></a><span class="lineno"> 4325</span>  }</div> |
| 4424 | <div class="line"><a name="l04326"></a><span class="lineno"> 4326</span>  </div> |
| 4425 | <div class="line"><a name="l04327"></a><span class="lineno"> 4327</span>  <span class="comment">// Outputs:</span></div> |
| 4426 | <div class="line"><a name="l04328"></a><span class="lineno"> 4328</span>  <span class="comment">// 0: The cell state: A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT16_SYMM and shape [numBatches, outputSize]</span></div> |
| 4427 | <div class="line"><a name="l04329"></a><span class="lineno"> 4329</span>  <span class="comment">// which contains a cell state from the current time step. Tensor is quantized using a quantization range</span></div> |
| 4428 | <div class="line"><a name="l04330"></a><span class="lineno"> 4330</span>  <span class="comment">// of -2^4, 2^4 * 32767/32768.</span></div> |
| 4429 | <div class="line"><a name="l04331"></a><span class="lineno"> 4331</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* cellStateOut = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4430 | <div class="line"><a name="l04332"></a><span class="lineno"> 4332</span>  <span class="keywordflow">if</span> (!cellStateOut)</div> |
| 4431 | <div class="line"><a name="l04333"></a><span class="lineno"> 4333</span>  {</div> |
| 4432 | <div class="line"><a name="l04334"></a><span class="lineno"> 4334</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0: cellStateOut"</span>, __func__);</div> |
| 4433 | <div class="line"><a name="l04335"></a><span class="lineno"> 4335</span>  }</div> |
| 4434 | <div class="line"><a name="l04336"></a><span class="lineno"> 4336</span>  </div> |
| 4435 | <div class="line"><a name="l04337"></a><span class="lineno"> 4337</span>  <span class="comment">// 1: The output: A 2-D tensor of type ANEURALNETWORKS_TENSOR_QUANT8_ASYMM and shape [numBathes, outputSize] which</span></div> |
| 4436 | <div class="line"><a name="l04338"></a><span class="lineno"> 4338</span>  <span class="comment">// contains the output value. Tensor is quantized with a fixed quantization range of -1, 127/128.</span></div> |
| 4437 | <div class="line"><a name="l04339"></a><span class="lineno"> 4339</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 1, model);</div> |
| 4438 | <div class="line"><a name="l04340"></a><span class="lineno"> 4340</span>  <span class="keywordflow">if</span> (!output)</div> |
| 4439 | <div class="line"><a name="l04341"></a><span class="lineno"> 4341</span>  {</div> |
| 4440 | <div class="line"><a name="l04342"></a><span class="lineno"> 4342</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 1: output"</span>, __func__);</div> |
| 4441 | <div class="line"><a name="l04343"></a><span class="lineno"> 4343</span>  }</div> |
| 4442 | <div class="line"><a name="l04344"></a><span class="lineno"> 4344</span>  </div> |
| 4443 | <div class="line"><a name="l04345"></a><span class="lineno"> 4345</span>  <span class="comment">// Inputs</span></div> |
| 4444 | <div class="line"><a name="l04346"></a><span class="lineno"> 4346</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4445 | <div class="line"><a name="l04347"></a><span class="lineno"> 4347</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& previousCellStateInInfo = previousCellStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4446 | <div class="line"><a name="l04348"></a><span class="lineno"> 4348</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& previousOutputInInfo = previousOutputIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4447 | <div class="line"><a name="l04349"></a><span class="lineno"> 4349</span>  </div> |
| 4448 | <div class="line"><a name="l04350"></a><span class="lineno"> 4350</span>  <span class="comment">// Outputs</span></div> |
| 4449 | <div class="line"><a name="l04351"></a><span class="lineno"> 4351</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& cellStateOutInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*cellStateOut);</div> |
| 4450 | <div class="line"><a name="l04352"></a><span class="lineno"> 4352</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 4451 | <div class="line"><a name="l04353"></a><span class="lineno"> 4353</span>  </div> |
| 4452 | <div class="line"><a name="l04354"></a><span class="lineno"> 4354</span>  <span class="comment">// Dynamic tensors currently not supported</span></div> |
| 4453 | <div class="line"><a name="l04355"></a><span class="lineno"> 4355</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(cellStateOutInfo) || <a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 4454 | <div class="line"><a name="l04356"></a><span class="lineno"> 4356</span>  {</div> |
| 4455 | <div class="line"><a name="l04357"></a><span class="lineno"> 4357</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Dynamic output tensors are not supported"</span>, __func__);</div> |
| 4456 | <div class="line"><a name="l04358"></a><span class="lineno"> 4358</span>  }</div> |
| 4457 | <div class="line"><a name="l04359"></a><span class="lineno"> 4359</span>  </div> |
| 4458 | <div class="line"><a name="l04360"></a><span class="lineno"> 4360</span>  <a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html">QuantizedLstmInputParams</a> params;</div> |
| 4459 | <div class="line"><a name="l04361"></a><span class="lineno"> 4361</span>  </div> |
| 4460 | <div class="line"><a name="l04362"></a><span class="lineno"> 4362</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a> = inputToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4461 | <div class="line"><a name="l04363"></a><span class="lineno"> 4363</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">m_InputToForgetWeights</a> = inputToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4462 | <div class="line"><a name="l04364"></a><span class="lineno"> 4364</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">m_InputToCellWeights</a> = inputToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4463 | <div class="line"><a name="l04365"></a><span class="lineno"> 4365</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">m_InputToOutputWeights</a> = inputToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4464 | <div class="line"><a name="l04366"></a><span class="lineno"> 4366</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a> = recurrentToInputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4465 | <div class="line"><a name="l04367"></a><span class="lineno"> 4367</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">m_RecurrentToForgetWeights</a> = recurrentToForgetWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4466 | <div class="line"><a name="l04368"></a><span class="lineno"> 4368</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">m_RecurrentToCellWeights</a> = recurrentToCellWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4467 | <div class="line"><a name="l04369"></a><span class="lineno"> 4369</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">m_RecurrentToOutputWeights</a> = recurrentToOutputWeightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4468 | <div class="line"><a name="l04370"></a><span class="lineno"> 4370</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a> = inputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4469 | <div class="line"><a name="l04371"></a><span class="lineno"> 4371</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">m_ForgetGateBias</a> = forgetGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4470 | <div class="line"><a name="l04372"></a><span class="lineno"> 4372</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">m_CellBias</a> = cellBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4471 | <div class="line"><a name="l04373"></a><span class="lineno"> 4373</span>  params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">m_OutputGateBias</a> = outputGateBiasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">GetConstTensorPtr</a>();</div> |
| 4472 | <div class="line"><a name="l04374"></a><span class="lineno"> 4374</span>  </div> |
| 4473 | <div class="line"><a name="l04375"></a><span class="lineno"> 4375</span>  <a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html">QuantizedLstmInputParamsInfo</a> paramsInfo;</div> |
| 4474 | <div class="line"><a name="l04376"></a><span class="lineno"> 4376</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a15f9f65126ad3d4d82d6c19d0662ae01">m_InputToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">m_InputToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4475 | <div class="line"><a name="l04377"></a><span class="lineno"> 4377</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a4b3f6e6f2268416ffd7a34fda95ffd0b">m_InputToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">m_InputToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4476 | <div class="line"><a name="l04378"></a><span class="lineno"> 4378</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#af07c65d3b7886a00f10c69093e76a341">m_InputToCellWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">m_InputToCellWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4477 | <div class="line"><a name="l04379"></a><span class="lineno"> 4379</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a33242cb43250f04f1736161ca8ad2db9">m_InputToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">m_InputToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4478 | <div class="line"><a name="l04380"></a><span class="lineno"> 4380</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a9ab864961ded72f5ce7ea4eb9921f826">m_RecurrentToInputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">m_RecurrentToInputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4479 | <div class="line"><a name="l04381"></a><span class="lineno"> 4381</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a11b90bd2694aa7f4b434aadd75fbb1c2">m_RecurrentToForgetWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">m_RecurrentToForgetWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4480 | <div class="line"><a name="l04382"></a><span class="lineno"> 4382</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a6c7873a3320a59a63aa26c006db905c0">m_RecurrentToCellWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">m_RecurrentToCellWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4481 | <div class="line"><a name="l04383"></a><span class="lineno"> 4383</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#af434b509cd5232ef762c4b21b2dfaae0">m_RecurrentToOutputWeights</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">m_RecurrentToOutputWeights</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4482 | <div class="line"><a name="l04384"></a><span class="lineno"> 4384</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a042ea9d6d3842b87f112db126e806a93">m_InputGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">m_InputGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4483 | <div class="line"><a name="l04385"></a><span class="lineno"> 4385</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a21626d0b91f8942c3ef8e48e17db9f21">m_ForgetGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">m_ForgetGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4484 | <div class="line"><a name="l04386"></a><span class="lineno"> 4386</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#ab173a067eeb7295d84f5327bcc05a6c1">m_CellBias</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">m_CellBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4485 | <div class="line"><a name="l04387"></a><span class="lineno"> 4387</span>  paramsInfo.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params_info.html#a48cb83dcb58c786ea36f5d37695e75b1">m_OutputGateBias</a> = &(params.<a class="code" href="structarmnn_1_1_quantized_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">m_OutputGateBias</a>-><a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 4486 | <div class="line"><a name="l04388"></a><span class="lineno"> 4388</span>  </div> |
| 4487 | <div class="line"><a name="l04389"></a><span class="lineno"> 4389</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4488 | <div class="line"><a name="l04390"></a><span class="lineno"> 4390</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4489 | <div class="line"><a name="l04391"></a><span class="lineno"> 4391</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 4490 | <div class="line"><a name="l04392"></a><span class="lineno"> 4392</span>  {</div> |
| 4491 | <div class="line"><a name="l04393"></a><span class="lineno"> 4393</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4492 | <div class="line"><a name="l04394"></a><span class="lineno"> 4394</span>  IsQuantizedLstmSupported,</div> |
| 4493 | <div class="line"><a name="l04395"></a><span class="lineno"> 4395</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4494 | <div class="line"><a name="l04396"></a><span class="lineno"> 4396</span>  isSupported,</div> |
| 4495 | <div class="line"><a name="l04397"></a><span class="lineno"> 4397</span>  setBackend,</div> |
| 4496 | <div class="line"><a name="l04398"></a><span class="lineno"> 4398</span>  inputInfo,</div> |
| 4497 | <div class="line"><a name="l04399"></a><span class="lineno"> 4399</span>  previousCellStateInInfo,</div> |
| 4498 | <div class="line"><a name="l04400"></a><span class="lineno"> 4400</span>  previousOutputInInfo,</div> |
| 4499 | <div class="line"><a name="l04401"></a><span class="lineno"> 4401</span>  cellStateOutInfo,</div> |
| 4500 | <div class="line"><a name="l04402"></a><span class="lineno"> 4402</span>  outputInfo,</div> |
| 4501 | <div class="line"><a name="l04403"></a><span class="lineno"> 4403</span>  paramsInfo);</div> |
| 4502 | <div class="line"><a name="l04404"></a><span class="lineno"> 4404</span>  };</div> |
| 4503 | <div class="line"><a name="l04405"></a><span class="lineno"> 4405</span>  </div> |
| 4504 | <div class="line"><a name="l04406"></a><span class="lineno"> 4406</span>  <span class="keywordtype">bool</span> isDynamic = <span class="keyword">false</span>;</div> |
| 4505 | <div class="line"><a name="l04407"></a><span class="lineno"> 4407</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(cellStateOutInfo) &&</div> |
| 4506 | <div class="line"><a name="l04408"></a><span class="lineno"> 4408</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 4507 | <div class="line"><a name="l04409"></a><span class="lineno"> 4409</span>  {</div> |
| 4508 | <div class="line"><a name="l04410"></a><span class="lineno"> 4410</span>  validateFunc(outputInfo, isSupported);</div> |
| 4509 | <div class="line"><a name="l04411"></a><span class="lineno"> 4411</span>  }</div> |
| 4510 | <div class="line"><a name="l04412"></a><span class="lineno"> 4412</span>  <span class="keywordflow">else</span></div> |
| 4511 | <div class="line"><a name="l04413"></a><span class="lineno"> 4413</span>  {</div> |
| 4512 | <div class="line"><a name="l04414"></a><span class="lineno"> 4414</span>  isDynamic = <span class="keyword">true</span>;</div> |
| 4513 | <div class="line"><a name="l04415"></a><span class="lineno"> 4415</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 4514 | <div class="line"><a name="l04416"></a><span class="lineno"> 4416</span>  }</div> |
| 4515 | <div class="line"><a name="l04417"></a><span class="lineno"> 4417</span>  </div> |
| 4516 | <div class="line"><a name="l04418"></a><span class="lineno"> 4418</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4517 | <div class="line"><a name="l04419"></a><span class="lineno"> 4419</span>  {</div> |
| 4518 | <div class="line"><a name="l04420"></a><span class="lineno"> 4420</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4519 | <div class="line"><a name="l04421"></a><span class="lineno"> 4421</span>  }</div> |
| 4520 | <div class="line"><a name="l04422"></a><span class="lineno"> 4422</span>  </div> |
| 4521 | <div class="line"><a name="l04423"></a><span class="lineno"> 4423</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddQuantizedLstmLayer(params, <span class="stringliteral">"QuantizedLstm"</span>);</div> |
| 4522 | <div class="line"><a name="l04424"></a><span class="lineno"> 4424</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4523 | <div class="line"><a name="l04425"></a><span class="lineno"> 4425</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4524 | <div class="line"><a name="l04426"></a><span class="lineno"> 4426</span>  previousCellStateIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 4525 | <div class="line"><a name="l04427"></a><span class="lineno"> 4427</span>  previousOutputIn.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(2));</div> |
| 4526 | <div class="line"><a name="l04428"></a><span class="lineno"> 4428</span>  </div> |
| 4527 | <div class="line"><a name="l04429"></a><span class="lineno"> 4429</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| 4528 | <div class="line"><a name="l04430"></a><span class="lineno"> 4430</span>  {</div> |
| 4529 | <div class="line"><a name="l04431"></a><span class="lineno"> 4431</span>  <span class="keywordflow">return</span> (<a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, 0, model, data) &&</div> |
| 4530 | <div class="line"><a name="l04432"></a><span class="lineno"> 4432</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 1, *layer, 1, model, data));</div> |
| 4531 | <div class="line"><a name="l04433"></a><span class="lineno"> 4433</span>  }</div> |
| 4532 | <div class="line"><a name="l04434"></a><span class="lineno"> 4434</span>  <span class="keywordflow">else</span></div> |
| 4533 | <div class="line"><a name="l04435"></a><span class="lineno"> 4435</span>  {</div> |
| 4534 | <div class="line"><a name="l04436"></a><span class="lineno"> 4436</span>  <span class="keywordflow">return</span> (<a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, 0, model, data) &&</div> |
| 4535 | <div class="line"><a name="l04437"></a><span class="lineno"> 4437</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| 4536 | <div class="line"><a name="l04438"></a><span class="lineno"> 4438</span>  operation, 1, *layer, 1, model, data, <span class="keyword">nullptr</span>, validateFunc, ActivationFn::kActivationNone, <span class="keyword">true</span>));</div> |
| 4537 | <div class="line"><a name="l04439"></a><span class="lineno"> 4439</span>  }</div> |
| 4538 | <div class="line"><a name="l04440"></a><span class="lineno"> 4440</span>  </div> |
| 4539 | <div class="line"><a name="l04441"></a><span class="lineno"> 4441</span> }</div> |
| 4540 | <div class="line"><a name="l04442"></a><span class="lineno"> 4442</span>  </div> |
| 4541 | <div class="line"><a name="l04443"></a><span class="lineno"> 4443</span> <span class="keywordtype">bool</span> Converter::ConvertRank(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4542 | <div class="line"><a name="l04444"></a><span class="lineno"> 4444</span> {</div> |
| 4543 | <div class="line"><a name="l04445"></a><span class="lineno"> 4445</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertRank()"</span>;</div> |
| 4544 | <div class="line"><a name="l04446"></a><span class="lineno"> 4446</span>  </div> |
| 4545 | <div class="line"><a name="l04447"></a><span class="lineno"> 4447</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* inputOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 0, model);</div> |
| 4546 | <div class="line"><a name="l04448"></a><span class="lineno"> 4448</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4547 | <div class="line"><a name="l04449"></a><span class="lineno"> 4449</span>  </div> |
| 4548 | <div class="line"><a name="l04450"></a><span class="lineno"> 4450</span>  <span class="keywordflow">if</span> (inputOperand == <span class="keyword">nullptr</span> || outputOperand == <span class="keyword">nullptr</span>)</div> |
| 4549 | <div class="line"><a name="l04451"></a><span class="lineno"> 4451</span>  {</div> |
| 4550 | <div class="line"><a name="l04452"></a><span class="lineno"> 4452</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 4551 | <div class="line"><a name="l04453"></a><span class="lineno"> 4453</span>  }</div> |
| 4552 | <div class="line"><a name="l04454"></a><span class="lineno"> 4454</span>  </div> |
| 4553 | <div class="line"><a name="l04455"></a><span class="lineno"> 4455</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9">Shape</a> inputOperandShape = GetOperandShape(*inputOperand);</div> |
| 4554 | <div class="line"><a name="l04456"></a><span class="lineno"> 4456</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9">Shape</a> outputOperandShape = GetOperandShape(*outputOperand);</div> |
| 4555 | <div class="line"><a name="l04457"></a><span class="lineno"> 4457</span>  </div> |
| 4556 | <div class="line"><a name="l04458"></a><span class="lineno"> 4458</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4557 | <div class="line"><a name="l04459"></a><span class="lineno"> 4459</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4558 | <div class="line"><a name="l04460"></a><span class="lineno"> 4460</span>  {</div> |
| 4559 | <div class="line"><a name="l04461"></a><span class="lineno"> 4461</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0"</span>, __func__);</div> |
| 4560 | <div class="line"><a name="l04462"></a><span class="lineno"> 4462</span>  }</div> |
| 4561 | <div class="line"><a name="l04463"></a><span class="lineno"> 4463</span>  </div> |
| 4562 | <div class="line"><a name="l04464"></a><span class="lineno"> 4464</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> outInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 4563 | <div class="line"><a name="l04465"></a><span class="lineno"> 4465</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outInfo))</div> |
| 4564 | <div class="line"><a name="l04466"></a><span class="lineno"> 4466</span>  {</div> |
| 4565 | <div class="line"><a name="l04467"></a><span class="lineno"> 4467</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Dynamic output tensors are not supported"</span>, __func__);</div> |
| 4566 | <div class="line"><a name="l04468"></a><span class="lineno"> 4468</span>  }</div> |
| 4567 | <div class="line"><a name="l04469"></a><span class="lineno"> 4469</span>  </div> |
| 4568 | <div class="line"><a name="l04470"></a><span class="lineno"> 4470</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4569 | <div class="line"><a name="l04471"></a><span class="lineno"> 4471</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4570 | <div class="line"><a name="l04472"></a><span class="lineno"> 4472</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4571 | <div class="line"><a name="l04473"></a><span class="lineno"> 4473</span>  IsRankSupported,</div> |
| 4572 | <div class="line"><a name="l04474"></a><span class="lineno"> 4474</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4573 | <div class="line"><a name="l04475"></a><span class="lineno"> 4475</span>  isSupported,</div> |
| 4574 | <div class="line"><a name="l04476"></a><span class="lineno"> 4476</span>  setBackend,</div> |
| 4575 | <div class="line"><a name="l04477"></a><span class="lineno"> 4477</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 4576 | <div class="line"><a name="l04478"></a><span class="lineno"> 4478</span>  outInfo);</div> |
| 4577 | <div class="line"><a name="l04479"></a><span class="lineno"> 4479</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4578 | <div class="line"><a name="l04480"></a><span class="lineno"> 4480</span>  {</div> |
| 4579 | <div class="line"><a name="l04481"></a><span class="lineno"> 4481</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4580 | <div class="line"><a name="l04482"></a><span class="lineno"> 4482</span>  }</div> |
| 4581 | <div class="line"><a name="l04483"></a><span class="lineno"> 4483</span>  </div> |
| 4582 | <div class="line"><a name="l04484"></a><span class="lineno"> 4484</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddRankLayer();</div> |
| 4583 | <div class="line"><a name="l04485"></a><span class="lineno"> 4485</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4584 | <div class="line"><a name="l04486"></a><span class="lineno"> 4486</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 4585 | <div class="line"><a name="l04487"></a><span class="lineno"> 4487</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4586 | <div class="line"><a name="l04488"></a><span class="lineno"> 4488</span>  </div> |
| 4587 | <div class="line"><a name="l04489"></a><span class="lineno"> 4489</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, &outInfo);</div> |
| 4588 | <div class="line"><a name="l04490"></a><span class="lineno"> 4490</span> }</div> |
| 4589 | <div class="line"><a name="l04491"></a><span class="lineno"> 4491</span>  </div> |
| 4590 | <div class="line"><a name="l04492"></a><span class="lineno"> 4492</span> <span class="keywordtype">bool</span> Converter::ConvertReLu(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4591 | <div class="line"><a name="l04493"></a><span class="lineno"> 4493</span> {</div> |
| 4592 | <div class="line"><a name="l04494"></a><span class="lineno"> 4494</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReLu()"</span>;</div> |
| 4593 | <div class="line"><a name="l04495"></a><span class="lineno"> 4495</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">armnn::ActivationDescriptor</a> desc;</div> |
| 4594 | <div class="line"><a name="l04496"></a><span class="lineno"> 4496</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = <a class="code" href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaa3d90c0a5ab3fcf8e6f6608cb3d3a1559">armnn::ActivationFunction::ReLu</a>;</div> |
| 4595 | <div class="line"><a name="l04497"></a><span class="lineno"> 4497</span>  </div> |
| 4596 | <div class="line"><a name="l04498"></a><span class="lineno"> 4498</span>  </div> |
| 4597 | <div class="line"><a name="l04499"></a><span class="lineno"> 4499</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4598 | <div class="line"><a name="l04500"></a><span class="lineno"> 4500</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4599 | <div class="line"><a name="l04501"></a><span class="lineno"> 4501</span>  {</div> |
| 4600 | <div class="line"><a name="l04502"></a><span class="lineno"> 4502</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Input 0 is invalid"</span>, <span class="stringliteral">"operationName"</span>, __func__);</div> |
| 4601 | <div class="line"><a name="l04503"></a><span class="lineno"> 4503</span>  }</div> |
| 4602 | <div class="line"><a name="l04504"></a><span class="lineno"> 4504</span>  </div> |
| 4603 | <div class="line"><a name="l04505"></a><span class="lineno"> 4505</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4604 | <div class="line"><a name="l04506"></a><span class="lineno"> 4506</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 4605 | <div class="line"><a name="l04507"></a><span class="lineno"> 4507</span>  {</div> |
| 4606 | <div class="line"><a name="l04508"></a><span class="lineno"> 4508</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4607 | <div class="line"><a name="l04509"></a><span class="lineno"> 4509</span>  }</div> |
| 4608 | <div class="line"><a name="l04510"></a><span class="lineno"> 4510</span>  </div> |
| 4609 | <div class="line"><a name="l04511"></a><span class="lineno"> 4511</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 4610 | <div class="line"><a name="l04512"></a><span class="lineno"> 4512</span>  </div> |
| 4611 | <div class="line"><a name="l04513"></a><span class="lineno"> 4513</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4612 | <div class="line"><a name="l04514"></a><span class="lineno"> 4514</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4613 | <div class="line"><a name="l04515"></a><span class="lineno"> 4515</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 4614 | <div class="line"><a name="l04516"></a><span class="lineno"> 4516</span>  {</div> |
| 4615 | <div class="line"><a name="l04517"></a><span class="lineno"> 4517</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4616 | <div class="line"><a name="l04518"></a><span class="lineno"> 4518</span>  IsActivationSupported,</div> |
| 4617 | <div class="line"><a name="l04519"></a><span class="lineno"> 4519</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4618 | <div class="line"><a name="l04520"></a><span class="lineno"> 4520</span>  isSupported,</div> |
| 4619 | <div class="line"><a name="l04521"></a><span class="lineno"> 4521</span>  setBackend,</div> |
| 4620 | <div class="line"><a name="l04522"></a><span class="lineno"> 4522</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 4621 | <div class="line"><a name="l04523"></a><span class="lineno"> 4523</span>  outInfo,</div> |
| 4622 | <div class="line"><a name="l04524"></a><span class="lineno"> 4524</span>  desc);</div> |
| 4623 | <div class="line"><a name="l04525"></a><span class="lineno"> 4525</span>  };</div> |
| 4624 | <div class="line"><a name="l04526"></a><span class="lineno"> 4526</span>  </div> |
| 4625 | <div class="line"><a name="l04527"></a><span class="lineno"> 4527</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outInfo))</div> |
| 4626 | <div class="line"><a name="l04528"></a><span class="lineno"> 4528</span>  {</div> |
| 4627 | <div class="line"><a name="l04529"></a><span class="lineno"> 4529</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 4628 | <div class="line"><a name="l04530"></a><span class="lineno"> 4530</span>  }</div> |
| 4629 | <div class="line"><a name="l04531"></a><span class="lineno"> 4531</span>  <span class="keywordflow">else</span></div> |
| 4630 | <div class="line"><a name="l04532"></a><span class="lineno"> 4532</span>  {</div> |
| 4631 | <div class="line"><a name="l04533"></a><span class="lineno"> 4533</span>  validateFunc(outInfo, isSupported);</div> |
| 4632 | <div class="line"><a name="l04534"></a><span class="lineno"> 4534</span>  }</div> |
| 4633 | <div class="line"><a name="l04535"></a><span class="lineno"> 4535</span>  </div> |
| 4634 | <div class="line"><a name="l04536"></a><span class="lineno"> 4536</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4635 | <div class="line"><a name="l04537"></a><span class="lineno"> 4537</span>  {</div> |
| 4636 | <div class="line"><a name="l04538"></a><span class="lineno"> 4538</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4637 | <div class="line"><a name="l04539"></a><span class="lineno"> 4539</span>  }</div> |
| 4638 | <div class="line"><a name="l04540"></a><span class="lineno"> 4540</span>  </div> |
| 4639 | <div class="line"><a name="l04541"></a><span class="lineno"> 4541</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddActivationLayer(desc);</div> |
| 4640 | <div class="line"><a name="l04542"></a><span class="lineno"> 4542</span>  <span class="keywordflow">if</span> (layer == <span class="keyword">nullptr</span>)</div> |
| 4641 | <div class="line"><a name="l04543"></a><span class="lineno"> 4543</span>  {</div> |
| 4642 | <div class="line"><a name="l04544"></a><span class="lineno"> 4544</span>  <span class="keywordflow">throw</span> <a class="code" href="classarmnn_1_1_null_pointer_exception.html">armnn::NullPointerException</a>(<span class="stringliteral">"failed to add Activation Layer to network"</span>);</div> |
| 4643 | <div class="line"><a name="l04545"></a><span class="lineno"> 4545</span>  }</div> |
| 4644 | <div class="line"><a name="l04546"></a><span class="lineno"> 4546</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4645 | <div class="line"><a name="l04547"></a><span class="lineno"> 4547</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4646 | <div class="line"><a name="l04548"></a><span class="lineno"> 4548</span>  </div> |
| 4647 | <div class="line"><a name="l04549"></a><span class="lineno"> 4549</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 4648 | <div class="line"><a name="l04550"></a><span class="lineno"> 4550</span> }</div> |
| 4649 | <div class="line"><a name="l04551"></a><span class="lineno"> 4551</span>  </div> |
| 4650 | <div class="line"><a name="l04552"></a><span class="lineno"> 4552</span> <span class="keywordtype">bool</span> Converter::ConvertReLu1(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4651 | <div class="line"><a name="l04553"></a><span class="lineno"> 4553</span> {</div> |
| 4652 | <div class="line"><a name="l04554"></a><span class="lineno"> 4554</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReLu1()"</span>;</div> |
| 4653 | <div class="line"><a name="l04555"></a><span class="lineno"> 4555</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">armnn::ActivationDescriptor</a> desc;</div> |
| 4654 | <div class="line"><a name="l04556"></a><span class="lineno"> 4556</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = <a class="code" href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaabc5a0f0d6e7cf7fca73299dcd46c0f0d">armnn::ActivationFunction::BoundedReLu</a>;</div> |
| 4655 | <div class="line"><a name="l04557"></a><span class="lineno"> 4557</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a017b2990003a014234f13e999dc7c689">m_A</a> = 1.0f;</div> |
| 4656 | <div class="line"><a name="l04558"></a><span class="lineno"> 4558</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a28c4c9cb15f6be3499abbc46b356060b">m_B</a> = -1.0f;</div> |
| 4657 | <div class="line"><a name="l04559"></a><span class="lineno"> 4559</span>  </div> |
| 4658 | <div class="line"><a name="l04560"></a><span class="lineno"> 4560</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 4659 | <div class="line"><a name="l04561"></a><span class="lineno"> 4561</span> }</div> |
| 4660 | <div class="line"><a name="l04562"></a><span class="lineno"> 4562</span>  </div> |
| 4661 | <div class="line"><a name="l04563"></a><span class="lineno"> 4563</span> <span class="keywordtype">bool</span> Converter::ConvertReLu6(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4662 | <div class="line"><a name="l04564"></a><span class="lineno"> 4564</span> {</div> |
| 4663 | <div class="line"><a name="l04565"></a><span class="lineno"> 4565</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReLu6()"</span>;</div> |
| 4664 | <div class="line"><a name="l04566"></a><span class="lineno"> 4566</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">armnn::ActivationDescriptor</a> desc;</div> |
| 4665 | <div class="line"><a name="l04567"></a><span class="lineno"> 4567</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = <a class="code" href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaabc5a0f0d6e7cf7fca73299dcd46c0f0d">armnn::ActivationFunction::BoundedReLu</a>;</div> |
| 4666 | <div class="line"><a name="l04568"></a><span class="lineno"> 4568</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a017b2990003a014234f13e999dc7c689">m_A</a> = 6.0f;</div> |
| 4667 | <div class="line"><a name="l04569"></a><span class="lineno"> 4569</span>  </div> |
| 4668 | <div class="line"><a name="l04570"></a><span class="lineno"> 4570</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 4669 | <div class="line"><a name="l04571"></a><span class="lineno"> 4571</span> }</div> |
| 4670 | <div class="line"><a name="l04572"></a><span class="lineno"> 4572</span>  </div> |
| 4671 | <div class="line"><a name="l04573"></a><span class="lineno"> 4573</span> <span class="keywordtype">bool</span> Converter::ConvertReshape(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4672 | <div class="line"><a name="l04574"></a><span class="lineno"> 4574</span> {</div> |
| 4673 | <div class="line"><a name="l04575"></a><span class="lineno"> 4575</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReshape()"</span>;</div> |
| 4674 | <div class="line"><a name="l04576"></a><span class="lineno"> 4576</span>  </div> |
| 4675 | <div class="line"><a name="l04577"></a><span class="lineno"> 4577</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* inputOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 0, model);</div> |
| 4676 | <div class="line"><a name="l04578"></a><span class="lineno"> 4578</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* requestedShapeOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 4677 | <div class="line"><a name="l04579"></a><span class="lineno"> 4579</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4678 | <div class="line"><a name="l04580"></a><span class="lineno"> 4580</span>  </div> |
| 4679 | <div class="line"><a name="l04581"></a><span class="lineno"> 4581</span>  <span class="keywordflow">if</span> (inputOperand == <span class="keyword">nullptr</span></div> |
| 4680 | <div class="line"><a name="l04582"></a><span class="lineno"> 4582</span>  || requestedShapeOperand == <span class="keyword">nullptr</span></div> |
| 4681 | <div class="line"><a name="l04583"></a><span class="lineno"> 4583</span>  || outputOperand == <span class="keyword">nullptr</span>)</div> |
| 4682 | <div class="line"><a name="l04584"></a><span class="lineno"> 4584</span>  {</div> |
| 4683 | <div class="line"><a name="l04585"></a><span class="lineno"> 4585</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 4684 | <div class="line"><a name="l04586"></a><span class="lineno"> 4586</span>  }</div> |
| 4685 | <div class="line"><a name="l04587"></a><span class="lineno"> 4587</span>  </div> |
| 4686 | <div class="line"><a name="l04588"></a><span class="lineno"> 4588</span>  <span class="keywordflow">if</span> (requestedShapeOperand->dimensions.size() != 1)</div> |
| 4687 | <div class="line"><a name="l04589"></a><span class="lineno"> 4589</span>  {</div> |
| 4688 | <div class="line"><a name="l04590"></a><span class="lineno"> 4590</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Input 1 expected to be one-dimensional (found %i dimensions)"</span>,</div> |
| 4689 | <div class="line"><a name="l04591"></a><span class="lineno"> 4591</span>  __func__, requestedShapeOperand->dimensions.size());</div> |
| 4690 | <div class="line"><a name="l04592"></a><span class="lineno"> 4592</span>  }</div> |
| 4691 | <div class="line"><a name="l04593"></a><span class="lineno"> 4593</span>  </div> |
| 4692 | <div class="line"><a name="l04594"></a><span class="lineno"> 4594</span>  std::vector<int32_t> targetDimensions;</div> |
| 4693 | <div class="line"><a name="l04595"></a><span class="lineno"> 4595</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*requestedShapeOperand, targetDimensions, model, data))</div> |
| 4694 | <div class="line"><a name="l04596"></a><span class="lineno"> 4596</span>  {</div> |
| 4695 | <div class="line"><a name="l04597"></a><span class="lineno"> 4597</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read values of input 1"</span>, __func__);</div> |
| 4696 | <div class="line"><a name="l04598"></a><span class="lineno"> 4598</span>  }</div> |
| 4697 | <div class="line"><a name="l04599"></a><span class="lineno"> 4599</span>  </div> |
| 4698 | <div class="line"><a name="l04600"></a><span class="lineno"> 4600</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9">Shape</a> inputOperandShape = GetOperandShape(*inputOperand);</div> |
| 4699 | <div class="line"><a name="l04601"></a><span class="lineno"> 4601</span>  </div> |
| 4700 | <div class="line"><a name="l04602"></a><span class="lineno"> 4602</span>  <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9">Shape</a> requestedShape;</div> |
| 4701 | <div class="line"><a name="l04603"></a><span class="lineno"> 4603</span>  <span class="comment">// targetDimensions may contain special values (e.g. -1). reshapePrepare() is an AndroidNN provided utility</span></div> |
| 4702 | <div class="line"><a name="l04604"></a><span class="lineno"> 4604</span>  <span class="comment">// function that resolves these values into a fully specified tensor shape.</span></div> |
| 4703 | <div class="line"><a name="l04605"></a><span class="lineno"> 4605</span>  <span class="keywordflow">if</span> (!reshapePrepare(inputOperandShape, targetDimensions.data(), targetDimensions.size(), &requestedShape))</div> |
| 4704 | <div class="line"><a name="l04606"></a><span class="lineno"> 4606</span>  {</div> |
| 4705 | <div class="line"><a name="l04607"></a><span class="lineno"> 4607</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Failed to resolve the requested shape"</span>, __func__);</div> |
| 4706 | <div class="line"><a name="l04608"></a><span class="lineno"> 4608</span>  }</div> |
| 4707 | <div class="line"><a name="l04609"></a><span class="lineno"> 4609</span>  </div> |
| 4708 | <div class="line"><a name="l04610"></a><span class="lineno"> 4610</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4709 | <div class="line"><a name="l04611"></a><span class="lineno"> 4611</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4710 | <div class="line"><a name="l04612"></a><span class="lineno"> 4612</span>  {</div> |
| 4711 | <div class="line"><a name="l04613"></a><span class="lineno"> 4613</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0"</span>, __func__);</div> |
| 4712 | <div class="line"><a name="l04614"></a><span class="lineno"> 4614</span>  }</div> |
| 4713 | <div class="line"><a name="l04615"></a><span class="lineno"> 4615</span>  </div> |
| 4714 | <div class="line"><a name="l04616"></a><span class="lineno"> 4616</span>  <a class="code" href="structarmnn_1_1_reshape_descriptor.html">armnn::ReshapeDescriptor</a> reshapeDescriptor;</div> |
| 4715 | <div class="line"><a name="l04617"></a><span class="lineno"> 4617</span>  reshapeDescriptor.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a> = <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>(requestedShape.dimensions.size(),</div> |
| 4716 | <div class="line"><a name="l04618"></a><span class="lineno"> 4618</span>  requestedShape.dimensions.data());</div> |
| 4717 | <div class="line"><a name="l04619"></a><span class="lineno"> 4619</span>  </div> |
| 4718 | <div class="line"><a name="l04620"></a><span class="lineno"> 4620</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 4719 | <div class="line"><a name="l04621"></a><span class="lineno"> 4621</span>  </div> |
| 4720 | <div class="line"><a name="l04622"></a><span class="lineno"> 4622</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4721 | <div class="line"><a name="l04623"></a><span class="lineno"> 4623</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4722 | <div class="line"><a name="l04624"></a><span class="lineno"> 4624</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 4723 | <div class="line"><a name="l04625"></a><span class="lineno"> 4625</span>  {</div> |
| 4724 | <div class="line"><a name="l04626"></a><span class="lineno"> 4626</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4725 | <div class="line"><a name="l04627"></a><span class="lineno"> 4627</span>  IsReshapeSupported,</div> |
| 4726 | <div class="line"><a name="l04628"></a><span class="lineno"> 4628</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4727 | <div class="line"><a name="l04629"></a><span class="lineno"> 4629</span>  isSupported,</div> |
| 4728 | <div class="line"><a name="l04630"></a><span class="lineno"> 4630</span>  setBackend,</div> |
| 4729 | <div class="line"><a name="l04631"></a><span class="lineno"> 4631</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 4730 | <div class="line"><a name="l04632"></a><span class="lineno"> 4632</span>  outputInfo,</div> |
| 4731 | <div class="line"><a name="l04633"></a><span class="lineno"> 4633</span>  reshapeDescriptor);</div> |
| 4732 | <div class="line"><a name="l04634"></a><span class="lineno"> 4634</span>  };</div> |
| 4733 | <div class="line"><a name="l04635"></a><span class="lineno"> 4635</span>  </div> |
| 4734 | <div class="line"><a name="l04636"></a><span class="lineno"> 4636</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 4735 | <div class="line"><a name="l04637"></a><span class="lineno"> 4637</span>  {</div> |
| 4736 | <div class="line"><a name="l04638"></a><span class="lineno"> 4638</span>  validateFunc(outputInfo, isSupported);</div> |
| 4737 | <div class="line"><a name="l04639"></a><span class="lineno"> 4639</span>  }</div> |
| 4738 | <div class="line"><a name="l04640"></a><span class="lineno"> 4640</span>  <span class="keywordflow">else</span></div> |
| 4739 | <div class="line"><a name="l04641"></a><span class="lineno"> 4641</span>  {</div> |
| 4740 | <div class="line"><a name="l04642"></a><span class="lineno"> 4642</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 4741 | <div class="line"><a name="l04643"></a><span class="lineno"> 4643</span>  }</div> |
| 4742 | <div class="line"><a name="l04644"></a><span class="lineno"> 4644</span>  </div> |
| 4743 | <div class="line"><a name="l04645"></a><span class="lineno"> 4645</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4744 | <div class="line"><a name="l04646"></a><span class="lineno"> 4646</span>  {</div> |
| 4745 | <div class="line"><a name="l04647"></a><span class="lineno"> 4647</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4746 | <div class="line"><a name="l04648"></a><span class="lineno"> 4648</span>  }</div> |
| 4747 | <div class="line"><a name="l04649"></a><span class="lineno"> 4649</span>  </div> |
| 4748 | <div class="line"><a name="l04650"></a><span class="lineno"> 4650</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddReshapeLayer(reshapeDescriptor);</div> |
| 4749 | <div class="line"><a name="l04651"></a><span class="lineno"> 4651</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4750 | <div class="line"><a name="l04652"></a><span class="lineno"> 4652</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 4751 | <div class="line"><a name="l04653"></a><span class="lineno"> 4653</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4752 | <div class="line"><a name="l04654"></a><span class="lineno"> 4654</span>  </div> |
| 4753 | <div class="line"><a name="l04655"></a><span class="lineno"> 4655</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 4754 | <div class="line"><a name="l04656"></a><span class="lineno"> 4656</span> }</div> |
| 4755 | <div class="line"><a name="l04657"></a><span class="lineno"> 4657</span>  </div> |
| 4756 | <div class="line"><a name="l04658"></a><span class="lineno"> 4658</span> <span class="keywordtype">bool</span> Converter::ConvertResize(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation,</div> |
| 4757 | <div class="line"><a name="l04659"></a><span class="lineno"> 4659</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model,</div> |
| 4758 | <div class="line"><a name="l04660"></a><span class="lineno"> 4660</span>  <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data,</div> |
| 4759 | <div class="line"><a name="l04661"></a><span class="lineno"> 4661</span>  <a class="code" href="namespacearmnn.html#a9a2af2f8c4af4f9efa8e79417d505ac4">ResizeMethod</a> resizeMethod)</div> |
| 4760 | <div class="line"><a name="l04662"></a><span class="lineno"> 4662</span> {</div> |
| 4761 | <div class="line"><a name="l04663"></a><span class="lineno"> 4663</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertResize()"</span>;</div> |
| 4762 | <div class="line"><a name="l04664"></a><span class="lineno"> 4664</span>  VLOG(DRIVER) << <span class="stringliteral">"resizeMethod = "</span> << <a class="code" href="namespacearmnn.html#aded981a42027bd3302b9c0f09d988659">GetResizeMethodAsCString</a>(resizeMethod);</div> |
| 4763 | <div class="line"><a name="l04665"></a><span class="lineno"> 4665</span>  </div> |
| 4764 | <div class="line"><a name="l04666"></a><span class="lineno"> 4666</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4765 | <div class="line"><a name="l04667"></a><span class="lineno"> 4667</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4766 | <div class="line"><a name="l04668"></a><span class="lineno"> 4668</span>  {</div> |
| 4767 | <div class="line"><a name="l04669"></a><span class="lineno"> 4669</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 0"</span>, __func__);</div> |
| 4768 | <div class="line"><a name="l04670"></a><span class="lineno"> 4670</span>  }</div> |
| 4769 | <div class="line"><a name="l04671"></a><span class="lineno"> 4671</span>  </div> |
| 4770 | <div class="line"><a name="l04672"></a><span class="lineno"> 4672</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4771 | <div class="line"><a name="l04673"></a><span class="lineno"> 4673</span>  <span class="keywordflow">if</span> (!output)</div> |
| 4772 | <div class="line"><a name="l04674"></a><span class="lineno"> 4674</span>  {</div> |
| 4773 | <div class="line"><a name="l04675"></a><span class="lineno"> 4675</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 4774 | <div class="line"><a name="l04676"></a><span class="lineno"> 4676</span>  }</div> |
| 4775 | <div class="line"><a name="l04677"></a><span class="lineno"> 4677</span>  </div> |
| 4776 | <div class="line"><a name="l04678"></a><span class="lineno"> 4678</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4777 | <div class="line"><a name="l04679"></a><span class="lineno"> 4679</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 4778 | <div class="line"><a name="l04680"></a><span class="lineno"> 4680</span>  </div> |
| 4779 | <div class="line"><a name="l04681"></a><span class="lineno"> 4681</span>  <a class="code" href="structarmnn_1_1_resize_descriptor.html">ResizeDescriptor</a> descriptor;</div> |
| 4780 | <div class="line"><a name="l04682"></a><span class="lineno"> 4682</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a869254cb56968986a78a79e1d6d4a86b">m_Method</a> = resizeMethod;</div> |
| 4781 | <div class="line"><a name="l04683"></a><span class="lineno"> 4683</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 3, model, data);</div> |
| 4782 | <div class="line"><a name="l04684"></a><span class="lineno"> 4684</span>  </div> |
| 4783 | <div class="line"><a name="l04685"></a><span class="lineno"> 4685</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType1;</div> |
| 4784 | <div class="line"><a name="l04686"></a><span class="lineno"> 4686</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType2;</div> |
| 4785 | <div class="line"><a name="l04687"></a><span class="lineno"> 4687</span>  </div> |
| 4786 | <div class="line"><a name="l04688"></a><span class="lineno"> 4688</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 1, model, operandType1) ||</div> |
| 4787 | <div class="line"><a name="l04689"></a><span class="lineno"> 4689</span>  !<a class="code" href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">GetOperandType</a>(operation, 2, model, operandType2))</div> |
| 4788 | <div class="line"><a name="l04690"></a><span class="lineno"> 4690</span>  {</div> |
| 4789 | <div class="line"><a name="l04691"></a><span class="lineno"> 4691</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 4790 | <div class="line"><a name="l04692"></a><span class="lineno"> 4692</span>  }</div> |
| 4791 | <div class="line"><a name="l04693"></a><span class="lineno"> 4693</span>  </div> |
| 4792 | <div class="line"><a name="l04694"></a><span class="lineno"> 4694</span>  <span class="keywordflow">if</span> (operandType1 != operandType2)</div> |
| 4793 | <div class="line"><a name="l04695"></a><span class="lineno"> 4695</span>  {</div> |
| 4794 | <div class="line"><a name="l04696"></a><span class="lineno"> 4696</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs. Type of input 1 and 2 should be the same"</span>, __func__);</div> |
| 4795 | <div class="line"><a name="l04697"></a><span class="lineno"> 4697</span>  }</div> |
| 4796 | <div class="line"><a name="l04698"></a><span class="lineno"> 4698</span>  </div> |
| 4797 | <div class="line"><a name="l04699"></a><span class="lineno"> 4699</span>  <span class="keywordflow">if</span> (operandType1 == OperandType::INT32)</div> |
| 4798 | <div class="line"><a name="l04700"></a><span class="lineno"> 4700</span>  {</div> |
| 4799 | <div class="line"><a name="l04701"></a><span class="lineno"> 4701</span>  <span class="comment">// Case 1: resizing by shape</span></div> |
| 4800 | <div class="line"><a name="l04702"></a><span class="lineno"> 4702</span>  int32_t targetWidth = 0;</div> |
| 4801 | <div class="line"><a name="l04703"></a><span class="lineno"> 4703</span>  int32_t targetHeight = 0;</div> |
| 4802 | <div class="line"><a name="l04704"></a><span class="lineno"> 4704</span>  </div> |
| 4803 | <div class="line"><a name="l04705"></a><span class="lineno"> 4705</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 1, targetWidth, model, data) ||</div> |
| 4804 | <div class="line"><a name="l04706"></a><span class="lineno"> 4706</span>  !<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 2, targetHeight, model, data))</div> |
| 4805 | <div class="line"><a name="l04707"></a><span class="lineno"> 4707</span>  {</div> |
| 4806 | <div class="line"><a name="l04708"></a><span class="lineno"> 4708</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs for resizing by shape"</span>, __func__);</div> |
| 4807 | <div class="line"><a name="l04709"></a><span class="lineno"> 4709</span>  }</div> |
| 4808 | <div class="line"><a name="l04710"></a><span class="lineno"> 4710</span>  </div> |
| 4809 | <div class="line"><a name="l04711"></a><span class="lineno"> 4711</span>  <span class="keywordflow">if</span> (targetWidth < 0 || targetHeight < 0)</div> |
| 4810 | <div class="line"><a name="l04712"></a><span class="lineno"> 4712</span>  {</div> |
| 4811 | <div class="line"><a name="l04713"></a><span class="lineno"> 4713</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs for resizing by shape. "</span></div> |
| 4812 | <div class="line"><a name="l04714"></a><span class="lineno"> 4714</span>  <span class="stringliteral">"Target width/height cannot be < 0"</span>, __func__);</div> |
| 4813 | <div class="line"><a name="l04715"></a><span class="lineno"> 4715</span>  }</div> |
| 4814 | <div class="line"><a name="l04716"></a><span class="lineno"> 4716</span>  </div> |
| 4815 | <div class="line"><a name="l04717"></a><span class="lineno"> 4717</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#adcf5037208faac36c0788239a073f75c">m_TargetWidth</a> = <span class="keyword">static_cast<</span>uint32_t<span class="keyword">></span>(targetWidth);</div> |
| 4816 | <div class="line"><a name="l04718"></a><span class="lineno"> 4718</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a46c3fa15c46fb0d1dcdc24d0ea5cb5cd">m_TargetHeight</a> = <span class="keyword">static_cast<</span>uint32_t<span class="keyword">></span>(targetHeight);</div> |
| 4817 | <div class="line"><a name="l04719"></a><span class="lineno"> 4719</span>  }</div> |
| 4818 | <div class="line"><a name="l04720"></a><span class="lineno"> 4720</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operandType1 == OperandType::FLOAT32)</div> |
| 4819 | <div class="line"><a name="l04721"></a><span class="lineno"> 4721</span>  {</div> |
| 4820 | <div class="line"><a name="l04722"></a><span class="lineno"> 4722</span>  <span class="comment">// Case 2: resizing by scale</span></div> |
| 4821 | <div class="line"><a name="l04723"></a><span class="lineno"> 4723</span>  <span class="keywordtype">float</span> widthScale = 1.0f;</div> |
| 4822 | <div class="line"><a name="l04724"></a><span class="lineno"> 4724</span>  <span class="keywordtype">float</span> heightScale = 1.0f;</div> |
| 4823 | <div class="line"><a name="l04725"></a><span class="lineno"> 4725</span>  </div> |
| 4824 | <div class="line"><a name="l04726"></a><span class="lineno"> 4726</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 1, widthScale, model, data) ||</div> |
| 4825 | <div class="line"><a name="l04727"></a><span class="lineno"> 4727</span>  !<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 2, heightScale, model, data))</div> |
| 4826 | <div class="line"><a name="l04728"></a><span class="lineno"> 4728</span>  {</div> |
| 4827 | <div class="line"><a name="l04729"></a><span class="lineno"> 4729</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs for resizing by scale"</span>, __func__);</div> |
| 4828 | <div class="line"><a name="l04730"></a><span class="lineno"> 4730</span>  }</div> |
| 4829 | <div class="line"><a name="l04731"></a><span class="lineno"> 4731</span>  </div> |
| 4830 | <div class="line"><a name="l04732"></a><span class="lineno"> 4732</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>& inputShape = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 4831 | <div class="line"><a name="l04733"></a><span class="lineno"> 4733</span>  <a class="code" href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a> dataLayoutIndexed(descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a>);</div> |
| 4832 | <div class="line"><a name="l04734"></a><span class="lineno"> 4734</span>  </div> |
| 4833 | <div class="line"><a name="l04735"></a><span class="lineno"> 4735</span>  <span class="keywordtype">float</span> width = inputShape[dataLayoutIndexed.GetWidthIndex()];</div> |
| 4834 | <div class="line"><a name="l04736"></a><span class="lineno"> 4736</span>  <span class="keywordtype">float</span> height = inputShape[dataLayoutIndexed.GetHeightIndex()];</div> |
| 4835 | <div class="line"><a name="l04737"></a><span class="lineno"> 4737</span>  </div> |
| 4836 | <div class="line"><a name="l04738"></a><span class="lineno"> 4738</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#adcf5037208faac36c0788239a073f75c">m_TargetWidth</a> = std::floor(width * widthScale);</div> |
| 4837 | <div class="line"><a name="l04739"></a><span class="lineno"> 4739</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a46c3fa15c46fb0d1dcdc24d0ea5cb5cd">m_TargetHeight</a> = std::floor(height * heightScale);</div> |
| 4838 | <div class="line"><a name="l04740"></a><span class="lineno"> 4740</span>  }</div> |
| 4839 | <div class="line"><a name="l04741"></a><span class="lineno"> 4741</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operandType1 == OperandType::FLOAT16)</div> |
| 4840 | <div class="line"><a name="l04742"></a><span class="lineno"> 4742</span>  {</div> |
| 4841 | <div class="line"><a name="l04743"></a><span class="lineno"> 4743</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> widthScale;</div> |
| 4842 | <div class="line"><a name="l04744"></a><span class="lineno"> 4744</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> heightScale;</div> |
| 4843 | <div class="line"><a name="l04745"></a><span class="lineno"> 4745</span>  </div> |
| 4844 | <div class="line"><a name="l04746"></a><span class="lineno"> 4746</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT16, widthScale, model, data) ||</div> |
| 4845 | <div class="line"><a name="l04747"></a><span class="lineno"> 4747</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 2, OperandType::FLOAT16, heightScale, model, data))</div> |
| 4846 | <div class="line"><a name="l04748"></a><span class="lineno"> 4748</span>  {</div> |
| 4847 | <div class="line"><a name="l04749"></a><span class="lineno"> 4749</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs for resizing by scale"</span>, __func__);</div> |
| 4848 | <div class="line"><a name="l04750"></a><span class="lineno"> 4750</span>  }</div> |
| 4849 | <div class="line"><a name="l04751"></a><span class="lineno"> 4751</span>  </div> |
| 4850 | <div class="line"><a name="l04752"></a><span class="lineno"> 4752</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>& inputShape = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 4851 | <div class="line"><a name="l04753"></a><span class="lineno"> 4753</span>  <a class="code" href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a> dataLayoutIndexed(descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a>);</div> |
| 4852 | <div class="line"><a name="l04754"></a><span class="lineno"> 4754</span>  </div> |
| 4853 | <div class="line"><a name="l04755"></a><span class="lineno"> 4755</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> width = <span class="keyword">static_cast<</span><a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a><span class="keyword">></span>(inputShape[dataLayoutIndexed.GetWidthIndex()]);</div> |
| 4854 | <div class="line"><a name="l04756"></a><span class="lineno"> 4756</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> height = <span class="keyword">static_cast<</span><a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a><span class="keyword">></span>(inputShape[dataLayoutIndexed.GetHeightIndex()]);</div> |
| 4855 | <div class="line"><a name="l04757"></a><span class="lineno"> 4757</span>  </div> |
| 4856 | <div class="line"><a name="l04758"></a><span class="lineno"> 4758</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#adcf5037208faac36c0788239a073f75c">m_TargetWidth</a> = std::floor(width * widthScale);</div> |
| 4857 | <div class="line"><a name="l04759"></a><span class="lineno"> 4759</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a46c3fa15c46fb0d1dcdc24d0ea5cb5cd">m_TargetHeight</a> = std::floor(height * heightScale);</div> |
| 4858 | <div class="line"><a name="l04760"></a><span class="lineno"> 4760</span>  }</div> |
| 4859 | <div class="line"><a name="l04761"></a><span class="lineno"> 4761</span>  <span class="keywordflow">else</span></div> |
| 4860 | <div class="line"><a name="l04762"></a><span class="lineno"> 4762</span>  {</div> |
| 4861 | <div class="line"><a name="l04763"></a><span class="lineno"> 4763</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operand has invalid data type for resizing by scale"</span>, __func__);</div> |
| 4862 | <div class="line"><a name="l04764"></a><span class="lineno"> 4764</span>  }</div> |
| 4863 | <div class="line"><a name="l04765"></a><span class="lineno"> 4765</span>  </div> |
| 4864 | <div class="line"><a name="l04766"></a><span class="lineno"> 4766</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#ae1a4b3b6c60552509b89747cebb900a2">m_AlignCorners</a> = <a class="code" href="namespacearmnn__driver.html#ae28f9841f6cb1775849b7921b54da32e">GetOptionalBool</a>(operation, 4, model, data);</div> |
| 4865 | <div class="line"><a name="l04767"></a><span class="lineno"> 4767</span>  descriptor.<a class="code" href="structarmnn_1_1_resize_descriptor.html#a4022d5107338aaf5eb7abebf78a1360b">m_HalfPixelCenters</a> = <a class="code" href="namespacearmnn__driver.html#ae28f9841f6cb1775849b7921b54da32e">GetOptionalBool</a>(operation, 5, model, data);</div> |
| 4866 | <div class="line"><a name="l04768"></a><span class="lineno"> 4768</span>  </div> |
| 4867 | <div class="line"><a name="l04769"></a><span class="lineno"> 4769</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4868 | <div class="line"><a name="l04770"></a><span class="lineno"> 4770</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4869 | <div class="line"><a name="l04771"></a><span class="lineno"> 4771</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 4870 | <div class="line"><a name="l04772"></a><span class="lineno"> 4772</span>  {</div> |
| 4871 | <div class="line"><a name="l04773"></a><span class="lineno"> 4773</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4872 | <div class="line"><a name="l04774"></a><span class="lineno"> 4774</span>  IsResizeSupported,</div> |
| 4873 | <div class="line"><a name="l04775"></a><span class="lineno"> 4775</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4874 | <div class="line"><a name="l04776"></a><span class="lineno"> 4776</span>  isSupported,</div> |
| 4875 | <div class="line"><a name="l04777"></a><span class="lineno"> 4777</span>  setBackend,</div> |
| 4876 | <div class="line"><a name="l04778"></a><span class="lineno"> 4778</span>  inputInfo,</div> |
| 4877 | <div class="line"><a name="l04779"></a><span class="lineno"> 4779</span>  outputInfo,</div> |
| 4878 | <div class="line"><a name="l04780"></a><span class="lineno"> 4780</span>  descriptor);</div> |
| 4879 | <div class="line"><a name="l04781"></a><span class="lineno"> 4781</span>  };</div> |
| 4880 | <div class="line"><a name="l04782"></a><span class="lineno"> 4782</span>  </div> |
| 4881 | <div class="line"><a name="l04783"></a><span class="lineno"> 4783</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 4882 | <div class="line"><a name="l04784"></a><span class="lineno"> 4784</span>  {</div> |
| 4883 | <div class="line"><a name="l04785"></a><span class="lineno"> 4785</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 4884 | <div class="line"><a name="l04786"></a><span class="lineno"> 4786</span>  }</div> |
| 4885 | <div class="line"><a name="l04787"></a><span class="lineno"> 4787</span>  <span class="keywordflow">else</span></div> |
| 4886 | <div class="line"><a name="l04788"></a><span class="lineno"> 4788</span>  {</div> |
| 4887 | <div class="line"><a name="l04789"></a><span class="lineno"> 4789</span>  validateFunc(outputInfo, isSupported);</div> |
| 4888 | <div class="line"><a name="l04790"></a><span class="lineno"> 4790</span>  }</div> |
| 4889 | <div class="line"><a name="l04791"></a><span class="lineno"> 4791</span>  </div> |
| 4890 | <div class="line"><a name="l04792"></a><span class="lineno"> 4792</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4891 | <div class="line"><a name="l04793"></a><span class="lineno"> 4793</span>  {</div> |
| 4892 | <div class="line"><a name="l04794"></a><span class="lineno"> 4794</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4893 | <div class="line"><a name="l04795"></a><span class="lineno"> 4795</span>  }</div> |
| 4894 | <div class="line"><a name="l04796"></a><span class="lineno"> 4796</span>  </div> |
| 4895 | <div class="line"><a name="l04797"></a><span class="lineno"> 4797</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddResizeLayer(descriptor);</div> |
| 4896 | <div class="line"><a name="l04798"></a><span class="lineno"> 4798</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4897 | <div class="line"><a name="l04799"></a><span class="lineno"> 4799</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 4898 | <div class="line"><a name="l04800"></a><span class="lineno"> 4800</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4899 | <div class="line"><a name="l04801"></a><span class="lineno"> 4801</span>  </div> |
| 4900 | <div class="line"><a name="l04802"></a><span class="lineno"> 4802</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 4901 | <div class="line"><a name="l04803"></a><span class="lineno"> 4803</span> }</div> |
| 4902 | <div class="line"><a name="l04804"></a><span class="lineno"> 4804</span>  </div> |
| 4903 | <div class="line"><a name="l04805"></a><span class="lineno"> 4805</span> <span class="keywordtype">bool</span> Converter::ConvertReverseV2(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4904 | <div class="line"><a name="l04806"></a><span class="lineno"> 4806</span> {</div> |
| 4905 | <div class="line"><a name="l04807"></a><span class="lineno"> 4807</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReverseV2()"</span>;</div> |
| 4906 | <div class="line"><a name="l04808"></a><span class="lineno"> 4808</span>  </div> |
| 4907 | <div class="line"><a name="l04809"></a><span class="lineno"> 4809</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input0 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4908 | <div class="line"><a name="l04810"></a><span class="lineno"> 4810</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input1 = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 1, model, data);</div> |
| 4909 | <div class="line"><a name="l04811"></a><span class="lineno"> 4811</span>  <span class="keywordflow">if</span> (!input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() || !input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4910 | <div class="line"><a name="l04812"></a><span class="lineno"> 4812</span>  {</div> |
| 4911 | <div class="line"><a name="l04813"></a><span class="lineno"> 4813</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 4912 | <div class="line"><a name="l04814"></a><span class="lineno"> 4814</span>  }</div> |
| 4913 | <div class="line"><a name="l04815"></a><span class="lineno"> 4815</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo0 = input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4914 | <div class="line"><a name="l04816"></a><span class="lineno"> 4816</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo1 = input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4915 | <div class="line"><a name="l04817"></a><span class="lineno"> 4817</span>  </div> |
| 4916 | <div class="line"><a name="l04818"></a><span class="lineno"> 4818</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4917 | <div class="line"><a name="l04819"></a><span class="lineno"> 4819</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 4918 | <div class="line"><a name="l04820"></a><span class="lineno"> 4820</span>  {</div> |
| 4919 | <div class="line"><a name="l04821"></a><span class="lineno"> 4821</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 4920 | <div class="line"><a name="l04822"></a><span class="lineno"> 4822</span>  }</div> |
| 4921 | <div class="line"><a name="l04823"></a><span class="lineno"> 4823</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 4922 | <div class="line"><a name="l04824"></a><span class="lineno"> 4824</span>  </div> |
| 4923 | <div class="line"><a name="l04825"></a><span class="lineno"> 4825</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 4924 | <div class="line"><a name="l04826"></a><span class="lineno"> 4826</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 4925 | <div class="line"><a name="l04827"></a><span class="lineno"> 4827</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 4926 | <div class="line"><a name="l04828"></a><span class="lineno"> 4828</span>  {</div> |
| 4927 | <div class="line"><a name="l04829"></a><span class="lineno"> 4829</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 4928 | <div class="line"><a name="l04830"></a><span class="lineno"> 4830</span>  IsReverseV2Supported,</div> |
| 4929 | <div class="line"><a name="l04831"></a><span class="lineno"> 4831</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 4930 | <div class="line"><a name="l04832"></a><span class="lineno"> 4832</span>  isSupported,</div> |
| 4931 | <div class="line"><a name="l04833"></a><span class="lineno"> 4833</span>  setBackend,</div> |
| 4932 | <div class="line"><a name="l04834"></a><span class="lineno"> 4834</span>  inputInfo0,</div> |
| 4933 | <div class="line"><a name="l04835"></a><span class="lineno"> 4835</span>  inputInfo1,</div> |
| 4934 | <div class="line"><a name="l04836"></a><span class="lineno"> 4836</span>  outputInfo);</div> |
| 4935 | <div class="line"><a name="l04837"></a><span class="lineno"> 4837</span>  };</div> |
| 4936 | <div class="line"><a name="l04838"></a><span class="lineno"> 4838</span>  </div> |
| 4937 | <div class="line"><a name="l04839"></a><span class="lineno"> 4839</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 4938 | <div class="line"><a name="l04840"></a><span class="lineno"> 4840</span>  {</div> |
| 4939 | <div class="line"><a name="l04841"></a><span class="lineno"> 4841</span>  validateFunc(outputInfo, isSupported);</div> |
| 4940 | <div class="line"><a name="l04842"></a><span class="lineno"> 4842</span>  }</div> |
| 4941 | <div class="line"><a name="l04843"></a><span class="lineno"> 4843</span>  <span class="keywordflow">else</span></div> |
| 4942 | <div class="line"><a name="l04844"></a><span class="lineno"> 4844</span>  {</div> |
| 4943 | <div class="line"><a name="l04845"></a><span class="lineno"> 4845</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 4944 | <div class="line"><a name="l04846"></a><span class="lineno"> 4846</span>  }</div> |
| 4945 | <div class="line"><a name="l04847"></a><span class="lineno"> 4847</span>  </div> |
| 4946 | <div class="line"><a name="l04848"></a><span class="lineno"> 4848</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 4947 | <div class="line"><a name="l04849"></a><span class="lineno"> 4849</span>  {</div> |
| 4948 | <div class="line"><a name="l04850"></a><span class="lineno"> 4850</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 4949 | <div class="line"><a name="l04851"></a><span class="lineno"> 4851</span>  }</div> |
| 4950 | <div class="line"><a name="l04852"></a><span class="lineno"> 4852</span>  </div> |
| 4951 | <div class="line"><a name="l04853"></a><span class="lineno"> 4853</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddReverseV2Layer();</div> |
| 4952 | <div class="line"><a name="l04854"></a><span class="lineno"> 4854</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 4953 | <div class="line"><a name="l04855"></a><span class="lineno"> 4855</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 4954 | <div class="line"><a name="l04856"></a><span class="lineno"> 4856</span>  input0.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 4955 | <div class="line"><a name="l04857"></a><span class="lineno"> 4857</span>  input1.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div> |
| 4956 | <div class="line"><a name="l04858"></a><span class="lineno"> 4858</span>  </div> |
| 4957 | <div class="line"><a name="l04859"></a><span class="lineno"> 4859</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 4958 | <div class="line"><a name="l04860"></a><span class="lineno"> 4860</span> }</div> |
| 4959 | <div class="line"><a name="l04861"></a><span class="lineno"> 4861</span>  </div> |
| 4960 | <div class="line"><a name="l04862"></a><span class="lineno"> 4862</span> <span class="keywordtype">bool</span> Converter::ConvertSpaceToBatchNd(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 4961 | <div class="line"><a name="l04863"></a><span class="lineno"> 4863</span> {</div> |
| 4962 | <div class="line"><a name="l04864"></a><span class="lineno"> 4864</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSpaceToBatchNd()"</span>;</div> |
| 4963 | <div class="line"><a name="l04865"></a><span class="lineno"> 4865</span>  </div> |
| 4964 | <div class="line"><a name="l04866"></a><span class="lineno"> 4866</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 4965 | <div class="line"><a name="l04867"></a><span class="lineno"> 4867</span>  <span class="keywordflow">if</span>(!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 4966 | <div class="line"><a name="l04868"></a><span class="lineno"> 4868</span>  {</div> |
| 4967 | <div class="line"><a name="l04869"></a><span class="lineno"> 4869</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 4968 | <div class="line"><a name="l04870"></a><span class="lineno"> 4870</span>  }</div> |
| 4969 | <div class="line"><a name="l04871"></a><span class="lineno"> 4871</span>  </div> |
| 4970 | <div class="line"><a name="l04872"></a><span class="lineno"> 4872</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 4971 | <div class="line"><a name="l04873"></a><span class="lineno"> 4873</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 4972 | <div class="line"><a name="l04874"></a><span class="lineno"> 4874</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> spatialDim = rank - 2;</div> |
| 4973 | <div class="line"><a name="l04875"></a><span class="lineno"> 4875</span>  </div> |
| 4974 | <div class="line"><a name="l04876"></a><span class="lineno"> 4876</span>  <span class="keywordflow">if</span>(rank != 4)</div> |
| 4975 | <div class="line"><a name="l04877"></a><span class="lineno"> 4877</span>  {</div> |
| 4976 | <div class="line"><a name="l04878"></a><span class="lineno"> 4878</span>  Fail(<span class="stringliteral">"%s: Only inputs with rank 4 are supported"</span>, __func__);</div> |
| 4977 | <div class="line"><a name="l04879"></a><span class="lineno"> 4879</span>  }</div> |
| 4978 | <div class="line"><a name="l04880"></a><span class="lineno"> 4880</span>  </div> |
| 4979 | <div class="line"><a name="l04881"></a><span class="lineno"> 4881</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a> *output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 4980 | <div class="line"><a name="l04882"></a><span class="lineno"> 4882</span>  <span class="keywordflow">if</span>(!output)</div> |
| 4981 | <div class="line"><a name="l04883"></a><span class="lineno"> 4883</span>  {</div> |
| 4982 | <div class="line"><a name="l04884"></a><span class="lineno"> 4884</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 4983 | <div class="line"><a name="l04885"></a><span class="lineno"> 4885</span>  }</div> |
| 4984 | <div class="line"><a name="l04886"></a><span class="lineno"> 4886</span>  </div> |
| 4985 | <div class="line"><a name="l04887"></a><span class="lineno"> 4887</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 4986 | <div class="line"><a name="l04888"></a><span class="lineno"> 4888</span>  </div> |
| 4987 | <div class="line"><a name="l04889"></a><span class="lineno"> 4889</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a> *blockShapeOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 4988 | <div class="line"><a name="l04890"></a><span class="lineno"> 4890</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a> *paddingsOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 2, model);</div> |
| 4989 | <div class="line"><a name="l04891"></a><span class="lineno"> 4891</span>  </div> |
| 4990 | <div class="line"><a name="l04892"></a><span class="lineno"> 4892</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> blockShapeOperandShape = GetTensorShapeForOperand(*blockShapeOperand);</div> |
| 4991 | <div class="line"><a name="l04893"></a><span class="lineno"> 4893</span>  <span class="keywordflow">if</span>(blockShapeOperandShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() != 1 || blockShapeOperandShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a8846406ac37fbd2204f0be16ee05d5b7">GetNumElements</a>() != spatialDim)</div> |
| 4992 | <div class="line"><a name="l04894"></a><span class="lineno"> 4894</span>  {</div> |
| 4993 | <div class="line"><a name="l04895"></a><span class="lineno"> 4895</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid block shape operand: expected shape [%d]"</span>, __func__, spatialDim);</div> |
| 4994 | <div class="line"><a name="l04896"></a><span class="lineno"> 4896</span>  }</div> |
| 4995 | <div class="line"><a name="l04897"></a><span class="lineno"> 4897</span>  </div> |
| 4996 | <div class="line"><a name="l04898"></a><span class="lineno"> 4898</span>  std::vector<int32_t> blockShape;</div> |
| 4997 | <div class="line"><a name="l04899"></a><span class="lineno"> 4899</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*blockShapeOperand, blockShape, model, data))</div> |
| 4998 | <div class="line"><a name="l04900"></a><span class="lineno"> 4900</span>  {</div> |
| 4999 | <div class="line"><a name="l04901"></a><span class="lineno"> 4901</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has an invalid or unsupported block size operand"</span>, __func__);</div> |
| 5000 | <div class="line"><a name="l04902"></a><span class="lineno"> 4902</span>  }</div> |
| 5001 | <div class="line"><a name="l04903"></a><span class="lineno"> 4903</span>  <span class="keywordflow">if</span>(std::any_of(blockShape.cbegin(), blockShape.cend(), [](int32_t i)</div> |
| 5002 | <div class="line"><a name="l04904"></a><span class="lineno"> 4904</span>  { return i < 1; }))</div> |
| 5003 | <div class="line"><a name="l04905"></a><span class="lineno"> 4905</span>  {</div> |
| 5004 | <div class="line"><a name="l04906"></a><span class="lineno"> 4906</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Block shape must be at least 1 in all dimensions."</span>, __func__);</div> |
| 5005 | <div class="line"><a name="l04907"></a><span class="lineno"> 4907</span>  }</div> |
| 5006 | <div class="line"><a name="l04908"></a><span class="lineno"> 4908</span>  </div> |
| 5007 | <div class="line"><a name="l04909"></a><span class="lineno"> 4909</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> paddingsOperandShape = GetTensorShapeForOperand(*paddingsOperand);</div> |
| 5008 | <div class="line"><a name="l04910"></a><span class="lineno"> 4910</span>  <span class="keywordflow">if</span>(paddingsOperandShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>() != 2 || paddingsOperandShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a8846406ac37fbd2204f0be16ee05d5b7">GetNumElements</a>() != 2 * spatialDim)</div> |
| 5009 | <div class="line"><a name="l04911"></a><span class="lineno"> 4911</span>  {</div> |
| 5010 | <div class="line"><a name="l04912"></a><span class="lineno"> 4912</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid paddings operand: expected shape [%d, 2]"</span>, __func__, spatialDim);</div> |
| 5011 | <div class="line"><a name="l04913"></a><span class="lineno"> 4913</span>  }</div> |
| 5012 | <div class="line"><a name="l04914"></a><span class="lineno"> 4914</span>  </div> |
| 5013 | <div class="line"><a name="l04915"></a><span class="lineno"> 4915</span>  std::vector<std::pair<unsigned int, unsigned int>> paddingList;</div> |
| 5014 | <div class="line"><a name="l04916"></a><span class="lineno"> 4916</span>  std::vector<int32_t> paddings;</div> |
| 5015 | <div class="line"><a name="l04917"></a><span class="lineno"> 4917</span>  <span class="keywordflow">if</span>(!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*paddingsOperand, paddings, model, data))</div> |
| 5016 | <div class="line"><a name="l04918"></a><span class="lineno"> 4918</span>  {</div> |
| 5017 | <div class="line"><a name="l04919"></a><span class="lineno"> 4919</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has an invalid or unsupported paddings operand"</span>, __func__);</div> |
| 5018 | <div class="line"><a name="l04920"></a><span class="lineno"> 4920</span>  }</div> |
| 5019 | <div class="line"><a name="l04921"></a><span class="lineno"> 4921</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i < paddings.size() - 1; i += 2)</div> |
| 5020 | <div class="line"><a name="l04922"></a><span class="lineno"> 4922</span>  {</div> |
| 5021 | <div class="line"><a name="l04923"></a><span class="lineno"> 4923</span>  <span class="keywordtype">int</span> paddingBeforeInput = paddings[i];</div> |
| 5022 | <div class="line"><a name="l04924"></a><span class="lineno"> 4924</span>  <span class="keywordtype">int</span> paddingAfterInput = paddings[i + 1];</div> |
| 5023 | <div class="line"><a name="l04925"></a><span class="lineno"> 4925</span>  <span class="keywordflow">if</span>(paddingBeforeInput < 0 || paddingAfterInput < 0)</div> |
| 5024 | <div class="line"><a name="l04926"></a><span class="lineno"> 4926</span>  {</div> |
| 5025 | <div class="line"><a name="l04927"></a><span class="lineno"> 4927</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid paddings operand, invalid padding values."</span>, __func__);</div> |
| 5026 | <div class="line"><a name="l04928"></a><span class="lineno"> 4928</span>  }</div> |
| 5027 | <div class="line"><a name="l04929"></a><span class="lineno"> 4929</span>  </div> |
| 5028 | <div class="line"><a name="l04930"></a><span class="lineno"> 4930</span>  paddingList.emplace_back((<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>) paddingBeforeInput, (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>) paddingAfterInput);</div> |
| 5029 | <div class="line"><a name="l04931"></a><span class="lineno"> 4931</span>  }</div> |
| 5030 | <div class="line"><a name="l04932"></a><span class="lineno"> 4932</span>  </div> |
| 5031 | <div class="line"><a name="l04933"></a><span class="lineno"> 4933</span>  <a class="code" href="structarmnn_1_1_space_to_batch_nd_descriptor.html">armnn::SpaceToBatchNdDescriptor</a> descriptor;</div> |
| 5032 | <div class="line"><a name="l04934"></a><span class="lineno"> 4934</span>  descriptor.<a class="code" href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 5033 | <div class="line"><a name="l04935"></a><span class="lineno"> 4935</span>  descriptor.<a class="code" href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a02e143524aefddd40b485fcf7dea6696">m_BlockShape</a>.assign(blockShape.cbegin(), blockShape.cend());</div> |
| 5034 | <div class="line"><a name="l04936"></a><span class="lineno"> 4936</span>  descriptor.<a class="code" href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a85f98c94e11f65a6b73f831735c040f3">m_PadList</a>.assign(paddingList.cbegin(), paddingList.cend());</div> |
| 5035 | <div class="line"><a name="l04937"></a><span class="lineno"> 4937</span>  </div> |
| 5036 | <div class="line"><a name="l04938"></a><span class="lineno"> 4938</span>  <span class="keywordflow">if</span>(Is12OrLaterOperand(*output))</div> |
| 5037 | <div class="line"><a name="l04939"></a><span class="lineno"> 4939</span>  {</div> |
| 5038 | <div class="line"><a name="l04940"></a><span class="lineno"> 4940</span>  descriptor.<a class="code" href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 3, model, data);</div> |
| 5039 | <div class="line"><a name="l04941"></a><span class="lineno"> 4941</span>  }</div> |
| 5040 | <div class="line"><a name="l04942"></a><span class="lineno"> 4942</span>  </div> |
| 5041 | <div class="line"><a name="l04943"></a><span class="lineno"> 4943</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5042 | <div class="line"><a name="l04944"></a><span class="lineno"> 4944</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5043 | <div class="line"><a name="l04945"></a><span class="lineno"> 4945</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &outputInfo, <span class="keywordtype">bool</span> &isSupported)</div> |
| 5044 | <div class="line"><a name="l04946"></a><span class="lineno"> 4946</span>  {</div> |
| 5045 | <div class="line"><a name="l04947"></a><span class="lineno"> 4947</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5046 | <div class="line"><a name="l04948"></a><span class="lineno"> 4948</span>  IsSpaceToBatchNdSupported,</div> |
| 5047 | <div class="line"><a name="l04949"></a><span class="lineno"> 4949</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5048 | <div class="line"><a name="l04950"></a><span class="lineno"> 4950</span>  isSupported,</div> |
| 5049 | <div class="line"><a name="l04951"></a><span class="lineno"> 4951</span>  setBackend,</div> |
| 5050 | <div class="line"><a name="l04952"></a><span class="lineno"> 4952</span>  inputInfo,</div> |
| 5051 | <div class="line"><a name="l04953"></a><span class="lineno"> 4953</span>  outputInfo,</div> |
| 5052 | <div class="line"><a name="l04954"></a><span class="lineno"> 4954</span>  descriptor);</div> |
| 5053 | <div class="line"><a name="l04955"></a><span class="lineno"> 4955</span>  };</div> |
| 5054 | <div class="line"><a name="l04956"></a><span class="lineno"> 4956</span>  </div> |
| 5055 | <div class="line"><a name="l04957"></a><span class="lineno"> 4957</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5056 | <div class="line"><a name="l04958"></a><span class="lineno"> 4958</span>  {</div> |
| 5057 | <div class="line"><a name="l04959"></a><span class="lineno"> 4959</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5058 | <div class="line"><a name="l04960"></a><span class="lineno"> 4960</span>  } <span class="keywordflow">else</span></div> |
| 5059 | <div class="line"><a name="l04961"></a><span class="lineno"> 4961</span>  {</div> |
| 5060 | <div class="line"><a name="l04962"></a><span class="lineno"> 4962</span>  validateFunc(outputInfo, isSupported);</div> |
| 5061 | <div class="line"><a name="l04963"></a><span class="lineno"> 4963</span>  }</div> |
| 5062 | <div class="line"><a name="l04964"></a><span class="lineno"> 4964</span>  </div> |
| 5063 | <div class="line"><a name="l04965"></a><span class="lineno"> 4965</span>  <span class="keywordflow">if</span>(!isSupported)</div> |
| 5064 | <div class="line"><a name="l04966"></a><span class="lineno"> 4966</span>  {</div> |
| 5065 | <div class="line"><a name="l04967"></a><span class="lineno"> 4967</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5066 | <div class="line"><a name="l04968"></a><span class="lineno"> 4968</span>  }</div> |
| 5067 | <div class="line"><a name="l04969"></a><span class="lineno"> 4969</span>  </div> |
| 5068 | <div class="line"><a name="l04970"></a><span class="lineno"> 4970</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a> *<span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddSpaceToBatchNdLayer(descriptor);</div> |
| 5069 | <div class="line"><a name="l04971"></a><span class="lineno"> 4971</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5070 | <div class="line"><a name="l04972"></a><span class="lineno"> 4972</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5071 | <div class="line"><a name="l04973"></a><span class="lineno"> 4973</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5072 | <div class="line"><a name="l04974"></a><span class="lineno"> 4974</span>  </div> |
| 5073 | <div class="line"><a name="l04975"></a><span class="lineno"> 4975</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 5074 | <div class="line"><a name="l04976"></a><span class="lineno"> 4976</span> }</div> |
| 5075 | <div class="line"><a name="l04977"></a><span class="lineno"> 4977</span>  </div> |
| 5076 | <div class="line"><a name="l04978"></a><span class="lineno"> 4978</span> <span class="keywordtype">bool</span> Converter::ConvertSpaceToDepth(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5077 | <div class="line"><a name="l04979"></a><span class="lineno"> 4979</span> {</div> |
| 5078 | <div class="line"><a name="l04980"></a><span class="lineno"> 4980</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSpaceToDepth()"</span>;</div> |
| 5079 | <div class="line"><a name="l04981"></a><span class="lineno"> 4981</span>  </div> |
| 5080 | <div class="line"><a name="l04982"></a><span class="lineno"> 4982</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5081 | <div class="line"><a name="l04983"></a><span class="lineno"> 4983</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>() )</div> |
| 5082 | <div class="line"><a name="l04984"></a><span class="lineno"> 4984</span>  {</div> |
| 5083 | <div class="line"><a name="l04985"></a><span class="lineno"> 4985</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5084 | <div class="line"><a name="l04986"></a><span class="lineno"> 4986</span>  }</div> |
| 5085 | <div class="line"><a name="l04987"></a><span class="lineno"> 4987</span>  </div> |
| 5086 | <div class="line"><a name="l04988"></a><span class="lineno"> 4988</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 5087 | <div class="line"><a name="l04989"></a><span class="lineno"> 4989</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 5088 | <div class="line"><a name="l04990"></a><span class="lineno"> 4990</span>  <span class="keywordflow">if</span> (rank != 4)</div> |
| 5089 | <div class="line"><a name="l04991"></a><span class="lineno"> 4991</span>  {</div> |
| 5090 | <div class="line"><a name="l04992"></a><span class="lineno"> 4992</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Only inputs with rank 4 are supported"</span>, __func__);</div> |
| 5091 | <div class="line"><a name="l04993"></a><span class="lineno"> 4993</span>  }</div> |
| 5092 | <div class="line"><a name="l04994"></a><span class="lineno"> 4994</span>  </div> |
| 5093 | <div class="line"><a name="l04995"></a><span class="lineno"> 4995</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5094 | <div class="line"><a name="l04996"></a><span class="lineno"> 4996</span>  <span class="keywordflow">if</span> (!output)</div> |
| 5095 | <div class="line"><a name="l04997"></a><span class="lineno"> 4997</span>  {</div> |
| 5096 | <div class="line"><a name="l04998"></a><span class="lineno"> 4998</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 5097 | <div class="line"><a name="l04999"></a><span class="lineno"> 4999</span>  }</div> |
| 5098 | <div class="line"><a name="l05000"></a><span class="lineno"> 5000</span>  </div> |
| 5099 | <div class="line"><a name="l05001"></a><span class="lineno"> 5001</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 5100 | <div class="line"><a name="l05002"></a><span class="lineno"> 5002</span>  </div> |
| 5101 | <div class="line"><a name="l05003"></a><span class="lineno"> 5003</span>  <a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html">SpaceToDepthDescriptor</a> desc;</div> |
| 5102 | <div class="line"><a name="l05004"></a><span class="lineno"> 5004</span>  </div> |
| 5103 | <div class="line"><a name="l05005"></a><span class="lineno"> 5005</span>  <a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6c6b8957f1e176867e5fb05b1a1a1486">m_BlockSize</a>, model, data);</div> |
| 5104 | <div class="line"><a name="l05006"></a><span class="lineno"> 5006</span>  </div> |
| 5105 | <div class="line"><a name="l05007"></a><span class="lineno"> 5007</span>  <span class="keywordflow">if</span> (desc.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6c6b8957f1e176867e5fb05b1a1a1486">m_BlockSize</a> <= 1)</div> |
| 5106 | <div class="line"><a name="l05008"></a><span class="lineno"> 5008</span>  {</div> |
| 5107 | <div class="line"><a name="l05009"></a><span class="lineno"> 5009</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Block size must be at least 1 in all dimensions"</span>, __func__);</div> |
| 5108 | <div class="line"><a name="l05010"></a><span class="lineno"> 5010</span>  }</div> |
| 5109 | <div class="line"><a name="l05011"></a><span class="lineno"> 5011</span>  </div> |
| 5110 | <div class="line"><a name="l05012"></a><span class="lineno"> 5012</span>  desc.<a class="code" href="structarmnn_1_1_space_to_depth_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 2, model, data);</div> |
| 5111 | <div class="line"><a name="l05013"></a><span class="lineno"> 5013</span>  </div> |
| 5112 | <div class="line"><a name="l05014"></a><span class="lineno"> 5014</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5113 | <div class="line"><a name="l05015"></a><span class="lineno"> 5015</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5114 | <div class="line"><a name="l05016"></a><span class="lineno"> 5016</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 5115 | <div class="line"><a name="l05017"></a><span class="lineno"> 5017</span>  {</div> |
| 5116 | <div class="line"><a name="l05018"></a><span class="lineno"> 5018</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5117 | <div class="line"><a name="l05019"></a><span class="lineno"> 5019</span>  IsSpaceToDepthSupported,</div> |
| 5118 | <div class="line"><a name="l05020"></a><span class="lineno"> 5020</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5119 | <div class="line"><a name="l05021"></a><span class="lineno"> 5021</span>  isSupported,</div> |
| 5120 | <div class="line"><a name="l05022"></a><span class="lineno"> 5022</span>  setBackend,</div> |
| 5121 | <div class="line"><a name="l05023"></a><span class="lineno"> 5023</span>  inputInfo,</div> |
| 5122 | <div class="line"><a name="l05024"></a><span class="lineno"> 5024</span>  outputInfo,</div> |
| 5123 | <div class="line"><a name="l05025"></a><span class="lineno"> 5025</span>  desc);</div> |
| 5124 | <div class="line"><a name="l05026"></a><span class="lineno"> 5026</span>  };</div> |
| 5125 | <div class="line"><a name="l05027"></a><span class="lineno"> 5027</span>  </div> |
| 5126 | <div class="line"><a name="l05028"></a><span class="lineno"> 5028</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5127 | <div class="line"><a name="l05029"></a><span class="lineno"> 5029</span>  {</div> |
| 5128 | <div class="line"><a name="l05030"></a><span class="lineno"> 5030</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5129 | <div class="line"><a name="l05031"></a><span class="lineno"> 5031</span>  }</div> |
| 5130 | <div class="line"><a name="l05032"></a><span class="lineno"> 5032</span>  <span class="keywordflow">else</span></div> |
| 5131 | <div class="line"><a name="l05033"></a><span class="lineno"> 5033</span>  {</div> |
| 5132 | <div class="line"><a name="l05034"></a><span class="lineno"> 5034</span>  validateFunc(outputInfo, isSupported);</div> |
| 5133 | <div class="line"><a name="l05035"></a><span class="lineno"> 5035</span>  }</div> |
| 5134 | <div class="line"><a name="l05036"></a><span class="lineno"> 5036</span>  </div> |
| 5135 | <div class="line"><a name="l05037"></a><span class="lineno"> 5037</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5136 | <div class="line"><a name="l05038"></a><span class="lineno"> 5038</span>  {</div> |
| 5137 | <div class="line"><a name="l05039"></a><span class="lineno"> 5039</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5138 | <div class="line"><a name="l05040"></a><span class="lineno"> 5040</span>  }</div> |
| 5139 | <div class="line"><a name="l05041"></a><span class="lineno"> 5041</span>  </div> |
| 5140 | <div class="line"><a name="l05042"></a><span class="lineno"> 5042</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddSpaceToDepthLayer(desc);</div> |
| 5141 | <div class="line"><a name="l05043"></a><span class="lineno"> 5043</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5142 | <div class="line"><a name="l05044"></a><span class="lineno"> 5044</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5143 | <div class="line"><a name="l05045"></a><span class="lineno"> 5045</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5144 | <div class="line"><a name="l05046"></a><span class="lineno"> 5046</span>  </div> |
| 5145 | <div class="line"><a name="l05047"></a><span class="lineno"> 5047</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 5146 | <div class="line"><a name="l05048"></a><span class="lineno"> 5048</span> }</div> |
| 5147 | <div class="line"><a name="l05049"></a><span class="lineno"> 5049</span>  </div> |
| 5148 | <div class="line"><a name="l05050"></a><span class="lineno"> 5050</span> <span class="keywordtype">bool</span> Converter::ConvertSoftmax(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5149 | <div class="line"><a name="l05051"></a><span class="lineno"> 5051</span> {</div> |
| 5150 | <div class="line"><a name="l05052"></a><span class="lineno"> 5052</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSoftmax()"</span>;</div> |
| 5151 | <div class="line"><a name="l05053"></a><span class="lineno"> 5053</span>  </div> |
| 5152 | <div class="line"><a name="l05054"></a><span class="lineno"> 5054</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5153 | <div class="line"><a name="l05055"></a><span class="lineno"> 5055</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5154 | <div class="line"><a name="l05056"></a><span class="lineno"> 5056</span>  {</div> |
| 5155 | <div class="line"><a name="l05057"></a><span class="lineno"> 5057</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5156 | <div class="line"><a name="l05058"></a><span class="lineno"> 5058</span>  }</div> |
| 5157 | <div class="line"><a name="l05059"></a><span class="lineno"> 5059</span>  </div> |
| 5158 | <div class="line"><a name="l05060"></a><span class="lineno"> 5060</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5159 | <div class="line"><a name="l05061"></a><span class="lineno"> 5061</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 5160 | <div class="line"><a name="l05062"></a><span class="lineno"> 5062</span>  {</div> |
| 5161 | <div class="line"><a name="l05063"></a><span class="lineno"> 5063</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has no outputs"</span>, __func__);</div> |
| 5162 | <div class="line"><a name="l05064"></a><span class="lineno"> 5064</span>  }</div> |
| 5163 | <div class="line"><a name="l05065"></a><span class="lineno"> 5065</span>  </div> |
| 5164 | <div class="line"><a name="l05066"></a><span class="lineno"> 5066</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 5165 | <div class="line"><a name="l05067"></a><span class="lineno"> 5067</span>  </div> |
| 5166 | <div class="line"><a name="l05068"></a><span class="lineno"> 5068</span>  <a class="code" href="structarmnn_1_1_softmax_descriptor.html">SoftmaxDescriptor</a> desc;</div> |
| 5167 | <div class="line"><a name="l05069"></a><span class="lineno"> 5069</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> outputType = outputOperand->type;</div> |
| 5168 | <div class="line"><a name="l05070"></a><span class="lineno"> 5070</span>  </div> |
| 5169 | <div class="line"><a name="l05071"></a><span class="lineno"> 5071</span>  <span class="comment">// Read beta value</span></div> |
| 5170 | <div class="line"><a name="l05072"></a><span class="lineno"> 5072</span>  <span class="keywordflow">if</span> (outputType == OperandType::TENSOR_FLOAT16)</div> |
| 5171 | <div class="line"><a name="l05073"></a><span class="lineno"> 5073</span>  {</div> |
| 5172 | <div class="line"><a name="l05074"></a><span class="lineno"> 5074</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> value;</div> |
| 5173 | <div class="line"><a name="l05075"></a><span class="lineno"> 5075</span>  </div> |
| 5174 | <div class="line"><a name="l05076"></a><span class="lineno"> 5076</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 1, OperandType::FLOAT16, value, model, data))</div> |
| 5175 | <div class="line"><a name="l05077"></a><span class="lineno"> 5077</span>  {</div> |
| 5176 | <div class="line"><a name="l05078"></a><span class="lineno"> 5078</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs %d"</span>, __func__, outputType);</div> |
| 5177 | <div class="line"><a name="l05079"></a><span class="lineno"> 5079</span>  }</div> |
| 5178 | <div class="line"><a name="l05080"></a><span class="lineno"> 5080</span>  </div> |
| 5179 | <div class="line"><a name="l05081"></a><span class="lineno"> 5081</span>  desc.<a class="code" href="structarmnn_1_1_softmax_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a> = <span class="keyword">static_cast<</span><span class="keywordtype">float</span><span class="keyword">></span>(value);</div> |
| 5180 | <div class="line"><a name="l05082"></a><span class="lineno"> 5082</span>  }</div> |
| 5181 | <div class="line"><a name="l05083"></a><span class="lineno"> 5083</span>  <span class="keywordflow">else</span></div> |
| 5182 | <div class="line"><a name="l05084"></a><span class="lineno"> 5084</span>  {</div> |
| 5183 | <div class="line"><a name="l05085"></a><span class="lineno"> 5085</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">GetInputFloat32</a>(operation, 1, desc.<a class="code" href="structarmnn_1_1_softmax_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">m_Beta</a>, model, data))</div> |
| 5184 | <div class="line"><a name="l05086"></a><span class="lineno"> 5086</span>  {</div> |
| 5185 | <div class="line"><a name="l05087"></a><span class="lineno"> 5087</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs %d"</span>, __func__, outputType);</div> |
| 5186 | <div class="line"><a name="l05088"></a><span class="lineno"> 5088</span>  }</div> |
| 5187 | <div class="line"><a name="l05089"></a><span class="lineno"> 5089</span>  }</div> |
| 5188 | <div class="line"><a name="l05090"></a><span class="lineno"> 5090</span>  </div> |
| 5189 | <div class="line"><a name="l05091"></a><span class="lineno"> 5091</span>  <span class="keywordflow">if</span> (operation.inputs.size() > 2 && !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation,</div> |
| 5190 | <div class="line"><a name="l05092"></a><span class="lineno"> 5092</span>  2,</div> |
| 5191 | <div class="line"><a name="l05093"></a><span class="lineno"> 5093</span>  OperandType::INT32,</div> |
| 5192 | <div class="line"><a name="l05094"></a><span class="lineno"> 5094</span>  desc.<a class="code" href="structarmnn_1_1_softmax_descriptor.html#a214c3636fdf0ea5bac8edb42d0e6c7f0">m_Axis</a>,</div> |
| 5193 | <div class="line"><a name="l05095"></a><span class="lineno"> 5095</span>  model,</div> |
| 5194 | <div class="line"><a name="l05096"></a><span class="lineno"> 5096</span>  data))</div> |
| 5195 | <div class="line"><a name="l05097"></a><span class="lineno"> 5097</span>  {</div> |
| 5196 | <div class="line"><a name="l05098"></a><span class="lineno"> 5098</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5197 | <div class="line"><a name="l05099"></a><span class="lineno"> 5099</span>  }</div> |
| 5198 | <div class="line"><a name="l05100"></a><span class="lineno"> 5100</span>  </div> |
| 5199 | <div class="line"><a name="l05101"></a><span class="lineno"> 5101</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5200 | <div class="line"><a name="l05102"></a><span class="lineno"> 5102</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5201 | <div class="line"><a name="l05103"></a><span class="lineno"> 5103</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 5202 | <div class="line"><a name="l05104"></a><span class="lineno"> 5104</span>  {</div> |
| 5203 | <div class="line"><a name="l05105"></a><span class="lineno"> 5105</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5204 | <div class="line"><a name="l05106"></a><span class="lineno"> 5106</span>  IsSoftmaxSupported,</div> |
| 5205 | <div class="line"><a name="l05107"></a><span class="lineno"> 5107</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5206 | <div class="line"><a name="l05108"></a><span class="lineno"> 5108</span>  isSupported,</div> |
| 5207 | <div class="line"><a name="l05109"></a><span class="lineno"> 5109</span>  setBackend,</div> |
| 5208 | <div class="line"><a name="l05110"></a><span class="lineno"> 5110</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>(),</div> |
| 5209 | <div class="line"><a name="l05111"></a><span class="lineno"> 5111</span>  outputInfo,</div> |
| 5210 | <div class="line"><a name="l05112"></a><span class="lineno"> 5112</span>  desc);</div> |
| 5211 | <div class="line"><a name="l05113"></a><span class="lineno"> 5113</span>  };</div> |
| 5212 | <div class="line"><a name="l05114"></a><span class="lineno"> 5114</span>  </div> |
| 5213 | <div class="line"><a name="l05115"></a><span class="lineno"> 5115</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5214 | <div class="line"><a name="l05116"></a><span class="lineno"> 5116</span>  {</div> |
| 5215 | <div class="line"><a name="l05117"></a><span class="lineno"> 5117</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5216 | <div class="line"><a name="l05118"></a><span class="lineno"> 5118</span>  }</div> |
| 5217 | <div class="line"><a name="l05119"></a><span class="lineno"> 5119</span>  <span class="keywordflow">else</span></div> |
| 5218 | <div class="line"><a name="l05120"></a><span class="lineno"> 5120</span>  {</div> |
| 5219 | <div class="line"><a name="l05121"></a><span class="lineno"> 5121</span>  validateFunc(outputInfo, isSupported);</div> |
| 5220 | <div class="line"><a name="l05122"></a><span class="lineno"> 5122</span>  }</div> |
| 5221 | <div class="line"><a name="l05123"></a><span class="lineno"> 5123</span>  </div> |
| 5222 | <div class="line"><a name="l05124"></a><span class="lineno"> 5124</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5223 | <div class="line"><a name="l05125"></a><span class="lineno"> 5125</span>  {</div> |
| 5224 | <div class="line"><a name="l05126"></a><span class="lineno"> 5126</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5225 | <div class="line"><a name="l05127"></a><span class="lineno"> 5127</span>  }</div> |
| 5226 | <div class="line"><a name="l05128"></a><span class="lineno"> 5128</span>  </div> |
| 5227 | <div class="line"><a name="l05129"></a><span class="lineno"> 5129</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddSoftmaxLayer(desc);</div> |
| 5228 | <div class="line"><a name="l05130"></a><span class="lineno"> 5130</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5229 | <div class="line"><a name="l05131"></a><span class="lineno"> 5131</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5230 | <div class="line"><a name="l05132"></a><span class="lineno"> 5132</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5231 | <div class="line"><a name="l05133"></a><span class="lineno"> 5133</span>  </div> |
| 5232 | <div class="line"><a name="l05134"></a><span class="lineno"> 5134</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 5233 | <div class="line"><a name="l05135"></a><span class="lineno"> 5135</span> }</div> |
| 5234 | <div class="line"><a name="l05136"></a><span class="lineno"> 5136</span>  </div> |
| 5235 | <div class="line"><a name="l05137"></a><span class="lineno"> 5137</span> <span class="keywordtype">bool</span> Converter::ConvertTanH(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5236 | <div class="line"><a name="l05138"></a><span class="lineno"> 5138</span> {</div> |
| 5237 | <div class="line"><a name="l05139"></a><span class="lineno"> 5139</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTanH()"</span>;</div> |
| 5238 | <div class="line"><a name="l05140"></a><span class="lineno"> 5140</span>  </div> |
| 5239 | <div class="line"><a name="l05141"></a><span class="lineno"> 5141</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">armnn::ActivationDescriptor</a> desc;</div> |
| 5240 | <div class="line"><a name="l05142"></a><span class="lineno"> 5142</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = <a class="code" href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaa23b68da1de2b77d74da9da2635722a3e">armnn::ActivationFunction::TanH</a>;</div> |
| 5241 | <div class="line"><a name="l05143"></a><span class="lineno"> 5143</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a017b2990003a014234f13e999dc7c689">m_A</a> = 1.0f; <span class="comment">// android nn does not support tanH parameters</span></div> |
| 5242 | <div class="line"><a name="l05144"></a><span class="lineno"> 5144</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#a28c4c9cb15f6be3499abbc46b356060b">m_B</a> = 1.0f; <span class="comment">// set to 1.0f for unity scaling</span></div> |
| 5243 | <div class="line"><a name="l05145"></a><span class="lineno"> 5145</span>  </div> |
| 5244 | <div class="line"><a name="l05146"></a><span class="lineno"> 5146</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 5245 | <div class="line"><a name="l05147"></a><span class="lineno"> 5147</span> }</div> |
| 5246 | <div class="line"><a name="l05148"></a><span class="lineno"> 5148</span>  </div> |
| 5247 | <div class="line"><a name="l05149"></a><span class="lineno"> 5149</span> <span class="keywordtype">bool</span> Converter::ConvertTile(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5248 | <div class="line"><a name="l05150"></a><span class="lineno"> 5150</span> {</div> |
| 5249 | <div class="line"><a name="l05151"></a><span class="lineno"> 5151</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTile()"</span>;</div> |
| 5250 | <div class="line"><a name="l05152"></a><span class="lineno"> 5152</span>  </div> |
| 5251 | <div class="line"><a name="l05153"></a><span class="lineno"> 5153</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5252 | <div class="line"><a name="l05154"></a><span class="lineno"> 5154</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5253 | <div class="line"><a name="l05155"></a><span class="lineno"> 5155</span>  {</div> |
| 5254 | <div class="line"><a name="l05156"></a><span class="lineno"> 5156</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5255 | <div class="line"><a name="l05157"></a><span class="lineno"> 5157</span>  }</div> |
| 5256 | <div class="line"><a name="l05158"></a><span class="lineno"> 5158</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 5257 | <div class="line"><a name="l05159"></a><span class="lineno"> 5159</span>  </div> |
| 5258 | <div class="line"><a name="l05160"></a><span class="lineno"> 5160</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputOperand = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5259 | <div class="line"><a name="l05161"></a><span class="lineno"> 5161</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| 5260 | <div class="line"><a name="l05162"></a><span class="lineno"> 5162</span>  {</div> |
| 5261 | <div class="line"><a name="l05163"></a><span class="lineno"> 5163</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has no outputs"</span>, __func__);</div> |
| 5262 | <div class="line"><a name="l05164"></a><span class="lineno"> 5164</span>  }</div> |
| 5263 | <div class="line"><a name="l05165"></a><span class="lineno"> 5165</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*outputOperand);</div> |
| 5264 | <div class="line"><a name="l05166"></a><span class="lineno"> 5166</span>  </div> |
| 5265 | <div class="line"><a name="l05167"></a><span class="lineno"> 5167</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* multiplesOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 5266 | <div class="line"><a name="l05168"></a><span class="lineno"> 5168</span>  <span class="keywordflow">if</span> (!multiplesOperand)</div> |
| 5267 | <div class="line"><a name="l05169"></a><span class="lineno"> 5169</span>  {</div> |
| 5268 | <div class="line"><a name="l05170"></a><span class="lineno"> 5170</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read input 1"</span>, __func__);</div> |
| 5269 | <div class="line"><a name="l05171"></a><span class="lineno"> 5171</span>  }</div> |
| 5270 | <div class="line"><a name="l05172"></a><span class="lineno"> 5172</span>  std::vector<int32_t> multiples;</div> |
| 5271 | <div class="line"><a name="l05173"></a><span class="lineno"> 5173</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*multiplesOperand, multiples, model, data))</div> |
| 5272 | <div class="line"><a name="l05174"></a><span class="lineno"> 5174</span>  {</div> |
| 5273 | <div class="line"><a name="l05175"></a><span class="lineno"> 5175</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Input 1 has invalid values"</span>, __func__);</div> |
| 5274 | <div class="line"><a name="l05176"></a><span class="lineno"> 5176</span>  }</div> |
| 5275 | <div class="line"><a name="l05177"></a><span class="lineno"> 5177</span>  </div> |
| 5276 | <div class="line"><a name="l05178"></a><span class="lineno"> 5178</span>  <a class="code" href="structarmnn_1_1_tile_descriptor.html">TileDescriptor</a> descriptor;</div> |
| 5277 | <div class="line"><a name="l05179"></a><span class="lineno"> 5179</span>  descriptor.<a class="code" href="structarmnn_1_1_tile_descriptor.html#a4e66396e5ed6ede8d87dc75103645def">m_Multiples</a>.assign(multiples.begin(), multiples.end());</div> |
| 5278 | <div class="line"><a name="l05180"></a><span class="lineno"> 5180</span>  </div> |
| 5279 | <div class="line"><a name="l05181"></a><span class="lineno"> 5181</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5280 | <div class="line"><a name="l05182"></a><span class="lineno"> 5182</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5281 | <div class="line"><a name="l05183"></a><span class="lineno"> 5183</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 5282 | <div class="line"><a name="l05184"></a><span class="lineno"> 5184</span>  {</div> |
| 5283 | <div class="line"><a name="l05185"></a><span class="lineno"> 5185</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5284 | <div class="line"><a name="l05186"></a><span class="lineno"> 5186</span>  IsTileSupported,</div> |
| 5285 | <div class="line"><a name="l05187"></a><span class="lineno"> 5187</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5286 | <div class="line"><a name="l05188"></a><span class="lineno"> 5188</span>  isSupported,</div> |
| 5287 | <div class="line"><a name="l05189"></a><span class="lineno"> 5189</span>  setBackend,</div> |
| 5288 | <div class="line"><a name="l05190"></a><span class="lineno"> 5190</span>  inputInfo,</div> |
| 5289 | <div class="line"><a name="l05191"></a><span class="lineno"> 5191</span>  outputInfo,</div> |
| 5290 | <div class="line"><a name="l05192"></a><span class="lineno"> 5192</span>  descriptor);</div> |
| 5291 | <div class="line"><a name="l05193"></a><span class="lineno"> 5193</span>  };</div> |
| 5292 | <div class="line"><a name="l05194"></a><span class="lineno"> 5194</span>  </div> |
| 5293 | <div class="line"><a name="l05195"></a><span class="lineno"> 5195</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5294 | <div class="line"><a name="l05196"></a><span class="lineno"> 5196</span>  {</div> |
| 5295 | <div class="line"><a name="l05197"></a><span class="lineno"> 5197</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5296 | <div class="line"><a name="l05198"></a><span class="lineno"> 5198</span>  }</div> |
| 5297 | <div class="line"><a name="l05199"></a><span class="lineno"> 5199</span>  <span class="keywordflow">else</span></div> |
| 5298 | <div class="line"><a name="l05200"></a><span class="lineno"> 5200</span>  {</div> |
| 5299 | <div class="line"><a name="l05201"></a><span class="lineno"> 5201</span>  validateFunc(outputInfo, isSupported);</div> |
| 5300 | <div class="line"><a name="l05202"></a><span class="lineno"> 5202</span>  }</div> |
| 5301 | <div class="line"><a name="l05203"></a><span class="lineno"> 5203</span>  </div> |
| 5302 | <div class="line"><a name="l05204"></a><span class="lineno"> 5204</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5303 | <div class="line"><a name="l05205"></a><span class="lineno"> 5205</span>  {</div> |
| 5304 | <div class="line"><a name="l05206"></a><span class="lineno"> 5206</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5305 | <div class="line"><a name="l05207"></a><span class="lineno"> 5207</span>  }</div> |
| 5306 | <div class="line"><a name="l05208"></a><span class="lineno"> 5208</span>  </div> |
| 5307 | <div class="line"><a name="l05209"></a><span class="lineno"> 5209</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddTileLayer(descriptor);</div> |
| 5308 | <div class="line"><a name="l05210"></a><span class="lineno"> 5210</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5309 | <div class="line"><a name="l05211"></a><span class="lineno"> 5211</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5310 | <div class="line"><a name="l05212"></a><span class="lineno"> 5212</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5311 | <div class="line"><a name="l05213"></a><span class="lineno"> 5213</span>  </div> |
| 5312 | <div class="line"><a name="l05214"></a><span class="lineno"> 5214</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 5313 | <div class="line"><a name="l05215"></a><span class="lineno"> 5215</span> }</div> |
| 5314 | <div class="line"><a name="l05216"></a><span class="lineno"> 5216</span>  </div> |
| 5315 | <div class="line"><a name="l05217"></a><span class="lineno"> 5217</span> <span class="keywordtype">bool</span> Converter::ConvertTransposeConv2d(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5316 | <div class="line"><a name="l05218"></a><span class="lineno"> 5218</span> {</div> |
| 5317 | <div class="line"><a name="l05219"></a><span class="lineno"> 5219</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTransposeConv2d()"</span>;</div> |
| 5318 | <div class="line"><a name="l05220"></a><span class="lineno"> 5220</span>  </div> |
| 5319 | <div class="line"><a name="l05221"></a><span class="lineno"> 5221</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5320 | <div class="line"><a name="l05222"></a><span class="lineno"> 5222</span>  </div> |
| 5321 | <div class="line"><a name="l05223"></a><span class="lineno"> 5223</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5322 | <div class="line"><a name="l05224"></a><span class="lineno"> 5224</span>  {</div> |
| 5323 | <div class="line"><a name="l05225"></a><span class="lineno"> 5225</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5324 | <div class="line"><a name="l05226"></a><span class="lineno"> 5226</span>  }</div> |
| 5325 | <div class="line"><a name="l05227"></a><span class="lineno"> 5227</span>  </div> |
| 5326 | <div class="line"><a name="l05228"></a><span class="lineno"> 5228</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5327 | <div class="line"><a name="l05229"></a><span class="lineno"> 5229</span>  </div> |
| 5328 | <div class="line"><a name="l05230"></a><span class="lineno"> 5230</span>  <span class="keywordflow">if</span> (!output)</div> |
| 5329 | <div class="line"><a name="l05231"></a><span class="lineno"> 5231</span>  {</div> |
| 5330 | <div class="line"><a name="l05232"></a><span class="lineno"> 5232</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 5331 | <div class="line"><a name="l05233"></a><span class="lineno"> 5233</span>  }</div> |
| 5332 | <div class="line"><a name="l05234"></a><span class="lineno"> 5234</span>  </div> |
| 5333 | <div class="line"><a name="l05235"></a><span class="lineno"> 5235</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 5334 | <div class="line"><a name="l05236"></a><span class="lineno"> 5236</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 5335 | <div class="line"><a name="l05237"></a><span class="lineno"> 5237</span>  </div> |
| 5336 | <div class="line"><a name="l05238"></a><span class="lineno"> 5238</span>  <span class="comment">// ArmNN does not currently support non-fixed weights or bias</span></div> |
| 5337 | <div class="line"><a name="l05239"></a><span class="lineno"> 5239</span>  <span class="comment">// Find the shape of the weights tensor. In AndroidNN this will be [ 1, H, W, I * M ]</span></div> |
| 5338 | <div class="line"><a name="l05240"></a><span class="lineno"> 5240</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* weightsOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 5339 | <div class="line"><a name="l05241"></a><span class="lineno"> 5241</span>  </div> |
| 5340 | <div class="line"><a name="l05242"></a><span class="lineno"> 5242</span>  <span class="keywordflow">if</span> (weightsOperand == <span class="keyword">nullptr</span>)</div> |
| 5341 | <div class="line"><a name="l05243"></a><span class="lineno"> 5243</span>  {</div> |
| 5342 | <div class="line"><a name="l05244"></a><span class="lineno"> 5244</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operand is invalid"</span>, __func__);</div> |
| 5343 | <div class="line"><a name="l05245"></a><span class="lineno"> 5245</span>  }</div> |
| 5344 | <div class="line"><a name="l05246"></a><span class="lineno"> 5246</span>  <a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html">TransposeConvolution2dDescriptor</a> desc;</div> |
| 5345 | <div class="line"><a name="l05247"></a><span class="lineno"> 5247</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = DataLayout::NHWC;</div> |
| 5346 | <div class="line"><a name="l05248"></a><span class="lineno"> 5248</span>  </div> |
| 5347 | <div class="line"><a name="l05249"></a><span class="lineno"> 5249</span>  <span class="comment">// Determine whether padding is implicit or explicit</span></div> |
| 5348 | <div class="line"><a name="l05250"></a><span class="lineno"> 5250</span>  <span class="keywordtype">bool</span> implicitPadding = operation.inputs.size() == 9;</div> |
| 5349 | <div class="line"><a name="l05251"></a><span class="lineno"> 5251</span>  </div> |
| 5350 | <div class="line"><a name="l05252"></a><span class="lineno"> 5252</span>  <span class="keywordflow">if</span> (implicitPadding )</div> |
| 5351 | <div class="line"><a name="l05253"></a><span class="lineno"> 5253</span>  {</div> |
| 5352 | <div class="line"><a name="l05254"></a><span class="lineno"> 5254</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 8, model, data);</div> |
| 5353 | <div class="line"><a name="l05255"></a><span class="lineno"> 5255</span>  }</div> |
| 5354 | <div class="line"><a name="l05256"></a><span class="lineno"> 5256</span>  <span class="keywordflow">else</span></div> |
| 5355 | <div class="line"><a name="l05257"></a><span class="lineno"> 5257</span>  {</div> |
| 5356 | <div class="line"><a name="l05258"></a><span class="lineno"> 5258</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">OptionalDataLayout</a>(operation, 10, model, data);</div> |
| 5357 | <div class="line"><a name="l05259"></a><span class="lineno"> 5259</span>  }</div> |
| 5358 | <div class="line"><a name="l05260"></a><span class="lineno"> 5260</span>  </div> |
| 5359 | <div class="line"><a name="l05261"></a><span class="lineno"> 5261</span>  <a class="code" href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a> dataLayoutIndexed(desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a>);</div> |
| 5360 | <div class="line"><a name="l05262"></a><span class="lineno"> 5262</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> widthIndex = dataLayoutIndexed.GetWidthIndex();</div> |
| 5361 | <div class="line"><a name="l05263"></a><span class="lineno"> 5263</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> heightIndex = dataLayoutIndexed.GetHeightIndex();</div> |
| 5362 | <div class="line"><a name="l05264"></a><span class="lineno"> 5264</span>  </div> |
| 5363 | <div class="line"><a name="l05265"></a><span class="lineno"> 5265</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_permutation_vector.html">PermutationVector</a> OHWIToOIHW = {0, 2, 3, 1};</div> |
| 5364 | <div class="line"><a name="l05266"></a><span class="lineno"> 5266</span>  </div> |
| 5365 | <div class="line"><a name="l05267"></a><span class="lineno"> 5267</span>  <span class="comment">// The shape of the weight is [depth_out, filter_height, filter_width, depth_in].</span></div> |
| 5366 | <div class="line"><a name="l05268"></a><span class="lineno"> 5268</span>  <span class="comment">// We have to permute it to OIHW if the data layout is NCHW.</span></div> |
| 5367 | <div class="line"><a name="l05269"></a><span class="lineno"> 5269</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> weightsPin = (desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> == DataLayout::NCHW) ?</div> |
| 5368 | <div class="line"><a name="l05270"></a><span class="lineno"> 5270</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 1,</div> |
| 5369 | <div class="line"><a name="l05271"></a><span class="lineno"> 5271</span>  model, data, OHWIToOIHW) :</div> |
| 5370 | <div class="line"><a name="l05272"></a><span class="lineno"> 5272</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 1, model, data);</div> |
| 5371 | <div class="line"><a name="l05273"></a><span class="lineno"> 5273</span>  </div> |
| 5372 | <div class="line"><a name="l05274"></a><span class="lineno"> 5274</span>  <span class="comment">// Bias is a 1D tensor</span></div> |
| 5373 | <div class="line"><a name="l05275"></a><span class="lineno"> 5275</span>  <span class="keyword">const</span> <a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html">ConstTensorPin</a> biasPin =</div> |
| 5374 | <div class="line"><a name="l05276"></a><span class="lineno"> 5276</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation, 2, model, data);</div> |
| 5375 | <div class="line"><a name="l05277"></a><span class="lineno"> 5277</span>  </div> |
| 5376 | <div class="line"><a name="l05278"></a><span class="lineno"> 5278</span>  <span class="keywordflow">if</span> (!weightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5377 | <div class="line"><a name="l05279"></a><span class="lineno"> 5279</span>  {</div> |
| 5378 | <div class="line"><a name="l05280"></a><span class="lineno"> 5280</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid weights"</span>, __func__);</div> |
| 5379 | <div class="line"><a name="l05281"></a><span class="lineno"> 5281</span>  }</div> |
| 5380 | <div class="line"><a name="l05282"></a><span class="lineno"> 5282</span>  </div> |
| 5381 | <div class="line"><a name="l05283"></a><span class="lineno"> 5283</span>  <span class="keywordflow">if</span> (!biasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5382 | <div class="line"><a name="l05284"></a><span class="lineno"> 5284</span>  {</div> |
| 5383 | <div class="line"><a name="l05285"></a><span class="lineno"> 5285</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid biases"</span>, __func__);</div> |
| 5384 | <div class="line"><a name="l05286"></a><span class="lineno"> 5286</span>  }</div> |
| 5385 | <div class="line"><a name="l05287"></a><span class="lineno"> 5287</span>  </div> |
| 5386 | <div class="line"><a name="l05288"></a><span class="lineno"> 5288</span>  <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> weights = weightsPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a971023b9874b0d773e8aa4c6d6744a2b">GetConstTensor</a>();</div> |
| 5387 | <div class="line"><a name="l05289"></a><span class="lineno"> 5289</span>  <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> bias = biasPin.<a class="code" href="classarmnn__driver_1_1_const_tensor_pin.html#a971023b9874b0d773e8aa4c6d6744a2b">GetConstTensor</a>();</div> |
| 5388 | <div class="line"><a name="l05290"></a><span class="lineno"> 5290</span>  SanitizeBiasQuantizationScale(bias.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>(), weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>(), inputInfo);</div> |
| 5389 | <div class="line"><a name="l05291"></a><span class="lineno"> 5291</span>  </div> |
| 5390 | <div class="line"><a name="l05292"></a><span class="lineno"> 5292</span>  ActivationFn activation;</div> |
| 5391 | <div class="line"><a name="l05293"></a><span class="lineno"> 5293</span>  </div> |
| 5392 | <div class="line"><a name="l05294"></a><span class="lineno"> 5294</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| 5393 | <div class="line"><a name="l05295"></a><span class="lineno"> 5295</span>  {</div> |
| 5394 | <div class="line"><a name="l05296"></a><span class="lineno"> 5296</span>  int32_t strideX{0};</div> |
| 5395 | <div class="line"><a name="l05297"></a><span class="lineno"> 5297</span>  int32_t strideY{0};</div> |
| 5396 | <div class="line"><a name="l05298"></a><span class="lineno"> 5298</span>  int32_t padLeft{0};</div> |
| 5397 | <div class="line"><a name="l05299"></a><span class="lineno"> 5299</span>  int32_t padRight{0};</div> |
| 5398 | <div class="line"><a name="l05300"></a><span class="lineno"> 5300</span>  int32_t padTop{0};</div> |
| 5399 | <div class="line"><a name="l05301"></a><span class="lineno"> 5301</span>  int32_t padBottom{0};</div> |
| 5400 | <div class="line"><a name="l05302"></a><span class="lineno"> 5302</span>  </div> |
| 5401 | <div class="line"><a name="l05303"></a><span class="lineno"> 5303</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| 5402 | <div class="line"><a name="l05304"></a><span class="lineno"> 5304</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa9ef723ace8562b9636e8387a120fbde">GetInputPaddingScheme</a>(operation, 4, paddingScheme, model, data) ||</div> |
| 5403 | <div class="line"><a name="l05305"></a><span class="lineno"> 5305</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, strideX, model, data) ||</div> |
| 5404 | <div class="line"><a name="l05306"></a><span class="lineno"> 5306</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 6, OperandType::INT32, strideY, model, data) ||</div> |
| 5405 | <div class="line"><a name="l05307"></a><span class="lineno"> 5307</span>  !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 7, activation, model, data))</div> |
| 5406 | <div class="line"><a name="l05308"></a><span class="lineno"> 5308</span>  {</div> |
| 5407 | <div class="line"><a name="l05309"></a><span class="lineno"> 5309</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (implicit padding)"</span>, __func__);</div> |
| 5408 | <div class="line"><a name="l05310"></a><span class="lineno"> 5310</span>  }</div> |
| 5409 | <div class="line"><a name="l05311"></a><span class="lineno"> 5311</span>  </div> |
| 5410 | <div class="line"><a name="l05312"></a><span class="lineno"> 5312</span>  <span class="keyword">const</span> uint32_t kernelX = weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[widthIndex];</div> |
| 5411 | <div class="line"><a name="l05313"></a><span class="lineno"> 5313</span>  <span class="keyword">const</span> uint32_t kernelY = weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[heightIndex];</div> |
| 5412 | <div class="line"><a name="l05314"></a><span class="lineno"> 5314</span>  </div> |
| 5413 | <div class="line"><a name="l05315"></a><span class="lineno"> 5315</span>  <span class="comment">// If output shape has been specified as a parameter then extract it and make it available.</span></div> |
| 5414 | <div class="line"><a name="l05316"></a><span class="lineno"> 5316</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* outputShapeOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 3, model, <span class="keyword">false</span>);</div> |
| 5415 | <div class="line"><a name="l05317"></a><span class="lineno"> 5317</span>  std::vector<int32_t> outputShape;</div> |
| 5416 | <div class="line"><a name="l05318"></a><span class="lineno"> 5318</span>  <span class="keywordflow">if</span> ((outputShapeOperand) && (<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*outputShapeOperand, outputShape, model, data)))</div> |
| 5417 | <div class="line"><a name="l05319"></a><span class="lineno"> 5319</span>  {</div> |
| 5418 | <div class="line"><a name="l05320"></a><span class="lineno"> 5320</span>  <span class="comment">// Change from signed to unsigned int to store in TransposeConvolution2dDescriptor.</span></div> |
| 5419 | <div class="line"><a name="l05321"></a><span class="lineno"> 5321</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> dimension : outputShape)</div> |
| 5420 | <div class="line"><a name="l05322"></a><span class="lineno"> 5322</span>  {</div> |
| 5421 | <div class="line"><a name="l05323"></a><span class="lineno"> 5323</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a78f5b213a603b55d0fabea389e89f203">m_OutputShape</a>.push_back(<span class="keyword">static_cast<</span><span class="keywordtype">unsigned</span> <span class="keywordtype">int</span><span class="keyword">></span>(dimension));</div> |
| 5422 | <div class="line"><a name="l05324"></a><span class="lineno"> 5324</span>  }</div> |
| 5423 | <div class="line"><a name="l05325"></a><span class="lineno"> 5325</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a9bc46a2ad56319786bad30be9ab2dc6a">m_OutputShapeEnabled</a> = <span class="keyword">true</span>;</div> |
| 5424 | <div class="line"><a name="l05326"></a><span class="lineno"> 5326</span>  }</div> |
| 5425 | <div class="line"><a name="l05327"></a><span class="lineno"> 5327</span>  </div> |
| 5426 | <div class="line"><a name="l05328"></a><span class="lineno"> 5328</span>  uint32_t outputX;</div> |
| 5427 | <div class="line"><a name="l05329"></a><span class="lineno"> 5329</span>  uint32_t outputY;</div> |
| 5428 | <div class="line"><a name="l05330"></a><span class="lineno"> 5330</span>  </div> |
| 5429 | <div class="line"><a name="l05331"></a><span class="lineno"> 5331</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5430 | <div class="line"><a name="l05332"></a><span class="lineno"> 5332</span>  {</div> |
| 5431 | <div class="line"><a name="l05333"></a><span class="lineno"> 5333</span>  <span class="keywordflow">if</span> (outputShape.size() == 0)</div> |
| 5432 | <div class="line"><a name="l05334"></a><span class="lineno"> 5334</span>  {</div> |
| 5433 | <div class="line"><a name="l05335"></a><span class="lineno"> 5335</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Padding sizes cannot be inferred"</span>, __func__);</div> |
| 5434 | <div class="line"><a name="l05336"></a><span class="lineno"> 5336</span>  }</div> |
| 5435 | <div class="line"><a name="l05337"></a><span class="lineno"> 5337</span>  </div> |
| 5436 | <div class="line"><a name="l05338"></a><span class="lineno"> 5338</span>  outputX = outputShape[widthIndex];</div> |
| 5437 | <div class="line"><a name="l05339"></a><span class="lineno"> 5339</span>  outputY = outputShape[heightIndex];</div> |
| 5438 | <div class="line"><a name="l05340"></a><span class="lineno"> 5340</span>  }</div> |
| 5439 | <div class="line"><a name="l05341"></a><span class="lineno"> 5341</span>  <span class="keywordflow">else</span></div> |
| 5440 | <div class="line"><a name="l05342"></a><span class="lineno"> 5342</span>  {</div> |
| 5441 | <div class="line"><a name="l05343"></a><span class="lineno"> 5343</span>  outputX = outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[widthIndex];</div> |
| 5442 | <div class="line"><a name="l05344"></a><span class="lineno"> 5344</span>  outputY = outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[heightIndex];</div> |
| 5443 | <div class="line"><a name="l05345"></a><span class="lineno"> 5345</span>  }</div> |
| 5444 | <div class="line"><a name="l05346"></a><span class="lineno"> 5346</span>  </div> |
| 5445 | <div class="line"><a name="l05347"></a><span class="lineno"> 5347</span>  CalcPaddingTransposeConv(outputX, kernelX, strideX, padLeft, padRight, paddingScheme);</div> |
| 5446 | <div class="line"><a name="l05348"></a><span class="lineno"> 5348</span>  CalcPaddingTransposeConv(outputY, kernelY, strideY, padTop, padBottom, paddingScheme);</div> |
| 5447 | <div class="line"><a name="l05349"></a><span class="lineno"> 5349</span>  </div> |
| 5448 | <div class="line"><a name="l05350"></a><span class="lineno"> 5350</span>  <span class="comment">// NOTE: The Android NN API allows for negative padding values in TransposeConv2d,</span></div> |
| 5449 | <div class="line"><a name="l05351"></a><span class="lineno"> 5351</span>  <span class="comment">// but Arm NN only supports values >= 0</span></div> |
| 5450 | <div class="line"><a name="l05352"></a><span class="lineno"> 5352</span>  <span class="keywordflow">if</span> (padLeft < 0 || padRight < 0 || padTop < 0 || padBottom < 0)</div> |
| 5451 | <div class="line"><a name="l05353"></a><span class="lineno"> 5353</span>  {</div> |
| 5452 | <div class="line"><a name="l05354"></a><span class="lineno"> 5354</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Negative padding values are not supported"</span>, __func__);</div> |
| 5453 | <div class="line"><a name="l05355"></a><span class="lineno"> 5355</span>  }</div> |
| 5454 | <div class="line"><a name="l05356"></a><span class="lineno"> 5356</span>  </div> |
| 5455 | <div class="line"><a name="l05357"></a><span class="lineno"> 5357</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a> = armnn::numeric_cast<uint32_t>(strideX);</div> |
| 5456 | <div class="line"><a name="l05358"></a><span class="lineno"> 5358</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a> = armnn::numeric_cast<uint32_t>(strideY);</div> |
| 5457 | <div class="line"><a name="l05359"></a><span class="lineno"> 5359</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a> = armnn::numeric_cast<uint32_t>(padLeft);</div> |
| 5458 | <div class="line"><a name="l05360"></a><span class="lineno"> 5360</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a> = armnn::numeric_cast<uint32_t>(padRight);</div> |
| 5459 | <div class="line"><a name="l05361"></a><span class="lineno"> 5361</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a> = armnn::numeric_cast<uint32_t>(padTop);</div> |
| 5460 | <div class="line"><a name="l05362"></a><span class="lineno"> 5362</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a> = armnn::numeric_cast<uint32_t>(padBottom);</div> |
| 5461 | <div class="line"><a name="l05363"></a><span class="lineno"> 5363</span>  }</div> |
| 5462 | <div class="line"><a name="l05364"></a><span class="lineno"> 5364</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (operation.inputs.size() == 11)</div> |
| 5463 | <div class="line"><a name="l05365"></a><span class="lineno"> 5365</span>  {</div> |
| 5464 | <div class="line"><a name="l05366"></a><span class="lineno"> 5366</span>  <span class="comment">// explicit padding</span></div> |
| 5465 | <div class="line"><a name="l05367"></a><span class="lineno"> 5367</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 3, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">m_PadLeft</a>, model, data) ||</div> |
| 5466 | <div class="line"><a name="l05368"></a><span class="lineno"> 5368</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 4, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">m_PadRight</a>, model, data) ||</div> |
| 5467 | <div class="line"><a name="l05369"></a><span class="lineno"> 5369</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 5, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">m_PadTop</a>, model, data) ||</div> |
| 5468 | <div class="line"><a name="l05370"></a><span class="lineno"> 5370</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 6, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">m_PadBottom</a>, model, data) ||</div> |
| 5469 | <div class="line"><a name="l05371"></a><span class="lineno"> 5371</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 7, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">m_StrideX</a>, model, data) ||</div> |
| 5470 | <div class="line"><a name="l05372"></a><span class="lineno"> 5372</span>  !<a class="code" href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">GetInputScalar</a>(operation, 8, OperandType::INT32, desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">m_StrideY</a>, model, data) ||</div> |
| 5471 | <div class="line"><a name="l05373"></a><span class="lineno"> 5373</span>  !<a class="code" href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">GetInputActivationFunction</a>(operation, 9, activation, model, data))</div> |
| 5472 | <div class="line"><a name="l05374"></a><span class="lineno"> 5374</span>  {</div> |
| 5473 | <div class="line"><a name="l05375"></a><span class="lineno"> 5375</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs (explicit padding)"</span>, __func__);</div> |
| 5474 | <div class="line"><a name="l05376"></a><span class="lineno"> 5376</span>  }</div> |
| 5475 | <div class="line"><a name="l05377"></a><span class="lineno"> 5377</span>  }</div> |
| 5476 | <div class="line"><a name="l05378"></a><span class="lineno"> 5378</span>  <span class="keywordflow">else</span></div> |
| 5477 | <div class="line"><a name="l05379"></a><span class="lineno"> 5379</span>  {</div> |
| 5478 | <div class="line"><a name="l05380"></a><span class="lineno"> 5380</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Unsupported number of operation inputs"</span>, __func__);</div> |
| 5479 | <div class="line"><a name="l05381"></a><span class="lineno"> 5381</span>  }</div> |
| 5480 | <div class="line"><a name="l05382"></a><span class="lineno"> 5382</span>  </div> |
| 5481 | <div class="line"><a name="l05383"></a><span class="lineno"> 5383</span>  desc.<a class="code" href="structarmnn_1_1_transpose_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">m_BiasEnabled</a> = <span class="keyword">true</span>;</div> |
| 5482 | <div class="line"><a name="l05384"></a><span class="lineno"> 5384</span>  <a class="code" href="classarmnn_1_1_optional.html">Optional<TensorInfo></a> biases(bias.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>());</div> |
| 5483 | <div class="line"><a name="l05385"></a><span class="lineno"> 5385</span>  </div> |
| 5484 | <div class="line"><a name="l05386"></a><span class="lineno"> 5386</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5485 | <div class="line"><a name="l05387"></a><span class="lineno"> 5387</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5486 | <div class="line"><a name="l05388"></a><span class="lineno"> 5388</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 5487 | <div class="line"><a name="l05389"></a><span class="lineno"> 5389</span>  {</div> |
| 5488 | <div class="line"><a name="l05390"></a><span class="lineno"> 5390</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5489 | <div class="line"><a name="l05391"></a><span class="lineno"> 5391</span>  IsTransposeConvolution2dSupported,</div> |
| 5490 | <div class="line"><a name="l05392"></a><span class="lineno"> 5392</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5491 | <div class="line"><a name="l05393"></a><span class="lineno"> 5393</span>  isSupported,</div> |
| 5492 | <div class="line"><a name="l05394"></a><span class="lineno"> 5394</span>  setBackend,</div> |
| 5493 | <div class="line"><a name="l05395"></a><span class="lineno"> 5395</span>  inputInfo,</div> |
| 5494 | <div class="line"><a name="l05396"></a><span class="lineno"> 5396</span>  outputInfo,</div> |
| 5495 | <div class="line"><a name="l05397"></a><span class="lineno"> 5397</span>  desc,</div> |
| 5496 | <div class="line"><a name="l05398"></a><span class="lineno"> 5398</span>  weights.<a class="code" href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">GetInfo</a>(),</div> |
| 5497 | <div class="line"><a name="l05399"></a><span class="lineno"> 5399</span>  biases);</div> |
| 5498 | <div class="line"><a name="l05400"></a><span class="lineno"> 5400</span>  };</div> |
| 5499 | <div class="line"><a name="l05401"></a><span class="lineno"> 5401</span>  </div> |
| 5500 | <div class="line"><a name="l05402"></a><span class="lineno"> 5402</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5501 | <div class="line"><a name="l05403"></a><span class="lineno"> 5403</span>  {</div> |
| 5502 | <div class="line"><a name="l05404"></a><span class="lineno"> 5404</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5503 | <div class="line"><a name="l05405"></a><span class="lineno"> 5405</span>  }</div> |
| 5504 | <div class="line"><a name="l05406"></a><span class="lineno"> 5406</span>  <span class="keywordflow">else</span></div> |
| 5505 | <div class="line"><a name="l05407"></a><span class="lineno"> 5407</span>  {</div> |
| 5506 | <div class="line"><a name="l05408"></a><span class="lineno"> 5408</span>  validateFunc(outputInfo, isSupported);</div> |
| 5507 | <div class="line"><a name="l05409"></a><span class="lineno"> 5409</span>  }</div> |
| 5508 | <div class="line"><a name="l05410"></a><span class="lineno"> 5410</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5509 | <div class="line"><a name="l05411"></a><span class="lineno"> 5411</span>  {</div> |
| 5510 | <div class="line"><a name="l05412"></a><span class="lineno"> 5412</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5511 | <div class="line"><a name="l05413"></a><span class="lineno"> 5413</span>  }</div> |
| 5512 | <div class="line"><a name="l05414"></a><span class="lineno"> 5414</span>  </div> |
| 5513 | <div class="line"><a name="l05415"></a><span class="lineno"> 5415</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* startLayer =</div> |
| 5514 | <div class="line"><a name="l05416"></a><span class="lineno"> 5416</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddTransposeConvolution2dLayer(desc, weights, <a class="code" href="classarmnn_1_1_optional.html">Optional<ConstTensor></a>(bias));</div> |
| 5515 | <div class="line"><a name="l05417"></a><span class="lineno"> 5417</span>  startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5516 | <div class="line"><a name="l05418"></a><span class="lineno"> 5418</span>  <span class="keywordflow">if</span> (!startLayer)</div> |
| 5517 | <div class="line"><a name="l05419"></a><span class="lineno"> 5419</span>  {</div> |
| 5518 | <div class="line"><a name="l05420"></a><span class="lineno"> 5420</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: AddTransposeConvolution2dLayer failed"</span>, __func__);</div> |
| 5519 | <div class="line"><a name="l05421"></a><span class="lineno"> 5421</span>  }</div> |
| 5520 | <div class="line"><a name="l05422"></a><span class="lineno"> 5422</span>  </div> |
| 5521 | <div class="line"><a name="l05423"></a><span class="lineno"> 5423</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(startLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5522 | <div class="line"><a name="l05424"></a><span class="lineno"> 5424</span>  </div> |
| 5523 | <div class="line"><a name="l05425"></a><span class="lineno"> 5425</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *startLayer, model,</div> |
| 5524 | <div class="line"><a name="l05426"></a><span class="lineno"> 5426</span>  data, <span class="keyword">nullptr</span>, validateFunc, activation);</div> |
| 5525 | <div class="line"><a name="l05427"></a><span class="lineno"> 5427</span> }</div> |
| 5526 | <div class="line"><a name="l05428"></a><span class="lineno"> 5428</span>  </div> |
| 5527 | <div class="line"><a name="l05429"></a><span class="lineno"> 5429</span> <span class="keywordtype">bool</span> Converter::ConvertSqrt(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5528 | <div class="line"><a name="l05430"></a><span class="lineno"> 5430</span> {</div> |
| 5529 | <div class="line"><a name="l05431"></a><span class="lineno"> 5431</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSqrt()"</span>;</div> |
| 5530 | <div class="line"><a name="l05432"></a><span class="lineno"> 5432</span>  <a class="code" href="structarmnn_1_1_activation_descriptor.html">ActivationDescriptor</a> desc;</div> |
| 5531 | <div class="line"><a name="l05433"></a><span class="lineno"> 5433</span>  desc.<a class="code" href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">m_Function</a> = ActivationFunction::Sqrt;</div> |
| 5532 | <div class="line"><a name="l05434"></a><span class="lineno"> 5434</span>  </div> |
| 5533 | <div class="line"><a name="l05435"></a><span class="lineno"> 5435</span>  <a class="code" href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">return ::ConvertToActivation</a>(operation, __func__, desc, model, data);</div> |
| 5534 | <div class="line"><a name="l05436"></a><span class="lineno"> 5436</span> }</div> |
| 5535 | <div class="line"><a name="l05437"></a><span class="lineno"> 5437</span>  </div> |
| 5536 | <div class="line"><a name="l05438"></a><span class="lineno"> 5438</span> <span class="keywordtype">bool</span> Converter::ConvertSqueeze(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5537 | <div class="line"><a name="l05439"></a><span class="lineno"> 5439</span> {</div> |
| 5538 | <div class="line"><a name="l05440"></a><span class="lineno"> 5440</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSqueeze()"</span>;</div> |
| 5539 | <div class="line"><a name="l05441"></a><span class="lineno"> 5441</span>  </div> |
| 5540 | <div class="line"><a name="l05442"></a><span class="lineno"> 5442</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5541 | <div class="line"><a name="l05443"></a><span class="lineno"> 5443</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5542 | <div class="line"><a name="l05444"></a><span class="lineno"> 5444</span>  {</div> |
| 5543 | <div class="line"><a name="l05445"></a><span class="lineno"> 5445</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5544 | <div class="line"><a name="l05446"></a><span class="lineno"> 5446</span>  }</div> |
| 5545 | <div class="line"><a name="l05447"></a><span class="lineno"> 5447</span>  </div> |
| 5546 | <div class="line"><a name="l05448"></a><span class="lineno"> 5448</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 5547 | <div class="line"><a name="l05449"></a><span class="lineno"> 5449</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 5548 | <div class="line"><a name="l05450"></a><span class="lineno"> 5450</span>  <span class="keywordflow">if</span> (rank > 4)</div> |
| 5549 | <div class="line"><a name="l05451"></a><span class="lineno"> 5451</span>  {</div> |
| 5550 | <div class="line"><a name="l05452"></a><span class="lineno"> 5452</span>  Fail(<span class="stringliteral">"%s: Inputs with rank greater than 4 are not supported"</span>, __func__);</div> |
| 5551 | <div class="line"><a name="l05453"></a><span class="lineno"> 5453</span>  }</div> |
| 5552 | <div class="line"><a name="l05454"></a><span class="lineno"> 5454</span>  </div> |
| 5553 | <div class="line"><a name="l05455"></a><span class="lineno"> 5455</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5554 | <div class="line"><a name="l05456"></a><span class="lineno"> 5456</span>  <span class="keywordflow">if</span> (!output)</div> |
| 5555 | <div class="line"><a name="l05457"></a><span class="lineno"> 5457</span>  {</div> |
| 5556 | <div class="line"><a name="l05458"></a><span class="lineno"> 5458</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 5557 | <div class="line"><a name="l05459"></a><span class="lineno"> 5459</span>  }</div> |
| 5558 | <div class="line"><a name="l05460"></a><span class="lineno"> 5460</span>  </div> |
| 5559 | <div class="line"><a name="l05461"></a><span class="lineno"> 5461</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(<a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output)) && !(<a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>()))</div> |
| 5560 | <div class="line"><a name="l05462"></a><span class="lineno"> 5462</span>  {</div> |
| 5561 | <div class="line"><a name="l05463"></a><span class="lineno"> 5463</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Dynamic output tensors are not supported"</span>, __func__);</div> |
| 5562 | <div class="line"><a name="l05464"></a><span class="lineno"> 5464</span>  }</div> |
| 5563 | <div class="line"><a name="l05465"></a><span class="lineno"> 5465</span>  </div> |
| 5564 | <div class="line"><a name="l05466"></a><span class="lineno"> 5466</span>  <span class="comment">// NOTE: Axis is an optional parameter to SQUEEZE, therefore we do not want to generate a failure</span></div> |
| 5565 | <div class="line"><a name="l05467"></a><span class="lineno"> 5467</span>  <span class="comment">// if the operand index is out of bounds.</span></div> |
| 5566 | <div class="line"><a name="l05468"></a><span class="lineno"> 5468</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* axisOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model, <span class="keyword">false</span>);</div> |
| 5567 | <div class="line"><a name="l05469"></a><span class="lineno"> 5469</span>  </div> |
| 5568 | <div class="line"><a name="l05470"></a><span class="lineno"> 5470</span>  <span class="keyword">const</span> uint32_t dimensionSequence[] = { 0, 1, 2, 3 };</div> |
| 5569 | <div class="line"><a name="l05471"></a><span class="lineno"> 5471</span>  </div> |
| 5570 | <div class="line"><a name="l05472"></a><span class="lineno"> 5472</span>  std::vector<int32_t> axis;</div> |
| 5571 | <div class="line"><a name="l05473"></a><span class="lineno"> 5473</span>  <span class="keywordflow">if</span> (!axisOperand)</div> |
| 5572 | <div class="line"><a name="l05474"></a><span class="lineno"> 5474</span>  {</div> |
| 5573 | <div class="line"><a name="l05475"></a><span class="lineno"> 5475</span>  axis.assign(dimensionSequence,</div> |
| 5574 | <div class="line"><a name="l05476"></a><span class="lineno"> 5476</span>  dimensionSequence + rank);</div> |
| 5575 | <div class="line"><a name="l05477"></a><span class="lineno"> 5477</span>  }</div> |
| 5576 | <div class="line"><a name="l05478"></a><span class="lineno"> 5478</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*axisOperand, axis, model, data))</div> |
| 5577 | <div class="line"><a name="l05479"></a><span class="lineno"> 5479</span>  {</div> |
| 5578 | <div class="line"><a name="l05480"></a><span class="lineno"> 5480</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has an invalid or unsupported axis operand"</span>, __func__);</div> |
| 5579 | <div class="line"><a name="l05481"></a><span class="lineno"> 5481</span>  }</div> |
| 5580 | <div class="line"><a name="l05482"></a><span class="lineno"> 5482</span>  </div> |
| 5581 | <div class="line"><a name="l05483"></a><span class="lineno"> 5483</span>  std::vector<uint32_t> outputDims;</div> |
| 5582 | <div class="line"><a name="l05484"></a><span class="lineno"> 5484</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i < rank; i++)</div> |
| 5583 | <div class="line"><a name="l05485"></a><span class="lineno"> 5485</span>  {</div> |
| 5584 | <div class="line"><a name="l05486"></a><span class="lineno"> 5486</span>  <span class="keywordtype">bool</span> skipSqueeze = (std::find(axis.begin(), axis.end(), i) == axis.end());</div> |
| 5585 | <div class="line"><a name="l05487"></a><span class="lineno"> 5487</span>  <span class="keyword">auto</span> currentDimension = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>()[i];</div> |
| 5586 | <div class="line"><a name="l05488"></a><span class="lineno"> 5488</span>  <span class="keywordflow">if</span> (skipSqueeze || currentDimension != 1)</div> |
| 5587 | <div class="line"><a name="l05489"></a><span class="lineno"> 5489</span>  {</div> |
| 5588 | <div class="line"><a name="l05490"></a><span class="lineno"> 5490</span>  outputDims.push_back(currentDimension);</div> |
| 5589 | <div class="line"><a name="l05491"></a><span class="lineno"> 5491</span>  }</div> |
| 5590 | <div class="line"><a name="l05492"></a><span class="lineno"> 5492</span>  }</div> |
| 5591 | <div class="line"><a name="l05493"></a><span class="lineno"> 5493</span>  </div> |
| 5592 | <div class="line"><a name="l05494"></a><span class="lineno"> 5494</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> outShape = <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a>(outputDims.size(), outputDims.data());</div> |
| 5593 | <div class="line"><a name="l05495"></a><span class="lineno"> 5495</span>  </div> |
| 5594 | <div class="line"><a name="l05496"></a><span class="lineno"> 5496</span>  <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> outputInfo = inputInfo;</div> |
| 5595 | <div class="line"><a name="l05497"></a><span class="lineno"> 5497</span>  outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#abe8889e8150beef5fd204b2d87b49298">SetShape</a>(outShape);</div> |
| 5596 | <div class="line"><a name="l05498"></a><span class="lineno"> 5498</span>  </div> |
| 5597 | <div class="line"><a name="l05499"></a><span class="lineno"> 5499</span>  <a class="code" href="structarmnn_1_1_reshape_descriptor.html">armnn::ReshapeDescriptor</a> reshapeDesc;</div> |
| 5598 | <div class="line"><a name="l05500"></a><span class="lineno"> 5500</span>  reshapeDesc.<a class="code" href="structarmnn_1_1_reshape_descriptor.html#a1178f4dafdda81f59c15145ec327f7d9">m_TargetShape</a> = outputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 5599 | <div class="line"><a name="l05501"></a><span class="lineno"> 5501</span>  </div> |
| 5600 | <div class="line"><a name="l05502"></a><span class="lineno"> 5502</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5601 | <div class="line"><a name="l05503"></a><span class="lineno"> 5503</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5602 | <div class="line"><a name="l05504"></a><span class="lineno"> 5504</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5603 | <div class="line"><a name="l05505"></a><span class="lineno"> 5505</span>  IsReshapeSupported,</div> |
| 5604 | <div class="line"><a name="l05506"></a><span class="lineno"> 5506</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5605 | <div class="line"><a name="l05507"></a><span class="lineno"> 5507</span>  isSupported,</div> |
| 5606 | <div class="line"><a name="l05508"></a><span class="lineno"> 5508</span>  setBackend,</div> |
| 5607 | <div class="line"><a name="l05509"></a><span class="lineno"> 5509</span>  inputInfo,</div> |
| 5608 | <div class="line"><a name="l05510"></a><span class="lineno"> 5510</span>  outputInfo,</div> |
| 5609 | <div class="line"><a name="l05511"></a><span class="lineno"> 5511</span>  reshapeDesc);</div> |
| 5610 | <div class="line"><a name="l05512"></a><span class="lineno"> 5512</span>  </div> |
| 5611 | <div class="line"><a name="l05513"></a><span class="lineno"> 5513</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5612 | <div class="line"><a name="l05514"></a><span class="lineno"> 5514</span>  {</div> |
| 5613 | <div class="line"><a name="l05515"></a><span class="lineno"> 5515</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5614 | <div class="line"><a name="l05516"></a><span class="lineno"> 5516</span>  }</div> |
| 5615 | <div class="line"><a name="l05517"></a><span class="lineno"> 5517</span>  </div> |
| 5616 | <div class="line"><a name="l05518"></a><span class="lineno"> 5518</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddReshapeLayer(reshapeDesc);</div> |
| 5617 | <div class="line"><a name="l05519"></a><span class="lineno"> 5519</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5618 | <div class="line"><a name="l05520"></a><span class="lineno"> 5520</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5619 | <div class="line"><a name="l05521"></a><span class="lineno"> 5521</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5620 | <div class="line"><a name="l05522"></a><span class="lineno"> 5522</span>  </div> |
| 5621 | <div class="line"><a name="l05523"></a><span class="lineno"> 5523</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data);</div> |
| 5622 | <div class="line"><a name="l05524"></a><span class="lineno"> 5524</span> }</div> |
| 5623 | <div class="line"><a name="l05525"></a><span class="lineno"> 5525</span>  </div> |
| 5624 | <div class="line"><a name="l05526"></a><span class="lineno"> 5526</span> <span class="keywordtype">bool</span> Converter::ConvertStridedSlice(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5625 | <div class="line"><a name="l05527"></a><span class="lineno"> 5527</span> {</div> |
| 5626 | <div class="line"><a name="l05528"></a><span class="lineno"> 5528</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertStridedSlice()"</span>;</div> |
| 5627 | <div class="line"><a name="l05529"></a><span class="lineno"> 5529</span>  </div> |
| 5628 | <div class="line"><a name="l05530"></a><span class="lineno"> 5530</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5629 | <div class="line"><a name="l05531"></a><span class="lineno"> 5531</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5630 | <div class="line"><a name="l05532"></a><span class="lineno"> 5532</span>  {</div> |
| 5631 | <div class="line"><a name="l05533"></a><span class="lineno"> 5533</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5632 | <div class="line"><a name="l05534"></a><span class="lineno"> 5534</span>  }</div> |
| 5633 | <div class="line"><a name="l05535"></a><span class="lineno"> 5535</span>  </div> |
| 5634 | <div class="line"><a name="l05536"></a><span class="lineno"> 5536</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 5635 | <div class="line"><a name="l05537"></a><span class="lineno"> 5537</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 5636 | <div class="line"><a name="l05538"></a><span class="lineno"> 5538</span>  <span class="keywordflow">if</span> (rank > 4)</div> |
| 5637 | <div class="line"><a name="l05539"></a><span class="lineno"> 5539</span>  {</div> |
| 5638 | <div class="line"><a name="l05540"></a><span class="lineno"> 5540</span>  Fail(<span class="stringliteral">"%s: Inputs with rank greater than 4 are not supported"</span>, __func__);</div> |
| 5639 | <div class="line"><a name="l05541"></a><span class="lineno"> 5541</span>  }</div> |
| 5640 | <div class="line"><a name="l05542"></a><span class="lineno"> 5542</span>  </div> |
| 5641 | <div class="line"><a name="l05543"></a><span class="lineno"> 5543</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5642 | <div class="line"><a name="l05544"></a><span class="lineno"> 5544</span>  <span class="keywordflow">if</span> (!output)</div> |
| 5643 | <div class="line"><a name="l05545"></a><span class="lineno"> 5545</span>  {</div> |
| 5644 | <div class="line"><a name="l05546"></a><span class="lineno"> 5546</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 5645 | <div class="line"><a name="l05547"></a><span class="lineno"> 5547</span>  }</div> |
| 5646 | <div class="line"><a name="l05548"></a><span class="lineno"> 5548</span>  </div> |
| 5647 | <div class="line"><a name="l05549"></a><span class="lineno"> 5549</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 5648 | <div class="line"><a name="l05550"></a><span class="lineno"> 5550</span>  </div> |
| 5649 | <div class="line"><a name="l05551"></a><span class="lineno"> 5551</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* beginOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model);</div> |
| 5650 | <div class="line"><a name="l05552"></a><span class="lineno"> 5552</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* endOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 2, model);</div> |
| 5651 | <div class="line"><a name="l05553"></a><span class="lineno"> 5553</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* stridesOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 3, model);</div> |
| 5652 | <div class="line"><a name="l05554"></a><span class="lineno"> 5554</span>  </div> |
| 5653 | <div class="line"><a name="l05555"></a><span class="lineno"> 5555</span>  std::vector<int32_t> beginValues;</div> |
| 5654 | <div class="line"><a name="l05556"></a><span class="lineno"> 5556</span>  std::vector<int32_t> endValues;</div> |
| 5655 | <div class="line"><a name="l05557"></a><span class="lineno"> 5557</span>  std::vector<int32_t> stridesValues;</div> |
| 5656 | <div class="line"><a name="l05558"></a><span class="lineno"> 5558</span>  </div> |
| 5657 | <div class="line"><a name="l05559"></a><span class="lineno"> 5559</span>  <span class="comment">// The length of the beginOperand, endOperand and stridesOperand must be of a rank(input)</span></div> |
| 5658 | <div class="line"><a name="l05560"></a><span class="lineno"> 5560</span>  <span class="keyword">auto</span> ValidateInputOperands = [&] (<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>& operand, std::vector<int32_t>& operandValues)</div> |
| 5659 | <div class="line"><a name="l05561"></a><span class="lineno"> 5561</span>  {</div> |
| 5660 | <div class="line"><a name="l05562"></a><span class="lineno"> 5562</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(operand, operandValues, model, data))</div> |
| 5661 | <div class="line"><a name="l05563"></a><span class="lineno"> 5563</span>  {</div> |
| 5662 | <div class="line"><a name="l05564"></a><span class="lineno"> 5564</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5663 | <div class="line"><a name="l05565"></a><span class="lineno"> 5565</span>  }</div> |
| 5664 | <div class="line"><a name="l05566"></a><span class="lineno"> 5566</span>  </div> |
| 5665 | <div class="line"><a name="l05567"></a><span class="lineno"> 5567</span>  <span class="keywordflow">if</span> (operandValues.size() != rank)</div> |
| 5666 | <div class="line"><a name="l05568"></a><span class="lineno"> 5568</span>  {</div> |
| 5667 | <div class="line"><a name="l05569"></a><span class="lineno"> 5569</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5668 | <div class="line"><a name="l05570"></a><span class="lineno"> 5570</span>  }</div> |
| 5669 | <div class="line"><a name="l05571"></a><span class="lineno"> 5571</span>  </div> |
| 5670 | <div class="line"><a name="l05572"></a><span class="lineno"> 5572</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| 5671 | <div class="line"><a name="l05573"></a><span class="lineno"> 5573</span>  };</div> |
| 5672 | <div class="line"><a name="l05574"></a><span class="lineno"> 5574</span>  </div> |
| 5673 | <div class="line"><a name="l05575"></a><span class="lineno"> 5575</span>  <span class="keywordflow">if</span> (!ValidateInputOperands(*beginOperand, beginValues)</div> |
| 5674 | <div class="line"><a name="l05576"></a><span class="lineno"> 5576</span>  || !ValidateInputOperands(*endOperand, endValues)</div> |
| 5675 | <div class="line"><a name="l05577"></a><span class="lineno"> 5577</span>  || !ValidateInputOperands(*stridesOperand, stridesValues))</div> |
| 5676 | <div class="line"><a name="l05578"></a><span class="lineno"> 5578</span>  {</div> |
| 5677 | <div class="line"><a name="l05579"></a><span class="lineno"> 5579</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid input operand"</span>, __func__);</div> |
| 5678 | <div class="line"><a name="l05580"></a><span class="lineno"> 5580</span>  }</div> |
| 5679 | <div class="line"><a name="l05581"></a><span class="lineno"> 5581</span>  </div> |
| 5680 | <div class="line"><a name="l05582"></a><span class="lineno"> 5582</span>  <span class="comment">// Stride cannot have value '0'</span></div> |
| 5681 | <div class="line"><a name="l05583"></a><span class="lineno"> 5583</span>  <span class="keywordflow">if</span> (std::any_of(stridesValues.cbegin(), stridesValues.cend(), [](int32_t i){ return i == 0; }))</div> |
| 5682 | <div class="line"><a name="l05584"></a><span class="lineno"> 5584</span>  {</div> |
| 5683 | <div class="line"><a name="l05585"></a><span class="lineno"> 5585</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Stride must be non-zero value."</span>, __func__);</div> |
| 5684 | <div class="line"><a name="l05586"></a><span class="lineno"> 5586</span>  }</div> |
| 5685 | <div class="line"><a name="l05587"></a><span class="lineno"> 5587</span>  </div> |
| 5686 | <div class="line"><a name="l05588"></a><span class="lineno"> 5588</span>  <a class="code" href="structarmnn_1_1_strided_slice_descriptor.html">armnn::StridedSliceDescriptor</a> descriptor;</div> |
| 5687 | <div class="line"><a name="l05589"></a><span class="lineno"> 5589</span>  descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a118fe06b7c2599da60398ee311ede923">m_Begin</a>.assign(beginValues.cbegin(), beginValues.cend());</div> |
| 5688 | <div class="line"><a name="l05590"></a><span class="lineno"> 5590</span>  descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#aa68194dd6258ab5b04123005a066ea25">m_End</a>.assign(endValues.cbegin(), endValues.cend());</div> |
| 5689 | <div class="line"><a name="l05591"></a><span class="lineno"> 5591</span>  descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a0d53caff836b84204adbd1c28752a201">m_Stride</a>.assign(stridesValues.cbegin(), stridesValues.cend());</div> |
| 5690 | <div class="line"><a name="l05592"></a><span class="lineno"> 5592</span>  descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a6089e1ca91914015777ea780a513131a">m_DataLayout</a> = <a class="code" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>;</div> |
| 5691 | <div class="line"><a name="l05593"></a><span class="lineno"> 5593</span>  </div> |
| 5692 | <div class="line"><a name="l05594"></a><span class="lineno"> 5594</span>  <span class="comment">// Get the "begin_mask", "end_mask", and "shrink_axis_mask" flags</span></div> |
| 5693 | <div class="line"><a name="l05595"></a><span class="lineno"> 5595</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 4, descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a61081be1483984e33db452c75d569f51">m_BeginMask</a>, model, data) ||</div> |
| 5694 | <div class="line"><a name="l05596"></a><span class="lineno"> 5596</span>  !<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 5, descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#ac37e49c0d6e6e54f9d2015d0f11f8ee7">m_EndMask</a>, model, data) ||</div> |
| 5695 | <div class="line"><a name="l05597"></a><span class="lineno"> 5597</span>  !<a class="code" href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">GetInputInt32</a>(operation, 6, descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a6d0384878432cfc9652b7ae8bc59506f">m_ShrinkAxisMask</a>, model, data))</div> |
| 5696 | <div class="line"><a name="l05598"></a><span class="lineno"> 5598</span>  {</div> |
| 5697 | <div class="line"><a name="l05599"></a><span class="lineno"> 5599</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5698 | <div class="line"><a name="l05600"></a><span class="lineno"> 5600</span>  }</div> |
| 5699 | <div class="line"><a name="l05601"></a><span class="lineno"> 5601</span>  </div> |
| 5700 | <div class="line"><a name="l05602"></a><span class="lineno"> 5602</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5701 | <div class="line"><a name="l05603"></a><span class="lineno"> 5603</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5702 | <div class="line"><a name="l05604"></a><span class="lineno"> 5604</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 5703 | <div class="line"><a name="l05605"></a><span class="lineno"> 5605</span>  {</div> |
| 5704 | <div class="line"><a name="l05606"></a><span class="lineno"> 5606</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5705 | <div class="line"><a name="l05607"></a><span class="lineno"> 5607</span>  IsStridedSliceSupported,</div> |
| 5706 | <div class="line"><a name="l05608"></a><span class="lineno"> 5608</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5707 | <div class="line"><a name="l05609"></a><span class="lineno"> 5609</span>  isSupported,</div> |
| 5708 | <div class="line"><a name="l05610"></a><span class="lineno"> 5610</span>  setBackend,</div> |
| 5709 | <div class="line"><a name="l05611"></a><span class="lineno"> 5611</span>  inputInfo,</div> |
| 5710 | <div class="line"><a name="l05612"></a><span class="lineno"> 5612</span>  outputInfo,</div> |
| 5711 | <div class="line"><a name="l05613"></a><span class="lineno"> 5613</span>  descriptor);</div> |
| 5712 | <div class="line"><a name="l05614"></a><span class="lineno"> 5614</span>  };</div> |
| 5713 | <div class="line"><a name="l05615"></a><span class="lineno"> 5615</span>  </div> |
| 5714 | <div class="line"><a name="l05616"></a><span class="lineno"> 5616</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5715 | <div class="line"><a name="l05617"></a><span class="lineno"> 5617</span>  {</div> |
| 5716 | <div class="line"><a name="l05618"></a><span class="lineno"> 5618</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5717 | <div class="line"><a name="l05619"></a><span class="lineno"> 5619</span>  }</div> |
| 5718 | <div class="line"><a name="l05620"></a><span class="lineno"> 5620</span>  <span class="keywordflow">else</span></div> |
| 5719 | <div class="line"><a name="l05621"></a><span class="lineno"> 5621</span>  {</div> |
| 5720 | <div class="line"><a name="l05622"></a><span class="lineno"> 5622</span>  validateFunc(outputInfo, isSupported);</div> |
| 5721 | <div class="line"><a name="l05623"></a><span class="lineno"> 5623</span>  }</div> |
| 5722 | <div class="line"><a name="l05624"></a><span class="lineno"> 5624</span>  </div> |
| 5723 | <div class="line"><a name="l05625"></a><span class="lineno"> 5625</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5724 | <div class="line"><a name="l05626"></a><span class="lineno"> 5626</span>  {</div> |
| 5725 | <div class="line"><a name="l05627"></a><span class="lineno"> 5627</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5726 | <div class="line"><a name="l05628"></a><span class="lineno"> 5628</span>  }</div> |
| 5727 | <div class="line"><a name="l05629"></a><span class="lineno"> 5629</span>  </div> |
| 5728 | <div class="line"><a name="l05630"></a><span class="lineno"> 5630</span>  <span class="comment">// Check if slice can fit in a inferred output</span></div> |
| 5729 | <div class="line"><a name="l05631"></a><span class="lineno"> 5631</span>  <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> inputShape = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">GetShape</a>();</div> |
| 5730 | <div class="line"><a name="l05632"></a><span class="lineno"> 5632</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i < inputShape.<a class="code" href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>(); i++)</div> |
| 5731 | <div class="line"><a name="l05633"></a><span class="lineno"> 5633</span>  {</div> |
| 5732 | <div class="line"><a name="l05634"></a><span class="lineno"> 5634</span>  <span class="keywordtype">int</span> stride = descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a0d53caff836b84204adbd1c28752a201">m_Stride</a>[i];</div> |
| 5733 | <div class="line"><a name="l05635"></a><span class="lineno"> 5635</span>  </div> |
| 5734 | <div class="line"><a name="l05636"></a><span class="lineno"> 5636</span>  <span class="keywordflow">if</span> (descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a6d0384878432cfc9652b7ae8bc59506f">m_ShrinkAxisMask</a> & (1 << i))</div> |
| 5735 | <div class="line"><a name="l05637"></a><span class="lineno"> 5637</span>  {</div> |
| 5736 | <div class="line"><a name="l05638"></a><span class="lineno"> 5638</span>  <span class="comment">// If the difference between the start point and the end point of the slice on an axis being shrunk</span></div> |
| 5737 | <div class="line"><a name="l05639"></a><span class="lineno"> 5639</span>  <span class="comment">// is greater than 1 then throw an error as the output will not be large enough to hold the slice</span></div> |
| 5738 | <div class="line"><a name="l05640"></a><span class="lineno"> 5640</span>  <span class="keywordflow">if</span> (((descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a118fe06b7c2599da60398ee311ede923">m_Begin</a>[i] - descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#aa68194dd6258ab5b04123005a066ea25">m_End</a>[i]) > 1)</div> |
| 5739 | <div class="line"><a name="l05641"></a><span class="lineno"> 5641</span>  || ((descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#a118fe06b7c2599da60398ee311ede923">m_Begin</a>[i] - descriptor.<a class="code" href="structarmnn_1_1_strided_slice_descriptor.html#aa68194dd6258ab5b04123005a066ea25">m_End</a>[i]) < -1))</div> |
| 5740 | <div class="line"><a name="l05642"></a><span class="lineno"> 5642</span>  {</div> |
| 5741 | <div class="line"><a name="l05643"></a><span class="lineno"> 5643</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: StridedSlice: Output will not be large enough to hold the slice"</span>, __func__);</div> |
| 5742 | <div class="line"><a name="l05644"></a><span class="lineno"> 5644</span>  }</div> |
| 5743 | <div class="line"><a name="l05645"></a><span class="lineno"> 5645</span>  </div> |
| 5744 | <div class="line"><a name="l05646"></a><span class="lineno"> 5646</span>  <span class="keywordflow">if</span>(stride < 0)</div> |
| 5745 | <div class="line"><a name="l05647"></a><span class="lineno"> 5647</span>  {</div> |
| 5746 | <div class="line"><a name="l05648"></a><span class="lineno"> 5648</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: StridedSlice: Stride can not be negative while ShrinkAxisMask is set."</span>, __func__);</div> |
| 5747 | <div class="line"><a name="l05649"></a><span class="lineno"> 5649</span>  }</div> |
| 5748 | <div class="line"><a name="l05650"></a><span class="lineno"> 5650</span>  }</div> |
| 5749 | <div class="line"><a name="l05651"></a><span class="lineno"> 5651</span>  }</div> |
| 5750 | <div class="line"><a name="l05652"></a><span class="lineno"> 5652</span>  </div> |
| 5751 | <div class="line"><a name="l05653"></a><span class="lineno"> 5653</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddStridedSliceLayer(descriptor);</div> |
| 5752 | <div class="line"><a name="l05654"></a><span class="lineno"> 5654</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5753 | <div class="line"><a name="l05655"></a><span class="lineno"> 5655</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5754 | <div class="line"><a name="l05656"></a><span class="lineno"> 5656</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5755 | <div class="line"><a name="l05657"></a><span class="lineno"> 5657</span>  </div> |
| 5756 | <div class="line"><a name="l05658"></a><span class="lineno"> 5658</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 5757 | <div class="line"><a name="l05659"></a><span class="lineno"> 5659</span> }</div> |
| 5758 | <div class="line"><a name="l05660"></a><span class="lineno"> 5660</span>  </div> |
| 5759 | <div class="line"><a name="l05661"></a><span class="lineno"> 5661</span> <span class="keywordtype">bool</span> Converter::ConvertTranspose(<span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">Operation</a>& operation, <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">Model</a>& model, <a class="code" href="structarmnn__driver_1_1_conversion_data.html">ConversionData</a>& data)</div> |
| 5760 | <div class="line"><a name="l05662"></a><span class="lineno"> 5662</span> {</div> |
| 5761 | <div class="line"><a name="l05663"></a><span class="lineno"> 5663</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTranspose()"</span>;</div> |
| 5762 | <div class="line"><a name="l05664"></a><span class="lineno"> 5664</span>  </div> |
| 5763 | <div class="line"><a name="l05665"></a><span class="lineno"> 5665</span>  <a class="code" href="classarmnn__driver_1_1_layer_input_handle.html">LayerInputHandle</a> input = <a class="code" href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">ConvertToLayerInputHandle</a>(operation, 0, model, data);</div> |
| 5764 | <div class="line"><a name="l05666"></a><span class="lineno"> 5666</span>  <span class="keywordflow">if</span> (!input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">IsValid</a>())</div> |
| 5765 | <div class="line"><a name="l05667"></a><span class="lineno"> 5667</span>  {</div> |
| 5766 | <div class="line"><a name="l05668"></a><span class="lineno"> 5668</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has invalid inputs"</span>, __func__);</div> |
| 5767 | <div class="line"><a name="l05669"></a><span class="lineno"> 5669</span>  }</div> |
| 5768 | <div class="line"><a name="l05670"></a><span class="lineno"> 5670</span>  </div> |
| 5769 | <div class="line"><a name="l05671"></a><span class="lineno"> 5671</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& inputInfo = input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">GetTensorInfo</a>();</div> |
| 5770 | <div class="line"><a name="l05672"></a><span class="lineno"> 5672</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> rank = inputInfo.<a class="code" href="classarmnn_1_1_tensor_info.html#a157e27d41e9f6b21f0d3c025fa47dc24">GetNumDimensions</a>();</div> |
| 5771 | <div class="line"><a name="l05673"></a><span class="lineno"> 5673</span>  <span class="keywordflow">if</span> (rank > 4)</div> |
| 5772 | <div class="line"><a name="l05674"></a><span class="lineno"> 5674</span>  {</div> |
| 5773 | <div class="line"><a name="l05675"></a><span class="lineno"> 5675</span>  Fail(<span class="stringliteral">"%s: Inputs with rank greater than 4 are not supported"</span>, __func__);</div> |
| 5774 | <div class="line"><a name="l05676"></a><span class="lineno"> 5676</span>  }</div> |
| 5775 | <div class="line"><a name="l05677"></a><span class="lineno"> 5677</span>  </div> |
| 5776 | <div class="line"><a name="l05678"></a><span class="lineno"> 5678</span>  <span class="comment">// NOTE: Axis is an optional parameter to TRANSPOSE, therefore we do not want to generate a failure</span></div> |
| 5777 | <div class="line"><a name="l05679"></a><span class="lineno"> 5679</span>  <span class="comment">// if the operand index is out of bounds.</span></div> |
| 5778 | <div class="line"><a name="l05680"></a><span class="lineno"> 5680</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* permOperand = <a class="code" href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">GetInputOperand</a>(operation, 1, model, <span class="keyword">false</span>);</div> |
| 5779 | <div class="line"><a name="l05681"></a><span class="lineno"> 5681</span>  </div> |
| 5780 | <div class="line"><a name="l05682"></a><span class="lineno"> 5682</span>  std::vector<int32_t> perm(rank);</div> |
| 5781 | <div class="line"><a name="l05683"></a><span class="lineno"> 5683</span>  <span class="keywordflow">if</span> (!permOperand || (permOperand->lifetime == OperandLifeTime::NO_VALUE))</div> |
| 5782 | <div class="line"><a name="l05684"></a><span class="lineno"> 5684</span>  {</div> |
| 5783 | <div class="line"><a name="l05685"></a><span class="lineno"> 5685</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = rank; i > 0; i--)</div> |
| 5784 | <div class="line"><a name="l05686"></a><span class="lineno"> 5686</span>  {</div> |
| 5785 | <div class="line"><a name="l05687"></a><span class="lineno"> 5687</span>  perm[rank - i] = armnn::numeric_cast<int> (i - 1);</div> |
| 5786 | <div class="line"><a name="l05688"></a><span class="lineno"> 5688</span>  }</div> |
| 5787 | <div class="line"><a name="l05689"></a><span class="lineno"> 5689</span>  }</div> |
| 5788 | <div class="line"><a name="l05690"></a><span class="lineno"> 5690</span>  <span class="keywordflow">else</span> <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">GetTensorInt32Values</a>(*permOperand, perm, model, data))</div> |
| 5789 | <div class="line"><a name="l05691"></a><span class="lineno"> 5691</span>  {</div> |
| 5790 | <div class="line"><a name="l05692"></a><span class="lineno"> 5692</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Operation has an invalid or unsupported permutation operand"</span>, __func__);</div> |
| 5791 | <div class="line"><a name="l05693"></a><span class="lineno"> 5693</span>  }</div> |
| 5792 | <div class="line"><a name="l05694"></a><span class="lineno"> 5694</span>  </div> |
| 5793 | <div class="line"><a name="l05695"></a><span class="lineno"> 5695</span>  std::vector<uint32_t> outputDims(perm.begin(), perm.begin() + rank);</div> |
| 5794 | <div class="line"><a name="l05696"></a><span class="lineno"> 5696</span>  </div> |
| 5795 | <div class="line"><a name="l05697"></a><span class="lineno"> 5697</span>  <a class="code" href="structarmnn_1_1_transpose_descriptor.html">armnn::TransposeDescriptor</a> transposeDesc;</div> |
| 5796 | <div class="line"><a name="l05698"></a><span class="lineno"> 5698</span>  transposeDesc.<a class="code" href="structarmnn_1_1_transpose_descriptor.html#a14433af2b223695b40d8c8f8ba2ebb8f">m_DimMappings</a> = <a class="code" href="classarmnn_1_1_permutation_vector.html">armnn::PermutationVector</a>(outputDims.data(), outputDims.size());</div> |
| 5797 | <div class="line"><a name="l05699"></a><span class="lineno"> 5699</span>  </div> |
| 5798 | <div class="line"><a name="l05700"></a><span class="lineno"> 5700</span>  <span class="keyword">const</span> <a class="code" href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">Operand</a>* output = <a class="code" href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">GetOutputOperand</a>(operation, 0, model);</div> |
| 5799 | <div class="line"><a name="l05701"></a><span class="lineno"> 5701</span>  <span class="keywordflow">if</span> (!output)</div> |
| 5800 | <div class="line"><a name="l05702"></a><span class="lineno"> 5702</span>  {</div> |
| 5801 | <div class="line"><a name="l05703"></a><span class="lineno"> 5703</span>  <span class="keywordflow">return</span> Fail(<span class="stringliteral">"%s: Could not read output 0"</span>, __func__);</div> |
| 5802 | <div class="line"><a name="l05704"></a><span class="lineno"> 5704</span>  }</div> |
| 5803 | <div class="line"><a name="l05705"></a><span class="lineno"> 5705</span>  </div> |
| 5804 | <div class="line"><a name="l05706"></a><span class="lineno"> 5706</span>  <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo = <a class="code" href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">GetTensorInfoForOperand</a>(*output);</div> |
| 5805 | <div class="line"><a name="l05707"></a><span class="lineno"> 5707</span>  </div> |
| 5806 | <div class="line"><a name="l05708"></a><span class="lineno"> 5708</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| 5807 | <div class="line"><a name="l05709"></a><span class="lineno"> 5709</span>  <a class="code" href="classarmnn_1_1_backend_id.html">armnn::BackendId</a> setBackend;</div> |
| 5808 | <div class="line"><a name="l05710"></a><span class="lineno"> 5710</span>  <span class="keyword">auto</span> validateFunc = [&](<span class="keyword">const</span> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& outputInfo, <span class="keywordtype">bool</span>& isSupported)</div> |
| 5809 | <div class="line"><a name="l05711"></a><span class="lineno"> 5711</span>  {</div> |
| 5810 | <div class="line"><a name="l05712"></a><span class="lineno"> 5712</span>  <a class="code" href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a>(__func__,</div> |
| 5811 | <div class="line"><a name="l05713"></a><span class="lineno"> 5713</span>  IsTransposeSupported,</div> |
| 5812 | <div class="line"><a name="l05714"></a><span class="lineno"> 5714</span>  data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">m_Backends</a>,</div> |
| 5813 | <div class="line"><a name="l05715"></a><span class="lineno"> 5715</span>  isSupported,</div> |
| 5814 | <div class="line"><a name="l05716"></a><span class="lineno"> 5716</span>  setBackend,</div> |
| 5815 | <div class="line"><a name="l05717"></a><span class="lineno"> 5717</span>  inputInfo,</div> |
| 5816 | <div class="line"><a name="l05718"></a><span class="lineno"> 5718</span>  outputInfo,</div> |
| 5817 | <div class="line"><a name="l05719"></a><span class="lineno"> 5719</span>  transposeDesc);</div> |
| 5818 | <div class="line"><a name="l05720"></a><span class="lineno"> 5720</span>  };</div> |
| 5819 | <div class="line"><a name="l05721"></a><span class="lineno"> 5721</span>  </div> |
| 5820 | <div class="line"><a name="l05722"></a><span class="lineno"> 5722</span>  <span class="keywordflow">if</span>(<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| 5821 | <div class="line"><a name="l05723"></a><span class="lineno"> 5723</span>  {</div> |
| 5822 | <div class="line"><a name="l05724"></a><span class="lineno"> 5724</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| 5823 | <div class="line"><a name="l05725"></a><span class="lineno"> 5725</span>  }</div> |
| 5824 | <div class="line"><a name="l05726"></a><span class="lineno"> 5726</span>  <span class="keywordflow">else</span></div> |
| 5825 | <div class="line"><a name="l05727"></a><span class="lineno"> 5727</span>  {</div> |
| 5826 | <div class="line"><a name="l05728"></a><span class="lineno"> 5728</span>  validateFunc(outputInfo, isSupported);</div> |
| 5827 | <div class="line"><a name="l05729"></a><span class="lineno"> 5729</span>  }</div> |
| 5828 | <div class="line"><a name="l05730"></a><span class="lineno"> 5730</span>  </div> |
| 5829 | <div class="line"><a name="l05731"></a><span class="lineno"> 5731</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| 5830 | <div class="line"><a name="l05732"></a><span class="lineno"> 5732</span>  {</div> |
| 5831 | <div class="line"><a name="l05733"></a><span class="lineno"> 5733</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| 5832 | <div class="line"><a name="l05734"></a><span class="lineno"> 5734</span>  }</div> |
| 5833 | <div class="line"><a name="l05735"></a><span class="lineno"> 5735</span>  </div> |
| 5834 | <div class="line"><a name="l05736"></a><span class="lineno"> 5736</span>  <a class="code" href="classarmnn_1_1_i_connectable_layer.html">armnn::IConnectableLayer</a>* <span class="keyword">const</span> layer = data.<a class="code" href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">m_Network</a>->AddTransposeLayer(transposeDesc);</div> |
| 5835 | <div class="line"><a name="l05737"></a><span class="lineno"> 5737</span>  layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">SetBackendId</a>(setBackend);</div> |
| 5836 | <div class="line"><a name="l05738"></a><span class="lineno"> 5738</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| 5837 | <div class="line"><a name="l05739"></a><span class="lineno"> 5739</span>  input.<a class="code" href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">Connect</a>(layer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div> |
| 5838 | <div class="line"><a name="l05740"></a><span class="lineno"> 5740</span>  </div> |
| 5839 | <div class="line"><a name="l05741"></a><span class="lineno"> 5741</span>  <span class="keywordflow">return</span> <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(operation, 0, *layer, model, data, <span class="keyword">nullptr</span>, validateFunc);</div> |
| 5840 | <div class="line"><a name="l05742"></a><span class="lineno"> 5742</span> }</div> |
| 5841 | <div class="line"><a name="l05743"></a><span class="lineno"> 5743</span>  </div> |
| 5842 | <div class="line"><a name="l05744"></a><span class="lineno"> 5744</span> } <span class="comment">// namespace armnn_driver</span></div> |
| 5843 | </div><!-- fragment --></div><!-- contents --> |
| 5844 | </div><!-- doc-content --> |
| 5845 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_a56b51f56cef50cdfa554258eecdab046"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">armnn::Convolution2dDescriptor::m_PadTop</a></div><div class="ttdeci">uint32_t m_PadTop</div><div class="ttdoc">Padding top value in the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00570">Descriptors.hpp:570</a></div></div> |
| 5846 | <div class="ttc" id="astructarmnn_1_1_instance_normalization_descriptor_html_a8275d51ef9a584feb95726ea0522f6e5"><div class="ttname"><a href="structarmnn_1_1_instance_normalization_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">armnn::InstanceNormalizationDescriptor::m_Beta</a></div><div class="ttdeci">float m_Beta</div><div class="ttdoc">Beta, the offset scalar value applied for the normalized tensor. Defaults to 1.0.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00867">Descriptors.hpp:867</a></div></div> |
| 5847 | <div class="ttc" id="anamespacearmnn_html_aed346b74795fd613f727ca8369778704"><div class="ttname"><a href="namespacearmnn.html#aed346b74795fd613f727ca8369778704">armnn::GetBinaryOperationAsCString</a></div><div class="ttdeci">constexpr char const * GetBinaryOperationAsCString(BinaryOperation operation)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00076">TypesUtils.hpp:76</a></div></div> |
| 5848 | <div class="ttc" id="aclassarmnn_1_1_i_connectable_layer_html_a99cc75f524d831824c5f3730a88e9ca3"><div class="ttname"><a href="classarmnn_1_1_i_connectable_layer.html#a99cc75f524d831824c5f3730a88e9ca3">armnn::IConnectableLayer::SetBackendId</a></div><div class="ttdeci">virtual void SetBackendId(const BackendId &id)=0</div><div class="ttdoc">Set the backend of the IConnectableLayer.</div></div> |
| 5849 | <div class="ttc" id="anamespacearmnn__driver_html_acd62ec697055015572a6ba61eaba4f1a"><div class="ttname"><a href="namespacearmnn__driver.html#acd62ec697055015572a6ba61eaba4f1a">armnn_driver::GetOptionalInputActivation</a></div><div class="ttdeci">bool GetOptionalInputActivation(const Operation &operation, uint32_t inputIndex, ActivationFn &activationFunction, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00886">ConversionUtils.hpp:886</a></div></div> |
| 5850 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_ae83131e16df1cace69395a5f99bc5ecb"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">armnn::LstmInputParams::m_RecurrentToForgetWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00045">LstmParams.hpp:45</a></div></div> |
| 5851 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a15f9f65126ad3d4d82d6c19d0662ae01"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a15f9f65126ad3d4d82d6c19d0662ae01">armnn::LstmInputParamsInfo::m_InputToInputWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00089">LstmParams.hpp:89</a></div></div> |
| 5852 | <div class="ttc" id="anamespacearmnn_html_ac4f8557279754ed7b3f749d55b0e3047a62b6d55816cf737bfc6f42e60df1a3f2"><div class="ttname"><a href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047a62b6d55816cf737bfc6f42e60df1a3f2">armnn::BinaryOperation::Mul</a></div><div class="ttdeci">@ Mul</div></div> |
| 5853 | <div class="ttc" id="astructarmnn_1_1_fully_connected_descriptor_html_a2d3dcfc10f90adedc995b64211dab6e9"><div class="ttname"><a href="structarmnn_1_1_fully_connected_descriptor.html#a2d3dcfc10f90adedc995b64211dab6e9">armnn::FullyConnectedDescriptor::m_ConstantWeights</a></div><div class="ttdeci">bool m_ConstantWeights</div><div class="ttdoc">Enable/disable constant weights and biases.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00530">Descriptors.hpp:530</a></div></div> |
| 5854 | <div class="ttc" id="astructarmnn_1_1_views_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_views_descriptor.html">armnn::ViewsDescriptor</a></div><div class="ttdoc">A ViewsDescriptor for the SplitterLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00244">Descriptors.hpp:244</a></div></div> |
| 5855 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a9b18daea2e9f42386055326fd016519a"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a9b18daea2e9f42386055326fd016519a">armnn::LstmInputParams::m_OutputLayerNormWeights</a></div><div class="ttdeci">const ConstTensor * m_OutputLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00060">LstmParams.hpp:60</a></div></div> |
| 5856 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_afec7f36158448f723b426a9527acb189"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#afec7f36158448f723b426a9527acb189">armnn::QLstmDescriptor::m_ForgetIntermediateScale</a></div><div class="ttdeci">float m_ForgetIntermediateScale</div><div class="ttdoc">Forget intermediate quantization scale.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01428">Descriptors.hpp:1428</a></div></div> |
| 5857 | <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> |
| 5858 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_ac18546ebbebbb32fe0a03baa9bf2c600"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">armnn::TransposeConvolution2dDescriptor::m_PadLeft</a></div><div class="ttdeci">uint32_t m_PadLeft</div><div class="ttdoc">Padding left value in the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01469">Descriptors.hpp:1469</a></div></div> |
| 5859 | <div class="ttc" id="astructarmnn_1_1_batch_mat_mul_descriptor_html_acb441bb8db19bcce78d15cdd8ceb5ea0"><div class="ttname"><a href="structarmnn_1_1_batch_mat_mul_descriptor.html#acb441bb8db19bcce78d15cdd8ceb5ea0">armnn::BatchMatMulDescriptor::m_TransposeX</a></div><div class="ttdeci">bool m_TransposeX</div><div class="ttdoc">Transpose the slices of each input tensor Transpose and Adjoint can not both be set to true for the s...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01612">Descriptors.hpp:1612</a></div></div> |
| 5860 | <div class="ttc" id="anamespacearmnn__driver_html_ac8643aaee3eafc54336c5462b51bd53f"><div class="ttname"><a href="namespacearmnn__driver.html#ac8643aaee3eafc54336c5462b51bd53f">armnn_driver::GetTensorInt32Values</a></div><div class="ttdeci">bool GetTensorInt32Values(const Operand &operand, std::vector< int32_t > &outValues, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00843">ConversionUtils.cpp:843</a></div></div> |
| 5861 | <div class="ttc" id="aclassarmnn__driver_1_1_converter_html_a066acd78f531b709b725a730bfbee887"><div class="ttname"><a href="classarmnn__driver_1_1_converter.html#a066acd78f531b709b725a730bfbee887">armnn_driver::Converter::ConvertOperation</a></div><div class="ttdeci">static bool ConvertOperation(const Operation &operation, const Model &model, ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_converter_8cpp_source.html#l00022">Converter.cpp:22</a></div></div> |
| 5862 | <div class="ttc" id="astructarmnn_1_1_fully_connected_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_fully_connected_descriptor.html">armnn::FullyConnectedDescriptor</a></div><div class="ttdoc">A FullyConnectedDescriptor for the FullyConnectedLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00507">Descriptors.hpp:507</a></div></div> |
| 5863 | <div class="ttc" id="anamespacearmnn_html_ac4f8557279754ed7b3f749d55b0e3047aec211f7c20af43e742bf2570c3cb84f9"><div class="ttname"><a href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047aec211f7c20af43e742bf2570c3cb84f9">armnn::BinaryOperation::Add</a></div><div class="ttdeci">@ Add</div></div> |
| 5864 | <div class="ttc" id="anamespacearmnn_html_a0ad431ff1600c845fda078ac58a47809"><div class="ttname"><a href="namespacearmnn.html#a0ad431ff1600c845fda078ac58a47809">armnn::GetLogicalBinaryOperationAsCString</a></div><div class="ttdeci">constexpr char const * GetLogicalBinaryOperationAsCString(LogicalBinaryOperation operation)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00109">TypesUtils.hpp:109</a></div></div> |
| 5865 | <div class="ttc" id="aclassarmnn_1_1_base_tensor_html_aa81f67ac64f0c249e26499600c45d996"><div class="ttname"><a href="classarmnn_1_1_base_tensor.html#aa81f67ac64f0c249e26499600c45d996">armnn::BaseTensor::GetMemoryArea</a></div><div class="ttdeci">MemoryType GetMemoryArea() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00307">Tensor.hpp:307</a></div></div> |
| 5866 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html">armnn::QLstmDescriptor</a></div><div class="ttdoc">A QLstmDescriptor for the QLstmLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01380">Descriptors.hpp:1380</a></div></div> |
| 5867 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_afe6a3377c4531315354def9023c8fdda"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">armnn::TransposeConvolution2dDescriptor::m_StrideX</a></div><div class="ttdeci">uint32_t m_StrideX</div><div class="ttdoc">Stride value when proceeding through input for the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01477">Descriptors.hpp:1477</a></div></div> |
| 5868 | <div class="ttc" id="aclassarmnn_1_1_optional_html"><div class="ttname"><a href="classarmnn_1_1_optional.html">armnn::Optional</a></div><div class="ttdef"><b>Definition:</b> <a href="_optional_8hpp_source.html#l00270">Optional.hpp:270</a></div></div> |
| 5869 | <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> |
| 5870 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_af07c65d3b7886a00f10c69093e76a341"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#af07c65d3b7886a00f10c69093e76a341">armnn::QuantizedLstmInputParamsInfo::m_InputToCellWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00141">QuantizedLstmParams.hpp:141</a></div></div> |
| 5871 | <div class="ttc" id="anamespacearmnn_html_aded981a42027bd3302b9c0f09d988659"><div class="ttname"><a href="namespacearmnn.html#aded981a42027bd3302b9c0f09d988659">armnn::GetResizeMethodAsCString</a></div><div class="ttdeci">constexpr const char * GetResizeMethodAsCString(ResizeMethod method)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00285">TypesUtils.hpp:285</a></div></div> |
| 5872 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_a6c9de81fc65b3c4924cab11907075a17"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">armnn::QLstmDescriptor::m_ProjectionEnabled</a></div><div class="ttdeci">bool m_ProjectionEnabled</div><div class="ttdoc">Enable/disable the projection layer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01422">Descriptors.hpp:1422</a></div></div> |
| 5873 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html_a4022d5107338aaf5eb7abebf78a1360b"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html#a4022d5107338aaf5eb7abebf78a1360b">armnn::ResizeDescriptor::m_HalfPixelCenters</a></div><div class="ttdeci">bool m_HalfPixelCenters</div><div class="ttdoc">Half Pixel Centers.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01018">Descriptors.hpp:1018</a></div></div> |
| 5874 | <div class="ttc" id="anamespacearmnn_html_a9a2af2f8c4af4f9efa8e79417d505ac4"><div class="ttname"><a href="namespacearmnn.html#a9a2af2f8c4af4f9efa8e79417d505ac4">armnn::ResizeMethod</a></div><div class="ttdeci">ResizeMethod</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00166">Types.hpp:166</a></div></div> |
| 5875 | <div class="ttc" id="anamespacearmnn_html_ad1d5cce2d9e9a5d61c243e5c989112e0"><div class="ttname"><a href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a></div><div class="ttdeci">DataLayout</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00062">Types.hpp:62</a></div></div> |
| 5876 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_ab8c7712c0c6bc427dede8c76945e8f89"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#ab8c7712c0c6bc427dede8c76945e8f89">armnn::LstmInputParamsInfo::m_InputLayerNormWeights</a></div><div class="ttdeci">const TensorInfo * m_InputLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00106">LstmParams.hpp:106</a></div></div> |
| 5877 | <div class="ttc" id="astructarmnn_1_1_space_to_batch_nd_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::SpaceToBatchNdDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01071">Descriptors.hpp:1071</a></div></div> |
| 5878 | <div class="ttc" id="astructarmnn_1_1_fully_connected_descriptor_html_a281fcaec86e17c97f7b8402633f6b55a"><div class="ttname"><a href="structarmnn_1_1_fully_connected_descriptor.html#a281fcaec86e17c97f7b8402633f6b55a">armnn::FullyConnectedDescriptor::m_TransposeWeightMatrix</a></div><div class="ttdeci">bool m_TransposeWeightMatrix</div><div class="ttdoc">Enable/disable transpose weight matrix.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00528">Descriptors.hpp:528</a></div></div> |
| 5879 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a21626d0b91f8942c3ef8e48e17db9f21"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a21626d0b91f8942c3ef8e48e17db9f21">armnn::QuantizedLstmInputParamsInfo::m_ForgetGateBias</a></div><div class="ttdeci">const TensorInfo * m_ForgetGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00150">QuantizedLstmParams.hpp:150</a></div></div> |
| 5880 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html_a46c3fa15c46fb0d1dcdc24d0ea5cb5cd"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html#a46c3fa15c46fb0d1dcdc24d0ea5cb5cd">armnn::ResizeDescriptor::m_TargetHeight</a></div><div class="ttdeci">uint32_t m_TargetHeight</div><div class="ttdoc">Target height value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01009">Descriptors.hpp:1009</a></div></div> |
| 5881 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_aea202e14d8874cefd9a0f778022b7e25"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">armnn::DepthwiseConvolution2dDescriptor::m_BiasEnabled</a></div><div class="ttdeci">bool m_BiasEnabled</div><div class="ttdoc">Enable/disable bias.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00708">Descriptors.hpp:708</a></div></div> |
| 5882 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a8bc11f1fa23ef42532f9fdd04d355270"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a8bc11f1fa23ef42532f9fdd04d355270">armnn::TensorInfo::GetQuantizationScales</a></div><div class="ttdeci">std::vector< float > GetQuantizationScales() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00451">Tensor.cpp:451</a></div></div> |
| 5883 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_a118fe06b7c2599da60398ee311ede923"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#a118fe06b7c2599da60398ee311ede923">armnn::StridedSliceDescriptor::m_Begin</a></div><div class="ttdeci">std::vector< int > m_Begin</div><div class="ttdoc">Begin values for the input that will be sliced.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01342">Descriptors.hpp:1342</a></div></div> |
| 5884 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a44b0e6b16708df7f0d2bbab141688aaa"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a44b0e6b16708df7f0d2bbab141688aaa">armnn::LstmInputParams::m_ProjectionBias</a></div><div class="ttdeci">const ConstTensor * m_ProjectionBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00056">LstmParams.hpp:56</a></div></div> |
| 5885 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a1759754ccb88ecc9af44f3aae6e244ee"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">armnn::LstmInputParams::m_RecurrentToCellWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00046">LstmParams.hpp:46</a></div></div> |
| 5886 | <div class="ttc" id="anamespacearmnn_html_ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51"><div class="ttname"><a href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a></div><div class="ttdeci">@ NHWC</div></div> |
| 5887 | <div class="ttc" id="anamespacearmnn__driver_html_aa9b4bc8d4cfb04be14771dbb1bc1a1ac"><div class="ttname"><a href="namespacearmnn__driver.html#aa9b4bc8d4cfb04be14771dbb1bc1a1ac">armnn_driver::Operand</a></div><div class="ttdeci">::android::nn::Operand Operand</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00046">ConversionUtils.hpp:46</a></div></div> |
| 5888 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a6e30c7b3451da3ea9cf4259fb602e6e6"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">armnn::QuantizedLstmInputParams::m_InputToOutputWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00036">QuantizedLstmParams.hpp:36</a></div></div> |
| 5889 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a4a9d678146f572808a361dbdc5489f38"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">armnn::LstmInputParams::m_CellBias</a></div><div class="ttdeci">const ConstTensor * m_CellBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00053">LstmParams.hpp:53</a></div></div> |
| 5890 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html">armnn::ResizeDescriptor</a></div><div class="ttdoc">A ResizeDescriptor for the ResizeLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00985">Descriptors.hpp:985</a></div></div> |
| 5891 | <div class="ttc" id="astructarmnn_1_1_arg_min_max_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_arg_min_max_descriptor.html">armnn::ArgMinMaxDescriptor</a></div><div class="ttdoc">An ArgMinMaxDescriptor for ArgMinMaxLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00067">Descriptors.hpp:67</a></div></div> |
| 5892 | <div class="ttc" id="anamespacearmnn__driver_html_a208149d5c08f84debf8b265bb508fe02"><div class="ttname"><a href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">armnn_driver::IsDynamicTensor</a></div><div class="ttdeci">bool IsDynamicTensor(const armnn::TensorInfo &tensorInfo)</div><div class="ttdoc">Checks if a tensor info represents a dynamic tensor.</div><div class="ttdef"><b>Definition:</b> <a href="_canonical_utils_8cpp_source.html#l00491">CanonicalUtils.cpp:491</a></div></div> |
| 5893 | <div class="ttc" id="astructarmnn_1_1_activation_descriptor_html_a017b2990003a014234f13e999dc7c689"><div class="ttname"><a href="structarmnn_1_1_activation_descriptor.html#a017b2990003a014234f13e999dc7c689">armnn::ActivationDescriptor::m_A</a></div><div class="ttdeci">float m_A</div><div class="ttdoc">Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH,...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00061">Descriptors.hpp:61</a></div></div> |
| 5894 | <div class="ttc" id="anamespacearmnn__driver_html_a2eb75cb48464cf87c290503d7bea79a4"><div class="ttname"><a href="namespacearmnn__driver.html#a2eb75cb48464cf87c290503d7bea79a4">armnn_driver::ConvertPooling2d</a></div><div class="ttdeci">bool ConvertPooling2d(const Operation &operation, const char *operationName, armnn::PoolingAlgorithm poolType, const Model &model, ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00392">ConversionUtils.cpp:392</a></div></div> |
| 5895 | <div class="ttc" id="aclassarmnn__driver_1_1_layer_input_handle_html_a9b63bdf08709bc5302b3c1a9198d06a3"><div class="ttname"><a href="classarmnn__driver_1_1_layer_input_handle.html#a9b63bdf08709bc5302b3c1a9198d06a3">armnn_driver::LayerInputHandle::Connect</a></div><div class="ttdeci">void Connect(armnn::IInputSlot &inputSlot)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00033">ConversionUtils.cpp:33</a></div></div> |
| 5896 | <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> |
| 5897 | <div class="ttc" id="anamespacearmnn__driver_html_a8c749c18e62c59a5c1254aac342c995a"><div class="ttname"><a href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">armnn_driver::ConvertOperationInputToConstTensorPin</a></div><div class="ttdeci">ConstTensorPin ConvertOperationInputToConstTensorPin(const Operation &operation, uint32_t inputIndex, const Model &model, const ConversionData &data, const armnn::PermutationVector &dimensionMappings=g_DontPermute, const armnn::TensorShape *overrideTensorShape=nullptr, bool optional=false)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00751">ConversionUtils.hpp:751</a></div></div> |
| 5898 | <div class="ttc" id="astructarmnn_1_1_softmax_descriptor_html_a8275d51ef9a584feb95726ea0522f6e5"><div class="ttname"><a href="structarmnn_1_1_softmax_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">armnn::SoftmaxDescriptor::m_Beta</a></div><div class="ttdeci">float m_Beta</div><div class="ttdoc">Exponentiation value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00190">Descriptors.hpp:190</a></div></div> |
| 5899 | <div class="ttc" id="astructarmnn_1_1_gather_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_gather_descriptor.html">armnn::GatherDescriptor</a></div><div class="ttdoc">A GatherDescriptor for the GatherLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00965">Descriptors.hpp:965</a></div></div> |
| 5900 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a047ca888c43bd7fb5702853bf72410d0"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a047ca888c43bd7fb5702853bf72410d0">armnn::TensorInfo::GetQuantizationScale</a></div><div class="ttdeci">float GetQuantizationScale() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00461">Tensor.cpp:461</a></div></div> |
| 5901 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::DepthwiseConvolution2dDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00710">Descriptors.hpp:710</a></div></div> |
| 5902 | <div class="ttc" id="astructarmnn_1_1_l2_normalization_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_l2_normalization_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::L2NormalizationDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00824">Descriptors.hpp:824</a></div></div> |
| 5903 | <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> |
| 5904 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_af434b509cd5232ef762c4b21b2dfaae0"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#af434b509cd5232ef762c4b21b2dfaae0">armnn::QuantizedLstmInputParamsInfo::m_RecurrentToOutputWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00147">QuantizedLstmParams.hpp:147</a></div></div> |
| 5905 | <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> |
| 5906 | <div class="ttc" id="anamespacearmnn_html_ad605d1661fa0d8c7fea651d82fbe11c9ac65d2e51c610dd3853a3c777aa8bfe9d"><div class="ttname"><a href="namespacearmnn.html#ad605d1661fa0d8c7fea651d82fbe11c9ac65d2e51c610dd3853a3c777aa8bfe9d">armnn::NormalizationAlgorithmMethod::LocalBrightness</a></div><div class="ttdeci">@ LocalBrightness</div><div class="ttdoc">Krichevsky 2012: Local Brightness Normalization.</div></div> |
| 5907 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_a8275d51ef9a584feb95726ea0522f6e5"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#a8275d51ef9a584feb95726ea0522f6e5">armnn::NormalizationDescriptor::m_Beta</a></div><div class="ttdeci">float m_Beta</div><div class="ttdoc">Beta value for the normalization equation.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00801">Descriptors.hpp:801</a></div></div> |
| 5908 | <div class="ttc" id="anamespacearmnn__driver_html_a1f6b85ab26d684241babcde2d659116b"><div class="ttname"><a href="namespacearmnn__driver.html#a1f6b85ab26d684241babcde2d659116b">armnn_driver::ConvertToLayerInputHandle</a></div><div class="ttdeci">LayerInputHandle ConvertToLayerInputHandle(const Operation &operation, uint32_t inputIndex, const Model &model, ConversionData &data, const armnn::PermutationVector &dimensionMappings, const LayerInputHandle *inputHandle)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00216">ConversionUtils.cpp:216</a></div></div> |
| 5909 | <div class="ttc" id="anamespacearmnn__driver_html_ab3c135a0d4b1152f6b989ea6afddd6bc"><div class="ttname"><a href="namespacearmnn__driver.html#ab3c135a0d4b1152f6b989ea6afddd6bc">armnn_driver::isQuantizedOperand</a></div><div class="ttdeci">bool isQuantizedOperand(const OperandType &operandType)</div><div class="ttdef"><b>Definition:</b> <a href="_canonical_utils_8cpp_source.html#l00510">CanonicalUtils.cpp:510</a></div></div> |
| 5910 | <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> |
| 5911 | <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> |
| 5912 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_ab173a067eeb7295d84f5327bcc05a6c1"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#ab173a067eeb7295d84f5327bcc05a6c1">armnn::QuantizedLstmInputParamsInfo::m_CellBias</a></div><div class="ttdeci">const TensorInfo * m_CellBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00151">QuantizedLstmParams.hpp:151</a></div></div> |
| 5913 | <div class="ttc" id="astructarmnn_1_1_batch_to_space_nd_descriptor_html_a02e143524aefddd40b485fcf7dea6696"><div class="ttname"><a href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a02e143524aefddd40b485fcf7dea6696">armnn::BatchToSpaceNdDescriptor::m_BlockShape</a></div><div class="ttdeci">std::vector< unsigned int > m_BlockShape</div><div class="ttdoc">Block shape values.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00898">Descriptors.hpp:898</a></div></div> |
| 5914 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a48cb83dcb58c786ea36f5d37695e75b1"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a48cb83dcb58c786ea36f5d37695e75b1">armnn::LstmInputParamsInfo::m_OutputGateBias</a></div><div class="ttdeci">const TensorInfo * m_OutputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00103">LstmParams.hpp:103</a></div></div> |
| 5915 | <div class="ttc" id="aclassarmnn_utils_1_1_data_layout_indexed_html"><div class="ttname"><a href="classarmnn_utils_1_1_data_layout_indexed.html">armnnUtils::DataLayoutIndexed</a></div><div class="ttdoc">Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout.</div><div class="ttdef"><b>Definition:</b> <a href="_data_layout_indexed_8hpp_source.html#l00017">DataLayoutIndexed.hpp:17</a></div></div> |
| 5916 | <div class="ttc" id="astructarmnn__driver_1_1_conversion_data_html"><div class="ttname"><a href="structarmnn__driver_1_1_conversion_data.html">armnn_driver::ConversionData</a></div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00053">ConversionUtils.hpp:53</a></div></div> |
| 5917 | <div class="ttc" id="aclassarmnn__driver_1_1_converter_html_a923b39ec5f83672f50289cbaf5de6798"><div class="ttname"><a href="classarmnn__driver_1_1_converter.html#a923b39ec5f83672f50289cbaf5de6798">armnn_driver::Converter::Model</a></div><div class="ttdeci">::android::nn::Model Model</div><div class="ttdef"><b>Definition:</b> <a href="_converter_8hpp_source.html#l00024">Converter.hpp:24</a></div></div> |
| 5918 | <div class="ttc" id="anamespacearmnn__driver_html_ae28f9841f6cb1775849b7921b54da32e"><div class="ttname"><a href="namespacearmnn__driver.html#ae28f9841f6cb1775849b7921b54da32e">armnn_driver::GetOptionalBool</a></div><div class="ttdeci">bool GetOptionalBool(const Operation &operation, uint32_t inputIndex, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00933">ConversionUtils.hpp:933</a></div></div> |
| 5919 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::ResizeDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01014">Descriptors.hpp:1014</a></div></div> |
| 5920 | <div class="ttc" id="anamespacearmnn_html_a56297e0f7b215eea46c818cb7528d9eaa23b68da1de2b77d74da9da2635722a3e"><div class="ttname"><a href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaa23b68da1de2b77d74da9da2635722a3e">armnn::ActivationFunction::TanH</a></div><div class="ttdeci">@ TanH</div></div> |
| 5921 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_ab99b2297c5a17114926b8b47c693fe43"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#ab99b2297c5a17114926b8b47c693fe43">armnn::LstmInputParamsInfo::m_CellToInputWeights</a></div><div class="ttdeci">const TensorInfo * m_CellToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00097">LstmParams.hpp:97</a></div></div> |
| 5922 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a435d3651482bbfcc11263b4e4e0c900f"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">armnn::QuantizedLstmInputParams::m_RecurrentToInputWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00038">QuantizedLstmParams.hpp:38</a></div></div> |
| 5923 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_ac18546ebbebbb32fe0a03baa9bf2c600"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">armnn::DepthwiseConvolution2dDescriptor::m_PadLeft</a></div><div class="ttdeci">uint32_t m_PadLeft</div><div class="ttdoc">Padding left value in the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00692">Descriptors.hpp:692</a></div></div> |
| 5924 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a567622530e51eaa2e82c9790a3a43f02"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a567622530e51eaa2e82c9790a3a43f02">armnn::LstmInputParamsInfo::m_CellLayerNormWeights</a></div><div class="ttdeci">const TensorInfo * m_CellLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00108">LstmParams.hpp:108</a></div></div> |
| 5925 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_ac1fe174bbadfb39a2b636940c2e647c8"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">armnn::Convolution2dDescriptor::m_StrideY</a></div><div class="ttdeci">uint32_t m_StrideY</div><div class="ttdoc">Stride value when proceeding through input for the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00576">Descriptors.hpp:576</a></div></div> |
| 5926 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_ae83131e16df1cace69395a5f99bc5ecb"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#ae83131e16df1cace69395a5f99bc5ecb">armnn::QuantizedLstmInputParams::m_RecurrentToForgetWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00039">QuantizedLstmParams.hpp:39</a></div></div> |
| 5927 | <div class="ttc" id="anamespacearmnn_html_a7863c179ff92feec660c48ab7b95ae55"><div class="ttname"><a href="namespacearmnn.html#a7863c179ff92feec660c48ab7b95ae55">armnn::ConcatDescriptor</a></div><div class="ttdeci">OriginsDescriptor ConcatDescriptor</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_fwd_8hpp_source.html#l00060">DescriptorsFwd.hpp:60</a></div></div> |
| 5928 | <div class="ttc" id="a_conversion_utils_8hpp_html_a00e2c074a81ec0d41495ed9287dcc542"><div class="ttname"><a href="_conversion_utils_8hpp.html#a00e2c074a81ec0d41495ed9287dcc542">FORWARD_LAYER_SUPPORT_FUNC</a></div><div class="ttdeci">#define FORWARD_LAYER_SUPPORT_FUNC(funcName, func, backends, supported, setBackend,...)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00155">ConversionUtils.hpp:155</a></div></div> |
| 5929 | <div class="ttc" id="anamespacearmnn__driver_html_a0f38fa92b2468d5378258a2b074c1a31"><div class="ttname"><a href="namespacearmnn__driver.html#a0f38fa92b2468d5378258a2b074c1a31">armnn_driver::Half</a></div><div class="ttdeci">half_float::half Half</div><div class="ttdef"><b>Definition:</b> <a href="_converter_8cpp_source.html#l00015">Converter.cpp:15</a></div></div> |
| 5930 | <div class="ttc" id="anamespacearmnn__driver_html_a5890dee5ebd8c09d4b3787136d8170d8"><div class="ttname"><a href="namespacearmnn__driver.html#a5890dee5ebd8c09d4b3787136d8170d8">armnn_driver::GetOutputOperand</a></div><div class="ttdeci">const Operand * GetOutputOperand(const Operation &operation, uint32_t outputIndex, const Model &model)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00688">ConversionUtils.hpp:688</a></div></div> |
| 5931 | <div class="ttc" id="astructarmnn_1_1_arg_min_max_descriptor_html_ab1ae6f520bb1a4da191a0ae907477f23"><div class="ttname"><a href="structarmnn_1_1_arg_min_max_descriptor.html#ab1ae6f520bb1a4da191a0ae907477f23">armnn::ArgMinMaxDescriptor::m_Function</a></div><div class="ttdeci">ArgMinMaxFunction m_Function</div><div class="ttdoc">Specify if the function is to find Min or Max.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00081">Descriptors.hpp:81</a></div></div> |
| 5932 | <div class="ttc" id="anamespacearmnn__driver_html_ad3a9c1424b37970a54624f514196b985"><div class="ttname"><a href="namespacearmnn__driver.html#ad3a9c1424b37970a54624f514196b985">armnn_driver::Operation</a></div><div class="ttdeci">::android::nn::Operation Operation</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00049">ConversionUtils.hpp:49</a></div></div> |
| 5933 | <div class="ttc" id="astructarmnn_1_1_space_to_batch_nd_descriptor_html_a02e143524aefddd40b485fcf7dea6696"><div class="ttname"><a href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a02e143524aefddd40b485fcf7dea6696">armnn::SpaceToBatchNdDescriptor::m_BlockShape</a></div><div class="ttdeci">std::vector< unsigned int > m_BlockShape</div><div class="ttdoc">Block shape value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01066">Descriptors.hpp:1066</a></div></div> |
| 5934 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_a61081be1483984e33db452c75d569f51"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#a61081be1483984e33db452c75d569f51">armnn::StridedSliceDescriptor::m_BeginMask</a></div><div class="ttdeci">int32_t m_BeginMask</div><div class="ttdoc">Begin mask value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01350">Descriptors.hpp:1350</a></div></div> |
| 5935 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a6c7873a3320a59a63aa26c006db905c0"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a6c7873a3320a59a63aa26c006db905c0">armnn::LstmInputParamsInfo::m_RecurrentToCellWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00095">LstmParams.hpp:95</a></div></div> |
| 5936 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_afe204ca375b74e9a72640c9227918d0a"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#afe204ca375b74e9a72640c9227918d0a">armnn::LstmInputParams::m_CellToOutputWeights</a></div><div class="ttdeci">const ConstTensor * m_CellToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00050">LstmParams.hpp:50</a></div></div> |
| 5937 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a49e11acda22742cbaf6f1b259ead0d84"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">armnn::LstmInputParams::m_InputToCellWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00042">LstmParams.hpp:42</a></div></div> |
| 5938 | <div class="ttc" id="anamespacearmnn__driver_html_a28243ca97e12ccb50e01f03d313ecc86"><div class="ttname"><a href="namespacearmnn__driver.html#a28243ca97e12ccb50e01f03d313ecc86">armnn_driver::ConvertToActivation</a></div><div class="ttdeci">bool ConvertToActivation(const Operation &operation, const char *operationName, const armnn::ActivationDescriptor &activationDesc, const Model &model, ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00604">ConversionUtils.cpp:604</a></div></div> |
| 5939 | <div class="ttc" id="anamespacearmnn_html_ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e"><div class="ttname"><a href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e">armnn::DataType::QAsymmU8</a></div><div class="ttdeci">@ QAsymmU8</div></div> |
| 5940 | <div class="ttc" id="aclassarmnn__driver_1_1_layer_input_handle_html_a594af6896fa4bf3de9ee961fa8e2aade"><div class="ttname"><a href="classarmnn__driver_1_1_layer_input_handle.html#a594af6896fa4bf3de9ee961fa8e2aade">armnn_driver::LayerInputHandle::SanitizeQuantizationScale</a></div><div class="ttdeci">void SanitizeQuantizationScale(LayerInputHandle &weight, LayerInputHandle &input)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00062">ConversionUtils.cpp:62</a></div></div> |
| 5941 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_a09e1f097944f61cc901240f9300364cf"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#a09e1f097944f61cc901240f9300364cf">armnn::QLstmDescriptor::m_InputIntermediateScale</a></div><div class="ttdeci">float m_InputIntermediateScale</div><div class="ttdoc">Input intermediate quantization scale.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01426">Descriptors.hpp:1426</a></div></div> |
| 5942 | <div class="ttc" id="anamespacearmnn_html_ae7e8cbf71db6a490789ca6dcaa8deeae"><div class="ttname"><a href="namespacearmnn.html#ae7e8cbf71db6a490789ca6dcaa8deeae">armnn::ArgMinMaxFunction</a></div><div class="ttdeci">ArgMinMaxFunction</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00103">Types.hpp:103</a></div></div> |
| 5943 | <div class="ttc" id="anamespacearmnn_html_a56297e0f7b215eea46c818cb7528d9eaabc5a0f0d6e7cf7fca73299dcd46c0f0d"><div class="ttname"><a href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaabc5a0f0d6e7cf7fca73299dcd46c0f0d">armnn::ActivationFunction::BoundedReLu</a></div><div class="ttdeci">@ BoundedReLu</div><div class="ttdoc">min(a, max(b, input)) ReLu1 & ReLu6.</div></div> |
| 5944 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_ace7a1f1f1041b412b7d8ef82b95ff352"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">armnn::QuantizedLstmInputParams::m_ForgetGateBias</a></div><div class="ttdeci">const ConstTensor * m_ForgetGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00044">QuantizedLstmParams.hpp:44</a></div></div> |
| 5945 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a1759754ccb88ecc9af44f3aae6e244ee"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a1759754ccb88ecc9af44f3aae6e244ee">armnn::QuantizedLstmInputParams::m_RecurrentToCellWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00040">QuantizedLstmParams.hpp:40</a></div></div> |
| 5946 | <div class="ttc" id="anamespacearmnn_html_a0b49aa352b84d572942185ce72cef751"><div class="ttname"><a href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">armnn::Half</a></div><div class="ttdeci">half_float::half Half</div><div class="ttdef"><b>Definition:</b> <a href="_half_8hpp_source.html#l00022">Half.hpp:22</a></div></div> |
| 5947 | <div class="ttc" id="aclassarmnn_1_1_i_output_slot_html_a9943775a364fc4ab53b85ac88f311886"><div class="ttname"><a href="classarmnn_1_1_i_output_slot.html#a9943775a364fc4ab53b85ac88f311886">armnn::IOutputSlot::GetTensorInfo</a></div><div class="ttdeci">virtual const TensorInfo & GetTensorInfo() const =0</div></div> |
| 5948 | <div class="ttc" id="anamespacearmnn__driver_html_a3f3a756df9dee4b33566cddd786b83c9"><div class="ttname"><a href="namespacearmnn__driver.html#a3f3a756df9dee4b33566cddd786b83c9">armnn_driver::OptionalDataLayout</a></div><div class="ttdeci">armnn::DataLayout OptionalDataLayout(const Operation &operation, uint32_t inputIndex, const Model &model, ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00874">ConversionUtils.cpp:874</a></div></div> |
| 5949 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_aa70c05f1aad12fbd9d9ec43ea4557b03"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#aa70c05f1aad12fbd9d9ec43ea4557b03">armnn::NormalizationDescriptor::m_NormSize</a></div><div class="ttdeci">uint32_t m_NormSize</div><div class="ttdoc">Depth radius value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00797">Descriptors.hpp:797</a></div></div> |
| 5950 | <div class="ttc" id="anamespacearmnn_utils_html_a428a9a6ffdf0e8d723b50c038c56c336"><div class="ttname"><a href="namespacearmnn_utils.html#a428a9a6ffdf0e8d723b50c038c56c336">armnnUtils::TransposeTensorShape</a></div><div class="ttdeci">armnn::TensorShape TransposeTensorShape(const armnn::TensorShape &srcShape, const armnn::PermutationVector &mappings)</div><div class="ttdef"><b>Definition:</b> <a href="_transpose_8cpp_source.html#l00125">Transpose.cpp:125</a></div></div> |
| 5951 | <div class="ttc" id="anamespacearmnn__driver_html_ae059aea66ee0867a204aa2f2d9abb42c"><div class="ttname"><a href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">armnn_driver::g_DontPermute</a></div><div class="ttdeci">const armnn::PermutationVector g_DontPermute</div><div class="ttdef"><b>Definition:</b> <a href="_canonical_utils_8cpp_source.html#l00038">CanonicalUtils.cpp:38</a></div></div> |
| 5952 | <div class="ttc" id="anamespacearmnn_html_a2da4db0140d1a6dc69c9c82e9ef5379e"><div class="ttname"><a href="namespacearmnn.html#a2da4db0140d1a6dc69c9c82e9ef5379e">armnn::LogicalBinaryOperation</a></div><div class="ttdeci">LogicalBinaryOperation</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00119">Types.hpp:119</a></div></div> |
| 5953 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_ac18546ebbebbb32fe0a03baa9bf2c600"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#ac18546ebbebbb32fe0a03baa9bf2c600">armnn::Convolution2dDescriptor::m_PadLeft</a></div><div class="ttdeci">uint32_t m_PadLeft</div><div class="ttdoc">Padding left value in the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00566">Descriptors.hpp:566</a></div></div> |
| 5954 | <div class="ttc" id="astructarmnn_1_1_batch_to_space_nd_descriptor_html_a3941f674c071c9503e00d2b59e92e454"><div class="ttname"><a href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a3941f674c071c9503e00d2b59e92e454">armnn::BatchToSpaceNdDescriptor::m_Crops</a></div><div class="ttdeci">std::vector< std::pair< unsigned int, unsigned int > > m_Crops</div><div class="ttdoc">The values to crop from the input dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00900">Descriptors.hpp:900</a></div></div> |
| 5955 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a9935ad8aff43cf1e0622bf6d82710513"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a9935ad8aff43cf1e0622bf6d82710513">armnn::LstmInputParamsInfo::m_OutputLayerNormWeights</a></div><div class="ttdeci">const TensorInfo * m_OutputLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00109">LstmParams.hpp:109</a></div></div> |
| 5956 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_ab85cd8cc10c96a7c99c14042c251fc48"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#ab85cd8cc10c96a7c99c14042c251fc48">armnn::TensorInfo::HasPerAxisQuantization</a></div><div class="ttdeci">bool HasPerAxisQuantization() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00446">Tensor.cpp:446</a></div></div> |
| 5957 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_ac1fe174bbadfb39a2b636940c2e647c8"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">armnn::DepthwiseConvolution2dDescriptor::m_StrideY</a></div><div class="ttdeci">uint32_t m_StrideY</div><div class="ttdoc">Stride value when proceeding through input for the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00702">Descriptors.hpp:702</a></div></div> |
| 5958 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_a302b688d88dd73cde0fb1faef6679907"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#a302b688d88dd73cde0fb1faef6679907">armnn::Convolution2dDescriptor::m_DilationY</a></div><div class="ttdeci">uint32_t m_DilationY</div><div class="ttdoc">Dilation along y axis.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00580">Descriptors.hpp:580</a></div></div> |
| 5959 | <div class="ttc" id="anamespacearmnn_html_a4dc0adc6737b5944e7671bee71788407acb5e100e5a9a3e7f6d1fd97512215282"><div class="ttname"><a href="namespacearmnn.html#a4dc0adc6737b5944e7671bee71788407acb5e100e5a9a3e7f6d1fd97512215282">armnn::BoostLogSeverityMapping::error</a></div><div class="ttdeci">@ error</div></div> |
| 5960 | <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> |
| 5961 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_a0477ee1b44ace6090119178eea78cb0b"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#a0477ee1b44ace6090119178eea78cb0b">armnn::QLstmDescriptor::m_CellIntermediateScale</a></div><div class="ttdeci">float m_CellIntermediateScale</div><div class="ttdoc">Cell intermediate quantization scale.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01430">Descriptors.hpp:1430</a></div></div> |
| 5962 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_a05945f080edf694b631960728b87aadb"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#a05945f080edf694b631960728b87aadb">armnn::NormalizationDescriptor::m_NormMethodType</a></div><div class="ttdeci">NormalizationAlgorithmMethod m_NormMethodType</div><div class="ttdoc">Normalization method algorithm to use (LocalBrightness, LocalContrast).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00795">Descriptors.hpp:795</a></div></div> |
| 5963 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a48cb83dcb58c786ea36f5d37695e75b1"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a48cb83dcb58c786ea36f5d37695e75b1">armnn::QuantizedLstmInputParamsInfo::m_OutputGateBias</a></div><div class="ttdeci">const TensorInfo * m_OutputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00152">QuantizedLstmParams.hpp:152</a></div></div> |
| 5964 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_aa9e49717ebdb741e8c767741647fc618"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">armnn::TransposeConvolution2dDescriptor::m_PadBottom</a></div><div class="ttdeci">uint32_t m_PadBottom</div><div class="ttdoc">Padding bottom value in the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01475">Descriptors.hpp:1475</a></div></div> |
| 5965 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_ace7a1f1f1041b412b7d8ef82b95ff352"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#ace7a1f1f1041b412b7d8ef82b95ff352">armnn::LstmInputParams::m_ForgetGateBias</a></div><div class="ttdeci">const ConstTensor * m_ForgetGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00052">LstmParams.hpp:52</a></div></div> |
| 5966 | <div class="ttc" id="anamespacearmnn__driver_html_a53bfb6eb38a90244ba166b641a6965d9"><div class="ttname"><a href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">armnn_driver::OperandType</a></div><div class="ttdeci">::android::nn::OperandType OperandType</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00048">ConversionUtils.hpp:48</a></div></div> |
| 5967 | <div class="ttc" id="anamespacearmnn_html_abe18a5033f2ab9c0de82c676b48f5437a810f43f3996922151c39b76143faeecc"><div class="ttname"><a href="namespacearmnn.html#abe18a5033f2ab9c0de82c676b48f5437a810f43f3996922151c39b76143faeecc">armnn::NormalizationAlgorithmChannel::Across</a></div><div class="ttdeci">@ Across</div></div> |
| 5968 | <div class="ttc" id="a_tensor_utils_8hpp_html"><div class="ttname"><a href="_tensor_utils_8hpp.html">TensorUtils.hpp</a></div></div> |
| 5969 | <div class="ttc" id="anamespacearmnn__driver_html_aa9ef723ace8562b9636e8387a120fbde"><div class="ttname"><a href="namespacearmnn__driver.html#aa9ef723ace8562b9636e8387a120fbde">armnn_driver::GetInputPaddingScheme</a></div><div class="ttdeci">bool GetInputPaddingScheme(const Operation &operation, uint32_t inputIndex, PaddingScheme &outPaddingScheme, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00782">ConversionUtils.cpp:782</a></div></div> |
| 5970 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a484bafa2f8453a7c5a4a00b92a61b006"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a484bafa2f8453a7c5a4a00b92a61b006">armnn::LstmInputParams::m_CellToInputWeights</a></div><div class="ttdeci">const ConstTensor * m_CellToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00048">LstmParams.hpp:48</a></div></div> |
| 5971 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html">armnn::QuantizedLstmInputParamsInfo</a></div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00119">QuantizedLstmParams.hpp:119</a></div></div> |
| 5972 | <div class="ttc" id="astructarmnn_1_1_mean_descriptor_html_a28e0548abfc4e79c48f29a3d11a062e9"><div class="ttname"><a href="structarmnn_1_1_mean_descriptor.html#a28e0548abfc4e79c48f29a3d11a062e9">armnn::MeanDescriptor::m_KeepDims</a></div><div class="ttdeci">bool m_KeepDims</div><div class="ttdoc">Enable/disable keep dimensions. If true, then the reduced dimensions that are of length 1 are kept.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01192">Descriptors.hpp:1192</a></div></div> |
| 5973 | <div class="ttc" id="astructarmnn_1_1_fill_descriptor_html_ab3ebc5cf4a617d43371a4cb7fecdeb32"><div class="ttname"><a href="structarmnn_1_1_fill_descriptor.html#ab3ebc5cf4a617d43371a4cb7fecdeb32">armnn::FillDescriptor::m_Value</a></div><div class="ttdeci">float m_Value</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00940">Descriptors.hpp:940</a></div></div> |
| 5974 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_af434b509cd5232ef762c4b21b2dfaae0"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#af434b509cd5232ef762c4b21b2dfaae0">armnn::LstmInputParamsInfo::m_RecurrentToOutputWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00096">LstmParams.hpp:96</a></div></div> |
| 5975 | <div class="ttc" id="astructarmnn_1_1_elementwise_binary_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_elementwise_binary_descriptor.html">armnn::ElementwiseBinaryDescriptor</a></div><div class="ttdoc">A ElementwiseBinaryDescriptor for the ElementwiseBinaryLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00109">Descriptors.hpp:109</a></div></div> |
| 5976 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html_a869254cb56968986a78a79e1d6d4a86b"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html#a869254cb56968986a78a79e1d6d4a86b">armnn::ResizeDescriptor::m_Method</a></div><div class="ttdeci">ResizeMethod m_Method</div><div class="ttdoc">The Interpolation method to use (Bilinear, NearestNeighbor).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01012">Descriptors.hpp:1012</a></div></div> |
| 5977 | <div class="ttc" id="astructarmnn_1_1_space_to_batch_nd_descriptor_html_a85f98c94e11f65a6b73f831735c040f3"><div class="ttname"><a href="structarmnn_1_1_space_to_batch_nd_descriptor.html#a85f98c94e11f65a6b73f831735c040f3">armnn::SpaceToBatchNdDescriptor::m_PadList</a></div><div class="ttdeci">std::vector< std::pair< unsigned int, unsigned int > > m_PadList</div><div class="ttdoc">Specifies the padding values for the input dimension: heightPad{top, bottom} widthPad{left,...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01069">Descriptors.hpp:1069</a></div></div> |
| 5978 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a6e30c7b3451da3ea9cf4259fb602e6e6"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a6e30c7b3451da3ea9cf4259fb602e6e6">armnn::LstmInputParams::m_InputToOutputWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00043">LstmParams.hpp:43</a></div></div> |
| 5979 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_a2837b4396f20c956952d1a7286cab5f8"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">armnn::LstmDescriptor::m_PeepholeEnabled</a></div><div class="ttdeci">bool m_PeepholeEnabled</div><div class="ttdoc">Enable/disable peephole.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01148">Descriptors.hpp:1148</a></div></div> |
| 5980 | <div class="ttc" id="aclassarmnn_1_1_tensor_shape_html"><div class="ttname"><a href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a></div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00020">Tensor.hpp:20</a></div></div> |
| 5981 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a33242cb43250f04f1736161ca8ad2db9"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a33242cb43250f04f1736161ca8ad2db9">armnn::QuantizedLstmInputParamsInfo::m_InputToOutputWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00142">QuantizedLstmParams.hpp:142</a></div></div> |
| 5982 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a513ad7774e0f8969ab68d91a04ecca65"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a513ad7774e0f8969ab68d91a04ecca65">armnn::LstmInputParamsInfo::m_ForgetLayerNormWeights</a></div><div class="ttdeci">const TensorInfo * m_ForgetLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00107">LstmParams.hpp:107</a></div></div> |
| 5983 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a685739c4eb65a580e075282cfe6787d6"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a685739c4eb65a580e075282cfe6787d6">armnn::TensorInfo::SetQuantizationScale</a></div><div class="ttdeci">void SetQuantizationScale(float scale)</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00477">Tensor.cpp:477</a></div></div> |
| 5984 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_afe1f0f09d49ad2befc01f8789187b7dd"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#afe1f0f09d49ad2befc01f8789187b7dd">armnn::NormalizationDescriptor::m_NormChannelType</a></div><div class="ttdeci">NormalizationAlgorithmChannel m_NormChannelType</div><div class="ttdoc">Normalization channel algorithm to use (Across, Within).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00793">Descriptors.hpp:793</a></div></div> |
| 5985 | <div class="ttc" id="anamespacearmnn_html_ac4f8557279754ed7b3f749d55b0e3047a8321e79c278ec510f63675c040594892"><div class="ttname"><a href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047a8321e79c278ec510f63675c040594892">armnn::BinaryOperation::Maximum</a></div><div class="ttdeci">@ Maximum</div></div> |
| 5986 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a35b112e30c3eb153806a2a8c16d178e3"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a35b112e30c3eb153806a2a8c16d178e3">armnn::LstmInputParams::m_CellToForgetWeights</a></div><div class="ttdeci">const ConstTensor * m_CellToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00049">LstmParams.hpp:49</a></div></div> |
| 5987 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a435d3651482bbfcc11263b4e4e0c900f"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a435d3651482bbfcc11263b4e4e0c900f">armnn::LstmInputParams::m_RecurrentToInputWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00044">LstmParams.hpp:44</a></div></div> |
| 5988 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_a86e88bef0df4df96df752b4b8955a3af"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#a86e88bef0df4df96df752b4b8955a3af">armnn::LstmDescriptor::m_ClippingThresProj</a></div><div class="ttdeci">float m_ClippingThresProj</div><div class="ttdoc">Clipping threshold value for the projection.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01144">Descriptors.hpp:1144</a></div></div> |
| 5989 | <div class="ttc" id="anamespacearmnn__driver_html_a8d4f662a4fb5b9272724bdaeacbd370a"><div class="ttname"><a href="namespacearmnn__driver.html#a8d4f662a4fb5b9272724bdaeacbd370a">armnn_driver::ConvertPaddings</a></div><div class="ttdeci">bool ConvertPaddings(const Operation &operation, const Model &model, ConversionData &data, unsigned int rank, armnn::PadDescriptor &padDescriptor)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00350">ConversionUtils.cpp:350</a></div></div> |
| 5990 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a11b90bd2694aa7f4b434aadd75fbb1c2"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a11b90bd2694aa7f4b434aadd75fbb1c2">armnn::LstmInputParamsInfo::m_RecurrentToForgetWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00094">LstmParams.hpp:94</a></div></div> |
| 5991 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::StridedSliceDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01363">Descriptors.hpp:1363</a></div></div> |
| 5992 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a1ddc4634491d95d37fe15049decadb8c"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a1ddc4634491d95d37fe15049decadb8c">armnn::LstmInputParamsInfo::m_CellToForgetWeights</a></div><div class="ttdeci">const TensorInfo * m_CellToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00098">LstmParams.hpp:98</a></div></div> |
| 5993 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_affcee5f4ab5994a21bee3b78b4e43de3"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">armnn::LstmInputParams::m_InputToInputWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00040">LstmParams.hpp:40</a></div></div> |
| 5994 | <div class="ttc" id="aclassarmnn_1_1_tensor_shape_html_a157e27d41e9f6b21f0d3c025fa47dc24"><div class="ttname"><a href="classarmnn_1_1_tensor_shape.html#a157e27d41e9f6b21f0d3c025fa47dc24">armnn::TensorShape::GetNumDimensions</a></div><div class="ttdeci">unsigned int GetNumDimensions() const</div><div class="ttdoc">Function that returns the tensor rank.</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00174">Tensor.cpp:174</a></div></div> |
| 5995 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_a302b688d88dd73cde0fb1faef6679907"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a302b688d88dd73cde0fb1faef6679907">armnn::DepthwiseConvolution2dDescriptor::m_DilationY</a></div><div class="ttdeci">uint32_t m_DilationY</div><div class="ttdoc">Dilation factor value for height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00706">Descriptors.hpp:706</a></div></div> |
| 5996 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a042ea9d6d3842b87f112db126e806a93"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a042ea9d6d3842b87f112db126e806a93">armnn::QuantizedLstmInputParamsInfo::m_InputGateBias</a></div><div class="ttdeci">const TensorInfo * m_InputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00149">QuantizedLstmParams.hpp:149</a></div></div> |
| 5997 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a56b81ca8ba4b4937e0787e4951f043fc"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">armnn::QuantizedLstmInputParams::m_RecurrentToOutputWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00041">QuantizedLstmParams.hpp:41</a></div></div> |
| 5998 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a56b81ca8ba4b4937e0787e4951f043fc"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a56b81ca8ba4b4937e0787e4951f043fc">armnn::LstmInputParams::m_RecurrentToOutputWeights</a></div><div class="ttdeci">const ConstTensor * m_RecurrentToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00047">LstmParams.hpp:47</a></div></div> |
| 5999 | <div class="ttc" id="astructarmnn_1_1_fully_connected_descriptor_html_aea202e14d8874cefd9a0f778022b7e25"><div class="ttname"><a href="structarmnn_1_1_fully_connected_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">armnn::FullyConnectedDescriptor::m_BiasEnabled</a></div><div class="ttdeci">bool m_BiasEnabled</div><div class="ttdoc">Enable/disable bias.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00526">Descriptors.hpp:526</a></div></div> |
| 6000 | <div class="ttc" id="anamespacearmnn__driver_html"><div class="ttname"><a href="namespacearmnn__driver.html">armnn_driver</a></div><div class="ttdoc">Helper classes.</div><div class="ttdef"><b>Definition:</b> <a href="_armnn_device_8cpp_source.html#l00037">ArmnnDevice.cpp:37</a></div></div> |
| 6001 | <div class="ttc" id="astructarmnn_1_1_pad_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_pad_descriptor.html">armnn::PadDescriptor</a></div><div class="ttdoc">A PadDescriptor for the PadLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01196">Descriptors.hpp:1196</a></div></div> |
| 6002 | <div class="ttc" id="aclassarmnn_1_1_i_output_slot_html_a5ee4a6c9a2481245487b1b1a70d20fd0"><div class="ttname"><a href="classarmnn_1_1_i_output_slot.html#a5ee4a6c9a2481245487b1b1a70d20fd0">armnn::IOutputSlot::SetTensorInfo</a></div><div class="ttdeci">virtual void SetTensorInfo(const TensorInfo &tensorInfo)=0</div></div> |
| 6003 | <div class="ttc" id="astructarmnn_1_1_transpose_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_transpose_descriptor.html">armnn::TransposeDescriptor</a></div><div class="ttdoc">A TransposeDescriptor for the TransposeLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01490">Descriptors.hpp:1490</a></div></div> |
| 6004 | <div class="ttc" id="astructarmnn_1_1_batch_mat_mul_descriptor_html_a112b466e5d2ab9d1887178adbe3afa1c"><div class="ttname"><a href="structarmnn_1_1_batch_mat_mul_descriptor.html#a112b466e5d2ab9d1887178adbe3afa1c">armnn::BatchMatMulDescriptor::m_TransposeY</a></div><div class="ttdeci">bool m_TransposeY</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01613">Descriptors.hpp:1613</a></div></div> |
| 6005 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a8b8fc85ce966c035d789cf22db5088a1"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a8b8fc85ce966c035d789cf22db5088a1">armnn::TensorInfo::GetQuantizationDim</a></div><div class="ttdeci">Optional< unsigned int > GetQuantizationDim() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00498">Tensor.cpp:498</a></div></div> |
| 6006 | <div class="ttc" id="anamespacearmnn__driver_html_a923b39ec5f83672f50289cbaf5de6798"><div class="ttname"><a href="namespacearmnn__driver.html#a923b39ec5f83672f50289cbaf5de6798">armnn_driver::Model</a></div><div class="ttdeci">::android::nn::Model Model</div><div class="ttdoc">Helper classes.</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00045">ConversionUtils.hpp:45</a></div></div> |
| 6007 | <div class="ttc" id="astructarmnn_1_1_instance_normalization_descriptor_html_a5e078fd505aef7bccaa05c8058e096cc"><div class="ttname"><a href="structarmnn_1_1_instance_normalization_descriptor.html#a5e078fd505aef7bccaa05c8058e096cc">armnn::InstanceNormalizationDescriptor::m_Gamma</a></div><div class="ttdeci">float m_Gamma</div><div class="ttdoc">Gamma, the scale scalar value applied for the normalized tensor. Defaults to 1.0.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00865">Descriptors.hpp:865</a></div></div> |
| 6008 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a9e081a9b94defb30d1558dc912507e0e"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">armnn::LstmInputParams::m_InputGateBias</a></div><div class="ttdeci">const ConstTensor * m_InputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00051">LstmParams.hpp:51</a></div></div> |
| 6009 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_aea202e14d8874cefd9a0f778022b7e25"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">armnn::Convolution2dDescriptor::m_BiasEnabled</a></div><div class="ttdeci">bool m_BiasEnabled</div><div class="ttdoc">Enable/disable bias.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00582">Descriptors.hpp:582</a></div></div> |
| 6010 | <div class="ttc" id="anamespacearmnn__driver_html_adfc2c5b834fdb89e5458ddb1d9aa7830"><div class="ttname"><a href="namespacearmnn__driver.html#adfc2c5b834fdb89e5458ddb1d9aa7830">armnn_driver::GetOperandType</a></div><div class="ttdeci">bool GetOperandType(const Operation &operation, uint32_t inputIndex, const Model &model, OperandType &type)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00716">ConversionUtils.hpp:716</a></div></div> |
| 6011 | <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> |
| 6012 | <div class="ttc" id="anamespacearmnn__driver_html_aa43f9c81f830559c15e5af86f8e11482"><div class="ttname"><a href="namespacearmnn__driver.html#aa43f9c81f830559c15e5af86f8e11482">armnn_driver::GetInputInt32</a></div><div class="ttdeci">bool GetInputInt32(const Operation &operation, uint32_t inputIndex, int32_t &outValue, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00815">ConversionUtils.hpp:815</a></div></div> |
| 6013 | <div class="ttc" id="anamespacearmnn__driver_html_acfe37581bc8296b5d4cdb311b1e3f142"><div class="ttname"><a href="namespacearmnn__driver.html#acfe37581bc8296b5d4cdb311b1e3f142">armnn_driver::GetOptionalConvolutionDilationParams</a></div><div class="ttdeci">bool GetOptionalConvolutionDilationParams(const Operation &operation, uint32_t dilationXIndex, ConvolutionDescriptor &descriptor, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00907">ConversionUtils.hpp:907</a></div></div> |
| 6014 | <div class="ttc" id="astructarmnn_1_1_pad_descriptor_html_a410fa919f78af0f0f100bd1594eca4ab"><div class="ttname"><a href="structarmnn_1_1_pad_descriptor.html#a410fa919f78af0f0f100bd1594eca4ab">armnn::PadDescriptor::m_PadValue</a></div><div class="ttdeci">float m_PadValue</div><div class="ttdoc">Optional value to use for padding, defaults to 0.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01221">Descriptors.hpp:1221</a></div></div> |
| 6015 | <div class="ttc" id="anamespacearmnn__driver_html_abf92490dc588f48af86fc12184841fd5"><div class="ttname"><a href="namespacearmnn__driver.html#abf92490dc588f48af86fc12184841fd5">armnn_driver::GetInputScalar</a></div><div class="ttdeci">bool GetInputScalar(const Operation &operation, uint32_t inputIndex, OperandType type, OutputType &outValue, const Model &model, const ConversionData &data, bool optional=false)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00775">ConversionUtils.hpp:775</a></div></div> |
| 6016 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_a002bc30e590d78cbb4f4d12171055ca7"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">armnn::DepthwiseConvolution2dDescriptor::m_PadRight</a></div><div class="ttdeci">uint32_t m_PadRight</div><div class="ttdoc">Padding right value in the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00694">Descriptors.hpp:694</a></div></div> |
| 6017 | <div class="ttc" id="astructarmnn_1_1_activation_descriptor_html_af10fa7883e3579950f477bee92a64844"><div class="ttname"><a href="structarmnn_1_1_activation_descriptor.html#af10fa7883e3579950f477bee92a64844">armnn::ActivationDescriptor::m_Function</a></div><div class="ttdeci">ActivationFunction m_Function</div><div class="ttdoc">The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu,...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00059">Descriptors.hpp:59</a></div></div> |
| 6018 | <div class="ttc" id="aclassarmnn__driver_1_1_layer_input_handle_html"><div class="ttname"><a href="classarmnn__driver_1_1_layer_input_handle.html">armnn_driver::LayerInputHandle</a></div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00068">ConversionUtils.hpp:68</a></div></div> |
| 6019 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::NormalizationDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00805">Descriptors.hpp:805</a></div></div> |
| 6020 | <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> |
| 6021 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a15f9f65126ad3d4d82d6c19d0662ae01"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a15f9f65126ad3d4d82d6c19d0662ae01">armnn::QuantizedLstmInputParamsInfo::m_InputToInputWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00139">QuantizedLstmParams.hpp:139</a></div></div> |
| 6022 | <div class="ttc" id="anamespacearmnn__driver_html_a5378da53b59c1c826e6320383ca0257e"><div class="ttname"><a href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">armnn_driver::SetupAndTrackLayerOutputSlot</a></div><div class="ttdeci">bool SetupAndTrackLayerOutputSlot(const Operation &operation, uint32_t operationOutputIndex, armnn::IConnectableLayer &layer, uint32_t layerOutputIndex, const Model &model, ConversionData &data, const armnn::TensorInfo *overrideOutputInfo, const std::function< void(const armnn::TensorInfo &, bool &)> &validateFunc, const ActivationFn &activationFunction, bool inferOutputShapes)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00987">ConversionUtils.cpp:987</a></div></div> |
| 6023 | <div class="ttc" id="anamespacearmnn_html_abc05539fc6e7907f32ef0fb242e3b3b0aa0ec87054b5e5b7847d0d8780a01a3d5"><div class="ttname"><a href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0aa0ec87054b5e5b7847d0d8780a01a3d5">armnn::ReduceOperation::Sum</a></div><div class="ttdeci">@ Sum</div></div> |
| 6024 | <div class="ttc" id="anamespacearmnn_utils_html_a0d3b1be320610515e0cac8d745d9f8c2"><div class="ttname"><a href="namespacearmnn_utils.html#a0d3b1be320610515e0cac8d745d9f8c2">armnnUtils::ExpandDims</a></div><div class="ttdeci">armnn::TensorShape ExpandDims(const armnn::TensorShape &tensorShape, int axis)</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_utils_8cpp_source.html#l00142">TensorUtils.cpp:142</a></div></div> |
| 6025 | <div class="ttc" id="anamespacearmnn_html_aa02b9e06fb20fa3c13da0427e6ee5ab2"><div class="ttname"><a href="namespacearmnn.html#aa02b9e06fb20fa3c13da0427e6ee5ab2">armnn::GetDataTypeSize</a></div><div class="ttdeci">constexpr unsigned int GetDataTypeSize(DataType dataType)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00182">TypesUtils.hpp:182</a></div></div> |
| 6026 | <div class="ttc" id="aclassarmnn__driver_1_1_const_tensor_pin_html_a971023b9874b0d773e8aa4c6d6744a2b"><div class="ttname"><a href="classarmnn__driver_1_1_const_tensor_pin.html#a971023b9874b0d773e8aa4c6d6744a2b">armnn_driver::ConstTensorPin::GetConstTensor</a></div><div class="ttdeci">const armnn::ConstTensor & GetConstTensor() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00122">ConversionUtils.cpp:122</a></div></div> |
| 6027 | <div class="ttc" id="aclassarmnn_1_1_base_tensor_html_a8b5d0f8a24e9d9238f412260a552acf8"><div class="ttname"><a href="classarmnn_1_1_base_tensor.html#a8b5d0f8a24e9d9238f412260a552acf8">armnn::BaseTensor::GetShape</a></div><div class="ttdeci">const TensorShape & GetShape() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00299">Tensor.hpp:299</a></div></div> |
| 6028 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::Convolution2dDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00584">Descriptors.hpp:584</a></div></div> |
| 6029 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_aa6a518b65088f34803b3214334bdff61"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#aa6a518b65088f34803b3214334bdff61">armnn::QLstmDescriptor::m_ProjectionClip</a></div><div class="ttdeci">float m_ProjectionClip</div><div class="ttdoc">Clipping threshold value for the projection.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01416">Descriptors.hpp:1416</a></div></div> |
| 6030 | <div class="ttc" id="astructarmnn_1_1_gather_descriptor_html_a35d11c7d509d1adbae1ae01c58394a7f"><div class="ttname"><a href="structarmnn_1_1_gather_descriptor.html#a35d11c7d509d1adbae1ae01c58394a7f">armnn::GatherDescriptor::m_Axis</a></div><div class="ttdeci">int32_t m_Axis</div><div class="ttdoc">The axis in params to gather indices from.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00981">Descriptors.hpp:981</a></div></div> |
| 6031 | <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> |
| 6032 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_aa9e49717ebdb741e8c767741647fc618"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">armnn::Convolution2dDescriptor::m_PadBottom</a></div><div class="ttdeci">uint32_t m_PadBottom</div><div class="ttdoc">Padding bottom value in the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00572">Descriptors.hpp:572</a></div></div> |
| 6033 | <div class="ttc" id="aclassarmnn_1_1_permutation_vector_html"><div class="ttname"><a href="classarmnn_1_1_permutation_vector.html">armnn::PermutationVector</a></div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00314">Types.hpp:314</a></div></div> |
| 6034 | <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> |
| 6035 | <div class="ttc" id="aclassarmnn_1_1_exception_html"><div class="ttname"><a href="classarmnn_1_1_exception.html">armnn::Exception</a></div><div class="ttdoc">Base class for all ArmNN exceptions so that users can filter to just those.</div><div class="ttdef"><b>Definition:</b> <a href="_exceptions_8hpp_source.html#l00046">Exceptions.hpp:46</a></div></div> |
| 6036 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a31da1ead6794dd64571afdd0b6efc771"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">armnn::QuantizedLstmInputParams::m_InputToForgetWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00034">QuantizedLstmParams.hpp:34</a></div></div> |
| 6037 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_ac1fe174bbadfb39a2b636940c2e647c8"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#ac1fe174bbadfb39a2b636940c2e647c8">armnn::TransposeConvolution2dDescriptor::m_StrideY</a></div><div class="ttdeci">uint32_t m_StrideY</div><div class="ttdoc">Stride value when proceeding through input for the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01479">Descriptors.hpp:1479</a></div></div> |
| 6038 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_ac37e49c0d6e6e54f9d2015d0f11f8ee7"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#ac37e49c0d6e6e54f9d2015d0f11f8ee7">armnn::StridedSliceDescriptor::m_EndMask</a></div><div class="ttdeci">int32_t m_EndMask</div><div class="ttdoc">End mask value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01353">Descriptors.hpp:1353</a></div></div> |
| 6039 | <div class="ttc" id="aclassarmnn_1_1_base_tensor_html_a8aeddebdcf02e1832b22203c08a6b678"><div class="ttname"><a href="classarmnn_1_1_base_tensor.html#a8aeddebdcf02e1832b22203c08a6b678">armnn::BaseTensor::GetInfo</a></div><div class="ttdeci">const TensorInfo & GetInfo() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00297">Tensor.hpp:297</a></div></div> |
| 6040 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a8c0f6d48705f40c5590dde09be262222"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">armnn::QuantizedLstmInputParams::m_OutputGateBias</a></div><div class="ttdeci">const ConstTensor * m_OutputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00046">QuantizedLstmParams.hpp:46</a></div></div> |
| 6041 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_af07c65d3b7886a00f10c69093e76a341"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#af07c65d3b7886a00f10c69093e76a341">armnn::LstmInputParamsInfo::m_InputToCellWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00091">LstmParams.hpp:91</a></div></div> |
| 6042 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_ab173a067eeb7295d84f5327bcc05a6c1"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#ab173a067eeb7295d84f5327bcc05a6c1">armnn::LstmInputParamsInfo::m_CellBias</a></div><div class="ttdeci">const TensorInfo * m_CellBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00102">LstmParams.hpp:102</a></div></div> |
| 6043 | <div class="ttc" id="anamespacearmnn_html_a1cfaa710db2a54673b21d2ea2da757c8"><div class="ttname"><a href="namespacearmnn.html#a1cfaa710db2a54673b21d2ea2da757c8">armnn::UnaryOperation</a></div><div class="ttdeci">UnaryOperation</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00125">Types.hpp:125</a></div></div> |
| 6044 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_afe6a3377c4531315354def9023c8fdda"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">armnn::Convolution2dDescriptor::m_StrideX</a></div><div class="ttdeci">uint32_t m_StrideX</div><div class="ttdoc">Stride value when proceeding through input for the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00574">Descriptors.hpp:574</a></div></div> |
| 6045 | <div class="ttc" id="aclassarmnn__driver_1_1_layer_input_handle_html_a7a886c1735beae3f31f2062e1b327ac4"><div class="ttname"><a href="classarmnn__driver_1_1_layer_input_handle.html#a7a886c1735beae3f31f2062e1b327ac4">armnn_driver::LayerInputHandle::GetOutputSlot</a></div><div class="ttdeci">armnn::IOutputSlot * GetOutputSlot() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00077">ConversionUtils.cpp:77</a></div></div> |
| 6046 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_aa43409f9b457352c95c89f20ce5d844d"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#aa43409f9b457352c95c89f20ce5d844d">armnn::QLstmDescriptor::m_OutputIntermediateScale</a></div><div class="ttdeci">float m_OutputIntermediateScale</div><div class="ttdoc">Output intermediate quantization scale.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01432">Descriptors.hpp:1432</a></div></div> |
| 6047 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_a002bc30e590d78cbb4f4d12171055ca7"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">armnn::Convolution2dDescriptor::m_PadRight</a></div><div class="ttdeci">uint32_t m_PadRight</div><div class="ttdoc">Padding right value in the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00568">Descriptors.hpp:568</a></div></div> |
| 6048 | <div class="ttc" id="aclassarmnn__driver_1_1_const_tensor_pin_html_a0325a5d8b692252432a6bf2069cefebf"><div class="ttname"><a href="classarmnn__driver_1_1_const_tensor_pin.html#a0325a5d8b692252432a6bf2069cefebf">armnn_driver::ConstTensorPin::IsOptional</a></div><div class="ttdeci">bool IsOptional() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00117">ConversionUtils.cpp:117</a></div></div> |
| 6049 | <div class="ttc" id="astructarmnn_1_1_origins_descriptor_html_a5b192c5fcd96a0f75542524cf646b355"><div class="ttname"><a href="structarmnn_1_1_origins_descriptor.html#a5b192c5fcd96a0f75542524cf646b355">armnn::OriginsDescriptor::SetConcatAxis</a></div><div class="ttdeci">void SetConcatAxis(unsigned int concatAxis)</div><div class="ttdoc">Set the concatenation axis value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8cpp_source.html#l00158">Descriptors.cpp:158</a></div></div> |
| 6050 | <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> |
| 6051 | <div class="ttc" id="anamespacearmnn__driver_html_a14977f94479f2f7e13bdf915ac8ce8d0"><div class="ttname"><a href="namespacearmnn__driver.html#a14977f94479f2f7e13bdf915ac8ce8d0">armnn_driver::GetInputOperand</a></div><div class="ttdeci">const Operand * GetInputOperand(const Operation &operation, uint32_t inputIndex, const Model &model, bool failOnIndexOutOfBounds=true)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00662">ConversionUtils.hpp:662</a></div></div> |
| 6052 | <div class="ttc" id="anamespacearmnn_html_a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9"><div class="ttname"><a href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9">armnn::LayerType::Shape</a></div><div class="ttdeci">@ Shape</div></div> |
| 6053 | <div class="ttc" id="anamespacearmnn_html_a5cda3502382f06a64c3cbeb1829bd850"><div class="ttname"><a href="namespacearmnn.html#a5cda3502382f06a64c3cbeb1829bd850">armnn::GetArgMinMaxFunctionAsCString</a></div><div class="ttdeci">constexpr char const * GetArgMinMaxFunctionAsCString(ArgMinMaxFunction function)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00052">TypesUtils.hpp:52</a></div></div> |
| 6054 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a0cd848f65ec31778d708852f0042fe37"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a0cd848f65ec31778d708852f0042fe37">armnn::LstmInputParams::m_InputLayerNormWeights</a></div><div class="ttdeci">const ConstTensor * m_InputLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00057">LstmParams.hpp:57</a></div></div> |
| 6055 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a9ab864961ded72f5ce7ea4eb9921f826"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a9ab864961ded72f5ce7ea4eb9921f826">armnn::LstmInputParamsInfo::m_RecurrentToInputWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00093">LstmParams.hpp:93</a></div></div> |
| 6056 | <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> |
| 6057 | <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> |
| 6058 | <div class="ttc" id="anamespacearmnn_html_abc05539fc6e7907f32ef0fb242e3b3b0a362a33c23b08e4a32a4ec53fbb82cccd"><div class="ttname"><a href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0a362a33c23b08e4a32a4ec53fbb82cccd">armnn::ReduceOperation::Prod</a></div><div class="ttdeci">@ Prod</div></div> |
| 6059 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a21626d0b91f8942c3ef8e48e17db9f21"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a21626d0b91f8942c3ef8e48e17db9f21">armnn::LstmInputParamsInfo::m_ForgetGateBias</a></div><div class="ttdeci">const TensorInfo * m_ForgetGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00101">LstmParams.hpp:101</a></div></div> |
| 6060 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_aa9e49717ebdb741e8c767741647fc618"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aa9e49717ebdb741e8c767741647fc618">armnn::DepthwiseConvolution2dDescriptor::m_PadBottom</a></div><div class="ttdeci">uint32_t m_PadBottom</div><div class="ttdoc">Padding bottom value in the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00698">Descriptors.hpp:698</a></div></div> |
| 6061 | <div class="ttc" id="astructarmnn_1_1_comparison_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_comparison_descriptor.html">armnn::ComparisonDescriptor</a></div><div class="ttdoc">A ComparisonDescriptor for the ComparisonLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00089">Descriptors.hpp:89</a></div></div> |
| 6062 | <div class="ttc" id="astructarmnn_1_1_fill_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_fill_descriptor.html">armnn::FillDescriptor</a></div><div class="ttdoc">A FillDescriptor for the FillLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00925">Descriptors.hpp:925</a></div></div> |
| 6063 | <div class="ttc" id="anamespacearmnn_html_ad8ed01ff3ff33333d8e19db4d2818bb6a9d02ea768c081d4bdb2b7cab0b3f510d"><div class="ttname"><a href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a9d02ea768c081d4bdb2b7cab0b3f510d">armnn::DataType::QAsymmS8</a></div><div class="ttdeci">@ QAsymmS8</div></div> |
| 6064 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_a6d0384878432cfc9652b7ae8bc59506f"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#a6d0384878432cfc9652b7ae8bc59506f">armnn::StridedSliceDescriptor::m_ShrinkAxisMask</a></div><div class="ttdeci">int32_t m_ShrinkAxisMask</div><div class="ttdoc">Shrink axis mask value. If set, the nth specification shrinks the dimensionality by 1.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01355">Descriptors.hpp:1355</a></div></div> |
| 6065 | <div class="ttc" id="astructarmnn_1_1_space_to_depth_descriptor_html_a6c6b8957f1e176867e5fb05b1a1a1486"><div class="ttname"><a href="structarmnn_1_1_space_to_depth_descriptor.html#a6c6b8957f1e176867e5fb05b1a1a1486">armnn::SpaceToDepthDescriptor::m_BlockSize</a></div><div class="ttdeci">unsigned int m_BlockSize</div><div class="ttdoc">Scalar specifying the input block size. It must be >= 1.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01092">Descriptors.hpp:1092</a></div></div> |
| 6066 | <div class="ttc" id="anamespacearmnn_html_a733ae6b70d0bfa43433c3e7606992328"><div class="ttname"><a href="namespacearmnn.html#a733ae6b70d0bfa43433c3e7606992328">armnn::CreateDescriptorForConcatenation</a></div><div class="ttdeci">OriginsDescriptor CreateDescriptorForConcatenation(TensorShapeIt first, TensorShapeIt last, unsigned int concatenationDimension)</div><div class="ttdoc">Convenience template to create an OriginsDescriptor to use when creating a ConcatLayer for performing...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00300">Descriptors.hpp:300</a></div></div> |
| 6067 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a63cbc581012c957f9d68d224ddc3e43c"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a63cbc581012c957f9d68d224ddc3e43c">armnn::TensorInfo::SetQuantizationOffset</a></div><div class="ttdeci">void SetQuantizationOffset(int32_t offset)</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00493">Tensor.cpp:493</a></div></div> |
| 6068 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_af0f796fba1a2be9c56b4c9ee534577ee"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#af0f796fba1a2be9c56b4c9ee534577ee">armnn::LstmInputParams::m_ForgetLayerNormWeights</a></div><div class="ttdeci">const ConstTensor * m_ForgetLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00058">LstmParams.hpp:58</a></div></div> |
| 6069 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html_adcf5037208faac36c0788239a073f75c"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html#adcf5037208faac36c0788239a073f75c">armnn::ResizeDescriptor::m_TargetWidth</a></div><div class="ttdeci">uint32_t m_TargetWidth</div><div class="ttdoc">Target width value.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01007">Descriptors.hpp:1007</a></div></div> |
| 6070 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_a0d53caff836b84204adbd1c28752a201"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#a0d53caff836b84204adbd1c28752a201">armnn::StridedSliceDescriptor::m_Stride</a></div><div class="ttdeci">std::vector< int > m_Stride</div><div class="ttdoc">Stride values for the input that will be sliced.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01346">Descriptors.hpp:1346</a></div></div> |
| 6071 | <div class="ttc" id="astructarmnn_1_1_instance_normalization_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_instance_normalization_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::InstanceNormalizationDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00871">Descriptors.hpp:871</a></div></div> |
| 6072 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html">armnn::LstmDescriptor</a></div><div class="ttdoc">An LstmDescriptor for the LstmLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01102">Descriptors.hpp:1102</a></div></div> |
| 6073 | <div class="ttc" id="anamespacearmnn_html_a2d299363c9fc33334c571fa29ca4f58c"><div class="ttname"><a href="namespacearmnn.html#a2d299363c9fc33334c571fa29ca4f58c">armnn::ComparisonOperation</a></div><div class="ttdeci">ComparisonOperation</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00109">Types.hpp:109</a></div></div> |
| 6074 | <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> |
| 6075 | <div class="ttc" id="astructarmnn_1_1_transpose_descriptor_html_a14433af2b223695b40d8c8f8ba2ebb8f"><div class="ttname"><a href="structarmnn_1_1_transpose_descriptor.html#a14433af2b223695b40d8c8f8ba2ebb8f">armnn::TransposeDescriptor::m_DimMappings</a></div><div class="ttdeci">PermutationVector m_DimMappings</div><div class="ttdoc">Indicates how to translate tensor elements from a given source into the target destination,...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01514">Descriptors.hpp:1514</a></div></div> |
| 6076 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_a78f5b213a603b55d0fabea389e89f203"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a78f5b213a603b55d0fabea389e89f203">armnn::TransposeConvolution2dDescriptor::m_OutputShape</a></div><div class="ttdeci">std::vector< unsigned int > m_OutputShape</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01486">Descriptors.hpp:1486</a></div></div> |
| 6077 | <div class="ttc" id="astructarmnn_1_1_resize_descriptor_html_ae1a4b3b6c60552509b89747cebb900a2"><div class="ttname"><a href="structarmnn_1_1_resize_descriptor.html#ae1a4b3b6c60552509b89747cebb900a2">armnn::ResizeDescriptor::m_AlignCorners</a></div><div class="ttdeci">bool m_AlignCorners</div><div class="ttdoc">Aligned corners.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01016">Descriptors.hpp:1016</a></div></div> |
| 6078 | <div class="ttc" id="astructarmnn_1_1_tile_descriptor_html_a4e66396e5ed6ede8d87dc75103645def"><div class="ttname"><a href="structarmnn_1_1_tile_descriptor.html#a4e66396e5ed6ede8d87dc75103645def">armnn::TileDescriptor::m_Multiples</a></div><div class="ttdeci">std::vector< uint32_t > m_Multiples</div><div class="ttdoc">The vector to multiply the input shape by.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01656">Descriptors.hpp:1656</a></div></div> |
| 6079 | <div class="ttc" id="astructarmnn_1_1_mean_descriptor_html_a1f0d67b087c491248bd1cde3ff995a95"><div class="ttname"><a href="structarmnn_1_1_mean_descriptor.html#a1f0d67b087c491248bd1cde3ff995a95">armnn::MeanDescriptor::m_Axis</a></div><div class="ttdeci">std::vector< unsigned int > m_Axis</div><div class="ttdoc">Values for the dimensions to reduce.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01190">Descriptors.hpp:1190</a></div></div> |
| 6080 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_ad474e5c51a0b194ef32e812b86c0cbdb"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">armnn::LstmDescriptor::m_CifgEnabled</a></div><div class="ttdeci">bool m_CifgEnabled</div><div class="ttdoc">Enable/disable cifg (coupled input & forget gate).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01146">Descriptors.hpp:1146</a></div></div> |
| 6081 | <div class="ttc" id="aclassarmnn_1_1_i_output_slot_html_ac1835f8756a9f03c02fcf9664e3a0fce"><div class="ttname"><a href="classarmnn_1_1_i_output_slot.html#ac1835f8756a9f03c02fcf9664e3a0fce">armnn::IOutputSlot::Connect</a></div><div class="ttdeci">virtual int Connect(IInputSlot &destination)=0</div></div> |
| 6082 | <div class="ttc" id="aclassarmnn_1_1_i_output_slot_html_ab4e57caf059794897b17266090ebcb39"><div class="ttname"><a href="classarmnn_1_1_i_output_slot.html#ab4e57caf059794897b17266090ebcb39">armnn::IOutputSlot::IsTensorInfoSet</a></div><div class="ttdeci">virtual bool IsTensorInfoSet() const =0</div></div> |
| 6083 | <div class="ttc" id="astructarmnn_1_1_logical_binary_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_logical_binary_descriptor.html">armnn::LogicalBinaryDescriptor</a></div><div class="ttdoc">A LogicalBinaryDescriptor for the LogicalBinaryLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01518">Descriptors.hpp:1518</a></div></div> |
| 6084 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_a174279be57d7596eeb04c6b7f7510f99"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#a174279be57d7596eeb04c6b7f7510f99">armnn::NormalizationDescriptor::m_Alpha</a></div><div class="ttdeci">float m_Alpha</div><div class="ttdoc">Alpha value for the normalization equation.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00799">Descriptors.hpp:799</a></div></div> |
| 6085 | <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> |
| 6086 | <div class="ttc" id="anamespacearmnn_html_a6dac966f265381903c8ef4f392becced"><div class="ttname"><a href="namespacearmnn.html#a6dac966f265381903c8ef4f392becced">armnn::GetUnaryOperationAsCString</a></div><div class="ttdeci">constexpr char const * GetUnaryOperationAsCString(UnaryOperation operation)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00092">TypesUtils.hpp:92</a></div></div> |
| 6087 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a8b5d0f8a24e9d9238f412260a552acf8"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a8b5d0f8a24e9d9238f412260a552acf8">armnn::TensorInfo::GetShape</a></div><div class="ttdeci">const TensorShape & GetShape() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00193">Tensor.hpp:193</a></div></div> |
| 6088 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_a4556cbd764d4848d8ad0637a9eed580d"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#a4556cbd764d4848d8ad0637a9eed580d">armnn::QLstmDescriptor::m_HiddenStateZeroPoint</a></div><div class="ttdeci">int32_t m_HiddenStateZeroPoint</div><div class="ttdoc">Hidden State zero point.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01434">Descriptors.hpp:1434</a></div></div> |
| 6089 | <div class="ttc" id="astructarmnn_1_1_convolution2d_descriptor_html_aa3c6a77a963a98ccb8ea7b8fd008a8c1"><div class="ttname"><a href="structarmnn_1_1_convolution2d_descriptor.html#aa3c6a77a963a98ccb8ea7b8fd008a8c1">armnn::Convolution2dDescriptor::m_DilationX</a></div><div class="ttdeci">uint32_t m_DilationX</div><div class="ttdoc">Dilation along x axis.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00578">Descriptors.hpp:578</a></div></div> |
| 6090 | <div class="ttc" id="astructarmnn_1_1_softmax_descriptor_html_a214c3636fdf0ea5bac8edb42d0e6c7f0"><div class="ttname"><a href="structarmnn_1_1_softmax_descriptor.html#a214c3636fdf0ea5bac8edb42d0e6c7f0">armnn::SoftmaxDescriptor::m_Axis</a></div><div class="ttdeci">int m_Axis</div><div class="ttdoc">Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed o...</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00192">Descriptors.hpp:192</a></div></div> |
| 6091 | <div class="ttc" id="astructarmnn__driver_1_1_conversion_data_html_a2db3ae8d422f17d455e0ba0cb6291d2a"><div class="ttname"><a href="structarmnn__driver_1_1_conversion_data.html#a2db3ae8d422f17d455e0ba0cb6291d2a">armnn_driver::ConversionData::m_Network</a></div><div class="ttdeci">armnn::INetworkPtr m_Network</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00062">ConversionUtils.hpp:62</a></div></div> |
| 6092 | <div class="ttc" id="anamespacearmnn__driver_html_ae780aad3075e52e802f9c0ed5d88e570"><div class="ttname"><a href="namespacearmnn__driver.html#ae780aad3075e52e802f9c0ed5d88e570">armnn_driver::IsWeightsValid</a></div><div class="ttdeci">bool IsWeightsValid(const Operation &operation, uint32_t inputIndex, const Model &model, const bool isOptional=true)</div><div class="ttdoc">Utility functions.</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00141">ConversionUtils.cpp:141</a></div></div> |
| 6093 | <div class="ttc" id="anamespacearmnn__driver_html_a9ddfceb499ef7b3d01053399bc79b852"><div class="ttname"><a href="namespacearmnn__driver.html#a9ddfceb499ef7b3d01053399bc79b852">armnn_driver::IsOperandConstant</a></div><div class="ttdeci">bool IsOperandConstant(const Operand &operand)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00731">ConversionUtils.hpp:731</a></div></div> |
| 6094 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_a4a8ec49f130084445d44297549254780"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">armnn::LstmDescriptor::m_LayerNormEnabled</a></div><div class="ttdeci">bool m_LayerNormEnabled</div><div class="ttdoc">Enable/disable layer normalization.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01152">Descriptors.hpp:1152</a></div></div> |
| 6095 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_affcee5f4ab5994a21bee3b78b4e43de3"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#affcee5f4ab5994a21bee3b78b4e43de3">armnn::QuantizedLstmInputParams::m_InputToInputWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00033">QuantizedLstmParams.hpp:33</a></div></div> |
| 6096 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_a56b51f56cef50cdfa554258eecdab046"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">armnn::TransposeConvolution2dDescriptor::m_PadTop</a></div><div class="ttdeci">uint32_t m_PadTop</div><div class="ttdoc">Padding top value in the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01473">Descriptors.hpp:1473</a></div></div> |
| 6097 | <div class="ttc" id="aclassarmnn__driver_1_1_const_tensor_pin_html_a4103db28319d5fc58ba0b7ee20b5da22"><div class="ttname"><a href="classarmnn__driver_1_1_const_tensor_pin.html#a4103db28319d5fc58ba0b7ee20b5da22">armnn_driver::ConstTensorPin::GetConstTensorPtr</a></div><div class="ttdeci">const armnn::ConstTensor * GetConstTensorPtr() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00127">ConversionUtils.cpp:127</a></div></div> |
| 6098 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_a002bc30e590d78cbb4f4d12171055ca7"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a002bc30e590d78cbb4f4d12171055ca7">armnn::TransposeConvolution2dDescriptor::m_PadRight</a></div><div class="ttdeci">uint32_t m_PadRight</div><div class="ttdoc">Padding right value in the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01471">Descriptors.hpp:1471</a></div></div> |
| 6099 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a4a9d678146f572808a361dbdc5489f38"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a4a9d678146f572808a361dbdc5489f38">armnn::QuantizedLstmInputParams::m_CellBias</a></div><div class="ttdeci">const ConstTensor * m_CellBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00045">QuantizedLstmParams.hpp:45</a></div></div> |
| 6100 | <div class="ttc" id="anamespaceandroid_1_1nn_html"><div class="ttname"><a href="namespaceandroid_1_1nn.html">android::nn</a></div><div class="ttdef"><b>Definition:</b> <a href="support__library__service_8cpp_source.html#l00010">support_library_service.cpp:10</a></div></div> |
| 6101 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a8c0f6d48705f40c5590dde09be262222"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a8c0f6d48705f40c5590dde09be262222">armnn::LstmInputParams::m_OutputGateBias</a></div><div class="ttdeci">const ConstTensor * m_OutputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00054">LstmParams.hpp:54</a></div></div> |
| 6102 | <div class="ttc" id="anamespacearmnn__driver_html_a93f4874062e34819806d36b74c73dc61"><div class="ttname"><a href="namespacearmnn__driver.html#a93f4874062e34819806d36b74c73dc61">armnn_driver::GetTensorInfoForOperand</a></div><div class="ttdeci">armnn::TensorInfo GetTensorInfoForOperand(const Operand &operand)</div><div class="ttdef"><b>Definition:</b> <a href="_canonical_utils_8cpp_source.html#l00097">CanonicalUtils.cpp:97</a></div></div> |
| 6103 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_ad474e5c51a0b194ef32e812b86c0cbdb"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#ad474e5c51a0b194ef32e812b86c0cbdb">armnn::QLstmDescriptor::m_CifgEnabled</a></div><div class="ttdeci">bool m_CifgEnabled</div><div class="ttdoc">Enable/disable CIFG (coupled input & forget gate).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01418">Descriptors.hpp:1418</a></div></div> |
| 6104 | <div class="ttc" id="anamespacearmnn__driver_html_a4c6b55ca5080b58c2cc6d7c6c7347d96"><div class="ttname"><a href="namespacearmnn__driver.html#a4c6b55ca5080b58c2cc6d7c6c7347d96">armnn_driver::GetInputActivationFunctionFromTensor</a></div><div class="ttdeci">bool GetInputActivationFunctionFromTensor(const Operation &operation, uint32_t inputIndex, ActivationFn &outActivationFunction, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00870">ConversionUtils.hpp:870</a></div></div> |
| 6105 | <div class="ttc" id="aclassarmnn_1_1_backend_id_html"><div class="ttname"><a href="classarmnn_1_1_backend_id.html">armnn::BackendId</a></div><div class="ttdef"><b>Definition:</b> <a href="_backend_id_8hpp_source.html#l00075">BackendId.hpp:75</a></div></div> |
| 6106 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a6c7873a3320a59a63aa26c006db905c0"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a6c7873a3320a59a63aa26c006db905c0">armnn::QuantizedLstmInputParamsInfo::m_RecurrentToCellWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00146">QuantizedLstmParams.hpp:146</a></div></div> |
| 6107 | <div class="ttc" id="anamespacearmnn_html_ac4f8557279754ed7b3f749d55b0e3047aa1d0ec6d56f8833a078b5a7ac4caf2d4"><div class="ttname"><a href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047aa1d0ec6d56f8833a078b5a7ac4caf2d4">armnn::BinaryOperation::Minimum</a></div><div class="ttdeci">@ Minimum</div></div> |
| 6108 | <div class="ttc" id="anamespacearmnn__driver_html_a3f6b313fd199d1552844f3f95fd25cd2"><div class="ttname"><a href="namespacearmnn__driver.html#a3f6b313fd199d1552844f3f95fd25cd2">armnn_driver::ConvertReduce</a></div><div class="ttdeci">bool ConvertReduce(const Operation &operation, const Model &model, ConversionData &data, armnn::ReduceOperation reduceOperation)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00520">ConversionUtils.cpp:520</a></div></div> |
| 6109 | <div class="ttc" id="astructarmnn_1_1_origins_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_origins_descriptor.html">armnn::OriginsDescriptor</a></div><div class="ttdoc">An OriginsDescriptor for the ConcatLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00201">Descriptors.hpp:201</a></div></div> |
| 6110 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_ab03e6e1514f74427916c892f473fe04c"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#ab03e6e1514f74427916c892f473fe04c">armnn::LstmInputParams::m_ProjectionWeights</a></div><div class="ttdeci">const ConstTensor * m_ProjectionWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00055">LstmParams.hpp:55</a></div></div> |
| 6111 | <div class="ttc" id="anamespacearmnn_html_a56297e0f7b215eea46c818cb7528d9eaa3d90c0a5ab3fcf8e6f6608cb3d3a1559"><div class="ttname"><a href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaa3d90c0a5ab3fcf8e6f6608cb3d3a1559">armnn::ActivationFunction::ReLu</a></div><div class="ttdeci">@ ReLu</div></div> |
| 6112 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_a31da1ead6794dd64571afdd0b6efc771"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#a31da1ead6794dd64571afdd0b6efc771">armnn::LstmInputParams::m_InputToForgetWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00041">LstmParams.hpp:41</a></div></div> |
| 6113 | <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 & GetOutputSlot(unsigned int index) const =0</div><div class="ttdoc">Get the const output slot handle by slot index.</div></div> |
| 6114 | <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 &newShape)</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00195">Tensor.hpp:195</a></div></div> |
| 6115 | <div class="ttc" id="a_exceptions_8hpp_html"><div class="ttname"><a href="_exceptions_8hpp.html">Exceptions.hpp</a></div></div> |
| 6116 | <div class="ttc" id="astructarmnn_1_1_elementwise_unary_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_elementwise_unary_descriptor.html">armnn::ElementwiseUnaryDescriptor</a></div><div class="ttdoc">A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00129">Descriptors.hpp:129</a></div></div> |
| 6117 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html">armnn::TransposeConvolution2dDescriptor</a></div><div class="ttdoc">A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01440">Descriptors.hpp:1440</a></div></div> |
| 6118 | <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 & GetInputSlot(unsigned int index) const =0</div><div class="ttdoc">Get a const input slot handle by slot index.</div></div> |
| 6119 | <div class="ttc" id="astructarmnn_1_1_arg_min_max_descriptor_html_a214c3636fdf0ea5bac8edb42d0e6c7f0"><div class="ttname"><a href="structarmnn_1_1_arg_min_max_descriptor.html#a214c3636fdf0ea5bac8edb42d0e6c7f0">armnn::ArgMinMaxDescriptor::m_Axis</a></div><div class="ttdeci">int m_Axis</div><div class="ttdoc">Axis to reduce across the input tensor.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00083">Descriptors.hpp:83</a></div></div> |
| 6120 | <div class="ttc" id="astructarmnn_1_1_activation_descriptor_html_a28c4c9cb15f6be3499abbc46b356060b"><div class="ttname"><a href="structarmnn_1_1_activation_descriptor.html#a28c4c9cb15f6be3499abbc46b356060b">armnn::ActivationDescriptor::m_B</a></div><div class="ttdeci">float m_B</div><div class="ttdoc">Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00063">Descriptors.hpp:63</a></div></div> |
| 6121 | <div class="ttc" id="astructarmnn_1_1_origins_descriptor_html_a2b125117aa61f9baf3a9cb8658aa61a2"><div class="ttname"><a href="structarmnn_1_1_origins_descriptor.html#a2b125117aa61f9baf3a9cb8658aa61a2">armnn::OriginsDescriptor::SetViewOriginCoord</a></div><div class="ttdeci">Status SetViewOriginCoord(uint32_t view, uint32_t coord, uint32_t value)</div><div class="ttdoc">@Brief Set the view origin coordinates.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8cpp_source.html#l00167">Descriptors.cpp:167</a></div></div> |
| 6122 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a16b4d4acd2a0d6d533ff3fe8b8e71765"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a16b4d4acd2a0d6d533ff3fe8b8e71765">armnn::LstmInputParamsInfo::m_CellToOutputWeights</a></div><div class="ttdeci">const TensorInfo * m_CellToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00099">LstmParams.hpp:99</a></div></div> |
| 6123 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html">armnn::LstmInputParamsInfo</a></div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00063">LstmParams.hpp:63</a></div></div> |
| 6124 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_af8f724af7210b52529216feefa993c98"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#af8f724af7210b52529216feefa993c98">armnn::QLstmDescriptor::m_HiddenStateScale</a></div><div class="ttdeci">float m_HiddenStateScale</div><div class="ttdoc">Hidden State quantization scale.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01436">Descriptors.hpp:1436</a></div></div> |
| 6125 | <div class="ttc" id="astructarmnn_1_1_normalization_descriptor_html_a8526ea7cf860d8e7f8340e9f9354f9f0"><div class="ttname"><a href="structarmnn_1_1_normalization_descriptor.html#a8526ea7cf860d8e7f8340e9f9354f9f0">armnn::NormalizationDescriptor::m_K</a></div><div class="ttdeci">float m_K</div><div class="ttdoc">Kappa value used for the across channel normalization equation.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00803">Descriptors.hpp:803</a></div></div> |
| 6126 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_a6c9de81fc65b3c4924cab11907075a17"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#a6c9de81fc65b3c4924cab11907075a17">armnn::LstmDescriptor::m_ProjectionEnabled</a></div><div class="ttdeci">bool m_ProjectionEnabled</div><div class="ttdoc">Enable/disable the projection layer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01150">Descriptors.hpp:1150</a></div></div> |
| 6127 | <div class="ttc" id="astructarmnn_1_1_strided_slice_descriptor_html_aa68194dd6258ab5b04123005a066ea25"><div class="ttname"><a href="structarmnn_1_1_strided_slice_descriptor.html#aa68194dd6258ab5b04123005a066ea25">armnn::StridedSliceDescriptor::m_End</a></div><div class="ttdeci">std::vector< int > m_End</div><div class="ttdoc">End values for the input that will be sliced.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01344">Descriptors.hpp:1344</a></div></div> |
| 6128 | <div class="ttc" id="anamespacearmnn_html_abc05539fc6e7907f32ef0fb242e3b3b0a78d811e98514cd165dda532286610fd2"><div class="ttname"><a href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0a78d811e98514cd165dda532286610fd2">armnn::ReduceOperation::Min</a></div><div class="ttdeci">@ Min</div></div> |
| 6129 | <div class="ttc" id="aclassarmnn__driver_1_1_converter_html_ad3a9c1424b37970a54624f514196b985"><div class="ttname"><a href="classarmnn__driver_1_1_converter.html#ad3a9c1424b37970a54624f514196b985">armnn_driver::Converter::Operation</a></div><div class="ttdeci">::android::nn::Operation Operation</div><div class="ttdef"><b>Definition:</b> <a href="_converter_8hpp_source.html#l00028">Converter.hpp:28</a></div></div> |
| 6130 | <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> |
| 6131 | <div class="ttc" id="aclassarmnn__driver_1_1_layer_input_handle_html_acbafd23115b06f0a82eab45bd1e3cff9"><div class="ttname"><a href="classarmnn__driver_1_1_layer_input_handle.html#acbafd23115b06f0a82eab45bd1e3cff9">armnn_driver::LayerInputHandle::GetTensorInfo</a></div><div class="ttdeci">const armnn::TensorInfo & GetTensorInfo() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00057">ConversionUtils.cpp:57</a></div></div> |
| 6132 | <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> |
| 6133 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a9ab864961ded72f5ce7ea4eb9921f826"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a9ab864961ded72f5ce7ea4eb9921f826">armnn::QuantizedLstmInputParamsInfo::m_RecurrentToInputWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToInputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00144">QuantizedLstmParams.hpp:144</a></div></div> |
| 6134 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a4b3f6e6f2268416ffd7a34fda95ffd0b"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a4b3f6e6f2268416ffd7a34fda95ffd0b">armnn::QuantizedLstmInputParamsInfo::m_InputToForgetWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00140">QuantizedLstmParams.hpp:140</a></div></div> |
| 6135 | <div class="ttc" id="anamespacearmnn__driver_html_a85d53debffd9611a3b5bc6031afd5073"><div class="ttname"><a href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">armnn_driver::AreDynamicTensorsSupported</a></div><div class="ttdeci">bool AreDynamicTensorsSupported()</div><div class="ttdoc">Checks for ArmNN support of dynamic tensors.</div><div class="ttdef"><b>Definition:</b> <a href="_canonical_utils_8cpp_source.html#l00505">CanonicalUtils.cpp:505</a></div></div> |
| 6136 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_a9bc46a2ad56319786bad30be9ab2dc6a"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a9bc46a2ad56319786bad30be9ab2dc6a">armnn::TransposeConvolution2dDescriptor::m_OutputShapeEnabled</a></div><div class="ttdeci">bool m_OutputShapeEnabled</div><div class="ttdoc">Output shape if it has been specified.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01485">Descriptors.hpp:1485</a></div></div> |
| 6137 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_aea202e14d8874cefd9a0f778022b7e25"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#aea202e14d8874cefd9a0f778022b7e25">armnn::TransposeConvolution2dDescriptor::m_BiasEnabled</a></div><div class="ttdeci">bool m_BiasEnabled</div><div class="ttdoc">Enable/disable bias.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01481">Descriptors.hpp:1481</a></div></div> |
| 6138 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_ac81fb0e66dc623dc37c77f219f53a6d3"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#ac81fb0e66dc623dc37c77f219f53a6d3">armnn::QLstmDescriptor::m_CellClip</a></div><div class="ttdeci">float m_CellClip</div><div class="ttdoc">Clipping threshold value for the cell state.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01414">Descriptors.hpp:1414</a></div></div> |
| 6139 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a042ea9d6d3842b87f112db126e806a93"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a042ea9d6d3842b87f112db126e806a93">armnn::LstmInputParamsInfo::m_InputGateBias</a></div><div class="ttdeci">const TensorInfo * m_InputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00100">LstmParams.hpp:100</a></div></div> |
| 6140 | <div class="ttc" id="aclassarmnn__driver_1_1_const_tensor_pin_html"><div class="ttname"><a href="classarmnn__driver_1_1_const_tensor_pin.html">armnn_driver::ConstTensorPin</a></div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00092">ConversionUtils.hpp:92</a></div></div> |
| 6141 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_a4a8ec49f130084445d44297549254780"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#a4a8ec49f130084445d44297549254780">armnn::QLstmDescriptor::m_LayerNormEnabled</a></div><div class="ttdeci">bool m_LayerNormEnabled</div><div class="ttdoc">Enable/disable layer normalization.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01424">Descriptors.hpp:1424</a></div></div> |
| 6142 | <div class="ttc" id="aclassarmnn__driver_1_1_const_tensor_pin_html_ac532c4b500b1a85ea22217f2c65a70ed"><div class="ttname"><a href="classarmnn__driver_1_1_const_tensor_pin.html#ac532c4b500b1a85ea22217f2c65a70ed">armnn_driver::ConstTensorPin::IsValid</a></div><div class="ttdeci">bool IsValid() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00112">ConversionUtils.cpp:112</a></div></div> |
| 6143 | <div class="ttc" id="anamespacearmnn__driver_html_a37ca67c646393225921543233a873803"><div class="ttname"><a href="namespacearmnn__driver.html#a37ca67c646393225921543233a873803">armnn_driver::DequantizeAndMakeConstTensorPin</a></div><div class="ttdeci">ConstTensorPin DequantizeAndMakeConstTensorPin(const Operation &operation, const Model &model, const ConversionData &data, size_t operandIndex, bool optional)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00752">ConversionUtils.cpp:752</a></div></div> |
| 6144 | <div class="ttc" id="anamespacearmnn_html_ac4f8557279754ed7b3f749d55b0e3047a43d11850893d2fe84a1e618121c1cc0a"><div class="ttname"><a href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047a43d11850893d2fe84a1e618121c1cc0a">armnn::BinaryOperation::Div</a></div><div class="ttdeci">@ Div</div></div> |
| 6145 | <div class="ttc" id="astructarmnn_1_1_transpose_convolution2d_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_transpose_convolution2d_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::TransposeConvolution2dDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01483">Descriptors.hpp:1483</a></div></div> |
| 6146 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a9e081a9b94defb30d1558dc912507e0e"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a9e081a9b94defb30d1558dc912507e0e">armnn::QuantizedLstmInputParams::m_InputGateBias</a></div><div class="ttdeci">const ConstTensor * m_InputGateBias</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00043">QuantizedLstmParams.hpp:43</a></div></div> |
| 6147 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_ae1b07ed928036004bd257169e5aeeef4"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#ae1b07ed928036004bd257169e5aeeef4">armnn::LstmDescriptor::m_ActivationFunc</a></div><div class="ttdeci">uint32_t m_ActivationFunc</div><div class="ttdoc">The activation function to use.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01140">Descriptors.hpp:1140</a></div></div> |
| 6148 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a4b3f6e6f2268416ffd7a34fda95ffd0b"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a4b3f6e6f2268416ffd7a34fda95ffd0b">armnn::LstmInputParamsInfo::m_InputToForgetWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00090">LstmParams.hpp:90</a></div></div> |
| 6149 | <div class="ttc" id="astructarmnn_1_1_batch_to_space_nd_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_batch_to_space_nd_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::BatchToSpaceNdDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00902">Descriptors.hpp:902</a></div></div> |
| 6150 | <div class="ttc" id="aclassarmnn_1_1_tensor_info_html_a770b51078da02f44a819e9f95d8058b5"><div class="ttname"><a href="classarmnn_1_1_tensor_info.html#a770b51078da02f44a819e9f95d8058b5">armnn::TensorInfo::GetQuantizationOffset</a></div><div class="ttdeci">int32_t GetQuantizationOffset() const</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00482">Tensor.cpp:482</a></div></div> |
| 6151 | <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> |
| 6152 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_aa3c6a77a963a98ccb8ea7b8fd008a8c1"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#aa3c6a77a963a98ccb8ea7b8fd008a8c1">armnn::DepthwiseConvolution2dDescriptor::m_DilationX</a></div><div class="ttdeci">uint32_t m_DilationX</div><div class="ttdoc">Dilation factor value for width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00704">Descriptors.hpp:704</a></div></div> |
| 6153 | <div class="ttc" id="anamespacearmnn__driver_html_ac092fff5c81f7938eea9a2479f014c66"><div class="ttname"><a href="namespacearmnn__driver.html#ac092fff5c81f7938eea9a2479f014c66">armnn_driver::GetInputActivationFunction</a></div><div class="ttdeci">bool GetInputActivationFunction(const Operation &operation, uint32_t inputIndex, ActivationFn &outActivationFunction, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00856">ConversionUtils.hpp:856</a></div></div> |
| 6154 | <div class="ttc" id="astructarmnn_1_1_lstm_descriptor_html_a531a3907ec13d3772370da88030191a5"><div class="ttname"><a href="structarmnn_1_1_lstm_descriptor.html#a531a3907ec13d3772370da88030191a5">armnn::LstmDescriptor::m_ClippingThresCell</a></div><div class="ttdeci">float m_ClippingThresCell</div><div class="ttdoc">Clipping threshold value for the cell state.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01142">Descriptors.hpp:1142</a></div></div> |
| 6155 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_a33242cb43250f04f1736161ca8ad2db9"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#a33242cb43250f04f1736161ca8ad2db9">armnn::LstmInputParamsInfo::m_InputToOutputWeights</a></div><div class="ttdeci">const TensorInfo * m_InputToOutputWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00092">LstmParams.hpp:92</a></div></div> |
| 6156 | <div class="ttc" id="aclassarmnn_1_1_null_pointer_exception_html"><div class="ttname"><a href="classarmnn_1_1_null_pointer_exception.html">armnn::NullPointerException</a></div><div class="ttdef"><b>Definition:</b> <a href="_exceptions_8hpp_source.html#l00146">Exceptions.hpp:146</a></div></div> |
| 6157 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_ae22fc962c59e7c24986718f5af0020db"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#ae22fc962c59e7c24986718f5af0020db">armnn::LstmInputParamsInfo::m_ProjectionBias</a></div><div class="ttdeci">const TensorInfo * m_ProjectionBias</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00105">LstmParams.hpp:105</a></div></div> |
| 6158 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html">armnn::LstmInputParams</a></div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00013">LstmParams.hpp:13</a></div></div> |
| 6159 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_info_html_ac6fb4a3a8e7e8a359ac34badc8178b1a"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params_info.html#ac6fb4a3a8e7e8a359ac34badc8178b1a">armnn::LstmInputParamsInfo::m_ProjectionWeights</a></div><div class="ttdeci">const TensorInfo * m_ProjectionWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00104">LstmParams.hpp:104</a></div></div> |
| 6160 | <div class="ttc" id="astructarmnn_1_1_lstm_input_params_html_ad0b8c32bb5381f4cc999093ba3b98b43"><div class="ttname"><a href="structarmnn_1_1_lstm_input_params.html#ad0b8c32bb5381f4cc999093ba3b98b43">armnn::LstmInputParams::m_CellLayerNormWeights</a></div><div class="ttdeci">const ConstTensor * m_CellLayerNormWeights</div><div class="ttdef"><b>Definition:</b> <a href="_lstm_params_8hpp_source.html#l00059">LstmParams.hpp:59</a></div></div> |
| 6161 | <div class="ttc" id="anamespacearmnn_html_aabb76a77e95921785f576bb29b495cd8"><div class="ttname"><a href="namespacearmnn.html#aabb76a77e95921785f576bb29b495cd8">armnn::GetComparisonOperationAsCString</a></div><div class="ttdeci">constexpr char const * GetComparisonOperationAsCString(ComparisonOperation operation)</div><div class="ttdef"><b>Definition:</b> <a href="_types_utils_8hpp_source.html#l00062">TypesUtils.hpp:62</a></div></div> |
| 6162 | <div class="ttc" id="aclassarmnn_1_1_tensor_shape_html_a8846406ac37fbd2204f0be16ee05d5b7"><div class="ttname"><a href="classarmnn_1_1_tensor_shape.html#a8846406ac37fbd2204f0be16ee05d5b7">armnn::TensorShape::GetNumElements</a></div><div class="ttdeci">unsigned int GetNumElements() const</div><div class="ttdoc">Function that calculates the tensor elements by multiplying all dimension size which are Specified.</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8cpp_source.html#l00181">Tensor.cpp:181</a></div></div> |
| 6163 | <div class="ttc" id="astructarmnn_1_1_mean_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_mean_descriptor.html">armnn::MeanDescriptor</a></div><div class="ttdoc">A MeanDescriptor for the MeanLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01172">Descriptors.hpp:1172</a></div></div> |
| 6164 | <div class="ttc" id="astructarmnn_1_1_q_lstm_descriptor_html_a2837b4396f20c956952d1a7286cab5f8"><div class="ttname"><a href="structarmnn_1_1_q_lstm_descriptor.html#a2837b4396f20c956952d1a7286cab5f8">armnn::QLstmDescriptor::m_PeepholeEnabled</a></div><div class="ttdeci">bool m_PeepholeEnabled</div><div class="ttdoc">Enable/disable peephole.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01420">Descriptors.hpp:1420</a></div></div> |
| 6165 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html">armnn::QuantizedLstmInputParams</a></div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00013">QuantizedLstmParams.hpp:13</a></div></div> |
| 6166 | <div class="ttc" id="astructarmnn__driver_1_1_conversion_data_html_af3239731a47cf4c08a040c42ee855edc"><div class="ttname"><a href="structarmnn__driver_1_1_conversion_data.html#af3239731a47cf4c08a040c42ee855edc">armnn_driver::ConversionData::m_Backends</a></div><div class="ttdeci">const std::vector< armnn::BackendId > m_Backends</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00061">ConversionUtils.hpp:61</a></div></div> |
| 6167 | <div class="ttc" id="aclassarmnn_1_1_optional_reference_switch_html_a77c7d528ac063d870b8c8426ec81c1c3"><div class="ttname"><a href="classarmnn_1_1_optional_reference_switch.html#a77c7d528ac063d870b8c8426ec81c1c3">armnn::OptionalReferenceSwitch::value</a></div><div class="ttdeci">const T & value() const</div><div class="ttdef"><b>Definition:</b> <a href="_optional_8hpp_source.html#l00146">Optional.hpp:146</a></div></div> |
| 6168 | <div class="ttc" id="astructarmnn_1_1_tile_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_tile_descriptor.html">armnn::TileDescriptor</a></div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01640">Descriptors.hpp:1640</a></div></div> |
| 6169 | <div class="ttc" id="a_converter_8hpp_html"><div class="ttname"><a href="_converter_8hpp.html">Converter.hpp</a></div></div> |
| 6170 | <div class="ttc" id="anamespacearmnn__driver_html_a7bf2bd59bc8ca21e0dd906044bfc2faf"><div class="ttname"><a href="namespacearmnn__driver.html#a7bf2bd59bc8ca21e0dd906044bfc2faf">armnn_driver::IsConnectedToDequantize</a></div><div class="ttdeci">bool IsConnectedToDequantize(armnn::IOutputSlot *ioutputSlot)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l01064">ConversionUtils.cpp:1064</a></div></div> |
| 6171 | <div class="ttc" id="astructarmnn_1_1_softmax_descriptor_html"><div class="ttname"><a href="structarmnn_1_1_softmax_descriptor.html">armnn::SoftmaxDescriptor</a></div><div class="ttdoc">A SoftmaxDescriptor for the SoftmaxLayer.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00177">Descriptors.hpp:177</a></div></div> |
| 6172 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_html_a49e11acda22742cbaf6f1b259ead0d84"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params.html#a49e11acda22742cbaf6f1b259ead0d84">armnn::QuantizedLstmInputParams::m_InputToCellWeights</a></div><div class="ttdeci">const ConstTensor * m_InputToCellWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00035">QuantizedLstmParams.hpp:35</a></div></div> |
| 6173 | <div class="ttc" id="astructarmnn_1_1_quantized_lstm_input_params_info_html_a11b90bd2694aa7f4b434aadd75fbb1c2"><div class="ttname"><a href="structarmnn_1_1_quantized_lstm_input_params_info.html#a11b90bd2694aa7f4b434aadd75fbb1c2">armnn::QuantizedLstmInputParamsInfo::m_RecurrentToForgetWeights</a></div><div class="ttdeci">const TensorInfo * m_RecurrentToForgetWeights</div><div class="ttdef"><b>Definition:</b> <a href="_quantized_lstm_params_8hpp_source.html#l00145">QuantizedLstmParams.hpp:145</a></div></div> |
| 6174 | <div class="ttc" id="astructarmnn_1_1_instance_normalization_descriptor_html_a11c821c7524251004a72ed13c510853c"><div class="ttname"><a href="structarmnn_1_1_instance_normalization_descriptor.html#a11c821c7524251004a72ed13c510853c">armnn::InstanceNormalizationDescriptor::m_Eps</a></div><div class="ttdeci">float m_Eps</div><div class="ttdoc">Epsilon, small scalar value added to variance to avoid dividing by zero. Defaults to 1e-12f.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00869">Descriptors.hpp:869</a></div></div> |
| 6175 | <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> |
| 6176 | <div class="ttc" id="aclassarmnn_1_1_optional_base_html_a86b749ce2c4bc627fa8a1fcfaf0e314f"><div class="ttname"><a href="classarmnn_1_1_optional_base.html#a86b749ce2c4bc627fa8a1fcfaf0e314f">armnn::OptionalBase::has_value</a></div><div class="ttdeci">bool has_value() const noexcept</div><div class="ttdef"><b>Definition:</b> <a href="_optional_8hpp_source.html#l00053">Optional.hpp:53</a></div></div> |
| 6177 | <div class="ttc" id="anamespacearmnn_html_abc05539fc6e7907f32ef0fb242e3b3b0a6a061313d22e51e0f25b7cd4dc065233"><div class="ttname"><a href="namespacearmnn.html#abc05539fc6e7907f32ef0fb242e3b3b0a6a061313d22e51e0f25b7cd4dc065233">armnn::ReduceOperation::Max</a></div><div class="ttdeci">@ Max</div></div> |
| 6178 | <div class="ttc" id="a_test_utils_8cpp_html_a0b295acb179f15eb3fb862b32008f782"><div class="ttname"><a href="_test_utils_8cpp.html#a0b295acb179f15eb3fb862b32008f782">Connect</a></div><div class="ttdeci">void Connect(armnn::IConnectableLayer *from, armnn::IConnectableLayer *to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex, unsigned int toIndex)</div><div class="ttdef"><b>Definition:</b> <a href="_test_utils_8cpp_source.html#l00014">TestUtils.cpp:14</a></div></div> |
| 6179 | <div class="ttc" id="aclassarmnn__driver_1_1_layer_input_handle_html_ac532c4b500b1a85ea22217f2c65a70ed"><div class="ttname"><a href="classarmnn__driver_1_1_layer_input_handle.html#ac532c4b500b1a85ea22217f2c65a70ed">armnn_driver::LayerInputHandle::IsValid</a></div><div class="ttdeci">bool IsValid() const</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8cpp_source.html#l00028">ConversionUtils.cpp:28</a></div></div> |
| 6180 | <div class="ttc" id="anamespacearmnn__driver_html_a560530462631e354561c1763f695351c"><div class="ttname"><a href="namespacearmnn__driver.html#a560530462631e354561c1763f695351c">armnn_driver::GetInputFloat32</a></div><div class="ttdeci">bool GetInputFloat32(const Operation &operation, uint32_t inputIndex, float &outValue, const Model &model, const ConversionData &data)</div><div class="ttdef"><b>Definition:</b> <a href="_conversion_utils_8hpp_source.html#l00824">ConversionUtils.hpp:824</a></div></div> |
| 6181 | <div class="ttc" id="anamespacearmnn_html_a56297e0f7b215eea46c818cb7528d9eaa21eebb164e4b8b9bcf64fdb4d8d5dff4"><div class="ttname"><a href="namespacearmnn.html#a56297e0f7b215eea46c818cb7528d9eaa21eebb164e4b8b9bcf64fdb4d8d5dff4">armnn::ActivationFunction::Sigmoid</a></div><div class="ttdeci">@ Sigmoid</div></div> |
| 6182 | <div class="ttc" id="astructarmnn_1_1_space_to_depth_descriptor_html_a6089e1ca91914015777ea780a513131a"><div class="ttname"><a href="structarmnn_1_1_space_to_depth_descriptor.html#a6089e1ca91914015777ea780a513131a">armnn::SpaceToDepthDescriptor::m_DataLayout</a></div><div class="ttdeci">DataLayout m_DataLayout</div><div class="ttdoc">The data layout to be used (NCHW, NHWC).</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l01095">Descriptors.hpp:1095</a></div></div> |
| 6183 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_afe6a3377c4531315354def9023c8fdda"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#afe6a3377c4531315354def9023c8fdda">armnn::DepthwiseConvolution2dDescriptor::m_StrideX</a></div><div class="ttdeci">uint32_t m_StrideX</div><div class="ttdoc">Stride value when proceeding through input for the width dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00700">Descriptors.hpp:700</a></div></div> |
| 6184 | <div class="ttc" id="astructarmnn_1_1_depthwise_convolution2d_descriptor_html_a56b51f56cef50cdfa554258eecdab046"><div class="ttname"><a href="structarmnn_1_1_depthwise_convolution2d_descriptor.html#a56b51f56cef50cdfa554258eecdab046">armnn::DepthwiseConvolution2dDescriptor::m_PadTop</a></div><div class="ttdeci">uint32_t m_PadTop</div><div class="ttdoc">Padding top value in the height dimension.</div><div class="ttdef"><b>Definition:</b> <a href="_descriptors_8hpp_source.html#l00696">Descriptors.hpp:696</a></div></div> |
| 6185 | <!-- start footer part --> |
| 6186 | <div id="nav-path" class="navpath"><!-- id is needed for treeview function! --> |
| 6187 | <ul> |
| 6188 | <li class="navelem"><a class="el" href="dir_b4dbf96392bc72a402a7206cbda88a03.html">shim</a></li><li class="navelem"><a class="el" href="dir_09eea26cf0279feb9fcfb9b796a3e57e.html">sl</a></li><li class="navelem"><a class="el" href="dir_6de13b49022bb7cfc6557e28050d5cbe.html">canonical</a></li><li class="navelem"><a class="el" href="_converter_8cpp.html">Converter.cpp</a></li> |
| 6189 | <li class="footer">Generated on Thu May 16 2024 09:31:44 for Arm NN by |
| 6190 | <a href="http://www.doxygen.org/index.html"> |
| 6191 | <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li> |
| 6192 | </ul> |
| 6193 | </div> |
| 6194 | </body> |
| 6195 | </html> |