Nikhil Raj | 03c7ff3 | 2023-08-22 12:00:04 +0100 | [diff] [blame^] | 1 | <!-- HTML header for doxygen 1.8.17--> |
| 2 | <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| 3 | <html xmlns="http://www.w3.org/1999/xhtml"> |
| 4 | <head> |
| 5 | <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| 6 | <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| 7 | <meta name="generator" content="Doxygen 1.8.17"/> |
| 8 | <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| 9 | <title>Arm NN: src/backends/aclCommon/ArmComputeUtils.hpp File Reference</title> |
| 10 | <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| 11 | <script type="text/javascript" src="jquery.js"></script> |
| 12 | <script type="text/javascript" src="dynsections.js"></script> |
| 13 | <link href="navtree.css" rel="stylesheet" type="text/css"/> |
| 14 | <script type="text/javascript" src="resize.js"></script> |
| 15 | <script type="text/javascript" src="navtreedata.js"></script> |
| 16 | <script type="text/javascript" src="navtree.js"></script> |
| 17 | <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| 18 | <script type="text/javascript" src="search/searchdata.js"></script> |
| 19 | <script type="text/javascript" src="search/search.js"></script> |
| 20 | <script type="text/x-mathjax-config"> |
| 21 | MathJax.Hub.Config({ |
| 22 | extensions: ["tex2jax.js"], |
| 23 | jax: ["input/TeX","output/HTML-CSS"], |
| 24 | }); |
| 25 | </script> |
| 26 | <script type="text/javascript" async="async" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script> |
| 27 | <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| 28 | <link href="customdoxygen.css" rel="stylesheet" type="text/css"/> |
| 29 | </head> |
| 30 | <body> |
| 31 | <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| 32 | <div id="titlearea"> |
| 33 | <table cellspacing="0" cellpadding="0"> |
| 34 | <tbody> |
| 35 | <tr style="height: 56px;"> |
| 36 | <img alt="ArmNN" src="Arm_NN_horizontal_blue.png" style="max-width: 15rem; margin-top: .5rem; margin-left 13px"/> |
| 37 | <td id="projectalign" style="padding-left: 0.9em;"> |
| 38 | <div id="projectname"> |
| 39 |  <span id="projectnumber">23.08</span> |
| 40 | </div> |
| 41 | </td> |
| 42 | </tr> |
| 43 | </tbody> |
| 44 | </table> |
| 45 | </div> |
| 46 | <!-- end header part --> |
| 47 | <!-- Generated by Doxygen 1.8.17 --> |
| 48 | <script type="text/javascript"> |
| 49 | /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| 50 | var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| 51 | /* @license-end */ |
| 52 | </script> |
| 53 | <script type="text/javascript" src="menudata.js"></script> |
| 54 | <script type="text/javascript" src="menu.js"></script> |
| 55 | <script type="text/javascript"> |
| 56 | /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| 57 | $(function() { |
| 58 | initMenu('',true,false,'search.php','Search'); |
| 59 | $(document).ready(function() { init_search(); }); |
| 60 | }); |
| 61 | /* @license-end */</script> |
| 62 | <div id="main-nav"></div> |
| 63 | </div><!-- top --> |
| 64 | <div id="side-nav" class="ui-resizable side-nav-resizable"> |
| 65 | <div id="nav-tree"> |
| 66 | <div id="nav-tree-contents"> |
| 67 | <div id="nav-sync" class="sync"></div> |
| 68 | </div> |
| 69 | </div> |
| 70 | <div id="splitbar" style="-moz-user-select:none;" |
| 71 | class="ui-resizable-handle"> |
| 72 | </div> |
| 73 | </div> |
| 74 | <script type="text/javascript"> |
| 75 | /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| 76 | $(document).ready(function(){initNavTree('_arm_compute_utils_8hpp.html',''); initResizable(); }); |
| 77 | /* @license-end */ |
| 78 | </script> |
| 79 | <div id="doc-content"> |
| 80 | <!-- window showing the filter options --> |
| 81 | <div id="MSearchSelectWindow" |
| 82 | onmouseover="return searchBox.OnSearchSelectShow()" |
| 83 | onmouseout="return searchBox.OnSearchSelectHide()" |
| 84 | onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| 85 | </div> |
| 86 | |
| 87 | <!-- iframe showing the search results (closed by default) --> |
| 88 | <div id="MSearchResultsWindow"> |
| 89 | <iframe src="javascript:void(0)" frameborder="0" |
| 90 | name="MSearchResults" id="MSearchResults"> |
| 91 | </iframe> |
| 92 | </div> |
| 93 | |
| 94 | <div class="header"> |
| 95 | <div class="summary"> |
| 96 | <a href="#namespaces">Namespaces</a> | |
| 97 | <a href="#define-members">Macros</a> | |
| 98 | <a href="#func-members">Functions</a> </div> |
| 99 | <div class="headertitle"> |
| 100 | <div class="title">ArmComputeUtils.hpp File Reference</div> </div> |
| 101 | </div><!--header--> |
| 102 | <div class="contents"> |
| 103 | <div class="textblock"><code>#include <<a class="el" href="_descriptors_8hpp_source.html">armnn/Descriptors.hpp</a>></code><br /> |
| 104 | <code>#include <<a class="el" href="_tensor_8hpp_source.html">armnn/Tensor.hpp</a>></code><br /> |
| 105 | <code>#include <<a class="el" href="_assert_8hpp_source.html">armnn/utility/Assert.hpp</a>></code><br /> |
| 106 | <code>#include <<a class="el" href="_numeric_cast_8hpp_source.html">armnn/utility/NumericCast.hpp</a>></code><br /> |
| 107 | <code>#include <<a class="el" href="_workload_data_8hpp_source.html">armnn/backends/WorkloadData.hpp</a>></code><br /> |
| 108 | <code>#include <arm_compute/runtime/FunctionDescriptors.h></code><br /> |
| 109 | <code>#include <arm_compute/function_info/FullyConnectedLayerInfo.h></code><br /> |
| 110 | <code>#include "<a class="el" href="_neon_reduce_workload_8hpp_source.html">neon/workloads/NeonReduceWorkload.hpp</a>"</code><br /> |
| 111 | <code>#include "<a class="el" href="_cl_reduce_workload_8hpp_source.html">cl/workloads/ClReduceWorkload.hpp</a>"</code><br /> |
| 112 | </div><div class="textblock"><div class="dynheader"> |
| 113 | Include dependency graph for ArmComputeUtils.hpp:</div> |
| 114 | <div class="dyncontent"> |
| 115 | <div class="center"><iframe scrolling="no" frameborder="0" src="_arm_compute_utils_8hpp__incl.svg" width="3430" height="947"><p><b>This browser is not able to show SVG: try Firefox, Chrome, Safari, or Opera instead.</b></p></iframe> |
| 116 | </div> |
| 117 | </div> |
| 118 | </div><div class="textblock"><div class="dynheader"> |
| 119 | This graph shows which files directly or indirectly include this file:</div> |
| 120 | <div class="dyncontent"> |
| 121 | <div class="center"><iframe scrolling="no" frameborder="0" src="_arm_compute_utils_8hpp__dep__incl.svg" width="17903" height="231"><p><b>This browser is not able to show SVG: try Firefox, Chrome, Safari, or Opera instead.</b></p></iframe> |
| 122 | </div> |
| 123 | </div> |
| 124 | </div> |
| 125 | <p><a href="_arm_compute_utils_8hpp_source.html">Go to the source code of this file.</a></p> |
| 126 | <table class="memberdecls"> |
| 127 | <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="namespaces"></a> |
| 128 | Namespaces</h2></td></tr> |
| 129 | <tr class="memitem:namespacearmnn"><td class="memItemLeft" align="right" valign="top">  </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html">armnn</a></td></tr> |
| 130 | <tr class="memdesc:namespacearmnn"><td class="mdescLeft"> </td><td class="mdescRight">Copyright (c) 2021 ARM Limited and Contributors. <br /></td></tr> |
| 131 | <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| 132 | </table><table class="memberdecls"> |
| 133 | <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="define-members"></a> |
| 134 | Macros</h2></td></tr> |
| 135 | <tr class="memitem:a97d67174b2c48969547ba4bac79e5ea2"><td class="memItemLeft" align="right" valign="top">#define </td><td class="memItemRight" valign="bottom"><a class="el" href="_arm_compute_utils_8hpp.html#a97d67174b2c48969547ba4bac79e5ea2">IS_MULTI_AXES_REDUCE_SUPPORTED</a>(func, input, desc, status)</td></tr> |
| 136 | <tr class="memdesc:a97d67174b2c48969547ba4bac79e5ea2"><td class="mdescLeft"> </td><td class="mdescRight">Macro function check if layer with multiple axes is supported on each backend. <a href="_arm_compute_utils_8hpp.html#a97d67174b2c48969547ba4bac79e5ea2">More...</a><br /></td></tr> |
| 137 | <tr class="separator:a97d67174b2c48969547ba4bac79e5ea2"><td class="memSeparator" colspan="2"> </td></tr> |
| 138 | </table><table class="memberdecls"> |
| 139 | <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a> |
| 140 | Functions</h2></td></tr> |
| 141 | <tr class="memitem:a5e783a951642781b9e7b55db06a514b7"><td class="memItemLeft" align="right" valign="top">arm_compute::NormalizationLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a5e783a951642781b9e7b55db06a514b7">CreateAclNormalizationLayerInfoForL2Normalization</a> (const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &tensorInfo, <a class="el" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> dataLayout)</td></tr> |
| 142 | <tr class="separator:a5e783a951642781b9e7b55db06a514b7"><td class="memSeparator" colspan="2"> </td></tr> |
| 143 | <tr class="memitem:afdba36f125621d775d471f0daf613df2"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo::ActivationFunction </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#afdba36f125621d775d471f0daf613df2">ConvertActivationFunctionToAclActivationFunction</a> (ActivationFunction armnnFunction)</td></tr> |
| 144 | <tr class="separator:afdba36f125621d775d471f0daf613df2"><td class="memSeparator" colspan="2"> </td></tr> |
| 145 | <tr class="memitem:ad701d0d29baa4266ab4d33b090aa661c"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ad701d0d29baa4266ab4d33b090aa661c">ConvertActivationDescriptorToAclActivationLayerInfo</a> (const ActivationDescriptor &actDesc)</td></tr> |
| 146 | <tr class="separator:ad701d0d29baa4266ab4d33b090aa661c"><td class="memSeparator" colspan="2"> </td></tr> |
| 147 | <tr class="memitem:af6d517245110f6b1ca702da0cbbf4518"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#af6d517245110f6b1ca702da0cbbf4518">ConvertActivationDescriptorToAclActivationLayerInfo</a> (const ActivationDescriptor *activationDescPtr)</td></tr> |
| 148 | <tr class="separator:af6d517245110f6b1ca702da0cbbf4518"><td class="memSeparator" colspan="2"> </td></tr> |
| 149 | <tr class="memitem:abfb0841058a8190d30851f07eca3991f"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#abfb0841058a8190d30851f07eca3991f">ConvertAdditionalInfoToAclActivationLayerInfo</a> (const QueueDescriptor &queueDescriptor)</td></tr> |
| 150 | <tr class="separator:abfb0841058a8190d30851f07eca3991f"><td class="memSeparator" colspan="2"> </td></tr> |
| 151 | <tr class="memitem:aa1e93ef5f9ee3dbb5e7faa9578f180ae"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aa1e93ef5f9ee3dbb5e7faa9578f180ae">ConvertLstmActivationFuncToAclLayerInfo</a> (uint32_t activationFunction)</td></tr> |
| 152 | <tr class="separator:aa1e93ef5f9ee3dbb5e7faa9578f180ae"><td class="memSeparator" colspan="2"> </td></tr> |
| 153 | <tr class="memitem:ac1438a97bbc2aee5cd8470d9d2a3aebc"><td class="memItemLeft" align="right" valign="top">arm_compute::ComparisonOperation </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ac1438a97bbc2aee5cd8470d9d2a3aebc">ConvertComparisonOperationToAcl</a> (const ComparisonDescriptor &descriptor)</td></tr> |
| 154 | <tr class="separator:ac1438a97bbc2aee5cd8470d9d2a3aebc"><td class="memSeparator" colspan="2"> </td></tr> |
| 155 | <tr class="memitem:ad256fcf8c7f4d5a240fa47f0b56d50af"><td class="memItemLeft" align="right" valign="top">arm_compute::PoolingType </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ad256fcf8c7f4d5a240fa47f0b56d50af">ConvertPoolingAlgorithmToAclPoolingType</a> (PoolingAlgorithm poolingAlgorithm)</td></tr> |
| 156 | <tr class="separator:ad256fcf8c7f4d5a240fa47f0b56d50af"><td class="memSeparator" colspan="2"> </td></tr> |
| 157 | <tr class="memitem:a8f3bfacadfd6d2146d6ccd299dabc7aa"><td class="memItemLeft" align="right" valign="top">arm_compute::DimensionRoundingType </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a8f3bfacadfd6d2146d6ccd299dabc7aa">ConvertOutputShapeRoundingToAclDimensionRoundingType</a> (OutputShapeRounding rounding)</td></tr> |
| 158 | <tr class="separator:a8f3bfacadfd6d2146d6ccd299dabc7aa"><td class="memSeparator" colspan="2"> </td></tr> |
| 159 | <tr class="memitem:aa5baabb8e3a4aa6cbdcab419d743e747"><td class="memItemLeft" align="right" valign="top">arm_compute::NormType </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aa5baabb8e3a4aa6cbdcab419d743e747">ConvertNormalizationAlgorithmChannelToAclNormType</a> (NormalizationAlgorithmChannel channelType)</td></tr> |
| 160 | <tr class="separator:aa5baabb8e3a4aa6cbdcab419d743e747"><td class="memSeparator" colspan="2"> </td></tr> |
| 161 | <tr class="memitem:ae2e0c1d8160fadce44d479379790718e"><td class="memItemLeft" align="right" valign="top">arm_compute::FullyConnectedLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ae2e0c1d8160fadce44d479379790718e">ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo</a> (const FullyConnectedDescriptor &fullyConnectedDesc, const ActivationDescriptor *activationDesc)</td></tr> |
| 162 | <tr class="separator:ae2e0c1d8160fadce44d479379790718e"><td class="memSeparator" colspan="2"> </td></tr> |
| 163 | <tr class="memitem:a2f12c1f572fb15978e51cf646d42d4dc"><td class="memItemLeft" align="right" valign="top">arm_compute::FullyConnectedLayerInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a2f12c1f572fb15978e51cf646d42d4dc">ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo</a> (const FullyConnectedDescriptor &fullyConnectedDesc, arm_compute::ActivationLayerInfo activationLayerInfo)</td></tr> |
| 164 | <tr class="separator:a2f12c1f572fb15978e51cf646d42d4dc"><td class="memSeparator" colspan="2"> </td></tr> |
| 165 | <tr class="memitem:ae9bdcb8ac91731109dc423d6ed476204"><td class="memItemLeft" align="right" valign="top">arm_compute::InterpolationPolicy </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ae9bdcb8ac91731109dc423d6ed476204">ConvertResizeMethodToAclInterpolationPolicy</a> (ResizeMethod resizeMethod)</td></tr> |
| 166 | <tr class="separator:ae9bdcb8ac91731109dc423d6ed476204"><td class="memSeparator" colspan="2"> </td></tr> |
| 167 | <tr class="memitem:ae33fcc729c3eeda77a443a11f13da7ba"><td class="memTemplParams" colspan="2">template<typename T > </td></tr> |
| 168 | <tr class="memitem:ae33fcc729c3eeda77a443a11f13da7ba"><td class="memTemplItemLeft" align="right" valign="top">T </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ae33fcc729c3eeda77a443a11f13da7ba">ComputeSoftmaxAclAxis</a> (const SoftmaxDescriptor &softmaxDesc, const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &tensor)</td></tr> |
| 169 | <tr class="separator:ae33fcc729c3eeda77a443a11f13da7ba"><td class="memSeparator" colspan="2"> </td></tr> |
| 170 | <tr class="memitem:a8cbabc875597b3bed0ccdc0adb289fde"><td class="memItemLeft" align="right" valign="top">std::set< unsigned int > </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a8cbabc875597b3bed0ccdc0adb289fde">ComputeSplitAxis</a> (const <a class="el" href="namespacearmnn.html#a60291543fe872b795e71e05bcd835fd1">armnn::SplitterDescriptor</a> &desc, const TensorShape &input)</td></tr> |
| 171 | <tr class="separator:a8cbabc875597b3bed0ccdc0adb289fde"><td class="memSeparator" colspan="2"> </td></tr> |
| 172 | <tr class="memitem:a44a3b98b37a25c995aa9e4dae7d7b456"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a44a3b98b37a25c995aa9e4dae7d7b456">ComputeAclAxis</a> (const int &armnnAxis, const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &tensor)</td></tr> |
| 173 | <tr class="memdesc:a44a3b98b37a25c995aa9e4dae7d7b456"><td class="mdescLeft"> </td><td class="mdescRight">Function to convert ArmNN axis (left to right) to ACL axis (right to left) ranging from [-rank, rank) <a href="namespacearmnn.html#a44a3b98b37a25c995aa9e4dae7d7b456">More...</a><br /></td></tr> |
| 174 | <tr class="separator:a44a3b98b37a25c995aa9e4dae7d7b456"><td class="memSeparator" colspan="2"> </td></tr> |
| 175 | <tr class="memitem:aa1e65a4dea8c65b76423e121b880d059"><td class="memItemLeft" align="right" valign="top">unsigned int </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aa1e65a4dea8c65b76423e121b880d059">ComputePositiveAxis</a> (const int &axis, const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &tensor)</td></tr> |
| 176 | <tr class="memdesc:aa1e65a4dea8c65b76423e121b880d059"><td class="mdescLeft"> </td><td class="mdescRight">Function to convert axis to its positive equivalent value. <a href="namespacearmnn.html#aa1e65a4dea8c65b76423e121b880d059">More...</a><br /></td></tr> |
| 177 | <tr class="separator:aa1e65a4dea8c65b76423e121b880d059"><td class="memSeparator" colspan="2"> </td></tr> |
| 178 | <tr class="memitem:a855760e957668804d33c0691a802496e"><td class="memItemLeft" align="right" valign="top">arm_compute::Conv3dInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a855760e957668804d33c0691a802496e">ComputeConv3DInfo</a> (const <a class="el" href="structarmnn_1_1_convolution3d_descriptor.html">armnn::Convolution3dDescriptor</a> descriptor, bool isFastMathEnabled, const ActivationDescriptor *activationDescriptor)</td></tr> |
| 179 | <tr class="memdesc:a855760e957668804d33c0691a802496e"><td class="mdescLeft"> </td><td class="mdescRight">Utility function used to setup an arm_compute::Conv3dInfo object from convolution3d descriptor. <a href="namespacearmnn.html#a855760e957668804d33c0691a802496e">More...</a><br /></td></tr> |
| 180 | <tr class="separator:a855760e957668804d33c0691a802496e"><td class="memSeparator" colspan="2"> </td></tr> |
| 181 | <tr class="memitem:a3f2c9ea2505a2cac1d05e17e989aff2b"><td class="memItemLeft" align="right" valign="top">arm_compute::Conv3dInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a3f2c9ea2505a2cac1d05e17e989aff2b">ComputeConv3DInfo</a> (const <a class="el" href="structarmnn_1_1_convolution3d_queue_descriptor.html">armnn::Convolution3dQueueDescriptor</a> queueDescriptor, bool isFastMathEnabled)</td></tr> |
| 182 | <tr class="separator:a3f2c9ea2505a2cac1d05e17e989aff2b"><td class="memSeparator" colspan="2"> </td></tr> |
| 183 | <tr class="memitem:adf5a024ace0dbe2af2d5b1a1b9623771"><td class="memItemLeft" align="right" valign="top">arm_compute::PaddingMode </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#adf5a024ace0dbe2af2d5b1a1b9623771">ConvertPaddingModeToAcl</a> (const PaddingMode &paddingMode)</td></tr> |
| 184 | <tr class="separator:adf5a024ace0dbe2af2d5b1a1b9623771"><td class="memSeparator" colspan="2"> </td></tr> |
| 185 | <tr class="memitem:a03ab984d93d0f240f31274499b72e6b0"><td class="memItemLeft" align="right" valign="top">arm_compute::ReductionOperation </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a03ab984d93d0f240f31274499b72e6b0">ConvertReductionOperationToAcl</a> (const ReduceDescriptor &descriptor)</td></tr> |
| 186 | <tr class="separator:a03ab984d93d0f240f31274499b72e6b0"><td class="memSeparator" colspan="2"> </td></tr> |
| 187 | <tr class="memitem:a9d43aa01b882d8776ab9403eea3cf779"><td class="memItemLeft" align="right" valign="top">const TensorInfo </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a9d43aa01b882d8776ab9403eea3cf779">ComputeReductionTensorShape</a> (const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &input, const std::vector< uint32_t > &vAxis, const bool keepDims)</td></tr> |
| 188 | <tr class="memdesc:a9d43aa01b882d8776ab9403eea3cf779"><td class="mdescLeft"> </td><td class="mdescRight">Function to compute the output tensor shape based on the axes and if keepDims is set. <a href="namespacearmnn.html#a9d43aa01b882d8776ab9403eea3cf779">More...</a><br /></td></tr> |
| 189 | <tr class="separator:a9d43aa01b882d8776ab9403eea3cf779"><td class="memSeparator" colspan="2"> </td></tr> |
| 190 | </table> |
| 191 | <h2 class="groupheader">Macro Definition Documentation</h2> |
| 192 | <a id="a97d67174b2c48969547ba4bac79e5ea2"></a> |
| 193 | <h2 class="memtitle"><span class="permalink"><a href="#a97d67174b2c48969547ba4bac79e5ea2">◆ </a></span>IS_MULTI_AXES_REDUCE_SUPPORTED</h2> |
| 194 | |
| 195 | <div class="memitem"> |
| 196 | <div class="memproto"> |
| 197 | <table class="memname"> |
| 198 | <tr> |
| 199 | <td class="memname">#define IS_MULTI_AXES_REDUCE_SUPPORTED</td> |
| 200 | <td>(</td> |
| 201 | <td class="paramtype"> </td> |
| 202 | <td class="paramname">func, </td> |
| 203 | </tr> |
| 204 | <tr> |
| 205 | <td class="paramkey"></td> |
| 206 | <td></td> |
| 207 | <td class="paramtype"> </td> |
| 208 | <td class="paramname">input, </td> |
| 209 | </tr> |
| 210 | <tr> |
| 211 | <td class="paramkey"></td> |
| 212 | <td></td> |
| 213 | <td class="paramtype"> </td> |
| 214 | <td class="paramname">desc, </td> |
| 215 | </tr> |
| 216 | <tr> |
| 217 | <td class="paramkey"></td> |
| 218 | <td></td> |
| 219 | <td class="paramtype"> </td> |
| 220 | <td class="paramname">status </td> |
| 221 | </tr> |
| 222 | <tr> |
| 223 | <td></td> |
| 224 | <td>)</td> |
| 225 | <td></td><td></td> |
| 226 | </tr> |
| 227 | </table> |
| 228 | </div><div class="memdoc"> |
| 229 | <b>Value:</b><div class="fragment"><div class="line"> <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> inputTensorInfo = input; \</div> |
| 230 | <div class="line"> unsigned <span class="keywordtype">int</span> recalulatedAxis = 0; \</div> |
| 231 | <div class="line"> std::vector<uint32_t> axes; \</div> |
| 232 | <div class="line"> \</div> |
| 233 | <div class="line"> for (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i != desc.m_vAxis.size(); ++i) \</div> |
| 234 | <div class="line"> { \</div> |
| 235 | <div class="line"> axes.emplace_back(desc.m_vAxis[i]); \</div> |
| 236 | <div class="line"> \</div> |
| 237 | <div class="line"> const <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>& reducedTensorInfo = \</div> |
| 238 | <div class="line"> ComputeReductionTensorShape(input, axes, desc.m_KeepDims); \</div> |
| 239 | <div class="line"> \</div> |
| 240 | <div class="line"> std::vector<uint32_t> singleAxis(1, desc.m_vAxis[i] - recalulatedAxis); \</div> |
| 241 | <div class="line"> \</div> |
| 242 | <div class="line"> armnn::ReduceDescriptor newReduceDescriptor = desc; \</div> |
| 243 | <div class="line"> newReduceDescriptor.m_vAxis.assign(singleAxis.begin(), singleAxis.end()); \</div> |
| 244 | <div class="line"> \</div> |
| 245 | <div class="line"> status = func(inputTensorInfo, reducedTensorInfo, newReduceDescriptor); \</div> |
| 246 | <div class="line"> if (!status) \</div> |
| 247 | <div class="line"> { \</div> |
| 248 | <div class="line"> break; \</div> |
| 249 | <div class="line"> } \</div> |
| 250 | <div class="line"> \</div> |
| 251 | <div class="line"> if (!desc.m_KeepDims) \</div> |
| 252 | <div class="line"> { \</div> |
| 253 | <div class="line"> recalulatedAxis++; \</div> |
| 254 | <div class="line"> } \</div> |
| 255 | <div class="line"> \</div> |
| 256 | <div class="line"> inputTensorInfo = reducedTensorInfo; \</div> |
| 257 | <div class="line"> }</div> |
| 258 | </div><!-- fragment --> |
| 259 | <p>Macro function check if layer with multiple axes is supported on each backend. </p> |
| 260 | |
| 261 | <p class="definition">Definition at line <a class="el" href="_arm_compute_utils_8hpp_source.html#l00405">405</a> of file <a class="el" href="_arm_compute_utils_8hpp_source.html">ArmComputeUtils.hpp</a>.</p> |
| 262 | |
| 263 | </div> |
| 264 | </div> |
| 265 | </div><!-- contents --> |
| 266 | </div><!-- doc-content --> |
| 267 | <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> |
| 268 | <!-- start footer part --> |
| 269 | <div id="nav-path" class="navpath"><!-- id is needed for treeview function! --> |
| 270 | <ul> |
| 271 | <li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.html">src</a></li><li class="navelem"><a class="el" href="dir_0f3cdec46afbc61a1ded8e1687c9c9a0.html">backends</a></li><li class="navelem"><a class="el" href="dir_c13beb47b846b3a63741c705c772cf8d.html">aclCommon</a></li><li class="navelem"><a class="el" href="_arm_compute_utils_8hpp.html">ArmComputeUtils.hpp</a></li> |
| 272 | <li class="footer">Generated on Tue Aug 22 2023 11:37:02 for Arm NN by |
| 273 | <a href="http://www.doxygen.org/index.html"> |
| 274 | <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li> |
| 275 | </ul> |
| 276 | </div> |
| 277 | </body> |
| 278 | </html> |