|
- <!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.16"/>
- <meta name="viewport" content="width=device-width, initial-scale=1"/>
- <title>aitisa_api: /home/amax107/hice/likesen/aitisa_api/src/nn/pooling.c File Reference</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="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>
- <link href="doxygen.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;">
- <td id="projectalign" style="padding-left: 0.5em;">
- <div id="projectname">aitisa_api
-  <span id="projectnumber">0.4.0</span>
- </div>
- <div id="projectbrief">Standard APIs for AI operations</div>
- </td>
- </tr>
- </tbody>
- </table>
- </div>
- <!-- end header part -->
- <!-- Generated by Doxygen 1.8.16 -->
- <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>
- <!-- 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 id="nav-path" class="navpath">
- <ul>
- <li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.html">src</a></li><li class="navelem"><a class="el" href="dir_8ddd85c06ebeec0a6efc6a7da38f01d9.html">nn</a></li> </ul>
- </div>
- </div><!-- top -->
- <div class="header">
- <div class="summary">
- <a href="#define-members">Macros</a> |
- <a href="#func-members">Functions</a> </div>
- <div class="headertitle">
- <div class="title">pooling.c File Reference</div> </div>
- </div><!--header-->
- <div class="contents">
- <div class="textblock"><code>#include "<a class="el" href="allocator_8h_source.html">src/core/allocator.h</a>"</code><br />
- <code>#include "<a class="el" href="dispatch_8h_source.html">src/core/dispatch.h</a>"</code><br />
- <code>#include "<a class="el" href="index__utils_8h_source.html">src/basic/index_utils.h</a>"</code><br />
- <code>#include "<a class="el" href="pooling_8h_source.html">src/nn/pooling.h</a>"</code><br />
- <code>#include <math.h></code><br />
- </div><table class="memberdecls">
- <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="define-members"></a>
- Macros</h2></td></tr>
- <tr class="memitem:a0bc47e22c55e189740a38029bf5e6a0f"><td class="memItemLeft" align="right" valign="top">#define </td><td class="memItemRight" valign="bottom"><a class="el" href="pooling_8c.html#a0bc47e22c55e189740a38029bf5e6a0f">pooling_kernel</a>(typename)</td></tr>
- <tr class="separator:a0bc47e22c55e189740a38029bf5e6a0f"><td class="memSeparator" colspan="2"> </td></tr>
- <tr class="memitem:a9c59fcd2e850ed15680c828d8daa0b5f"><td class="memItemLeft" align="right" valign="top">#define </td><td class="memItemRight" valign="bottom"><a class="el" href="pooling_8c.html#a9c59fcd2e850ed15680c828d8daa0b5f">pooling_output_element</a>(typename)</td></tr>
- <tr class="separator:a9c59fcd2e850ed15680c828d8daa0b5f"><td class="memSeparator" colspan="2"> </td></tr>
- </table><table class="memberdecls">
- <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
- Functions</h2></td></tr>
- <tr class="memitem:a30bf04d198c7a41f4a4f5839c89100d6"><td class="memItemLeft" align="right" valign="top">static <a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="pooling_8c.html#a30bf04d198c7a41f4a4f5839c89100d6">pooling_create_output</a> (const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> input, int64_t *out_dims, const int *ksize, const int *stride, const int *padding, const int *dilation, const int axis, int64_t c_start, <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> *output)</td></tr>
- <tr class="separator:a30bf04d198c7a41f4a4f5839c89100d6"><td class="memSeparator" colspan="2"> </td></tr>
- <tr class="memitem:af1d691ac69f6b1554bd8383efc496ee3"><td class="memItemLeft" align="right" valign="top">static <a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="pooling_8c.html#af1d691ac69f6b1554bd8383efc496ee3">pooling_double</a> (const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> input, const char *mode, const int *ksize, const int *stride, const int *padding, const int *dilation, const int axis, <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> *output)</td></tr>
- <tr class="separator:af1d691ac69f6b1554bd8383efc496ee3"><td class="memSeparator" colspan="2"> </td></tr>
- <tr class="memitem:ad061e28180ec47569970fbe30e7ddd97"><td class="memItemLeft" align="right" valign="top"><a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="pooling_8c.html#ad061e28180ec47569970fbe30e7ddd97">aitisa_pooling</a> (const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> input, const char *mode, const int *ksize, const int *stride, const int *padding, const int *dilation, <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> *output)</td></tr>
- <tr class="memdesc:ad061e28180ec47569970fbe30e7ddd97"><td class="mdescLeft"> </td><td class="mdescRight">Applies a pooling over an input signal composed of several feature maps. <a href="pooling_8c.html#ad061e28180ec47569970fbe30e7ddd97">More...</a><br /></td></tr>
- <tr class="separator:ad061e28180ec47569970fbe30e7ddd97"><td class="memSeparator" colspan="2"> </td></tr>
- </table>
- <h2 class="groupheader">Macro Definition Documentation</h2>
- <a id="a0bc47e22c55e189740a38029bf5e6a0f"></a>
- <h2 class="memtitle"><span class="permalink"><a href="#a0bc47e22c55e189740a38029bf5e6a0f">◆ </a></span>pooling_kernel</h2>
-
- <div class="memitem">
- <div class="memproto">
- <table class="memname">
- <tr>
- <td class="memname">#define pooling_kernel</td>
- <td>(</td>
- <td class="paramtype"> </td>
- <td class="paramname">typename</td><td>)</td>
- <td></td>
- </tr>
- </table>
- </div><div class="memdoc">
- <b>Value:</b><div class="fragment"><div class="line"><span class="keywordflow">if</span>(!strcmp(mode, <span class="stringliteral">"avg"</span>)){ \</div>
- <div class="line"> temp += ((<span class="keyword">typename</span>*)in_data)[linear_idx]; \</div>
- <div class="line"> } \</div>
- <div class="line"> else <span class="keywordflow">if</span> (!strcmp(mode, <span class="stringliteral">"max"</span>)) { \</div>
- <div class="line"> if (temp < ((<span class="keyword">typename</span>*)in_data)[linear_idx]) { \</div>
- <div class="line"> temp = ((<span class="keyword">typename</span>*)in_data)[linear_idx]; \</div>
- <div class="line"> } \</div>
- <div class="line"> }</div>
- </div><!-- fragment -->
- </div>
- </div>
- <a id="a9c59fcd2e850ed15680c828d8daa0b5f"></a>
- <h2 class="memtitle"><span class="permalink"><a href="#a9c59fcd2e850ed15680c828d8daa0b5f">◆ </a></span>pooling_output_element</h2>
-
- <div class="memitem">
- <div class="memproto">
- <table class="memname">
- <tr>
- <td class="memname">#define pooling_output_element</td>
- <td>(</td>
- <td class="paramtype"> </td>
- <td class="paramname">typename</td><td>)</td>
- <td></td>
- </tr>
- </table>
- </div><div class="memdoc">
- <b>Value:</b><div class="fragment"><div class="line"><span class="keywordflow">if</span>(!strcmp(mode, <span class="stringliteral">"avg"</span>)){ \</div>
- <div class="line"> ((<span class="keyword">typename</span>*)out_data)[out_linear_idx] = (<span class="keyword">typename</span>)(temp / window_size); \</div>
- <div class="line"> } \</div>
- <div class="line"> else <span class="keywordflow">if</span> (!strcmp(mode, <span class="stringliteral">"max"</span>)) { \</div>
- <div class="line"> ((<span class="keyword">typename</span>*)out_data)[out_linear_idx] = (<span class="keyword">typename</span>)temp; \</div>
- <div class="line"> }</div>
- </div><!-- fragment -->
- </div>
- </div>
- <h2 class="groupheader">Function Documentation</h2>
- <a id="ad061e28180ec47569970fbe30e7ddd97"></a>
- <h2 class="memtitle"><span class="permalink"><a href="#ad061e28180ec47569970fbe30e7ddd97">◆ </a></span>aitisa_pooling()</h2>
-
- <div class="memitem">
- <div class="memproto">
- <table class="memname">
- <tr>
- <td class="memname"><a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> aitisa_pooling </td>
- <td>(</td>
- <td class="paramtype">const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> </td>
- <td class="paramname"><em>input</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const char * </td>
- <td class="paramname"><em>mode</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>ksize</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>stride</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>padding</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>dilation</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype"><a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> * </td>
- <td class="paramname"><em>output</em> </td>
- </tr>
- <tr>
- <td></td>
- <td>)</td>
- <td></td><td></td>
- </tr>
- </table>
- </div><div class="memdoc">
-
- <p>Applies a pooling over an input signal composed of several feature maps. </p>
- <dl class="params"><dt>Parameters</dt><dd>
- <table class="params">
- <tr><td class="paramname">input</td><td>The input tensor of a shape [num_batches, in_channels, in_spatial_shapes]. </td></tr>
- <tr><td class="paramname">mode</td><td>The pooling mode including "max" and "avg". </td></tr>
- <tr><td class="paramname">ksize</td><td>The window size of a shape each of which is >= 1. </td></tr>
- <tr><td class="paramname">stride</td><td>The stride of a shape [spatial_shapes] each of which is >= 1. </td></tr>
- <tr><td class="paramname">padding</td><td>The padding of a shape [spatial_shapes] each of which is >= 0. If the padding number of some axis is non-zero, implicit zero padding is added to input on both sides. If the padding number of some axis is odd, then the rest zeros will be placed on the high-dimension side. </td></tr>
- <tr><td class="paramname">dilation</td><td>The dilation of a shape [spatial_shapes] each of which is >= 1. This is a regular pooling if all values are 1. </td></tr>
- <tr><td class="paramname">output</td><td>The output tensor pointer of a shape [num_batches, out_channels, * * out_spatial_shapes]. </td></tr>
- </table>
- </dd>
- </dl>
- <dl class="section return"><dt>Returns</dt><dd>Status The Status enum indicates whether the routine is OK. </dd></dl>
-
- </div>
- </div>
- <a id="a30bf04d198c7a41f4a4f5839c89100d6"></a>
- <h2 class="memtitle"><span class="permalink"><a href="#a30bf04d198c7a41f4a4f5839c89100d6">◆ </a></span>pooling_create_output()</h2>
-
- <div class="memitem">
- <div class="memproto">
- <table class="mlabels">
- <tr>
- <td class="mlabels-left">
- <table class="memname">
- <tr>
- <td class="memname">static <a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> pooling_create_output </td>
- <td>(</td>
- <td class="paramtype">const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> </td>
- <td class="paramname"><em>input</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">int64_t * </td>
- <td class="paramname"><em>out_dims</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>ksize</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>stride</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>padding</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>dilation</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int </td>
- <td class="paramname"><em>axis</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">int64_t </td>
- <td class="paramname"><em>c_start</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype"><a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> * </td>
- <td class="paramname"><em>output</em> </td>
- </tr>
- <tr>
- <td></td>
- <td>)</td>
- <td></td><td></td>
- </tr>
- </table>
- </td>
- <td class="mlabels-right">
- <span class="mlabels"><span class="mlabel">static</span></span> </td>
- </tr>
- </table>
- </div><div class="memdoc">
-
- </div>
- </div>
- <a id="af1d691ac69f6b1554bd8383efc496ee3"></a>
- <h2 class="memtitle"><span class="permalink"><a href="#af1d691ac69f6b1554bd8383efc496ee3">◆ </a></span>pooling_double()</h2>
-
- <div class="memitem">
- <div class="memproto">
- <table class="mlabels">
- <tr>
- <td class="mlabels-left">
- <table class="memname">
- <tr>
- <td class="memname">static <a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> pooling_double </td>
- <td>(</td>
- <td class="paramtype">const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> </td>
- <td class="paramname"><em>input</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const char * </td>
- <td class="paramname"><em>mode</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>ksize</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>stride</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>padding</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int * </td>
- <td class="paramname"><em>dilation</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const int </td>
- <td class="paramname"><em>axis</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype"><a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> * </td>
- <td class="paramname"><em>output</em> </td>
- </tr>
- <tr>
- <td></td>
- <td>)</td>
- <td></td><td></td>
- </tr>
- </table>
- </td>
- <td class="mlabels-right">
- <span class="mlabels"><span class="mlabel">static</span></span> </td>
- </tr>
- </table>
- </div><div class="memdoc">
-
- </div>
- </div>
- </div><!-- contents -->
- <!-- start footer part -->
- <hr class="footer"/><address class="footer"><small>
- Generated by  <a href="http://www.doxygen.org/index.html">
- <img class="footer" src="doxygen.png" alt="doxygen"/>
- </a> 1.8.16
- </small></address>
- </body>
- </html>
|