| <!-- HTML header for doxygen 1.8.17--> |
| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.17"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>Arm NN: shim/sl/canonical/Converter.cpp Source File</title> |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="jquery.js"></script> |
| <script type="text/javascript" src="dynsections.js"></script> |
| <link href="navtree.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="resize.js"></script> |
| <script type="text/javascript" src="navtreedata.js"></script> |
| <script type="text/javascript" src="navtree.js"></script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="search/searchdata.js"></script> |
| <script type="text/javascript" src="search/search.js"></script> |
| <script type="text/x-mathjax-config"> |
| MathJax.Hub.Config({ |
| extensions: ["tex2jax.js"], |
| jax: ["input/TeX","output/HTML-CSS"], |
| }); |
| </script> |
| <script type="text/javascript" async="async" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script> |
| <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| <link href="customdoxygen.css" rel="stylesheet" type="text/css"/> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| <div id="titlearea"> |
| <table cellspacing="0" cellpadding="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| <img alt="ArmNN" src="Arm_NN_horizontal_blue.png" style="max-width: 15rem; margin-top: .5rem; margin-left 13px"/> |
| <td id="projectalign" style="padding-left: 0.9em;"> |
| <div id="projectname"> |
|  <span id="projectnumber">24.05</span> |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.17 --> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| /* @license-end */ |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| /* @license-end */</script> |
| <div id="main-nav"></div> |
| </div><!-- top --> |
| <div id="side-nav" class="ui-resizable side-nav-resizable"> |
| <div id="nav-tree"> |
| <div id="nav-tree-contents"> |
| <div id="nav-sync" class="sync"></div> |
| </div> |
| </div> |
| <div id="splitbar" style="-moz-user-select:none;" |
| class="ui-resizable-handle"> |
| </div> |
| </div> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| $(document).ready(function(){initNavTree('_converter_8cpp_source.html',''); initResizable(); }); |
| /* @license-end */ |
| </script> |
| <div id="doc-content"> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">Converter.cpp</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <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> |
| <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> |
| <div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment">// SPDX-License-Identifier: MIT</span></div> |
| <div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment">//</span></div> |
| <div class="line"><a name="l00005"></a><span class="lineno"> 5</span>  </div> |
| <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> |
| <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="preprocessor">#include <half/half.hpp></span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span>  </div> |
| <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> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> {</div> |
| <div class="line"><a name="l00013"></a><span class="lineno"> 13</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00016"></a><span class="lineno"> 16</span>  </div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="keyword">namespace</span></div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> {</div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span>  </div> |
| <div class="line"><a name="l00020"></a><span class="lineno"> 20</span> } <span class="comment">// anonymouse namespace</span></div> |
| <div class="line"><a name="l00021"></a><span class="lineno"> 21</span>  </div> |
| <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> |
| <div class="line"><a name="l00023"></a><span class="lineno"> 23</span> {</div> |
| <div class="line"><a name="l00024"></a><span class="lineno"> 24</span>  <span class="keywordflow">switch</span> (operation.type)</div> |
| <div class="line"><a name="l00025"></a><span class="lineno"> 25</span>  {</div> |
| <div class="line"><a name="l00026"></a><span class="lineno"> 26</span>  <span class="keywordflow">case</span> OperationType::ABS:</div> |
| <div class="line"><a name="l00027"></a><span class="lineno"> 27</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Abs);</div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span>  <span class="keywordflow">case</span> OperationType::ADD:</div> |
| <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> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</span>  <span class="keywordflow">case</span> OperationType::ARGMAX:</div> |
| <div class="line"><a name="l00031"></a><span class="lineno"> 31</span>  <span class="keywordflow">return</span> ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Max);</div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span>  <span class="keywordflow">case</span> OperationType::ARGMIN:</div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span>  <span class="keywordflow">return</span> ConvertArgMinMax(operation, model, data, ArgMinMaxFunction::Min);</div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span>  <span class="keywordflow">case</span> OperationType::AVERAGE_POOL_2D:</div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span>  <span class="keywordflow">return</span> ConvertAveragePool2d(operation, model, data);</div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span>  <span class="keywordflow">case</span> OperationType::BATCH_MATMUL:</div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span>  <span class="keywordflow">return</span> ConvertBatchMatMul(operation, model, data);</div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span>  <span class="keywordflow">case</span> OperationType::BATCH_TO_SPACE_ND:</div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span>  <span class="keywordflow">return</span> ConvertBatchToSpaceNd(operation, model, data);</div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span>  <span class="keywordflow">case</span> OperationType::CAST:</div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span>  <span class="keywordflow">return</span> ConvertCast(operation, model, data);</div> |
| <div class="line"><a name="l00042"></a><span class="lineno"> 42</span>  <span class="keywordflow">case</span> OperationType::CONCATENATION:</div> |
| <div class="line"><a name="l00043"></a><span class="lineno"> 43</span>  <span class="keywordflow">return</span> ConvertConcatenation(operation, model, data);</div> |
| <div class="line"><a name="l00044"></a><span class="lineno"> 44</span>  <span class="keywordflow">case</span> OperationType::CONV_2D:</div> |
| <div class="line"><a name="l00045"></a><span class="lineno"> 45</span>  <span class="keywordflow">return</span> ConvertConv2d(operation, model, data);</div> |
| <div class="line"><a name="l00046"></a><span class="lineno"> 46</span>  <span class="keywordflow">case</span> OperationType::DEPTH_TO_SPACE:</div> |
| <div class="line"><a name="l00047"></a><span class="lineno"> 47</span>  <span class="keywordflow">return</span> ConvertDepthToSpace(operation, model, data);</div> |
| <div class="line"><a name="l00048"></a><span class="lineno"> 48</span>  <span class="keywordflow">case</span> OperationType::DEPTHWISE_CONV_2D:</div> |
| <div class="line"><a name="l00049"></a><span class="lineno"> 49</span>  <span class="keywordflow">return</span> ConvertDepthwiseConv2d(operation, model, data);</div> |
| <div class="line"><a name="l00050"></a><span class="lineno"> 50</span>  <span class="keywordflow">case</span> OperationType::DEQUANTIZE:</div> |
| <div class="line"><a name="l00051"></a><span class="lineno"> 51</span>  <span class="keywordflow">return</span> ConvertDequantize(operation, model, data);</div> |
| <div class="line"><a name="l00052"></a><span class="lineno"> 52</span>  <span class="keywordflow">case</span> OperationType::DIV:</div> |
| <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> |
| <div class="line"><a name="l00054"></a><span class="lineno"> 54</span>  <span class="keywordflow">case</span> OperationType::ELU:</div> |
| <div class="line"><a name="l00055"></a><span class="lineno"> 55</span>  <span class="keywordflow">return</span> ConvertElu(operation, model, data);</div> |
| <div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  <span class="keywordflow">case</span> OperationType::EQUAL:</div> |
| <div class="line"><a name="l00057"></a><span class="lineno"> 57</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::Equal);</div> |
| <div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  <span class="keywordflow">case</span> OperationType::EXP:</div> |
| <div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Exp);</div> |
| <div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  <span class="keywordflow">case</span> OperationType::EXPAND_DIMS:</div> |
| <div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  <span class="keywordflow">return</span> ConvertExpandDims(operation, model, data);</div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  <span class="keywordflow">case</span> OperationType::FILL:</div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span>  <span class="keywordflow">return</span> ConvertFill(operation, model, data);</div> |
| <div class="line"><a name="l00064"></a><span class="lineno"> 64</span>  <span class="keywordflow">case</span> OperationType::FLOOR:</div> |
| <div class="line"><a name="l00065"></a><span class="lineno"> 65</span>  <span class="keywordflow">return</span> ConvertFloor(operation, model, data);</div> |
| <div class="line"><a name="l00066"></a><span class="lineno"> 66</span>  <span class="keywordflow">case</span> OperationType::FULLY_CONNECTED:</div> |
| <div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  <span class="keywordflow">return</span> ConvertFullyConnected(operation, model, data);</div> |
| <div class="line"><a name="l00068"></a><span class="lineno"> 68</span>  <span class="keywordflow">case</span> OperationType::GATHER:</div> |
| <div class="line"><a name="l00069"></a><span class="lineno"> 69</span>  <span class="keywordflow">return</span> ConvertGather(operation, model, data);</div> |
| <div class="line"><a name="l00070"></a><span class="lineno"> 70</span>  <span class="keywordflow">case</span> OperationType::GREATER:</div> |
| <div class="line"><a name="l00071"></a><span class="lineno"> 71</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::Greater);</div> |
| <div class="line"><a name="l00072"></a><span class="lineno"> 72</span>  <span class="keywordflow">case</span> OperationType::GREATER_EQUAL:</div> |
| <div class="line"><a name="l00073"></a><span class="lineno"> 73</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::GreaterOrEqual);</div> |
| <div class="line"><a name="l00074"></a><span class="lineno"> 74</span>  <span class="keywordflow">case</span> OperationType::GROUPED_CONV_2D:</div> |
| <div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  <span class="keywordflow">return</span> ConvertGroupedConv2d(operation, model, data);</div> |
| <div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  <span class="keywordflow">case</span> OperationType::HARD_SWISH:</div> |
| <div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  <span class="keywordflow">return</span> ConvertHardSwish(operation, model, data);</div> |
| <div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  <span class="keywordflow">case</span> OperationType::INSTANCE_NORMALIZATION:</div> |
| <div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  <span class="keywordflow">return</span> ConvertInstanceNormalization(operation, model, data);</div> |
| <div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  <span class="keywordflow">case</span> OperationType::L2_NORMALIZATION:</div> |
| <div class="line"><a name="l00081"></a><span class="lineno"> 81</span>  <span class="keywordflow">return</span> ConvertL2Normalization(operation, model, data);</div> |
| <div class="line"><a name="l00082"></a><span class="lineno"> 82</span>  <span class="keywordflow">case</span> OperationType::L2_POOL_2D:</div> |
| <div class="line"><a name="l00083"></a><span class="lineno"> 83</span>  <span class="keywordflow">return</span> ConvertL2Pool2d(operation, model, data);</div> |
| <div class="line"><a name="l00084"></a><span class="lineno"> 84</span>  <span class="keywordflow">case</span> OperationType::LESS:</div> |
| <div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::Less);</div> |
| <div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  <span class="keywordflow">case</span> OperationType::LESS_EQUAL:</div> |
| <div class="line"><a name="l00087"></a><span class="lineno"> 87</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::LessOrEqual);</div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span>  <span class="keywordflow">case</span> OperationType::LOCAL_RESPONSE_NORMALIZATION:</div> |
| <div class="line"><a name="l00089"></a><span class="lineno"> 89</span>  <span class="keywordflow">return</span> ConvertLocalResponseNormalization(operation, model, data);</div> |
| <div class="line"><a name="l00090"></a><span class="lineno"> 90</span>  <span class="keywordflow">case</span> OperationType::LOG:</div> |
| <div class="line"><a name="l00091"></a><span class="lineno"> 91</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Log);</div> |
| <div class="line"><a name="l00092"></a><span class="lineno"> 92</span>  <span class="keywordflow">case</span> OperationType::LOGICAL_AND:</div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  <span class="keywordflow">return</span> ConvertLogicalBinary(operation, model, data, LogicalBinaryOperation::LogicalAnd);</div> |
| <div class="line"><a name="l00094"></a><span class="lineno"> 94</span>  <span class="keywordflow">case</span> OperationType::LOGICAL_NOT:</div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::LogicalNot);</div> |
| <div class="line"><a name="l00096"></a><span class="lineno"> 96</span>  <span class="keywordflow">case</span> OperationType::LOGICAL_OR:</div> |
| <div class="line"><a name="l00097"></a><span class="lineno"> 97</span>  <span class="keywordflow">return</span> ConvertLogicalBinary(operation, model, data, LogicalBinaryOperation::LogicalOr);</div> |
| <div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  <span class="keywordflow">case</span> OperationType::LOGISTIC:</div> |
| <div class="line"><a name="l00099"></a><span class="lineno"> 99</span>  <span class="keywordflow">return</span> ConvertLogistic(operation, model, data);</div> |
| <div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  <span class="keywordflow">case</span> OperationType::LOG_SOFTMAX:</div> |
| <div class="line"><a name="l00101"></a><span class="lineno"> 101</span>  <span class="keywordflow">return</span> ConvertLogSoftmax(operation, model, data);</div> |
| <div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  <span class="keywordflow">case</span> OperationType::LSTM:</div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span>  <span class="keywordflow">return</span> ConvertLstm(operation, model, data);</div> |
| <div class="line"><a name="l00104"></a><span class="lineno"> 104</span>  <span class="keywordflow">case</span> OperationType::MAX_POOL_2D:</div> |
| <div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  <span class="keywordflow">return</span> ConvertMaxPool2d(operation, model, data);</div> |
| <div class="line"><a name="l00106"></a><span class="lineno"> 106</span>  <span class="keywordflow">case</span> OperationType::MAXIMUM:</div> |
| <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> |
| <div class="line"><a name="l00108"></a><span class="lineno"> 108</span>  <span class="keywordflow">case</span> OperationType::MEAN:</div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  <span class="keywordflow">return</span> ConvertMean(operation, model, data);</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  <span class="keywordflow">case</span> OperationType::MINIMUM:</div> |
| <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> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  <span class="keywordflow">case</span> OperationType::MUL:</div> |
| <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> |
| <div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <span class="keywordflow">case</span> OperationType::NEG:</div> |
| <div class="line"><a name="l00115"></a><span class="lineno"> 115</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Neg);</div> |
| <div class="line"><a name="l00116"></a><span class="lineno"> 116</span>  <span class="keywordflow">case</span> OperationType::NOT_EQUAL:</div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  <span class="keywordflow">return</span> ConvertComparison(operation, model, data, ComparisonOperation::NotEqual);</div> |
| <div class="line"><a name="l00118"></a><span class="lineno"> 118</span>  <span class="keywordflow">case</span> OperationType::PAD:</div> |
| <div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <span class="keywordflow">return</span> ConvertPad(operation, model, data);</div> |
| <div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  <span class="keywordflow">case</span> OperationType::PAD_V2:</div> |
| <div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  <span class="keywordflow">return</span> ConvertPadV2(operation, model, data);</div> |
| <div class="line"><a name="l00122"></a><span class="lineno"> 122</span>  <span class="keywordflow">case</span> OperationType::PRELU:</div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  <span class="keywordflow">return</span> ConvertPrelu(operation, model, data);</div> |
| <div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  <span class="keywordflow">case</span> OperationType::POW:</div> |
| <div class="line"><a name="l00125"></a><span class="lineno"> 125</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, BinaryOperation::Power);</div> |
| <div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  <span class="keywordflow">case</span> OperationType::QUANTIZE:</div> |
| <div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  <span class="keywordflow">return</span> ConvertQuantize(operation, model, data);</div> |
| <div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  <span class="keywordflow">case</span> OperationType::QUANTIZED_LSTM:</div> |
| <div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <span class="keywordflow">return</span> ConvertQuantizedLstm(operation, model, data);</div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  <span class="keywordflow">case</span> OperationType::QUANTIZED_16BIT_LSTM:</div> |
| <div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  <span class="keywordflow">return</span> ConvertQuantized16BitLstm(operation, model, data);</div> |
| <div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  <span class="keywordflow">case</span> OperationType::RANK:</div> |
| <div class="line"><a name="l00133"></a><span class="lineno"> 133</span>  <span class="keywordflow">return</span> ConvertRank(operation, model, data);</div> |
| <div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  <span class="keywordflow">case</span> OperationType::REDUCE_MAX:</div> |
| <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> |
| <div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  <span class="keywordflow">case</span> OperationType::REDUCE_MIN:</div> |
| <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> |
| <div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  <span class="keywordflow">case</span> OperationType::REDUCE_PROD:</div> |
| <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> |
| <div class="line"><a name="l00140"></a><span class="lineno"> 140</span>  <span class="keywordflow">case</span> OperationType::REDUCE_SUM:</div> |
| <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> |
| <div class="line"><a name="l00142"></a><span class="lineno"> 142</span>  <span class="keywordflow">case</span> OperationType::RELU:</div> |
| <div class="line"><a name="l00143"></a><span class="lineno"> 143</span>  <span class="keywordflow">return</span> ConvertReLu(operation, model, data);</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keywordflow">case</span> OperationType::RELU1:</div> |
| <div class="line"><a name="l00145"></a><span class="lineno"> 145</span>  <span class="keywordflow">return</span> ConvertReLu1(operation, model, data);</div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  <span class="keywordflow">case</span> OperationType::RELU6:</div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  <span class="keywordflow">return</span> ConvertReLu6(operation, model, data);</div> |
| <div class="line"><a name="l00148"></a><span class="lineno"> 148</span>  <span class="keywordflow">case</span> OperationType::RESHAPE:</div> |
| <div class="line"><a name="l00149"></a><span class="lineno"> 149</span>  <span class="keywordflow">return</span> ConvertReshape(operation, model, data);</div> |
| <div class="line"><a name="l00150"></a><span class="lineno"> 150</span>  <span class="keywordflow">case</span> OperationType::RESIZE_BILINEAR:</div> |
| <div class="line"><a name="l00151"></a><span class="lineno"> 151</span>  <span class="keywordflow">return</span> ConvertResize(operation, model, data, ResizeMethod::Bilinear);</div> |
| <div class="line"><a name="l00152"></a><span class="lineno"> 152</span>  <span class="keywordflow">case</span> OperationType::RESIZE_NEAREST_NEIGHBOR:</div> |
| <div class="line"><a name="l00153"></a><span class="lineno"> 153</span>  <span class="keywordflow">return</span> ConvertResize(operation, model, data, ResizeMethod::NearestNeighbor);</div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  <span class="keywordflow">case</span> OperationType::REVERSE:</div> |
| <div class="line"><a name="l00155"></a><span class="lineno"> 155</span>  <span class="keywordflow">return</span> ConvertReverseV2(operation, model, data);</div> |
| <div class="line"><a name="l00156"></a><span class="lineno"> 156</span>  <span class="keywordflow">case</span> OperationType::RSQRT:</div> |
| <div class="line"><a name="l00157"></a><span class="lineno"> 157</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Rsqrt);</div> |
| <div class="line"><a name="l00158"></a><span class="lineno"> 158</span>  <span class="keywordflow">case</span> OperationType::SIN:</div> |
| <div class="line"><a name="l00159"></a><span class="lineno"> 159</span>  <span class="keywordflow">return</span> ConvertElementwiseUnary(operation, model, data, UnaryOperation::Sin);</div> |
| <div class="line"><a name="l00160"></a><span class="lineno"> 160</span>  <span class="keywordflow">case</span> OperationType::SOFTMAX:</div> |
| <div class="line"><a name="l00161"></a><span class="lineno"> 161</span>  <span class="keywordflow">return</span> ConvertSoftmax(operation, model, data);</div> |
| <div class="line"><a name="l00162"></a><span class="lineno"> 162</span>  <span class="keywordflow">case</span> OperationType::SPACE_TO_BATCH_ND :</div> |
| <div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  <span class="keywordflow">return</span> ConvertSpaceToBatchNd(operation, model, data);</div> |
| <div class="line"><a name="l00164"></a><span class="lineno"> 164</span>  <span class="keywordflow">case</span> OperationType::SPACE_TO_DEPTH:</div> |
| <div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  <span class="keywordflow">return</span> ConvertSpaceToDepth(operation, model, data);</div> |
| <div class="line"><a name="l00166"></a><span class="lineno"> 166</span>  <span class="keywordflow">case</span> OperationType::SQRT:</div> |
| <div class="line"><a name="l00167"></a><span class="lineno"> 167</span>  <span class="keywordflow">return</span> ConvertSqrt(operation, model, data);</div> |
| <div class="line"><a name="l00168"></a><span class="lineno"> 168</span>  <span class="keywordflow">case</span> OperationType::SQUEEZE:</div> |
| <div class="line"><a name="l00169"></a><span class="lineno"> 169</span>  <span class="keywordflow">return</span> ConvertSqueeze(operation, model, data);</div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  <span class="keywordflow">case</span> OperationType::STRIDED_SLICE:</div> |
| <div class="line"><a name="l00171"></a><span class="lineno"> 171</span>  <span class="keywordflow">return</span> ConvertStridedSlice(operation, model, data);</div> |
| <div class="line"><a name="l00172"></a><span class="lineno"> 172</span>  <span class="keywordflow">case</span> OperationType::SUB:</div> |
| <div class="line"><a name="l00173"></a><span class="lineno"> 173</span>  <span class="keywordflow">return</span> ConvertElementwiseBinary(operation, model, data, BinaryOperation::Sub);</div> |
| <div class="line"><a name="l00174"></a><span class="lineno"> 174</span>  <span class="keywordflow">case</span> OperationType::TILE:</div> |
| <div class="line"><a name="l00175"></a><span class="lineno"> 175</span>  <span class="keywordflow">return</span> ConvertTile(operation, model, data);</div> |
| <div class="line"><a name="l00176"></a><span class="lineno"> 176</span>  <span class="keywordflow">case</span> OperationType::TRANSPOSE:</div> |
| <div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  <span class="keywordflow">return</span> ConvertTranspose(operation, model, data);</div> |
| <div class="line"><a name="l00178"></a><span class="lineno"> 178</span>  <span class="keywordflow">case</span> OperationType::TRANSPOSE_CONV_2D:</div> |
| <div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  <span class="keywordflow">return</span> ConvertTransposeConv2d(operation, model, data);</div> |
| <div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  <span class="keywordflow">case</span> OperationType::TANH:</div> |
| <div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  <span class="keywordflow">return</span> ConvertTanH(operation, model, data);</div> |
| <div class="line"><a name="l00182"></a><span class="lineno"> 182</span>  <span class="keywordflow">default</span>:</div> |
| <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> |
| <div class="line"><a name="l00184"></a><span class="lineno"> 184</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00185"></a><span class="lineno"> 185</span>  }</div> |
| <div class="line"><a name="l00186"></a><span class="lineno"> 186</span> }</div> |
| <div class="line"><a name="l00187"></a><span class="lineno"> 187</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00191"></a><span class="lineno"> 191</span>  <a class="code" href="namespacearmnn.html#ae7e8cbf71db6a490789ca6dcaa8deeae">armnn::ArgMinMaxFunction</a> argMinMaxFunction)</div> |
| <div class="line"><a name="l00192"></a><span class="lineno"> 192</span> {</div> |
| <div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertArgMinMax()"</span>;</div> |
| <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> |
| <div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  </div> |
| <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> |
| <div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  </div> |
| <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> |
| <div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  {</div> |
| <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> |
| <div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  }</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  </div> |
| <div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  int32_t axis;</div> |
| <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> |
| <div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  {</div> |
| <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> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  }</div> |
| <div class="line"><a name="l00208"></a><span class="lineno"> 208</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  </div> |
| <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> |
| <div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  }</div> |
| <div class="line"><a name="l00219"></a><span class="lineno"> 219</span>  </div> |
| <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> |
| <div class="line"><a name="l00221"></a><span class="lineno"> 221</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l00222"></a><span class="lineno"> 222</span>  {</div> |
| <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> |
| <div class="line"><a name="l00224"></a><span class="lineno"> 224</span>  }</div> |
| <div class="line"><a name="l00225"></a><span class="lineno"> 225</span>  </div> |
| <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> |
| <div class="line"><a name="l00227"></a><span class="lineno"> 227</span>  </div> |
| <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> |
| <div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  </div> |
| <div class="line"><a name="l00234"></a><span class="lineno"> 234</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00237"></a><span class="lineno"> 237</span>  {</div> |
| <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> |
| <div class="line"><a name="l00239"></a><span class="lineno"> 239</span>  IsArgMinMaxSupported,</div> |
| <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> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  isSupported,</div> |
| <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  setBackend,</div> |
| <div class="line"><a name="l00243"></a><span class="lineno"> 243</span>  inputInfo0,</div> |
| <div class="line"><a name="l00244"></a><span class="lineno"> 244</span>  outputInfo,</div> |
| <div class="line"><a name="l00245"></a><span class="lineno"> 245</span>  descriptor);</div> |
| <div class="line"><a name="l00246"></a><span class="lineno"> 246</span>  };</div> |
| <div class="line"><a name="l00247"></a><span class="lineno"> 247</span>  </div> |
| <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> |
| <div class="line"><a name="l00249"></a><span class="lineno"> 249</span>  {</div> |
| <div class="line"><a name="l00250"></a><span class="lineno"> 250</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00251"></a><span class="lineno"> 251</span>  }</div> |
| <div class="line"><a name="l00252"></a><span class="lineno"> 252</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00253"></a><span class="lineno"> 253</span>  {</div> |
| <div class="line"><a name="l00254"></a><span class="lineno"> 254</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l00255"></a><span class="lineno"> 255</span>  }</div> |
| <div class="line"><a name="l00256"></a><span class="lineno"> 256</span>  </div> |
| <div class="line"><a name="l00257"></a><span class="lineno"> 257</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  {</div> |
| <div class="line"><a name="l00259"></a><span class="lineno"> 259</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00260"></a><span class="lineno"> 260</span>  }</div> |
| <div class="line"><a name="l00261"></a><span class="lineno"> 261</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00264"></a><span class="lineno"> 264</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <div class="line"><a name="l00265"></a><span class="lineno"> 265</span>  </div> |
| <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> |
| <div class="line"><a name="l00267"></a><span class="lineno"> 267</span>  </div> |
| <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> |
| <div class="line"><a name="l00269"></a><span class="lineno"> 269</span> }</div> |
| <div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  </div> |
| <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> |
| <div class="line"><a name="l00272"></a><span class="lineno"> 272</span> {</div> |
| <div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertAveragePool2d()"</span>;</div> |
| <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> |
| <div class="line"><a name="l00275"></a><span class="lineno"> 275</span> }</div> |
| <div class="line"><a name="l00276"></a><span class="lineno"> 276</span>  </div> |
| <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> |
| <div class="line"><a name="l00278"></a><span class="lineno"> 278</span> {</div> |
| <div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertBatchMatMul()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00282"></a><span class="lineno"> 282</span>  </div> |
| <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> |
| <div class="line"><a name="l00284"></a><span class="lineno"> 284</span>  {</div> |
| <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> |
| <div class="line"><a name="l00286"></a><span class="lineno"> 286</span>  }</div> |
| <div class="line"><a name="l00287"></a><span class="lineno"> 287</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00290"></a><span class="lineno"> 290</span>  </div> |
| <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> |
| <div class="line"><a name="l00292"></a><span class="lineno"> 292</span>  <span class="keywordflow">if</span> (rankInput0 > 4 || rankInput0 < 2)</div> |
| <div class="line"><a name="l00293"></a><span class="lineno"> 293</span>  {</div> |
| <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> |
| <div class="line"><a name="l00295"></a><span class="lineno"> 295</span>  }</div> |
| <div class="line"><a name="l00296"></a><span class="lineno"> 296</span>  </div> |
| <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> |
| <div class="line"><a name="l00298"></a><span class="lineno"> 298</span>  <span class="keywordflow">if</span> (rankInput1 > 4 || rankInput1 < 2)</div> |
| <div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  {</div> |
| <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> |
| <div class="line"><a name="l00301"></a><span class="lineno"> 301</span>  }</div> |
| <div class="line"><a name="l00302"></a><span class="lineno"> 302</span>  </div> |
| <div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  <span class="comment">// Determine data type of input tensor 0</span></div> |
| <div class="line"><a name="l00304"></a><span class="lineno"> 304</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> input0Type;</div> |
| <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> |
| <div class="line"><a name="l00306"></a><span class="lineno"> 306</span>  {</div> |
| <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> |
| <div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  }</div> |
| <div class="line"><a name="l00309"></a><span class="lineno"> 309</span>  </div> |
| <div class="line"><a name="l00310"></a><span class="lineno"> 310</span>  <span class="comment">// Determine data type of input tensor 0</span></div> |
| <div class="line"><a name="l00311"></a><span class="lineno"> 311</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> input1Type;</div> |
| <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> |
| <div class="line"><a name="l00313"></a><span class="lineno"> 313</span>  {</div> |
| <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> |
| <div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  }</div> |
| <div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  </div> |
| <div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  <span class="keywordflow">if</span> (input0Type != input1Type)</div> |
| <div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  {</div> |
| <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> |
| <div class="line"><a name="l00320"></a><span class="lineno"> 320</span>  }</div> |
| <div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  </div> |
| <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> |
| <div class="line"><a name="l00323"></a><span class="lineno"> 323</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l00324"></a><span class="lineno"> 324</span>  {</div> |
| <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> |
| <div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  }</div> |
| <div class="line"><a name="l00327"></a><span class="lineno"> 327</span>  </div> |
| <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> |
| <div class="line"><a name="l00329"></a><span class="lineno"> 329</span>  </div> |
| <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> |
| <div class="line"><a name="l00331"></a><span class="lineno"> 331</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00336"></a><span class="lineno"> 336</span>  </div> |
| <div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00340"></a><span class="lineno"> 340</span>  {</div> |
| <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> |
| <div class="line"><a name="l00342"></a><span class="lineno"> 342</span>  IsBatchMatMulSupported,</div> |
| <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> |
| <div class="line"><a name="l00344"></a><span class="lineno"> 344</span>  isSupported,</div> |
| <div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  setBackend,</div> |
| <div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  inputInfo0,</div> |
| <div class="line"><a name="l00347"></a><span class="lineno"> 347</span>  inputInfo1,</div> |
| <div class="line"><a name="l00348"></a><span class="lineno"> 348</span>  outputInfo,</div> |
| <div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  batchMatMulDesc);</div> |
| <div class="line"><a name="l00350"></a><span class="lineno"> 350</span>  };</div> |
| <div class="line"><a name="l00351"></a><span class="lineno"> 351</span>  </div> |
| <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> |
| <div class="line"><a name="l00353"></a><span class="lineno"> 353</span>  {</div> |
| <div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l00355"></a><span class="lineno"> 355</span>  }</div> |
| <div class="line"><a name="l00356"></a><span class="lineno"> 356</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00357"></a><span class="lineno"> 357</span>  {</div> |
| <div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00359"></a><span class="lineno"> 359</span>  }</div> |
| <div class="line"><a name="l00360"></a><span class="lineno"> 360</span>  </div> |
| <div class="line"><a name="l00361"></a><span class="lineno"> 361</span>  </div> |
| <div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  {</div> |
| <div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  }</div> |
| <div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00372"></a><span class="lineno"> 372</span>  </div> |
| <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> |
| <div class="line"><a name="l00374"></a><span class="lineno"> 374</span> }</div> |
| <div class="line"><a name="l00375"></a><span class="lineno"> 375</span>  </div> |
| <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> |
| <div class="line"><a name="l00377"></a><span class="lineno"> 377</span> {</div> |
| <div class="line"><a name="l00378"></a><span class="lineno"> 378</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertBatchToSpaceNd()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00381"></a><span class="lineno"> 381</span>  {</div> |
| <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> |
| <div class="line"><a name="l00383"></a><span class="lineno"> 383</span>  }</div> |
| <div class="line"><a name="l00384"></a><span class="lineno"> 384</span>  </div> |
| <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> |
| <div class="line"><a name="l00386"></a><span class="lineno"> 386</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l00387"></a><span class="lineno"> 387</span>  {</div> |
| <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> |
| <div class="line"><a name="l00389"></a><span class="lineno"> 389</span>  }</div> |
| <div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  </div> |
| <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> |
| <div class="line"><a name="l00392"></a><span class="lineno"> 392</span>  </div> |
| <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> |
| <div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  <span class="keywordflow">if</span> (!blockOperand)</div> |
| <div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  {</div> |
| <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> |
| <div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  }</div> |
| <div class="line"><a name="l00398"></a><span class="lineno"> 398</span>  </div> |
| <div class="line"><a name="l00399"></a><span class="lineno"> 399</span>  <span class="comment">// Convert the block operand to int32</span></div> |
| <div class="line"><a name="l00400"></a><span class="lineno"> 400</span>  std::vector<int32_t> block;</div> |
| <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> |
| <div class="line"><a name="l00402"></a><span class="lineno"> 402</span>  {</div> |
| <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> |
| <div class="line"><a name="l00404"></a><span class="lineno"> 404</span>  }</div> |
| <div class="line"><a name="l00405"></a><span class="lineno"> 405</span>  </div> |
| <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> |
| <div class="line"><a name="l00407"></a><span class="lineno"> 407</span>  </div> |
| <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> |
| <div class="line"><a name="l00409"></a><span class="lineno"> 409</span>  <span class="keywordflow">if</span> (rank != 4)</div> |
| <div class="line"><a name="l00410"></a><span class="lineno"> 410</span>  {</div> |
| <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> |
| <div class="line"><a name="l00412"></a><span class="lineno"> 412</span>  }</div> |
| <div class="line"><a name="l00413"></a><span class="lineno"> 413</span>  </div> |
| <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> |
| <div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  {</div> |
| <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> |
| <div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  <span class="stringliteral">" greater than or equal to 1"</span>, __func__);</div> |
| <div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  }</div> |
| <div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  </div> |
| <div class="line"><a name="l00424"></a><span class="lineno"> 424</span>  <span class="keywordflow">if</span> (Is12OrLaterOperand(*output))</div> |
| <div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  {</div> |
| <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> |
| <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00430"></a><span class="lineno"> 430</span>  </div> |
| <div class="line"><a name="l00431"></a><span class="lineno"> 431</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00434"></a><span class="lineno"> 434</span>  {</div> |
| <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> |
| <div class="line"><a name="l00436"></a><span class="lineno"> 436</span>  IsBatchToSpaceNdSupported,</div> |
| <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> |
| <div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  isSupported,</div> |
| <div class="line"><a name="l00439"></a><span class="lineno"> 439</span>  setBackend,</div> |
| <div class="line"><a name="l00440"></a><span class="lineno"> 440</span>  inputInfo,</div> |
| <div class="line"><a name="l00441"></a><span class="lineno"> 441</span>  outputInfo,</div> |
| <div class="line"><a name="l00442"></a><span class="lineno"> 442</span>  batchToSpaceNdDesc);</div> |
| <div class="line"><a name="l00443"></a><span class="lineno"> 443</span>  };</div> |
| <div class="line"><a name="l00444"></a><span class="lineno"> 444</span>  </div> |
| <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> |
| <div class="line"><a name="l00446"></a><span class="lineno"> 446</span>  {</div> |
| <div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l00448"></a><span class="lineno"> 448</span>  }</div> |
| <div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  {</div> |
| <div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  }</div> |
| <div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  </div> |
| <div class="line"><a name="l00454"></a><span class="lineno"> 454</span>  </div> |
| <div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  {</div> |
| <div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  }</div> |
| <div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00462"></a><span class="lineno"> 462</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l00464"></a><span class="lineno"> 464</span>  </div> |
| <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> |
| <div class="line"><a name="l00466"></a><span class="lineno"> 466</span> }</div> |
| <div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  </div> |
| <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> |
| <div class="line"><a name="l00469"></a><span class="lineno"> 469</span> {</div> |
| <div class="line"><a name="l00470"></a><span class="lineno"> 470</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertCast()"</span>;</div> |
| <div class="line"><a name="l00471"></a><span class="lineno"> 471</span>  </div> |
| <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> |
| <div class="line"><a name="l00473"></a><span class="lineno"> 473</span>  </div> |
| <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> |
| <div class="line"><a name="l00475"></a><span class="lineno"> 475</span>  {</div> |
| <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> |
| <div class="line"><a name="l00477"></a><span class="lineno"> 477</span>  }</div> |
| <div class="line"><a name="l00478"></a><span class="lineno"> 478</span>  </div> |
| <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> |
| <div class="line"><a name="l00480"></a><span class="lineno"> 480</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l00481"></a><span class="lineno"> 481</span>  {</div> |
| <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> |
| <div class="line"><a name="l00483"></a><span class="lineno"> 483</span>  }</div> |
| <div class="line"><a name="l00484"></a><span class="lineno"> 484</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00487"></a><span class="lineno"> 487</span>  </div> |
| <div class="line"><a name="l00488"></a><span class="lineno"> 488</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00491"></a><span class="lineno"> 491</span>  {</div> |
| <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> |
| <div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  IsCastSupported,</div> |
| <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> |
| <div class="line"><a name="l00495"></a><span class="lineno"> 495</span>  isSupported,</div> |
| <div class="line"><a name="l00496"></a><span class="lineno"> 496</span>  setBackend,</div> |
| <div class="line"><a name="l00497"></a><span class="lineno"> 497</span>  inputInfo,</div> |
| <div class="line"><a name="l00498"></a><span class="lineno"> 498</span>  outputInfo);</div> |
| <div class="line"><a name="l00499"></a><span class="lineno"> 499</span>  };</div> |
| <div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  </div> |
| <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> |
| <div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  {</div> |
| <div class="line"><a name="l00503"></a><span class="lineno"> 503</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  }</div> |
| <div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00506"></a><span class="lineno"> 506</span>  {</div> |
| <div class="line"><a name="l00507"></a><span class="lineno"> 507</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00508"></a><span class="lineno"> 508</span>  }</div> |
| <div class="line"><a name="l00509"></a><span class="lineno"> 509</span>  </div> |
| <div class="line"><a name="l00510"></a><span class="lineno"> 510</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  {</div> |
| <div class="line"><a name="l00512"></a><span class="lineno"> 512</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00513"></a><span class="lineno"> 513</span>  }</div> |
| <div class="line"><a name="l00514"></a><span class="lineno"> 514</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l00519"></a><span class="lineno"> 519</span>  </div> |
| <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> |
| <div class="line"><a name="l00521"></a><span class="lineno"> 521</span> }</div> |
| <div class="line"><a name="l00522"></a><span class="lineno"> 522</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00526"></a><span class="lineno"> 526</span>  <a class="code" href="namespacearmnn.html#a2d299363c9fc33334c571fa29ca4f58c">ComparisonOperation</a> comparisonOperation)</div> |
| <div class="line"><a name="l00527"></a><span class="lineno"> 527</span> {</div> |
| <div class="line"><a name="l00528"></a><span class="lineno"> 528</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertComparison()"</span>;</div> |
| <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> |
| <div class="line"><a name="l00530"></a><span class="lineno"> 530</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00533"></a><span class="lineno"> 533</span>  </div> |
| <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> |
| <div class="line"><a name="l00535"></a><span class="lineno"> 535</span>  {</div> |
| <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> |
| <div class="line"><a name="l00537"></a><span class="lineno"> 537</span>  }</div> |
| <div class="line"><a name="l00538"></a><span class="lineno"> 538</span>  </div> |
| <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> |
| <div class="line"><a name="l00540"></a><span class="lineno"> 540</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l00541"></a><span class="lineno"> 541</span>  {</div> |
| <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> |
| <div class="line"><a name="l00543"></a><span class="lineno"> 543</span>  }</div> |
| <div class="line"><a name="l00544"></a><span class="lineno"> 544</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00548"></a><span class="lineno"> 548</span>  </div> |
| <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> |
| <div class="line"><a name="l00550"></a><span class="lineno"> 550</span>  </div> |
| <div class="line"><a name="l00551"></a><span class="lineno"> 551</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00554"></a><span class="lineno"> 554</span>  {</div> |
| <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> |
| <div class="line"><a name="l00556"></a><span class="lineno"> 556</span>  IsComparisonSupported,</div> |
| <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> |
| <div class="line"><a name="l00558"></a><span class="lineno"> 558</span>  isSupported,</div> |
| <div class="line"><a name="l00559"></a><span class="lineno"> 559</span>  setBackend,</div> |
| <div class="line"><a name="l00560"></a><span class="lineno"> 560</span>  inputInfo0,</div> |
| <div class="line"><a name="l00561"></a><span class="lineno"> 561</span>  inputInfo1,</div> |
| <div class="line"><a name="l00562"></a><span class="lineno"> 562</span>  outputInfo,</div> |
| <div class="line"><a name="l00563"></a><span class="lineno"> 563</span>  descriptor);</div> |
| <div class="line"><a name="l00564"></a><span class="lineno"> 564</span>  };</div> |
| <div class="line"><a name="l00565"></a><span class="lineno"> 565</span>  </div> |
| <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> |
| <div class="line"><a name="l00567"></a><span class="lineno"> 567</span>  {</div> |
| <div class="line"><a name="l00568"></a><span class="lineno"> 568</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l00569"></a><span class="lineno"> 569</span>  }</div> |
| <div class="line"><a name="l00570"></a><span class="lineno"> 570</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00571"></a><span class="lineno"> 571</span>  {</div> |
| <div class="line"><a name="l00572"></a><span class="lineno"> 572</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00573"></a><span class="lineno"> 573</span>  }</div> |
| <div class="line"><a name="l00574"></a><span class="lineno"> 574</span>  </div> |
| <div class="line"><a name="l00575"></a><span class="lineno"> 575</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00576"></a><span class="lineno"> 576</span>  {</div> |
| <div class="line"><a name="l00577"></a><span class="lineno"> 577</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00578"></a><span class="lineno"> 578</span>  }</div> |
| <div class="line"><a name="l00579"></a><span class="lineno"> 579</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00582"></a><span class="lineno"> 582</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <div class="line"><a name="l00583"></a><span class="lineno"> 583</span>  </div> |
| <div class="line"><a name="l00584"></a><span class="lineno"> 584</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input0, input1, layer, data);</div> |
| <div class="line"><a name="l00585"></a><span class="lineno"> 585</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| <div class="line"><a name="l00586"></a><span class="lineno"> 586</span>  {</div> |
| <div class="line"><a name="l00587"></a><span class="lineno"> 587</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00588"></a><span class="lineno"> 588</span>  }</div> |
| <div class="line"><a name="l00589"></a><span class="lineno"> 589</span>  </div> |
| <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> |
| <div class="line"><a name="l00591"></a><span class="lineno"> 591</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00594"></a><span class="lineno"> 594</span>  }</div> |
| <div class="line"><a name="l00595"></a><span class="lineno"> 595</span>  </div> |
| <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> |
| <div class="line"><a name="l00597"></a><span class="lineno"> 597</span> }</div> |
| <div class="line"><a name="l00598"></a><span class="lineno"> 598</span>  </div> |
| <div class="line"><a name="l00599"></a><span class="lineno"> 599</span>  </div> |
| <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> |
| <div class="line"><a name="l00601"></a><span class="lineno"> 601</span> {</div> |
| <div class="line"><a name="l00602"></a><span class="lineno"> 602</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConcatenation()"</span>;</div> |
| <div class="line"><a name="l00603"></a><span class="lineno"> 603</span>  </div> |
| <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> |
| <div class="line"><a name="l00605"></a><span class="lineno"> 605</span>  <span class="keywordflow">if</span> (operation.inputs.size() <= 1)</div> |
| <div class="line"><a name="l00606"></a><span class="lineno"> 606</span>  {</div> |
| <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> |
| <div class="line"><a name="l00608"></a><span class="lineno"> 608</span>  }</div> |
| <div class="line"><a name="l00609"></a><span class="lineno"> 609</span>  </div> |
| <div class="line"><a name="l00610"></a><span class="lineno"> 610</span>  <span class="comment">// Get inputs and outputs</span></div> |
| <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> |
| <div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  </div> |
| <div class="line"><a name="l00613"></a><span class="lineno"> 613</span>  int32_t concatDim;</div> |
| <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> |
| <div class="line"><a name="l00615"></a><span class="lineno"> 615</span>  {</div> |
| <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> |
| <div class="line"><a name="l00617"></a><span class="lineno"> 617</span>  }</div> |
| <div class="line"><a name="l00618"></a><span class="lineno"> 618</span>  </div> |
| <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> |
| <div class="line"><a name="l00620"></a><span class="lineno"> 620</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l00621"></a><span class="lineno"> 621</span>  {</div> |
| <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> |
| <div class="line"><a name="l00623"></a><span class="lineno"> 623</span>  }</div> |
| <div class="line"><a name="l00624"></a><span class="lineno"> 624</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00628"></a><span class="lineno"> 628</span>  <span class="comment">//</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00632"></a><span class="lineno"> 632</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l00633"></a><span class="lineno"> 633</span>  <span class="keywordflow">if</span> (concatDim < 0)</div> |
| <div class="line"><a name="l00634"></a><span class="lineno"> 634</span>  {</div> |
| <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> |
| <div class="line"><a name="l00636"></a><span class="lineno"> 636</span>  }</div> |
| <div class="line"><a name="l00637"></a><span class="lineno"> 637</span>  </div> |
| <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> |
| <div class="line"><a name="l00639"></a><span class="lineno"> 639</span>  {</div> |
| <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> |
| <div class="line"><a name="l00641"></a><span class="lineno"> 641</span>  }</div> |
| <div class="line"><a name="l00642"></a><span class="lineno"> 642</span>  </div> |
| <div class="line"><a name="l00643"></a><span class="lineno"> 643</span>  std::vector<LayerInputHandle> inputHandles;</div> |
| <div class="line"><a name="l00644"></a><span class="lineno"> 644</span>  std::vector<armnn::TensorShape> inputShapes;</div> |
| <div class="line"><a name="l00645"></a><span class="lineno"> 645</span>  </div> |
| <div class="line"><a name="l00646"></a><span class="lineno"> 646</span>  inputHandles.reserve(numInputTensors);</div> |
| <div class="line"><a name="l00647"></a><span class="lineno"> 647</span>  inputShapes.reserve(numInputTensors);</div> |
| <div class="line"><a name="l00648"></a><span class="lineno"> 648</span>  </div> |
| <div class="line"><a name="l00649"></a><span class="lineno"> 649</span>  <span class="keywordtype">bool</span> inputsHaveBeenReshaped = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00652"></a><span class="lineno"> 652</span>  {</div> |
| <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> |
| <div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  <span class="keywordflow">if</span> (!operand)</div> |
| <div class="line"><a name="l00655"></a><span class="lineno"> 655</span>  {</div> |
| <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> |
| <div class="line"><a name="l00657"></a><span class="lineno"> 657</span>  }</div> |
| <div class="line"><a name="l00658"></a><span class="lineno"> 658</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00661"></a><span class="lineno"> 661</span>  {</div> |
| <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> |
| <div class="line"><a name="l00663"></a><span class="lineno"> 663</span>  }</div> |
| <div class="line"><a name="l00664"></a><span class="lineno"> 664</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00667"></a><span class="lineno"> 667</span>  {</div> |
| <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> |
| <div class="line"><a name="l00669"></a><span class="lineno"> 669</span>  }</div> |
| <div class="line"><a name="l00670"></a><span class="lineno"> 670</span>  </div> |
| <div class="line"><a name="l00671"></a><span class="lineno"> 671</span>  <span class="keywordflow">if</span> (RequiresReshape(operandShape))</div> |
| <div class="line"><a name="l00672"></a><span class="lineno"> 672</span>  {</div> |
| <div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  inputsHaveBeenReshaped = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00674"></a><span class="lineno"> 674</span>  </div> |
| <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> |
| <div class="line"><a name="l00676"></a><span class="lineno"> 676</span>  </div> |
| <div class="line"><a name="l00677"></a><span class="lineno"> 677</span>  <span class="comment">// Expand the tensor to three dimensions</span></div> |
| <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> |
| <div class="line"><a name="l00679"></a><span class="lineno"> 679</span>  {</div> |
| <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> |
| <div class="line"><a name="l00681"></a><span class="lineno"> 681</span>  tensorDimensionsAdded = 1;</div> |
| <div class="line"><a name="l00682"></a><span class="lineno"> 682</span>  }</div> |
| <div class="line"><a name="l00683"></a><span class="lineno"> 683</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00684"></a><span class="lineno"> 684</span>  {</div> |
| <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> |
| <div class="line"><a name="l00686"></a><span class="lineno"> 686</span>  tensorDimensionsAdded = 2;</div> |
| <div class="line"><a name="l00687"></a><span class="lineno"> 687</span>  }</div> |
| <div class="line"><a name="l00688"></a><span class="lineno"> 688</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00691"></a><span class="lineno"> 691</span>  </div> |
| <div class="line"><a name="l00692"></a><span class="lineno"> 692</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00695"></a><span class="lineno"> 695</span>  IsReshapeSupported,</div> |
| <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> |
| <div class="line"><a name="l00697"></a><span class="lineno"> 697</span>  isSupported,</div> |
| <div class="line"><a name="l00698"></a><span class="lineno"> 698</span>  setBackendReshape,</div> |
| <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> |
| <div class="line"><a name="l00700"></a><span class="lineno"> 700</span>  reshapeInfo,</div> |
| <div class="line"><a name="l00701"></a><span class="lineno"> 701</span>  reshapeDescriptor);</div> |
| <div class="line"><a name="l00702"></a><span class="lineno"> 702</span>  </div> |
| <div class="line"><a name="l00703"></a><span class="lineno"> 703</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00704"></a><span class="lineno"> 704</span>  {</div> |
| <div class="line"><a name="l00705"></a><span class="lineno"> 705</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00706"></a><span class="lineno"> 706</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00709"></a><span class="lineno"> 709</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00713"></a><span class="lineno"> 713</span>  }</div> |
| <div class="line"><a name="l00714"></a><span class="lineno"> 714</span>  </div> |
| <div class="line"><a name="l00715"></a><span class="lineno"> 715</span>  inputShapes.emplace_back(operandShape);</div> |
| <div class="line"><a name="l00716"></a><span class="lineno"> 716</span>  inputHandles.emplace_back(operandInputHandle);</div> |
| <div class="line"><a name="l00717"></a><span class="lineno"> 717</span>  </div> |
| <div class="line"><a name="l00718"></a><span class="lineno"> 718</span>  <span class="keywordflow">if</span> (!inputHandles.back().IsValid())</div> |
| <div class="line"><a name="l00719"></a><span class="lineno"> 719</span>  {</div> |
| <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> |
| <div class="line"><a name="l00721"></a><span class="lineno"> 721</span>  }</div> |
| <div class="line"><a name="l00722"></a><span class="lineno"> 722</span>  }</div> |
| <div class="line"><a name="l00723"></a><span class="lineno"> 723</span>  </div> |
| <div class="line"><a name="l00724"></a><span class="lineno"> 724</span>  <span class="keywordflow">if</span> (inputShapes.size() != inputHandles.size())</div> |
| <div class="line"><a name="l00725"></a><span class="lineno"> 725</span>  {</div> |
| <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> |
| <div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  inputShapes.size(), inputHandles.size());</div> |
| <div class="line"><a name="l00728"></a><span class="lineno"> 728</span>  }</div> |
| <div class="line"><a name="l00729"></a><span class="lineno"> 729</span>  </div> |
| <div class="line"><a name="l00730"></a><span class="lineno"> 730</span>  <span class="keywordflow">if</span> (inputsHaveBeenReshaped)</div> |
| <div class="line"><a name="l00731"></a><span class="lineno"> 731</span>  {</div> |
| <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> |
| <div class="line"><a name="l00733"></a><span class="lineno"> 733</span>  concatDim += tensorDimensionsAdded;</div> |
| <div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  </div> |
| <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> |
| <div class="line"><a name="l00736"></a><span class="lineno"> 736</span>  <span class="keywordflow">if</span> (tensorDimensionsAdded == 1)</div> |
| <div class="line"><a name="l00737"></a><span class="lineno"> 737</span>  {</div> |
| <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> |
| <div class="line"><a name="l00739"></a><span class="lineno"> 739</span>  {</div> |
| <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> |
| <div class="line"><a name="l00741"></a><span class="lineno"> 741</span>  }</div> |
| <div class="line"><a name="l00742"></a><span class="lineno"> 742</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00743"></a><span class="lineno"> 743</span>  {</div> |
| <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> |
| <div class="line"><a name="l00745"></a><span class="lineno"> 745</span>  }</div> |
| <div class="line"><a name="l00746"></a><span class="lineno"> 746</span>  }</div> |
| <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> |
| <div class="line"><a name="l00748"></a><span class="lineno"> 748</span>  {</div> |
| <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> |
| <div class="line"><a name="l00750"></a><span class="lineno"> 750</span>  {</div> |
| <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> |
| <div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  }</div> |
| <div class="line"><a name="l00753"></a><span class="lineno"> 753</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00754"></a><span class="lineno"> 754</span>  {</div> |
| <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> |
| <div class="line"><a name="l00756"></a><span class="lineno"> 756</span>  }</div> |
| <div class="line"><a name="l00757"></a><span class="lineno"> 757</span>  }</div> |
| <div class="line"><a name="l00758"></a><span class="lineno"> 758</span>  }</div> |
| <div class="line"><a name="l00759"></a><span class="lineno"> 759</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00762"></a><span class="lineno"> 762</span>  std::pair<armnn::PermutationVector, armnn::PermutationVector> permutationPair =</div> |
| <div class="line"><a name="l00763"></a><span class="lineno"> 763</span>  std::make_pair(IdentityPermutation4D, IdentityPermutation4D);</div> |
| <div class="line"><a name="l00764"></a><span class="lineno"> 764</span>  <span class="keywordtype">bool</span> needPermute = CreateConcatPermutationParameters(inputShapes[0].GetNumDimensions(),</div> |
| <div class="line"><a name="l00765"></a><span class="lineno"> 765</span>  concatDim,</div> |
| <div class="line"><a name="l00766"></a><span class="lineno"> 766</span>  permutationPair);</div> |
| <div class="line"><a name="l00767"></a><span class="lineno"> 767</span>  </div> |
| <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> |
| <div class="line"><a name="l00769"></a><span class="lineno"> 769</span>  <span class="keywordflow">if</span> (!isDynamicTensor)</div> |
| <div class="line"><a name="l00770"></a><span class="lineno"> 770</span>  {</div> |
| <div class="line"><a name="l00771"></a><span class="lineno"> 771</span>  <span class="keywordflow">if</span> (needPermute)</div> |
| <div class="line"><a name="l00772"></a><span class="lineno"> 772</span>  {</div> |
| <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> |
| <div class="line"><a name="l00774"></a><span class="lineno"> 774</span>  }</div> |
| <div class="line"><a name="l00775"></a><span class="lineno"> 775</span>  </div> |
| <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> |
| <div class="line"><a name="l00777"></a><span class="lineno"> 777</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00780"></a><span class="lineno"> 780</span>  <span class="keywordflow">if</span> (!TransposeInputTensors(data, inputHandles, inputShapes, permutationPair.first))</div> |
| <div class="line"><a name="l00781"></a><span class="lineno"> 781</span>  {</div> |
| <div class="line"><a name="l00782"></a><span class="lineno"> 782</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00783"></a><span class="lineno"> 783</span>  }</div> |
| <div class="line"><a name="l00784"></a><span class="lineno"> 784</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00787"></a><span class="lineno"> 787</span>  </div> |
| <div class="line"><a name="l00788"></a><span class="lineno"> 788</span>  <span class="keywordflow">try</span></div> |
| <div class="line"><a name="l00789"></a><span class="lineno"> 789</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00793"></a><span class="lineno"> 793</span>  inputShapes.end(),</div> |
| <div class="line"><a name="l00794"></a><span class="lineno"> 794</span>  concatDim);</div> |
| <div class="line"><a name="l00795"></a><span class="lineno"> 795</span>  } <span class="keywordflow">catch</span> (std::exception& error)</div> |
| <div class="line"><a name="l00796"></a><span class="lineno"> 796</span>  {</div> |
| <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> |
| <div class="line"><a name="l00798"></a><span class="lineno"> 798</span>  }</div> |
| <div class="line"><a name="l00799"></a><span class="lineno"> 799</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00802"></a><span class="lineno"> 802</span>  <span class="keywordflow">if</span> (!isDynamicTensor)</div> |
| <div class="line"><a name="l00803"></a><span class="lineno"> 803</span>  {</div> |
| <div class="line"><a name="l00804"></a><span class="lineno"> 804</span>  <span class="keywordflow">if</span> (!ValidateConcatOutputShape(inputShapes, outputShape, concatDim))</div> |
| <div class="line"><a name="l00805"></a><span class="lineno"> 805</span>  {</div> |
| <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> |
| <div class="line"><a name="l00807"></a><span class="lineno"> 807</span>  }</div> |
| <div class="line"><a name="l00808"></a><span class="lineno"> 808</span>  }</div> |
| <div class="line"><a name="l00809"></a><span class="lineno"> 809</span>  </div> |
| <div class="line"><a name="l00810"></a><span class="lineno"> 810</span>  std::vector<const armnn::TensorInfo*> inputTensorInfos;</div> |
| <div class="line"><a name="l00811"></a><span class="lineno"> 811</span>  std::transform(inputHandles.begin(), inputHandles.end(), std::back_inserter(inputTensorInfos),</div> |
| <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> |
| <div class="line"><a name="l00813"></a><span class="lineno"> 813</span>  </div> |
| <div class="line"><a name="l00814"></a><span class="lineno"> 814</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00818"></a><span class="lineno"> 818</span>  IsConcatSupported,</div> |
| <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> |
| <div class="line"><a name="l00820"></a><span class="lineno"> 820</span>  isSupported,</div> |
| <div class="line"><a name="l00821"></a><span class="lineno"> 821</span>  setBackendConcat,</div> |
| <div class="line"><a name="l00822"></a><span class="lineno"> 822</span>  inputTensorInfos,</div> |
| <div class="line"><a name="l00823"></a><span class="lineno"> 823</span>  outputInfo,</div> |
| <div class="line"><a name="l00824"></a><span class="lineno"> 824</span>  concatDescriptor);</div> |
| <div class="line"><a name="l00825"></a><span class="lineno"> 825</span>  };</div> |
| <div class="line"><a name="l00826"></a><span class="lineno"> 826</span>  </div> |
| <div class="line"><a name="l00827"></a><span class="lineno"> 827</span>  <span class="keywordflow">if</span> (!isDynamicTensor)</div> |
| <div class="line"><a name="l00828"></a><span class="lineno"> 828</span>  {</div> |
| <div class="line"><a name="l00829"></a><span class="lineno"> 829</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l00830"></a><span class="lineno"> 830</span>  }</div> |
| <div class="line"><a name="l00831"></a><span class="lineno"> 831</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00832"></a><span class="lineno"> 832</span>  {</div> |
| <div class="line"><a name="l00833"></a><span class="lineno"> 833</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00834"></a><span class="lineno"> 834</span>  }</div> |
| <div class="line"><a name="l00835"></a><span class="lineno"> 835</span>  </div> |
| <div class="line"><a name="l00836"></a><span class="lineno"> 836</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00837"></a><span class="lineno"> 837</span>  {</div> |
| <div class="line"><a name="l00838"></a><span class="lineno"> 838</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00839"></a><span class="lineno"> 839</span>  }</div> |
| <div class="line"><a name="l00840"></a><span class="lineno"> 840</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00843"></a><span class="lineno"> 843</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l00845"></a><span class="lineno"> 845</span>  <span class="comment">// Connect inputs to the layer</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00849"></a><span class="lineno"> 849</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00852"></a><span class="lineno"> 852</span>  }</div> |
| <div class="line"><a name="l00853"></a><span class="lineno"> 853</span>  </div> |
| <div class="line"><a name="l00854"></a><span class="lineno"> 854</span>  <span class="comment">// Transpose the output shape</span></div> |
| <div class="line"><a name="l00855"></a><span class="lineno"> 855</span>  <span class="keyword">auto</span> transposeOutputShape = [&](){</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00860"></a><span class="lineno"> 860</span>  permutationPair.second);</div> |
| <div class="line"><a name="l00861"></a><span class="lineno"> 861</span>  isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00864"></a><span class="lineno"> 864</span>  IsTransposeSupported,</div> |
| <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> |
| <div class="line"><a name="l00866"></a><span class="lineno"> 866</span>  isSupported,</div> |
| <div class="line"><a name="l00867"></a><span class="lineno"> 867</span>  setBackendTranspose,</div> |
| <div class="line"><a name="l00868"></a><span class="lineno"> 868</span>  inputTransposeInfo,</div> |
| <div class="line"><a name="l00869"></a><span class="lineno"> 869</span>  outputTransposeInfo,</div> |
| <div class="line"><a name="l00870"></a><span class="lineno"> 870</span>  transposeDesc);</div> |
| <div class="line"><a name="l00871"></a><span class="lineno"> 871</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00872"></a><span class="lineno"> 872</span>  {</div> |
| <div class="line"><a name="l00873"></a><span class="lineno"> 873</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00874"></a><span class="lineno"> 874</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00877"></a><span class="lineno"> 877</span>  permutationPair.second);</div> |
| <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> |
| <div class="line"><a name="l00879"></a><span class="lineno"> 879</span>  layer = &deswizzleLayer;</div> |
| <div class="line"><a name="l00880"></a><span class="lineno"> 880</span>  </div> |
| <div class="line"><a name="l00881"></a><span class="lineno"> 881</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00882"></a><span class="lineno"> 882</span>  };</div> |
| <div class="line"><a name="l00883"></a><span class="lineno"> 883</span>  </div> |
| <div class="line"><a name="l00884"></a><span class="lineno"> 884</span>  <span class="keywordflow">if</span> (needPermute && !isDynamicTensor)</div> |
| <div class="line"><a name="l00885"></a><span class="lineno"> 885</span>  {</div> |
| <div class="line"><a name="l00886"></a><span class="lineno"> 886</span>  transposeOutputShape();</div> |
| <div class="line"><a name="l00887"></a><span class="lineno"> 887</span>  }</div> |
| <div class="line"><a name="l00888"></a><span class="lineno"> 888</span>  </div> |
| <div class="line"><a name="l00889"></a><span class="lineno"> 889</span>  <span class="keywordflow">if</span> (inputsHaveBeenReshaped)</div> |
| <div class="line"><a name="l00890"></a><span class="lineno"> 890</span>  {</div> |
| <div class="line"><a name="l00891"></a><span class="lineno"> 891</span>  <span class="keywordflow">if</span> (isDynamicTensor)</div> |
| <div class="line"><a name="l00892"></a><span class="lineno"> 892</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00895"></a><span class="lineno"> 895</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00898"></a><span class="lineno"> 898</span>  }</div> |
| <div class="line"><a name="l00899"></a><span class="lineno"> 899</span>  <span class="keywordflow">if</span> (!ValidateConcatOutputShape(inputShapes,</div> |
| <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> |
| <div class="line"><a name="l00901"></a><span class="lineno"> 901</span>  concatDim))</div> |
| <div class="line"><a name="l00902"></a><span class="lineno"> 902</span>  {</div> |
| <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> |
| <div class="line"><a name="l00904"></a><span class="lineno"> 904</span>  }</div> |
| <div class="line"><a name="l00905"></a><span class="lineno"> 905</span>  transposeOutputShape();</div> |
| <div class="line"><a name="l00906"></a><span class="lineno"> 906</span>  }</div> |
| <div class="line"><a name="l00907"></a><span class="lineno"> 907</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00910"></a><span class="lineno"> 910</span>  <span class="keywordflow">if</span> (tensorDimensionsAdded == 1)</div> |
| <div class="line"><a name="l00911"></a><span class="lineno"> 911</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l00914"></a><span class="lineno"> 914</span>  }</div> |
| <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> |
| <div class="line"><a name="l00916"></a><span class="lineno"> 916</span>  {</div> |
| <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> |
| <div class="line"><a name="l00918"></a><span class="lineno"> 918</span>  }</div> |
| <div class="line"><a name="l00919"></a><span class="lineno"> 919</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00923"></a><span class="lineno"> 923</span>  </div> |
| <div class="line"><a name="l00924"></a><span class="lineno"> 924</span>  isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00928"></a><span class="lineno"> 928</span>  IsReshapeSupported,</div> |
| <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> |
| <div class="line"><a name="l00930"></a><span class="lineno"> 930</span>  isSupported,</div> |
| <div class="line"><a name="l00931"></a><span class="lineno"> 931</span>  setBackendReshape2,</div> |
| <div class="line"><a name="l00932"></a><span class="lineno"> 932</span>  concatInfo,</div> |
| <div class="line"><a name="l00933"></a><span class="lineno"> 933</span>  afterConcatInfo,</div> |
| <div class="line"><a name="l00934"></a><span class="lineno"> 934</span>  reshapeDescriptor);</div> |
| <div class="line"><a name="l00935"></a><span class="lineno"> 935</span>  };</div> |
| <div class="line"><a name="l00936"></a><span class="lineno"> 936</span>  </div> |
| <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> |
| <div class="line"><a name="l00938"></a><span class="lineno"> 938</span>  {</div> |
| <div class="line"><a name="l00939"></a><span class="lineno"> 939</span>  validateReshapeFunc(afterConcatInfo, isSupported);</div> |
| <div class="line"><a name="l00940"></a><span class="lineno"> 940</span>  }</div> |
| <div class="line"><a name="l00941"></a><span class="lineno"> 941</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l00942"></a><span class="lineno"> 942</span>  {</div> |
| <div class="line"><a name="l00943"></a><span class="lineno"> 943</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l00944"></a><span class="lineno"> 944</span>  }</div> |
| <div class="line"><a name="l00945"></a><span class="lineno"> 945</span>  </div> |
| <div class="line"><a name="l00946"></a><span class="lineno"> 946</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l00947"></a><span class="lineno"> 947</span>  {</div> |
| <div class="line"><a name="l00948"></a><span class="lineno"> 948</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00949"></a><span class="lineno"> 949</span>  }</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l00953"></a><span class="lineno"> 953</span>  0,</div> |
| <div class="line"><a name="l00954"></a><span class="lineno"> 954</span>  *layer,</div> |
| <div class="line"><a name="l00955"></a><span class="lineno"> 955</span>  model,</div> |
| <div class="line"><a name="l00956"></a><span class="lineno"> 956</span>  data,</div> |
| <div class="line"><a name="l00957"></a><span class="lineno"> 957</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l00958"></a><span class="lineno"> 958</span>  validateReshapeFunc);</div> |
| <div class="line"><a name="l00959"></a><span class="lineno"> 959</span>  }</div> |
| <div class="line"><a name="l00960"></a><span class="lineno"> 960</span>  </div> |
| <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> |
| <div class="line"><a name="l00962"></a><span class="lineno"> 962</span> }</div> |
| <div class="line"><a name="l00963"></a><span class="lineno"> 963</span>  </div> |
| <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> |
| <div class="line"><a name="l00965"></a><span class="lineno"> 965</span> {</div> |
| <div class="line"><a name="l00966"></a><span class="lineno"> 966</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConv2d()"</span>;</div> |
| <div class="line"><a name="l00967"></a><span class="lineno"> 967</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00970"></a><span class="lineno"> 970</span>  {</div> |
| <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> |
| <div class="line"><a name="l00972"></a><span class="lineno"> 972</span>  }</div> |
| <div class="line"><a name="l00973"></a><span class="lineno"> 973</span>  </div> |
| <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> |
| <div class="line"><a name="l00975"></a><span class="lineno"> 975</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l00976"></a><span class="lineno"> 976</span>  {</div> |
| <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> |
| <div class="line"><a name="l00978"></a><span class="lineno"> 978</span>  }</div> |
| <div class="line"><a name="l00979"></a><span class="lineno"> 979</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00982"></a><span class="lineno"> 982</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l00985"></a><span class="lineno"> 985</span>  </div> |
| <div class="line"><a name="l00986"></a><span class="lineno"> 986</span>  <span class="comment">// Determine whether padding is implicit or explicit</span></div> |
| <div class="line"><a name="l00987"></a><span class="lineno"> 987</span>  <span class="keywordtype">bool</span> implicitPadding = operation.inputs.size() == 7</div> |
| <div class="line"><a name="l00988"></a><span class="lineno"> 988</span>  || (operation.inputs.size() >= 8</div> |
| <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> |
| <div class="line"><a name="l00990"></a><span class="lineno"> 990</span>  </div> |
| <div class="line"><a name="l00991"></a><span class="lineno"> 991</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| <div class="line"><a name="l00992"></a><span class="lineno"> 992</span>  {</div> |
| <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> |
| <div class="line"><a name="l00994"></a><span class="lineno"> 994</span>  }</div> |
| <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> |
| <div class="line"><a name="l00996"></a><span class="lineno"> 996</span>  {</div> |
| <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> |
| <div class="line"><a name="l00998"></a><span class="lineno"> 998</span>  }</div> |
| <div class="line"><a name="l00999"></a><span class="lineno"> 999</span>  </div> |
| <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> |
| <div class="line"><a name="l01001"></a><span class="lineno"> 1001</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01005"></a><span class="lineno"> 1005</span>  <span class="comment">// the DataLayout is NCHW</span></div> |
| <div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>  </div> |
| <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> |
| <div class="line"><a name="l01008"></a><span class="lineno"> 1008</span>  {</div> |
| <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> |
| <div class="line"><a name="l01010"></a><span class="lineno"> 1010</span>  }</div> |
| <div class="line"><a name="l01011"></a><span class="lineno"> 1011</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01015"></a><span class="lineno"> 1015</span>  </div> |
| <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> |
| <div class="line"><a name="l01017"></a><span class="lineno"> 1017</span>  {</div> |
| <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> |
| <div class="line"><a name="l01019"></a><span class="lineno"> 1019</span>  }</div> |
| <div class="line"><a name="l01020"></a><span class="lineno"> 1020</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01023"></a><span class="lineno"> 1023</span>  {</div> |
| <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> |
| <div class="line"><a name="l01025"></a><span class="lineno"> 1025</span>  }</div> |
| <div class="line"><a name="l01026"></a><span class="lineno"> 1026</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01030"></a><span class="lineno"> 1030</span>  </div> |
| <div class="line"><a name="l01031"></a><span class="lineno"> 1031</span>  ActivationFn activation;</div> |
| <div class="line"><a name="l01032"></a><span class="lineno"> 1032</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| <div class="line"><a name="l01033"></a><span class="lineno"> 1033</span>  {</div> |
| <div class="line"><a name="l01034"></a><span class="lineno"> 1034</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01040"></a><span class="lineno"> 1040</span>  {</div> |
| <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> |
| <div class="line"><a name="l01042"></a><span class="lineno"> 1042</span>  }</div> |
| <div class="line"><a name="l01043"></a><span class="lineno"> 1043</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01051"></a><span class="lineno"> 1051</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01054"></a><span class="lineno"> 1054</span>  </div> |
| <div class="line"><a name="l01055"></a><span class="lineno"> 1055</span>  }</div> |
| <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> |
| <div class="line"><a name="l01057"></a><span class="lineno"> 1057</span>  {</div> |
| <div class="line"><a name="l01058"></a><span class="lineno"> 1058</span>  <span class="comment">// explicit padding</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01067"></a><span class="lineno"> 1067</span>  {</div> |
| <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> |
| <div class="line"><a name="l01069"></a><span class="lineno"> 1069</span>  }</div> |
| <div class="line"><a name="l01070"></a><span class="lineno"> 1070</span>  }</div> |
| <div class="line"><a name="l01071"></a><span class="lineno"> 1071</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>  {</div> |
| <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> |
| <div class="line"><a name="l01074"></a><span class="lineno"> 1074</span>  }</div> |
| <div class="line"><a name="l01075"></a><span class="lineno"> 1075</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01078"></a><span class="lineno"> 1078</span>  </div> |
| <div class="line"><a name="l01079"></a><span class="lineno"> 1079</span>  <span class="keywordtype">bool</span> requiresValidation = <span class="keyword">true</span>;</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01084"></a><span class="lineno"> 1084</span>  {</div> |
| <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> |
| <div class="line"><a name="l01086"></a><span class="lineno"> 1086</span>  <span class="comment">// [ConvertConstDequantisationLayersToConstLayers] will convert layers to Constant layers</span></div> |
| <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> |
| <div class="line"><a name="l01088"></a><span class="lineno"> 1088</span>  requiresValidation = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>  }</div> |
| <div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01095"></a><span class="lineno"> 1095</span>  IsConvolution2dSupported,</div> |
| <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> |
| <div class="line"><a name="l01097"></a><span class="lineno"> 1097</span>  isSupported,</div> |
| <div class="line"><a name="l01098"></a><span class="lineno"> 1098</span>  setBackend,</div> |
| <div class="line"><a name="l01099"></a><span class="lineno"> 1099</span>  inputInfo,</div> |
| <div class="line"><a name="l01100"></a><span class="lineno"> 1100</span>  outputInfo,</div> |
| <div class="line"><a name="l01101"></a><span class="lineno"> 1101</span>  desc,</div> |
| <div class="line"><a name="l01102"></a><span class="lineno"> 1102</span>  weightsInfo,</div> |
| <div class="line"><a name="l01103"></a><span class="lineno"> 1103</span>  biases);</div> |
| <div class="line"><a name="l01104"></a><span class="lineno"> 1104</span>  };</div> |
| <div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>  </div> |
| <div class="line"><a name="l01106"></a><span class="lineno"> 1106</span>  <span class="keywordflow">if</span> (requiresValidation)</div> |
| <div class="line"><a name="l01107"></a><span class="lineno"> 1107</span>  {</div> |
| <div class="line"><a name="l01108"></a><span class="lineno"> 1108</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertConv2d(): Requires Validation!"</span>;</div> |
| <div class="line"><a name="l01109"></a><span class="lineno"> 1109</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l01111"></a><span class="lineno"> 1111</span>  {</div> |
| <div class="line"><a name="l01112"></a><span class="lineno"> 1112</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>  }</div> |
| <div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>  {</div> |
| <div class="line"><a name="l01116"></a><span class="lineno"> 1116</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>  }</div> |
| <div class="line"><a name="l01118"></a><span class="lineno"> 1118</span>  </div> |
| <div class="line"><a name="l01119"></a><span class="lineno"> 1119</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01120"></a><span class="lineno"> 1120</span>  {</div> |
| <div class="line"><a name="l01121"></a><span class="lineno"> 1121</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01122"></a><span class="lineno"> 1122</span>  }</div> |
| <div class="line"><a name="l01123"></a><span class="lineno"> 1123</span>  }</div> |
| <div class="line"><a name="l01124"></a><span class="lineno"> 1124</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01127"></a><span class="lineno"> 1127</span>  </div> |
| <div class="line"><a name="l01128"></a><span class="lineno"> 1128</span>  <span class="keywordflow">if</span> (!startLayer)</div> |
| <div class="line"><a name="l01129"></a><span class="lineno"> 1129</span>  {</div> |
| <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> |
| <div class="line"><a name="l01131"></a><span class="lineno"> 1131</span>  }</div> |
| <div class="line"><a name="l01132"></a><span class="lineno"> 1132</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01136"></a><span class="lineno"> 1136</span>  </div> |
| <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> |
| <div class="line"><a name="l01138"></a><span class="lineno"> 1138</span> }</div> |
| <div class="line"><a name="l01139"></a><span class="lineno"> 1139</span>  </div> |
| <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> |
| <div class="line"><a name="l01141"></a><span class="lineno"> 1141</span> {</div> |
| <div class="line"><a name="l01142"></a><span class="lineno"> 1142</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthToSpace()"</span>;</div> |
| <div class="line"><a name="l01143"></a><span class="lineno"> 1143</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01146"></a><span class="lineno"> 1146</span>  {</div> |
| <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> |
| <div class="line"><a name="l01148"></a><span class="lineno"> 1148</span>  }</div> |
| <div class="line"><a name="l01149"></a><span class="lineno"> 1149</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01152"></a><span class="lineno"> 1152</span>  <span class="keywordflow">if</span> (rank != 4)</div> |
| <div class="line"><a name="l01153"></a><span class="lineno"> 1153</span>  {</div> |
| <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> |
| <div class="line"><a name="l01155"></a><span class="lineno"> 1155</span>  }</div> |
| <div class="line"><a name="l01156"></a><span class="lineno"> 1156</span>  </div> |
| <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> |
| <div class="line"><a name="l01158"></a><span class="lineno"> 1158</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01159"></a><span class="lineno"> 1159</span>  {</div> |
| <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> |
| <div class="line"><a name="l01161"></a><span class="lineno"> 1161</span>  }</div> |
| <div class="line"><a name="l01162"></a><span class="lineno"> 1162</span>  </div> |
| <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> |
| <div class="line"><a name="l01164"></a><span class="lineno"> 1164</span>  </div> |
| <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> |
| <div class="line"><a name="l01166"></a><span class="lineno"> 1166</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01169"></a><span class="lineno"> 1169</span>  {</div> |
| <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> |
| <div class="line"><a name="l01171"></a><span class="lineno"> 1171</span>  }</div> |
| <div class="line"><a name="l01172"></a><span class="lineno"> 1172</span>  </div> |
| <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> |
| <div class="line"><a name="l01174"></a><span class="lineno"> 1174</span>  <span class="keywordflow">if</span> (Is12OrLaterOperand(*output))</div> |
| <div class="line"><a name="l01175"></a><span class="lineno"> 1175</span>  {</div> |
| <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> |
| <div class="line"><a name="l01177"></a><span class="lineno"> 1177</span>  }</div> |
| <div class="line"><a name="l01178"></a><span class="lineno"> 1178</span>  </div> |
| <div class="line"><a name="l01179"></a><span class="lineno"> 1179</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01182"></a><span class="lineno"> 1182</span>  {</div> |
| <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> |
| <div class="line"><a name="l01184"></a><span class="lineno"> 1184</span>  IsDepthToSpaceSupported,</div> |
| <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> |
| <div class="line"><a name="l01186"></a><span class="lineno"> 1186</span>  isSupported,</div> |
| <div class="line"><a name="l01187"></a><span class="lineno"> 1187</span>  setBackend,</div> |
| <div class="line"><a name="l01188"></a><span class="lineno"> 1188</span>  inputInfo,</div> |
| <div class="line"><a name="l01189"></a><span class="lineno"> 1189</span>  outputInfo,</div> |
| <div class="line"><a name="l01190"></a><span class="lineno"> 1190</span>  descriptor);</div> |
| <div class="line"><a name="l01191"></a><span class="lineno"> 1191</span>  };</div> |
| <div class="line"><a name="l01192"></a><span class="lineno"> 1192</span>  </div> |
| <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> |
| <div class="line"><a name="l01194"></a><span class="lineno"> 1194</span>  {</div> |
| <div class="line"><a name="l01195"></a><span class="lineno"> 1195</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01196"></a><span class="lineno"> 1196</span>  }</div> |
| <div class="line"><a name="l01197"></a><span class="lineno"> 1197</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01198"></a><span class="lineno"> 1198</span>  {</div> |
| <div class="line"><a name="l01199"></a><span class="lineno"> 1199</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01200"></a><span class="lineno"> 1200</span>  }</div> |
| <div class="line"><a name="l01201"></a><span class="lineno"> 1201</span>  </div> |
| <div class="line"><a name="l01202"></a><span class="lineno"> 1202</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01203"></a><span class="lineno"> 1203</span>  {</div> |
| <div class="line"><a name="l01204"></a><span class="lineno"> 1204</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01205"></a><span class="lineno"> 1205</span>  }</div> |
| <div class="line"><a name="l01206"></a><span class="lineno"> 1206</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01209"></a><span class="lineno"> 1209</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l01211"></a><span class="lineno"> 1211</span>  </div> |
| <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> |
| <div class="line"><a name="l01213"></a><span class="lineno"> 1213</span> }</div> |
| <div class="line"><a name="l01214"></a><span class="lineno"> 1214</span>  </div> |
| <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> |
| <div class="line"><a name="l01216"></a><span class="lineno"> 1216</span> {</div> |
| <div class="line"><a name="l01217"></a><span class="lineno"> 1217</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthwiseConv2d()"</span>;</div> |
| <div class="line"><a name="l01218"></a><span class="lineno"> 1218</span>  </div> |
| <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> |
| <div class="line"><a name="l01220"></a><span class="lineno"> 1220</span>  </div> |
| <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> |
| <div class="line"><a name="l01222"></a><span class="lineno"> 1222</span>  {</div> |
| <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> |
| <div class="line"><a name="l01224"></a><span class="lineno"> 1224</span>  }</div> |
| <div class="line"><a name="l01225"></a><span class="lineno"> 1225</span>  </div> |
| <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> |
| <div class="line"><a name="l01227"></a><span class="lineno"> 1227</span>  </div> |
| <div class="line"><a name="l01228"></a><span class="lineno"> 1228</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01229"></a><span class="lineno"> 1229</span>  {</div> |
| <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> |
| <div class="line"><a name="l01231"></a><span class="lineno"> 1231</span>  }</div> |
| <div class="line"><a name="l01232"></a><span class="lineno"> 1232</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01235"></a><span class="lineno"> 1235</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01238"></a><span class="lineno"> 1238</span>  {</div> |
| <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> |
| <div class="line"><a name="l01240"></a><span class="lineno"> 1240</span>  }</div> |
| <div class="line"><a name="l01241"></a><span class="lineno"> 1241</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01244"></a><span class="lineno"> 1244</span>  </div> |
| <div class="line"><a name="l01245"></a><span class="lineno"> 1245</span>  <span class="comment">// Basic sanity check on the weights shape.</span></div> |
| <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> |
| <div class="line"><a name="l01247"></a><span class="lineno"> 1247</span>  <span class="comment">// [1, filter_height, filter_width, depth_out]</span></div> |
| <div class="line"><a name="l01248"></a><span class="lineno"> 1248</span>  <span class="keywordflow">if</span> (weightsOperand->dimensions[0] != 1)</div> |
| <div class="line"><a name="l01249"></a><span class="lineno"> 1249</span>  {</div> |
| <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> |
| <div class="line"><a name="l01251"></a><span class="lineno"> 1251</span>  }</div> |
| <div class="line"><a name="l01252"></a><span class="lineno"> 1252</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01255"></a><span class="lineno"> 1255</span>  </div> |
| <div class="line"><a name="l01256"></a><span class="lineno"> 1256</span>  <span class="comment">// Determine whether padding is implicit or explicit</span></div> |
| <div class="line"><a name="l01257"></a><span class="lineno"> 1257</span>  <span class="keywordtype">bool</span> implicitPadding = operation.inputs.size() == 8</div> |
| <div class="line"><a name="l01258"></a><span class="lineno"> 1258</span>  || (operation.inputs.size() >= 9</div> |
| <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> |
| <div class="line"><a name="l01260"></a><span class="lineno"> 1260</span>  </div> |
| <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> |
| <div class="line"><a name="l01262"></a><span class="lineno"> 1262</span>  <span class="keyword">const</span> uint32_t dataLayoutFlagIndex = implicitPadding ? 8 : 11;</div> |
| <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> |
| <div class="line"><a name="l01264"></a><span class="lineno"> 1264</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01268"></a><span class="lineno"> 1268</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01271"></a><span class="lineno"> 1271</span>  {</div> |
| <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> |
| <div class="line"><a name="l01273"></a><span class="lineno"> 1273</span>  }</div> |
| <div class="line"><a name="l01274"></a><span class="lineno"> 1274</span>  </div> |
| <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> |
| <div class="line"><a name="l01276"></a><span class="lineno"> 1276</span>  <span class="keywordflow">if</span> (!biasOperand)</div> |
| <div class="line"><a name="l01277"></a><span class="lineno"> 1277</span>  {</div> |
| <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> |
| <div class="line"><a name="l01279"></a><span class="lineno"> 1279</span>  }</div> |
| <div class="line"><a name="l01280"></a><span class="lineno"> 1280</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01283"></a><span class="lineno"> 1283</span>  {</div> |
| <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> |
| <div class="line"><a name="l01285"></a><span class="lineno"> 1285</span>  }</div> |
| <div class="line"><a name="l01286"></a><span class="lineno"> 1286</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01290"></a><span class="lineno"> 1290</span>  </div> |
| <div class="line"><a name="l01291"></a><span class="lineno"> 1291</span>  ActivationFn activation;</div> |
| <div class="line"><a name="l01292"></a><span class="lineno"> 1292</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| <div class="line"><a name="l01293"></a><span class="lineno"> 1293</span>  {</div> |
| <div class="line"><a name="l01294"></a><span class="lineno"> 1294</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01300"></a><span class="lineno"> 1300</span>  {</div> |
| <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> |
| <div class="line"><a name="l01302"></a><span class="lineno"> 1302</span>  }</div> |
| <div class="line"><a name="l01303"></a><span class="lineno"> 1303</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01308"></a><span class="lineno"> 1308</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01311"></a><span class="lineno"> 1311</span>  }</div> |
| <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> |
| <div class="line"><a name="l01313"></a><span class="lineno"> 1313</span>  {</div> |
| <div class="line"><a name="l01314"></a><span class="lineno"> 1314</span>  <span class="comment">// explicit padding</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01323"></a><span class="lineno"> 1323</span>  {</div> |
| <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> |
| <div class="line"><a name="l01325"></a><span class="lineno"> 1325</span>  }</div> |
| <div class="line"><a name="l01326"></a><span class="lineno"> 1326</span>  }</div> |
| <div class="line"><a name="l01327"></a><span class="lineno"> 1327</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01328"></a><span class="lineno"> 1328</span>  {</div> |
| <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> |
| <div class="line"><a name="l01330"></a><span class="lineno"> 1330</span>  }</div> |
| <div class="line"><a name="l01331"></a><span class="lineno"> 1331</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01334"></a><span class="lineno"> 1334</span>  </div> |
| <div class="line"><a name="l01335"></a><span class="lineno"> 1335</span>  <span class="keywordtype">bool</span> requiresValidation = <span class="keyword">true</span>;</div> |
| <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> |
| <div class="line"><a name="l01337"></a><span class="lineno"> 1337</span>  {</div> |
| <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> |
| <div class="line"><a name="l01339"></a><span class="lineno"> 1339</span>  <span class="comment">// [ConvertConstDequantisationLayersToConstLayers] will convert layers to Constant layers</span></div> |
| <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> |
| <div class="line"><a name="l01341"></a><span class="lineno"> 1341</span>  requiresValidation = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l01343"></a><span class="lineno"> 1343</span>  }</div> |
| <div class="line"><a name="l01344"></a><span class="lineno"> 1344</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01348"></a><span class="lineno"> 1348</span>  IsDepthwiseConvolutionSupported,</div> |
| <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> |
| <div class="line"><a name="l01350"></a><span class="lineno"> 1350</span>  isSupported,</div> |
| <div class="line"><a name="l01351"></a><span class="lineno"> 1351</span>  setBackend,</div> |
| <div class="line"><a name="l01352"></a><span class="lineno"> 1352</span>  inputInfo,</div> |
| <div class="line"><a name="l01353"></a><span class="lineno"> 1353</span>  outputInfo,</div> |
| <div class="line"><a name="l01354"></a><span class="lineno"> 1354</span>  desc,</div> |
| <div class="line"><a name="l01355"></a><span class="lineno"> 1355</span>  weightsInfo,</div> |
| <div class="line"><a name="l01356"></a><span class="lineno"> 1356</span>  biases);</div> |
| <div class="line"><a name="l01357"></a><span class="lineno"> 1357</span>  };</div> |
| <div class="line"><a name="l01358"></a><span class="lineno"> 1358</span>  </div> |
| <div class="line"><a name="l01359"></a><span class="lineno"> 1359</span>  <span class="keywordflow">if</span> (requiresValidation)</div> |
| <div class="line"><a name="l01360"></a><span class="lineno"> 1360</span>  {</div> |
| <div class="line"><a name="l01361"></a><span class="lineno"> 1361</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDepthwiseConv2d(): Requires Validation!"</span>;</div> |
| <div class="line"><a name="l01362"></a><span class="lineno"> 1362</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l01364"></a><span class="lineno"> 1364</span>  {</div> |
| <div class="line"><a name="l01365"></a><span class="lineno"> 1365</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01366"></a><span class="lineno"> 1366</span>  }</div> |
| <div class="line"><a name="l01367"></a><span class="lineno"> 1367</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01368"></a><span class="lineno"> 1368</span>  {</div> |
| <div class="line"><a name="l01369"></a><span class="lineno"> 1369</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01370"></a><span class="lineno"> 1370</span>  }</div> |
| <div class="line"><a name="l01371"></a><span class="lineno"> 1371</span>  </div> |
| <div class="line"><a name="l01372"></a><span class="lineno"> 1372</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01373"></a><span class="lineno"> 1373</span>  {</div> |
| <div class="line"><a name="l01374"></a><span class="lineno"> 1374</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01375"></a><span class="lineno"> 1375</span>  }</div> |
| <div class="line"><a name="l01376"></a><span class="lineno"> 1376</span>  }</div> |
| <div class="line"><a name="l01377"></a><span class="lineno"> 1377</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01380"></a><span class="lineno"> 1380</span>  </div> |
| <div class="line"><a name="l01381"></a><span class="lineno"> 1381</span>  <span class="keywordflow">if</span> (!startLayer)</div> |
| <div class="line"><a name="l01382"></a><span class="lineno"> 1382</span>  {</div> |
| <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> |
| <div class="line"><a name="l01384"></a><span class="lineno"> 1384</span>  }</div> |
| <div class="line"><a name="l01385"></a><span class="lineno"> 1385</span>  </div> |
| <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> |
| <div class="line"><a name="l01387"></a><span class="lineno"> 1387</span>  </div> |
| <div class="line"><a name="l01388"></a><span class="lineno"> 1388</span>  <span class="comment">// Connect weights and bias inputs</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l01391"></a><span class="lineno"> 1391</span>  </div> |
| <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> |
| <div class="line"><a name="l01393"></a><span class="lineno"> 1393</span> }</div> |
| <div class="line"><a name="l01394"></a><span class="lineno"> 1394</span>  </div> |
| <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> |
| <div class="line"><a name="l01396"></a><span class="lineno"> 1396</span> {</div> |
| <div class="line"><a name="l01397"></a><span class="lineno"> 1397</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertDequantize()"</span>;</div> |
| <div class="line"><a name="l01398"></a><span class="lineno"> 1398</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01401"></a><span class="lineno"> 1401</span>  {</div> |
| <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> |
| <div class="line"><a name="l01403"></a><span class="lineno"> 1403</span>  }</div> |
| <div class="line"><a name="l01404"></a><span class="lineno"> 1404</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01408"></a><span class="lineno"> 1408</span>  {</div> |
| <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> |
| <div class="line"><a name="l01410"></a><span class="lineno"> 1410</span>  }</div> |
| <div class="line"><a name="l01411"></a><span class="lineno"> 1411</span>  </div> |
| <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> |
| <div class="line"><a name="l01413"></a><span class="lineno"> 1413</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l01414"></a><span class="lineno"> 1414</span>  {</div> |
| <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> |
| <div class="line"><a name="l01416"></a><span class="lineno"> 1416</span>  }</div> |
| <div class="line"><a name="l01417"></a><span class="lineno"> 1417</span>  </div> |
| <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> |
| <div class="line"><a name="l01419"></a><span class="lineno"> 1419</span>  </div> |
| <div class="line"><a name="l01420"></a><span class="lineno"> 1420</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01423"></a><span class="lineno"> 1423</span>  {</div> |
| <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> |
| <div class="line"><a name="l01425"></a><span class="lineno"> 1425</span>  IsDequantizeSupported,</div> |
| <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> |
| <div class="line"><a name="l01427"></a><span class="lineno"> 1427</span>  isSupported,</div> |
| <div class="line"><a name="l01428"></a><span class="lineno"> 1428</span>  setBackend,</div> |
| <div class="line"><a name="l01429"></a><span class="lineno"> 1429</span>  inputInfo,</div> |
| <div class="line"><a name="l01430"></a><span class="lineno"> 1430</span>  outputInfo);</div> |
| <div class="line"><a name="l01431"></a><span class="lineno"> 1431</span>  };</div> |
| <div class="line"><a name="l01432"></a><span class="lineno"> 1432</span>  </div> |
| <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> |
| <div class="line"><a name="l01434"></a><span class="lineno"> 1434</span>  {</div> |
| <div class="line"><a name="l01435"></a><span class="lineno"> 1435</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01436"></a><span class="lineno"> 1436</span>  }</div> |
| <div class="line"><a name="l01437"></a><span class="lineno"> 1437</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01438"></a><span class="lineno"> 1438</span>  {</div> |
| <div class="line"><a name="l01439"></a><span class="lineno"> 1439</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01440"></a><span class="lineno"> 1440</span>  }</div> |
| <div class="line"><a name="l01441"></a><span class="lineno"> 1441</span>  </div> |
| <div class="line"><a name="l01442"></a><span class="lineno"> 1442</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01443"></a><span class="lineno"> 1443</span>  {</div> |
| <div class="line"><a name="l01444"></a><span class="lineno"> 1444</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01445"></a><span class="lineno"> 1445</span>  }</div> |
| <div class="line"><a name="l01446"></a><span class="lineno"> 1446</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01449"></a><span class="lineno"> 1449</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l01451"></a><span class="lineno"> 1451</span>  </div> |
| <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> |
| <div class="line"><a name="l01453"></a><span class="lineno"> 1453</span> }</div> |
| <div class="line"><a name="l01454"></a><span class="lineno"> 1454</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01458"></a><span class="lineno"> 1458</span>  <a class="code" href="namespacearmnn.html#ac4f8557279754ed7b3f749d55b0e3047">armnn::BinaryOperation</a> binaryOperation)</div> |
| <div class="line"><a name="l01459"></a><span class="lineno"> 1459</span> {</div> |
| <div class="line"><a name="l01460"></a><span class="lineno"> 1460</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertElementwiseBinary()"</span>;</div> |
| <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> |
| <div class="line"><a name="l01462"></a><span class="lineno"> 1462</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01465"></a><span class="lineno"> 1465</span>  </div> |
| <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> |
| <div class="line"><a name="l01467"></a><span class="lineno"> 1467</span>  {</div> |
| <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> |
| <div class="line"><a name="l01469"></a><span class="lineno"> 1469</span>  }</div> |
| <div class="line"><a name="l01470"></a><span class="lineno"> 1470</span>  </div> |
| <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> |
| <div class="line"><a name="l01472"></a><span class="lineno"> 1472</span>  ActivationFn activationFunction;</div> |
| <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> |
| <div class="line"><a name="l01474"></a><span class="lineno"> 1474</span>  {</div> |
| <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> |
| <div class="line"><a name="l01476"></a><span class="lineno"> 1476</span>  }</div> |
| <div class="line"><a name="l01477"></a><span class="lineno"> 1477</span>  </div> |
| <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> |
| <div class="line"><a name="l01479"></a><span class="lineno"> 1479</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01480"></a><span class="lineno"> 1480</span>  {</div> |
| <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> |
| <div class="line"><a name="l01482"></a><span class="lineno"> 1482</span>  }</div> |
| <div class="line"><a name="l01483"></a><span class="lineno"> 1483</span>  </div> |
| <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> |
| <div class="line"><a name="l01485"></a><span class="lineno"> 1485</span>  </div> |
| <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> |
| <div class="line"><a name="l01487"></a><span class="lineno"> 1487</span>  </div> |
| <div class="line"><a name="l01488"></a><span class="lineno"> 1488</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l01490"></a><span class="lineno"> 1490</span>  {</div> |
| <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> |
| <div class="line"><a name="l01492"></a><span class="lineno"> 1492</span>  IsElementwiseBinarySupported,</div> |
| <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> |
| <div class="line"><a name="l01494"></a><span class="lineno"> 1494</span>  isSupported,</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01498"></a><span class="lineno"> 1498</span>  outputInfo,</div> |
| <div class="line"><a name="l01499"></a><span class="lineno"> 1499</span>  binaryOperation);</div> |
| <div class="line"><a name="l01500"></a><span class="lineno"> 1500</span>  };</div> |
| <div class="line"><a name="l01501"></a><span class="lineno"> 1501</span>  </div> |
| <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> |
| <div class="line"><a name="l01503"></a><span class="lineno"> 1503</span>  {</div> |
| <div class="line"><a name="l01504"></a><span class="lineno"> 1504</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01505"></a><span class="lineno"> 1505</span>  }</div> |
| <div class="line"><a name="l01506"></a><span class="lineno"> 1506</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01507"></a><span class="lineno"> 1507</span>  {</div> |
| <div class="line"><a name="l01508"></a><span class="lineno"> 1508</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01509"></a><span class="lineno"> 1509</span>  }</div> |
| <div class="line"><a name="l01510"></a><span class="lineno"> 1510</span>  </div> |
| <div class="line"><a name="l01511"></a><span class="lineno"> 1511</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01512"></a><span class="lineno"> 1512</span>  {</div> |
| <div class="line"><a name="l01513"></a><span class="lineno"> 1513</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01514"></a><span class="lineno"> 1514</span>  }</div> |
| <div class="line"><a name="l01515"></a><span class="lineno"> 1515</span>  </div> |
| <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> |
| <div class="line"><a name="l01517"></a><span class="lineno"> 1517</span>  <span class="keywordflow">if</span> (!layer)</div> |
| <div class="line"><a name="l01518"></a><span class="lineno"> 1518</span>  {</div> |
| <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> |
| <div class="line"><a name="l01520"></a><span class="lineno"> 1520</span>  }</div> |
| <div class="line"><a name="l01521"></a><span class="lineno"> 1521</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input0, input1, layer, data);</div> |
| <div class="line"><a name="l01522"></a><span class="lineno"> 1522</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| <div class="line"><a name="l01523"></a><span class="lineno"> 1523</span>  {</div> |
| <div class="line"><a name="l01524"></a><span class="lineno"> 1524</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01525"></a><span class="lineno"> 1525</span>  }</div> |
| <div class="line"><a name="l01526"></a><span class="lineno"> 1526</span>  </div> |
| <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> |
| <div class="line"><a name="l01528"></a><span class="lineno"> 1528</span>  data, <span class="keyword">nullptr</span>, validateFunc, activationFunction);</div> |
| <div class="line"><a name="l01529"></a><span class="lineno"> 1529</span> }</div> |
| <div class="line"><a name="l01530"></a><span class="lineno"> 1530</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01534"></a><span class="lineno"> 1534</span>  <a class="code" href="namespacearmnn.html#a1cfaa710db2a54673b21d2ea2da757c8">UnaryOperation</a> unaryOperation)</div> |
| <div class="line"><a name="l01535"></a><span class="lineno"> 1535</span> {</div> |
| <div class="line"><a name="l01536"></a><span class="lineno"> 1536</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertElementwiseUnary()"</span>;</div> |
| <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> |
| <div class="line"><a name="l01538"></a><span class="lineno"> 1538</span>  </div> |
| <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> |
| <div class="line"><a name="l01540"></a><span class="lineno"> 1540</span>  </div> |
| <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> |
| <div class="line"><a name="l01542"></a><span class="lineno"> 1542</span>  {</div> |
| <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> |
| <div class="line"><a name="l01544"></a><span class="lineno"> 1544</span>  }</div> |
| <div class="line"><a name="l01545"></a><span class="lineno"> 1545</span>  </div> |
| <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> |
| <div class="line"><a name="l01547"></a><span class="lineno"> 1547</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01548"></a><span class="lineno"> 1548</span>  {</div> |
| <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> |
| <div class="line"><a name="l01550"></a><span class="lineno"> 1550</span>  }</div> |
| <div class="line"><a name="l01551"></a><span class="lineno"> 1551</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01554"></a><span class="lineno"> 1554</span>  </div> |
| <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> |
| <div class="line"><a name="l01556"></a><span class="lineno"> 1556</span>  </div> |
| <div class="line"><a name="l01557"></a><span class="lineno"> 1557</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01560"></a><span class="lineno"> 1560</span>  {</div> |
| <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> |
| <div class="line"><a name="l01562"></a><span class="lineno"> 1562</span>  IsElementwiseUnarySupported,</div> |
| <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> |
| <div class="line"><a name="l01564"></a><span class="lineno"> 1564</span>  isSupported,</div> |
| <div class="line"><a name="l01565"></a><span class="lineno"> 1565</span>  setBackend,</div> |
| <div class="line"><a name="l01566"></a><span class="lineno"> 1566</span>  inputInfo,</div> |
| <div class="line"><a name="l01567"></a><span class="lineno"> 1567</span>  outputInfo,</div> |
| <div class="line"><a name="l01568"></a><span class="lineno"> 1568</span>  descriptor);</div> |
| <div class="line"><a name="l01569"></a><span class="lineno"> 1569</span>  };</div> |
| <div class="line"><a name="l01570"></a><span class="lineno"> 1570</span>  </div> |
| <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> |
| <div class="line"><a name="l01572"></a><span class="lineno"> 1572</span>  {</div> |
| <div class="line"><a name="l01573"></a><span class="lineno"> 1573</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01574"></a><span class="lineno"> 1574</span>  }</div> |
| <div class="line"><a name="l01575"></a><span class="lineno"> 1575</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01576"></a><span class="lineno"> 1576</span>  {</div> |
| <div class="line"><a name="l01577"></a><span class="lineno"> 1577</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01578"></a><span class="lineno"> 1578</span>  }</div> |
| <div class="line"><a name="l01579"></a><span class="lineno"> 1579</span>  </div> |
| <div class="line"><a name="l01580"></a><span class="lineno"> 1580</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01581"></a><span class="lineno"> 1581</span>  {</div> |
| <div class="line"><a name="l01582"></a><span class="lineno"> 1582</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01583"></a><span class="lineno"> 1583</span>  }</div> |
| <div class="line"><a name="l01584"></a><span class="lineno"> 1584</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01587"></a><span class="lineno"> 1587</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l01589"></a><span class="lineno"> 1589</span>  </div> |
| <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> |
| <div class="line"><a name="l01591"></a><span class="lineno"> 1591</span> }</div> |
| <div class="line"><a name="l01592"></a><span class="lineno"> 1592</span>  </div> |
| <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> |
| <div class="line"><a name="l01594"></a><span class="lineno"> 1594</span> {</div> |
| <div class="line"><a name="l01595"></a><span class="lineno"> 1595</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertElu()"</span>;</div> |
| <div class="line"><a name="l01596"></a><span class="lineno"> 1596</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01599"></a><span class="lineno"> 1599</span>  {</div> |
| <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> |
| <div class="line"><a name="l01601"></a><span class="lineno"> 1601</span>  }</div> |
| <div class="line"><a name="l01602"></a><span class="lineno"> 1602</span>  </div> |
| <div class="line"><a name="l01603"></a><span class="lineno"> 1603</span>  <span class="comment">// Determine data type of input tensor</span></div> |
| <div class="line"><a name="l01604"></a><span class="lineno"> 1604</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> inputType;</div> |
| <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> |
| <div class="line"><a name="l01606"></a><span class="lineno"> 1606</span>  {</div> |
| <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> |
| <div class="line"><a name="l01608"></a><span class="lineno"> 1608</span>  }</div> |
| <div class="line"><a name="l01609"></a><span class="lineno"> 1609</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01612"></a><span class="lineno"> 1612</span>  </div> |
| <div class="line"><a name="l01613"></a><span class="lineno"> 1613</span>  <span class="comment">// Read alpha</span></div> |
| <div class="line"><a name="l01614"></a><span class="lineno"> 1614</span>  <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT16)</div> |
| <div class="line"><a name="l01615"></a><span class="lineno"> 1615</span>  {</div> |
| <div class="line"><a name="l01616"></a><span class="lineno"> 1616</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> alpha;</div> |
| <div class="line"><a name="l01617"></a><span class="lineno"> 1617</span>  </div> |
| <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> |
| <div class="line"><a name="l01619"></a><span class="lineno"> 1619</span>  {</div> |
| <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> |
| <div class="line"><a name="l01621"></a><span class="lineno"> 1621</span>  }</div> |
| <div class="line"><a name="l01622"></a><span class="lineno"> 1622</span>  </div> |
| <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> |
| <div class="line"><a name="l01624"></a><span class="lineno"> 1624</span>  }</div> |
| <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> |
| <div class="line"><a name="l01626"></a><span class="lineno"> 1626</span>  {</div> |
| <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> |
| <div class="line"><a name="l01628"></a><span class="lineno"> 1628</span>  {</div> |
| <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> |
| <div class="line"><a name="l01630"></a><span class="lineno"> 1630</span>  }</div> |
| <div class="line"><a name="l01631"></a><span class="lineno"> 1631</span>  }</div> |
| <div class="line"><a name="l01632"></a><span class="lineno"> 1632</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01633"></a><span class="lineno"> 1633</span>  {</div> |
| <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> |
| <div class="line"><a name="l01635"></a><span class="lineno"> 1635</span>  }</div> |
| <div class="line"><a name="l01636"></a><span class="lineno"> 1636</span>  </div> |
| <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> |
| <div class="line"><a name="l01638"></a><span class="lineno"> 1638</span> }</div> |
| <div class="line"><a name="l01639"></a><span class="lineno"> 1639</span>  </div> |
| <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> |
| <div class="line"><a name="l01641"></a><span class="lineno"> 1641</span> {</div> |
| <div class="line"><a name="l01642"></a><span class="lineno"> 1642</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertExpandDims()"</span>;</div> |
| <div class="line"><a name="l01643"></a><span class="lineno"> 1643</span>  </div> |
| <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> |
| <div class="line"><a name="l01645"></a><span class="lineno"> 1645</span>  </div> |
| <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> |
| <div class="line"><a name="l01647"></a><span class="lineno"> 1647</span>  {</div> |
| <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> |
| <div class="line"><a name="l01649"></a><span class="lineno"> 1649</span>  }</div> |
| <div class="line"><a name="l01650"></a><span class="lineno"> 1650</span>  </div> |
| <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> |
| <div class="line"><a name="l01652"></a><span class="lineno"> 1652</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01653"></a><span class="lineno"> 1653</span>  {</div> |
| <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> |
| <div class="line"><a name="l01655"></a><span class="lineno"> 1655</span>  }</div> |
| <div class="line"><a name="l01656"></a><span class="lineno"> 1656</span>  </div> |
| <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> |
| <div class="line"><a name="l01658"></a><span class="lineno"> 1658</span>  </div> |
| <div class="line"><a name="l01659"></a><span class="lineno"> 1659</span>  int32_t axis;</div> |
| <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> |
| <div class="line"><a name="l01661"></a><span class="lineno"> 1661</span>  {</div> |
| <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> |
| <div class="line"><a name="l01663"></a><span class="lineno"> 1663</span>  }</div> |
| <div class="line"><a name="l01664"></a><span class="lineno"> 1664</span>  </div> |
| <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> |
| <div class="line"><a name="l01666"></a><span class="lineno"> 1666</span>  </div> |
| <div class="line"><a name="l01667"></a><span class="lineno"> 1667</span>  <span class="keywordflow">try</span></div> |
| <div class="line"><a name="l01668"></a><span class="lineno"> 1668</span>  {</div> |
| <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> |
| <div class="line"><a name="l01670"></a><span class="lineno"> 1670</span>  }</div> |
| <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> |
| <div class="line"><a name="l01672"></a><span class="lineno"> 1672</span>  {</div> |
| <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> |
| <div class="line"><a name="l01674"></a><span class="lineno"> 1674</span>  }</div> |
| <div class="line"><a name="l01675"></a><span class="lineno"> 1675</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01678"></a><span class="lineno"> 1678</span>  </div> |
| <div class="line"><a name="l01679"></a><span class="lineno"> 1679</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01682"></a><span class="lineno"> 1682</span>  {</div> |
| <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> |
| <div class="line"><a name="l01684"></a><span class="lineno"> 1684</span>  IsReshapeSupported,</div> |
| <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> |
| <div class="line"><a name="l01686"></a><span class="lineno"> 1686</span>  isSupported,</div> |
| <div class="line"><a name="l01687"></a><span class="lineno"> 1687</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l01689"></a><span class="lineno"> 1689</span>  outputInfo,</div> |
| <div class="line"><a name="l01690"></a><span class="lineno"> 1690</span>  reshapeDescriptor);</div> |
| <div class="line"><a name="l01691"></a><span class="lineno"> 1691</span>  };</div> |
| <div class="line"><a name="l01692"></a><span class="lineno"> 1692</span>  </div> |
| <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> |
| <div class="line"><a name="l01694"></a><span class="lineno"> 1694</span>  {</div> |
| <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> |
| <div class="line"><a name="l01696"></a><span class="lineno"> 1696</span>  {</div> |
| <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> |
| <div class="line"><a name="l01698"></a><span class="lineno"> 1698</span>  }</div> |
| <div class="line"><a name="l01699"></a><span class="lineno"> 1699</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01700"></a><span class="lineno"> 1700</span>  }</div> |
| <div class="line"><a name="l01701"></a><span class="lineno"> 1701</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01702"></a><span class="lineno"> 1702</span>  {</div> |
| <div class="line"><a name="l01703"></a><span class="lineno"> 1703</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01704"></a><span class="lineno"> 1704</span>  }</div> |
| <div class="line"><a name="l01705"></a><span class="lineno"> 1705</span>  </div> |
| <div class="line"><a name="l01706"></a><span class="lineno"> 1706</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01707"></a><span class="lineno"> 1707</span>  {</div> |
| <div class="line"><a name="l01708"></a><span class="lineno"> 1708</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01709"></a><span class="lineno"> 1709</span>  }</div> |
| <div class="line"><a name="l01710"></a><span class="lineno"> 1710</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01713"></a><span class="lineno"> 1713</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l01715"></a><span class="lineno"> 1715</span>  </div> |
| <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> |
| <div class="line"><a name="l01717"></a><span class="lineno"> 1717</span> }</div> |
| <div class="line"><a name="l01718"></a><span class="lineno"> 1718</span>  </div> |
| <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> |
| <div class="line"><a name="l01720"></a><span class="lineno"> 1720</span> {</div> |
| <div class="line"><a name="l01721"></a><span class="lineno"> 1721</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertFill()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01724"></a><span class="lineno"> 1724</span>  {</div> |
| <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> |
| <div class="line"><a name="l01726"></a><span class="lineno"> 1726</span>  }</div> |
| <div class="line"><a name="l01727"></a><span class="lineno"> 1727</span>  </div> |
| <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> |
| <div class="line"><a name="l01729"></a><span class="lineno"> 1729</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01730"></a><span class="lineno"> 1730</span>  {</div> |
| <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> |
| <div class="line"><a name="l01732"></a><span class="lineno"> 1732</span>  }</div> |
| <div class="line"><a name="l01733"></a><span class="lineno"> 1733</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01737"></a><span class="lineno"> 1737</span>  {</div> |
| <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> |
| <div class="line"><a name="l01739"></a><span class="lineno"> 1739</span>  }</div> |
| <div class="line"><a name="l01740"></a><span class="lineno"> 1740</span>  </div> |
| <div class="line"><a name="l01741"></a><span class="lineno"> 1741</span>  <span class="comment">// Determine data type of output tensor</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l01744"></a><span class="lineno"> 1744</span>  <span class="comment">// Read the scalar fill value</span></div> |
| <div class="line"><a name="l01745"></a><span class="lineno"> 1745</span>  <span class="keywordflow">if</span> (outputType == OperandType::TENSOR_FLOAT16)</div> |
| <div class="line"><a name="l01746"></a><span class="lineno"> 1746</span>  {</div> |
| <div class="line"><a name="l01747"></a><span class="lineno"> 1747</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> value;</div> |
| <div class="line"><a name="l01748"></a><span class="lineno"> 1748</span>  </div> |
| <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> |
| <div class="line"><a name="l01750"></a><span class="lineno"> 1750</span>  {</div> |
| <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> |
| <div class="line"><a name="l01752"></a><span class="lineno"> 1752</span>  }</div> |
| <div class="line"><a name="l01753"></a><span class="lineno"> 1753</span>  </div> |
| <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> |
| <div class="line"><a name="l01755"></a><span class="lineno"> 1755</span>  }</div> |
| <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> |
| <div class="line"><a name="l01757"></a><span class="lineno"> 1757</span>  {</div> |
| <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> |
| <div class="line"><a name="l01759"></a><span class="lineno"> 1759</span>  {</div> |
| <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> |
| <div class="line"><a name="l01761"></a><span class="lineno"> 1761</span>  }</div> |
| <div class="line"><a name="l01762"></a><span class="lineno"> 1762</span>  }</div> |
| <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> |
| <div class="line"><a name="l01764"></a><span class="lineno"> 1764</span>  {</div> |
| <div class="line"><a name="l01765"></a><span class="lineno"> 1765</span>  int32_t value;</div> |
| <div class="line"><a name="l01766"></a><span class="lineno"> 1766</span>  </div> |
| <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> |
| <div class="line"><a name="l01768"></a><span class="lineno"> 1768</span>  {</div> |
| <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> |
| <div class="line"><a name="l01770"></a><span class="lineno"> 1770</span>  }</div> |
| <div class="line"><a name="l01771"></a><span class="lineno"> 1771</span>  </div> |
| <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> |
| <div class="line"><a name="l01773"></a><span class="lineno"> 1773</span>  }</div> |
| <div class="line"><a name="l01774"></a><span class="lineno"> 1774</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01775"></a><span class="lineno"> 1775</span>  {</div> |
| <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> |
| <div class="line"><a name="l01777"></a><span class="lineno"> 1777</span>  }</div> |
| <div class="line"><a name="l01778"></a><span class="lineno"> 1778</span>  </div> |
| <div class="line"><a name="l01779"></a><span class="lineno"> 1779</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01782"></a><span class="lineno"> 1782</span>  IsFillSupported,</div> |
| <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> |
| <div class="line"><a name="l01784"></a><span class="lineno"> 1784</span>  isSupported,</div> |
| <div class="line"><a name="l01785"></a><span class="lineno"> 1785</span>  setBackend,</div> |
| <div class="line"><a name="l01786"></a><span class="lineno"> 1786</span>  inputInfo,</div> |
| <div class="line"><a name="l01787"></a><span class="lineno"> 1787</span>  outputInfo,</div> |
| <div class="line"><a name="l01788"></a><span class="lineno"> 1788</span>  descriptor);</div> |
| <div class="line"><a name="l01789"></a><span class="lineno"> 1789</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01790"></a><span class="lineno"> 1790</span>  {</div> |
| <div class="line"><a name="l01791"></a><span class="lineno"> 1791</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01792"></a><span class="lineno"> 1792</span>  }</div> |
| <div class="line"><a name="l01793"></a><span class="lineno"> 1793</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01796"></a><span class="lineno"> 1796</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l01798"></a><span class="lineno"> 1798</span>  </div> |
| <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> |
| <div class="line"><a name="l01800"></a><span class="lineno"> 1800</span> }</div> |
| <div class="line"><a name="l01801"></a><span class="lineno"> 1801</span>  </div> |
| <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> |
| <div class="line"><a name="l01803"></a><span class="lineno"> 1803</span> {</div> |
| <div class="line"><a name="l01804"></a><span class="lineno"> 1804</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertFloor()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01807"></a><span class="lineno"> 1807</span>  {</div> |
| <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> |
| <div class="line"><a name="l01809"></a><span class="lineno"> 1809</span>  }</div> |
| <div class="line"><a name="l01810"></a><span class="lineno"> 1810</span>  </div> |
| <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> |
| <div class="line"><a name="l01812"></a><span class="lineno"> 1812</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l01813"></a><span class="lineno"> 1813</span>  {</div> |
| <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> |
| <div class="line"><a name="l01815"></a><span class="lineno"> 1815</span>  }</div> |
| <div class="line"><a name="l01816"></a><span class="lineno"> 1816</span>  </div> |
| <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> |
| <div class="line"><a name="l01818"></a><span class="lineno"> 1818</span>  </div> |
| <div class="line"><a name="l01819"></a><span class="lineno"> 1819</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01822"></a><span class="lineno"> 1822</span>  {</div> |
| <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> |
| <div class="line"><a name="l01824"></a><span class="lineno"> 1824</span>  IsFloorSupported,</div> |
| <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> |
| <div class="line"><a name="l01826"></a><span class="lineno"> 1826</span>  isSupported,</div> |
| <div class="line"><a name="l01827"></a><span class="lineno"> 1827</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l01829"></a><span class="lineno"> 1829</span>  outputInfo);</div> |
| <div class="line"><a name="l01830"></a><span class="lineno"> 1830</span>  };</div> |
| <div class="line"><a name="l01831"></a><span class="lineno"> 1831</span>  </div> |
| <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> |
| <div class="line"><a name="l01833"></a><span class="lineno"> 1833</span>  {</div> |
| <div class="line"><a name="l01834"></a><span class="lineno"> 1834</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01835"></a><span class="lineno"> 1835</span>  }</div> |
| <div class="line"><a name="l01836"></a><span class="lineno"> 1836</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01837"></a><span class="lineno"> 1837</span>  {</div> |
| <div class="line"><a name="l01838"></a><span class="lineno"> 1838</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01839"></a><span class="lineno"> 1839</span>  }</div> |
| <div class="line"><a name="l01840"></a><span class="lineno"> 1840</span>  </div> |
| <div class="line"><a name="l01841"></a><span class="lineno"> 1841</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01842"></a><span class="lineno"> 1842</span>  {</div> |
| <div class="line"><a name="l01843"></a><span class="lineno"> 1843</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01844"></a><span class="lineno"> 1844</span>  }</div> |
| <div class="line"><a name="l01845"></a><span class="lineno"> 1845</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01848"></a><span class="lineno"> 1848</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l01850"></a><span class="lineno"> 1850</span>  </div> |
| <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> |
| <div class="line"><a name="l01852"></a><span class="lineno"> 1852</span> }</div> |
| <div class="line"><a name="l01853"></a><span class="lineno"> 1853</span>  </div> |
| <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> |
| <div class="line"><a name="l01855"></a><span class="lineno"> 1855</span> {</div> |
| <div class="line"><a name="l01856"></a><span class="lineno"> 1856</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertFullyConnected()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01859"></a><span class="lineno"> 1859</span>  {</div> |
| <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> |
| <div class="line"><a name="l01861"></a><span class="lineno"> 1861</span>  }</div> |
| <div class="line"><a name="l01862"></a><span class="lineno"> 1862</span>  </div> |
| <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> |
| <div class="line"><a name="l01864"></a><span class="lineno"> 1864</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l01865"></a><span class="lineno"> 1865</span>  {</div> |
| <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> |
| <div class="line"><a name="l01867"></a><span class="lineno"> 1867</span>  }</div> |
| <div class="line"><a name="l01868"></a><span class="lineno"> 1868</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01871"></a><span class="lineno"> 1871</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01874"></a><span class="lineno"> 1874</span>  <span class="keywordflow">if</span> (!weightsOperand)</div> |
| <div class="line"><a name="l01875"></a><span class="lineno"> 1875</span>  {</div> |
| <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> |
| <div class="line"><a name="l01877"></a><span class="lineno"> 1877</span>  }</div> |
| <div class="line"><a name="l01878"></a><span class="lineno"> 1878</span>  </div> |
| <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> |
| <div class="line"><a name="l01880"></a><span class="lineno"> 1880</span>  <span class="comment">// Otherwise handle non const weights as inputs.</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l01883"></a><span class="lineno"> 1883</span>  {</div> |
| <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> |
| <div class="line"><a name="l01885"></a><span class="lineno"> 1885</span>  }</div> |
| <div class="line"><a name="l01886"></a><span class="lineno"> 1886</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01889"></a><span class="lineno"> 1889</span>  <span class="keywordflow">if</span> (!biasOperand)</div> |
| <div class="line"><a name="l01890"></a><span class="lineno"> 1890</span>  {</div> |
| <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> |
| <div class="line"><a name="l01892"></a><span class="lineno"> 1892</span>  }</div> |
| <div class="line"><a name="l01893"></a><span class="lineno"> 1893</span>  </div> |
| <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> |
| <div class="line"><a name="l01895"></a><span class="lineno"> 1895</span>  <span class="comment">// Otherwise handle non const bias as inputs.</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l01898"></a><span class="lineno"> 1898</span>  {</div> |
| <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> |
| <div class="line"><a name="l01900"></a><span class="lineno"> 1900</span>  }</div> |
| <div class="line"><a name="l01901"></a><span class="lineno"> 1901</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01904"></a><span class="lineno"> 1904</span>  <span class="keywordflow">try</span></div> |
| <div class="line"><a name="l01905"></a><span class="lineno"> 1905</span>  {</div> |
| <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> |
| <div class="line"><a name="l01907"></a><span class="lineno"> 1907</span>  }</div> |
| <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> |
| <div class="line"><a name="l01909"></a><span class="lineno"> 1909</span>  {</div> |
| <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> |
| <div class="line"><a name="l01911"></a><span class="lineno"> 1911</span>  }</div> |
| <div class="line"><a name="l01912"></a><span class="lineno"> 1912</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l01915"></a><span class="lineno"> 1915</span>  SanitizeBiasQuantizationScale(biasInfo, weightsInfo, reshapedInfo);</div> |
| <div class="line"><a name="l01916"></a><span class="lineno"> 1916</span>  </div> |
| <div class="line"><a name="l01917"></a><span class="lineno"> 1917</span>  ActivationFn activationFunction;</div> |
| <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> |
| <div class="line"><a name="l01919"></a><span class="lineno"> 1919</span>  {</div> |
| <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> |
| <div class="line"><a name="l01921"></a><span class="lineno"> 1921</span>  }</div> |
| <div class="line"><a name="l01922"></a><span class="lineno"> 1922</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01927"></a><span class="lineno"> 1927</span>  </div> |
| <div class="line"><a name="l01928"></a><span class="lineno"> 1928</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01931"></a><span class="lineno"> 1931</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01936"></a><span class="lineno"> 1936</span>  {</div> |
| <div class="line"><a name="l01937"></a><span class="lineno"> 1937</span>  isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l01939"></a><span class="lineno"> 1939</span>  <span class="keywordflow">return</span>;</div> |
| <div class="line"><a name="l01940"></a><span class="lineno"> 1940</span>  }</div> |
| <div class="line"><a name="l01941"></a><span class="lineno"> 1941</span>  </div> |
| <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> |
| <div class="line"><a name="l01943"></a><span class="lineno"> 1943</span>  IsFullyConnectedSupported,</div> |
| <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> |
| <div class="line"><a name="l01945"></a><span class="lineno"> 1945</span>  isSupported,</div> |
| <div class="line"><a name="l01946"></a><span class="lineno"> 1946</span>  setBackend,</div> |
| <div class="line"><a name="l01947"></a><span class="lineno"> 1947</span>  reshapedInfo,</div> |
| <div class="line"><a name="l01948"></a><span class="lineno"> 1948</span>  outputInfo,</div> |
| <div class="line"><a name="l01949"></a><span class="lineno"> 1949</span>  weightsInfo,</div> |
| <div class="line"><a name="l01950"></a><span class="lineno"> 1950</span>  biasInfo,</div> |
| <div class="line"><a name="l01951"></a><span class="lineno"> 1951</span>  desc);</div> |
| <div class="line"><a name="l01952"></a><span class="lineno"> 1952</span>  };</div> |
| <div class="line"><a name="l01953"></a><span class="lineno"> 1953</span>  </div> |
| <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> |
| <div class="line"><a name="l01955"></a><span class="lineno"> 1955</span>  {</div> |
| <div class="line"><a name="l01956"></a><span class="lineno"> 1956</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l01957"></a><span class="lineno"> 1957</span>  }</div> |
| <div class="line"><a name="l01958"></a><span class="lineno"> 1958</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01959"></a><span class="lineno"> 1959</span>  {</div> |
| <div class="line"><a name="l01960"></a><span class="lineno"> 1960</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l01961"></a><span class="lineno"> 1961</span>  }</div> |
| <div class="line"><a name="l01962"></a><span class="lineno"> 1962</span>  </div> |
| <div class="line"><a name="l01963"></a><span class="lineno"> 1963</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l01964"></a><span class="lineno"> 1964</span>  {</div> |
| <div class="line"><a name="l01965"></a><span class="lineno"> 1965</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l01966"></a><span class="lineno"> 1966</span>  }</div> |
| <div class="line"><a name="l01967"></a><span class="lineno"> 1967</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01971"></a><span class="lineno"> 1971</span>  </div> |
| <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> |
| <div class="line"><a name="l01973"></a><span class="lineno"> 1973</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l01976"></a><span class="lineno"> 1976</span>  </div> |
| <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> |
| <div class="line"><a name="l01978"></a><span class="lineno"> 1978</span>  assert(reshapeLayer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l01982"></a><span class="lineno"> 1982</span>  }</div> |
| <div class="line"><a name="l01983"></a><span class="lineno"> 1983</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l01984"></a><span class="lineno"> 1984</span>  {</div> |
| <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> |
| <div class="line"><a name="l01986"></a><span class="lineno"> 1986</span>  }</div> |
| <div class="line"><a name="l01987"></a><span class="lineno"> 1987</span>  </div> |
| <div class="line"><a name="l01988"></a><span class="lineno"> 1988</span>  <span class="comment">// Connect weights and bias inputs</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l01991"></a><span class="lineno"> 1991</span>  </div> |
| <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> |
| <div class="line"><a name="l01993"></a><span class="lineno"> 1993</span>  data, <span class="keyword">nullptr</span>, validateFunc, activationFunction);</div> |
| <div class="line"><a name="l01994"></a><span class="lineno"> 1994</span> }</div> |
| <div class="line"><a name="l01995"></a><span class="lineno"> 1995</span>  </div> |
| <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> |
| <div class="line"><a name="l01997"></a><span class="lineno"> 1997</span> {</div> |
| <div class="line"><a name="l01998"></a><span class="lineno"> 1998</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertGather()"</span>;</div> |
| <div class="line"><a name="l01999"></a><span class="lineno"> 1999</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02002"></a><span class="lineno"> 2002</span>  {</div> |
| <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> |
| <div class="line"><a name="l02004"></a><span class="lineno"> 2004</span>  }</div> |
| <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> |
| <div class="line"><a name="l02006"></a><span class="lineno"> 2006</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02009"></a><span class="lineno"> 2009</span>  {</div> |
| <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> |
| <div class="line"><a name="l02011"></a><span class="lineno"> 2011</span>  }</div> |
| <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> |
| <div class="line"><a name="l02013"></a><span class="lineno"> 2013</span>  </div> |
| <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> |
| <div class="line"><a name="l02015"></a><span class="lineno"> 2015</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02016"></a><span class="lineno"> 2016</span>  {</div> |
| <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> |
| <div class="line"><a name="l02018"></a><span class="lineno"> 2018</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02021"></a><span class="lineno"> 2021</span>  <span class="keywordflow">if</span> (outputDimensions != inputDimensions + indicesDimensions - 1)</div> |
| <div class="line"><a name="l02022"></a><span class="lineno"> 2022</span>  {</div> |
| <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> |
| <div class="line"><a name="l02024"></a><span class="lineno"> 2024</span>  __func__, outputDimensions, inputDimensions, indicesDimensions);</div> |
| <div class="line"><a name="l02025"></a><span class="lineno"> 2025</span>  }</div> |
| <div class="line"><a name="l02026"></a><span class="lineno"> 2026</span>  </div> |
| <div class="line"><a name="l02027"></a><span class="lineno"> 2027</span>  int32_t axis;</div> |
| <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> |
| <div class="line"><a name="l02029"></a><span class="lineno"> 2029</span>  {</div> |
| <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> |
| <div class="line"><a name="l02031"></a><span class="lineno"> 2031</span>  }</div> |
| <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> |
| <div class="line"><a name="l02033"></a><span class="lineno"> 2033</span>  <span class="keywordflow">if</span> ((axis < -inputDimensions_int) || (inputDimensions_int <= axis))</div> |
| <div class="line"><a name="l02034"></a><span class="lineno"> 2034</span>  {</div> |
| <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> |
| <div class="line"><a name="l02036"></a><span class="lineno"> 2036</span>  inputDimensions, inputDimensions);</div> |
| <div class="line"><a name="l02037"></a><span class="lineno"> 2037</span>  }</div> |
| <div class="line"><a name="l02038"></a><span class="lineno"> 2038</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02041"></a><span class="lineno"> 2041</span>  </div> |
| <div class="line"><a name="l02042"></a><span class="lineno"> 2042</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02045"></a><span class="lineno"> 2045</span>  {</div> |
| <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> |
| <div class="line"><a name="l02047"></a><span class="lineno"> 2047</span>  IsGatherSupported,</div> |
| <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> |
| <div class="line"><a name="l02049"></a><span class="lineno"> 2049</span>  isSupported,</div> |
| <div class="line"><a name="l02050"></a><span class="lineno"> 2050</span>  setBackend,</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02053"></a><span class="lineno"> 2053</span>  outputInfo,</div> |
| <div class="line"><a name="l02054"></a><span class="lineno"> 2054</span>  desc);</div> |
| <div class="line"><a name="l02055"></a><span class="lineno"> 2055</span>  };</div> |
| <div class="line"><a name="l02056"></a><span class="lineno"> 2056</span>  </div> |
| <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> |
| <div class="line"><a name="l02058"></a><span class="lineno"> 2058</span>  {</div> |
| <div class="line"><a name="l02059"></a><span class="lineno"> 2059</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l02060"></a><span class="lineno"> 2060</span>  }</div> |
| <div class="line"><a name="l02061"></a><span class="lineno"> 2061</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02062"></a><span class="lineno"> 2062</span>  {</div> |
| <div class="line"><a name="l02063"></a><span class="lineno"> 2063</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02064"></a><span class="lineno"> 2064</span>  }</div> |
| <div class="line"><a name="l02065"></a><span class="lineno"> 2065</span>  </div> |
| <div class="line"><a name="l02066"></a><span class="lineno"> 2066</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02067"></a><span class="lineno"> 2067</span>  {</div> |
| <div class="line"><a name="l02068"></a><span class="lineno"> 2068</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02069"></a><span class="lineno"> 2069</span>  }</div> |
| <div class="line"><a name="l02070"></a><span class="lineno"> 2070</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02073"></a><span class="lineno"> 2073</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02076"></a><span class="lineno"> 2076</span>  </div> |
| <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> |
| <div class="line"><a name="l02078"></a><span class="lineno"> 2078</span> }</div> |
| <div class="line"><a name="l02079"></a><span class="lineno"> 2079</span>  </div> |
| <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> |
| <div class="line"><a name="l02081"></a><span class="lineno"> 2081</span> {</div> |
| <div class="line"><a name="l02082"></a><span class="lineno"> 2082</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertGroupedConv2d()"</span>;</div> |
| <div class="line"><a name="l02083"></a><span class="lineno"> 2083</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02084"></a><span class="lineno"> 2084</span>  <span class="comment">// Parse data</span></div> |
| <div class="line"><a name="l02085"></a><span class="lineno"> 2085</span>  <span class="comment">//</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l02088"></a><span class="lineno"> 2088</span>  {</div> |
| <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> |
| <div class="line"><a name="l02090"></a><span class="lineno"> 2090</span>  }</div> |
| <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> |
| <div class="line"><a name="l02092"></a><span class="lineno"> 2092</span>  </div> |
| <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> |
| <div class="line"><a name="l02094"></a><span class="lineno"> 2094</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02095"></a><span class="lineno"> 2095</span>  {</div> |
| <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> |
| <div class="line"><a name="l02097"></a><span class="lineno"> 2097</span>  }</div> |
| <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> |
| <div class="line"><a name="l02099"></a><span class="lineno"> 2099</span>  </div> |
| <div class="line"><a name="l02100"></a><span class="lineno"> 2100</span>  <span class="comment">// Look ahead to determine data layout</span></div> |
| <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> |
| <div class="line"><a name="l02102"></a><span class="lineno"> 2102</span>  <span class="keywordflow">if</span> (operation.inputs.size() == 12)</div> |
| <div class="line"><a name="l02103"></a><span class="lineno"> 2103</span>  {</div> |
| <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> |
| <div class="line"><a name="l02105"></a><span class="lineno"> 2105</span>  }</div> |
| <div class="line"><a name="l02106"></a><span class="lineno"> 2106</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02107"></a><span class="lineno"> 2107</span>  {</div> |
| <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> |
| <div class="line"><a name="l02109"></a><span class="lineno"> 2109</span>  }</div> |
| <div class="line"><a name="l02110"></a><span class="lineno"> 2110</span>  </div> |
| <div class="line"><a name="l02111"></a><span class="lineno"> 2111</span>  <span class="comment">// NOTE:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02118"></a><span class="lineno"> 2118</span>  model, data, ohwiToOihw) :</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02123"></a><span class="lineno"> 2123</span>  {</div> |
| <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> |
| <div class="line"><a name="l02125"></a><span class="lineno"> 2125</span>  }</div> |
| <div class="line"><a name="l02126"></a><span class="lineno"> 2126</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02130"></a><span class="lineno"> 2130</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02135"></a><span class="lineno"> 2135</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02140"></a><span class="lineno"> 2140</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02144"></a><span class="lineno"> 2144</span>  </div> |
| <div class="line"><a name="l02145"></a><span class="lineno"> 2145</span>  <span class="keywordtype">int</span> numGroups;</div> |
| <div class="line"><a name="l02146"></a><span class="lineno"> 2146</span>  ActivationFn activation;</div> |
| <div class="line"><a name="l02147"></a><span class="lineno"> 2147</span>  </div> |
| <div class="line"><a name="l02148"></a><span class="lineno"> 2148</span>  <span class="keywordflow">if</span> (operation.inputs.size() == 12)</div> |
| <div class="line"><a name="l02149"></a><span class="lineno"> 2149</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02158"></a><span class="lineno"> 2158</span>  {</div> |
| <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> |
| <div class="line"><a name="l02160"></a><span class="lineno"> 2160</span>  }</div> |
| <div class="line"><a name="l02161"></a><span class="lineno"> 2161</span>  </div> |
| <div class="line"><a name="l02162"></a><span class="lineno"> 2162</span>  }</div> |
| <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> |
| <div class="line"><a name="l02164"></a><span class="lineno"> 2164</span>  {</div> |
| <div class="line"><a name="l02165"></a><span class="lineno"> 2165</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02171"></a><span class="lineno"> 2171</span>  {</div> |
| <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> |
| <div class="line"><a name="l02173"></a><span class="lineno"> 2173</span>  }</div> |
| <div class="line"><a name="l02174"></a><span class="lineno"> 2174</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02177"></a><span class="lineno"> 2177</span>  </div> |
| <div class="line"><a name="l02178"></a><span class="lineno"> 2178</span>  <span class="keyword">const</span> uint32_t kernelX = weightsShape[widthIndex];</div> |
| <div class="line"><a name="l02179"></a><span class="lineno"> 2179</span>  <span class="keyword">const</span> uint32_t kernelY = weightsShape[heightIndex];</div> |
| <div class="line"><a name="l02180"></a><span class="lineno"> 2180</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02183"></a><span class="lineno"> 2183</span>  }</div> |
| <div class="line"><a name="l02184"></a><span class="lineno"> 2184</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02185"></a><span class="lineno"> 2185</span>  {</div> |
| <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> |
| <div class="line"><a name="l02187"></a><span class="lineno"> 2187</span>  }</div> |
| <div class="line"><a name="l02188"></a><span class="lineno"> 2188</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02191"></a><span class="lineno"> 2191</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02194"></a><span class="lineno"> 2194</span>  </div> |
| <div class="line"><a name="l02195"></a><span class="lineno"> 2195</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02196"></a><span class="lineno"> 2196</span>  <span class="comment">// Validate all relevant inputs</span></div> |
| <div class="line"><a name="l02197"></a><span class="lineno"> 2197</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02198"></a><span class="lineno"> 2198</span>  <span class="keywordflow">if</span> (numGroups <= 0)</div> |
| <div class="line"><a name="l02199"></a><span class="lineno"> 2199</span>  {</div> |
| <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> |
| <div class="line"><a name="l02201"></a><span class="lineno"> 2201</span>  }</div> |
| <div class="line"><a name="l02202"></a><span class="lineno"> 2202</span>  </div> |
| <div class="line"><a name="l02203"></a><span class="lineno"> 2203</span>  <span class="keywordflow">if</span> (outputChannels % numGroups != 0u)</div> |
| <div class="line"><a name="l02204"></a><span class="lineno"> 2204</span>  {</div> |
| <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> |
| <div class="line"><a name="l02206"></a><span class="lineno"> 2206</span>  }</div> |
| <div class="line"><a name="l02207"></a><span class="lineno"> 2207</span>  </div> |
| <div class="line"><a name="l02208"></a><span class="lineno"> 2208</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02209"></a><span class="lineno"> 2209</span>  <span class="comment">// Set up Splitter layer</span></div> |
| <div class="line"><a name="l02210"></a><span class="lineno"> 2210</span>  <span class="comment">//</span></div> |
| <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> |
| <div class="line"><a name="l02212"></a><span class="lineno"> 2212</span>  splitterDimSizes[channelsIndex] /= numGroups; <span class="comment">// split in depth</span></div> |
| <div class="line"><a name="l02213"></a><span class="lineno"> 2213</span>  </div> |
| <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> |
| <div class="line"><a name="l02215"></a><span class="lineno"> 2215</span>  splitterDimSizes,</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02219"></a><span class="lineno"> 2219</span>  </div> |
| <div class="line"><a name="l02220"></a><span class="lineno"> 2220</span>  std::vector<std::reference_wrapper<TensorInfo>> splitterOutputInfos(numGroups, std::ref(splitterOutputInfo));</div> |
| <div class="line"><a name="l02221"></a><span class="lineno"> 2221</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02224"></a><span class="lineno"> 2224</span>  {</div> |
| <div class="line"><a name="l02225"></a><span class="lineno"> 2225</span>  splitterDesc.SetViewOriginCoord(group, channelsIndex, splitterDimSizes[channelsIndex] * group);</div> |
| <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> |
| <div class="line"><a name="l02227"></a><span class="lineno"> 2227</span>  {</div> |
| <div class="line"><a name="l02228"></a><span class="lineno"> 2228</span>  splitterDesc.SetViewSize(group, dimIdx, splitterDimSizes[dimIdx]);</div> |
| <div class="line"><a name="l02229"></a><span class="lineno"> 2229</span>  }</div> |
| <div class="line"><a name="l02230"></a><span class="lineno"> 2230</span>  }</div> |
| <div class="line"><a name="l02231"></a><span class="lineno"> 2231</span>  </div> |
| <div class="line"><a name="l02232"></a><span class="lineno"> 2232</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02235"></a><span class="lineno"> 2235</span>  IsSplitterSupported,</div> |
| <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> |
| <div class="line"><a name="l02237"></a><span class="lineno"> 2237</span>  isSupported,</div> |
| <div class="line"><a name="l02238"></a><span class="lineno"> 2238</span>  setBackendSplit,</div> |
| <div class="line"><a name="l02239"></a><span class="lineno"> 2239</span>  inputInfo,</div> |
| <div class="line"><a name="l02240"></a><span class="lineno"> 2240</span>  splitterOutputInfos,</div> |
| <div class="line"><a name="l02241"></a><span class="lineno"> 2241</span>  splitterDesc);</div> |
| <div class="line"><a name="l02242"></a><span class="lineno"> 2242</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02243"></a><span class="lineno"> 2243</span>  {</div> |
| <div class="line"><a name="l02244"></a><span class="lineno"> 2244</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02245"></a><span class="lineno"> 2245</span>  }</div> |
| <div class="line"><a name="l02246"></a><span class="lineno"> 2246</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02249"></a><span class="lineno"> 2249</span>  <span class="keywordflow">if</span> (!splitterLayer)</div> |
| <div class="line"><a name="l02250"></a><span class="lineno"> 2250</span>  {</div> |
| <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> |
| <div class="line"><a name="l02252"></a><span class="lineno"> 2252</span>  }</div> |
| <div class="line"><a name="l02253"></a><span class="lineno"> 2253</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02256"></a><span class="lineno"> 2256</span>  {</div> |
| <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> |
| <div class="line"><a name="l02258"></a><span class="lineno"> 2258</span>  }</div> |
| <div class="line"><a name="l02259"></a><span class="lineno"> 2259</span>  </div> |
| <div class="line"><a name="l02260"></a><span class="lineno"> 2260</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02261"></a><span class="lineno"> 2261</span>  <span class="comment">// Set up Convolution2d layers for each group</span></div> |
| <div class="line"><a name="l02262"></a><span class="lineno"> 2262</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02263"></a><span class="lineno"> 2263</span>  </div> |
| <div class="line"><a name="l02264"></a><span class="lineno"> 2264</span>  <span class="comment">// Set up group tensor shapes</span></div> |
| <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> |
| <div class="line"><a name="l02266"></a><span class="lineno"> 2266</span>  groupInputShape[channelsIndex] = channelsPerGroup;</div> |
| <div class="line"><a name="l02267"></a><span class="lineno"> 2267</span>  </div> |
| <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> |
| <div class="line"><a name="l02269"></a><span class="lineno"> 2269</span>  groupWeightsShape[0] /= channelMultiplier * numGroups;</div> |
| <div class="line"><a name="l02270"></a><span class="lineno"> 2270</span>  </div> |
| <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> |
| <div class="line"><a name="l02272"></a><span class="lineno"> 2272</span>  </div> |
| <div class="line"><a name="l02273"></a><span class="lineno"> 2273</span>  <span class="comment">// Set up group tensor infos</span></div> |
| <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> |
| <div class="line"><a name="l02275"></a><span class="lineno"> 2275</span>  groupInputInfo.SetShape(groupInputShape);</div> |
| <div class="line"><a name="l02276"></a><span class="lineno"> 2276</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02279"></a><span class="lineno"> 2279</span>  groupWeightsInfo.SetShape(groupWeightsShape);</div> |
| <div class="line"><a name="l02280"></a><span class="lineno"> 2280</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02283"></a><span class="lineno"> 2283</span>  groupBiasesInfo.SetShape(groupBiasesShape);</div> |
| <div class="line"><a name="l02284"></a><span class="lineno"> 2284</span>  </div> |
| <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> |
| <div class="line"><a name="l02286"></a><span class="lineno"> 2286</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02289"></a><span class="lineno"> 2289</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| <div class="line"><a name="l02290"></a><span class="lineno"> 2290</span>  {</div> |
| <div class="line"><a name="l02291"></a><span class="lineno"> 2291</span>  groupOutputShape[channelsIndex] = 1;</div> |
| <div class="line"><a name="l02292"></a><span class="lineno"> 2292</span>  }</div> |
| <div class="line"><a name="l02293"></a><span class="lineno"> 2293</span>  groupOutputInfo.SetShape(groupOutputShape);</div> |
| <div class="line"><a name="l02294"></a><span class="lineno"> 2294</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02297"></a><span class="lineno"> 2297</span>  </div> |
| <div class="line"><a name="l02298"></a><span class="lineno"> 2298</span>  std::vector<IConnectableLayer*> convLayers(numGroups * channelMultiplier, <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l02300"></a><span class="lineno"> 2300</span>  {</div> |
| <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> |
| <div class="line"><a name="l02302"></a><span class="lineno"> 2302</span>  {</div> |
| <div class="line"><a name="l02303"></a><span class="lineno"> 2303</span>  <span class="keyword">auto</span> index = group * channelMultiplier + m;</div> |
| <div class="line"><a name="l02304"></a><span class="lineno"> 2304</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02307"></a><span class="lineno"> 2307</span>  </div> |
| <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> |
| <div class="line"><a name="l02309"></a><span class="lineno"> 2309</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02312"></a><span class="lineno"> 2312</span>  groupWeightsInfo.SetQuantizationScales(</div> |
| <div class="line"><a name="l02313"></a><span class="lineno"> 2313</span>  std::vector<float>(weightsQuantScales.begin() + index,</div> |
| <div class="line"><a name="l02314"></a><span class="lineno"> 2314</span>  weightsQuantScales.begin() + index + groupWeightsShape[0]));</div> |
| <div class="line"><a name="l02315"></a><span class="lineno"> 2315</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02318"></a><span class="lineno"> 2318</span>  groupBiasesInfo.SetQuantizationScales(</div> |
| <div class="line"><a name="l02319"></a><span class="lineno"> 2319</span>  std::vector<float>(biasesQuantScales.begin() + index,</div> |
| <div class="line"><a name="l02320"></a><span class="lineno"> 2320</span>  biasesQuantScales.begin() + index + groupWeightsShape[0]));</div> |
| <div class="line"><a name="l02321"></a><span class="lineno"> 2321</span>  }</div> |
| <div class="line"><a name="l02322"></a><span class="lineno"> 2322</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02326"></a><span class="lineno"> 2326</span>  weightsDataOffset));</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02329"></a><span class="lineno"> 2329</span>  biasesDataOffset));</div> |
| <div class="line"><a name="l02330"></a><span class="lineno"> 2330</span>  </div> |
| <div class="line"><a name="l02331"></a><span class="lineno"> 2331</span>  isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02334"></a><span class="lineno"> 2334</span>  {</div> |
| <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> |
| <div class="line"><a name="l02336"></a><span class="lineno"> 2336</span>  IsConvolution2dSupported,</div> |
| <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> |
| <div class="line"><a name="l02338"></a><span class="lineno"> 2338</span>  isSupported,</div> |
| <div class="line"><a name="l02339"></a><span class="lineno"> 2339</span>  setBackendConv,</div> |
| <div class="line"><a name="l02340"></a><span class="lineno"> 2340</span>  groupInputInfo,</div> |
| <div class="line"><a name="l02341"></a><span class="lineno"> 2341</span>  outputInfo,</div> |
| <div class="line"><a name="l02342"></a><span class="lineno"> 2342</span>  desc,</div> |
| <div class="line"><a name="l02343"></a><span class="lineno"> 2343</span>  groupWeightsInfo,</div> |
| <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> |
| <div class="line"><a name="l02345"></a><span class="lineno"> 2345</span>  };</div> |
| <div class="line"><a name="l02346"></a><span class="lineno"> 2346</span>  </div> |
| <div class="line"><a name="l02347"></a><span class="lineno"> 2347</span>  <span class="keywordflow">if</span>(!isDynamic)</div> |
| <div class="line"><a name="l02348"></a><span class="lineno"> 2348</span>  {</div> |
| <div class="line"><a name="l02349"></a><span class="lineno"> 2349</span>  validateFunc(groupOutputInfo, isSupported);</div> |
| <div class="line"><a name="l02350"></a><span class="lineno"> 2350</span>  }</div> |
| <div class="line"><a name="l02351"></a><span class="lineno"> 2351</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02352"></a><span class="lineno"> 2352</span>  {</div> |
| <div class="line"><a name="l02353"></a><span class="lineno"> 2353</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02354"></a><span class="lineno"> 2354</span>  }</div> |
| <div class="line"><a name="l02355"></a><span class="lineno"> 2355</span>  </div> |
| <div class="line"><a name="l02356"></a><span class="lineno"> 2356</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02357"></a><span class="lineno"> 2357</span>  {</div> |
| <div class="line"><a name="l02358"></a><span class="lineno"> 2358</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02359"></a><span class="lineno"> 2359</span>  }</div> |
| <div class="line"><a name="l02360"></a><span class="lineno"> 2360</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02364"></a><span class="lineno"> 2364</span>  </div> |
| <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> |
| <div class="line"><a name="l02366"></a><span class="lineno"> 2366</span>  </div> |
| <div class="line"><a name="l02367"></a><span class="lineno"> 2367</span>  <span class="keywordflow">if</span> (!convLayer)</div> |
| <div class="line"><a name="l02368"></a><span class="lineno"> 2368</span>  {</div> |
| <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> |
| <div class="line"><a name="l02370"></a><span class="lineno"> 2370</span>  }</div> |
| <div class="line"><a name="l02371"></a><span class="lineno"> 2371</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02375"></a><span class="lineno"> 2375</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02379"></a><span class="lineno"> 2379</span>  </div> |
| <div class="line"><a name="l02380"></a><span class="lineno"> 2380</span>  <span class="keywordflow">if</span>(isDynamic)</div> |
| <div class="line"><a name="l02381"></a><span class="lineno"> 2381</span>  {</div> |
| <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> |
| <div class="line"><a name="l02383"></a><span class="lineno"> 2383</span>  </div> |
| <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> |
| <div class="line"><a name="l02385"></a><span class="lineno"> 2385</span>  </div> |
| <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> |
| <div class="line"><a name="l02387"></a><span class="lineno"> 2387</span>  </div> |
| <div class="line"><a name="l02388"></a><span class="lineno"> 2388</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02389"></a><span class="lineno"> 2389</span>  {</div> |
| <div class="line"><a name="l02390"></a><span class="lineno"> 2390</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02391"></a><span class="lineno"> 2391</span>  }</div> |
| <div class="line"><a name="l02392"></a><span class="lineno"> 2392</span>  }</div> |
| <div class="line"><a name="l02393"></a><span class="lineno"> 2393</span>  </div> |
| <div class="line"><a name="l02394"></a><span class="lineno"> 2394</span>  convLayers[index] = convLayer;</div> |
| <div class="line"><a name="l02395"></a><span class="lineno"> 2395</span>  }</div> |
| <div class="line"><a name="l02396"></a><span class="lineno"> 2396</span>  }</div> |
| <div class="line"><a name="l02397"></a><span class="lineno"> 2397</span>  </div> |
| <div class="line"><a name="l02398"></a><span class="lineno"> 2398</span>  <span class="comment">//</span></div> |
| <div class="line"><a name="l02399"></a><span class="lineno"> 2399</span>  <span class="comment">// Set up Concat layer</span></div> |
| <div class="line"><a name="l02400"></a><span class="lineno"> 2400</span>  <span class="comment">//</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02405"></a><span class="lineno"> 2405</span>  {</div> |
| <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> |
| <div class="line"><a name="l02407"></a><span class="lineno"> 2407</span>  {</div> |
| <div class="line"><a name="l02408"></a><span class="lineno"> 2408</span>  <span class="keyword">auto</span> index = group * channelMultiplier + m;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02411"></a><span class="lineno"> 2411</span>  }</div> |
| <div class="line"><a name="l02412"></a><span class="lineno"> 2412</span>  }</div> |
| <div class="line"><a name="l02413"></a><span class="lineno"> 2413</span>  </div> |
| <div class="line"><a name="l02414"></a><span class="lineno"> 2414</span>  isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02417"></a><span class="lineno"> 2417</span>  IsConcatSupported,</div> |
| <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> |
| <div class="line"><a name="l02419"></a><span class="lineno"> 2419</span>  isSupported,</div> |
| <div class="line"><a name="l02420"></a><span class="lineno"> 2420</span>  setBackendConcat,</div> |
| <div class="line"><a name="l02421"></a><span class="lineno"> 2421</span>  std::vector<const TensorInfo*>(numGroups * channelMultiplier, &groupOutputInfo),</div> |
| <div class="line"><a name="l02422"></a><span class="lineno"> 2422</span>  outputInfo,</div> |
| <div class="line"><a name="l02423"></a><span class="lineno"> 2423</span>  concatDescriptor);</div> |
| <div class="line"><a name="l02424"></a><span class="lineno"> 2424</span>  </div> |
| <div class="line"><a name="l02425"></a><span class="lineno"> 2425</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02426"></a><span class="lineno"> 2426</span>  {</div> |
| <div class="line"><a name="l02427"></a><span class="lineno"> 2427</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02428"></a><span class="lineno"> 2428</span>  }</div> |
| <div class="line"><a name="l02429"></a><span class="lineno"> 2429</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02432"></a><span class="lineno"> 2432</span>  <span class="keywordflow">if</span> (!concatLayer)</div> |
| <div class="line"><a name="l02433"></a><span class="lineno"> 2433</span>  {</div> |
| <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> |
| <div class="line"><a name="l02435"></a><span class="lineno"> 2435</span>  }</div> |
| <div class="line"><a name="l02436"></a><span class="lineno"> 2436</span>  </div> |
| <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> |
| <div class="line"><a name="l02438"></a><span class="lineno"> 2438</span>  {</div> |
| <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> |
| <div class="line"><a name="l02440"></a><span class="lineno"> 2440</span>  {</div> |
| <div class="line"><a name="l02441"></a><span class="lineno"> 2441</span>  <span class="keyword">auto</span> index = group * channelMultiplier + m;</div> |
| <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> |
| <div class="line"><a name="l02443"></a><span class="lineno"> 2443</span>  }</div> |
| <div class="line"><a name="l02444"></a><span class="lineno"> 2444</span>  }</div> |
| <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> |
| <div class="line"><a name="l02446"></a><span class="lineno"> 2446</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02449"></a><span class="lineno"> 2449</span> }</div> |
| <div class="line"><a name="l02450"></a><span class="lineno"> 2450</span>  </div> |
| <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> |
| <div class="line"><a name="l02452"></a><span class="lineno"> 2452</span> {</div> |
| <div class="line"><a name="l02453"></a><span class="lineno"> 2453</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertHardSwish()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02456"></a><span class="lineno"> 2456</span>  </div> |
| <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> |
| <div class="line"><a name="l02458"></a><span class="lineno"> 2458</span> }</div> |
| <div class="line"><a name="l02459"></a><span class="lineno"> 2459</span>  </div> |
| <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> |
| <div class="line"><a name="l02461"></a><span class="lineno"> 2461</span> {</div> |
| <div class="line"><a name="l02462"></a><span class="lineno"> 2462</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertInstanceNormalization()"</span>;</div> |
| <div class="line"><a name="l02463"></a><span class="lineno"> 2463</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02466"></a><span class="lineno"> 2466</span>  {</div> |
| <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> |
| <div class="line"><a name="l02468"></a><span class="lineno"> 2468</span>  }</div> |
| <div class="line"><a name="l02469"></a><span class="lineno"> 2469</span>  </div> |
| <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> |
| <div class="line"><a name="l02471"></a><span class="lineno"> 2471</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02472"></a><span class="lineno"> 2472</span>  {</div> |
| <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> |
| <div class="line"><a name="l02474"></a><span class="lineno"> 2474</span>  }</div> |
| <div class="line"><a name="l02475"></a><span class="lineno"> 2475</span>  </div> |
| <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> |
| <div class="line"><a name="l02477"></a><span class="lineno"> 2477</span>  </div> |
| <div class="line"><a name="l02478"></a><span class="lineno"> 2478</span>  <span class="comment">// Determine data type of input tensor</span></div> |
| <div class="line"><a name="l02479"></a><span class="lineno"> 2479</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> inputType;</div> |
| <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> |
| <div class="line"><a name="l02481"></a><span class="lineno"> 2481</span>  {</div> |
| <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> |
| <div class="line"><a name="l02483"></a><span class="lineno"> 2483</span>  }</div> |
| <div class="line"><a name="l02484"></a><span class="lineno"> 2484</span>  </div> |
| <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> |
| <div class="line"><a name="l02486"></a><span class="lineno"> 2486</span>  </div> |
| <div class="line"><a name="l02487"></a><span class="lineno"> 2487</span>  <span class="comment">// Read gamma, beta & epsilon</span></div> |
| <div class="line"><a name="l02488"></a><span class="lineno"> 2488</span>  <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT16)</div> |
| <div class="line"><a name="l02489"></a><span class="lineno"> 2489</span>  {</div> |
| <div class="line"><a name="l02490"></a><span class="lineno"> 2490</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Gamma;</div> |
| <div class="line"><a name="l02491"></a><span class="lineno"> 2491</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Beta;</div> |
| <div class="line"><a name="l02492"></a><span class="lineno"> 2492</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Epsilon;</div> |
| <div class="line"><a name="l02493"></a><span class="lineno"> 2493</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02497"></a><span class="lineno"> 2497</span>  {</div> |
| <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> |
| <div class="line"><a name="l02499"></a><span class="lineno"> 2499</span>  }</div> |
| <div class="line"><a name="l02500"></a><span class="lineno"> 2500</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02504"></a><span class="lineno"> 2504</span>  }</div> |
| <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> |
| <div class="line"><a name="l02506"></a><span class="lineno"> 2506</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02510"></a><span class="lineno"> 2510</span>  {</div> |
| <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> |
| <div class="line"><a name="l02512"></a><span class="lineno"> 2512</span>  }</div> |
| <div class="line"><a name="l02513"></a><span class="lineno"> 2513</span>  }</div> |
| <div class="line"><a name="l02514"></a><span class="lineno"> 2514</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02515"></a><span class="lineno"> 2515</span>  {</div> |
| <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> |
| <div class="line"><a name="l02517"></a><span class="lineno"> 2517</span>  }</div> |
| <div class="line"><a name="l02518"></a><span class="lineno"> 2518</span>  </div> |
| <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> |
| <div class="line"><a name="l02520"></a><span class="lineno"> 2520</span>  </div> |
| <div class="line"><a name="l02521"></a><span class="lineno"> 2521</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02524"></a><span class="lineno"> 2524</span>  {</div> |
| <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> |
| <div class="line"><a name="l02526"></a><span class="lineno"> 2526</span>  IsInstanceNormalizationSupported,</div> |
| <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> |
| <div class="line"><a name="l02528"></a><span class="lineno"> 2528</span>  isSupported,</div> |
| <div class="line"><a name="l02529"></a><span class="lineno"> 2529</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l02531"></a><span class="lineno"> 2531</span>  outputInfo,</div> |
| <div class="line"><a name="l02532"></a><span class="lineno"> 2532</span>  desc);</div> |
| <div class="line"><a name="l02533"></a><span class="lineno"> 2533</span>  };</div> |
| <div class="line"><a name="l02534"></a><span class="lineno"> 2534</span>  </div> |
| <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> |
| <div class="line"><a name="l02536"></a><span class="lineno"> 2536</span>  {</div> |
| <div class="line"><a name="l02537"></a><span class="lineno"> 2537</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02538"></a><span class="lineno"> 2538</span>  }</div> |
| <div class="line"><a name="l02539"></a><span class="lineno"> 2539</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02540"></a><span class="lineno"> 2540</span>  {</div> |
| <div class="line"><a name="l02541"></a><span class="lineno"> 2541</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l02542"></a><span class="lineno"> 2542</span>  }</div> |
| <div class="line"><a name="l02543"></a><span class="lineno"> 2543</span>  </div> |
| <div class="line"><a name="l02544"></a><span class="lineno"> 2544</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02545"></a><span class="lineno"> 2545</span>  {</div> |
| <div class="line"><a name="l02546"></a><span class="lineno"> 2546</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02547"></a><span class="lineno"> 2547</span>  }</div> |
| <div class="line"><a name="l02548"></a><span class="lineno"> 2548</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02552"></a><span class="lineno"> 2552</span>  </div> |
| <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> |
| <div class="line"><a name="l02554"></a><span class="lineno"> 2554</span> }</div> |
| <div class="line"><a name="l02555"></a><span class="lineno"> 2555</span>  </div> |
| <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> |
| <div class="line"><a name="l02557"></a><span class="lineno"> 2557</span> {</div> |
| <div class="line"><a name="l02558"></a><span class="lineno"> 2558</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertL2Normalization()"</span>;</div> |
| <div class="line"><a name="l02559"></a><span class="lineno"> 2559</span>  </div> |
| <div class="line"><a name="l02560"></a><span class="lineno"> 2560</span>  <span class="keywordflow">if</span> (operation.inputs.size() != 1)</div> |
| <div class="line"><a name="l02561"></a><span class="lineno"> 2561</span>  {</div> |
| <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> |
| <div class="line"><a name="l02563"></a><span class="lineno"> 2563</span>  }</div> |
| <div class="line"><a name="l02564"></a><span class="lineno"> 2564</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02567"></a><span class="lineno"> 2567</span>  {</div> |
| <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> |
| <div class="line"><a name="l02569"></a><span class="lineno"> 2569</span>  }</div> |
| <div class="line"><a name="l02570"></a><span class="lineno"> 2570</span>  </div> |
| <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> |
| <div class="line"><a name="l02572"></a><span class="lineno"> 2572</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02573"></a><span class="lineno"> 2573</span>  {</div> |
| <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> |
| <div class="line"><a name="l02575"></a><span class="lineno"> 2575</span>  }</div> |
| <div class="line"><a name="l02576"></a><span class="lineno"> 2576</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02579"></a><span class="lineno"> 2579</span>  </div> |
| <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> |
| <div class="line"><a name="l02581"></a><span class="lineno"> 2581</span>  {</div> |
| <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> |
| <div class="line"><a name="l02583"></a><span class="lineno"> 2583</span>  }</div> |
| <div class="line"><a name="l02584"></a><span class="lineno"> 2584</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02587"></a><span class="lineno"> 2587</span>  </div> |
| <div class="line"><a name="l02588"></a><span class="lineno"> 2588</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02591"></a><span class="lineno"> 2591</span>  {</div> |
| <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> |
| <div class="line"><a name="l02593"></a><span class="lineno"> 2593</span>  IsL2NormalizationSupported,</div> |
| <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> |
| <div class="line"><a name="l02595"></a><span class="lineno"> 2595</span>  isSupported,</div> |
| <div class="line"><a name="l02596"></a><span class="lineno"> 2596</span>  setBackend,</div> |
| <div class="line"><a name="l02597"></a><span class="lineno"> 2597</span>  inputInfo,</div> |
| <div class="line"><a name="l02598"></a><span class="lineno"> 2598</span>  outputInfo,</div> |
| <div class="line"><a name="l02599"></a><span class="lineno"> 2599</span>  desc);</div> |
| <div class="line"><a name="l02600"></a><span class="lineno"> 2600</span>  };</div> |
| <div class="line"><a name="l02601"></a><span class="lineno"> 2601</span>  </div> |
| <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> |
| <div class="line"><a name="l02603"></a><span class="lineno"> 2603</span>  {</div> |
| <div class="line"><a name="l02604"></a><span class="lineno"> 2604</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l02605"></a><span class="lineno"> 2605</span>  }</div> |
| <div class="line"><a name="l02606"></a><span class="lineno"> 2606</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02607"></a><span class="lineno"> 2607</span>  {</div> |
| <div class="line"><a name="l02608"></a><span class="lineno"> 2608</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02609"></a><span class="lineno"> 2609</span>  }</div> |
| <div class="line"><a name="l02610"></a><span class="lineno"> 2610</span>  </div> |
| <div class="line"><a name="l02611"></a><span class="lineno"> 2611</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02612"></a><span class="lineno"> 2612</span>  {</div> |
| <div class="line"><a name="l02613"></a><span class="lineno"> 2613</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02614"></a><span class="lineno"> 2614</span>  }</div> |
| <div class="line"><a name="l02615"></a><span class="lineno"> 2615</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02618"></a><span class="lineno"> 2618</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l02620"></a><span class="lineno"> 2620</span>  </div> |
| <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> |
| <div class="line"><a name="l02622"></a><span class="lineno"> 2622</span> }</div> |
| <div class="line"><a name="l02623"></a><span class="lineno"> 2623</span>  </div> |
| <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> |
| <div class="line"><a name="l02625"></a><span class="lineno"> 2625</span> {</div> |
| <div class="line"><a name="l02626"></a><span class="lineno"> 2626</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertL2Pool2d()"</span>;</div> |
| <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> |
| <div class="line"><a name="l02628"></a><span class="lineno"> 2628</span> }</div> |
| <div class="line"><a name="l02629"></a><span class="lineno"> 2629</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02633"></a><span class="lineno"> 2633</span> {</div> |
| <div class="line"><a name="l02634"></a><span class="lineno"> 2634</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLocalResponseNormalization()"</span>;</div> |
| <div class="line"><a name="l02635"></a><span class="lineno"> 2635</span>  </div> |
| <div class="line"><a name="l02636"></a><span class="lineno"> 2636</span>  <span class="keywordflow">if</span> (operation.inputs.size() != 5)</div> |
| <div class="line"><a name="l02637"></a><span class="lineno"> 2637</span>  {</div> |
| <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> |
| <div class="line"><a name="l02639"></a><span class="lineno"> 2639</span>  }</div> |
| <div class="line"><a name="l02640"></a><span class="lineno"> 2640</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02643"></a><span class="lineno"> 2643</span>  {</div> |
| <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> |
| <div class="line"><a name="l02645"></a><span class="lineno"> 2645</span>  }</div> |
| <div class="line"><a name="l02646"></a><span class="lineno"> 2646</span>  </div> |
| <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> |
| <div class="line"><a name="l02648"></a><span class="lineno"> 2648</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02649"></a><span class="lineno"> 2649</span>  {</div> |
| <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> |
| <div class="line"><a name="l02651"></a><span class="lineno"> 2651</span>  }</div> |
| <div class="line"><a name="l02652"></a><span class="lineno"> 2652</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02655"></a><span class="lineno"> 2655</span>  </div> |
| <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> |
| <div class="line"><a name="l02657"></a><span class="lineno"> 2657</span>  {</div> |
| <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> |
| <div class="line"><a name="l02659"></a><span class="lineno"> 2659</span>  }</div> |
| <div class="line"><a name="l02660"></a><span class="lineno"> 2660</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02665"></a><span class="lineno"> 2665</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02671"></a><span class="lineno"> 2671</span>  {</div> |
| <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> |
| <div class="line"><a name="l02673"></a><span class="lineno"> 2673</span>  }</div> |
| <div class="line"><a name="l02674"></a><span class="lineno"> 2674</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02678"></a><span class="lineno"> 2678</span>  </div> |
| <div class="line"><a name="l02679"></a><span class="lineno"> 2679</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02682"></a><span class="lineno"> 2682</span>  {</div> |
| <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> |
| <div class="line"><a name="l02684"></a><span class="lineno"> 2684</span>  IsNormalizationSupported,</div> |
| <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> |
| <div class="line"><a name="l02686"></a><span class="lineno"> 2686</span>  isSupported,</div> |
| <div class="line"><a name="l02687"></a><span class="lineno"> 2687</span>  setBackend,</div> |
| <div class="line"><a name="l02688"></a><span class="lineno"> 2688</span>  inputInfo,</div> |
| <div class="line"><a name="l02689"></a><span class="lineno"> 2689</span>  outputInfo,</div> |
| <div class="line"><a name="l02690"></a><span class="lineno"> 2690</span>  descriptor);</div> |
| <div class="line"><a name="l02691"></a><span class="lineno"> 2691</span>  };</div> |
| <div class="line"><a name="l02692"></a><span class="lineno"> 2692</span>  </div> |
| <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> |
| <div class="line"><a name="l02694"></a><span class="lineno"> 2694</span>  {</div> |
| <div class="line"><a name="l02695"></a><span class="lineno"> 2695</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l02696"></a><span class="lineno"> 2696</span>  }</div> |
| <div class="line"><a name="l02697"></a><span class="lineno"> 2697</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02698"></a><span class="lineno"> 2698</span>  {</div> |
| <div class="line"><a name="l02699"></a><span class="lineno"> 2699</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02700"></a><span class="lineno"> 2700</span>  }</div> |
| <div class="line"><a name="l02701"></a><span class="lineno"> 2701</span>  </div> |
| <div class="line"><a name="l02702"></a><span class="lineno"> 2702</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02703"></a><span class="lineno"> 2703</span>  {</div> |
| <div class="line"><a name="l02704"></a><span class="lineno"> 2704</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02705"></a><span class="lineno"> 2705</span>  }</div> |
| <div class="line"><a name="l02706"></a><span class="lineno"> 2706</span>  </div> |
| <div class="line"><a name="l02707"></a><span class="lineno"> 2707</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02710"></a><span class="lineno"> 2710</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l02712"></a><span class="lineno"> 2712</span>  </div> |
| <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> |
| <div class="line"><a name="l02714"></a><span class="lineno"> 2714</span> }</div> |
| <div class="line"><a name="l02715"></a><span class="lineno"> 2715</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02719"></a><span class="lineno"> 2719</span>  <a class="code" href="namespacearmnn.html#a2da4db0140d1a6dc69c9c82e9ef5379e">armnn::LogicalBinaryOperation</a> logicalOperation)</div> |
| <div class="line"><a name="l02720"></a><span class="lineno"> 2720</span> {</div> |
| <div class="line"><a name="l02721"></a><span class="lineno"> 2721</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLogicalBinary()"</span>;</div> |
| <div class="line"><a name="l02722"></a><span class="lineno"> 2722</span>  VLOG(DRIVER) << <span class="stringliteral">"ConvertLogicalBinary()"</span>;</div> |
| <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> |
| <div class="line"><a name="l02724"></a><span class="lineno"> 2724</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02727"></a><span class="lineno"> 2727</span>  </div> |
| <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> |
| <div class="line"><a name="l02729"></a><span class="lineno"> 2729</span>  {</div> |
| <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> |
| <div class="line"><a name="l02731"></a><span class="lineno"> 2731</span>  }</div> |
| <div class="line"><a name="l02732"></a><span class="lineno"> 2732</span>  </div> |
| <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> |
| <div class="line"><a name="l02734"></a><span class="lineno"> 2734</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02735"></a><span class="lineno"> 2735</span>  {</div> |
| <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> |
| <div class="line"><a name="l02737"></a><span class="lineno"> 2737</span>  }</div> |
| <div class="line"><a name="l02738"></a><span class="lineno"> 2738</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02742"></a><span class="lineno"> 2742</span>  </div> |
| <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> |
| <div class="line"><a name="l02744"></a><span class="lineno"> 2744</span>  </div> |
| <div class="line"><a name="l02745"></a><span class="lineno"> 2745</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02748"></a><span class="lineno"> 2748</span>  {</div> |
| <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> |
| <div class="line"><a name="l02750"></a><span class="lineno"> 2750</span>  IsLogicalBinarySupported,</div> |
| <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> |
| <div class="line"><a name="l02752"></a><span class="lineno"> 2752</span>  isSupported,</div> |
| <div class="line"><a name="l02753"></a><span class="lineno"> 2753</span>  setBackend,</div> |
| <div class="line"><a name="l02754"></a><span class="lineno"> 2754</span>  inputInfo0,</div> |
| <div class="line"><a name="l02755"></a><span class="lineno"> 2755</span>  inputInfo1,</div> |
| <div class="line"><a name="l02756"></a><span class="lineno"> 2756</span>  outputInfo,</div> |
| <div class="line"><a name="l02757"></a><span class="lineno"> 2757</span>  descriptor);</div> |
| <div class="line"><a name="l02758"></a><span class="lineno"> 2758</span>  };</div> |
| <div class="line"><a name="l02759"></a><span class="lineno"> 2759</span>  </div> |
| <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> |
| <div class="line"><a name="l02761"></a><span class="lineno"> 2761</span>  {</div> |
| <div class="line"><a name="l02762"></a><span class="lineno"> 2762</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l02763"></a><span class="lineno"> 2763</span>  }</div> |
| <div class="line"><a name="l02764"></a><span class="lineno"> 2764</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02765"></a><span class="lineno"> 2765</span>  {</div> |
| <div class="line"><a name="l02766"></a><span class="lineno"> 2766</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02767"></a><span class="lineno"> 2767</span>  }</div> |
| <div class="line"><a name="l02768"></a><span class="lineno"> 2768</span>  </div> |
| <div class="line"><a name="l02769"></a><span class="lineno"> 2769</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02770"></a><span class="lineno"> 2770</span>  {</div> |
| <div class="line"><a name="l02771"></a><span class="lineno"> 2771</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02772"></a><span class="lineno"> 2772</span>  }</div> |
| <div class="line"><a name="l02773"></a><span class="lineno"> 2773</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02776"></a><span class="lineno"> 2776</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <div class="line"><a name="l02777"></a><span class="lineno"> 2777</span>  </div> |
| <div class="line"><a name="l02778"></a><span class="lineno"> 2778</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input0, input1, layer, data);</div> |
| <div class="line"><a name="l02779"></a><span class="lineno"> 2779</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| <div class="line"><a name="l02780"></a><span class="lineno"> 2780</span>  {</div> |
| <div class="line"><a name="l02781"></a><span class="lineno"> 2781</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02782"></a><span class="lineno"> 2782</span>  }</div> |
| <div class="line"><a name="l02783"></a><span class="lineno"> 2783</span>  </div> |
| <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> |
| <div class="line"><a name="l02785"></a><span class="lineno"> 2785</span> }</div> |
| <div class="line"><a name="l02786"></a><span class="lineno"> 2786</span>  </div> |
| <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> |
| <div class="line"><a name="l02788"></a><span class="lineno"> 2788</span> {</div> |
| <div class="line"><a name="l02789"></a><span class="lineno"> 2789</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLogistic()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02792"></a><span class="lineno"> 2792</span>  </div> |
| <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> |
| <div class="line"><a name="l02794"></a><span class="lineno"> 2794</span> }</div> |
| <div class="line"><a name="l02795"></a><span class="lineno"> 2795</span>  </div> |
| <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> |
| <div class="line"><a name="l02797"></a><span class="lineno"> 2797</span> {</div> |
| <div class="line"><a name="l02798"></a><span class="lineno"> 2798</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLogSoftmax()"</span>;</div> |
| <div class="line"><a name="l02799"></a><span class="lineno"> 2799</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02802"></a><span class="lineno"> 2802</span>  {</div> |
| <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> |
| <div class="line"><a name="l02804"></a><span class="lineno"> 2804</span>  }</div> |
| <div class="line"><a name="l02805"></a><span class="lineno"> 2805</span>  </div> |
| <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> |
| <div class="line"><a name="l02807"></a><span class="lineno"> 2807</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l02808"></a><span class="lineno"> 2808</span>  {</div> |
| <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> |
| <div class="line"><a name="l02810"></a><span class="lineno"> 2810</span>  }</div> |
| <div class="line"><a name="l02811"></a><span class="lineno"> 2811</span>  </div> |
| <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> |
| <div class="line"><a name="l02813"></a><span class="lineno"> 2813</span>  </div> |
| <div class="line"><a name="l02814"></a><span class="lineno"> 2814</span>  <span class="comment">// Determine data type of input tensor</span></div> |
| <div class="line"><a name="l02815"></a><span class="lineno"> 2815</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> inputType;</div> |
| <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> |
| <div class="line"><a name="l02817"></a><span class="lineno"> 2817</span>  {</div> |
| <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> |
| <div class="line"><a name="l02819"></a><span class="lineno"> 2819</span>  }</div> |
| <div class="line"><a name="l02820"></a><span class="lineno"> 2820</span>  </div> |
| <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> |
| <div class="line"><a name="l02822"></a><span class="lineno"> 2822</span>  </div> |
| <div class="line"><a name="l02823"></a><span class="lineno"> 2823</span>  <span class="comment">// Read beta</span></div> |
| <div class="line"><a name="l02824"></a><span class="lineno"> 2824</span>  <span class="keywordflow">if</span> (inputType == OperandType::TENSOR_FLOAT16)</div> |
| <div class="line"><a name="l02825"></a><span class="lineno"> 2825</span>  {</div> |
| <div class="line"><a name="l02826"></a><span class="lineno"> 2826</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> fp16Beta;</div> |
| <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> |
| <div class="line"><a name="l02828"></a><span class="lineno"> 2828</span>  {</div> |
| <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> |
| <div class="line"><a name="l02830"></a><span class="lineno"> 2830</span>  }</div> |
| <div class="line"><a name="l02831"></a><span class="lineno"> 2831</span>  </div> |
| <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> |
| <div class="line"><a name="l02833"></a><span class="lineno"> 2833</span>  }</div> |
| <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> |
| <div class="line"><a name="l02835"></a><span class="lineno"> 2835</span>  {</div> |
| <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> |
| <div class="line"><a name="l02837"></a><span class="lineno"> 2837</span>  {</div> |
| <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> |
| <div class="line"><a name="l02839"></a><span class="lineno"> 2839</span>  }</div> |
| <div class="line"><a name="l02840"></a><span class="lineno"> 2840</span>  }</div> |
| <div class="line"><a name="l02841"></a><span class="lineno"> 2841</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02842"></a><span class="lineno"> 2842</span>  {</div> |
| <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> |
| <div class="line"><a name="l02844"></a><span class="lineno"> 2844</span>  }</div> |
| <div class="line"><a name="l02845"></a><span class="lineno"> 2845</span>  </div> |
| <div class="line"><a name="l02846"></a><span class="lineno"> 2846</span>  <span class="comment">// Read axis</span></div> |
| <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> |
| <div class="line"><a name="l02848"></a><span class="lineno"> 2848</span>  {</div> |
| <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> |
| <div class="line"><a name="l02850"></a><span class="lineno"> 2850</span>  }</div> |
| <div class="line"><a name="l02851"></a><span class="lineno"> 2851</span>  </div> |
| <div class="line"><a name="l02852"></a><span class="lineno"> 2852</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l02855"></a><span class="lineno"> 2855</span>  {</div> |
| <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> |
| <div class="line"><a name="l02857"></a><span class="lineno"> 2857</span>  IsLogSoftmaxSupported,</div> |
| <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> |
| <div class="line"><a name="l02859"></a><span class="lineno"> 2859</span>  isSupported,</div> |
| <div class="line"><a name="l02860"></a><span class="lineno"> 2860</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l02862"></a><span class="lineno"> 2862</span>  outputInfo,</div> |
| <div class="line"><a name="l02863"></a><span class="lineno"> 2863</span>  descriptor);</div> |
| <div class="line"><a name="l02864"></a><span class="lineno"> 2864</span>  };</div> |
| <div class="line"><a name="l02865"></a><span class="lineno"> 2865</span>  </div> |
| <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> |
| <div class="line"><a name="l02867"></a><span class="lineno"> 2867</span>  {</div> |
| <div class="line"><a name="l02868"></a><span class="lineno"> 2868</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l02869"></a><span class="lineno"> 2869</span>  }</div> |
| <div class="line"><a name="l02870"></a><span class="lineno"> 2870</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l02871"></a><span class="lineno"> 2871</span>  {</div> |
| <div class="line"><a name="l02872"></a><span class="lineno"> 2872</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l02873"></a><span class="lineno"> 2873</span>  }</div> |
| <div class="line"><a name="l02874"></a><span class="lineno"> 2874</span>  </div> |
| <div class="line"><a name="l02875"></a><span class="lineno"> 2875</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l02876"></a><span class="lineno"> 2876</span>  {</div> |
| <div class="line"><a name="l02877"></a><span class="lineno"> 2877</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l02878"></a><span class="lineno"> 2878</span>  }</div> |
| <div class="line"><a name="l02879"></a><span class="lineno"> 2879</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l02882"></a><span class="lineno"> 2882</span>  <span class="keywordflow">if</span> (!layer)</div> |
| <div class="line"><a name="l02883"></a><span class="lineno"> 2883</span>  {</div> |
| <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> |
| <div class="line"><a name="l02885"></a><span class="lineno"> 2885</span>  }</div> |
| <div class="line"><a name="l02886"></a><span class="lineno"> 2886</span>  </div> |
| <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> |
| <div class="line"><a name="l02888"></a><span class="lineno"> 2888</span>  </div> |
| <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> |
| <div class="line"><a name="l02890"></a><span class="lineno"> 2890</span> }</div> |
| <div class="line"><a name="l02891"></a><span class="lineno"> 2891</span>  </div> |
| <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> |
| <div class="line"><a name="l02893"></a><span class="lineno"> 2893</span> {</div> |
| <div class="line"><a name="l02894"></a><span class="lineno"> 2894</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertLstm()"</span>;</div> |
| <div class="line"><a name="l02895"></a><span class="lineno"> 2895</span>  </div> |
| <div class="line"><a name="l02896"></a><span class="lineno"> 2896</span>  <span class="comment">// Inputs:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02901"></a><span class="lineno"> 2901</span>  {</div> |
| <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> |
| <div class="line"><a name="l02903"></a><span class="lineno"> 2903</span>  }</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02907"></a><span class="lineno"> 2907</span>  {</div> |
| <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> |
| <div class="line"><a name="l02909"></a><span class="lineno"> 2909</span>  }</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02913"></a><span class="lineno"> 2913</span>  {</div> |
| <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> |
| <div class="line"><a name="l02915"></a><span class="lineno"> 2915</span>  }</div> |
| <div class="line"><a name="l02916"></a><span class="lineno"> 2916</span>  </div> |
| <div class="line"><a name="l02917"></a><span class="lineno"> 2917</span>  <span class="comment">// Get the mandatory input tensors:</span></div> |
| <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> |
| <div class="line"><a name="l02919"></a><span class="lineno"> 2919</span>  <span class="comment">// [num_units, input_size].</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02923"></a><span class="lineno"> 2923</span>  <span class="comment">// [num_units, input_size].</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02927"></a><span class="lineno"> 2927</span>  <span class="comment">// [num_units, input_size].</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02931"></a><span class="lineno"> 2931</span>  <span class="comment">// [num_units, output_size].</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02935"></a><span class="lineno"> 2935</span>  <span class="comment">// [num_units, output_size].</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02939"></a><span class="lineno"> 2939</span>  <span class="comment">// [num_units, output_size].</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02951"></a><span class="lineno"> 2951</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02961"></a><span class="lineno"> 2961</span>  {</div> |
| <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> |
| <div class="line"><a name="l02963"></a><span class="lineno"> 2963</span>  }</div> |
| <div class="line"><a name="l02964"></a><span class="lineno"> 2964</span>  </div> |
| <div class="line"><a name="l02965"></a><span class="lineno"> 2965</span>  <span class="comment">// Get the optional input tensors:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l02986"></a><span class="lineno"> 2986</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l02987"></a><span class="lineno"> 2987</span>  12,</div> |
| <div class="line"><a name="l02988"></a><span class="lineno"> 2988</span>  model,</div> |
| <div class="line"><a name="l02989"></a><span class="lineno"> 2989</span>  data,</div> |
| <div class="line"><a name="l02990"></a><span class="lineno"> 2990</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l02991"></a><span class="lineno"> 2991</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l02992"></a><span class="lineno"> 2992</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l02993"></a><span class="lineno"> 2993</span>  </div> |
| <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> |
| <div class="line"><a name="l02995"></a><span class="lineno"> 2995</span>  <span class="comment">// [output_size, num_units].</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03000"></a><span class="lineno"> 3000</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03001"></a><span class="lineno"> 3001</span>  17,</div> |
| <div class="line"><a name="l03002"></a><span class="lineno"> 3002</span>  model,</div> |
| <div class="line"><a name="l03003"></a><span class="lineno"> 3003</span>  data,</div> |
| <div class="line"><a name="l03004"></a><span class="lineno"> 3004</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03005"></a><span class="lineno"> 3005</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03006"></a><span class="lineno"> 3006</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03007"></a><span class="lineno"> 3007</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03016"></a><span class="lineno"> 3016</span>  {</div> |
| <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> |
| <div class="line"><a name="l03018"></a><span class="lineno"> 3018</span>  }</div> |
| <div class="line"><a name="l03019"></a><span class="lineno"> 3019</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03027"></a><span class="lineno"> 3027</span>  ActivationFn activation = ActivationFn::kActivationNone;</div> |
| <div class="line"><a name="l03028"></a><span class="lineno"> 3028</span>  <span class="keywordtype">float</span> cellClip;</div> |
| <div class="line"><a name="l03029"></a><span class="lineno"> 3029</span>  <span class="keywordtype">float</span> projClip;</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03033"></a><span class="lineno"> 3033</span>  {</div> |
| <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> |
| <div class="line"><a name="l03035"></a><span class="lineno"> 3035</span>  }</div> |
| <div class="line"><a name="l03036"></a><span class="lineno"> 3036</span>  </div> |
| <div class="line"><a name="l03037"></a><span class="lineno"> 3037</span>  <span class="comment">// Get the normalization tensors</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03042"></a><span class="lineno"> 3042</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03046"></a><span class="lineno"> 3046</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03047"></a><span class="lineno"> 3047</span>  24,</div> |
| <div class="line"><a name="l03048"></a><span class="lineno"> 3048</span>  model,</div> |
| <div class="line"><a name="l03049"></a><span class="lineno"> 3049</span>  data,</div> |
| <div class="line"><a name="l03050"></a><span class="lineno"> 3050</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03051"></a><span class="lineno"> 3051</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03052"></a><span class="lineno"> 3052</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03053"></a><span class="lineno"> 3053</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03057"></a><span class="lineno"> 3057</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03058"></a><span class="lineno"> 3058</span>  25,</div> |
| <div class="line"><a name="l03059"></a><span class="lineno"> 3059</span>  model,</div> |
| <div class="line"><a name="l03060"></a><span class="lineno"> 3060</span>  data,</div> |
| <div class="line"><a name="l03061"></a><span class="lineno"> 3061</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03062"></a><span class="lineno"> 3062</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03063"></a><span class="lineno"> 3063</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03064"></a><span class="lineno"> 3064</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03068"></a><span class="lineno"> 3068</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03069"></a><span class="lineno"> 3069</span>  26,</div> |
| <div class="line"><a name="l03070"></a><span class="lineno"> 3070</span>  model,</div> |
| <div class="line"><a name="l03071"></a><span class="lineno"> 3071</span>  data,</div> |
| <div class="line"><a name="l03072"></a><span class="lineno"> 3072</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03073"></a><span class="lineno"> 3073</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03074"></a><span class="lineno"> 3074</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03075"></a><span class="lineno"> 3075</span>  </div> |
| <div class="line"><a name="l03076"></a><span class="lineno"> 3076</span>  <span class="comment">// Outputs:</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03080"></a><span class="lineno"> 3080</span>  <span class="keywordflow">if</span> (!scratchBuffer)</div> |
| <div class="line"><a name="l03081"></a><span class="lineno"> 3081</span>  {</div> |
| <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> |
| <div class="line"><a name="l03083"></a><span class="lineno"> 3083</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03086"></a><span class="lineno"> 3086</span>  <span class="keywordflow">if</span> (!outputStateOut)</div> |
| <div class="line"><a name="l03087"></a><span class="lineno"> 3087</span>  {</div> |
| <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> |
| <div class="line"><a name="l03089"></a><span class="lineno"> 3089</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03092"></a><span class="lineno"> 3092</span>  <span class="keywordflow">if</span> (!cellStateOut)</div> |
| <div class="line"><a name="l03093"></a><span class="lineno"> 3093</span>  {</div> |
| <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> |
| <div class="line"><a name="l03095"></a><span class="lineno"> 3095</span>  }</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03099"></a><span class="lineno"> 3099</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l03100"></a><span class="lineno"> 3100</span>  {</div> |
| <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> |
| <div class="line"><a name="l03102"></a><span class="lineno"> 3102</span>  }</div> |
| <div class="line"><a name="l03103"></a><span class="lineno"> 3103</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03127"></a><span class="lineno"> 3127</span>  </div> |
| <div class="line"><a name="l03128"></a><span class="lineno"> 3128</span>  <span class="comment">// set the layer descriptor</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03143"></a><span class="lineno"> 3143</span>  </div> |
| <div class="line"><a name="l03144"></a><span class="lineno"> 3144</span>  <span class="comment">// validate the optional input groups</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03149"></a><span class="lineno"> 3149</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03152"></a><span class="lineno"> 3152</span>  }</div> |
| <div class="line"><a name="l03153"></a><span class="lineno"> 3153</span>  </div> |
| <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> |
| <div class="line"><a name="l03155"></a><span class="lineno"> 3155</span>  {</div> |
| <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> |
| <div class="line"><a name="l03157"></a><span class="lineno"> 3157</span>  }</div> |
| <div class="line"><a name="l03158"></a><span class="lineno"> 3158</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03163"></a><span class="lineno"> 3163</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03166"></a><span class="lineno"> 3166</span>  }</div> |
| <div class="line"><a name="l03167"></a><span class="lineno"> 3167</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03173"></a><span class="lineno"> 3173</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03176"></a><span class="lineno"> 3176</span>  }</div> |
| <div class="line"><a name="l03177"></a><span class="lineno"> 3177</span>  </div> |
| <div class="line"><a name="l03178"></a><span class="lineno"> 3178</span>  <span class="comment">// Check if the layer is supported</span></div> |
| <div class="line"><a name="l03179"></a><span class="lineno"> 3179</span>  <span class="comment">// Inputs</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03183"></a><span class="lineno"> 3183</span>  </div> |
| <div class="line"><a name="l03184"></a><span class="lineno"> 3184</span>  <span class="comment">// Outputs</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03189"></a><span class="lineno"> 3189</span>  </div> |
| <div class="line"><a name="l03190"></a><span class="lineno"> 3190</span>  <span class="comment">// Basic parameters</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03201"></a><span class="lineno"> 3201</span>  </div> |
| <div class="line"><a name="l03202"></a><span class="lineno"> 3202</span>  <span class="comment">// Optional parameters</span></div> |
| <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> |
| <div class="line"><a name="l03204"></a><span class="lineno"> 3204</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03208"></a><span class="lineno"> 3208</span>  {</div> |
| <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> |
| <div class="line"><a name="l03210"></a><span class="lineno"> 3210</span>  }</div> |
| <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> |
| <div class="line"><a name="l03212"></a><span class="lineno"> 3212</span>  }</div> |
| <div class="line"><a name="l03213"></a><span class="lineno"> 3213</span>  </div> |
| <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> |
| <div class="line"><a name="l03215"></a><span class="lineno"> 3215</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03218"></a><span class="lineno"> 3218</span>  {</div> |
| <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> |
| <div class="line"><a name="l03220"></a><span class="lineno"> 3220</span>  }</div> |
| <div class="line"><a name="l03221"></a><span class="lineno"> 3221</span>  }</div> |
| <div class="line"><a name="l03222"></a><span class="lineno"> 3222</span>  </div> |
| <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> |
| <div class="line"><a name="l03224"></a><span class="lineno"> 3224</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03227"></a><span class="lineno"> 3227</span>  }</div> |
| <div class="line"><a name="l03228"></a><span class="lineno"> 3228</span>  </div> |
| <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> |
| <div class="line"><a name="l03230"></a><span class="lineno"> 3230</span>  {</div> |
| <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> |
| <div class="line"><a name="l03232"></a><span class="lineno"> 3232</span>  {</div> |
| <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> |
| <div class="line"><a name="l03234"></a><span class="lineno"> 3234</span>  }</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03238"></a><span class="lineno"> 3238</span>  }</div> |
| <div class="line"><a name="l03239"></a><span class="lineno"> 3239</span>  </div> |
| <div class="line"><a name="l03240"></a><span class="lineno"> 3240</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03243"></a><span class="lineno"> 3243</span>  {</div> |
| <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> |
| <div class="line"><a name="l03245"></a><span class="lineno"> 3245</span>  IsLstmSupported,</div> |
| <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> |
| <div class="line"><a name="l03247"></a><span class="lineno"> 3247</span>  isSupported,</div> |
| <div class="line"><a name="l03248"></a><span class="lineno"> 3248</span>  setBackend,</div> |
| <div class="line"><a name="l03249"></a><span class="lineno"> 3249</span>  inputInfo,</div> |
| <div class="line"><a name="l03250"></a><span class="lineno"> 3250</span>  outputStateInInfo,</div> |
| <div class="line"><a name="l03251"></a><span class="lineno"> 3251</span>  cellStateInInfo,</div> |
| <div class="line"><a name="l03252"></a><span class="lineno"> 3252</span>  scratchBufferInfo,</div> |
| <div class="line"><a name="l03253"></a><span class="lineno"> 3253</span>  outputStateOutInfo,</div> |
| <div class="line"><a name="l03254"></a><span class="lineno"> 3254</span>  cellStateOutInfo,</div> |
| <div class="line"><a name="l03255"></a><span class="lineno"> 3255</span>  outputInfo,</div> |
| <div class="line"><a name="l03256"></a><span class="lineno"> 3256</span>  desc,</div> |
| <div class="line"><a name="l03257"></a><span class="lineno"> 3257</span>  paramsInfo);</div> |
| <div class="line"><a name="l03258"></a><span class="lineno"> 3258</span>  };</div> |
| <div class="line"><a name="l03259"></a><span class="lineno"> 3259</span>  </div> |
| <div class="line"><a name="l03260"></a><span class="lineno"> 3260</span>  <span class="keywordtype">bool</span> isDynamic = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l03262"></a><span class="lineno"> 3262</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(scratchBufferInfo) &&</div> |
| <div class="line"><a name="l03263"></a><span class="lineno"> 3263</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(cellStateOutInfo) &&</div> |
| <div class="line"><a name="l03264"></a><span class="lineno"> 3264</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| <div class="line"><a name="l03265"></a><span class="lineno"> 3265</span>  {</div> |
| <div class="line"><a name="l03266"></a><span class="lineno"> 3266</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l03267"></a><span class="lineno"> 3267</span>  }</div> |
| <div class="line"><a name="l03268"></a><span class="lineno"> 3268</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03269"></a><span class="lineno"> 3269</span>  {</div> |
| <div class="line"><a name="l03270"></a><span class="lineno"> 3270</span>  isDynamic = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l03271"></a><span class="lineno"> 3271</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l03272"></a><span class="lineno"> 3272</span>  }</div> |
| <div class="line"><a name="l03273"></a><span class="lineno"> 3273</span>  </div> |
| <div class="line"><a name="l03274"></a><span class="lineno"> 3274</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l03275"></a><span class="lineno"> 3275</span>  {</div> |
| <div class="line"><a name="l03276"></a><span class="lineno"> 3276</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03277"></a><span class="lineno"> 3277</span>  }</div> |
| <div class="line"><a name="l03278"></a><span class="lineno"> 3278</span>  </div> |
| <div class="line"><a name="l03279"></a><span class="lineno"> 3279</span>  <span class="comment">// Add the layer</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03282"></a><span class="lineno"> 3282</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03286"></a><span class="lineno"> 3286</span>  </div> |
| <div class="line"><a name="l03287"></a><span class="lineno"> 3287</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| <div class="line"><a name="l03288"></a><span class="lineno"> 3288</span>  {</div> |
| <div class="line"><a name="l03289"></a><span class="lineno"> 3289</span>  <span class="keywordflow">return</span> (</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03294"></a><span class="lineno"> 3294</span>  }</div> |
| <div class="line"><a name="l03295"></a><span class="lineno"> 3295</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03296"></a><span class="lineno"> 3296</span>  {</div> |
| <div class="line"><a name="l03297"></a><span class="lineno"> 3297</span>  <span class="keywordflow">return</span> (</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03301"></a><span class="lineno"> 3301</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| <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> |
| <div class="line"><a name="l03303"></a><span class="lineno"> 3303</span>  }</div> |
| <div class="line"><a name="l03304"></a><span class="lineno"> 3304</span>  </div> |
| <div class="line"><a name="l03305"></a><span class="lineno"> 3305</span> }</div> |
| <div class="line"><a name="l03306"></a><span class="lineno"> 3306</span>  </div> |
| <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> |
| <div class="line"><a name="l03308"></a><span class="lineno"> 3308</span> {</div> |
| <div class="line"><a name="l03309"></a><span class="lineno"> 3309</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertMaxPool2d()"</span>;</div> |
| <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> |
| <div class="line"><a name="l03311"></a><span class="lineno"> 3311</span> }</div> |
| <div class="line"><a name="l03312"></a><span class="lineno"> 3312</span>  </div> |
| <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> |
| <div class="line"><a name="l03314"></a><span class="lineno"> 3314</span> {</div> |
| <div class="line"><a name="l03315"></a><span class="lineno"> 3315</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertMean()"</span>;</div> |
| <div class="line"><a name="l03316"></a><span class="lineno"> 3316</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03319"></a><span class="lineno"> 3319</span>  {</div> |
| <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> |
| <div class="line"><a name="l03321"></a><span class="lineno"> 3321</span>  }</div> |
| <div class="line"><a name="l03322"></a><span class="lineno"> 3322</span>  </div> |
| <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> |
| <div class="line"><a name="l03324"></a><span class="lineno"> 3324</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l03325"></a><span class="lineno"> 3325</span>  {</div> |
| <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> |
| <div class="line"><a name="l03327"></a><span class="lineno"> 3327</span>  }</div> |
| <div class="line"><a name="l03328"></a><span class="lineno"> 3328</span>  </div> |
| <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> |
| <div class="line"><a name="l03330"></a><span class="lineno"> 3330</span>  </div> |
| <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> |
| <div class="line"><a name="l03332"></a><span class="lineno"> 3332</span>  <span class="keywordflow">if</span> (!axisOperand)</div> |
| <div class="line"><a name="l03333"></a><span class="lineno"> 3333</span>  {</div> |
| <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> |
| <div class="line"><a name="l03335"></a><span class="lineno"> 3335</span>  }</div> |
| <div class="line"><a name="l03336"></a><span class="lineno"> 3336</span>  </div> |
| <div class="line"><a name="l03337"></a><span class="lineno"> 3337</span>  std::vector<int32_t> axis;</div> |
| <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> |
| <div class="line"><a name="l03339"></a><span class="lineno"> 3339</span>  {</div> |
| <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> |
| <div class="line"><a name="l03341"></a><span class="lineno"> 3341</span>  }</div> |
| <div class="line"><a name="l03342"></a><span class="lineno"> 3342</span>  </div> |
| <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> |
| <div class="line"><a name="l03344"></a><span class="lineno"> 3344</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03347"></a><span class="lineno"> 3347</span>  std::set<unsigned int> uniqueAxis;</div> |
| <div class="line"><a name="l03348"></a><span class="lineno"> 3348</span>  std::transform(axis.begin(), axis.end(),</div> |
| <div class="line"><a name="l03349"></a><span class="lineno"> 3349</span>  std::inserter(uniqueAxis, uniqueAxis.begin()),</div> |
| <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> |
| <div class="line"><a name="l03351"></a><span class="lineno"> 3351</span>  </div> |
| <div class="line"><a name="l03352"></a><span class="lineno"> 3352</span>  <span class="comment">// Get the "keep dims" flag.</span></div> |
| <div class="line"><a name="l03353"></a><span class="lineno"> 3353</span>  int32_t keepDims = 0;</div> |
| <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> |
| <div class="line"><a name="l03355"></a><span class="lineno"> 3355</span>  {</div> |
| <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> |
| <div class="line"><a name="l03357"></a><span class="lineno"> 3357</span>  }</div> |
| <div class="line"><a name="l03358"></a><span class="lineno"> 3358</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03362"></a><span class="lineno"> 3362</span>  </div> |
| <div class="line"><a name="l03363"></a><span class="lineno"> 3363</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03366"></a><span class="lineno"> 3366</span>  {</div> |
| <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> |
| <div class="line"><a name="l03368"></a><span class="lineno"> 3368</span>  IsMeanSupported,</div> |
| <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> |
| <div class="line"><a name="l03370"></a><span class="lineno"> 3370</span>  isSupported,</div> |
| <div class="line"><a name="l03371"></a><span class="lineno"> 3371</span>  setBackend,</div> |
| <div class="line"><a name="l03372"></a><span class="lineno"> 3372</span>  inputInfo,</div> |
| <div class="line"><a name="l03373"></a><span class="lineno"> 3373</span>  outputInfo,</div> |
| <div class="line"><a name="l03374"></a><span class="lineno"> 3374</span>  descriptor);</div> |
| <div class="line"><a name="l03375"></a><span class="lineno"> 3375</span>  };</div> |
| <div class="line"><a name="l03376"></a><span class="lineno"> 3376</span>  </div> |
| <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> |
| <div class="line"><a name="l03378"></a><span class="lineno"> 3378</span>  {</div> |
| <div class="line"><a name="l03379"></a><span class="lineno"> 3379</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l03380"></a><span class="lineno"> 3380</span>  }</div> |
| <div class="line"><a name="l03381"></a><span class="lineno"> 3381</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03382"></a><span class="lineno"> 3382</span>  {</div> |
| <div class="line"><a name="l03383"></a><span class="lineno"> 3383</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l03384"></a><span class="lineno"> 3384</span>  }</div> |
| <div class="line"><a name="l03385"></a><span class="lineno"> 3385</span>  </div> |
| <div class="line"><a name="l03386"></a><span class="lineno"> 3386</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l03387"></a><span class="lineno"> 3387</span>  {</div> |
| <div class="line"><a name="l03388"></a><span class="lineno"> 3388</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03389"></a><span class="lineno"> 3389</span>  }</div> |
| <div class="line"><a name="l03390"></a><span class="lineno"> 3390</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03393"></a><span class="lineno"> 3393</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l03395"></a><span class="lineno"> 3395</span>  </div> |
| <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> |
| <div class="line"><a name="l03397"></a><span class="lineno"> 3397</span> }</div> |
| <div class="line"><a name="l03398"></a><span class="lineno"> 3398</span>  </div> |
| <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> |
| <div class="line"><a name="l03400"></a><span class="lineno"> 3400</span> {</div> |
| <div class="line"><a name="l03401"></a><span class="lineno"> 3401</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertPad()"</span>;</div> |
| <div class="line"><a name="l03402"></a><span class="lineno"> 3402</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03405"></a><span class="lineno"> 3405</span>  {</div> |
| <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> |
| <div class="line"><a name="l03407"></a><span class="lineno"> 3407</span>  }</div> |
| <div class="line"><a name="l03408"></a><span class="lineno"> 3408</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03411"></a><span class="lineno"> 3411</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03414"></a><span class="lineno"> 3414</span>  {</div> |
| <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> |
| <div class="line"><a name="l03416"></a><span class="lineno"> 3416</span>  }</div> |
| <div class="line"><a name="l03417"></a><span class="lineno"> 3417</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03422"></a><span class="lineno"> 3422</span>  <span class="comment">// (QuantizationOffset - QuantizationOffset) * scale = 0.</span></div> |
| <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> |
| <div class="line"><a name="l03424"></a><span class="lineno"> 3424</span>  {</div> |
| <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> |
| <div class="line"><a name="l03426"></a><span class="lineno"> 3426</span>  }</div> |
| <div class="line"><a name="l03427"></a><span class="lineno"> 3427</span>  </div> |
| <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> |
| <div class="line"><a name="l03429"></a><span class="lineno"> 3429</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l03430"></a><span class="lineno"> 3430</span>  {</div> |
| <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> |
| <div class="line"><a name="l03432"></a><span class="lineno"> 3432</span>  }</div> |
| <div class="line"><a name="l03433"></a><span class="lineno"> 3433</span>  </div> |
| <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> |
| <div class="line"><a name="l03435"></a><span class="lineno"> 3435</span>  </div> |
| <div class="line"><a name="l03436"></a><span class="lineno"> 3436</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03439"></a><span class="lineno"> 3439</span>  {</div> |
| <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> |
| <div class="line"><a name="l03441"></a><span class="lineno"> 3441</span>  IsPadSupported,</div> |
| <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> |
| <div class="line"><a name="l03443"></a><span class="lineno"> 3443</span>  isSupported,</div> |
| <div class="line"><a name="l03444"></a><span class="lineno"> 3444</span>  setBackend,</div> |
| <div class="line"><a name="l03445"></a><span class="lineno"> 3445</span>  inputInfo,</div> |
| <div class="line"><a name="l03446"></a><span class="lineno"> 3446</span>  outputInfo,</div> |
| <div class="line"><a name="l03447"></a><span class="lineno"> 3447</span>  descriptor);</div> |
| <div class="line"><a name="l03448"></a><span class="lineno"> 3448</span>  };</div> |
| <div class="line"><a name="l03449"></a><span class="lineno"> 3449</span>  </div> |
| <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> |
| <div class="line"><a name="l03451"></a><span class="lineno"> 3451</span>  {</div> |
| <div class="line"><a name="l03452"></a><span class="lineno"> 3452</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l03453"></a><span class="lineno"> 3453</span>  }</div> |
| <div class="line"><a name="l03454"></a><span class="lineno"> 3454</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03455"></a><span class="lineno"> 3455</span>  {</div> |
| <div class="line"><a name="l03456"></a><span class="lineno"> 3456</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l03457"></a><span class="lineno"> 3457</span>  }</div> |
| <div class="line"><a name="l03458"></a><span class="lineno"> 3458</span>  </div> |
| <div class="line"><a name="l03459"></a><span class="lineno"> 3459</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l03460"></a><span class="lineno"> 3460</span>  {</div> |
| <div class="line"><a name="l03461"></a><span class="lineno"> 3461</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03462"></a><span class="lineno"> 3462</span>  }</div> |
| <div class="line"><a name="l03463"></a><span class="lineno"> 3463</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03466"></a><span class="lineno"> 3466</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l03468"></a><span class="lineno"> 3468</span>  </div> |
| <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> |
| <div class="line"><a name="l03470"></a><span class="lineno"> 3470</span> }</div> |
| <div class="line"><a name="l03471"></a><span class="lineno"> 3471</span>  </div> |
| <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> |
| <div class="line"><a name="l03473"></a><span class="lineno"> 3473</span> {</div> |
| <div class="line"><a name="l03474"></a><span class="lineno"> 3474</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertPadV2()"</span>;</div> |
| <div class="line"><a name="l03475"></a><span class="lineno"> 3475</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03478"></a><span class="lineno"> 3478</span>  {</div> |
| <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> |
| <div class="line"><a name="l03480"></a><span class="lineno"> 3480</span>  }</div> |
| <div class="line"><a name="l03481"></a><span class="lineno"> 3481</span>  </div> |
| <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> |
| <div class="line"><a name="l03483"></a><span class="lineno"> 3483</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l03484"></a><span class="lineno"> 3484</span>  {</div> |
| <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> |
| <div class="line"><a name="l03486"></a><span class="lineno"> 3486</span>  }</div> |
| <div class="line"><a name="l03487"></a><span class="lineno"> 3487</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03490"></a><span class="lineno"> 3490</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03493"></a><span class="lineno"> 3493</span>  {</div> |
| <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> |
| <div class="line"><a name="l03495"></a><span class="lineno"> 3495</span>  }</div> |
| <div class="line"><a name="l03496"></a><span class="lineno"> 3496</span>  </div> |
| <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> |
| <div class="line"><a name="l03498"></a><span class="lineno"> 3498</span>  </div> |
| <div class="line"><a name="l03499"></a><span class="lineno"> 3499</span>  <span class="comment">// Determine type of padding value</span></div> |
| <div class="line"><a name="l03500"></a><span class="lineno"> 3500</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType0;</div> |
| <div class="line"><a name="l03501"></a><span class="lineno"> 3501</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType2;</div> |
| <div class="line"><a name="l03502"></a><span class="lineno"> 3502</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03505"></a><span class="lineno"> 3505</span>  {</div> |
| <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> |
| <div class="line"><a name="l03507"></a><span class="lineno"> 3507</span>  }</div> |
| <div class="line"><a name="l03508"></a><span class="lineno"> 3508</span>  </div> |
| <div class="line"><a name="l03509"></a><span class="lineno"> 3509</span>  <span class="comment">// Read value to use for padding</span></div> |
| <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> |
| <div class="line"><a name="l03511"></a><span class="lineno"> 3511</span>  {</div> |
| <div class="line"><a name="l03512"></a><span class="lineno"> 3512</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> f16PadValue;</div> |
| <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> |
| <div class="line"><a name="l03514"></a><span class="lineno"> 3514</span>  {</div> |
| <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> |
| <div class="line"><a name="l03516"></a><span class="lineno"> 3516</span>  }</div> |
| <div class="line"><a name="l03517"></a><span class="lineno"> 3517</span>  </div> |
| <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> |
| <div class="line"><a name="l03519"></a><span class="lineno"> 3519</span>  }</div> |
| <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> |
| <div class="line"><a name="l03521"></a><span class="lineno"> 3521</span>  {</div> |
| <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> |
| <div class="line"><a name="l03523"></a><span class="lineno"> 3523</span>  {</div> |
| <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> |
| <div class="line"><a name="l03525"></a><span class="lineno"> 3525</span>  }</div> |
| <div class="line"><a name="l03526"></a><span class="lineno"> 3526</span>  }</div> |
| <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> |
| <div class="line"><a name="l03528"></a><span class="lineno"> 3528</span>  {</div> |
| <div class="line"><a name="l03529"></a><span class="lineno"> 3529</span>  int32_t intPadValue = 0;</div> |
| <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> |
| <div class="line"><a name="l03531"></a><span class="lineno"> 3531</span>  {</div> |
| <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> |
| <div class="line"><a name="l03533"></a><span class="lineno"> 3533</span>  }</div> |
| <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> |
| <div class="line"><a name="l03535"></a><span class="lineno"> 3535</span>  }</div> |
| <div class="line"><a name="l03536"></a><span class="lineno"> 3536</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03537"></a><span class="lineno"> 3537</span>  {</div> |
| <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> |
| <div class="line"><a name="l03539"></a><span class="lineno"> 3539</span>  }</div> |
| <div class="line"><a name="l03540"></a><span class="lineno"> 3540</span>  </div> |
| <div class="line"><a name="l03541"></a><span class="lineno"> 3541</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03544"></a><span class="lineno"> 3544</span>  {</div> |
| <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> |
| <div class="line"><a name="l03546"></a><span class="lineno"> 3546</span>  IsPadSupported,</div> |
| <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> |
| <div class="line"><a name="l03548"></a><span class="lineno"> 3548</span>  isSupported,</div> |
| <div class="line"><a name="l03549"></a><span class="lineno"> 3549</span>  setBackend,</div> |
| <div class="line"><a name="l03550"></a><span class="lineno"> 3550</span>  inputInfo,</div> |
| <div class="line"><a name="l03551"></a><span class="lineno"> 3551</span>  outputInfo,</div> |
| <div class="line"><a name="l03552"></a><span class="lineno"> 3552</span>  descriptor);</div> |
| <div class="line"><a name="l03553"></a><span class="lineno"> 3553</span>  };</div> |
| <div class="line"><a name="l03554"></a><span class="lineno"> 3554</span>  </div> |
| <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> |
| <div class="line"><a name="l03556"></a><span class="lineno"> 3556</span>  {</div> |
| <div class="line"><a name="l03557"></a><span class="lineno"> 3557</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l03558"></a><span class="lineno"> 3558</span>  }</div> |
| <div class="line"><a name="l03559"></a><span class="lineno"> 3559</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03560"></a><span class="lineno"> 3560</span>  {</div> |
| <div class="line"><a name="l03561"></a><span class="lineno"> 3561</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l03562"></a><span class="lineno"> 3562</span>  }</div> |
| <div class="line"><a name="l03563"></a><span class="lineno"> 3563</span>  </div> |
| <div class="line"><a name="l03564"></a><span class="lineno"> 3564</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l03565"></a><span class="lineno"> 3565</span>  {</div> |
| <div class="line"><a name="l03566"></a><span class="lineno"> 3566</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03567"></a><span class="lineno"> 3567</span>  }</div> |
| <div class="line"><a name="l03568"></a><span class="lineno"> 3568</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03571"></a><span class="lineno"> 3571</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l03573"></a><span class="lineno"> 3573</span>  </div> |
| <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> |
| <div class="line"><a name="l03575"></a><span class="lineno"> 3575</span> }</div> |
| <div class="line"><a name="l03576"></a><span class="lineno"> 3576</span>  </div> |
| <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> |
| <div class="line"><a name="l03578"></a><span class="lineno"> 3578</span> {</div> |
| <div class="line"><a name="l03579"></a><span class="lineno"> 3579</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertPrelu()"</span>;</div> |
| <div class="line"><a name="l03580"></a><span class="lineno"> 3580</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03583"></a><span class="lineno"> 3583</span>  </div> |
| <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> |
| <div class="line"><a name="l03585"></a><span class="lineno"> 3585</span>  {</div> |
| <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> |
| <div class="line"><a name="l03587"></a><span class="lineno"> 3587</span>  }</div> |
| <div class="line"><a name="l03588"></a><span class="lineno"> 3588</span>  </div> |
| <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> |
| <div class="line"><a name="l03590"></a><span class="lineno"> 3590</span>  </div> |
| <div class="line"><a name="l03591"></a><span class="lineno"> 3591</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l03592"></a><span class="lineno"> 3592</span>  {</div> |
| <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> |
| <div class="line"><a name="l03594"></a><span class="lineno"> 3594</span>  }</div> |
| <div class="line"><a name="l03595"></a><span class="lineno"> 3595</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03599"></a><span class="lineno"> 3599</span>  </div> |
| <div class="line"><a name="l03600"></a><span class="lineno"> 3600</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03603"></a><span class="lineno"> 3603</span>  {</div> |
| <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> |
| <div class="line"><a name="l03605"></a><span class="lineno"> 3605</span>  IsPreluSupported,</div> |
| <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> |
| <div class="line"><a name="l03607"></a><span class="lineno"> 3607</span>  isSupported,</div> |
| <div class="line"><a name="l03608"></a><span class="lineno"> 3608</span>  setBackend,</div> |
| <div class="line"><a name="l03609"></a><span class="lineno"> 3609</span>  inputInfo,</div> |
| <div class="line"><a name="l03610"></a><span class="lineno"> 3610</span>  alphaInfo,</div> |
| <div class="line"><a name="l03611"></a><span class="lineno"> 3611</span>  outputInfo);</div> |
| <div class="line"><a name="l03612"></a><span class="lineno"> 3612</span>  };</div> |
| <div class="line"><a name="l03613"></a><span class="lineno"> 3613</span>  </div> |
| <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> |
| <div class="line"><a name="l03615"></a><span class="lineno"> 3615</span>  {</div> |
| <div class="line"><a name="l03616"></a><span class="lineno"> 3616</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l03617"></a><span class="lineno"> 3617</span>  }</div> |
| <div class="line"><a name="l03618"></a><span class="lineno"> 3618</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03619"></a><span class="lineno"> 3619</span>  {</div> |
| <div class="line"><a name="l03620"></a><span class="lineno"> 3620</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l03621"></a><span class="lineno"> 3621</span>  }</div> |
| <div class="line"><a name="l03622"></a><span class="lineno"> 3622</span>  </div> |
| <div class="line"><a name="l03623"></a><span class="lineno"> 3623</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l03624"></a><span class="lineno"> 3624</span>  {</div> |
| <div class="line"><a name="l03625"></a><span class="lineno"> 3625</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03626"></a><span class="lineno"> 3626</span>  }</div> |
| <div class="line"><a name="l03627"></a><span class="lineno"> 3627</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03630"></a><span class="lineno"> 3630</span>  </div> |
| <div class="line"><a name="l03631"></a><span class="lineno"> 3631</span>  <span class="keywordflow">if</span> (!layer)</div> |
| <div class="line"><a name="l03632"></a><span class="lineno"> 3632</span>  {</div> |
| <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> |
| <div class="line"><a name="l03634"></a><span class="lineno"> 3634</span>  }</div> |
| <div class="line"><a name="l03635"></a><span class="lineno"> 3635</span>  </div> |
| <div class="line"><a name="l03636"></a><span class="lineno"> 3636</span>  <span class="keywordtype">bool</span> isReshapeSupported = BroadcastTensor(input, alpha, layer, data);</div> |
| <div class="line"><a name="l03637"></a><span class="lineno"> 3637</span>  <span class="keywordflow">if</span> (!isReshapeSupported)</div> |
| <div class="line"><a name="l03638"></a><span class="lineno"> 3638</span>  {</div> |
| <div class="line"><a name="l03639"></a><span class="lineno"> 3639</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03640"></a><span class="lineno"> 3640</span>  }</div> |
| <div class="line"><a name="l03641"></a><span class="lineno"> 3641</span>  </div> |
| <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> |
| <div class="line"><a name="l03643"></a><span class="lineno"> 3643</span> }</div> |
| <div class="line"><a name="l03644"></a><span class="lineno"> 3644</span>  </div> |
| <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> |
| <div class="line"><a name="l03646"></a><span class="lineno"> 3646</span> {</div> |
| <div class="line"><a name="l03647"></a><span class="lineno"> 3647</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertQuantize()"</span>;</div> |
| <div class="line"><a name="l03648"></a><span class="lineno"> 3648</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03651"></a><span class="lineno"> 3651</span>  {</div> |
| <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> |
| <div class="line"><a name="l03653"></a><span class="lineno"> 3653</span>  }</div> |
| <div class="line"><a name="l03654"></a><span class="lineno"> 3654</span>  </div> |
| <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> |
| <div class="line"><a name="l03656"></a><span class="lineno"> 3656</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l03657"></a><span class="lineno"> 3657</span>  {</div> |
| <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> |
| <div class="line"><a name="l03659"></a><span class="lineno"> 3659</span>  }</div> |
| <div class="line"><a name="l03660"></a><span class="lineno"> 3660</span>  </div> |
| <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> |
| <div class="line"><a name="l03662"></a><span class="lineno"> 3662</span>  </div> |
| <div class="line"><a name="l03663"></a><span class="lineno"> 3663</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l03666"></a><span class="lineno"> 3666</span>  {</div> |
| <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> |
| <div class="line"><a name="l03668"></a><span class="lineno"> 3668</span>  IsQuantizeSupported,</div> |
| <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> |
| <div class="line"><a name="l03670"></a><span class="lineno"> 3670</span>  isSupported,</div> |
| <div class="line"><a name="l03671"></a><span class="lineno"> 3671</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l03673"></a><span class="lineno"> 3673</span>  outputInfo);</div> |
| <div class="line"><a name="l03674"></a><span class="lineno"> 3674</span>  };</div> |
| <div class="line"><a name="l03675"></a><span class="lineno"> 3675</span>  </div> |
| <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> |
| <div class="line"><a name="l03677"></a><span class="lineno"> 3677</span>  {</div> |
| <div class="line"><a name="l03678"></a><span class="lineno"> 3678</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l03679"></a><span class="lineno"> 3679</span>  }</div> |
| <div class="line"><a name="l03680"></a><span class="lineno"> 3680</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l03681"></a><span class="lineno"> 3681</span>  {</div> |
| <div class="line"><a name="l03682"></a><span class="lineno"> 3682</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l03683"></a><span class="lineno"> 3683</span>  }</div> |
| <div class="line"><a name="l03684"></a><span class="lineno"> 3684</span>  </div> |
| <div class="line"><a name="l03685"></a><span class="lineno"> 3685</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l03686"></a><span class="lineno"> 3686</span>  {</div> |
| <div class="line"><a name="l03687"></a><span class="lineno"> 3687</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l03688"></a><span class="lineno"> 3688</span>  }</div> |
| <div class="line"><a name="l03689"></a><span class="lineno"> 3689</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03692"></a><span class="lineno"> 3692</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l03694"></a><span class="lineno"> 3694</span>  </div> |
| <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> |
| <div class="line"><a name="l03696"></a><span class="lineno"> 3696</span> }</div> |
| <div class="line"><a name="l03697"></a><span class="lineno"> 3697</span>  </div> |
| <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> |
| <div class="line"><a name="l03699"></a><span class="lineno"> 3699</span> {</div> |
| <div class="line"><a name="l03700"></a><span class="lineno"> 3700</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertQuantizedLstm()"</span>;</div> |
| <div class="line"><a name="l03701"></a><span class="lineno"> 3701</span>  </div> |
| <div class="line"><a name="l03702"></a><span class="lineno"> 3702</span>  VLOG(DRIVER) << <span class="stringliteral">"ConvertQuantizedLstm()"</span>;</div> |
| <div class="line"><a name="l03703"></a><span class="lineno"> 3703</span>  </div> |
| <div class="line"><a name="l03704"></a><span class="lineno"> 3704</span>  <span class="comment">//Inputs:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03709"></a><span class="lineno"> 3709</span>  {</div> |
| <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> |
| <div class="line"><a name="l03711"></a><span class="lineno"> 3711</span>  }</div> |
| <div class="line"><a name="l03712"></a><span class="lineno"> 3712</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03716"></a><span class="lineno"> 3716</span>  {</div> |
| <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> |
| <div class="line"><a name="l03718"></a><span class="lineno"> 3718</span>  }</div> |
| <div class="line"><a name="l03719"></a><span class="lineno"> 3719</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03723"></a><span class="lineno"> 3723</span>  {</div> |
| <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> |
| <div class="line"><a name="l03725"></a><span class="lineno"> 3725</span>  }</div> |
| <div class="line"><a name="l03726"></a><span class="lineno"> 3726</span>  </div> |
| <div class="line"><a name="l03727"></a><span class="lineno"> 3727</span>  <span class="comment">// Get the mandatory input tensors:</span></div> |
| <div class="line"><a name="l03728"></a><span class="lineno"> 3728</span>  </div> |
| <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> |
| <div class="line"><a name="l03730"></a><span class="lineno"> 3730</span>  <span class="comment">// [num_units, input_size].</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03733"></a><span class="lineno"> 3733</span>  </div> |
| <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> |
| <div class="line"><a name="l03735"></a><span class="lineno"> 3735</span>  <span class="comment">// [num_units, input_size].</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03738"></a><span class="lineno"> 3738</span>  </div> |
| <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> |
| <div class="line"><a name="l03740"></a><span class="lineno"> 3740</span>  <span class="comment">// [num_units, input_size].</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03743"></a><span class="lineno"> 3743</span>  </div> |
| <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> |
| <div class="line"><a name="l03745"></a><span class="lineno"> 3745</span>  <span class="comment">// [num_units, output_size].</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03748"></a><span class="lineno"> 3748</span>  </div> |
| <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> |
| <div class="line"><a name="l03750"></a><span class="lineno"> 3750</span>  <span class="comment">// [num_units, output_size].</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03753"></a><span class="lineno"> 3753</span>  </div> |
| <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> |
| <div class="line"><a name="l03755"></a><span class="lineno"> 3755</span>  <span class="comment">// [num_units, output_size].</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03758"></a><span class="lineno"> 3758</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03762"></a><span class="lineno"> 3762</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03766"></a><span class="lineno"> 3766</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03770"></a><span class="lineno"> 3770</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03780"></a><span class="lineno"> 3780</span>  {</div> |
| <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> |
| <div class="line"><a name="l03782"></a><span class="lineno"> 3782</span>  }</div> |
| <div class="line"><a name="l03783"></a><span class="lineno"> 3783</span>  </div> |
| <div class="line"><a name="l03784"></a><span class="lineno"> 3784</span>  <span class="comment">// Get the optional input tensors:</span></div> |
| <div class="line"><a name="l03785"></a><span class="lineno"> 3785</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03789"></a><span class="lineno"> 3789</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03790"></a><span class="lineno"> 3790</span>  1,</div> |
| <div class="line"><a name="l03791"></a><span class="lineno"> 3791</span>  model,</div> |
| <div class="line"><a name="l03792"></a><span class="lineno"> 3792</span>  data,</div> |
| <div class="line"><a name="l03793"></a><span class="lineno"> 3793</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03794"></a><span class="lineno"> 3794</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03795"></a><span class="lineno"> 3795</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03796"></a><span class="lineno"> 3796</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03801"></a><span class="lineno"> 3801</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03802"></a><span class="lineno"> 3802</span>  5,</div> |
| <div class="line"><a name="l03803"></a><span class="lineno"> 3803</span>  model,</div> |
| <div class="line"><a name="l03804"></a><span class="lineno"> 3804</span>  data,</div> |
| <div class="line"><a name="l03805"></a><span class="lineno"> 3805</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03806"></a><span class="lineno"> 3806</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03807"></a><span class="lineno"> 3807</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03808"></a><span class="lineno"> 3808</span>  </div> |
| <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> |
| <div class="line"><a name="l03810"></a><span class="lineno"> 3810</span>  <span class="comment">// [num_units].</span></div> |
| <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> |
| <div class="line"><a name="l03812"></a><span class="lineno"> 3812</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03813"></a><span class="lineno"> 3813</span>  9,</div> |
| <div class="line"><a name="l03814"></a><span class="lineno"> 3814</span>  model,</div> |
| <div class="line"><a name="l03815"></a><span class="lineno"> 3815</span>  data,</div> |
| <div class="line"><a name="l03816"></a><span class="lineno"> 3816</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03817"></a><span class="lineno"> 3817</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03818"></a><span class="lineno"> 3818</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03819"></a><span class="lineno"> 3819</span>  </div> |
| <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> |
| <div class="line"><a name="l03821"></a><span class="lineno"> 3821</span>  <span class="comment">// [num_units].</span></div> |
| <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> |
| <div class="line"><a name="l03823"></a><span class="lineno"> 3823</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03824"></a><span class="lineno"> 3824</span>  10,</div> |
| <div class="line"><a name="l03825"></a><span class="lineno"> 3825</span>  model,</div> |
| <div class="line"><a name="l03826"></a><span class="lineno"> 3826</span>  data,</div> |
| <div class="line"><a name="l03827"></a><span class="lineno"> 3827</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03828"></a><span class="lineno"> 3828</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03829"></a><span class="lineno"> 3829</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03830"></a><span class="lineno"> 3830</span>  </div> |
| <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> |
| <div class="line"><a name="l03832"></a><span class="lineno"> 3832</span>  <span class="comment">// [num_units].</span></div> |
| <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> |
| <div class="line"><a name="l03834"></a><span class="lineno"> 3834</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03835"></a><span class="lineno"> 3835</span>  11,</div> |
| <div class="line"><a name="l03836"></a><span class="lineno"> 3836</span>  model,</div> |
| <div class="line"><a name="l03837"></a><span class="lineno"> 3837</span>  data,</div> |
| <div class="line"><a name="l03838"></a><span class="lineno"> 3838</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03839"></a><span class="lineno"> 3839</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03840"></a><span class="lineno"> 3840</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03841"></a><span class="lineno"> 3841</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03844"></a><span class="lineno"> 3844</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03845"></a><span class="lineno"> 3845</span>  12,</div> |
| <div class="line"><a name="l03846"></a><span class="lineno"> 3846</span>  model,</div> |
| <div class="line"><a name="l03847"></a><span class="lineno"> 3847</span>  data,</div> |
| <div class="line"><a name="l03848"></a><span class="lineno"> 3848</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03849"></a><span class="lineno"> 3849</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03850"></a><span class="lineno"> 3850</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03851"></a><span class="lineno"> 3851</span>  </div> |
| <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> |
| <div class="line"><a name="l03853"></a><span class="lineno"> 3853</span>  <span class="comment">// [output_size, num_units].</span></div> |
| <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> |
| <div class="line"><a name="l03855"></a><span class="lineno"> 3855</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03856"></a><span class="lineno"> 3856</span>  16,</div> |
| <div class="line"><a name="l03857"></a><span class="lineno"> 3857</span>  model,</div> |
| <div class="line"><a name="l03858"></a><span class="lineno"> 3858</span>  data,</div> |
| <div class="line"><a name="l03859"></a><span class="lineno"> 3859</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03860"></a><span class="lineno"> 3860</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03861"></a><span class="lineno"> 3861</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03862"></a><span class="lineno"> 3862</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03865"></a><span class="lineno"> 3865</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03866"></a><span class="lineno"> 3866</span>  17,</div> |
| <div class="line"><a name="l03867"></a><span class="lineno"> 3867</span>  model,</div> |
| <div class="line"><a name="l03868"></a><span class="lineno"> 3868</span>  data,</div> |
| <div class="line"><a name="l03869"></a><span class="lineno"> 3869</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03870"></a><span class="lineno"> 3870</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03871"></a><span class="lineno"> 3871</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03872"></a><span class="lineno"> 3872</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03881"></a><span class="lineno"> 3881</span>  {</div> |
| <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> |
| <div class="line"><a name="l03883"></a><span class="lineno"> 3883</span>  }</div> |
| <div class="line"><a name="l03884"></a><span class="lineno"> 3884</span>  </div> |
| <div class="line"><a name="l03885"></a><span class="lineno"> 3885</span>  </div> |
| <div class="line"><a name="l03886"></a><span class="lineno"> 3886</span>  <span class="comment">// Get the optional normalization tensors</span></div> |
| <div class="line"><a name="l03887"></a><span class="lineno"> 3887</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03891"></a><span class="lineno"> 3891</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03892"></a><span class="lineno"> 3892</span>  20,</div> |
| <div class="line"><a name="l03893"></a><span class="lineno"> 3893</span>  model,</div> |
| <div class="line"><a name="l03894"></a><span class="lineno"> 3894</span>  data,</div> |
| <div class="line"><a name="l03895"></a><span class="lineno"> 3895</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03896"></a><span class="lineno"> 3896</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03897"></a><span class="lineno"> 3897</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03898"></a><span class="lineno"> 3898</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03902"></a><span class="lineno"> 3902</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03903"></a><span class="lineno"> 3903</span>  21,</div> |
| <div class="line"><a name="l03904"></a><span class="lineno"> 3904</span>  model,</div> |
| <div class="line"><a name="l03905"></a><span class="lineno"> 3905</span>  data,</div> |
| <div class="line"><a name="l03906"></a><span class="lineno"> 3906</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03907"></a><span class="lineno"> 3907</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03908"></a><span class="lineno"> 3908</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03909"></a><span class="lineno"> 3909</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03913"></a><span class="lineno"> 3913</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03914"></a><span class="lineno"> 3914</span>  22,</div> |
| <div class="line"><a name="l03915"></a><span class="lineno"> 3915</span>  model,</div> |
| <div class="line"><a name="l03916"></a><span class="lineno"> 3916</span>  data,</div> |
| <div class="line"><a name="l03917"></a><span class="lineno"> 3917</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03918"></a><span class="lineno"> 3918</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03919"></a><span class="lineno"> 3919</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03920"></a><span class="lineno"> 3920</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03924"></a><span class="lineno"> 3924</span>  <a class="code" href="namespacearmnn__driver.html#a8c749c18e62c59a5c1254aac342c995a">ConvertOperationInputToConstTensorPin</a>(operation,</div> |
| <div class="line"><a name="l03925"></a><span class="lineno"> 3925</span>  23,</div> |
| <div class="line"><a name="l03926"></a><span class="lineno"> 3926</span>  model,</div> |
| <div class="line"><a name="l03927"></a><span class="lineno"> 3927</span>  data,</div> |
| <div class="line"><a name="l03928"></a><span class="lineno"> 3928</span>  <a class="code" href="namespacearmnn__driver.html#ae059aea66ee0867a204aa2f2d9abb42c">g_DontPermute</a>,</div> |
| <div class="line"><a name="l03929"></a><span class="lineno"> 3929</span>  <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l03930"></a><span class="lineno"> 3930</span>  <span class="keyword">true</span>);</div> |
| <div class="line"><a name="l03931"></a><span class="lineno"> 3931</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03936"></a><span class="lineno"> 3936</span>  {</div> |
| <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> |
| <div class="line"><a name="l03938"></a><span class="lineno"> 3938</span>  }</div> |
| <div class="line"><a name="l03939"></a><span class="lineno"> 3939</span>  </div> |
| <div class="line"><a name="l03940"></a><span class="lineno"> 3940</span>  <span class="comment">// Get the optional input scalars:</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l03943"></a><span class="lineno"> 3943</span>  </div> |
| <div class="line"><a name="l03944"></a><span class="lineno"> 3944</span>  <span class="comment">// Get the mandatory input scalars:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03952"></a><span class="lineno"> 3952</span>  <span class="keywordtype">int</span> projInputZeroPoint;</div> |
| <div class="line"><a name="l03953"></a><span class="lineno"> 3953</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03962"></a><span class="lineno"> 3962</span>  {</div> |
| <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> |
| <div class="line"><a name="l03964"></a><span class="lineno"> 3964</span>  }</div> |
| <div class="line"><a name="l03965"></a><span class="lineno"> 3965</span>  </div> |
| <div class="line"><a name="l03966"></a><span class="lineno"> 3966</span>  <span class="comment">// Outputs:</span></div> |
| <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> |
| <div class="line"><a name="l03968"></a><span class="lineno"> 3968</span>  <span class="comment">// output_size].</span></div> |
| <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> |
| <div class="line"><a name="l03970"></a><span class="lineno"> 3970</span>  <span class="keywordflow">if</span> (!outputStateOut)</div> |
| <div class="line"><a name="l03971"></a><span class="lineno"> 3971</span>  {</div> |
| <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> |
| <div class="line"><a name="l03973"></a><span class="lineno"> 3973</span>  }</div> |
| <div class="line"><a name="l03974"></a><span class="lineno"> 3974</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l03977"></a><span class="lineno"> 3977</span>  <span class="keywordflow">if</span> (!cellStateOut)</div> |
| <div class="line"><a name="l03978"></a><span class="lineno"> 3978</span>  {</div> |
| <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> |
| <div class="line"><a name="l03980"></a><span class="lineno"> 3980</span>  }</div> |
| <div class="line"><a name="l03981"></a><span class="lineno"> 3981</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l03985"></a><span class="lineno"> 3985</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l03986"></a><span class="lineno"> 3986</span>  {</div> |
| <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> |
| <div class="line"><a name="l03988"></a><span class="lineno"> 3988</span>  }</div> |
| <div class="line"><a name="l03989"></a><span class="lineno"> 3989</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04013"></a><span class="lineno"> 4013</span>  </div> |
| <div class="line"><a name="l04014"></a><span class="lineno"> 4014</span>  <span class="comment">// set the layer descriptor</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04034"></a><span class="lineno"> 4034</span>  </div> |
| <div class="line"><a name="l04035"></a><span class="lineno"> 4035</span>  <span class="comment">// validate the optional input groups</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04040"></a><span class="lineno"> 4040</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04043"></a><span class="lineno"> 4043</span>  }</div> |
| <div class="line"><a name="l04044"></a><span class="lineno"> 4044</span>  </div> |
| <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> |
| <div class="line"><a name="l04046"></a><span class="lineno"> 4046</span>  {</div> |
| <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> |
| <div class="line"><a name="l04048"></a><span class="lineno"> 4048</span>  }</div> |
| <div class="line"><a name="l04049"></a><span class="lineno"> 4049</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04054"></a><span class="lineno"> 4054</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04057"></a><span class="lineno"> 4057</span>  }</div> |
| <div class="line"><a name="l04058"></a><span class="lineno"> 4058</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04064"></a><span class="lineno"> 4064</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04067"></a><span class="lineno"> 4067</span>  }</div> |
| <div class="line"><a name="l04068"></a><span class="lineno"> 4068</span>  </div> |
| <div class="line"><a name="l04069"></a><span class="lineno"> 4069</span>  <span class="comment">// Basic parameters</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04080"></a><span class="lineno"> 4080</span>  </div> |
| <div class="line"><a name="l04081"></a><span class="lineno"> 4081</span>  <span class="comment">// Inputs</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04085"></a><span class="lineno"> 4085</span>  </div> |
| <div class="line"><a name="l04086"></a><span class="lineno"> 4086</span>  <span class="comment">// Outputs</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04090"></a><span class="lineno"> 4090</span>  </div> |
| <div class="line"><a name="l04091"></a><span class="lineno"> 4091</span>  <span class="comment">// Optional parameters</span></div> |
| <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> |
| <div class="line"><a name="l04093"></a><span class="lineno"> 4093</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04097"></a><span class="lineno"> 4097</span>  {</div> |
| <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> |
| <div class="line"><a name="l04099"></a><span class="lineno"> 4099</span>  }</div> |
| <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> |
| <div class="line"><a name="l04101"></a><span class="lineno"> 4101</span>  }</div> |
| <div class="line"><a name="l04102"></a><span class="lineno"> 4102</span>  </div> |
| <div class="line"><a name="l04103"></a><span class="lineno"> 4103</span>  </div> |
| <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> |
| <div class="line"><a name="l04105"></a><span class="lineno"> 4105</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04108"></a><span class="lineno"> 4108</span>  {</div> |
| <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> |
| <div class="line"><a name="l04110"></a><span class="lineno"> 4110</span>  }</div> |
| <div class="line"><a name="l04111"></a><span class="lineno"> 4111</span>  }</div> |
| <div class="line"><a name="l04112"></a><span class="lineno"> 4112</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04113"></a><span class="lineno"> 4113</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04120"></a><span class="lineno"> 4120</span>  }</div> |
| <div class="line"><a name="l04121"></a><span class="lineno"> 4121</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04124"></a><span class="lineno"> 4124</span>  </div> |
| <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> |
| <div class="line"><a name="l04126"></a><span class="lineno"> 4126</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04129"></a><span class="lineno"> 4129</span>  }</div> |
| <div class="line"><a name="l04130"></a><span class="lineno"> 4130</span>  </div> |
| <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> |
| <div class="line"><a name="l04132"></a><span class="lineno"> 4132</span>  {</div> |
| <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> |
| <div class="line"><a name="l04134"></a><span class="lineno"> 4134</span>  {</div> |
| <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> |
| <div class="line"><a name="l04136"></a><span class="lineno"> 4136</span>  }</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04140"></a><span class="lineno"> 4140</span>  }</div> |
| <div class="line"><a name="l04141"></a><span class="lineno"> 4141</span>  </div> |
| <div class="line"><a name="l04142"></a><span class="lineno"> 4142</span>  <span class="comment">// Check if the layer is supported</span></div> |
| <div class="line"><a name="l04143"></a><span class="lineno"> 4143</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04146"></a><span class="lineno"> 4146</span>  {</div> |
| <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> |
| <div class="line"><a name="l04148"></a><span class="lineno"> 4148</span>  IsQLstmSupported,</div> |
| <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> |
| <div class="line"><a name="l04150"></a><span class="lineno"> 4150</span>  isSupported,</div> |
| <div class="line"><a name="l04151"></a><span class="lineno"> 4151</span>  setBackend,</div> |
| <div class="line"><a name="l04152"></a><span class="lineno"> 4152</span>  inputInfo,</div> |
| <div class="line"><a name="l04153"></a><span class="lineno"> 4153</span>  outputStatePrevTimeStepInfo,</div> |
| <div class="line"><a name="l04154"></a><span class="lineno"> 4154</span>  cellStatePrevTimeStepInfo,</div> |
| <div class="line"><a name="l04155"></a><span class="lineno"> 4155</span>  constOutputStateOutInfo,</div> |
| <div class="line"><a name="l04156"></a><span class="lineno"> 4156</span>  cellStateOutInfo,</div> |
| <div class="line"><a name="l04157"></a><span class="lineno"> 4157</span>  constOutputInfo,</div> |
| <div class="line"><a name="l04158"></a><span class="lineno"> 4158</span>  desc,</div> |
| <div class="line"><a name="l04159"></a><span class="lineno"> 4159</span>  paramsInfo);</div> |
| <div class="line"><a name="l04160"></a><span class="lineno"> 4160</span>  };</div> |
| <div class="line"><a name="l04161"></a><span class="lineno"> 4161</span>  </div> |
| <div class="line"><a name="l04162"></a><span class="lineno"> 4162</span>  <span class="keywordtype">bool</span> isDynamic = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l04164"></a><span class="lineno"> 4164</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(cellStateOutInfo) &&</div> |
| <div class="line"><a name="l04165"></a><span class="lineno"> 4165</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(constOutputInfo))</div> |
| <div class="line"><a name="l04166"></a><span class="lineno"> 4166</span>  {</div> |
| <div class="line"><a name="l04167"></a><span class="lineno"> 4167</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l04168"></a><span class="lineno"> 4168</span>  }</div> |
| <div class="line"><a name="l04169"></a><span class="lineno"> 4169</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04170"></a><span class="lineno"> 4170</span>  {</div> |
| <div class="line"><a name="l04171"></a><span class="lineno"> 4171</span>  isDynamic = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l04172"></a><span class="lineno"> 4172</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04173"></a><span class="lineno"> 4173</span>  }</div> |
| <div class="line"><a name="l04174"></a><span class="lineno"> 4174</span>  </div> |
| <div class="line"><a name="l04175"></a><span class="lineno"> 4175</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04176"></a><span class="lineno"> 4176</span>  {</div> |
| <div class="line"><a name="l04177"></a><span class="lineno"> 4177</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04178"></a><span class="lineno"> 4178</span>  }</div> |
| <div class="line"><a name="l04179"></a><span class="lineno"> 4179</span>  </div> |
| <div class="line"><a name="l04180"></a><span class="lineno"> 4180</span>  <span class="comment">// Add the layer</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l04183"></a><span class="lineno"> 4183</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04187"></a><span class="lineno"> 4187</span>  </div> |
| <div class="line"><a name="l04188"></a><span class="lineno"> 4188</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| <div class="line"><a name="l04189"></a><span class="lineno"> 4189</span>  {</div> |
| <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> |
| <div class="line"><a name="l04191"></a><span class="lineno"> 4191</span>  operation, 0, *layer, 0, model, data, &constOutputStateOutInfo) &&</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04194"></a><span class="lineno"> 4194</span>  }</div> |
| <div class="line"><a name="l04195"></a><span class="lineno"> 4195</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04196"></a><span class="lineno"> 4196</span>  {</div> |
| <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> |
| <div class="line"><a name="l04198"></a><span class="lineno"> 4198</span>  operation, 0, *layer, 0, model, data, &constOutputStateOutInfo) &&</div> |
| <div class="line"><a name="l04199"></a><span class="lineno"> 4199</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| <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> |
| <div class="line"><a name="l04201"></a><span class="lineno"> 4201</span>  ActivationFn::kActivationNone, <span class="keyword">true</span>) &&</div> |
| <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> |
| <div class="line"><a name="l04203"></a><span class="lineno"> 4203</span>  }</div> |
| <div class="line"><a name="l04204"></a><span class="lineno"> 4204</span> }</div> |
| <div class="line"><a name="l04205"></a><span class="lineno"> 4205</span>  </div> |
| <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> |
| <div class="line"><a name="l04207"></a><span class="lineno"> 4207</span> {</div> |
| <div class="line"><a name="l04208"></a><span class="lineno"> 4208</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertQuantized16BitLstm()"</span>;</div> |
| <div class="line"><a name="l04209"></a><span class="lineno"> 4209</span>  VLOG(DRIVER) << <span class="stringliteral">"Policy::ConvertQuantized16BitLstm()"</span>;</div> |
| <div class="line"><a name="l04210"></a><span class="lineno"> 4210</span>  </div> |
| <div class="line"><a name="l04211"></a><span class="lineno"> 4211</span>  <span class="comment">//Inputs:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04216"></a><span class="lineno"> 4216</span>  {</div> |
| <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> |
| <div class="line"><a name="l04218"></a><span class="lineno"> 4218</span>  }</div> |
| <div class="line"><a name="l04219"></a><span class="lineno"> 4219</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04225"></a><span class="lineno"> 4225</span>  {</div> |
| <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> |
| <div class="line"><a name="l04227"></a><span class="lineno"> 4227</span>  }</div> |
| <div class="line"><a name="l04228"></a><span class="lineno"> 4228</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04234"></a><span class="lineno"> 4234</span>  {</div> |
| <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> |
| <div class="line"><a name="l04236"></a><span class="lineno"> 4236</span>  }</div> |
| <div class="line"><a name="l04237"></a><span class="lineno"> 4237</span>  </div> |
| <div class="line"><a name="l04238"></a><span class="lineno"> 4238</span>  <span class="comment">// Get the input tensors:</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04244"></a><span class="lineno"> 4244</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04250"></a><span class="lineno"> 4250</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04256"></a><span class="lineno"> 4256</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04262"></a><span class="lineno"> 4262</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04268"></a><span class="lineno"> 4268</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04274"></a><span class="lineno"> 4274</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04280"></a><span class="lineno"> 4280</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04286"></a><span class="lineno"> 4286</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04292"></a><span class="lineno"> 4292</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04298"></a><span class="lineno"> 4298</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04304"></a><span class="lineno"> 4304</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04310"></a><span class="lineno"> 4310</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04323"></a><span class="lineno"> 4323</span>  {</div> |
| <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> |
| <div class="line"><a name="l04325"></a><span class="lineno"> 4325</span>  }</div> |
| <div class="line"><a name="l04326"></a><span class="lineno"> 4326</span>  </div> |
| <div class="line"><a name="l04327"></a><span class="lineno"> 4327</span>  <span class="comment">// Outputs:</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l04330"></a><span class="lineno"> 4330</span>  <span class="comment">// of -2^4, 2^4 * 32767/32768.</span></div> |
| <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> |
| <div class="line"><a name="l04332"></a><span class="lineno"> 4332</span>  <span class="keywordflow">if</span> (!cellStateOut)</div> |
| <div class="line"><a name="l04333"></a><span class="lineno"> 4333</span>  {</div> |
| <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> |
| <div class="line"><a name="l04335"></a><span class="lineno"> 4335</span>  }</div> |
| <div class="line"><a name="l04336"></a><span class="lineno"> 4336</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04340"></a><span class="lineno"> 4340</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l04341"></a><span class="lineno"> 4341</span>  {</div> |
| <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> |
| <div class="line"><a name="l04343"></a><span class="lineno"> 4343</span>  }</div> |
| <div class="line"><a name="l04344"></a><span class="lineno"> 4344</span>  </div> |
| <div class="line"><a name="l04345"></a><span class="lineno"> 4345</span>  <span class="comment">// Inputs</span></div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04349"></a><span class="lineno"> 4349</span>  </div> |
| <div class="line"><a name="l04350"></a><span class="lineno"> 4350</span>  <span class="comment">// Outputs</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l04353"></a><span class="lineno"> 4353</span>  </div> |
| <div class="line"><a name="l04354"></a><span class="lineno"> 4354</span>  <span class="comment">// Dynamic tensors currently not supported</span></div> |
| <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> |
| <div class="line"><a name="l04356"></a><span class="lineno"> 4356</span>  {</div> |
| <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> |
| <div class="line"><a name="l04358"></a><span class="lineno"> 4358</span>  }</div> |
| <div class="line"><a name="l04359"></a><span class="lineno"> 4359</span>  </div> |
| <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> |
| <div class="line"><a name="l04361"></a><span class="lineno"> 4361</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04374"></a><span class="lineno"> 4374</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04388"></a><span class="lineno"> 4388</span>  </div> |
| <div class="line"><a name="l04389"></a><span class="lineno"> 4389</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04392"></a><span class="lineno"> 4392</span>  {</div> |
| <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> |
| <div class="line"><a name="l04394"></a><span class="lineno"> 4394</span>  IsQuantizedLstmSupported,</div> |
| <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> |
| <div class="line"><a name="l04396"></a><span class="lineno"> 4396</span>  isSupported,</div> |
| <div class="line"><a name="l04397"></a><span class="lineno"> 4397</span>  setBackend,</div> |
| <div class="line"><a name="l04398"></a><span class="lineno"> 4398</span>  inputInfo,</div> |
| <div class="line"><a name="l04399"></a><span class="lineno"> 4399</span>  previousCellStateInInfo,</div> |
| <div class="line"><a name="l04400"></a><span class="lineno"> 4400</span>  previousOutputInInfo,</div> |
| <div class="line"><a name="l04401"></a><span class="lineno"> 4401</span>  cellStateOutInfo,</div> |
| <div class="line"><a name="l04402"></a><span class="lineno"> 4402</span>  outputInfo,</div> |
| <div class="line"><a name="l04403"></a><span class="lineno"> 4403</span>  paramsInfo);</div> |
| <div class="line"><a name="l04404"></a><span class="lineno"> 4404</span>  };</div> |
| <div class="line"><a name="l04405"></a><span class="lineno"> 4405</span>  </div> |
| <div class="line"><a name="l04406"></a><span class="lineno"> 4406</span>  <span class="keywordtype">bool</span> isDynamic = <span class="keyword">false</span>;</div> |
| <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> |
| <div class="line"><a name="l04408"></a><span class="lineno"> 4408</span>  !<a class="code" href="namespacearmnn__driver.html#a208149d5c08f84debf8b265bb508fe02">IsDynamicTensor</a>(outputInfo))</div> |
| <div class="line"><a name="l04409"></a><span class="lineno"> 4409</span>  {</div> |
| <div class="line"><a name="l04410"></a><span class="lineno"> 4410</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l04411"></a><span class="lineno"> 4411</span>  }</div> |
| <div class="line"><a name="l04412"></a><span class="lineno"> 4412</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04413"></a><span class="lineno"> 4413</span>  {</div> |
| <div class="line"><a name="l04414"></a><span class="lineno"> 4414</span>  isDynamic = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l04415"></a><span class="lineno"> 4415</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04416"></a><span class="lineno"> 4416</span>  }</div> |
| <div class="line"><a name="l04417"></a><span class="lineno"> 4417</span>  </div> |
| <div class="line"><a name="l04418"></a><span class="lineno"> 4418</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04419"></a><span class="lineno"> 4419</span>  {</div> |
| <div class="line"><a name="l04420"></a><span class="lineno"> 4420</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04421"></a><span class="lineno"> 4421</span>  }</div> |
| <div class="line"><a name="l04422"></a><span class="lineno"> 4422</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04428"></a><span class="lineno"> 4428</span>  </div> |
| <div class="line"><a name="l04429"></a><span class="lineno"> 4429</span>  <span class="keywordflow">if</span> (!isDynamic)</div> |
| <div class="line"><a name="l04430"></a><span class="lineno"> 4430</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04433"></a><span class="lineno"> 4433</span>  }</div> |
| <div class="line"><a name="l04434"></a><span class="lineno"> 4434</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04435"></a><span class="lineno"> 4435</span>  {</div> |
| <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> |
| <div class="line"><a name="l04437"></a><span class="lineno"> 4437</span>  <a class="code" href="namespacearmnn__driver.html#a5378da53b59c1c826e6320383ca0257e">SetupAndTrackLayerOutputSlot</a>(</div> |
| <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> |
| <div class="line"><a name="l04439"></a><span class="lineno"> 4439</span>  }</div> |
| <div class="line"><a name="l04440"></a><span class="lineno"> 4440</span>  </div> |
| <div class="line"><a name="l04441"></a><span class="lineno"> 4441</span> }</div> |
| <div class="line"><a name="l04442"></a><span class="lineno"> 4442</span>  </div> |
| <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> |
| <div class="line"><a name="l04444"></a><span class="lineno"> 4444</span> {</div> |
| <div class="line"><a name="l04445"></a><span class="lineno"> 4445</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertRank()"</span>;</div> |
| <div class="line"><a name="l04446"></a><span class="lineno"> 4446</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04449"></a><span class="lineno"> 4449</span>  </div> |
| <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> |
| <div class="line"><a name="l04451"></a><span class="lineno"> 4451</span>  {</div> |
| <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> |
| <div class="line"><a name="l04453"></a><span class="lineno"> 4453</span>  }</div> |
| <div class="line"><a name="l04454"></a><span class="lineno"> 4454</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04457"></a><span class="lineno"> 4457</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04460"></a><span class="lineno"> 4460</span>  {</div> |
| <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> |
| <div class="line"><a name="l04462"></a><span class="lineno"> 4462</span>  }</div> |
| <div class="line"><a name="l04463"></a><span class="lineno"> 4463</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04466"></a><span class="lineno"> 4466</span>  {</div> |
| <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> |
| <div class="line"><a name="l04468"></a><span class="lineno"> 4468</span>  }</div> |
| <div class="line"><a name="l04469"></a><span class="lineno"> 4469</span>  </div> |
| <div class="line"><a name="l04470"></a><span class="lineno"> 4470</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04473"></a><span class="lineno"> 4473</span>  IsRankSupported,</div> |
| <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> |
| <div class="line"><a name="l04475"></a><span class="lineno"> 4475</span>  isSupported,</div> |
| <div class="line"><a name="l04476"></a><span class="lineno"> 4476</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l04478"></a><span class="lineno"> 4478</span>  outInfo);</div> |
| <div class="line"><a name="l04479"></a><span class="lineno"> 4479</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04480"></a><span class="lineno"> 4480</span>  {</div> |
| <div class="line"><a name="l04481"></a><span class="lineno"> 4481</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04482"></a><span class="lineno"> 4482</span>  }</div> |
| <div class="line"><a name="l04483"></a><span class="lineno"> 4483</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04486"></a><span class="lineno"> 4486</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l04488"></a><span class="lineno"> 4488</span>  </div> |
| <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> |
| <div class="line"><a name="l04490"></a><span class="lineno"> 4490</span> }</div> |
| <div class="line"><a name="l04491"></a><span class="lineno"> 4491</span>  </div> |
| <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> |
| <div class="line"><a name="l04493"></a><span class="lineno"> 4493</span> {</div> |
| <div class="line"><a name="l04494"></a><span class="lineno"> 4494</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReLu()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04497"></a><span class="lineno"> 4497</span>  </div> |
| <div class="line"><a name="l04498"></a><span class="lineno"> 4498</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04501"></a><span class="lineno"> 4501</span>  {</div> |
| <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> |
| <div class="line"><a name="l04503"></a><span class="lineno"> 4503</span>  }</div> |
| <div class="line"><a name="l04504"></a><span class="lineno"> 4504</span>  </div> |
| <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> |
| <div class="line"><a name="l04506"></a><span class="lineno"> 4506</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l04507"></a><span class="lineno"> 4507</span>  {</div> |
| <div class="line"><a name="l04508"></a><span class="lineno"> 4508</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04509"></a><span class="lineno"> 4509</span>  }</div> |
| <div class="line"><a name="l04510"></a><span class="lineno"> 4510</span>  </div> |
| <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> |
| <div class="line"><a name="l04512"></a><span class="lineno"> 4512</span>  </div> |
| <div class="line"><a name="l04513"></a><span class="lineno"> 4513</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04516"></a><span class="lineno"> 4516</span>  {</div> |
| <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> |
| <div class="line"><a name="l04518"></a><span class="lineno"> 4518</span>  IsActivationSupported,</div> |
| <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> |
| <div class="line"><a name="l04520"></a><span class="lineno"> 4520</span>  isSupported,</div> |
| <div class="line"><a name="l04521"></a><span class="lineno"> 4521</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l04523"></a><span class="lineno"> 4523</span>  outInfo,</div> |
| <div class="line"><a name="l04524"></a><span class="lineno"> 4524</span>  desc);</div> |
| <div class="line"><a name="l04525"></a><span class="lineno"> 4525</span>  };</div> |
| <div class="line"><a name="l04526"></a><span class="lineno"> 4526</span>  </div> |
| <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> |
| <div class="line"><a name="l04528"></a><span class="lineno"> 4528</span>  {</div> |
| <div class="line"><a name="l04529"></a><span class="lineno"> 4529</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04530"></a><span class="lineno"> 4530</span>  }</div> |
| <div class="line"><a name="l04531"></a><span class="lineno"> 4531</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04532"></a><span class="lineno"> 4532</span>  {</div> |
| <div class="line"><a name="l04533"></a><span class="lineno"> 4533</span>  validateFunc(outInfo, isSupported);</div> |
| <div class="line"><a name="l04534"></a><span class="lineno"> 4534</span>  }</div> |
| <div class="line"><a name="l04535"></a><span class="lineno"> 4535</span>  </div> |
| <div class="line"><a name="l04536"></a><span class="lineno"> 4536</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04537"></a><span class="lineno"> 4537</span>  {</div> |
| <div class="line"><a name="l04538"></a><span class="lineno"> 4538</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04539"></a><span class="lineno"> 4539</span>  }</div> |
| <div class="line"><a name="l04540"></a><span class="lineno"> 4540</span>  </div> |
| <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> |
| <div class="line"><a name="l04542"></a><span class="lineno"> 4542</span>  <span class="keywordflow">if</span> (layer == <span class="keyword">nullptr</span>)</div> |
| <div class="line"><a name="l04543"></a><span class="lineno"> 4543</span>  {</div> |
| <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> |
| <div class="line"><a name="l04545"></a><span class="lineno"> 4545</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04548"></a><span class="lineno"> 4548</span>  </div> |
| <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> |
| <div class="line"><a name="l04550"></a><span class="lineno"> 4550</span> }</div> |
| <div class="line"><a name="l04551"></a><span class="lineno"> 4551</span>  </div> |
| <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> |
| <div class="line"><a name="l04553"></a><span class="lineno"> 4553</span> {</div> |
| <div class="line"><a name="l04554"></a><span class="lineno"> 4554</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReLu1()"</span>;</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04559"></a><span class="lineno"> 4559</span>  </div> |
| <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> |
| <div class="line"><a name="l04561"></a><span class="lineno"> 4561</span> }</div> |
| <div class="line"><a name="l04562"></a><span class="lineno"> 4562</span>  </div> |
| <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> |
| <div class="line"><a name="l04564"></a><span class="lineno"> 4564</span> {</div> |
| <div class="line"><a name="l04565"></a><span class="lineno"> 4565</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReLu6()"</span>;</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04569"></a><span class="lineno"> 4569</span>  </div> |
| <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> |
| <div class="line"><a name="l04571"></a><span class="lineno"> 4571</span> }</div> |
| <div class="line"><a name="l04572"></a><span class="lineno"> 4572</span>  </div> |
| <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> |
| <div class="line"><a name="l04574"></a><span class="lineno"> 4574</span> {</div> |
| <div class="line"><a name="l04575"></a><span class="lineno"> 4575</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReshape()"</span>;</div> |
| <div class="line"><a name="l04576"></a><span class="lineno"> 4576</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04580"></a><span class="lineno"> 4580</span>  </div> |
| <div class="line"><a name="l04581"></a><span class="lineno"> 4581</span>  <span class="keywordflow">if</span> (inputOperand == <span class="keyword">nullptr</span></div> |
| <div class="line"><a name="l04582"></a><span class="lineno"> 4582</span>  || requestedShapeOperand == <span class="keyword">nullptr</span></div> |
| <div class="line"><a name="l04583"></a><span class="lineno"> 4583</span>  || outputOperand == <span class="keyword">nullptr</span>)</div> |
| <div class="line"><a name="l04584"></a><span class="lineno"> 4584</span>  {</div> |
| <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> |
| <div class="line"><a name="l04586"></a><span class="lineno"> 4586</span>  }</div> |
| <div class="line"><a name="l04587"></a><span class="lineno"> 4587</span>  </div> |
| <div class="line"><a name="l04588"></a><span class="lineno"> 4588</span>  <span class="keywordflow">if</span> (requestedShapeOperand->dimensions.size() != 1)</div> |
| <div class="line"><a name="l04589"></a><span class="lineno"> 4589</span>  {</div> |
| <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> |
| <div class="line"><a name="l04591"></a><span class="lineno"> 4591</span>  __func__, requestedShapeOperand->dimensions.size());</div> |
| <div class="line"><a name="l04592"></a><span class="lineno"> 4592</span>  }</div> |
| <div class="line"><a name="l04593"></a><span class="lineno"> 4593</span>  </div> |
| <div class="line"><a name="l04594"></a><span class="lineno"> 4594</span>  std::vector<int32_t> targetDimensions;</div> |
| <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> |
| <div class="line"><a name="l04596"></a><span class="lineno"> 4596</span>  {</div> |
| <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> |
| <div class="line"><a name="l04598"></a><span class="lineno"> 4598</span>  }</div> |
| <div class="line"><a name="l04599"></a><span class="lineno"> 4599</span>  </div> |
| <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> |
| <div class="line"><a name="l04601"></a><span class="lineno"> 4601</span>  </div> |
| <div class="line"><a name="l04602"></a><span class="lineno"> 4602</span>  <a class="code" href="namespacearmnn.html#a56943a0946e5f15e5e58054b8e7a04a4a880c1273b27d27cfc82004c3a4b205c9">Shape</a> requestedShape;</div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04606"></a><span class="lineno"> 4606</span>  {</div> |
| <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> |
| <div class="line"><a name="l04608"></a><span class="lineno"> 4608</span>  }</div> |
| <div class="line"><a name="l04609"></a><span class="lineno"> 4609</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04612"></a><span class="lineno"> 4612</span>  {</div> |
| <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> |
| <div class="line"><a name="l04614"></a><span class="lineno"> 4614</span>  }</div> |
| <div class="line"><a name="l04615"></a><span class="lineno"> 4615</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04618"></a><span class="lineno"> 4618</span>  requestedShape.dimensions.data());</div> |
| <div class="line"><a name="l04619"></a><span class="lineno"> 4619</span>  </div> |
| <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> |
| <div class="line"><a name="l04621"></a><span class="lineno"> 4621</span>  </div> |
| <div class="line"><a name="l04622"></a><span class="lineno"> 4622</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04625"></a><span class="lineno"> 4625</span>  {</div> |
| <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> |
| <div class="line"><a name="l04627"></a><span class="lineno"> 4627</span>  IsReshapeSupported,</div> |
| <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> |
| <div class="line"><a name="l04629"></a><span class="lineno"> 4629</span>  isSupported,</div> |
| <div class="line"><a name="l04630"></a><span class="lineno"> 4630</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l04632"></a><span class="lineno"> 4632</span>  outputInfo,</div> |
| <div class="line"><a name="l04633"></a><span class="lineno"> 4633</span>  reshapeDescriptor);</div> |
| <div class="line"><a name="l04634"></a><span class="lineno"> 4634</span>  };</div> |
| <div class="line"><a name="l04635"></a><span class="lineno"> 4635</span>  </div> |
| <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> |
| <div class="line"><a name="l04637"></a><span class="lineno"> 4637</span>  {</div> |
| <div class="line"><a name="l04638"></a><span class="lineno"> 4638</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l04639"></a><span class="lineno"> 4639</span>  }</div> |
| <div class="line"><a name="l04640"></a><span class="lineno"> 4640</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04641"></a><span class="lineno"> 4641</span>  {</div> |
| <div class="line"><a name="l04642"></a><span class="lineno"> 4642</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04643"></a><span class="lineno"> 4643</span>  }</div> |
| <div class="line"><a name="l04644"></a><span class="lineno"> 4644</span>  </div> |
| <div class="line"><a name="l04645"></a><span class="lineno"> 4645</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04646"></a><span class="lineno"> 4646</span>  {</div> |
| <div class="line"><a name="l04647"></a><span class="lineno"> 4647</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04648"></a><span class="lineno"> 4648</span>  }</div> |
| <div class="line"><a name="l04649"></a><span class="lineno"> 4649</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04652"></a><span class="lineno"> 4652</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l04654"></a><span class="lineno"> 4654</span>  </div> |
| <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> |
| <div class="line"><a name="l04656"></a><span class="lineno"> 4656</span> }</div> |
| <div class="line"><a name="l04657"></a><span class="lineno"> 4657</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04661"></a><span class="lineno"> 4661</span>  <a class="code" href="namespacearmnn.html#a9a2af2f8c4af4f9efa8e79417d505ac4">ResizeMethod</a> resizeMethod)</div> |
| <div class="line"><a name="l04662"></a><span class="lineno"> 4662</span> {</div> |
| <div class="line"><a name="l04663"></a><span class="lineno"> 4663</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertResize()"</span>;</div> |
| <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> |
| <div class="line"><a name="l04665"></a><span class="lineno"> 4665</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04668"></a><span class="lineno"> 4668</span>  {</div> |
| <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> |
| <div class="line"><a name="l04670"></a><span class="lineno"> 4670</span>  }</div> |
| <div class="line"><a name="l04671"></a><span class="lineno"> 4671</span>  </div> |
| <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> |
| <div class="line"><a name="l04673"></a><span class="lineno"> 4673</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l04674"></a><span class="lineno"> 4674</span>  {</div> |
| <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> |
| <div class="line"><a name="l04676"></a><span class="lineno"> 4676</span>  }</div> |
| <div class="line"><a name="l04677"></a><span class="lineno"> 4677</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04680"></a><span class="lineno"> 4680</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04684"></a><span class="lineno"> 4684</span>  </div> |
| <div class="line"><a name="l04685"></a><span class="lineno"> 4685</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType1;</div> |
| <div class="line"><a name="l04686"></a><span class="lineno"> 4686</span>  <a class="code" href="namespacearmnn__driver.html#a53bfb6eb38a90244ba166b641a6965d9">OperandType</a> operandType2;</div> |
| <div class="line"><a name="l04687"></a><span class="lineno"> 4687</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04690"></a><span class="lineno"> 4690</span>  {</div> |
| <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> |
| <div class="line"><a name="l04692"></a><span class="lineno"> 4692</span>  }</div> |
| <div class="line"><a name="l04693"></a><span class="lineno"> 4693</span>  </div> |
| <div class="line"><a name="l04694"></a><span class="lineno"> 4694</span>  <span class="keywordflow">if</span> (operandType1 != operandType2)</div> |
| <div class="line"><a name="l04695"></a><span class="lineno"> 4695</span>  {</div> |
| <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> |
| <div class="line"><a name="l04697"></a><span class="lineno"> 4697</span>  }</div> |
| <div class="line"><a name="l04698"></a><span class="lineno"> 4698</span>  </div> |
| <div class="line"><a name="l04699"></a><span class="lineno"> 4699</span>  <span class="keywordflow">if</span> (operandType1 == OperandType::INT32)</div> |
| <div class="line"><a name="l04700"></a><span class="lineno"> 4700</span>  {</div> |
| <div class="line"><a name="l04701"></a><span class="lineno"> 4701</span>  <span class="comment">// Case 1: resizing by shape</span></div> |
| <div class="line"><a name="l04702"></a><span class="lineno"> 4702</span>  int32_t targetWidth = 0;</div> |
| <div class="line"><a name="l04703"></a><span class="lineno"> 4703</span>  int32_t targetHeight = 0;</div> |
| <div class="line"><a name="l04704"></a><span class="lineno"> 4704</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04707"></a><span class="lineno"> 4707</span>  {</div> |
| <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> |
| <div class="line"><a name="l04709"></a><span class="lineno"> 4709</span>  }</div> |
| <div class="line"><a name="l04710"></a><span class="lineno"> 4710</span>  </div> |
| <div class="line"><a name="l04711"></a><span class="lineno"> 4711</span>  <span class="keywordflow">if</span> (targetWidth < 0 || targetHeight < 0)</div> |
| <div class="line"><a name="l04712"></a><span class="lineno"> 4712</span>  {</div> |
| <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> |
| <div class="line"><a name="l04714"></a><span class="lineno"> 4714</span>  <span class="stringliteral">"Target width/height cannot be < 0"</span>, __func__);</div> |
| <div class="line"><a name="l04715"></a><span class="lineno"> 4715</span>  }</div> |
| <div class="line"><a name="l04716"></a><span class="lineno"> 4716</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04719"></a><span class="lineno"> 4719</span>  }</div> |
| <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> |
| <div class="line"><a name="l04721"></a><span class="lineno"> 4721</span>  {</div> |
| <div class="line"><a name="l04722"></a><span class="lineno"> 4722</span>  <span class="comment">// Case 2: resizing by scale</span></div> |
| <div class="line"><a name="l04723"></a><span class="lineno"> 4723</span>  <span class="keywordtype">float</span> widthScale = 1.0f;</div> |
| <div class="line"><a name="l04724"></a><span class="lineno"> 4724</span>  <span class="keywordtype">float</span> heightScale = 1.0f;</div> |
| <div class="line"><a name="l04725"></a><span class="lineno"> 4725</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04728"></a><span class="lineno"> 4728</span>  {</div> |
| <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> |
| <div class="line"><a name="l04730"></a><span class="lineno"> 4730</span>  }</div> |
| <div class="line"><a name="l04731"></a><span class="lineno"> 4731</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04734"></a><span class="lineno"> 4734</span>  </div> |
| <div class="line"><a name="l04735"></a><span class="lineno"> 4735</span>  <span class="keywordtype">float</span> width = inputShape[dataLayoutIndexed.GetWidthIndex()];</div> |
| <div class="line"><a name="l04736"></a><span class="lineno"> 4736</span>  <span class="keywordtype">float</span> height = inputShape[dataLayoutIndexed.GetHeightIndex()];</div> |
| <div class="line"><a name="l04737"></a><span class="lineno"> 4737</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04740"></a><span class="lineno"> 4740</span>  }</div> |
| <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> |
| <div class="line"><a name="l04742"></a><span class="lineno"> 4742</span>  {</div> |
| <div class="line"><a name="l04743"></a><span class="lineno"> 4743</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> widthScale;</div> |
| <div class="line"><a name="l04744"></a><span class="lineno"> 4744</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> heightScale;</div> |
| <div class="line"><a name="l04745"></a><span class="lineno"> 4745</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04748"></a><span class="lineno"> 4748</span>  {</div> |
| <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> |
| <div class="line"><a name="l04750"></a><span class="lineno"> 4750</span>  }</div> |
| <div class="line"><a name="l04751"></a><span class="lineno"> 4751</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04754"></a><span class="lineno"> 4754</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04757"></a><span class="lineno"> 4757</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04760"></a><span class="lineno"> 4760</span>  }</div> |
| <div class="line"><a name="l04761"></a><span class="lineno"> 4761</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04762"></a><span class="lineno"> 4762</span>  {</div> |
| <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> |
| <div class="line"><a name="l04764"></a><span class="lineno"> 4764</span>  }</div> |
| <div class="line"><a name="l04765"></a><span class="lineno"> 4765</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04768"></a><span class="lineno"> 4768</span>  </div> |
| <div class="line"><a name="l04769"></a><span class="lineno"> 4769</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04772"></a><span class="lineno"> 4772</span>  {</div> |
| <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> |
| <div class="line"><a name="l04774"></a><span class="lineno"> 4774</span>  IsResizeSupported,</div> |
| <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> |
| <div class="line"><a name="l04776"></a><span class="lineno"> 4776</span>  isSupported,</div> |
| <div class="line"><a name="l04777"></a><span class="lineno"> 4777</span>  setBackend,</div> |
| <div class="line"><a name="l04778"></a><span class="lineno"> 4778</span>  inputInfo,</div> |
| <div class="line"><a name="l04779"></a><span class="lineno"> 4779</span>  outputInfo,</div> |
| <div class="line"><a name="l04780"></a><span class="lineno"> 4780</span>  descriptor);</div> |
| <div class="line"><a name="l04781"></a><span class="lineno"> 4781</span>  };</div> |
| <div class="line"><a name="l04782"></a><span class="lineno"> 4782</span>  </div> |
| <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> |
| <div class="line"><a name="l04784"></a><span class="lineno"> 4784</span>  {</div> |
| <div class="line"><a name="l04785"></a><span class="lineno"> 4785</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04786"></a><span class="lineno"> 4786</span>  }</div> |
| <div class="line"><a name="l04787"></a><span class="lineno"> 4787</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04788"></a><span class="lineno"> 4788</span>  {</div> |
| <div class="line"><a name="l04789"></a><span class="lineno"> 4789</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l04790"></a><span class="lineno"> 4790</span>  }</div> |
| <div class="line"><a name="l04791"></a><span class="lineno"> 4791</span>  </div> |
| <div class="line"><a name="l04792"></a><span class="lineno"> 4792</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04793"></a><span class="lineno"> 4793</span>  {</div> |
| <div class="line"><a name="l04794"></a><span class="lineno"> 4794</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04795"></a><span class="lineno"> 4795</span>  }</div> |
| <div class="line"><a name="l04796"></a><span class="lineno"> 4796</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04799"></a><span class="lineno"> 4799</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l04801"></a><span class="lineno"> 4801</span>  </div> |
| <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> |
| <div class="line"><a name="l04803"></a><span class="lineno"> 4803</span> }</div> |
| <div class="line"><a name="l04804"></a><span class="lineno"> 4804</span>  </div> |
| <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> |
| <div class="line"><a name="l04806"></a><span class="lineno"> 4806</span> {</div> |
| <div class="line"><a name="l04807"></a><span class="lineno"> 4807</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertReverseV2()"</span>;</div> |
| <div class="line"><a name="l04808"></a><span class="lineno"> 4808</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04812"></a><span class="lineno"> 4812</span>  {</div> |
| <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> |
| <div class="line"><a name="l04814"></a><span class="lineno"> 4814</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04817"></a><span class="lineno"> 4817</span>  </div> |
| <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> |
| <div class="line"><a name="l04819"></a><span class="lineno"> 4819</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l04820"></a><span class="lineno"> 4820</span>  {</div> |
| <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> |
| <div class="line"><a name="l04822"></a><span class="lineno"> 4822</span>  }</div> |
| <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> |
| <div class="line"><a name="l04824"></a><span class="lineno"> 4824</span>  </div> |
| <div class="line"><a name="l04825"></a><span class="lineno"> 4825</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04828"></a><span class="lineno"> 4828</span>  {</div> |
| <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> |
| <div class="line"><a name="l04830"></a><span class="lineno"> 4830</span>  IsReverseV2Supported,</div> |
| <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> |
| <div class="line"><a name="l04832"></a><span class="lineno"> 4832</span>  isSupported,</div> |
| <div class="line"><a name="l04833"></a><span class="lineno"> 4833</span>  setBackend,</div> |
| <div class="line"><a name="l04834"></a><span class="lineno"> 4834</span>  inputInfo0,</div> |
| <div class="line"><a name="l04835"></a><span class="lineno"> 4835</span>  inputInfo1,</div> |
| <div class="line"><a name="l04836"></a><span class="lineno"> 4836</span>  outputInfo);</div> |
| <div class="line"><a name="l04837"></a><span class="lineno"> 4837</span>  };</div> |
| <div class="line"><a name="l04838"></a><span class="lineno"> 4838</span>  </div> |
| <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> |
| <div class="line"><a name="l04840"></a><span class="lineno"> 4840</span>  {</div> |
| <div class="line"><a name="l04841"></a><span class="lineno"> 4841</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l04842"></a><span class="lineno"> 4842</span>  }</div> |
| <div class="line"><a name="l04843"></a><span class="lineno"> 4843</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04844"></a><span class="lineno"> 4844</span>  {</div> |
| <div class="line"><a name="l04845"></a><span class="lineno"> 4845</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04846"></a><span class="lineno"> 4846</span>  }</div> |
| <div class="line"><a name="l04847"></a><span class="lineno"> 4847</span>  </div> |
| <div class="line"><a name="l04848"></a><span class="lineno"> 4848</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l04849"></a><span class="lineno"> 4849</span>  {</div> |
| <div class="line"><a name="l04850"></a><span class="lineno"> 4850</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04851"></a><span class="lineno"> 4851</span>  }</div> |
| <div class="line"><a name="l04852"></a><span class="lineno"> 4852</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04855"></a><span class="lineno"> 4855</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04858"></a><span class="lineno"> 4858</span>  </div> |
| <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> |
| <div class="line"><a name="l04860"></a><span class="lineno"> 4860</span> }</div> |
| <div class="line"><a name="l04861"></a><span class="lineno"> 4861</span>  </div> |
| <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> |
| <div class="line"><a name="l04863"></a><span class="lineno"> 4863</span> {</div> |
| <div class="line"><a name="l04864"></a><span class="lineno"> 4864</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSpaceToBatchNd()"</span>;</div> |
| <div class="line"><a name="l04865"></a><span class="lineno"> 4865</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04868"></a><span class="lineno"> 4868</span>  {</div> |
| <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> |
| <div class="line"><a name="l04870"></a><span class="lineno"> 4870</span>  }</div> |
| <div class="line"><a name="l04871"></a><span class="lineno"> 4871</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04875"></a><span class="lineno"> 4875</span>  </div> |
| <div class="line"><a name="l04876"></a><span class="lineno"> 4876</span>  <span class="keywordflow">if</span>(rank != 4)</div> |
| <div class="line"><a name="l04877"></a><span class="lineno"> 4877</span>  {</div> |
| <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> |
| <div class="line"><a name="l04879"></a><span class="lineno"> 4879</span>  }</div> |
| <div class="line"><a name="l04880"></a><span class="lineno"> 4880</span>  </div> |
| <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> |
| <div class="line"><a name="l04882"></a><span class="lineno"> 4882</span>  <span class="keywordflow">if</span>(!output)</div> |
| <div class="line"><a name="l04883"></a><span class="lineno"> 4883</span>  {</div> |
| <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> |
| <div class="line"><a name="l04885"></a><span class="lineno"> 4885</span>  }</div> |
| <div class="line"><a name="l04886"></a><span class="lineno"> 4886</span>  </div> |
| <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> |
| <div class="line"><a name="l04888"></a><span class="lineno"> 4888</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04891"></a><span class="lineno"> 4891</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04894"></a><span class="lineno"> 4894</span>  {</div> |
| <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> |
| <div class="line"><a name="l04896"></a><span class="lineno"> 4896</span>  }</div> |
| <div class="line"><a name="l04897"></a><span class="lineno"> 4897</span>  </div> |
| <div class="line"><a name="l04898"></a><span class="lineno"> 4898</span>  std::vector<int32_t> blockShape;</div> |
| <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> |
| <div class="line"><a name="l04900"></a><span class="lineno"> 4900</span>  {</div> |
| <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> |
| <div class="line"><a name="l04902"></a><span class="lineno"> 4902</span>  }</div> |
| <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> |
| <div class="line"><a name="l04904"></a><span class="lineno"> 4904</span>  { return i < 1; }))</div> |
| <div class="line"><a name="l04905"></a><span class="lineno"> 4905</span>  {</div> |
| <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> |
| <div class="line"><a name="l04907"></a><span class="lineno"> 4907</span>  }</div> |
| <div class="line"><a name="l04908"></a><span class="lineno"> 4908</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04911"></a><span class="lineno"> 4911</span>  {</div> |
| <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> |
| <div class="line"><a name="l04913"></a><span class="lineno"> 4913</span>  }</div> |
| <div class="line"><a name="l04914"></a><span class="lineno"> 4914</span>  </div> |
| <div class="line"><a name="l04915"></a><span class="lineno"> 4915</span>  std::vector<std::pair<unsigned int, unsigned int>> paddingList;</div> |
| <div class="line"><a name="l04916"></a><span class="lineno"> 4916</span>  std::vector<int32_t> paddings;</div> |
| <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> |
| <div class="line"><a name="l04918"></a><span class="lineno"> 4918</span>  {</div> |
| <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> |
| <div class="line"><a name="l04920"></a><span class="lineno"> 4920</span>  }</div> |
| <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> |
| <div class="line"><a name="l04922"></a><span class="lineno"> 4922</span>  {</div> |
| <div class="line"><a name="l04923"></a><span class="lineno"> 4923</span>  <span class="keywordtype">int</span> paddingBeforeInput = paddings[i];</div> |
| <div class="line"><a name="l04924"></a><span class="lineno"> 4924</span>  <span class="keywordtype">int</span> paddingAfterInput = paddings[i + 1];</div> |
| <div class="line"><a name="l04925"></a><span class="lineno"> 4925</span>  <span class="keywordflow">if</span>(paddingBeforeInput < 0 || paddingAfterInput < 0)</div> |
| <div class="line"><a name="l04926"></a><span class="lineno"> 4926</span>  {</div> |
| <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> |
| <div class="line"><a name="l04928"></a><span class="lineno"> 4928</span>  }</div> |
| <div class="line"><a name="l04929"></a><span class="lineno"> 4929</span>  </div> |
| <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> |
| <div class="line"><a name="l04931"></a><span class="lineno"> 4931</span>  }</div> |
| <div class="line"><a name="l04932"></a><span class="lineno"> 4932</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l04937"></a><span class="lineno"> 4937</span>  </div> |
| <div class="line"><a name="l04938"></a><span class="lineno"> 4938</span>  <span class="keywordflow">if</span>(Is12OrLaterOperand(*output))</div> |
| <div class="line"><a name="l04939"></a><span class="lineno"> 4939</span>  {</div> |
| <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> |
| <div class="line"><a name="l04941"></a><span class="lineno"> 4941</span>  }</div> |
| <div class="line"><a name="l04942"></a><span class="lineno"> 4942</span>  </div> |
| <div class="line"><a name="l04943"></a><span class="lineno"> 4943</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l04946"></a><span class="lineno"> 4946</span>  {</div> |
| <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> |
| <div class="line"><a name="l04948"></a><span class="lineno"> 4948</span>  IsSpaceToBatchNdSupported,</div> |
| <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> |
| <div class="line"><a name="l04950"></a><span class="lineno"> 4950</span>  isSupported,</div> |
| <div class="line"><a name="l04951"></a><span class="lineno"> 4951</span>  setBackend,</div> |
| <div class="line"><a name="l04952"></a><span class="lineno"> 4952</span>  inputInfo,</div> |
| <div class="line"><a name="l04953"></a><span class="lineno"> 4953</span>  outputInfo,</div> |
| <div class="line"><a name="l04954"></a><span class="lineno"> 4954</span>  descriptor);</div> |
| <div class="line"><a name="l04955"></a><span class="lineno"> 4955</span>  };</div> |
| <div class="line"><a name="l04956"></a><span class="lineno"> 4956</span>  </div> |
| <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> |
| <div class="line"><a name="l04958"></a><span class="lineno"> 4958</span>  {</div> |
| <div class="line"><a name="l04959"></a><span class="lineno"> 4959</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l04960"></a><span class="lineno"> 4960</span>  } <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l04961"></a><span class="lineno"> 4961</span>  {</div> |
| <div class="line"><a name="l04962"></a><span class="lineno"> 4962</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l04963"></a><span class="lineno"> 4963</span>  }</div> |
| <div class="line"><a name="l04964"></a><span class="lineno"> 4964</span>  </div> |
| <div class="line"><a name="l04965"></a><span class="lineno"> 4965</span>  <span class="keywordflow">if</span>(!isSupported)</div> |
| <div class="line"><a name="l04966"></a><span class="lineno"> 4966</span>  {</div> |
| <div class="line"><a name="l04967"></a><span class="lineno"> 4967</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l04968"></a><span class="lineno"> 4968</span>  }</div> |
| <div class="line"><a name="l04969"></a><span class="lineno"> 4969</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04972"></a><span class="lineno"> 4972</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l04974"></a><span class="lineno"> 4974</span>  </div> |
| <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> |
| <div class="line"><a name="l04976"></a><span class="lineno"> 4976</span> }</div> |
| <div class="line"><a name="l04977"></a><span class="lineno"> 4977</span>  </div> |
| <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> |
| <div class="line"><a name="l04979"></a><span class="lineno"> 4979</span> {</div> |
| <div class="line"><a name="l04980"></a><span class="lineno"> 4980</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSpaceToDepth()"</span>;</div> |
| <div class="line"><a name="l04981"></a><span class="lineno"> 4981</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04984"></a><span class="lineno"> 4984</span>  {</div> |
| <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> |
| <div class="line"><a name="l04986"></a><span class="lineno"> 4986</span>  }</div> |
| <div class="line"><a name="l04987"></a><span class="lineno"> 4987</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l04990"></a><span class="lineno"> 4990</span>  <span class="keywordflow">if</span> (rank != 4)</div> |
| <div class="line"><a name="l04991"></a><span class="lineno"> 4991</span>  {</div> |
| <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> |
| <div class="line"><a name="l04993"></a><span class="lineno"> 4993</span>  }</div> |
| <div class="line"><a name="l04994"></a><span class="lineno"> 4994</span>  </div> |
| <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> |
| <div class="line"><a name="l04996"></a><span class="lineno"> 4996</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l04997"></a><span class="lineno"> 4997</span>  {</div> |
| <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> |
| <div class="line"><a name="l04999"></a><span class="lineno"> 4999</span>  }</div> |
| <div class="line"><a name="l05000"></a><span class="lineno"> 5000</span>  </div> |
| <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> |
| <div class="line"><a name="l05002"></a><span class="lineno"> 5002</span>  </div> |
| <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> |
| <div class="line"><a name="l05004"></a><span class="lineno"> 5004</span>  </div> |
| <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> |
| <div class="line"><a name="l05006"></a><span class="lineno"> 5006</span>  </div> |
| <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> |
| <div class="line"><a name="l05008"></a><span class="lineno"> 5008</span>  {</div> |
| <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> |
| <div class="line"><a name="l05010"></a><span class="lineno"> 5010</span>  }</div> |
| <div class="line"><a name="l05011"></a><span class="lineno"> 5011</span>  </div> |
| <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> |
| <div class="line"><a name="l05013"></a><span class="lineno"> 5013</span>  </div> |
| <div class="line"><a name="l05014"></a><span class="lineno"> 5014</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05017"></a><span class="lineno"> 5017</span>  {</div> |
| <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> |
| <div class="line"><a name="l05019"></a><span class="lineno"> 5019</span>  IsSpaceToDepthSupported,</div> |
| <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> |
| <div class="line"><a name="l05021"></a><span class="lineno"> 5021</span>  isSupported,</div> |
| <div class="line"><a name="l05022"></a><span class="lineno"> 5022</span>  setBackend,</div> |
| <div class="line"><a name="l05023"></a><span class="lineno"> 5023</span>  inputInfo,</div> |
| <div class="line"><a name="l05024"></a><span class="lineno"> 5024</span>  outputInfo,</div> |
| <div class="line"><a name="l05025"></a><span class="lineno"> 5025</span>  desc);</div> |
| <div class="line"><a name="l05026"></a><span class="lineno"> 5026</span>  };</div> |
| <div class="line"><a name="l05027"></a><span class="lineno"> 5027</span>  </div> |
| <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> |
| <div class="line"><a name="l05029"></a><span class="lineno"> 5029</span>  {</div> |
| <div class="line"><a name="l05030"></a><span class="lineno"> 5030</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l05031"></a><span class="lineno"> 5031</span>  }</div> |
| <div class="line"><a name="l05032"></a><span class="lineno"> 5032</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05033"></a><span class="lineno"> 5033</span>  {</div> |
| <div class="line"><a name="l05034"></a><span class="lineno"> 5034</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l05035"></a><span class="lineno"> 5035</span>  }</div> |
| <div class="line"><a name="l05036"></a><span class="lineno"> 5036</span>  </div> |
| <div class="line"><a name="l05037"></a><span class="lineno"> 5037</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05038"></a><span class="lineno"> 5038</span>  {</div> |
| <div class="line"><a name="l05039"></a><span class="lineno"> 5039</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05040"></a><span class="lineno"> 5040</span>  }</div> |
| <div class="line"><a name="l05041"></a><span class="lineno"> 5041</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05044"></a><span class="lineno"> 5044</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l05046"></a><span class="lineno"> 5046</span>  </div> |
| <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> |
| <div class="line"><a name="l05048"></a><span class="lineno"> 5048</span> }</div> |
| <div class="line"><a name="l05049"></a><span class="lineno"> 5049</span>  </div> |
| <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> |
| <div class="line"><a name="l05051"></a><span class="lineno"> 5051</span> {</div> |
| <div class="line"><a name="l05052"></a><span class="lineno"> 5052</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSoftmax()"</span>;</div> |
| <div class="line"><a name="l05053"></a><span class="lineno"> 5053</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05056"></a><span class="lineno"> 5056</span>  {</div> |
| <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> |
| <div class="line"><a name="l05058"></a><span class="lineno"> 5058</span>  }</div> |
| <div class="line"><a name="l05059"></a><span class="lineno"> 5059</span>  </div> |
| <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> |
| <div class="line"><a name="l05061"></a><span class="lineno"> 5061</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l05062"></a><span class="lineno"> 5062</span>  {</div> |
| <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> |
| <div class="line"><a name="l05064"></a><span class="lineno"> 5064</span>  }</div> |
| <div class="line"><a name="l05065"></a><span class="lineno"> 5065</span>  </div> |
| <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> |
| <div class="line"><a name="l05067"></a><span class="lineno"> 5067</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05070"></a><span class="lineno"> 5070</span>  </div> |
| <div class="line"><a name="l05071"></a><span class="lineno"> 5071</span>  <span class="comment">// Read beta value</span></div> |
| <div class="line"><a name="l05072"></a><span class="lineno"> 5072</span>  <span class="keywordflow">if</span> (outputType == OperandType::TENSOR_FLOAT16)</div> |
| <div class="line"><a name="l05073"></a><span class="lineno"> 5073</span>  {</div> |
| <div class="line"><a name="l05074"></a><span class="lineno"> 5074</span>  <a class="code" href="namespacearmnn.html#a0b49aa352b84d572942185ce72cef751">Half</a> value;</div> |
| <div class="line"><a name="l05075"></a><span class="lineno"> 5075</span>  </div> |
| <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> |
| <div class="line"><a name="l05077"></a><span class="lineno"> 5077</span>  {</div> |
| <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> |
| <div class="line"><a name="l05079"></a><span class="lineno"> 5079</span>  }</div> |
| <div class="line"><a name="l05080"></a><span class="lineno"> 5080</span>  </div> |
| <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> |
| <div class="line"><a name="l05082"></a><span class="lineno"> 5082</span>  }</div> |
| <div class="line"><a name="l05083"></a><span class="lineno"> 5083</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05084"></a><span class="lineno"> 5084</span>  {</div> |
| <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> |
| <div class="line"><a name="l05086"></a><span class="lineno"> 5086</span>  {</div> |
| <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> |
| <div class="line"><a name="l05088"></a><span class="lineno"> 5088</span>  }</div> |
| <div class="line"><a name="l05089"></a><span class="lineno"> 5089</span>  }</div> |
| <div class="line"><a name="l05090"></a><span class="lineno"> 5090</span>  </div> |
| <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> |
| <div class="line"><a name="l05092"></a><span class="lineno"> 5092</span>  2,</div> |
| <div class="line"><a name="l05093"></a><span class="lineno"> 5093</span>  OperandType::INT32,</div> |
| <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> |
| <div class="line"><a name="l05095"></a><span class="lineno"> 5095</span>  model,</div> |
| <div class="line"><a name="l05096"></a><span class="lineno"> 5096</span>  data))</div> |
| <div class="line"><a name="l05097"></a><span class="lineno"> 5097</span>  {</div> |
| <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> |
| <div class="line"><a name="l05099"></a><span class="lineno"> 5099</span>  }</div> |
| <div class="line"><a name="l05100"></a><span class="lineno"> 5100</span>  </div> |
| <div class="line"><a name="l05101"></a><span class="lineno"> 5101</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05104"></a><span class="lineno"> 5104</span>  {</div> |
| <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> |
| <div class="line"><a name="l05106"></a><span class="lineno"> 5106</span>  IsSoftmaxSupported,</div> |
| <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> |
| <div class="line"><a name="l05108"></a><span class="lineno"> 5108</span>  isSupported,</div> |
| <div class="line"><a name="l05109"></a><span class="lineno"> 5109</span>  setBackend,</div> |
| <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> |
| <div class="line"><a name="l05111"></a><span class="lineno"> 5111</span>  outputInfo,</div> |
| <div class="line"><a name="l05112"></a><span class="lineno"> 5112</span>  desc);</div> |
| <div class="line"><a name="l05113"></a><span class="lineno"> 5113</span>  };</div> |
| <div class="line"><a name="l05114"></a><span class="lineno"> 5114</span>  </div> |
| <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> |
| <div class="line"><a name="l05116"></a><span class="lineno"> 5116</span>  {</div> |
| <div class="line"><a name="l05117"></a><span class="lineno"> 5117</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l05118"></a><span class="lineno"> 5118</span>  }</div> |
| <div class="line"><a name="l05119"></a><span class="lineno"> 5119</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05120"></a><span class="lineno"> 5120</span>  {</div> |
| <div class="line"><a name="l05121"></a><span class="lineno"> 5121</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l05122"></a><span class="lineno"> 5122</span>  }</div> |
| <div class="line"><a name="l05123"></a><span class="lineno"> 5123</span>  </div> |
| <div class="line"><a name="l05124"></a><span class="lineno"> 5124</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05125"></a><span class="lineno"> 5125</span>  {</div> |
| <div class="line"><a name="l05126"></a><span class="lineno"> 5126</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05127"></a><span class="lineno"> 5127</span>  }</div> |
| <div class="line"><a name="l05128"></a><span class="lineno"> 5128</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05131"></a><span class="lineno"> 5131</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l05133"></a><span class="lineno"> 5133</span>  </div> |
| <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> |
| <div class="line"><a name="l05135"></a><span class="lineno"> 5135</span> }</div> |
| <div class="line"><a name="l05136"></a><span class="lineno"> 5136</span>  </div> |
| <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> |
| <div class="line"><a name="l05138"></a><span class="lineno"> 5138</span> {</div> |
| <div class="line"><a name="l05139"></a><span class="lineno"> 5139</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTanH()"</span>;</div> |
| <div class="line"><a name="l05140"></a><span class="lineno"> 5140</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05145"></a><span class="lineno"> 5145</span>  </div> |
| <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> |
| <div class="line"><a name="l05147"></a><span class="lineno"> 5147</span> }</div> |
| <div class="line"><a name="l05148"></a><span class="lineno"> 5148</span>  </div> |
| <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> |
| <div class="line"><a name="l05150"></a><span class="lineno"> 5150</span> {</div> |
| <div class="line"><a name="l05151"></a><span class="lineno"> 5151</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTile()"</span>;</div> |
| <div class="line"><a name="l05152"></a><span class="lineno"> 5152</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05155"></a><span class="lineno"> 5155</span>  {</div> |
| <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> |
| <div class="line"><a name="l05157"></a><span class="lineno"> 5157</span>  }</div> |
| <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> |
| <div class="line"><a name="l05159"></a><span class="lineno"> 5159</span>  </div> |
| <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> |
| <div class="line"><a name="l05161"></a><span class="lineno"> 5161</span>  <span class="keywordflow">if</span> (!outputOperand)</div> |
| <div class="line"><a name="l05162"></a><span class="lineno"> 5162</span>  {</div> |
| <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> |
| <div class="line"><a name="l05164"></a><span class="lineno"> 5164</span>  }</div> |
| <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> |
| <div class="line"><a name="l05166"></a><span class="lineno"> 5166</span>  </div> |
| <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> |
| <div class="line"><a name="l05168"></a><span class="lineno"> 5168</span>  <span class="keywordflow">if</span> (!multiplesOperand)</div> |
| <div class="line"><a name="l05169"></a><span class="lineno"> 5169</span>  {</div> |
| <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> |
| <div class="line"><a name="l05171"></a><span class="lineno"> 5171</span>  }</div> |
| <div class="line"><a name="l05172"></a><span class="lineno"> 5172</span>  std::vector<int32_t> multiples;</div> |
| <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> |
| <div class="line"><a name="l05174"></a><span class="lineno"> 5174</span>  {</div> |
| <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> |
| <div class="line"><a name="l05176"></a><span class="lineno"> 5176</span>  }</div> |
| <div class="line"><a name="l05177"></a><span class="lineno"> 5177</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05180"></a><span class="lineno"> 5180</span>  </div> |
| <div class="line"><a name="l05181"></a><span class="lineno"> 5181</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05184"></a><span class="lineno"> 5184</span>  {</div> |
| <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> |
| <div class="line"><a name="l05186"></a><span class="lineno"> 5186</span>  IsTileSupported,</div> |
| <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> |
| <div class="line"><a name="l05188"></a><span class="lineno"> 5188</span>  isSupported,</div> |
| <div class="line"><a name="l05189"></a><span class="lineno"> 5189</span>  setBackend,</div> |
| <div class="line"><a name="l05190"></a><span class="lineno"> 5190</span>  inputInfo,</div> |
| <div class="line"><a name="l05191"></a><span class="lineno"> 5191</span>  outputInfo,</div> |
| <div class="line"><a name="l05192"></a><span class="lineno"> 5192</span>  descriptor);</div> |
| <div class="line"><a name="l05193"></a><span class="lineno"> 5193</span>  };</div> |
| <div class="line"><a name="l05194"></a><span class="lineno"> 5194</span>  </div> |
| <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> |
| <div class="line"><a name="l05196"></a><span class="lineno"> 5196</span>  {</div> |
| <div class="line"><a name="l05197"></a><span class="lineno"> 5197</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l05198"></a><span class="lineno"> 5198</span>  }</div> |
| <div class="line"><a name="l05199"></a><span class="lineno"> 5199</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05200"></a><span class="lineno"> 5200</span>  {</div> |
| <div class="line"><a name="l05201"></a><span class="lineno"> 5201</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l05202"></a><span class="lineno"> 5202</span>  }</div> |
| <div class="line"><a name="l05203"></a><span class="lineno"> 5203</span>  </div> |
| <div class="line"><a name="l05204"></a><span class="lineno"> 5204</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05205"></a><span class="lineno"> 5205</span>  {</div> |
| <div class="line"><a name="l05206"></a><span class="lineno"> 5206</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05207"></a><span class="lineno"> 5207</span>  }</div> |
| <div class="line"><a name="l05208"></a><span class="lineno"> 5208</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05211"></a><span class="lineno"> 5211</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l05213"></a><span class="lineno"> 5213</span>  </div> |
| <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> |
| <div class="line"><a name="l05215"></a><span class="lineno"> 5215</span> }</div> |
| <div class="line"><a name="l05216"></a><span class="lineno"> 5216</span>  </div> |
| <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> |
| <div class="line"><a name="l05218"></a><span class="lineno"> 5218</span> {</div> |
| <div class="line"><a name="l05219"></a><span class="lineno"> 5219</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTransposeConv2d()"</span>;</div> |
| <div class="line"><a name="l05220"></a><span class="lineno"> 5220</span>  </div> |
| <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> |
| <div class="line"><a name="l05222"></a><span class="lineno"> 5222</span>  </div> |
| <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> |
| <div class="line"><a name="l05224"></a><span class="lineno"> 5224</span>  {</div> |
| <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> |
| <div class="line"><a name="l05226"></a><span class="lineno"> 5226</span>  }</div> |
| <div class="line"><a name="l05227"></a><span class="lineno"> 5227</span>  </div> |
| <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> |
| <div class="line"><a name="l05229"></a><span class="lineno"> 5229</span>  </div> |
| <div class="line"><a name="l05230"></a><span class="lineno"> 5230</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l05231"></a><span class="lineno"> 5231</span>  {</div> |
| <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> |
| <div class="line"><a name="l05233"></a><span class="lineno"> 5233</span>  }</div> |
| <div class="line"><a name="l05234"></a><span class="lineno"> 5234</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05237"></a><span class="lineno"> 5237</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05241"></a><span class="lineno"> 5241</span>  </div> |
| <div class="line"><a name="l05242"></a><span class="lineno"> 5242</span>  <span class="keywordflow">if</span> (weightsOperand == <span class="keyword">nullptr</span>)</div> |
| <div class="line"><a name="l05243"></a><span class="lineno"> 5243</span>  {</div> |
| <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> |
| <div class="line"><a name="l05245"></a><span class="lineno"> 5245</span>  }</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05248"></a><span class="lineno"> 5248</span>  </div> |
| <div class="line"><a name="l05249"></a><span class="lineno"> 5249</span>  <span class="comment">// Determine whether padding is implicit or explicit</span></div> |
| <div class="line"><a name="l05250"></a><span class="lineno"> 5250</span>  <span class="keywordtype">bool</span> implicitPadding = operation.inputs.size() == 9;</div> |
| <div class="line"><a name="l05251"></a><span class="lineno"> 5251</span>  </div> |
| <div class="line"><a name="l05252"></a><span class="lineno"> 5252</span>  <span class="keywordflow">if</span> (implicitPadding )</div> |
| <div class="line"><a name="l05253"></a><span class="lineno"> 5253</span>  {</div> |
| <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> |
| <div class="line"><a name="l05255"></a><span class="lineno"> 5255</span>  }</div> |
| <div class="line"><a name="l05256"></a><span class="lineno"> 5256</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05257"></a><span class="lineno"> 5257</span>  {</div> |
| <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> |
| <div class="line"><a name="l05259"></a><span class="lineno"> 5259</span>  }</div> |
| <div class="line"><a name="l05260"></a><span class="lineno"> 5260</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05264"></a><span class="lineno"> 5264</span>  </div> |
| <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> |
| <div class="line"><a name="l05266"></a><span class="lineno"> 5266</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05271"></a><span class="lineno"> 5271</span>  model, data, OHWIToOIHW) :</div> |
| <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> |
| <div class="line"><a name="l05273"></a><span class="lineno"> 5273</span>  </div> |
| <div class="line"><a name="l05274"></a><span class="lineno"> 5274</span>  <span class="comment">// Bias is a 1D tensor</span></div> |
| <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> |
| <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> |
| <div class="line"><a name="l05277"></a><span class="lineno"> 5277</span>  </div> |
| <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> |
| <div class="line"><a name="l05279"></a><span class="lineno"> 5279</span>  {</div> |
| <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> |
| <div class="line"><a name="l05281"></a><span class="lineno"> 5281</span>  }</div> |
| <div class="line"><a name="l05282"></a><span class="lineno"> 5282</span>  </div> |
| <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> |
| <div class="line"><a name="l05284"></a><span class="lineno"> 5284</span>  {</div> |
| <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> |
| <div class="line"><a name="l05286"></a><span class="lineno"> 5286</span>  }</div> |
| <div class="line"><a name="l05287"></a><span class="lineno"> 5287</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05291"></a><span class="lineno"> 5291</span>  </div> |
| <div class="line"><a name="l05292"></a><span class="lineno"> 5292</span>  ActivationFn activation;</div> |
| <div class="line"><a name="l05293"></a><span class="lineno"> 5293</span>  </div> |
| <div class="line"><a name="l05294"></a><span class="lineno"> 5294</span>  <span class="keywordflow">if</span> (implicitPadding)</div> |
| <div class="line"><a name="l05295"></a><span class="lineno"> 5295</span>  {</div> |
| <div class="line"><a name="l05296"></a><span class="lineno"> 5296</span>  int32_t strideX{0};</div> |
| <div class="line"><a name="l05297"></a><span class="lineno"> 5297</span>  int32_t strideY{0};</div> |
| <div class="line"><a name="l05298"></a><span class="lineno"> 5298</span>  int32_t padLeft{0};</div> |
| <div class="line"><a name="l05299"></a><span class="lineno"> 5299</span>  int32_t padRight{0};</div> |
| <div class="line"><a name="l05300"></a><span class="lineno"> 5300</span>  int32_t padTop{0};</div> |
| <div class="line"><a name="l05301"></a><span class="lineno"> 5301</span>  int32_t padBottom{0};</div> |
| <div class="line"><a name="l05302"></a><span class="lineno"> 5302</span>  </div> |
| <div class="line"><a name="l05303"></a><span class="lineno"> 5303</span>  ::android::nn::PaddingScheme paddingScheme;</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05308"></a><span class="lineno"> 5308</span>  {</div> |
| <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> |
| <div class="line"><a name="l05310"></a><span class="lineno"> 5310</span>  }</div> |
| <div class="line"><a name="l05311"></a><span class="lineno"> 5311</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05314"></a><span class="lineno"> 5314</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05317"></a><span class="lineno"> 5317</span>  std::vector<int32_t> outputShape;</div> |
| <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> |
| <div class="line"><a name="l05319"></a><span class="lineno"> 5319</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05322"></a><span class="lineno"> 5322</span>  {</div> |
| <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> |
| <div class="line"><a name="l05324"></a><span class="lineno"> 5324</span>  }</div> |
| <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> |
| <div class="line"><a name="l05326"></a><span class="lineno"> 5326</span>  }</div> |
| <div class="line"><a name="l05327"></a><span class="lineno"> 5327</span>  </div> |
| <div class="line"><a name="l05328"></a><span class="lineno"> 5328</span>  uint32_t outputX;</div> |
| <div class="line"><a name="l05329"></a><span class="lineno"> 5329</span>  uint32_t outputY;</div> |
| <div class="line"><a name="l05330"></a><span class="lineno"> 5330</span>  </div> |
| <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> |
| <div class="line"><a name="l05332"></a><span class="lineno"> 5332</span>  {</div> |
| <div class="line"><a name="l05333"></a><span class="lineno"> 5333</span>  <span class="keywordflow">if</span> (outputShape.size() == 0)</div> |
| <div class="line"><a name="l05334"></a><span class="lineno"> 5334</span>  {</div> |
| <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> |
| <div class="line"><a name="l05336"></a><span class="lineno"> 5336</span>  }</div> |
| <div class="line"><a name="l05337"></a><span class="lineno"> 5337</span>  </div> |
| <div class="line"><a name="l05338"></a><span class="lineno"> 5338</span>  outputX = outputShape[widthIndex];</div> |
| <div class="line"><a name="l05339"></a><span class="lineno"> 5339</span>  outputY = outputShape[heightIndex];</div> |
| <div class="line"><a name="l05340"></a><span class="lineno"> 5340</span>  }</div> |
| <div class="line"><a name="l05341"></a><span class="lineno"> 5341</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05342"></a><span class="lineno"> 5342</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05345"></a><span class="lineno"> 5345</span>  }</div> |
| <div class="line"><a name="l05346"></a><span class="lineno"> 5346</span>  </div> |
| <div class="line"><a name="l05347"></a><span class="lineno"> 5347</span>  CalcPaddingTransposeConv(outputX, kernelX, strideX, padLeft, padRight, paddingScheme);</div> |
| <div class="line"><a name="l05348"></a><span class="lineno"> 5348</span>  CalcPaddingTransposeConv(outputY, kernelY, strideY, padTop, padBottom, paddingScheme);</div> |
| <div class="line"><a name="l05349"></a><span class="lineno"> 5349</span>  </div> |
| <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> |
| <div class="line"><a name="l05351"></a><span class="lineno"> 5351</span>  <span class="comment">// but Arm NN only supports values >= 0</span></div> |
| <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> |
| <div class="line"><a name="l05353"></a><span class="lineno"> 5353</span>  {</div> |
| <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> |
| <div class="line"><a name="l05355"></a><span class="lineno"> 5355</span>  }</div> |
| <div class="line"><a name="l05356"></a><span class="lineno"> 5356</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05363"></a><span class="lineno"> 5363</span>  }</div> |
| <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> |
| <div class="line"><a name="l05365"></a><span class="lineno"> 5365</span>  {</div> |
| <div class="line"><a name="l05366"></a><span class="lineno"> 5366</span>  <span class="comment">// explicit padding</span></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05374"></a><span class="lineno"> 5374</span>  {</div> |
| <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> |
| <div class="line"><a name="l05376"></a><span class="lineno"> 5376</span>  }</div> |
| <div class="line"><a name="l05377"></a><span class="lineno"> 5377</span>  }</div> |
| <div class="line"><a name="l05378"></a><span class="lineno"> 5378</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05379"></a><span class="lineno"> 5379</span>  {</div> |
| <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> |
| <div class="line"><a name="l05381"></a><span class="lineno"> 5381</span>  }</div> |
| <div class="line"><a name="l05382"></a><span class="lineno"> 5382</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05385"></a><span class="lineno"> 5385</span>  </div> |
| <div class="line"><a name="l05386"></a><span class="lineno"> 5386</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05389"></a><span class="lineno"> 5389</span>  {</div> |
| <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> |
| <div class="line"><a name="l05391"></a><span class="lineno"> 5391</span>  IsTransposeConvolution2dSupported,</div> |
| <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> |
| <div class="line"><a name="l05393"></a><span class="lineno"> 5393</span>  isSupported,</div> |
| <div class="line"><a name="l05394"></a><span class="lineno"> 5394</span>  setBackend,</div> |
| <div class="line"><a name="l05395"></a><span class="lineno"> 5395</span>  inputInfo,</div> |
| <div class="line"><a name="l05396"></a><span class="lineno"> 5396</span>  outputInfo,</div> |
| <div class="line"><a name="l05397"></a><span class="lineno"> 5397</span>  desc,</div> |
| <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> |
| <div class="line"><a name="l05399"></a><span class="lineno"> 5399</span>  biases);</div> |
| <div class="line"><a name="l05400"></a><span class="lineno"> 5400</span>  };</div> |
| <div class="line"><a name="l05401"></a><span class="lineno"> 5401</span>  </div> |
| <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> |
| <div class="line"><a name="l05403"></a><span class="lineno"> 5403</span>  {</div> |
| <div class="line"><a name="l05404"></a><span class="lineno"> 5404</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l05405"></a><span class="lineno"> 5405</span>  }</div> |
| <div class="line"><a name="l05406"></a><span class="lineno"> 5406</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05407"></a><span class="lineno"> 5407</span>  {</div> |
| <div class="line"><a name="l05408"></a><span class="lineno"> 5408</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l05409"></a><span class="lineno"> 5409</span>  }</div> |
| <div class="line"><a name="l05410"></a><span class="lineno"> 5410</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05411"></a><span class="lineno"> 5411</span>  {</div> |
| <div class="line"><a name="l05412"></a><span class="lineno"> 5412</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05413"></a><span class="lineno"> 5413</span>  }</div> |
| <div class="line"><a name="l05414"></a><span class="lineno"> 5414</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05418"></a><span class="lineno"> 5418</span>  <span class="keywordflow">if</span> (!startLayer)</div> |
| <div class="line"><a name="l05419"></a><span class="lineno"> 5419</span>  {</div> |
| <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> |
| <div class="line"><a name="l05421"></a><span class="lineno"> 5421</span>  }</div> |
| <div class="line"><a name="l05422"></a><span class="lineno"> 5422</span>  </div> |
| <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> |
| <div class="line"><a name="l05424"></a><span class="lineno"> 5424</span>  </div> |
| <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> |
| <div class="line"><a name="l05426"></a><span class="lineno"> 5426</span>  data, <span class="keyword">nullptr</span>, validateFunc, activation);</div> |
| <div class="line"><a name="l05427"></a><span class="lineno"> 5427</span> }</div> |
| <div class="line"><a name="l05428"></a><span class="lineno"> 5428</span>  </div> |
| <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> |
| <div class="line"><a name="l05430"></a><span class="lineno"> 5430</span> {</div> |
| <div class="line"><a name="l05431"></a><span class="lineno"> 5431</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSqrt()"</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05434"></a><span class="lineno"> 5434</span>  </div> |
| <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> |
| <div class="line"><a name="l05436"></a><span class="lineno"> 5436</span> }</div> |
| <div class="line"><a name="l05437"></a><span class="lineno"> 5437</span>  </div> |
| <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> |
| <div class="line"><a name="l05439"></a><span class="lineno"> 5439</span> {</div> |
| <div class="line"><a name="l05440"></a><span class="lineno"> 5440</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertSqueeze()"</span>;</div> |
| <div class="line"><a name="l05441"></a><span class="lineno"> 5441</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05444"></a><span class="lineno"> 5444</span>  {</div> |
| <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> |
| <div class="line"><a name="l05446"></a><span class="lineno"> 5446</span>  }</div> |
| <div class="line"><a name="l05447"></a><span class="lineno"> 5447</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05450"></a><span class="lineno"> 5450</span>  <span class="keywordflow">if</span> (rank > 4)</div> |
| <div class="line"><a name="l05451"></a><span class="lineno"> 5451</span>  {</div> |
| <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> |
| <div class="line"><a name="l05453"></a><span class="lineno"> 5453</span>  }</div> |
| <div class="line"><a name="l05454"></a><span class="lineno"> 5454</span>  </div> |
| <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> |
| <div class="line"><a name="l05456"></a><span class="lineno"> 5456</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l05457"></a><span class="lineno"> 5457</span>  {</div> |
| <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> |
| <div class="line"><a name="l05459"></a><span class="lineno"> 5459</span>  }</div> |
| <div class="line"><a name="l05460"></a><span class="lineno"> 5460</span>  </div> |
| <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> |
| <div class="line"><a name="l05462"></a><span class="lineno"> 5462</span>  {</div> |
| <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> |
| <div class="line"><a name="l05464"></a><span class="lineno"> 5464</span>  }</div> |
| <div class="line"><a name="l05465"></a><span class="lineno"> 5465</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05469"></a><span class="lineno"> 5469</span>  </div> |
| <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> |
| <div class="line"><a name="l05471"></a><span class="lineno"> 5471</span>  </div> |
| <div class="line"><a name="l05472"></a><span class="lineno"> 5472</span>  std::vector<int32_t> axis;</div> |
| <div class="line"><a name="l05473"></a><span class="lineno"> 5473</span>  <span class="keywordflow">if</span> (!axisOperand)</div> |
| <div class="line"><a name="l05474"></a><span class="lineno"> 5474</span>  {</div> |
| <div class="line"><a name="l05475"></a><span class="lineno"> 5475</span>  axis.assign(dimensionSequence,</div> |
| <div class="line"><a name="l05476"></a><span class="lineno"> 5476</span>  dimensionSequence + rank);</div> |
| <div class="line"><a name="l05477"></a><span class="lineno"> 5477</span>  }</div> |
| <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> |
| <div class="line"><a name="l05479"></a><span class="lineno"> 5479</span>  {</div> |
| <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> |
| <div class="line"><a name="l05481"></a><span class="lineno"> 5481</span>  }</div> |
| <div class="line"><a name="l05482"></a><span class="lineno"> 5482</span>  </div> |
| <div class="line"><a name="l05483"></a><span class="lineno"> 5483</span>  std::vector<uint32_t> outputDims;</div> |
| <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> |
| <div class="line"><a name="l05485"></a><span class="lineno"> 5485</span>  {</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05488"></a><span class="lineno"> 5488</span>  <span class="keywordflow">if</span> (skipSqueeze || currentDimension != 1)</div> |
| <div class="line"><a name="l05489"></a><span class="lineno"> 5489</span>  {</div> |
| <div class="line"><a name="l05490"></a><span class="lineno"> 5490</span>  outputDims.push_back(currentDimension);</div> |
| <div class="line"><a name="l05491"></a><span class="lineno"> 5491</span>  }</div> |
| <div class="line"><a name="l05492"></a><span class="lineno"> 5492</span>  }</div> |
| <div class="line"><a name="l05493"></a><span class="lineno"> 5493</span>  </div> |
| <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> |
| <div class="line"><a name="l05495"></a><span class="lineno"> 5495</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05498"></a><span class="lineno"> 5498</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05501"></a><span class="lineno"> 5501</span>  </div> |
| <div class="line"><a name="l05502"></a><span class="lineno"> 5502</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05505"></a><span class="lineno"> 5505</span>  IsReshapeSupported,</div> |
| <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> |
| <div class="line"><a name="l05507"></a><span class="lineno"> 5507</span>  isSupported,</div> |
| <div class="line"><a name="l05508"></a><span class="lineno"> 5508</span>  setBackend,</div> |
| <div class="line"><a name="l05509"></a><span class="lineno"> 5509</span>  inputInfo,</div> |
| <div class="line"><a name="l05510"></a><span class="lineno"> 5510</span>  outputInfo,</div> |
| <div class="line"><a name="l05511"></a><span class="lineno"> 5511</span>  reshapeDesc);</div> |
| <div class="line"><a name="l05512"></a><span class="lineno"> 5512</span>  </div> |
| <div class="line"><a name="l05513"></a><span class="lineno"> 5513</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05514"></a><span class="lineno"> 5514</span>  {</div> |
| <div class="line"><a name="l05515"></a><span class="lineno"> 5515</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05516"></a><span class="lineno"> 5516</span>  }</div> |
| <div class="line"><a name="l05517"></a><span class="lineno"> 5517</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05520"></a><span class="lineno"> 5520</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l05522"></a><span class="lineno"> 5522</span>  </div> |
| <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> |
| <div class="line"><a name="l05524"></a><span class="lineno"> 5524</span> }</div> |
| <div class="line"><a name="l05525"></a><span class="lineno"> 5525</span>  </div> |
| <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> |
| <div class="line"><a name="l05527"></a><span class="lineno"> 5527</span> {</div> |
| <div class="line"><a name="l05528"></a><span class="lineno"> 5528</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertStridedSlice()"</span>;</div> |
| <div class="line"><a name="l05529"></a><span class="lineno"> 5529</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05532"></a><span class="lineno"> 5532</span>  {</div> |
| <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> |
| <div class="line"><a name="l05534"></a><span class="lineno"> 5534</span>  }</div> |
| <div class="line"><a name="l05535"></a><span class="lineno"> 5535</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05538"></a><span class="lineno"> 5538</span>  <span class="keywordflow">if</span> (rank > 4)</div> |
| <div class="line"><a name="l05539"></a><span class="lineno"> 5539</span>  {</div> |
| <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> |
| <div class="line"><a name="l05541"></a><span class="lineno"> 5541</span>  }</div> |
| <div class="line"><a name="l05542"></a><span class="lineno"> 5542</span>  </div> |
| <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> |
| <div class="line"><a name="l05544"></a><span class="lineno"> 5544</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l05545"></a><span class="lineno"> 5545</span>  {</div> |
| <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> |
| <div class="line"><a name="l05547"></a><span class="lineno"> 5547</span>  }</div> |
| <div class="line"><a name="l05548"></a><span class="lineno"> 5548</span>  </div> |
| <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> |
| <div class="line"><a name="l05550"></a><span class="lineno"> 5550</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05554"></a><span class="lineno"> 5554</span>  </div> |
| <div class="line"><a name="l05555"></a><span class="lineno"> 5555</span>  std::vector<int32_t> beginValues;</div> |
| <div class="line"><a name="l05556"></a><span class="lineno"> 5556</span>  std::vector<int32_t> endValues;</div> |
| <div class="line"><a name="l05557"></a><span class="lineno"> 5557</span>  std::vector<int32_t> stridesValues;</div> |
| <div class="line"><a name="l05558"></a><span class="lineno"> 5558</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05561"></a><span class="lineno"> 5561</span>  {</div> |
| <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> |
| <div class="line"><a name="l05563"></a><span class="lineno"> 5563</span>  {</div> |
| <div class="line"><a name="l05564"></a><span class="lineno"> 5564</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05565"></a><span class="lineno"> 5565</span>  }</div> |
| <div class="line"><a name="l05566"></a><span class="lineno"> 5566</span>  </div> |
| <div class="line"><a name="l05567"></a><span class="lineno"> 5567</span>  <span class="keywordflow">if</span> (operandValues.size() != rank)</div> |
| <div class="line"><a name="l05568"></a><span class="lineno"> 5568</span>  {</div> |
| <div class="line"><a name="l05569"></a><span class="lineno"> 5569</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05570"></a><span class="lineno"> 5570</span>  }</div> |
| <div class="line"><a name="l05571"></a><span class="lineno"> 5571</span>  </div> |
| <div class="line"><a name="l05572"></a><span class="lineno"> 5572</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l05573"></a><span class="lineno"> 5573</span>  };</div> |
| <div class="line"><a name="l05574"></a><span class="lineno"> 5574</span>  </div> |
| <div class="line"><a name="l05575"></a><span class="lineno"> 5575</span>  <span class="keywordflow">if</span> (!ValidateInputOperands(*beginOperand, beginValues)</div> |
| <div class="line"><a name="l05576"></a><span class="lineno"> 5576</span>  || !ValidateInputOperands(*endOperand, endValues)</div> |
| <div class="line"><a name="l05577"></a><span class="lineno"> 5577</span>  || !ValidateInputOperands(*stridesOperand, stridesValues))</div> |
| <div class="line"><a name="l05578"></a><span class="lineno"> 5578</span>  {</div> |
| <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> |
| <div class="line"><a name="l05580"></a><span class="lineno"> 5580</span>  }</div> |
| <div class="line"><a name="l05581"></a><span class="lineno"> 5581</span>  </div> |
| <div class="line"><a name="l05582"></a><span class="lineno"> 5582</span>  <span class="comment">// Stride cannot have value '0'</span></div> |
| <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> |
| <div class="line"><a name="l05584"></a><span class="lineno"> 5584</span>  {</div> |
| <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> |
| <div class="line"><a name="l05586"></a><span class="lineno"> 5586</span>  }</div> |
| <div class="line"><a name="l05587"></a><span class="lineno"> 5587</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05593"></a><span class="lineno"> 5593</span>  </div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05598"></a><span class="lineno"> 5598</span>  {</div> |
| <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> |
| <div class="line"><a name="l05600"></a><span class="lineno"> 5600</span>  }</div> |
| <div class="line"><a name="l05601"></a><span class="lineno"> 5601</span>  </div> |
| <div class="line"><a name="l05602"></a><span class="lineno"> 5602</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05605"></a><span class="lineno"> 5605</span>  {</div> |
| <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> |
| <div class="line"><a name="l05607"></a><span class="lineno"> 5607</span>  IsStridedSliceSupported,</div> |
| <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> |
| <div class="line"><a name="l05609"></a><span class="lineno"> 5609</span>  isSupported,</div> |
| <div class="line"><a name="l05610"></a><span class="lineno"> 5610</span>  setBackend,</div> |
| <div class="line"><a name="l05611"></a><span class="lineno"> 5611</span>  inputInfo,</div> |
| <div class="line"><a name="l05612"></a><span class="lineno"> 5612</span>  outputInfo,</div> |
| <div class="line"><a name="l05613"></a><span class="lineno"> 5613</span>  descriptor);</div> |
| <div class="line"><a name="l05614"></a><span class="lineno"> 5614</span>  };</div> |
| <div class="line"><a name="l05615"></a><span class="lineno"> 5615</span>  </div> |
| <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> |
| <div class="line"><a name="l05617"></a><span class="lineno"> 5617</span>  {</div> |
| <div class="line"><a name="l05618"></a><span class="lineno"> 5618</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l05619"></a><span class="lineno"> 5619</span>  }</div> |
| <div class="line"><a name="l05620"></a><span class="lineno"> 5620</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05621"></a><span class="lineno"> 5621</span>  {</div> |
| <div class="line"><a name="l05622"></a><span class="lineno"> 5622</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l05623"></a><span class="lineno"> 5623</span>  }</div> |
| <div class="line"><a name="l05624"></a><span class="lineno"> 5624</span>  </div> |
| <div class="line"><a name="l05625"></a><span class="lineno"> 5625</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05626"></a><span class="lineno"> 5626</span>  {</div> |
| <div class="line"><a name="l05627"></a><span class="lineno"> 5627</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05628"></a><span class="lineno"> 5628</span>  }</div> |
| <div class="line"><a name="l05629"></a><span class="lineno"> 5629</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05633"></a><span class="lineno"> 5633</span>  {</div> |
| <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> |
| <div class="line"><a name="l05635"></a><span class="lineno"> 5635</span>  </div> |
| <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> |
| <div class="line"><a name="l05637"></a><span class="lineno"> 5637</span>  {</div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05642"></a><span class="lineno"> 5642</span>  {</div> |
| <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> |
| <div class="line"><a name="l05644"></a><span class="lineno"> 5644</span>  }</div> |
| <div class="line"><a name="l05645"></a><span class="lineno"> 5645</span>  </div> |
| <div class="line"><a name="l05646"></a><span class="lineno"> 5646</span>  <span class="keywordflow">if</span>(stride < 0)</div> |
| <div class="line"><a name="l05647"></a><span class="lineno"> 5647</span>  {</div> |
| <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> |
| <div class="line"><a name="l05649"></a><span class="lineno"> 5649</span>  }</div> |
| <div class="line"><a name="l05650"></a><span class="lineno"> 5650</span>  }</div> |
| <div class="line"><a name="l05651"></a><span class="lineno"> 5651</span>  }</div> |
| <div class="line"><a name="l05652"></a><span class="lineno"> 5652</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05655"></a><span class="lineno"> 5655</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l05657"></a><span class="lineno"> 5657</span>  </div> |
| <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> |
| <div class="line"><a name="l05659"></a><span class="lineno"> 5659</span> }</div> |
| <div class="line"><a name="l05660"></a><span class="lineno"> 5660</span>  </div> |
| <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> |
| <div class="line"><a name="l05662"></a><span class="lineno"> 5662</span> {</div> |
| <div class="line"><a name="l05663"></a><span class="lineno"> 5663</span>  VLOG(DRIVER) << <span class="stringliteral">"Converter::ConvertTranspose()"</span>;</div> |
| <div class="line"><a name="l05664"></a><span class="lineno"> 5664</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05667"></a><span class="lineno"> 5667</span>  {</div> |
| <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> |
| <div class="line"><a name="l05669"></a><span class="lineno"> 5669</span>  }</div> |
| <div class="line"><a name="l05670"></a><span class="lineno"> 5670</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05673"></a><span class="lineno"> 5673</span>  <span class="keywordflow">if</span> (rank > 4)</div> |
| <div class="line"><a name="l05674"></a><span class="lineno"> 5674</span>  {</div> |
| <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> |
| <div class="line"><a name="l05676"></a><span class="lineno"> 5676</span>  }</div> |
| <div class="line"><a name="l05677"></a><span class="lineno"> 5677</span>  </div> |
| <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> |
| <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> |
| <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> |
| <div class="line"><a name="l05681"></a><span class="lineno"> 5681</span>  </div> |
| <div class="line"><a name="l05682"></a><span class="lineno"> 5682</span>  std::vector<int32_t> perm(rank);</div> |
| <div class="line"><a name="l05683"></a><span class="lineno"> 5683</span>  <span class="keywordflow">if</span> (!permOperand || (permOperand->lifetime == OperandLifeTime::NO_VALUE))</div> |
| <div class="line"><a name="l05684"></a><span class="lineno"> 5684</span>  {</div> |
| <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> |
| <div class="line"><a name="l05686"></a><span class="lineno"> 5686</span>  {</div> |
| <div class="line"><a name="l05687"></a><span class="lineno"> 5687</span>  perm[rank - i] = armnn::numeric_cast<int> (i - 1);</div> |
| <div class="line"><a name="l05688"></a><span class="lineno"> 5688</span>  }</div> |
| <div class="line"><a name="l05689"></a><span class="lineno"> 5689</span>  }</div> |
| <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> |
| <div class="line"><a name="l05691"></a><span class="lineno"> 5691</span>  {</div> |
| <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> |
| <div class="line"><a name="l05693"></a><span class="lineno"> 5693</span>  }</div> |
| <div class="line"><a name="l05694"></a><span class="lineno"> 5694</span>  </div> |
| <div class="line"><a name="l05695"></a><span class="lineno"> 5695</span>  std::vector<uint32_t> outputDims(perm.begin(), perm.begin() + rank);</div> |
| <div class="line"><a name="l05696"></a><span class="lineno"> 5696</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05699"></a><span class="lineno"> 5699</span>  </div> |
| <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> |
| <div class="line"><a name="l05701"></a><span class="lineno"> 5701</span>  <span class="keywordflow">if</span> (!output)</div> |
| <div class="line"><a name="l05702"></a><span class="lineno"> 5702</span>  {</div> |
| <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> |
| <div class="line"><a name="l05704"></a><span class="lineno"> 5704</span>  }</div> |
| <div class="line"><a name="l05705"></a><span class="lineno"> 5705</span>  </div> |
| <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> |
| <div class="line"><a name="l05707"></a><span class="lineno"> 5707</span>  </div> |
| <div class="line"><a name="l05708"></a><span class="lineno"> 5708</span>  <span class="keywordtype">bool</span> isSupported = <span class="keyword">false</span>;</div> |
| <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> |
| <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> |
| <div class="line"><a name="l05711"></a><span class="lineno"> 5711</span>  {</div> |
| <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> |
| <div class="line"><a name="l05713"></a><span class="lineno"> 5713</span>  IsTransposeSupported,</div> |
| <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> |
| <div class="line"><a name="l05715"></a><span class="lineno"> 5715</span>  isSupported,</div> |
| <div class="line"><a name="l05716"></a><span class="lineno"> 5716</span>  setBackend,</div> |
| <div class="line"><a name="l05717"></a><span class="lineno"> 5717</span>  inputInfo,</div> |
| <div class="line"><a name="l05718"></a><span class="lineno"> 5718</span>  outputInfo,</div> |
| <div class="line"><a name="l05719"></a><span class="lineno"> 5719</span>  transposeDesc);</div> |
| <div class="line"><a name="l05720"></a><span class="lineno"> 5720</span>  };</div> |
| <div class="line"><a name="l05721"></a><span class="lineno"> 5721</span>  </div> |
| <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> |
| <div class="line"><a name="l05723"></a><span class="lineno"> 5723</span>  {</div> |
| <div class="line"><a name="l05724"></a><span class="lineno"> 5724</span>  isSupported = <a class="code" href="namespacearmnn__driver.html#a85d53debffd9611a3b5bc6031afd5073">AreDynamicTensorsSupported</a>();</div> |
| <div class="line"><a name="l05725"></a><span class="lineno"> 5725</span>  }</div> |
| <div class="line"><a name="l05726"></a><span class="lineno"> 5726</span>  <span class="keywordflow">else</span></div> |
| <div class="line"><a name="l05727"></a><span class="lineno"> 5727</span>  {</div> |
| <div class="line"><a name="l05728"></a><span class="lineno"> 5728</span>  validateFunc(outputInfo, isSupported);</div> |
| <div class="line"><a name="l05729"></a><span class="lineno"> 5729</span>  }</div> |
| <div class="line"><a name="l05730"></a><span class="lineno"> 5730</span>  </div> |
| <div class="line"><a name="l05731"></a><span class="lineno"> 5731</span>  <span class="keywordflow">if</span> (!isSupported)</div> |
| <div class="line"><a name="l05732"></a><span class="lineno"> 5732</span>  {</div> |
| <div class="line"><a name="l05733"></a><span class="lineno"> 5733</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l05734"></a><span class="lineno"> 5734</span>  }</div> |
| <div class="line"><a name="l05735"></a><span class="lineno"> 5735</span>  </div> |
| <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> |
| <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> |
| <div class="line"><a name="l05738"></a><span class="lineno"> 5738</span>  assert(layer != <span class="keyword">nullptr</span>);</div> |
| <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> |
| <div class="line"><a name="l05740"></a><span class="lineno"> 5740</span>  </div> |
| <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> |
| <div class="line"><a name="l05742"></a><span class="lineno"> 5742</span> }</div> |
| <div class="line"><a name="l05743"></a><span class="lineno"> 5743</span>  </div> |
| <div class="line"><a name="l05744"></a><span class="lineno"> 5744</span> } <span class="comment">// namespace armnn_driver</span></div> |
| </div><!-- fragment --></div><!-- contents --> |
| </div><!-- doc-content --> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="ttc" id="a_tensor_utils_8hpp_html"><div class="ttname"><a href="_tensor_utils_8hpp.html">TensorUtils.hpp</a></div></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="ttc" id="a_exceptions_8hpp_html"><div class="ttname"><a href="_exceptions_8hpp.html">Exceptions.hpp</a></div></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <div class="ttc" id="a_converter_8hpp_html"><div class="ttname"><a href="_converter_8hpp.html">Converter.hpp</a></div></div> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <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> |
| <!-- start footer part --> |
| <div id="nav-path" class="navpath"><!-- id is needed for treeview function! --> |
| <ul> |
| <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> |
| <li class="footer">Generated on Thu May 16 2024 09:31:44 for Arm NN by |
| <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li> |
| </ul> |
| </div> |
| </body> |
| </html> |