Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / refs/heads/gh-pages / . / latest / _workload_utils_8hpp.html
blob: 28740142d8706749757a68e7c482edb031be83ba [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/backendsCommon/WorkloadUtils.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('_workload_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="#func-members">Functions</a> </div>
98 <div class="headertitle">
99<div class="title">WorkloadUtils.hpp File Reference</div> </div>
100</div><!--header-->
101<div class="contents">
102<div class="textblock"><code>#include &lt;<a class="el" href="_i_tensor_handle_8hpp_source.html">armnn/backends/ITensorHandle.hpp</a>&gt;</code><br />
103<code>#include &lt;<a class="el" href="_tensor_handle_8hpp_source.html">armnn/backends/TensorHandle.hpp</a>&gt;</code><br />
104<code>#include &lt;<a class="el" href="_tensor_8hpp_source.html">armnn/Tensor.hpp</a>&gt;</code><br />
105<code>#include &lt;<a class="el" href="_polymorphic_downcast_8hpp_source.html">armnn/utility/PolymorphicDowncast.hpp</a>&gt;</code><br />
106<code>#include &lt;<a class="el" href="_permute_8hpp_source.html">armnnUtils/Permute.hpp</a>&gt;</code><br />
107<code>#include &lt;<a class="el" href="_half_8hpp_source.html">Half.hpp</a>&gt;</code><br />
108<code>#include &lt;<a class="el" href="_profiling_8hpp_source.html">Profiling.hpp</a>&gt;</code><br />
109</div><div class="textblock"><div class="dynheader">
110Include dependency graph for WorkloadUtils.hpp:</div>
111<div class="dyncontent">
112<div class="center"><iframe scrolling="no" frameborder="0" src="_workload_utils_8hpp__incl.svg" width="3263" height="664"><p><b>This browser is not able to show SVG: try Firefox, Chrome, Safari, or Opera instead.</b></p></iframe>
113</div>
114</div>
115</div><div class="textblock"><div class="dynheader">
116This graph shows which files directly or indirectly include this file:</div>
117<div class="dyncontent">
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]118<div class="center"><iframe scrolling="no" frameborder="0" src="_workload_utils_8hpp__dep__incl.svg" width="7406" height="246"><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]119</div>
120</div>
121</div>
122<p><a href="_workload_utils_8hpp_source.html">Go to the source code of this file.</a></p>
123<table class="memberdecls">
124<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="namespaces"></a>
125Namespaces</h2></td></tr>
126<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>
127<tr class="memdesc:namespacearmnn"><td class="mdescLeft">&#160;</td><td class="mdescRight">Copyright (c) 2021 ARM Limited and Contributors. <br /></td></tr>
128<tr class="separator:"><td class="memSeparator" colspan="2">&#160;</td></tr>
129</table><table class="memberdecls">
130<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
131Functions</h2></td></tr>
132<tr class="memitem:a92c91193007aa49f4732d6dba5397f8d"><td class="memTemplParams" colspan="2">template&lt;typename CopyFunc &gt; </td></tr>
133<tr class="memitem:a92c91193007aa49f4732d6dba5397f8d"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a92c91193007aa49f4732d6dba5397f8d">CopyTensorContentsGeneric</a> (const ITensorHandle *srcTensor, ITensorHandle *dstTensor, CopyFunc copy)</td></tr>
134<tr class="separator:a92c91193007aa49f4732d6dba5397f8d"><td class="memSeparator" colspan="2">&#160;</td></tr>
135<tr class="memitem:a844d7fa4dde82e1c6b0606e1c68890bb"><td class="memTemplParams" colspan="2">template&lt;typename SrcTensorHandleType , typename DstTensorHandleType , typename DescriptorType &gt; </td></tr>
136<tr class="memitem:a844d7fa4dde82e1c6b0606e1c68890bb"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a844d7fa4dde82e1c6b0606e1c68890bb">GatherTensorHandlePairs</a> (const DescriptorType &amp;descriptor, std::vector&lt; std::pair&lt; SrcTensorHandleType *, DstTensorHandleType * &gt;&gt; &amp;tensorHandlePairs)</td></tr>
137<tr class="separator:a844d7fa4dde82e1c6b0606e1c68890bb"><td class="memSeparator" colspan="2">&#160;</td></tr>
138<tr class="memitem:ad69ffa576a596b9eb20ab6a41420c541"><td class="memItemLeft" align="right" valign="top">int32_t&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ad69ffa576a596b9eb20ab6a41420c541">ConvertMaskToACLFormat</a> (int32_t mask, int32_t numDim)</td></tr>
139<tr class="separator:ad69ffa576a596b9eb20ab6a41420c541"><td class="memSeparator" colspan="2">&#160;</td></tr>
140<tr class="memitem:aafe6180ef80d9f334f3a3ba9cc0db35d"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classarmnn_1_1_const_tensor.html">armnn::ConstTensor</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aafe6180ef80d9f334f3a3ba9cc0db35d">PermuteTensor</a> (const ConstTensorHandle *tensor, const PermutationVector &amp;permutationVector, void *permuteBuffer)</td></tr>
141<tr class="separator:aafe6180ef80d9f334f3a3ba9cc0db35d"><td class="memSeparator" colspan="2">&#160;</td></tr>
142<tr class="memitem:a3170fdd696155a247ecd81d445c0e2e1"><td class="memItemLeft" align="right" valign="top">void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a3170fdd696155a247ecd81d445c0e2e1">ReshapeWeightsForAcl</a> (TensorInfo &amp;weightInfo, DataLayout dataLayout)</td></tr>
143<tr class="separator:a3170fdd696155a247ecd81d445c0e2e1"><td class="memSeparator" colspan="2">&#160;</td></tr>
144<tr class="memitem:a1e8288eac7e909fdb58b6113d816763a"><td class="memItemLeft" align="right" valign="top">TensorInfo&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a1e8288eac7e909fdb58b6113d816763a">ConvertWeightTensorInfoFromArmnnToAcl</a> (const TensorInfo &amp;weightInfo, DataLayout dataLayout)</td></tr>
145<tr class="separator:a1e8288eac7e909fdb58b6113d816763a"><td class="memSeparator" colspan="2">&#160;</td></tr>
146<tr class="memitem:ac4aa9e41515b354234645f115c49de32"><td class="memItemLeft" align="right" valign="top">std::tuple&lt; TensorInfo, unsigned int &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ac4aa9e41515b354234645f115c49de32">Convert1HWOTensorInfoToAcl</a> (const TensorInfo &amp;weightInfo, const TensorInfo &amp;inputInfo, const DataLayout dataLayout)</td></tr>
147<tr class="memdesc:ac4aa9e41515b354234645f115c49de32"><td class="mdescLeft">&#160;</td><td class="mdescRight">Weights for depthwise have a datalayout of [1,H,W,O] = [1,H,W,I*M] This function coverts a <a class="el" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> from [1,H,W,I*M] to [1,I*M,H,W] (if NCHW) or keeps it at [1,H,W,I*M] (if NHWC) as required by the compute library Returns a tuple of converted weights tensor info and depth multiplier. <a href="namespacearmnn.html#ac4aa9e41515b354234645f115c49de32">More...</a><br /></td></tr>
148<tr class="separator:ac4aa9e41515b354234645f115c49de32"><td class="memSeparator" colspan="2">&#160;</td></tr>
149<tr class="memitem:a8ca9f249dc67c111b8234b2c78d672cd"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classarmnn_1_1_const_tensor.html">armnn::ConstTensor</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a8ca9f249dc67c111b8234b2c78d672cd">ConvertWeightTensorFromArmnnToAcl</a> (const ConstTensorHandle *weightTensor, DataLayout dataLayout, void *permuteBuffer)</td></tr>
150<tr class="separator:a8ca9f249dc67c111b8234b2c78d672cd"><td class="memSeparator" colspan="2">&#160;</td></tr>
151<tr class="memitem:af35f79341ec6c10a8bd4c8caf0585ffb"><td class="memItemLeft" align="right" valign="top">std::tuple&lt; ConstTensor, unsigned int &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#af35f79341ec6c10a8bd4c8caf0585ffb">Convert1HWOTensorToAcl</a> (const ConstTensorHandle *weightTensor, const TensorInfo &amp;inputInfo, const DataLayout dataLayout, void *permuteBuffer)</td></tr>
152<tr class="memdesc:af35f79341ec6c10a8bd4c8caf0585ffb"><td class="mdescLeft">&#160;</td><td class="mdescRight">Weights for depthwise have a datalayout of [1,H,W,O] = [1,H,W,I*M] This function coverts a ConstCpuTensorHandle from [1,H,W,I*M] to [1,I*M,H,W] (if NCHW) or keeps it at [1,H,W,I*M] (if NHWC) as required by the compute library. <a href="namespacearmnn.html#af35f79341ec6c10a8bd4c8caf0585ffb">More...</a><br /></td></tr>
153<tr class="separator:af35f79341ec6c10a8bd4c8caf0585ffb"><td class="memSeparator" colspan="2">&#160;</td></tr>
154<tr class="memitem:aa22a82f5240a0eb0d61135345080aa2d"><td class="memItemLeft" align="right" valign="top">std::tuple&lt; ConstTensor, unsigned int &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#aa22a82f5240a0eb0d61135345080aa2d">Convert1HWOtoMIHW</a> (const ConstTensorHandle *weightTensor, const TensorInfo &amp;inputInfo, const DataLayout &amp;dataLayout, void *permuteBuffer)</td></tr>
155<tr class="memdesc:aa22a82f5240a0eb0d61135345080aa2d"><td class="mdescLeft">&#160;</td><td class="mdescRight">Converts a (weights) tensor from [1, H, W, I*M] = [1, H, W, O] to [M, I, H, W]. <a href="namespacearmnn.html#aa22a82f5240a0eb0d61135345080aa2d">More...</a><br /></td></tr>
156<tr class="separator:aa22a82f5240a0eb0d61135345080aa2d"><td class="memSeparator" colspan="2">&#160;</td></tr>
157<tr class="memitem:ac40d3e4035af5fbe68d9e126a8d6367c"><td class="memItemLeft" align="right" valign="top">std::map&lt; std::string, unsigned int &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#ac40d3e4035af5fbe68d9e126a8d6367c">CalculateGatherNdKeyIndices</a> (TensorInfo inputInfo0, TensorInfo inputInfo1)</td></tr>
158<tr class="memdesc:ac40d3e4035af5fbe68d9e126a8d6367c"><td class="mdescLeft">&#160;</td><td class="mdescRight">Calculates the key index values needed for GatherNd: N, ND, K, W, C (N is always 1) <a href="namespacearmnn.html#ac40d3e4035af5fbe68d9e126a8d6367c">More...</a><br /></td></tr>
159<tr class="separator:ac40d3e4035af5fbe68d9e126a8d6367c"><td class="memSeparator" colspan="2">&#160;</td></tr>
160<tr class="memitem:a15e2ba06d2ecd7ff6013118838e5d1be"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classarmnn_1_1_permutation_vector.html">armnn::PermutationVector</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a15e2ba06d2ecd7ff6013118838e5d1be">GeneratePermutationVectorOnLastTwoDimensions</a> (unsigned int rank)</td></tr>
161<tr class="memdesc:a15e2ba06d2ecd7ff6013118838e5d1be"><td class="mdescLeft">&#160;</td><td class="mdescRight">Generates a permutation vector of size rank that permutes the 2 most right dimensions. <a href="namespacearmnn.html#a15e2ba06d2ecd7ff6013118838e5d1be">More...</a><br /></td></tr>
162<tr class="separator:a15e2ba06d2ecd7ff6013118838e5d1be"><td class="memSeparator" colspan="2">&#160;</td></tr>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]163<tr class="memitem:a2f68926d52d1aa3590bd467e041def05"><td class="memItemLeft" align="right" valign="top">std::set&lt; unsigned int &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacearmnn.html#a2f68926d52d1aa3590bd467e041def05">ComputeSplitAxis</a> (const <a class="el" href="namespacearmnn.html#a60291543fe872b795e71e05bcd835fd1">armnn::SplitterDescriptor</a> &amp;desc, const TensorShape &amp;input)</td></tr>
164<tr class="memdesc:a2f68926d52d1aa3590bd467e041def05"><td class="mdescLeft">&#160;</td><td class="mdescRight">Calculates the axis values for split operation. <a href="namespacearmnn.html#a2f68926d52d1aa3590bd467e041def05">More...</a><br /></td></tr>
165<tr class="separator:a2f68926d52d1aa3590bd467e041def05"><td class="memSeparator" colspan="2">&#160;</td></tr>
Nikhil Raj03c7ff32023-08-22 12:00:04 +0100[diff] [blame]166</table>
167</div><!-- contents -->
168</div><!-- doc-content -->
169<!-- start footer part -->
170<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
171 <ul>
172 <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_797a213d7d01b98ef12d53b0820ea64e.html">backendsCommon</a></li><li class="navelem"><a class="el" href="_workload_utils_8hpp.html">WorkloadUtils.hpp</a></li>
Nikhil Raj1dc83fe2024-05-16 09:47:51 +0100[diff] [blame]173 <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]174 <a href="http://www.doxygen.org/index.html">
175 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.17 </li>
176 </ul>
177</div>
178</body>
179</html>