Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / refs/heads/gh-pages / . / latest / _arm_compute_utils_8hpp.html
blob: 81ecf0970b6272e20d3bc89a55ef1c28e59ddabb [file] [log] [blame]
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]1<!-- HTML header for doxygen 1.8.17-->
2<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
3<html xmlns="http://www.w3.org/1999/xhtml">
4<head>
5<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
6<meta http-equiv="X-UA-Compatible" content="IE=9"/>
7<meta name="generator" content="Doxygen 1.8.17"/>
8<meta name="viewport" content="width=device-width, initial-scale=1"/>
9<title>Arm NN: src/backends/aclCommon/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">
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]39 &#160;<span id="projectnumber">24.05</span>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]40 </div>
41 </td>
42 </tr>
43 </tbody>
44</table>
45</div>
46<!-- end header part -->
47<!-- Generated by Doxygen 1.8.17 -->
48<script type="text/javascript">
49/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
50var searchBox = new SearchBox("searchBox", "search",false,'Search');
51/* @license-end */
52</script>
53<script type="text/javascript" src="menudata.js"></script>
54<script type="text/javascript" src="menu.js"></script>
55<script type="text/javascript">
56/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
57$(function() {
58 initMenu('',true,false,'search.php','Search');
59 $(document).ready(function() { init_search(); });
60});
61/* @license-end */</script>
62<div id="main-nav"></div>
63</div><!-- top -->
64<div id="side-nav" class="ui-resizable side-nav-resizable">
65 <div id="nav-tree">
66 <div id="nav-tree-contents">
67 <div id="nav-sync" class="sync"></div>
68 </div>
69 </div>
70 <div id="splitbar" style="-moz-user-select:none;"
71 class="ui-resizable-handle">
72 </div>
73</div>
74<script type="text/javascript">
75/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
76$(document).ready(function(){initNavTree('_arm_compute_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> &#124;
97<a href="#define-members">Macros</a> &#124;
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 &lt;<a class="el" href="_descriptors_8hpp_source.html">armnn/Descriptors.hpp</a>&gt;</code><br />
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]104<code>#include &lt;<a class="el" href="_exceptions_8hpp_source.html">armnn/Exceptions.hpp</a>&gt;</code><br />
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]105<code>#include &lt;<a class="el" href="_tensor_8hpp_source.html">armnn/Tensor.hpp</a>&gt;</code><br />
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]106<code>#include &lt;<a class="el" href="_numeric_cast_8hpp_source.html">armnn/utility/NumericCast.hpp</a>&gt;</code><br />
107<code>#include &lt;<a class="el" href="_workload_data_8hpp_source.html">armnn/backends/WorkloadData.hpp</a>&gt;</code><br />
Nikhil Raj6f92c8e2023-11-22 11:41:15 +0000[diff] [blame]108<code>#include &lt;<a class="el" href="_tensor_utils_8hpp_source.html">armnnUtils/TensorUtils.hpp</a>&gt;</code><br />
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]109<code>#include &lt;arm_compute/runtime/FunctionDescriptors.h&gt;</code><br />
110<code>#include &lt;arm_compute/function_info/FullyConnectedLayerInfo.h&gt;</code><br />
111<code>#include &quot;<a class="el" href="_neon_reduce_workload_8hpp_source.html">neon/workloads/NeonReduceWorkload.hpp</a>&quot;</code><br />
112<code>#include &quot;<a class="el" href="_cl_reduce_workload_8hpp_source.html">cl/workloads/ClReduceWorkload.hpp</a>&quot;</code><br />
113</div><div class="textblock"><div class="dynheader">
114Include dependency graph for ArmComputeUtils.hpp:</div>
115<div class="dyncontent">
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]116<div class="center"><iframe scrolling="no" frameborder="0" src="_arm_compute_utils_8hpp__incl.svg" width="3772" height="947"><p><b>This browser is not able to show SVG: try Firefox, Chrome, Safari, or Opera instead.</b></p></iframe>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]117</div>
118</div>
119</div><div class="textblock"><div class="dynheader">
120This graph shows which files directly or indirectly include this file:</div>
121<div class="dyncontent">
Nikhil Raj38b600d2024-02-15 15:02:19 +0000[diff] [blame]122<div class="center"><iframe scrolling="no" frameborder="0" src="_arm_compute_utils_8hpp__dep__incl.svg" width="19199" height="320"><p><b>This browser is not able to show SVG: try Firefox, Chrome, Safari, or Opera instead.</b></p></iframe>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]123</div>
124</div>
125</div>
126<p><a href="_arm_compute_utils_8hpp_source.html">Go to the source code of this file.</a></p>
127<table class="memberdecls">
128<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="namespaces"></a>
129Namespaces</h2></td></tr>
130<tr class="memitem:namespacearmnn"><td class="memItemLeft" align="right" valign="top"> &#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html">armnn</a></td></tr>
131<tr class="memdesc:namespacearmnn"><td class="mdescLeft">&#160;</td><td class="mdescRight">Copyright (c) 2021 ARM Limited and Contributors. <br /></td></tr>
132<tr class="separator:"><td class="memSeparator" colspan="2">&#160;</td></tr>
133</table><table class="memberdecls">
134<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="define-members"></a>
135Macros</h2></td></tr>
136<tr class="memitem:a97d67174b2c48969547ba4bac79e5ea2"><td class="memItemLeft" align="right" valign="top">#define&#160;</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>
137<tr class="memdesc:a97d67174b2c48969547ba4bac79e5ea2"><td class="mdescLeft">&#160;</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>
138<tr class="separator:a97d67174b2c48969547ba4bac79e5ea2"><td class="memSeparator" colspan="2">&#160;</td></tr>
139</table><table class="memberdecls">
140<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
141Functions</h2></td></tr>
142<tr class="memitem:a5e783a951642781b9e7b55db06a514b7"><td class="memItemLeft" align="right" valign="top">arm_compute::NormalizationLayerInfo&#160;</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> &amp;tensorInfo, <a class="el" href="namespacearmnn.html#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> dataLayout)</td></tr>
143<tr class="separator:a5e783a951642781b9e7b55db06a514b7"><td class="memSeparator" colspan="2">&#160;</td></tr>
144<tr class="memitem:afdba36f125621d775d471f0daf613df2"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo::ActivationFunction&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#afdba36f125621d775d471f0daf613df2">ConvertActivationFunctionToAclActivationFunction</a> (ActivationFunction armnnFunction)</td></tr>
145<tr class="separator:afdba36f125621d775d471f0daf613df2"><td class="memSeparator" colspan="2">&#160;</td></tr>
146<tr class="memitem:ad701d0d29baa4266ab4d33b090aa661c"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ad701d0d29baa4266ab4d33b090aa661c">ConvertActivationDescriptorToAclActivationLayerInfo</a> (const ActivationDescriptor &amp;actDesc)</td></tr>
147<tr class="separator:ad701d0d29baa4266ab4d33b090aa661c"><td class="memSeparator" colspan="2">&#160;</td></tr>
148<tr class="memitem:af6d517245110f6b1ca702da0cbbf4518"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#af6d517245110f6b1ca702da0cbbf4518">ConvertActivationDescriptorToAclActivationLayerInfo</a> (const ActivationDescriptor *activationDescPtr)</td></tr>
149<tr class="separator:af6d517245110f6b1ca702da0cbbf4518"><td class="memSeparator" colspan="2">&#160;</td></tr>
150<tr class="memitem:abfb0841058a8190d30851f07eca3991f"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#abfb0841058a8190d30851f07eca3991f">ConvertAdditionalInfoToAclActivationLayerInfo</a> (const QueueDescriptor &amp;queueDescriptor)</td></tr>
151<tr class="separator:abfb0841058a8190d30851f07eca3991f"><td class="memSeparator" colspan="2">&#160;</td></tr>
152<tr class="memitem:aa1e93ef5f9ee3dbb5e7faa9578f180ae"><td class="memItemLeft" align="right" valign="top">arm_compute::ActivationLayerInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aa1e93ef5f9ee3dbb5e7faa9578f180ae">ConvertLstmActivationFuncToAclLayerInfo</a> (uint32_t activationFunction)</td></tr>
153<tr class="separator:aa1e93ef5f9ee3dbb5e7faa9578f180ae"><td class="memSeparator" colspan="2">&#160;</td></tr>
154<tr class="memitem:ac1438a97bbc2aee5cd8470d9d2a3aebc"><td class="memItemLeft" align="right" valign="top">arm_compute::ComparisonOperation&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ac1438a97bbc2aee5cd8470d9d2a3aebc">ConvertComparisonOperationToAcl</a> (const ComparisonDescriptor &amp;descriptor)</td></tr>
155<tr class="separator:ac1438a97bbc2aee5cd8470d9d2a3aebc"><td class="memSeparator" colspan="2">&#160;</td></tr>
156<tr class="memitem:ad256fcf8c7f4d5a240fa47f0b56d50af"><td class="memItemLeft" align="right" valign="top">arm_compute::PoolingType&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ad256fcf8c7f4d5a240fa47f0b56d50af">ConvertPoolingAlgorithmToAclPoolingType</a> (PoolingAlgorithm poolingAlgorithm)</td></tr>
157<tr class="separator:ad256fcf8c7f4d5a240fa47f0b56d50af"><td class="memSeparator" colspan="2">&#160;</td></tr>
158<tr class="memitem:a8f3bfacadfd6d2146d6ccd299dabc7aa"><td class="memItemLeft" align="right" valign="top">arm_compute::DimensionRoundingType&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a8f3bfacadfd6d2146d6ccd299dabc7aa">ConvertOutputShapeRoundingToAclDimensionRoundingType</a> (OutputShapeRounding rounding)</td></tr>
159<tr class="separator:a8f3bfacadfd6d2146d6ccd299dabc7aa"><td class="memSeparator" colspan="2">&#160;</td></tr>
160<tr class="memitem:aa5baabb8e3a4aa6cbdcab419d743e747"><td class="memItemLeft" align="right" valign="top">arm_compute::NormType&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aa5baabb8e3a4aa6cbdcab419d743e747">ConvertNormalizationAlgorithmChannelToAclNormType</a> (NormalizationAlgorithmChannel channelType)</td></tr>
161<tr class="separator:aa5baabb8e3a4aa6cbdcab419d743e747"><td class="memSeparator" colspan="2">&#160;</td></tr>
162<tr class="memitem:ae2e0c1d8160fadce44d479379790718e"><td class="memItemLeft" align="right" valign="top">arm_compute::FullyConnectedLayerInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ae2e0c1d8160fadce44d479379790718e">ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo</a> (const FullyConnectedDescriptor &amp;fullyConnectedDesc, const ActivationDescriptor *activationDesc)</td></tr>
163<tr class="separator:ae2e0c1d8160fadce44d479379790718e"><td class="memSeparator" colspan="2">&#160;</td></tr>
164<tr class="memitem:a2f12c1f572fb15978e51cf646d42d4dc"><td class="memItemLeft" align="right" valign="top">arm_compute::FullyConnectedLayerInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a2f12c1f572fb15978e51cf646d42d4dc">ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo</a> (const FullyConnectedDescriptor &amp;fullyConnectedDesc, arm_compute::ActivationLayerInfo activationLayerInfo)</td></tr>
165<tr class="separator:a2f12c1f572fb15978e51cf646d42d4dc"><td class="memSeparator" colspan="2">&#160;</td></tr>
166<tr class="memitem:ae9bdcb8ac91731109dc423d6ed476204"><td class="memItemLeft" align="right" valign="top">arm_compute::InterpolationPolicy&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ae9bdcb8ac91731109dc423d6ed476204">ConvertResizeMethodToAclInterpolationPolicy</a> (ResizeMethod resizeMethod)</td></tr>
167<tr class="separator:ae9bdcb8ac91731109dc423d6ed476204"><td class="memSeparator" colspan="2">&#160;</td></tr>
168<tr class="memitem:ae33fcc729c3eeda77a443a11f13da7ba"><td class="memTemplParams" colspan="2">template&lt;typename T &gt; </td></tr>
169<tr class="memitem:ae33fcc729c3eeda77a443a11f13da7ba"><td class="memTemplItemLeft" align="right" valign="top">T&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ae33fcc729c3eeda77a443a11f13da7ba">ComputeSoftmaxAclAxis</a> (const SoftmaxDescriptor &amp;softmaxDesc, const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &amp;tensor)</td></tr>
170<tr class="separator:ae33fcc729c3eeda77a443a11f13da7ba"><td class="memSeparator" colspan="2">&#160;</td></tr>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]171<tr class="memitem:a44a3b98b37a25c995aa9e4dae7d7b456"><td class="memItemLeft" align="right" valign="top">int&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a44a3b98b37a25c995aa9e4dae7d7b456">ComputeAclAxis</a> (const int &amp;armnnAxis, const <a class="el" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a> &amp;tensor)</td></tr>
172<tr class="memdesc:a44a3b98b37a25c995aa9e4dae7d7b456"><td class="mdescLeft">&#160;</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>
173<tr class="separator:a44a3b98b37a25c995aa9e4dae7d7b456"><td class="memSeparator" colspan="2">&#160;</td></tr>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]174<tr class="memitem:a855760e957668804d33c0691a802496e"><td class="memItemLeft" align="right" valign="top">arm_compute::Conv3dInfo&#160;</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>
175<tr class="memdesc:a855760e957668804d33c0691a802496e"><td class="mdescLeft">&#160;</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>
176<tr class="separator:a855760e957668804d33c0691a802496e"><td class="memSeparator" colspan="2">&#160;</td></tr>
177<tr class="memitem:a3f2c9ea2505a2cac1d05e17e989aff2b"><td class="memItemLeft" align="right" valign="top">arm_compute::Conv3dInfo&#160;</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>
178<tr class="separator:a3f2c9ea2505a2cac1d05e17e989aff2b"><td class="memSeparator" colspan="2">&#160;</td></tr>
179<tr class="memitem:adf5a024ace0dbe2af2d5b1a1b9623771"><td class="memItemLeft" align="right" valign="top">arm_compute::PaddingMode&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#adf5a024ace0dbe2af2d5b1a1b9623771">ConvertPaddingModeToAcl</a> (const PaddingMode &amp;paddingMode)</td></tr>
180<tr class="separator:adf5a024ace0dbe2af2d5b1a1b9623771"><td class="memSeparator" colspan="2">&#160;</td></tr>
181<tr class="memitem:a03ab984d93d0f240f31274499b72e6b0"><td class="memItemLeft" align="right" valign="top">arm_compute::ReductionOperation&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a03ab984d93d0f240f31274499b72e6b0">ConvertReductionOperationToAcl</a> (const ReduceDescriptor &amp;descriptor)</td></tr>
182<tr class="separator:a03ab984d93d0f240f31274499b72e6b0"><td class="memSeparator" colspan="2">&#160;</td></tr>
183<tr class="memitem:a9d43aa01b882d8776ab9403eea3cf779"><td class="memItemLeft" align="right" valign="top">const TensorInfo&#160;</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> &amp;input, const std::vector&lt; uint32_t &gt; &amp;vAxis, const bool keepDims)</td></tr>
184<tr class="memdesc:a9d43aa01b882d8776ab9403eea3cf779"><td class="mdescLeft">&#160;</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>
185<tr class="separator:a9d43aa01b882d8776ab9403eea3cf779"><td class="memSeparator" colspan="2">&#160;</td></tr>
186</table>
187<h2 class="groupheader">Macro Definition Documentation</h2>
188<a id="a97d67174b2c48969547ba4bac79e5ea2"></a>
189<h2 class="memtitle"><span class="permalink"><a href="#a97d67174b2c48969547ba4bac79e5ea2">&#9670;&nbsp;</a></span>IS_MULTI_AXES_REDUCE_SUPPORTED</h2>
190
191<div class="memitem">
192<div class="memproto">
193 <table class="memname">
194 <tr>
195 <td class="memname">#define IS_MULTI_AXES_REDUCE_SUPPORTED</td>
196 <td>(</td>
197 <td class="paramtype">&#160;</td>
198 <td class="paramname">func, </td>
199 </tr>
200 <tr>
201 <td class="paramkey"></td>
202 <td></td>
203 <td class="paramtype">&#160;</td>
204 <td class="paramname">input, </td>
205 </tr>
206 <tr>
207 <td class="paramkey"></td>
208 <td></td>
209 <td class="paramtype">&#160;</td>
210 <td class="paramname">desc, </td>
211 </tr>
212 <tr>
213 <td class="paramkey"></td>
214 <td></td>
215 <td class="paramtype">&#160;</td>
216 <td class="paramname">status&#160;</td>
217 </tr>
218 <tr>
219 <td></td>
220 <td>)</td>
221 <td></td><td></td>
222 </tr>
223 </table>
224</div><div class="memdoc">
225<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>
226<div class="line"> unsigned <span class="keywordtype">int</span> recalulatedAxis = 0; \</div>
227<div class="line"> std::vector&lt;uint32_t&gt; axes; \</div>
228<div class="line"> \</div>
229<div class="line"> for (<span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> i = 0; i != desc.m_vAxis.size(); ++i) \</div>
230<div class="line"> { \</div>
231<div class="line"> axes.emplace_back(desc.m_vAxis[i]); \</div>
232<div class="line"> \</div>
233<div class="line"> const <a class="code" href="classarmnn_1_1_tensor_info.html">armnn::TensorInfo</a>&amp; reducedTensorInfo = \</div>
234<div class="line"> ComputeReductionTensorShape(input, axes, desc.m_KeepDims); \</div>
235<div class="line"> \</div>
236<div class="line"> std::vector&lt;uint32_t&gt; singleAxis(1, desc.m_vAxis[i] - recalulatedAxis); \</div>
237<div class="line"> \</div>
238<div class="line"> armnn::ReduceDescriptor newReduceDescriptor = desc; \</div>
239<div class="line"> newReduceDescriptor.m_vAxis.assign(singleAxis.begin(), singleAxis.end()); \</div>
240<div class="line"> \</div>
241<div class="line"> status = func(inputTensorInfo, reducedTensorInfo, newReduceDescriptor); \</div>
242<div class="line"> if (!status) \</div>
243<div class="line"> { \</div>
244<div class="line"> break; \</div>
245<div class="line"> } \</div>
246<div class="line"> \</div>
247<div class="line"> if (!desc.m_KeepDims) \</div>
248<div class="line"> { \</div>
249<div class="line"> recalulatedAxis++; \</div>
250<div class="line"> } \</div>
251<div class="line"> \</div>
252<div class="line"> inputTensorInfo = reducedTensorInfo; \</div>
253<div class="line"> }</div>
254</div><!-- fragment -->
255<p>Macro function check if layer with multiple axes is supported on each backend. </p>
256
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]257<p class="definition">Definition at line <a class="el" href="_arm_compute_utils_8hpp_source.html#l00373">373</a> of file <a class="el" href="_arm_compute_utils_8hpp_source.html">ArmComputeUtils.hpp</a>.</p>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]258
259</div>
260</div>
261</div><!-- contents -->
262</div><!-- doc-content -->
263<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>
264<!-- start footer part -->
265<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
266 <ul>
267 <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>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]268 <li class="footer">Generated on Thu May 16 2024 09:31:52 for Arm NN by
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]269 <a href="http://www.doxygen.org/index.html">
270 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li>
271 </ul>
272</div>
273</body>
274</html>