Gitiles
Code Review Sign In
gerrit-public.fairphone.software / platform / external / ComputeLibrary / 52ba29e936b8e711e8acdfe819e36f884d4f3fe1 / . / documentation / pooling__layer__quantized_8cl.xhtml
blob: 45b8075fcc81b179174e9483a27122ae3dfa7a94 [file] [log] [blame]
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]1<!-- HTML header for doxygen 1.8.9.1-->
2<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://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.11"/>
8<meta name="robots" content="NOINDEX, NOFOLLOW" /> <!-- Prevent indexing by search engines -->
9<title>Compute Library: src/core/CL/cl_kernels/pooling_layer_quantized.cl 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<script type="text/javascript">
18 $(document).ready(initResizable);
19 $(window).load(resizeHeight);
20</script>
21<link href="search/search.css" rel="stylesheet" type="text/css"/>
22<script type="text/javascript" src="search/searchdata.js"></script>
23<script type="text/javascript" src="search/search.js"></script>
24<script type="text/javascript">
25 $(document).ready(function() { init_search(); });
26</script>
27<script type="text/x-mathjax-config">
28 MathJax.Hub.Config({
29 extensions: ["tex2jax.js"],
30 jax: ["input/TeX","output/HTML-CSS"],
31});
32</script><script type="text/javascript" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script>
33<link href="doxygen.css" rel="stylesheet" type="text/css" />
34</head>
35<body>
36<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
37<div id="titlearea">
38<table cellspacing="0" cellpadding="0">
39 <tbody>
40 <tr style="height: 56px;">
41 <td style="padding-left: 0.5em;">
42 <div id="projectname">Compute Library
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]43 &#160;<span id="projectnumber">18.08</span>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]44 </div>
45 </td>
46 </tr>
47 </tbody>
48</table>
49</div>
50<!-- end header part -->
51<!-- Generated by Doxygen 1.8.11 -->
52<script type="text/javascript">
53var searchBox = new SearchBox("searchBox", "search",false,'Search');
54</script>
55 <div id="navrow1" class="tabs">
56 <ul class="tablist">
57 <li><a href="index.xhtml"><span>Main&#160;Page</span></a></li>
58 <li><a href="pages.xhtml"><span>Related&#160;Pages</span></a></li>
59 <li><a href="namespaces.xhtml"><span>Namespaces</span></a></li>
60 <li><a href="annotated.xhtml"><span>Data&#160;Structures</span></a></li>
61 <li class="current"><a href="files.xhtml"><span>Files</span></a></li>
62 <li>
63 <div id="MSearchBox" class="MSearchBoxInactive">
64 <span class="left">
65 <img id="MSearchSelect" src="search/mag_sel.png"
66 onmouseover="return searchBox.OnSearchSelectShow()"
67 onmouseout="return searchBox.OnSearchSelectHide()"
68 alt=""/>
69 <input type="text" id="MSearchField" value="Search" accesskey="S"
70 onfocus="searchBox.OnSearchFieldFocus(true)"
71 onblur="searchBox.OnSearchFieldFocus(false)"
72 onkeyup="searchBox.OnSearchFieldChange(event)"/>
73 </span><span class="right">
74 <a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a>
75 </span>
76 </div>
77 </li>
78 </ul>
79 </div>
80 <div id="navrow2" class="tabs2">
81 <ul class="tablist">
82 <li><a href="files.xhtml"><span>File&#160;List</span></a></li>
83 <li><a href="globals.xhtml"><span>Globals</span></a></li>
84 </ul>
85 </div>
86</div><!-- top -->
87<div id="side-nav" class="ui-resizable side-nav-resizable">
88 <div id="nav-tree">
89 <div id="nav-tree-contents">
90 <div id="nav-sync" class="sync"></div>
91 </div>
92 </div>
93 <div id="splitbar" style="-moz-user-select:none;"
94 class="ui-resizable-handle">
95 </div>
96</div>
97<script type="text/javascript">
98$(document).ready(function(){initNavTree('pooling__layer__quantized_8cl.xhtml','');});
99</script>
100<div id="doc-content">
101<!-- window showing the filter options -->
102<div id="MSearchSelectWindow"
103 onmouseover="return searchBox.OnSearchSelectShow()"
104 onmouseout="return searchBox.OnSearchSelectHide()"
105 onkeydown="return searchBox.OnSearchSelectKey(event)">
106</div>
107
108<!-- iframe showing the search results (closed by default) -->
109<div id="MSearchResultsWindow">
110<iframe src="javascript:void(0)" frameborder="0"
111 name="MSearchResults" id="MSearchResults">
112</iframe>
113</div>
114
115<div class="header">
116 <div class="summary">
117<a href="#define-members">Macros</a> &#124;
118<a href="#func-members">Functions</a> </div>
119 <div class="headertitle">
120<div class="title">pooling_layer_quantized.cl File Reference</div> </div>
121</div><!--header-->
122<div class="contents">
123<div class="textblock"><code>#include &quot;<a class="el" href="helpers_8h_source.xhtml">helpers.h</a>&quot;</code><br />
124</div>
125<p><a href="pooling__layer__quantized_8cl_source.xhtml">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="define-members"></a>
128Macros</h2></td></tr>
129<tr class="memitem:a482ef7d59a5f474ca126e737c7f0978a"><td class="memItemLeft" align="right" valign="top">#define&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(x, y)&#160;&#160;&#160;((x) + (y))</td></tr>
130<tr class="separator:a482ef7d59a5f474ca126e737c7f0978a"><td class="memSeparator" colspan="2">&#160;</td></tr>
131<tr class="memitem:a5db17889d824975fefb2ce2f4690637f"><td class="memItemLeft" align="right" valign="top">#define&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a5db17889d824975fefb2ce2f4690637f">DIV_OP</a>(x, y)&#160;&#160;&#160;(x * (1.f / y))</td></tr>
132<tr class="separator:a5db17889d824975fefb2ce2f4690637f"><td class="memSeparator" colspan="2">&#160;</td></tr>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]133<tr class="memitem:a6c01fa98d360a9d52926dc6a5a599711"><td class="memItemLeft" align="right" valign="top">#define&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a6c01fa98d360a9d52926dc6a5a599711">DIV_OP_NHWC</a>(x, y)&#160;&#160;&#160;(convert_float8(x) * (float8)(1.f / y))</td></tr>
134<tr class="separator:a6c01fa98d360a9d52926dc6a5a599711"><td class="memSeparator" colspan="2">&#160;</td></tr>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]135</table><table class="memberdecls">
136<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
137Functions</h2></td></tr>
Anthony Barbier06ea0482018-02-22 15:45:35 +0000[diff] [blame]138<tr class="memitem:a4057c01dd49145c5e463f1cea52542df"><td class="memItemLeft" align="right" valign="top">int&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a4057c01dd49145c5e463f1cea52542df">calculate_avg_scale</a> (const int pool_size_x, const int pool_size_y, const int upper_bound_w, const int upper_bound_h, const int pad_x, const int pad_y, const int stride_x, const int stride_y)</td></tr>
139<tr class="separator:a4057c01dd49145c5e463f1cea52542df"><td class="memSeparator" colspan="2">&#160;</td></tr>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]140<tr class="memitem:a5c74cefcb6318451b5785dd00bd9b05f"><td class="memItemLeft" align="right" valign="top">__kernel void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a5c74cefcb6318451b5785dd00bd9b05f">pooling_layer_MxN_quantized_nchw</a> (__global uchar *input_ptr, uint input_stride_x, uint input_step_x, uint input_stride_y, uint input_step_y, uint input_stride_z, uint input_step_z, uint input_offset_first_element_in_bytes, __global uchar *output_ptr, uint output_stride_x, uint output_step_x, uint output_stride_y, uint output_step_y, uint output_stride_z, uint output_step_z, uint output_offset_first_element_in_bytes)</td></tr>
141<tr class="memdesc:a5c74cefcb6318451b5785dd00bd9b05f"><td class="mdescLeft">&#160;</td><td class="mdescRight">Performs a pooling function of pool size equal to N (NCHW) <a href="#a5c74cefcb6318451b5785dd00bd9b05f">More...</a><br /></td></tr>
142<tr class="separator:a5c74cefcb6318451b5785dd00bd9b05f"><td class="memSeparator" colspan="2">&#160;</td></tr>
143<tr class="memitem:a235920cb0f8577da6e104ccb03fbda25"><td class="memItemLeft" align="right" valign="top">int&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a235920cb0f8577da6e104ccb03fbda25">calculate_avg_scale_nhwc</a> (const int pool_size_x, const int pool_size_y, int upper_bound_w, int upper_bound_h, const int pad_x, const int pad_y, const int stride_x, const int stride_y)</td></tr>
144<tr class="separator:a235920cb0f8577da6e104ccb03fbda25"><td class="memSeparator" colspan="2">&#160;</td></tr>
145<tr class="memitem:a747409ec1866adc96ee62d1e08ec502f"><td class="memItemLeft" align="right" valign="top">__kernel void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="pooling__layer__quantized_8cl.xhtml#a747409ec1866adc96ee62d1e08ec502f">pooling_layer_MxN_quantized_nhwc</a> (__global uchar *input_ptr, uint input_stride_x, uint input_step_x, uint input_stride_y, uint input_step_y, uint input_stride_z, uint input_step_z, uint input_offset_first_element_in_bytes, __global uchar *output_ptr, uint output_stride_x, uint output_step_x, uint output_stride_y, uint output_step_y, uint output_stride_z, uint output_step_z, uint output_offset_first_element_in_bytes)</td></tr>
146<tr class="memdesc:a747409ec1866adc96ee62d1e08ec502f"><td class="mdescLeft">&#160;</td><td class="mdescRight">Performs a pooling function of pool size equal to N (NHWC) <a href="#a747409ec1866adc96ee62d1e08ec502f">More...</a><br /></td></tr>
147<tr class="separator:a747409ec1866adc96ee62d1e08ec502f"><td class="memSeparator" colspan="2">&#160;</td></tr>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]148</table>
149<h2 class="groupheader">Macro Definition Documentation</h2>
150<a class="anchor" id="a5db17889d824975fefb2ce2f4690637f"></a>
151<div class="memitem">
152<div class="memproto">
153 <table class="memname">
154 <tr>
155 <td class="memname">#define DIV_OP</td>
156 <td>(</td>
157 <td class="paramtype">&#160;</td>
158 <td class="paramname">x, </td>
159 </tr>
160 <tr>
161 <td class="paramkey"></td>
162 <td></td>
163 <td class="paramtype">&#160;</td>
164 <td class="paramname">y&#160;</td>
165 </tr>
166 <tr>
167 <td></td>
168 <td>)</td>
169 <td></td><td>&#160;&#160;&#160;(x * (1.f / y))</td>
170 </tr>
171 </table>
172</div><div class="memdoc">
173
174<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00032">32</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
175
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]176<p>Referenced by <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00080">pooling_layer_MxN_quantized_nchw()</a>.</p>
177
178</div>
179</div>
180<a class="anchor" id="a6c01fa98d360a9d52926dc6a5a599711"></a>
181<div class="memitem">
182<div class="memproto">
183 <table class="memname">
184 <tr>
185 <td class="memname">#define DIV_OP_NHWC</td>
186 <td>(</td>
187 <td class="paramtype">&#160;</td>
188 <td class="paramname">x, </td>
189 </tr>
190 <tr>
191 <td class="paramkey"></td>
192 <td></td>
193 <td class="paramtype">&#160;</td>
194 <td class="paramname">y&#160;</td>
195 </tr>
196 <tr>
197 <td></td>
198 <td>)</td>
199 <td></td><td>&#160;&#160;&#160;(convert_float8(x) * (float8)(1.f / y))</td>
200 </tr>
201 </table>
202</div><div class="memdoc">
203
204<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00034">34</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
205
206<p>Referenced by <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00166">pooling_layer_MxN_quantized_nhwc()</a>.</p>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]207
208</div>
209</div>
210<a class="anchor" id="a482ef7d59a5f474ca126e737c7f0978a"></a>
211<div class="memitem">
212<div class="memproto">
213 <table class="memname">
214 <tr>
215 <td class="memname">#define POOL_OP</td>
216 <td>(</td>
217 <td class="paramtype">&#160;</td>
218 <td class="paramname">x, </td>
219 </tr>
220 <tr>
221 <td class="paramkey"></td>
222 <td></td>
223 <td class="paramtype">&#160;</td>
224 <td class="paramname">y&#160;</td>
225 </tr>
226 <tr>
227 <td></td>
228 <td>)</td>
229 <td></td><td>&#160;&#160;&#160;((x) + (y))</td>
230 </tr>
231 </table>
232</div><div class="memdoc">
233
234<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00027">27</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
235
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]236<p>Referenced by <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00080">pooling_layer_MxN_quantized_nchw()</a>, and <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00166">pooling_layer_MxN_quantized_nhwc()</a>.</p>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]237
238</div>
239</div>
240<h2 class="groupheader">Function Documentation</h2>
Anthony Barbier06ea0482018-02-22 15:45:35 +0000[diff] [blame]241<a class="anchor" id="a4057c01dd49145c5e463f1cea52542df"></a>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]242<div class="memitem">
243<div class="memproto">
244 <table class="memname">
245 <tr>
246 <td class="memname">int calculate_avg_scale </td>
247 <td>(</td>
248 <td class="paramtype">const int&#160;</td>
Anthony Barbier06ea0482018-02-22 15:45:35 +0000[diff] [blame]249 <td class="paramname"><em>pool_size_x</em>, </td>
250 </tr>
251 <tr>
252 <td class="paramkey"></td>
253 <td></td>
254 <td class="paramtype">const int&#160;</td>
255 <td class="paramname"><em>pool_size_y</em>, </td>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]256 </tr>
257 <tr>
258 <td class="paramkey"></td>
259 <td></td>
260 <td class="paramtype">const int&#160;</td>
261 <td class="paramname"><em>upper_bound_w</em>, </td>
262 </tr>
263 <tr>
264 <td class="paramkey"></td>
265 <td></td>
266 <td class="paramtype">const int&#160;</td>
267 <td class="paramname"><em>upper_bound_h</em>, </td>
268 </tr>
269 <tr>
270 <td class="paramkey"></td>
271 <td></td>
272 <td class="paramtype">const int&#160;</td>
273 <td class="paramname"><em>pad_x</em>, </td>
274 </tr>
275 <tr>
276 <td class="paramkey"></td>
277 <td></td>
278 <td class="paramtype">const int&#160;</td>
279 <td class="paramname"><em>pad_y</em>, </td>
280 </tr>
281 <tr>
282 <td class="paramkey"></td>
283 <td></td>
284 <td class="paramtype">const int&#160;</td>
285 <td class="paramname"><em>stride_x</em>, </td>
286 </tr>
287 <tr>
288 <td class="paramkey"></td>
289 <td></td>
290 <td class="paramtype">const int&#160;</td>
291 <td class="paramname"><em>stride_y</em>&#160;</td>
292 </tr>
293 <tr>
294 <td></td>
295 <td>)</td>
296 <td></td><td></td>
297 </tr>
298 </table>
299</div><div class="memdoc">
300
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]301<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00040">40</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]302
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]303<p>Referenced by <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00080">pooling_layer_MxN_quantized_nchw()</a>.</p>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]304<div class="fragment"><div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160;{</div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; <span class="keywordtype">int</span> start_x = get_global_id(0) * stride_x - pad_x;</div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; <span class="keywordtype">int</span> start_y = get_global_id(1) * stride_y - pad_y;</div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> end_x = min(start_x + pool_size_x, upper_bound_w);</div><div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> end_y = min(start_y + pool_size_y, upper_bound_h);</div><div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160;<span class="preprocessor">#if defined(EXCLUDE_PADDING)</span></div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; start_x = max(0, start_x);</div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; start_y = max(0, start_y);</div><div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160;<span class="preprocessor">#endif </span><span class="comment">/* defined(EXCLUDE_PADDING) */</span><span class="preprocessor"></span></div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160; <span class="keywordflow">return</span> ((end_y - start_y) * (end_x - start_x));</div><div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160;}</div></div><!-- fragment -->
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]305</div>
306</div>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]307<a class="anchor" id="a235920cb0f8577da6e104ccb03fbda25"></a>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]308<div class="memitem">
309<div class="memproto">
310 <table class="memname">
311 <tr>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]312 <td class="memname">int calculate_avg_scale_nhwc </td>
313 <td>(</td>
314 <td class="paramtype">const int&#160;</td>
315 <td class="paramname"><em>pool_size_x</em>, </td>
316 </tr>
317 <tr>
318 <td class="paramkey"></td>
319 <td></td>
320 <td class="paramtype">const int&#160;</td>
321 <td class="paramname"><em>pool_size_y</em>, </td>
322 </tr>
323 <tr>
324 <td class="paramkey"></td>
325 <td></td>
326 <td class="paramtype">int&#160;</td>
327 <td class="paramname"><em>upper_bound_w</em>, </td>
328 </tr>
329 <tr>
330 <td class="paramkey"></td>
331 <td></td>
332 <td class="paramtype">int&#160;</td>
333 <td class="paramname"><em>upper_bound_h</em>, </td>
334 </tr>
335 <tr>
336 <td class="paramkey"></td>
337 <td></td>
338 <td class="paramtype">const int&#160;</td>
339 <td class="paramname"><em>pad_x</em>, </td>
340 </tr>
341 <tr>
342 <td class="paramkey"></td>
343 <td></td>
344 <td class="paramtype">const int&#160;</td>
345 <td class="paramname"><em>pad_y</em>, </td>
346 </tr>
347 <tr>
348 <td class="paramkey"></td>
349 <td></td>
350 <td class="paramtype">const int&#160;</td>
351 <td class="paramname"><em>stride_x</em>, </td>
352 </tr>
353 <tr>
354 <td class="paramkey"></td>
355 <td></td>
356 <td class="paramtype">const int&#160;</td>
357 <td class="paramname"><em>stride_y</em>&#160;</td>
358 </tr>
359 <tr>
360 <td></td>
361 <td>)</td>
362 <td></td><td></td>
363 </tr>
364 </table>
365</div><div class="memdoc">
366
367<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00125">125</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
368
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]369<p>Referenced by <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00166">pooling_layer_MxN_quantized_nhwc()</a>.</p>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]370<div class="fragment"><div class="line"><a name="l00127"></a><span class="lineno"> 127</span>&#160;{</div><div class="line"><a name="l00128"></a><span class="lineno"> 128</span>&#160; <span class="keywordtype">int</span> start_x = get_global_id(1) * stride_x - pad_x;</div><div class="line"><a name="l00129"></a><span class="lineno"> 129</span>&#160; <span class="keywordtype">int</span> start_y = get_global_id(2) * stride_y - pad_y;</div><div class="line"><a name="l00130"></a><span class="lineno"> 130</span>&#160;</div><div class="line"><a name="l00131"></a><span class="lineno"> 131</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> end_x = min(start_x + pool_size_x, upper_bound_w);</div><div class="line"><a name="l00132"></a><span class="lineno"> 132</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> end_y = min(start_y + pool_size_y, upper_bound_h);</div><div class="line"><a name="l00133"></a><span class="lineno"> 133</span>&#160;</div><div class="line"><a name="l00134"></a><span class="lineno"> 134</span>&#160; start_x = max(0, start_x);</div><div class="line"><a name="l00135"></a><span class="lineno"> 135</span>&#160; start_y = max(0, start_y);</div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span>&#160;</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160; <span class="keywordflow">return</span> ((end_y - start_y) * (end_x - start_x));</div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>&#160;}</div></div><!-- fragment -->
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]371</div>
372</div>
373<a class="anchor" id="a5c74cefcb6318451b5785dd00bd9b05f"></a>
374<div class="memitem">
375<div class="memproto">
376 <table class="memname">
377 <tr>
378 <td class="memname">__kernel void pooling_layer_MxN_quantized_nchw </td>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]379 <td>(</td>
380 <td class="paramtype">__global uchar *&#160;</td>
381 <td class="paramname"><em>input_ptr</em>, </td>
382 </tr>
383 <tr>
384 <td class="paramkey"></td>
385 <td></td>
386 <td class="paramtype">uint&#160;</td>
387 <td class="paramname"><em>input_stride_x</em>, </td>
388 </tr>
389 <tr>
390 <td class="paramkey"></td>
391 <td></td>
392 <td class="paramtype">uint&#160;</td>
393 <td class="paramname"><em>input_step_x</em>, </td>
394 </tr>
395 <tr>
396 <td class="paramkey"></td>
397 <td></td>
398 <td class="paramtype">uint&#160;</td>
399 <td class="paramname"><em>input_stride_y</em>, </td>
400 </tr>
401 <tr>
402 <td class="paramkey"></td>
403 <td></td>
404 <td class="paramtype">uint&#160;</td>
405 <td class="paramname"><em>input_step_y</em>, </td>
406 </tr>
407 <tr>
408 <td class="paramkey"></td>
409 <td></td>
410 <td class="paramtype">uint&#160;</td>
411 <td class="paramname"><em>input_stride_z</em>, </td>
412 </tr>
413 <tr>
414 <td class="paramkey"></td>
415 <td></td>
416 <td class="paramtype">uint&#160;</td>
417 <td class="paramname"><em>input_step_z</em>, </td>
418 </tr>
419 <tr>
420 <td class="paramkey"></td>
421 <td></td>
422 <td class="paramtype">uint&#160;</td>
423 <td class="paramname"><em>input_offset_first_element_in_bytes</em>, </td>
424 </tr>
425 <tr>
426 <td class="paramkey"></td>
427 <td></td>
428 <td class="paramtype">__global uchar *&#160;</td>
429 <td class="paramname"><em>output_ptr</em>, </td>
430 </tr>
431 <tr>
432 <td class="paramkey"></td>
433 <td></td>
434 <td class="paramtype">uint&#160;</td>
435 <td class="paramname"><em>output_stride_x</em>, </td>
436 </tr>
437 <tr>
438 <td class="paramkey"></td>
439 <td></td>
440 <td class="paramtype">uint&#160;</td>
441 <td class="paramname"><em>output_step_x</em>, </td>
442 </tr>
443 <tr>
444 <td class="paramkey"></td>
445 <td></td>
446 <td class="paramtype">uint&#160;</td>
447 <td class="paramname"><em>output_stride_y</em>, </td>
448 </tr>
449 <tr>
450 <td class="paramkey"></td>
451 <td></td>
452 <td class="paramtype">uint&#160;</td>
453 <td class="paramname"><em>output_step_y</em>, </td>
454 </tr>
455 <tr>
456 <td class="paramkey"></td>
457 <td></td>
458 <td class="paramtype">uint&#160;</td>
459 <td class="paramname"><em>output_stride_z</em>, </td>
460 </tr>
461 <tr>
462 <td class="paramkey"></td>
463 <td></td>
464 <td class="paramtype">uint&#160;</td>
465 <td class="paramname"><em>output_step_z</em>, </td>
466 </tr>
467 <tr>
468 <td class="paramkey"></td>
469 <td></td>
470 <td class="paramtype">uint&#160;</td>
471 <td class="paramname"><em>output_offset_first_element_in_bytes</em>&#160;</td>
472 </tr>
473 <tr>
474 <td></td>
475 <td>)</td>
476 <td></td><td></td>
477 </tr>
478 </table>
479</div><div class="memdoc">
480
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]481<p>Performs a pooling function of pool size equal to N (NCHW) </p>
Anthony Barbier06ea0482018-02-22 15:45:35 +0000[diff] [blame]482<dl class="section note"><dt>Note</dt><dd>Pool sizes must be passed using -DPOOL_SIZE_X and -DPOOL_SIZE_Y e.g. -DPOOL_SIZE_X=13; </dd>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]483<dd>
484In case of average pooling the following information must be passed at compile time: -DPOOL_AVG must be provided otherwise max pooling will be performed. -DMAX_WIDTH and -DMAX_HEIGHT which are the maximum accessible indeces in x and y dimensions (width + pad) -DSTRIDE_X and -DSTRIDE_Y which are the steps of the window along the x and y directions -DPAD_X and -DPAD_Y which are the pooling paddings in x and y dimension</dd></dl>
485<dl class="params"><dt>Parameters</dt><dd>
486 <table class="params">
487 <tr><td class="paramdir">[in]</td><td class="paramname">input_ptr</td><td>Pointer to the source image. Supported data types: QASYMM8 </td></tr>
488 <tr><td class="paramdir">[in]</td><td class="paramname">input_stride_x</td><td>Stride of the source image in X dimension (in bytes) </td></tr>
489 <tr><td class="paramdir">[in]</td><td class="paramname">input_step_x</td><td>input_stride_x * number of elements along X processed per workitem(in bytes) </td></tr>
490 <tr><td class="paramdir">[in]</td><td class="paramname">input_stride_y</td><td>Stride of the source image in Y dimension (in bytes) </td></tr>
491 <tr><td class="paramdir">[in]</td><td class="paramname">input_step_y</td><td>input_stride_y * number of elements along Y processed per workitem(in bytes) </td></tr>
492 <tr><td class="paramdir">[in]</td><td class="paramname">input_stride_z</td><td>Stride of the source tensor in Z dimension (in bytes) </td></tr>
493 <tr><td class="paramdir">[in]</td><td class="paramname">input_step_z</td><td>input_stride_z * number of elements along Z processed per workitem(in bytes) </td></tr>
494 <tr><td class="paramdir">[in]</td><td class="paramname">input_offset_first_element_in_bytes</td><td>The offset of the first element in the source image </td></tr>
495 <tr><td class="paramdir">[out]</td><td class="paramname">output_ptr</td><td>Pointer to the destination image. Supported data types: same as <code>input_ptr</code> </td></tr>
496 <tr><td class="paramdir">[in]</td><td class="paramname">output_stride_x</td><td>Stride of the destination image in X dimension (in bytes) </td></tr>
497 <tr><td class="paramdir">[in]</td><td class="paramname">output_step_x</td><td>output_stride_x * number of elements along X processed per workitem(in bytes) </td></tr>
498 <tr><td class="paramdir">[in]</td><td class="paramname">output_stride_y</td><td>Stride of the destination image in Y dimension (in bytes) </td></tr>
499 <tr><td class="paramdir">[in]</td><td class="paramname">output_step_y</td><td>output_stride_y * number of elements along Y processed per workitem(in bytes) </td></tr>
500 <tr><td class="paramdir">[in]</td><td class="paramname">output_stride_z</td><td>Stride of the source tensor in Z dimension (in bytes) </td></tr>
501 <tr><td class="paramdir">[in]</td><td class="paramname">output_step_z</td><td>output_stride_z * number of elements along Z processed per workitem(in bytes) </td></tr>
502 <tr><td class="paramdir">[in]</td><td class="paramname">output_offset_first_element_in_bytes</td><td>The offset of the first element in the destination image </td></tr>
503 </table>
504 </dd>
505</dl>
506
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]507<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00080">80</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]508
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]509<p>References <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00040">calculate_avg_scale()</a>, <a class="el" href="helpers_8h_source.xhtml#l00127">CONVERT_TO_TENSOR3D_STRUCT</a>, <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00032">DIV_OP</a>, <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00027">POOL_OP</a>, <a class="el" href="helpers_8h_source.xhtml#l00161">Tensor3D::ptr</a>, <a class="el" href="namespacearm__compute.xhtml#a6452ac376d4adb94d93a93d678bb6757">arm_compute::round()</a>, and <a class="el" href="helpers_8h_source.xhtml#l00323">tensor3D_offset()</a>.</p>
510<div class="fragment"><div class="line"><a name="l00083"></a><span class="lineno"> 83</span>&#160;{</div><div class="line"><a name="l00084"></a><span class="lineno"> 84</span>&#160; <span class="comment">// Get pixels pointer</span></div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> input = <a class="code" href="helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(input);</div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> output = <a class="code" href="helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(output);</div><div class="line"><a name="l00087"></a><span class="lineno"> 87</span>&#160;</div><div class="line"><a name="l00088"></a><span class="lineno"> 88</span>&#160; int8 vdata = 0;</div><div class="line"><a name="l00089"></a><span class="lineno"> 89</span>&#160; <span class="keywordtype">int</span> sdata = 0;</div><div class="line"><a name="l00090"></a><span class="lineno"> 90</span>&#160;</div><div class="line"><a name="l00091"></a><span class="lineno"> 91</span>&#160; <span class="comment">// Load data</span></div><div class="line"><a name="l00092"></a><span class="lineno"> 92</span>&#160; <span class="keywordflow">for</span>(<span class="keywordtype">int</span> y = 0; y &lt; POOL_SIZE_Y; y++)</div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span>&#160; {</div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span>&#160; <span class="keywordtype">int</span> x = 0;</div><div class="line"><a name="l00095"></a><span class="lineno"> 95</span>&#160; <span class="keywordflow">for</span>(; x &lt;= ((int)POOL_SIZE_X - 8); x += 8)</div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span>&#160; {</div><div class="line"><a name="l00097"></a><span class="lineno"> 97</span>&#160; uchar8 data = vload8(0, (__global uchar *)<a class="code" href="helpers_8h.xhtml#a2101b2fe0193ce227ae4e0945e321d85">tensor3D_offset</a>(&amp;input, x, y, 0));</div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span>&#160; int8 data0 = convert_int8(data);</div><div class="line"><a name="l00099"></a><span class="lineno"> 99</span>&#160; vdata = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(vdata, data0);</div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span>&#160; }</div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span>&#160;</div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span>&#160; <span class="comment">// Leftover</span></div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160; <span class="keywordflow">for</span>(; x &lt; (int)POOL_SIZE_X; ++x)</div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>&#160; {</div><div class="line"><a name="l00105"></a><span class="lineno"> 105</span>&#160; uchar data = *((__global uchar *)<a class="code" href="helpers_8h.xhtml#a2101b2fe0193ce227ae4e0945e321d85">tensor3D_offset</a>(&amp;input, x, y, 0));</div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160; <span class="keywordtype">int</span> data0 = convert_int(data);</div><div class="line"><a name="l00107"></a><span class="lineno"> 107</span>&#160; sdata = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(sdata, data0);</div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span>&#160; }</div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span>&#160; }</div><div class="line"><a name="l00110"></a><span class="lineno"> 110</span>&#160;</div><div class="line"><a name="l00111"></a><span class="lineno"> 111</span>&#160; <span class="comment">// Reduce result</span></div><div class="line"><a name="l00112"></a><span class="lineno"> 112</span>&#160; int4 reduce4 = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(vdata.s0123, vdata.s4567);</div><div class="line"><a name="l00113"></a><span class="lineno"> 113</span>&#160; int2 reduce2 = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(reduce4.s01, reduce4.s23);</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>&#160; <span class="keywordtype">int</span> res = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(reduce2.s0, reduce2.s1);</div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span>&#160; res = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(res, sdata);</div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span>&#160;</div><div class="line"><a name="l00117"></a><span class="lineno"> 117</span>&#160;<span class="preprocessor">#if defined(POOL_AVG)</span></div><div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160; res = <a class="code" href="namespacearm__compute.xhtml#a6452ac376d4adb94d93a93d678bb6757">round</a>(<a class="code" href="pooling__layer__quantized_8cl.xhtml#a5db17889d824975fefb2ce2f4690637f">DIV_OP</a>(res, <a class="code" href="pooling__layer__quantized_8cl.xhtml#a4057c01dd49145c5e463f1cea52542df">calculate_avg_scale</a>(POOL_SIZE_X, POOL_SIZE_Y, MAX_WIDTH, MAX_HEIGHT, PAD_X, PAD_Y, STRIDE_X, STRIDE_Y)));</div><div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160;<span class="preprocessor">#endif </span><span class="comment">/* defined(POOL_AVG) */</span><span class="preprocessor"></span></div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span>&#160;</div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span>&#160; <span class="comment">// Store result</span></div><div class="line"><a name="l00122"></a><span class="lineno"> 122</span>&#160; *(__global uchar *)output.<a class="code" href="struct_tensor3_d.xhtml#acf52c23cbd7424606c10a606524e3e32">ptr</a> = convert_uchar(res);</div><div class="line"><a name="l00123"></a><span class="lineno"> 123</span>&#160;}</div><div class="ttc" id="helpers_8h_xhtml_a31c8c760f08fb1a331b16b7c204321dc"><div class="ttname"><a href="helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a></div><div class="ttdeci">#define CONVERT_TO_TENSOR3D_STRUCT(name)</div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00127">helpers.h:127</a></div></div>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]511<div class="ttc" id="pooling__layer__quantized_8cl_xhtml_a482ef7d59a5f474ca126e737c7f0978a"><div class="ttname"><a href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a></div><div class="ttdeci">#define POOL_OP(x, y)</div><div class="ttdef"><b>Definition:</b> <a href="pooling__layer__quantized_8cl_source.xhtml#l00027">pooling_layer_quantized.cl:27</a></div></div>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]512<div class="ttc" id="struct_tensor3_d_xhtml"><div class="ttname"><a href="struct_tensor3_d.xhtml">Tensor3D</a></div><div class="ttdoc">Structure to hold 3D tensor information. </div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00159">helpers.h:159</a></div></div>
513<div class="ttc" id="helpers_8h_xhtml_a2101b2fe0193ce227ae4e0945e321d85"><div class="ttname"><a href="helpers_8h.xhtml#a2101b2fe0193ce227ae4e0945e321d85">tensor3D_offset</a></div><div class="ttdeci">__global const uchar * tensor3D_offset(const Tensor3D *tensor, int x, int y, int z)</div><div class="ttdoc">Get the pointer position of a Tensor3D. </div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00323">helpers.h:323</a></div></div>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]514<div class="ttc" id="pooling__layer__quantized_8cl_xhtml_a5db17889d824975fefb2ce2f4690637f"><div class="ttname"><a href="pooling__layer__quantized_8cl.xhtml#a5db17889d824975fefb2ce2f4690637f">DIV_OP</a></div><div class="ttdeci">#define DIV_OP(x, y)</div><div class="ttdef"><b>Definition:</b> <a href="pooling__layer__quantized_8cl_source.xhtml#l00032">pooling_layer_quantized.cl:32</a></div></div>
515<div class="ttc" id="namespacearm__compute_xhtml_a6452ac376d4adb94d93a93d678bb6757"><div class="ttname"><a href="namespacearm__compute.xhtml#a6452ac376d4adb94d93a93d678bb6757">arm_compute::round</a></div><div class="ttdeci">int round(float x, RoundingPolicy rounding_policy)</div><div class="ttdoc">Return a rounded value of x. </div></div>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]516<div class="ttc" id="struct_tensor3_d_xhtml_acf52c23cbd7424606c10a606524e3e32"><div class="ttname"><a href="struct_tensor3_d.xhtml#acf52c23cbd7424606c10a606524e3e32">Tensor3D::ptr</a></div><div class="ttdeci">__global uchar * ptr</div><div class="ttdoc">Pointer to the starting postion of the buffer. </div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00161">helpers.h:161</a></div></div>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]517<div class="ttc" id="pooling__layer__quantized_8cl_xhtml_a4057c01dd49145c5e463f1cea52542df"><div class="ttname"><a href="pooling__layer__quantized_8cl.xhtml#a4057c01dd49145c5e463f1cea52542df">calculate_avg_scale</a></div><div class="ttdeci">int calculate_avg_scale(const int pool_size_x, const int pool_size_y, const int upper_bound_w, const int upper_bound_h, const int pad_x, const int pad_y, const int stride_x, const int stride_y)</div><div class="ttdef"><b>Definition:</b> <a href="pooling__layer__quantized_8cl_source.xhtml#l00040">pooling_layer_quantized.cl:40</a></div></div>
518</div><!-- fragment -->
519</div>
520</div>
521<a class="anchor" id="a747409ec1866adc96ee62d1e08ec502f"></a>
522<div class="memitem">
523<div class="memproto">
524 <table class="memname">
525 <tr>
526 <td class="memname">__kernel void pooling_layer_MxN_quantized_nhwc </td>
527 <td>(</td>
528 <td class="paramtype">__global uchar *&#160;</td>
529 <td class="paramname"><em>input_ptr</em>, </td>
530 </tr>
531 <tr>
532 <td class="paramkey"></td>
533 <td></td>
534 <td class="paramtype">uint&#160;</td>
535 <td class="paramname"><em>input_stride_x</em>, </td>
536 </tr>
537 <tr>
538 <td class="paramkey"></td>
539 <td></td>
540 <td class="paramtype">uint&#160;</td>
541 <td class="paramname"><em>input_step_x</em>, </td>
542 </tr>
543 <tr>
544 <td class="paramkey"></td>
545 <td></td>
546 <td class="paramtype">uint&#160;</td>
547 <td class="paramname"><em>input_stride_y</em>, </td>
548 </tr>
549 <tr>
550 <td class="paramkey"></td>
551 <td></td>
552 <td class="paramtype">uint&#160;</td>
553 <td class="paramname"><em>input_step_y</em>, </td>
554 </tr>
555 <tr>
556 <td class="paramkey"></td>
557 <td></td>
558 <td class="paramtype">uint&#160;</td>
559 <td class="paramname"><em>input_stride_z</em>, </td>
560 </tr>
561 <tr>
562 <td class="paramkey"></td>
563 <td></td>
564 <td class="paramtype">uint&#160;</td>
565 <td class="paramname"><em>input_step_z</em>, </td>
566 </tr>
567 <tr>
568 <td class="paramkey"></td>
569 <td></td>
570 <td class="paramtype">uint&#160;</td>
571 <td class="paramname"><em>input_offset_first_element_in_bytes</em>, </td>
572 </tr>
573 <tr>
574 <td class="paramkey"></td>
575 <td></td>
576 <td class="paramtype">__global uchar *&#160;</td>
577 <td class="paramname"><em>output_ptr</em>, </td>
578 </tr>
579 <tr>
580 <td class="paramkey"></td>
581 <td></td>
582 <td class="paramtype">uint&#160;</td>
583 <td class="paramname"><em>output_stride_x</em>, </td>
584 </tr>
585 <tr>
586 <td class="paramkey"></td>
587 <td></td>
588 <td class="paramtype">uint&#160;</td>
589 <td class="paramname"><em>output_step_x</em>, </td>
590 </tr>
591 <tr>
592 <td class="paramkey"></td>
593 <td></td>
594 <td class="paramtype">uint&#160;</td>
595 <td class="paramname"><em>output_stride_y</em>, </td>
596 </tr>
597 <tr>
598 <td class="paramkey"></td>
599 <td></td>
600 <td class="paramtype">uint&#160;</td>
601 <td class="paramname"><em>output_step_y</em>, </td>
602 </tr>
603 <tr>
604 <td class="paramkey"></td>
605 <td></td>
606 <td class="paramtype">uint&#160;</td>
607 <td class="paramname"><em>output_stride_z</em>, </td>
608 </tr>
609 <tr>
610 <td class="paramkey"></td>
611 <td></td>
612 <td class="paramtype">uint&#160;</td>
613 <td class="paramname"><em>output_step_z</em>, </td>
614 </tr>
615 <tr>
616 <td class="paramkey"></td>
617 <td></td>
618 <td class="paramtype">uint&#160;</td>
619 <td class="paramname"><em>output_offset_first_element_in_bytes</em>&#160;</td>
620 </tr>
621 <tr>
622 <td></td>
623 <td>)</td>
624 <td></td><td></td>
625 </tr>
626 </table>
627</div><div class="memdoc">
628
629<p>Performs a pooling function of pool size equal to N (NHWC) </p>
630<dl class="section note"><dt>Note</dt><dd>Pool sizes must be passed using -DPOOL_SIZE_X and -DPOOL_SIZE_Y e.g. -DPOOL_SIZE_X=13; </dd>
631<dd>
632Tensors width and height must be passed at compile time using -DMAX_WIDTH and -DMAX_HEIGHT </dd>
633<dd>
634Strides must be passed at compile time using -DSTRIDE_X and -DSTRIDE_Y which are the steps of the window along the x and y directions </dd>
635<dd>
636Pad values must be passed at compile time using -DPAD_X and -DPAD_Y which are the pooling paddings in x and y dimension </dd>
637<dd>
638In case of average pooling the following information must be passed at compile time: -DPOOL_AVG must be provided otherwise max pooling will be performed.</dd></dl>
639<dl class="params"><dt>Parameters</dt><dd>
640 <table class="params">
641 <tr><td class="paramdir">[in]</td><td class="paramname">input_ptr</td><td>Pointer to the source image. Supported data types: QASYMM8 </td></tr>
642 <tr><td class="paramdir">[in]</td><td class="paramname">input_stride_x</td><td>Stride of the source image in X dimension (in bytes) </td></tr>
643 <tr><td class="paramdir">[in]</td><td class="paramname">input_step_x</td><td>input_stride_x * number of elements along X processed per workitem(in bytes) </td></tr>
644 <tr><td class="paramdir">[in]</td><td class="paramname">input_stride_y</td><td>Stride of the source image in Y dimension (in bytes) </td></tr>
645 <tr><td class="paramdir">[in]</td><td class="paramname">input_step_y</td><td>input_stride_y * number of elements along Y processed per workitem(in bytes) </td></tr>
646 <tr><td class="paramdir">[in]</td><td class="paramname">input_stride_z</td><td>Stride of the source tensor in Z dimension (in bytes) </td></tr>
647 <tr><td class="paramdir">[in]</td><td class="paramname">input_step_z</td><td>input_stride_z * number of elements along Z processed per workitem(in bytes) </td></tr>
648 <tr><td class="paramdir">[in]</td><td class="paramname">input_offset_first_element_in_bytes</td><td>The offset of the first element in the source image </td></tr>
649 <tr><td class="paramdir">[out]</td><td class="paramname">output_ptr</td><td>Pointer to the destination image. Supported data types: same as <code>input_ptr</code> </td></tr>
650 <tr><td class="paramdir">[in]</td><td class="paramname">output_stride_x</td><td>Stride of the destination image in X dimension (in bytes) </td></tr>
651 <tr><td class="paramdir">[in]</td><td class="paramname">output_step_x</td><td>output_stride_x * number of elements along X processed per workitem(in bytes) </td></tr>
652 <tr><td class="paramdir">[in]</td><td class="paramname">output_stride_y</td><td>Stride of the destination image in Y dimension (in bytes) </td></tr>
653 <tr><td class="paramdir">[in]</td><td class="paramname">output_step_y</td><td>output_stride_y * number of elements along Y processed per workitem(in bytes) </td></tr>
654 <tr><td class="paramdir">[in]</td><td class="paramname">output_stride_z</td><td>Stride of the source tensor in Z dimension (in bytes) </td></tr>
655 <tr><td class="paramdir">[in]</td><td class="paramname">output_step_z</td><td>output_stride_z * number of elements along Z processed per workitem(in bytes) </td></tr>
656 <tr><td class="paramdir">[in]</td><td class="paramname">output_offset_first_element_in_bytes</td><td>The offset of the first element in the destination image </td></tr>
657 </table>
658 </dd>
659</dl>
660
661<p>Definition at line <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00166">166</a> of file <a class="el" href="pooling__layer__quantized_8cl_source.xhtml">pooling_layer_quantized.cl</a>.</p>
662
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]663<p>References <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00125">calculate_avg_scale_nhwc()</a>, <a class="el" href="helpers_8h_source.xhtml#l00127">CONVERT_TO_TENSOR3D_STRUCT</a>, <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00034">DIV_OP_NHWC</a>, <a class="el" href="pooling__layer__quantized_8cl_source.xhtml#l00027">POOL_OP</a>, <a class="el" href="helpers_8h_source.xhtml#l00161">Tensor3D::ptr</a>, <a class="el" href="namespacearm__compute.xhtml#a6452ac376d4adb94d93a93d678bb6757">arm_compute::round()</a>, and <a class="el" href="helpers_8h_source.xhtml#l00323">tensor3D_offset()</a>.</p>
664<div class="fragment"><div class="line"><a name="l00169"></a><span class="lineno"> 169</span>&#160;{</div><div class="line"><a name="l00170"></a><span class="lineno"> 170</span>&#160; <span class="comment">// Get pixels pointer</span></div><div class="line"><a name="l00171"></a><span class="lineno"> 171</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> input = <a class="code" href="helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(input);</div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> output = <a class="code" href="helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(output);</div><div class="line"><a name="l00173"></a><span class="lineno"> 173</span>&#160;</div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span>&#160; int8 vdata = 0;</div><div class="line"><a name="l00175"></a><span class="lineno"> 175</span>&#160;</div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> idx_width = get_global_id(1) * STRIDE_X;</div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> idx_height = get_global_id(2) * STRIDE_Y;</div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span>&#160;</div><div class="line"><a name="l00179"></a><span class="lineno"> 179</span>&#160; <span class="keywordflow">for</span>(<span class="keywordtype">int</span> y = 0; y &lt; POOL_SIZE_Y; ++y)</div><div class="line"><a name="l00180"></a><span class="lineno"> 180</span>&#160; {</div><div class="line"><a name="l00181"></a><span class="lineno"> 181</span>&#160; <span class="keywordtype">int</span> y1 = select(y, PAD_Y - idx_height, y + idx_height &lt; PAD_Y || y + idx_height &gt; MAX_HEIGHT);</div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160; <span class="keywordflow">for</span>(<span class="keywordtype">int</span> x = 0; x &lt; POOL_SIZE_X; ++x)</div><div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160; {</div><div class="line"><a name="l00184"></a><span class="lineno"> 184</span>&#160; <span class="keywordtype">int</span> x1 = select(x, PAD_X - idx_width - 1, x + idx_width &lt; PAD_X || x + idx_width &gt; MAX_WIDTH);</div><div class="line"><a name="l00185"></a><span class="lineno"> 185</span>&#160; x1 = select(x1, PAD_X - idx_width - 1, y != y1);</div><div class="line"><a name="l00186"></a><span class="lineno"> 186</span>&#160; uchar8 data = vload8(0, (__global uchar *)<a class="code" href="helpers_8h.xhtml#a2101b2fe0193ce227ae4e0945e321d85">tensor3D_offset</a>(&amp;input, 0, x1 - PAD_X, y1 - PAD_Y));</div><div class="line"><a name="l00187"></a><span class="lineno"> 187</span>&#160; int8 data0 = convert_int8(data);</div><div class="line"><a name="l00188"></a><span class="lineno"> 188</span>&#160; vdata = <a class="code" href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a>(vdata, data0);</div><div class="line"><a name="l00189"></a><span class="lineno"> 189</span>&#160; }</div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>&#160; }</div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span>&#160;</div><div class="line"><a name="l00192"></a><span class="lineno"> 192</span>&#160;<span class="preprocessor">#if defined(POOL_AVG)</span></div><div class="line"><a name="l00193"></a><span class="lineno"> 193</span>&#160; <span class="comment">// Divide by pool region in case of average pooling</span></div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span>&#160; vdata = convert_int8(<a class="code" href="namespacearm__compute.xhtml#a6452ac376d4adb94d93a93d678bb6757">round</a>(<a class="code" href="pooling__layer__quantized_8cl.xhtml#a6c01fa98d360a9d52926dc6a5a599711">DIV_OP_NHWC</a>(vdata, <a class="code" href="pooling__layer__quantized_8cl.xhtml#a235920cb0f8577da6e104ccb03fbda25">calculate_avg_scale_nhwc</a>(POOL_SIZE_X, POOL_SIZE_Y, MAX_WIDTH, MAX_HEIGHT, PAD_X, PAD_Y, STRIDE_X, STRIDE_Y))));</div><div class="line"><a name="l00195"></a><span class="lineno"> 195</span>&#160;<span class="preprocessor">#endif </span><span class="comment">/* defined(POOL_AVG) */</span><span class="preprocessor"></span></div><div class="line"><a name="l00196"></a><span class="lineno"> 196</span>&#160;</div><div class="line"><a name="l00197"></a><span class="lineno"> 197</span>&#160; <span class="comment">// Store result</span></div><div class="line"><a name="l00198"></a><span class="lineno"> 198</span>&#160; vstore8(convert_uchar8(vdata), 0, (__global uchar *)output.<a class="code" href="struct_tensor3_d.xhtml#acf52c23cbd7424606c10a606524e3e32">ptr</a>);</div><div class="line"><a name="l00199"></a><span class="lineno"> 199</span>&#160;}<div class="ttc" id="helpers_8h_xhtml_a31c8c760f08fb1a331b16b7c204321dc"><div class="ttname"><a href="helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a></div><div class="ttdeci">#define CONVERT_TO_TENSOR3D_STRUCT(name)</div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00127">helpers.h:127</a></div></div>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]665<div class="ttc" id="pooling__layer__quantized_8cl_xhtml_a482ef7d59a5f474ca126e737c7f0978a"><div class="ttname"><a href="pooling__layer__quantized_8cl.xhtml#a482ef7d59a5f474ca126e737c7f0978a">POOL_OP</a></div><div class="ttdeci">#define POOL_OP(x, y)</div><div class="ttdef"><b>Definition:</b> <a href="pooling__layer__quantized_8cl_source.xhtml#l00027">pooling_layer_quantized.cl:27</a></div></div>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]666<div class="ttc" id="struct_tensor3_d_xhtml"><div class="ttname"><a href="struct_tensor3_d.xhtml">Tensor3D</a></div><div class="ttdoc">Structure to hold 3D tensor information. </div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00159">helpers.h:159</a></div></div>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]667<div class="ttc" id="pooling__layer__quantized_8cl_xhtml_a235920cb0f8577da6e104ccb03fbda25"><div class="ttname"><a href="pooling__layer__quantized_8cl.xhtml#a235920cb0f8577da6e104ccb03fbda25">calculate_avg_scale_nhwc</a></div><div class="ttdeci">int calculate_avg_scale_nhwc(const int pool_size_x, const int pool_size_y, int upper_bound_w, int upper_bound_h, const int pad_x, const int pad_y, const int stride_x, const int stride_y)</div><div class="ttdef"><b>Definition:</b> <a href="pooling__layer__quantized_8cl_source.xhtml#l00125">pooling_layer_quantized.cl:125</a></div></div>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]668<div class="ttc" id="helpers_8h_xhtml_a2101b2fe0193ce227ae4e0945e321d85"><div class="ttname"><a href="helpers_8h.xhtml#a2101b2fe0193ce227ae4e0945e321d85">tensor3D_offset</a></div><div class="ttdeci">__global const uchar * tensor3D_offset(const Tensor3D *tensor, int x, int y, int z)</div><div class="ttdoc">Get the pointer position of a Tensor3D. </div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00323">helpers.h:323</a></div></div>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]669<div class="ttc" id="namespacearm__compute_xhtml_a6452ac376d4adb94d93a93d678bb6757"><div class="ttname"><a href="namespacearm__compute.xhtml#a6452ac376d4adb94d93a93d678bb6757">arm_compute::round</a></div><div class="ttdeci">int round(float x, RoundingPolicy rounding_policy)</div><div class="ttdoc">Return a rounded value of x. </div></div>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]670<div class="ttc" id="struct_tensor3_d_xhtml_acf52c23cbd7424606c10a606524e3e32"><div class="ttname"><a href="struct_tensor3_d.xhtml#acf52c23cbd7424606c10a606524e3e32">Tensor3D::ptr</a></div><div class="ttdeci">__global uchar * ptr</div><div class="ttdoc">Pointer to the starting postion of the buffer. </div><div class="ttdef"><b>Definition:</b> <a href="helpers_8h_source.xhtml#l00161">helpers.h:161</a></div></div>
Jenkinsb3a371b2018-05-23 11:36:53 +0100[diff] [blame]671<div class="ttc" id="pooling__layer__quantized_8cl_xhtml_a6c01fa98d360a9d52926dc6a5a599711"><div class="ttname"><a href="pooling__layer__quantized_8cl.xhtml#a6c01fa98d360a9d52926dc6a5a599711">DIV_OP_NHWC</a></div><div class="ttdeci">#define DIV_OP_NHWC(x, y)</div><div class="ttdef"><b>Definition:</b> <a href="pooling__layer__quantized_8cl_source.xhtml#l00034">pooling_layer_quantized.cl:34</a></div></div>
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]672</div><!-- fragment -->
673</div>
674</div>
675</div><!-- contents -->
676</div><!-- doc-content -->
677<!-- start footer part -->
678<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
679 <ul>
680 <li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.xhtml">src</a></li><li class="navelem"><a class="el" href="dir_aebb8dcc11953d78e620bbef0b9e2183.xhtml">core</a></li><li class="navelem"><a class="el" href="dir_8c278f79c760e5c5fbd911f9870614c1.xhtml">CL</a></li><li class="navelem"><a class="el" href="dir_25885286e9dad4fa105b7b25a8031bbf.xhtml">cl_kernels</a></li><li class="navelem"><a class="el" href="pooling__layer__quantized_8cl.xhtml">pooling_layer_quantized.cl</a></li>
Jenkins52ba29e2018-08-29 15:32:11 +0000[diff] [blame^]681 <li class="footer">Generated on Wed Aug 29 2018 15:31:38 for Compute Library by
Anthony Barbier8140e1e2017-12-14 23:48:46 +0000[diff] [blame]682 <a href="http://www.doxygen.org/index.html">
683 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.11 </li>
684 </ul>
685</div>
686</body>
687</html>