1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
|
<!-- 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: CustomMemoryAllocatorSample.cpp</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">23.11</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('_custom_memory_allocator_sample_8cpp-example.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">CustomMemoryAllocatorSample.cpp</div> </div>
</div><!--header-->
<div class="contents">
<p>This example is basically a copy of the SimpleSample example. But it makes use of a CustomAllocator to allocate memory for the inputs, outputs and inter layer memory.</p>
<dl class="section note"><dt>Note</dt><dd>This is currently an experimental interface</dd></dl>
<div class="fragment"><div class="line"><span class="comment">//</span></div>
<div class="line"><span class="comment">// Copyright © 2021, 2023 Arm Ltd and Contributors. All rights reserved.</span></div>
<div class="line"><span class="comment">// SPDX-License-Identifier: MIT</span></div>
<div class="line"><span class="comment">//</span></div>
<div class="line"> </div>
<div class="line"><span class="preprocessor">#include <<a class="code" href="_arm_n_n_8hpp.html">armnn/ArmNN.hpp</a>></span></div>
<div class="line"><span class="preprocessor">#include <<a class="code" href="_i_custom_allocator_8hpp.html">armnn/backends/ICustomAllocator.hpp</a>></span></div>
<div class="line"> </div>
<div class="line"><span class="preprocessor">#include <arm_compute/core/CL/CLKernelLibrary.h></span></div>
<div class="line"><span class="preprocessor">#include <arm_compute/runtime/CL/CLScheduler.h></span></div>
<div class="line"> </div>
<div class="line"><span class="preprocessor">#include <iostream></span></div>
<div class="line"><span class="comment"></span> </div>
<div class="line"><span class="comment">/** Sample implementation of ICustomAllocator for use with the ClBackend.</span></div>
<div class="line"><span class="comment"> * Note: any memory allocated must be host addressable with write access</span></div>
<div class="line"><span class="comment"> * in order for ArmNN to be able to properly use it. */</span></div>
<div class="line"><span class="keyword">class </span>SampleClBackendCustomAllocator : <span class="keyword">public</span> <a name="_a0"></a><a class="code" href="classarmnn_1_1_i_custom_allocator.html">armnn::ICustomAllocator</a></div>
<div class="line">{</div>
<div class="line"><span class="keyword">public</span>:</div>
<div class="line"> SampleClBackendCustomAllocator() = <span class="keywordflow">default</span>;</div>
<div class="line"> </div>
<div class="line"> <span class="keywordtype">void</span>* <a name="a1"></a><a class="code" href="classarmnn_1_1_i_custom_allocator.html#a6ce9dda753afa9abb498f3d6b673e595">allocate</a>(<span class="keywordtype">size_t</span> size, <span class="keywordtype">size_t</span> alignment)<span class="keyword"> override</span></div>
<div class="line"><span class="keyword"> </span>{</div>
<div class="line"> <span class="comment">// If alignment is 0 just use the CL_DEVICE_GLOBAL_MEM_CACHELINE_SIZE for alignment</span></div>
<div class="line"> <span class="keywordflow">if</span> (alignment == 0)</div>
<div class="line"> {</div>
<div class="line"> alignment = arm_compute::CLKernelLibrary::get().get_device().getInfo<CL_DEVICE_GLOBAL_MEM_CACHELINE_SIZE>();</div>
<div class="line"> }</div>
<div class="line"> <span class="keywordtype">size_t</span> space = size + alignment + alignment;</div>
<div class="line"> <span class="keyword">auto</span> allocatedMemPtr = std::malloc(space * <span class="keyword">sizeof</span>(<span class="keywordtype">size_t</span>));</div>
<div class="line"> </div>
<div class="line"> <span class="keywordflow">if</span> (std::align(alignment, size, allocatedMemPtr, space) == <span class="keyword">nullptr</span>)</div>
<div class="line"> {</div>
<div class="line"> <span class="keywordflow">throw</span> <a name="_a2"></a><a class="code" href="classarmnn_1_1_exception.html">armnn::Exception</a>(<span class="stringliteral">"SampleClBackendCustomAllocator::Alignment failed"</span>);</div>
<div class="line"> }</div>
<div class="line"> <span class="keywordflow">return</span> allocatedMemPtr;</div>
<div class="line"> }</div>
<div class="line"> </div>
<div class="line"> <span class="keywordtype">void</span> <a name="a3"></a><a class="code" href="classarmnn_1_1_i_custom_allocator.html#a7dbeba9295a3f52ea54698c31e555dff">free</a>(<span class="keywordtype">void</span>* ptr)<span class="keyword"> override</span></div>
<div class="line"><span class="keyword"> </span>{</div>
<div class="line"> std::free(ptr);</div>
<div class="line"> }</div>
<div class="line"> </div>
<div class="line"> <a class="code" href="namespacearmnn.html#a14fcd7f88d11cea0a018269dca5f9277">armnn::MemorySource</a> <a name="a4"></a><a class="code" href="classarmnn_1_1_i_custom_allocator.html#abf02353c52af045a9af48bb40f857ad2">GetMemorySourceType</a>()<span class="keyword"> override</span></div>
<div class="line"><span class="keyword"> </span>{</div>
<div class="line"> <span class="keywordflow">return</span> <a name="a5"></a><a class="code" href="namespacearmnn.html#a14fcd7f88d11cea0a018269dca5f9277a1131a914388fac73e5f07b0ba0aad523">armnn::MemorySource::Malloc</a>;</div>
<div class="line"> }</div>
<div class="line">};</div>
<div class="line"> </div>
<div class="line"> </div>
<div class="line"><span class="comment">// A simple example application to show the usage of a custom memory allocator. In this sample, the users single</span></div>
<div class="line"><span class="comment">// input number is multiplied by 1.0f using a fully connected layer with a single neuron to produce an output</span></div>
<div class="line"><span class="comment">// number that is the same as the input. All memory required to execute this mini network is allocated with</span></div>
<div class="line"><span class="comment">// the provided custom allocator.</span></div>
<div class="line"><span class="comment">//</span></div>
<div class="line"><span class="comment">// Using a Custom Allocator is required for use with Protected Mode and Protected Memory.</span></div>
<div class="line"><span class="comment">// This example is provided using only unprotected malloc as Protected Memory is platform</span></div>
<div class="line"><span class="comment">// and implementation specific.</span></div>
<div class="line"><span class="comment">//</span></div>
<div class="line"><span class="comment">// Note: This example is similar to the SimpleSample application that can also be found in armnn/samples.</span></div>
<div class="line"><span class="comment">// The differences are in the use of a custom allocator, the backend is GpuAcc, and the inputs/outputs</span></div>
<div class="line"><span class="comment">// are being imported instead of copied. (Import must be enabled when using a Custom Allocator)</span></div>
<div class="line"><span class="comment">// You might find this useful for comparison.</span></div>
<div class="line"><span class="keywordtype">int</span> <a name="a6"></a><a class="code" href="_armnn_converter_8cpp.html#a0ddf1224851353fc92bfbff6f499fa97">main</a>()</div>
<div class="line">{</div>
<div class="line"> <span class="keyword">using namespace </span><a class="code" href="namespacearmnn.html">armnn</a>;</div>
<div class="line"> </div>
<div class="line"> <span class="keywordtype">float</span> number;</div>
<div class="line"> std::cout << <span class="stringliteral">"Please enter a number: "</span> << std::endl;</div>
<div class="line"> std::cin >> number;</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Turn on logging to standard output</span></div>
<div class="line"> <span class="comment">// This is useful in this sample so that users can learn more about what is going on</span></div>
<div class="line"> <a name="a7"></a><a class="code" href="namespacearmnn.html#aa59f7a819c3e29d10ffc41e5c0616872">ConfigureLogging</a>(<span class="keyword">true</span>, <span class="keyword">false</span>, <a name="a8"></a><a class="code" href="namespacearmnn.html#a93a3ba385cad27c4774e5fe64c025d3da4059b0251f66a18cb56f544728796875">LogSeverity::Info</a>);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Construct ArmNN network</span></div>
<div class="line"> <a class="code" href="namespacearmnn.html#a0d8160388a127c1a23b37bc88dc6e2ec">NetworkId</a> networkIdentifier;</div>
<div class="line"> <a class="code" href="namespacearmnn.html#ace74f6f9feb95a964a49d79458232703">INetworkPtr</a> network = <a name="a9"></a><a class="code" href="classarmnn_1_1_i_network.html#a41ce159095e95f7cd4174ce5d4662697">INetwork::Create</a>();</div>
<div class="line"> <a name="_a10"></a><a class="code" href="structarmnn_1_1_fully_connected_descriptor.html">FullyConnectedDescriptor</a> fullyConnectedDesc;</div>
<div class="line"> <span class="keywordtype">float</span> weightsData[] = {1.0f}; <span class="comment">// Identity</span></div>
<div class="line"> <a name="_a11"></a><a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> weightsInfo(<a name="_a12"></a><a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>({1, 1}), <a name="a13"></a><a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">DataType::Float32</a>, 0.0f, 0, <span class="keyword">true</span>);</div>
<div class="line"> weightsInfo.SetConstant(<span class="keyword">true</span>);</div>
<div class="line"> <a name="_a14"></a><a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a> weights(weightsInfo, weightsData);</div>
<div class="line"> </div>
<div class="line"> <a name="_a15"></a><a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* inputLayer = network->AddInputLayer(0);</div>
<div class="line"> <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* weightsLayer = network->AddConstantLayer(weights, <span class="stringliteral">"Weights"</span>);</div>
<div class="line"> <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* fullyConnectedLayer =</div>
<div class="line"> network->AddFullyConnectedLayer(fullyConnectedDesc, <span class="stringliteral">"fully connected"</span>);</div>
<div class="line"> <a class="code" href="classarmnn_1_1_i_connectable_layer.html">IConnectableLayer</a>* outputLayer = network->AddOutputLayer(0);</div>
<div class="line"> </div>
<div class="line"> inputLayer-><a name="a16"></a><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(0).<a name="a17"></a><a class="code" href="classarmnn_1_1_i_output_slot.html#ac1835f8756a9f03c02fcf9664e3a0fce">Connect</a>(fullyConnectedLayer-><a name="a18"></a><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div>
<div class="line"> 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>(fullyConnectedLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(1));</div>
<div class="line"> fullyConnectedLayer-><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>(outputLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a6ec9e0eb66d7d6a01240492a0b18104c">GetInputSlot</a>(0));</div>
<div class="line"> weightsLayer-><a class="code" href="classarmnn_1_1_i_connectable_layer.html#a80ac4eda2e7f2757ec9dd96fc96dbd16">GetOutputSlot</a>(0).<a name="a19"></a><a class="code" href="classarmnn_1_1_i_output_slot.html#a5ee4a6c9a2481245487b1b1a70d20fd0">SetTensorInfo</a>(weightsInfo);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Create ArmNN runtime:</span></div>
<div class="line"> <span class="comment">//</span></div>
<div class="line"> <span class="comment">// This is the interesting bit when executing a model with a custom allocator.</span></div>
<div class="line"> <span class="comment">// You can have different allocators for different backends. To support this</span></div>
<div class="line"> <span class="comment">// the runtime creation option has a map that takes a BackendId and the corresponding</span></div>
<div class="line"> <span class="comment">// allocator that should be used for that backend.</span></div>
<div class="line"> <span class="comment">// Only GpuAcc supports a Custom Allocator for now</span></div>
<div class="line"> <span class="comment">//</span></div>
<div class="line"> <span class="comment">// Note: This is not covered in this example but if you want to run a model on</span></div>
<div class="line"> <span class="comment">// protected memory a custom allocator needs to be provided that supports</span></div>
<div class="line"> <span class="comment">// protected memory allocations and the MemorySource of that allocator is</span></div>
<div class="line"> <span class="comment">// set to MemorySource::DmaBufProtected</span></div>
<div class="line"> <a name="_a20"></a><a class="code" href="structarmnn_1_1_i_runtime_1_1_creation_options.html">IRuntime::CreationOptions</a> options;</div>
<div class="line"> <span class="keyword">auto</span> customAllocator = std::make_shared<SampleClBackendCustomAllocator>();</div>
<div class="line"> options.<a name="a21"></a><a class="code" href="structarmnn_1_1_i_runtime_1_1_creation_options.html#a939528f239e70c85f833c87c5fe41d83">m_CustomAllocatorMap</a> = {{<span class="stringliteral">"GpuAcc"</span>, std::move(customAllocator)}};</div>
<div class="line"> <a class="code" href="namespacearmnn.html#a150468a02bd7b2d2d061c4aaaee939f0">IRuntimePtr</a> runtime = <a name="a22"></a><a class="code" href="classarmnn_1_1_i_runtime.html#ad44ecd3700748dc30dc4bbe34ba5bde7">IRuntime::Create</a>(options);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">//Set the tensors in the network.</span></div>
<div class="line"> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> inputTensorInfo(<a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>({1, 1}), <a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">DataType::Float32</a>);</div>
<div class="line"> inputLayer-><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>(inputTensorInfo);</div>
<div class="line"> </div>
<div class="line"> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> numElements = inputTensorInfo.GetNumElements();</div>
<div class="line"> <span class="keywordtype">size_t</span> totalBytes = numElements * <span class="keyword">sizeof</span>(float);</div>
<div class="line"> </div>
<div class="line"> <a class="code" href="classarmnn_1_1_tensor_info.html">TensorInfo</a> outputTensorInfo(<a class="code" href="classarmnn_1_1_tensor_shape.html">TensorShape</a>({1, 1}), <a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">DataType::Float32</a>);</div>
<div class="line"> fullyConnectedLayer-><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>(outputTensorInfo);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Optimise ArmNN network</span></div>
<div class="line"> <a name="_a23"></a><a class="code" href="classarmnn_1_1_optimizer_options_opaque.html">OptimizerOptionsOpaque</a> optOptions;</div>
<div class="line"> optOptions.<a name="a24"></a><a class="code" href="classarmnn_1_1_optimizer_options_opaque.html#a9258f2763407b5dec4c88f7f6109cbd7">SetImportEnabled</a>(<span class="keyword">true</span>);</div>
<div class="line"> <a class="code" href="namespacearmnn.html#a674efcf6cbdb9e831d653ff0e821fb38">IOptimizedNetworkPtr</a> optNet =</div>
<div class="line"> <a name="a25"></a><a class="code" href="namespacearmnn.html#aa42e128b41f4e966fc901f9bf42c5a1c">Optimize</a>(*network, {<span class="stringliteral">"GpuAcc"</span>}, runtime->GetDeviceSpec(), optOptions);</div>
<div class="line"> <span class="keywordflow">if</span> (!optNet)</div>
<div class="line"> {</div>
<div class="line"> <span class="comment">// This shouldn't happen for this simple sample, with GpuAcc backend.</span></div>
<div class="line"> <span class="comment">// But in general usage Optimize could fail if the backend at runtime cannot</span></div>
<div class="line"> <span class="comment">// support the model that has been provided.</span></div>
<div class="line"> std::cerr << <span class="stringliteral">"Error: Failed to optimise the input network."</span> << std::endl;</div>
<div class="line"> <span class="keywordflow">return</span> 1;</div>
<div class="line"> }</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Load graph into runtime</span></div>
<div class="line"> std::string ignoredErrorMessage;</div>
<div class="line"> <a name="_a26"></a><a class="code" href="structarmnn_1_1_i_network_properties.html">INetworkProperties</a> networkProperties(<span class="keyword">false</span>, <a class="code" href="namespacearmnn.html#a14fcd7f88d11cea0a018269dca5f9277a1131a914388fac73e5f07b0ba0aad523">MemorySource::Malloc</a>, <a class="code" href="namespacearmnn.html#a14fcd7f88d11cea0a018269dca5f9277a1131a914388fac73e5f07b0ba0aad523">MemorySource::Malloc</a>);</div>
<div class="line"> runtime->LoadNetwork(networkIdentifier, std::move(optNet), ignoredErrorMessage, networkProperties);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Creates structures for input & output</span></div>
<div class="line"> <span class="keyword">const</span> <span class="keywordtype">size_t</span> alignment =</div>
<div class="line"> arm_compute::CLKernelLibrary::get().get_device().getInfo<CL_DEVICE_GLOBAL_MEM_CACHELINE_SIZE>();</div>
<div class="line"> </div>
<div class="line"> <span class="keywordtype">void</span>* alignedInputPtr = options.<a class="code" href="structarmnn_1_1_i_runtime_1_1_creation_options.html#a939528f239e70c85f833c87c5fe41d83">m_CustomAllocatorMap</a>[<span class="stringliteral">"GpuAcc"</span>]->allocate(totalBytes, alignment);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Input with negative values</span></div>
<div class="line"> <span class="keyword">auto</span>* inputPtr = <span class="keyword">reinterpret_cast<</span><span class="keywordtype">float</span>*<span class="keyword">></span>(alignedInputPtr);</div>
<div class="line"> std::fill_n(inputPtr, numElements, number);</div>
<div class="line"> </div>
<div class="line"> <span class="keywordtype">void</span>* alignedOutputPtr = options.<a class="code" href="structarmnn_1_1_i_runtime_1_1_creation_options.html#a939528f239e70c85f833c87c5fe41d83">m_CustomAllocatorMap</a>[<span class="stringliteral">"GpuAcc"</span>]->allocate(totalBytes, alignment);</div>
<div class="line"> <span class="keyword">auto</span>* outputPtr = <span class="keyword">reinterpret_cast<</span><span class="keywordtype">float</span>*<span class="keyword">></span>(alignedOutputPtr);</div>
<div class="line"> std::fill_n(outputPtr, numElements, -10.0f);</div>
<div class="line"> </div>
<div class="line"> inputTensorInfo = runtime->GetInputTensorInfo(networkIdentifier, 0);</div>
<div class="line"> inputTensorInfo.SetConstant(<span class="keyword">true</span>);</div>
<div class="line"> <a class="code" href="namespacearmnn.html#aa01bce88f89975a5a031db4cc8861527">InputTensors</a> inputTensors</div>
<div class="line"> {</div>
<div class="line"> {0, <a class="code" href="classarmnn_1_1_const_tensor.html">ConstTensor</a>(inputTensorInfo, alignedInputPtr)},</div>
<div class="line"> };</div>
<div class="line"> <a class="code" href="namespacearmnn.html#a8f091a512915d1cb29a4ebf13dfc53ea">OutputTensors</a> outputTensors</div>
<div class="line"> {</div>
<div class="line"> {0, <a name="_a27"></a><a class="code" href="classarmnn_1_1_tensor.html">Tensor</a>(runtime->GetOutputTensorInfo(networkIdentifier, 0), alignedOutputPtr)}</div>
<div class="line"> };</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Execute network</span></div>
<div class="line"> runtime->EnqueueWorkload(networkIdentifier, inputTensors, outputTensors);</div>
<div class="line"> </div>
<div class="line"> <span class="comment">// Tell the CLBackend to sync memory so we can read the output.</span></div>
<div class="line"> arm_compute::CLScheduler::get().sync();</div>
<div class="line"> <span class="keyword">auto</span>* outputResult = <span class="keyword">reinterpret_cast<</span><span class="keywordtype">float</span>*<span class="keyword">></span>(alignedOutputPtr);</div>
<div class="line"> std::cout << <span class="stringliteral">"Your number was "</span> << outputResult[0] << std::endl;</div>
<div class="line"> runtime->UnloadNetwork(networkIdentifier);</div>
<div class="line"> <span class="keywordflow">return</span> 0;</div>
<div class="line"> </div>
<div class="line">}</div>
</div><!-- fragment --> </div><!-- contents -->
</div><!-- doc-content -->
<div class="ttc" id="anamespacearmnn_html_a14fcd7f88d11cea0a018269dca5f9277a1131a914388fac73e5f07b0ba0aad523"><div class="ttname"><a href="namespacearmnn.html#a14fcd7f88d11cea0a018269dca5f9277a1131a914388fac73e5f07b0ba0aad523">armnn::MemorySource::Malloc</a></div><div class="ttdeci">@ Malloc</div></div>
<div class="ttc" id="anamespacearmnn_html_ace74f6f9feb95a964a49d79458232703"><div class="ttname"><a href="namespacearmnn.html#ace74f6f9feb95a964a49d79458232703">armnn::INetworkPtr</a></div><div class="ttdeci">std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr</div><div class="ttdef"><b>Definition:</b> <a href="_i_network_8hpp_source.html#l00339">INetwork.hpp:339</a></div></div>
<div class="ttc" id="anamespacearmnn_html_a674efcf6cbdb9e831d653ff0e821fb38"><div class="ttname"><a href="namespacearmnn.html#a674efcf6cbdb9e831d653ff0e821fb38">armnn::IOptimizedNetworkPtr</a></div><div class="ttdeci">std::unique_ptr< IOptimizedNetwork, void(*)(IOptimizedNetwork *network)> IOptimizedNetworkPtr</div><div class="ttdef"><b>Definition:</b> <a href="_i_network_8hpp_source.html#l00340">INetwork.hpp:340</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_tensor_html"><div class="ttname"><a href="classarmnn_1_1_tensor.html">armnn::Tensor</a></div><div class="ttdoc">A tensor defined by a TensorInfo (shape and data type) and a mutable backing store.</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00319">Tensor.hpp:319</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_aa01bce88f89975a5a031db4cc8861527"><div class="ttname"><a href="namespacearmnn.html#aa01bce88f89975a5a031db4cc8861527">armnn::InputTensors</a></div><div class="ttdeci">std::vector< std::pair< LayerBindingId, class ConstTensor > > InputTensors</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00392">Tensor.hpp:392</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_i_custom_allocator_html_a7dbeba9295a3f52ea54698c31e555dff"><div class="ttname"><a href="classarmnn_1_1_i_custom_allocator.html#a7dbeba9295a3f52ea54698c31e555dff">armnn::ICustomAllocator::free</a></div><div class="ttdeci">virtual void free(void *ptr)=0</div><div class="ttdoc">Interface to be implemented by the child class to free the allocated bytes.</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="aclassarmnn_1_1_i_custom_allocator_html"><div class="ttname"><a href="classarmnn_1_1_i_custom_allocator.html">armnn::ICustomAllocator</a></div><div class="ttdoc">Custom Allocator interface.</div><div class="ttdef"><b>Definition:</b> <a href="_i_custom_allocator_8hpp_source.html#l00016">ICustomAllocator.hpp:16</a></div></div>
<div class="ttc" id="anamespacearmnn_html_aa59f7a819c3e29d10ffc41e5c0616872"><div class="ttname"><a href="namespacearmnn.html#aa59f7a819c3e29d10ffc41e5c0616872">armnn::ConfigureLogging</a></div><div class="ttdeci">void ConfigureLogging(bool printToStandardOutput, bool printToDebugOutput, LogSeverity severity)</div><div class="ttdoc">Configures the logging behaviour of the ARMNN library.</div><div class="ttdef"><b>Definition:</b> <a href="_utils_8cpp_source.html#l00018">Utils.cpp:18</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_i_custom_allocator_html_a6ce9dda753afa9abb498f3d6b673e595"><div class="ttname"><a href="classarmnn_1_1_i_custom_allocator.html#a6ce9dda753afa9abb498f3d6b673e595">armnn::ICustomAllocator::allocate</a></div><div class="ttdeci">virtual void * allocate(size_t size, size_t alignment)=0</div><div class="ttdoc">Interface to be implemented by the child class to allocate bytes.</div></div>
<div class="ttc" id="anamespacearmnn_html_ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204"><div class="ttname"><a href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">armnn::DataType::Float32</a></div><div class="ttdeci">@ Float32</div></div>
<div class="ttc" id="anamespacearmnn_html_a93a3ba385cad27c4774e5fe64c025d3da4059b0251f66a18cb56f544728796875"><div class="ttname"><a href="namespacearmnn.html#a93a3ba385cad27c4774e5fe64c025d3da4059b0251f66a18cb56f544728796875">armnn::LogSeverity::Info</a></div><div class="ttdeci">@ Info</div></div>
<div class="ttc" id="a_i_custom_allocator_8hpp_html"><div class="ttname"><a href="_i_custom_allocator_8hpp.html">ICustomAllocator.hpp</a></div></div>
<div class="ttc" id="anamespacearmnn_html_a8f091a512915d1cb29a4ebf13dfc53ea"><div class="ttname"><a href="namespacearmnn.html#a8f091a512915d1cb29a4ebf13dfc53ea">armnn::OutputTensors</a></div><div class="ttdeci">std::vector< std::pair< LayerBindingId, class Tensor > > OutputTensors</div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00393">Tensor.hpp:393</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="anamespacearmnn_html_a0d8160388a127c1a23b37bc88dc6e2ec"><div class="ttname"><a href="namespacearmnn.html#a0d8160388a127c1a23b37bc88dc6e2ec">armnn::NetworkId</a></div><div class="ttdeci">int NetworkId</div><div class="ttdef"><b>Definition:</b> <a href="_i_runtime_8hpp_source.html#l00035">IRuntime.hpp:35</a></div></div>
<div class="ttc" id="anamespacearmnn_html_a150468a02bd7b2d2d061c4aaaee939f0"><div class="ttname"><a href="namespacearmnn.html#a150468a02bd7b2d2d061c4aaaee939f0">armnn::IRuntimePtr</a></div><div class="ttdeci">std::unique_ptr< IRuntime, void(*)(IRuntime *runtime)> IRuntimePtr</div><div class="ttdef"><b>Definition:</b> <a href="_i_runtime_8hpp_source.html#l00041">IRuntime.hpp:41</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_i_network_properties_html"><div class="ttname"><a href="structarmnn_1_1_i_network_properties.html">armnn::INetworkProperties</a></div><div class="ttdef"><b>Definition:</b> <a href="_i_runtime_8hpp_source.html#l00043">IRuntime.hpp:43</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="a_armnn_converter_8cpp_html_a0ddf1224851353fc92bfbff6f499fa97"><div class="ttname"><a href="_armnn_converter_8cpp.html#a0ddf1224851353fc92bfbff6f499fa97">main</a></div><div class="ttdeci">int main(int argc, char *argv[])</div><div class="ttdef"><b>Definition:</b> <a href="_armnn_converter_8cpp_source.html#l00327">ArmnnConverter.cpp:327</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="astructarmnn_1_1_i_runtime_1_1_creation_options_html"><div class="ttname"><a href="structarmnn_1_1_i_runtime_1_1_creation_options.html">armnn::IRuntime::CreationOptions</a></div><div class="ttdef"><b>Definition:</b> <a href="_i_runtime_8hpp_source.html#l00078">IRuntime.hpp:78</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_i_runtime_html_ad44ecd3700748dc30dc4bbe34ba5bde7"><div class="ttname"><a href="classarmnn_1_1_i_runtime.html#ad44ecd3700748dc30dc4bbe34ba5bde7">armnn::IRuntime::Create</a></div><div class="ttdeci">static IRuntimePtr Create(const CreationOptions &options)</div><div class="ttdef"><b>Definition:</b> <a href="_runtime_8cpp_source.html#l00052">Runtime.cpp:52</a></div></div>
<div class="ttc" id="anamespacearmnn_html_a14fcd7f88d11cea0a018269dca5f9277"><div class="ttname"><a href="namespacearmnn.html#a14fcd7f88d11cea0a018269dca5f9277">armnn::MemorySource</a></div><div class="ttdeci">MemorySource</div><div class="ttdoc">Define the Memory Source to reduce copies.</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00244">Types.hpp:244</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="anamespacearmnn_html"><div class="ttname"><a href="namespacearmnn.html">armnn</a></div><div class="ttdoc">Copyright (c) 2021 ARM Limited and Contributors.</div><div class="ttdef"><b>Definition:</b> <a href="01__00__quick__start_8dox_source.html#l00006">01_00_quick_start.dox:6</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="a_arm_n_n_8hpp_html"><div class="ttname"><a href="_arm_n_n_8hpp.html">ArmNN.hpp</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#l00327">Tensor.hpp:327</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="aclassarmnn_1_1_i_custom_allocator_html_abf02353c52af045a9af48bb40f857ad2"><div class="ttname"><a href="classarmnn_1_1_i_custom_allocator.html#abf02353c52af045a9af48bb40f857ad2">armnn::ICustomAllocator::GetMemorySourceType</a></div><div class="ttdeci">virtual armnn::MemorySource GetMemorySourceType()=0</div><div class="ttdoc">Used to specify what type of memory is being allocated by this allocator.</div></div>
<div class="ttc" id="astructarmnn_1_1_i_runtime_1_1_creation_options_html_a939528f239e70c85f833c87c5fe41d83"><div class="ttname"><a href="structarmnn_1_1_i_runtime_1_1_creation_options.html#a939528f239e70c85f833c87c5fe41d83">armnn::IRuntime::CreationOptions::m_CustomAllocatorMap</a></div><div class="ttdeci">std::map< BackendId, std::shared_ptr< ICustomAllocator > > m_CustomAllocatorMap</div><div class="ttdoc">A map to define a custom memory allocator for specific backend Ids.</div><div class="ttdef"><b>Definition:</b> <a href="_i_runtime_8hpp_source.html#l00115">IRuntime.hpp:115</a></div></div>
<div class="ttc" id="anamespacearmnn_html_aa42e128b41f4e966fc901f9bf42c5a1c"><div class="ttname"><a href="namespacearmnn.html#aa42e128b41f4e966fc901f9bf42c5a1c">armnn::Optimize</a></div><div class="ttdeci">IOptimizedNetworkPtr Optimize(const INetwork &network, const std::vector< BackendId > &backendPreferences, const IDeviceSpec &deviceSpec, const OptimizerOptionsOpaque &options=OptimizerOptionsOpaque(), Optional< std::vector< std::string > & > messages=EmptyOptional())</div><div class="ttdoc">Create an optimized version of the network.</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l02121">Network.cpp:2121</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_i_network_html_a41ce159095e95f7cd4174ce5d4662697"><div class="ttname"><a href="classarmnn_1_1_i_network.html#a41ce159095e95f7cd4174ce5d4662697">armnn::INetwork::Create</a></div><div class="ttdeci">static INetworkPtr Create(const NetworkOptions &networkOptions={})</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00676">Network.cpp:676</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_optimizer_options_opaque_html_a9258f2763407b5dec4c88f7f6109cbd7"><div class="ttname"><a href="classarmnn_1_1_optimizer_options_opaque.html#a9258f2763407b5dec4c88f7f6109cbd7">armnn::OptimizerOptionsOpaque::SetImportEnabled</a></div><div class="ttdeci">void SetImportEnabled(bool ImportState)</div><div class="ttdef"><b>Definition:</b> <a href="_network_8cpp_source.html#l00111">Network.cpp:111</a></div></div>
<div class="ttc" id="aclassarmnn_1_1_optimizer_options_opaque_html"><div class="ttname"><a href="classarmnn_1_1_optimizer_options_opaque.html">armnn::OptimizerOptionsOpaque</a></div><div class="ttdef"><b>Definition:</b> <a href="_i_network_8hpp_source.html#l00272">INetwork.hpp:272</a></div></div>
<!-- start footer part -->
<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
<ul>
<li class="footer">Generated on Wed Nov 22 2023 10:57:04 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>
|
