| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://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.13"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>mxnet: mshadow::expr::BLASEngine< gpu, half::half_t > Struct Template 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">mxnet |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.13 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| </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="namespacemshadow.html">mshadow</a></li><li class="navelem"><a class="el" href="namespacemshadow_1_1expr.html">expr</a></li><li class="navelem"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html">BLASEngine< gpu, half::half_t ></a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="summary"> |
| <a href="#pub-static-methods">Static Public Member Functions</a> | |
| <a href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4-members.html">List of all members</a> </div> |
| <div class="headertitle"> |
| <div class="title">mshadow::expr::BLASEngine< gpu, half::half_t > Struct Template Reference</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| |
| <p><code>#include <<a class="el" href="dot__engine-inl_8h_source.html">dot_engine-inl.h</a>></code></p> |
| <div class="dynheader"> |
| Collaboration diagram for mshadow::expr::BLASEngine< gpu, half::half_t >:</div> |
| <div class="dyncontent"> |
| <div class="center"><img src="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4__coll__graph.png" border="0" usemap="#mshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4_coll__map" alt="Collaboration graph"/></div> |
| <map name="mshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4_coll__map" id="mshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4_coll__map"> |
| </map> |
| </div> |
| <table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-static-methods"></a> |
| Static Public Member Functions</h2></td></tr> |
| <tr class="memitem:ac742f67973dd9ce1c444f0c476749d10"><td class="memItemLeft" align="right" valign="top">static cublasOperation_t </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#ac742f67973dd9ce1c444f0c476749d10">GetT</a> (bool t)</td></tr> |
| <tr class="separator:ac742f67973dd9ce1c444f0c476749d10"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a2091ee54add7a9ff95c6e1f08bed21d5"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#a2091ee54add7a9ff95c6e1f08bed21d5">SetStream</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream)</td></tr> |
| <tr class="separator:a2091ee54add7a9ff95c6e1f08bed21d5"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a8d6d8e682966822bb85b6ff6fee32ce2"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#a8d6d8e682966822bb85b6ff6fee32ce2">gemm</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, bool transa, bool transb, int m, int n, int k, half::half_t alpha, const half::half_t *A, int lda, const half::half_t *B, int ldb, half::half_t beta, half::half_t *C, int ldc)</td></tr> |
| <tr class="separator:a8d6d8e682966822bb85b6ff6fee32ce2"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a517affe439b1378dff42a2efe8c21dd8"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#a517affe439b1378dff42a2efe8c21dd8">batched_gemm</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, bool transa, bool transb, int m, int n, int k, half::half_t alpha, const half::half_t *A, int lda, const half::half_t *B, int ldb, half::half_t beta, half::half_t *C, int ldc, int batch_count, half::half_t **workspace)</td></tr> |
| <tr class="separator:a517affe439b1378dff42a2efe8c21dd8"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ae014aad49673f34d8f4149ea4525820b"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#ae014aad49673f34d8f4149ea4525820b">gemv</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, bool trans, int m, int n, half::half_t alpha, const half::half_t *A, int lda, const half::half_t *X, int incX, half::half_t beta, half::half_t *Y, int incY)</td></tr> |
| <tr class="separator:ae014aad49673f34d8f4149ea4525820b"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a7dbffea32aea2ed9b58e8098c80cd05f"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#a7dbffea32aea2ed9b58e8098c80cd05f">batched_gemv</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, bool trans, int m, int n, half::half_t alpha, const half::half_t *A, int lda, const half::half_t *X, int incX, half::half_t beta, half::half_t *Y, int incY, int batch_count)</td></tr> |
| <tr class="separator:a7dbffea32aea2ed9b58e8098c80cd05f"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a6d1f6461bd234ab76f380ac6da8f8f1c"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#a6d1f6461bd234ab76f380ac6da8f8f1c">ger</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, int m, int n, half::half_t alpha, const half::half_t *X, int incX, const half::half_t *Y, int incY, half::half_t *A, int lda)</td></tr> |
| <tr class="separator:a6d1f6461bd234ab76f380ac6da8f8f1c"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:af75b98880cf90282c31c002966e2c3ca"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#af75b98880cf90282c31c002966e2c3ca">batched_ger</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, int m, int n, half::half_t alpha, const half::half_t *X, int incX, const half::half_t *Y, int incY, half::half_t *A, int lda, int batch_count)</td></tr> |
| <tr class="separator:af75b98880cf90282c31c002966e2c3ca"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a88faac9a1b47990aed30b1b284154624"><td class="memItemLeft" align="right" valign="top">static void </td><td class="memItemRight" valign="bottom"><a class="el" href="structmshadow_1_1expr_1_1BLASEngine_3_01gpu_00_01half_1_1half__t_01_4.html#a88faac9a1b47990aed30b1b284154624">dot</a> (<a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > *stream, int n, const half::half_t *X, int incX, const half::half_t *Y, int incY, half::half_t *ret)</td></tr> |
| <tr class="separator:a88faac9a1b47990aed30b1b284154624"><td class="memSeparator" colspan="2"> </td></tr> |
| </table> |
| <h2 class="groupheader">Member Function Documentation</h2> |
| <a id="a517affe439b1378dff42a2efe8c21dd8"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a517affe439b1378dff42a2efe8c21dd8">◆ </a></span>batched_gemm()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::batched_gemm </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>transa</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>transb</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>m</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>k</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>alpha</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>A</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>lda</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>B</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>ldb</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>beta</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>C</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>ldc</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>batch_count</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t ** </td> |
| <td class="paramname"><em>workspace</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a7dbffea32aea2ed9b58e8098c80cd05f"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a7dbffea32aea2ed9b58e8098c80cd05f">◆ </a></span>batched_gemv()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::batched_gemv </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>trans</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>m</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>alpha</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>A</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>lda</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>X</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incX</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>beta</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>Y</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incY</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>batch_count</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="af75b98880cf90282c31c002966e2c3ca"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af75b98880cf90282c31c002966e2c3ca">◆ </a></span>batched_ger()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::batched_ger </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>m</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>alpha</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>X</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incX</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>Y</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incY</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>A</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>lda</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>batch_count</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a88faac9a1b47990aed30b1b284154624"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a88faac9a1b47990aed30b1b284154624">◆ </a></span>dot()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::dot </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>X</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incX</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>Y</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incY</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>ret</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a8d6d8e682966822bb85b6ff6fee32ce2"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a8d6d8e682966822bb85b6ff6fee32ce2">◆ </a></span>gemm()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::gemm </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>transa</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>transb</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>m</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>k</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>alpha</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>A</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>lda</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>B</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>ldb</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>beta</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>C</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>ldc</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ae014aad49673f34d8f4149ea4525820b"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ae014aad49673f34d8f4149ea4525820b">◆ </a></span>gemv()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::gemv </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>trans</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>m</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>alpha</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>A</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>lda</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>X</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incX</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>beta</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>Y</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incY</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a6d1f6461bd234ab76f380ac6da8f8f1c"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a6d1f6461bd234ab76f380ac6da8f8f1c">◆ </a></span>ger()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::ger </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>m</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>n</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t </td> |
| <td class="paramname"><em>alpha</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>X</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incX</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const half::half_t * </td> |
| <td class="paramname"><em>Y</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>incY</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">half::half_t * </td> |
| <td class="paramname"><em>A</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>lda</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">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ac742f67973dd9ce1c444f0c476749d10"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ac742f67973dd9ce1c444f0c476749d10">◆ </a></span>GetT()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static cublasOperation_t <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::GetT </td> |
| <td>(</td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>t</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a2091ee54add7a9ff95c6e1f08bed21d5"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a2091ee54add7a9ff95c6e1f08bed21d5">◆ </a></span>SetStream()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static void <a class="el" href="structmshadow_1_1expr_1_1BLASEngine.html">mshadow::expr::BLASEngine</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a>, half::half_t >::SetStream </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structmshadow_1_1Stream.html">Stream</a>< <a class="el" href="structmshadow_1_1gpu.html">gpu</a> > * </td> |
| <td class="paramname"><em>stream</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <hr/>The documentation for this struct was generated from the following file:<ul> |
| <li>/work/mxnet/3rdparty/mshadow/mshadow/<a class="el" href="dot__engine-inl_8h_source.html">dot_engine-inl.h</a></li> |
| </ul> |
| </div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Apr 28 2022 17:47:37 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.13 |
| </small></address> |
| </body> |
| </html> |