|
- <!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/math/matmul.h 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_ae9a93452e2a84339148a16bcf2eb561.html">math</a></li> </ul>
- </div>
- </div><!-- top -->
- <div class="header">
- <div class="summary">
- <a href="#func-members">Functions</a> </div>
- <div class="headertitle">
- <div class="title">matmul.h File Reference</div> </div>
- </div><!--header-->
- <div class="contents">
- <div class="textblock"><code>#include "<a class="el" href="tensor_8h_source.html">src/core/tensor.h</a>"</code><br />
- </div>
- <p><a href="matmul_8h_source.html">Go to the source code of this file.</a></p>
- <table class="memberdecls">
- <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
- Functions</h2></td></tr>
- <tr class="memitem:a004e7e1d896fc56f91a81230024deef8"><td class="memItemLeft" align="right" valign="top"><a class="el" href="macros_8h.html#ac8b488663e34cddcd0952bc63ed91178">AITISA_API_PUBLIC</a> <a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="matmul_8h.html#a004e7e1d896fc56f91a81230024deef8">aitisa_matmul</a> (const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> tensor1, const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> tensor2, <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> *output)</td></tr>
- <tr class="memdesc:a004e7e1d896fc56f91a81230024deef8"><td class="mdescLeft"> </td><td class="mdescRight">Matrix multiplication of tensor1 and tensor2. <a href="matmul_8h.html#a004e7e1d896fc56f91a81230024deef8">More...</a><br /></td></tr>
- <tr class="separator:a004e7e1d896fc56f91a81230024deef8"><td class="memSeparator" colspan="2"> </td></tr>
- </table>
- <h2 class="groupheader">Function Documentation</h2>
- <a id="a004e7e1d896fc56f91a81230024deef8"></a>
- <h2 class="memtitle"><span class="permalink"><a href="#a004e7e1d896fc56f91a81230024deef8">◆ </a></span>aitisa_matmul()</h2>
-
- <div class="memitem">
- <div class="memproto">
- <table class="memname">
- <tr>
- <td class="memname"><a class="el" href="macros_8h.html#ac8b488663e34cddcd0952bc63ed91178">AITISA_API_PUBLIC</a> <a class="el" href="status_8h.html#a67a0db04d321a74b7e7fcfd3f1a3f70b">Status</a> aitisa_matmul </td>
- <td>(</td>
- <td class="paramtype">const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> </td>
- <td class="paramname"><em>tensor1</em>, </td>
- </tr>
- <tr>
- <td class="paramkey"></td>
- <td></td>
- <td class="paramtype">const <a class="el" href="tensor_8h.html#a621a4e61b2dc779bfe3f7b216051f100">Tensor</a> </td>
- <td class="paramname"><em>tensor2</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>Matrix multiplication of tensor1 and tensor2. </p>
- <p>Apply different kernel(dot, gemv, gemm, batch_gemm) according the dim size of inputs: (0) ndim_tensor1 < 1 & ndim_tensor2 < 1 ==> INVALID ERROR. (1) ndim_tensor1 == 1 & ndim_tensor2 == 1 ==> output = dot(tensor1, tensor2); (2) ndim_tensor1 == 1 & ndim_tensor2 == 2 ==> tensor1 is considered as a one-row matrix, output = gemm(tensor1, tensor2); (3) ndim_tensor1 == 1 & ndim_tensor2 >= 3 ==> tensor1 is considered as a one-row matrix, output = batch_gemm(tensor1, tensor2); (4) ndim_tensor1 == 2 & ndim_tensor2 == 1 ==> output = gemv(tensor1, tensor2); (5) ndim_tensor1 == 2 & ndim_tensor2 == 2 ==> output = gemm(tensor1, tensor2); (6) ndim_tensor1 == 2 & ndim_tensor2 >= 3 ==> output = batch_gemm(tensor1, tensor2); (7) ndim_tensor1 >= 3 & ndim_tensor2 == 1 ==> tensor2 is considered as a one-column matrix, output = batch_gemm(tensor1, tensor2); (8) ndim_tensor1 >= 3 & ndim_tensor2 >= 2 ==> output = batch_gemm(tensor1, tensor2);</p>
- <dl class="params"><dt>Parameters</dt><dd>
- <table class="params">
- <tr><td class="paramname">tensor1</td><td>The first input. </td></tr>
- <tr><td class="paramname">tensor2</td><td>The second input. </td></tr>
- <tr><td class="paramname">output</td><td>Pointer to the result tensor, the output will be resized inside the function.</td></tr>
- </table>
- </dd>
- </dl>
- <dl class="section return"><dt>Returns</dt><dd>Status. </dd></dl>
- <dl class="retval"><dt>Return values</dt><dd>
- <table class="retval">
- <tr><td class="paramname">STATUS_SUCCESS</td><td>Success. </td></tr>
- <tr><td class="paramname">STATUS_NOT_SUPPORTED</td><td>The device type or data type is not supported. </td></tr>
- <tr><td class="paramname">STATUS_INVALID_ARGUMENT</td><td>The dims of inputs are invalid. </td></tr>
- <tr><td class="paramname">STATUS_DIMENSIONS_MISMATCH</td><td>The dims of inputs are mismatched.</td></tr>
- </table>
- </dd>
- </dl>
- <dl class="section note"><dt>Note</dt><dd>(1) Only CPU is supported for now. (2) before doing batch_gemm, the batch dims of tensor1 and tensor2 will be broadcasted. </dd></dl>
-
- </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>
|