blob: 88c5e36940141c73049efa414b4bf1875ea32e91 [file] [log] [blame]
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<meta name="generator" content="Doxygen 1.8.17"/>
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<title>mxnet: mxnet::ext::Node Class 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.17 -->
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;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&amp;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="namespacemxnet.html">mxnet</a></li><li class="navelem"><a class="el" href="namespacemxnet_1_1ext.html">ext</a></li><li class="navelem"><a class="el" href="classmxnet_1_1ext_1_1Node.html">Node</a></li> </ul>
</div>
</div><!-- top -->
<div class="header">
<div class="summary">
<a href="#pub-methods">Public Member Functions</a> &#124;
<a href="#pub-attribs">Public Attributes</a> &#124;
<a href="classmxnet_1_1ext_1_1Node-members.html">List of all members</a> </div>
<div class="headertitle">
<div class="title">mxnet::ext::Node Class Reference</div> </div>
</div><!--header-->
<div class="contents">
<p><code>#include &lt;<a class="el" href="lib__api_8h_source.html">lib_api.h</a>&gt;</code></p>
<div class="dynheader">
Collaboration diagram for mxnet::ext::Node:</div>
<div class="dyncontent">
<div class="center"><img src="classmxnet_1_1ext_1_1Node__coll__graph.png" border="0" usemap="#mxnet_1_1ext_1_1Node_coll__map" alt="Collaboration graph"/></div>
<!-- MAP 0 -->
</div>
<table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-methods"></a>
Public Member Functions</h2></td></tr>
<tr class="memitem:a2c1ec7bd5c9132e2636e70a93a53e452"><td class="memItemLeft" align="right" valign="top">&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a2c1ec7bd5c9132e2636e70a93a53e452">Node</a> ()</td></tr>
<tr class="separator:a2c1ec7bd5c9132e2636e70a93a53e452"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:aac12981007cdb017f652d1242cf9ab05"><td class="memItemLeft" align="right" valign="top">void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#aac12981007cdb017f652d1242cf9ab05">_setPassResource</a> (<a class="el" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a> *res_)</td></tr>
<tr class="separator:aac12981007cdb017f652d1242cf9ab05"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a391e1c2ccc10a9edc19417fa3fad3169"><td class="memItemLeft" align="right" valign="top">void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a391e1c2ccc10a9edc19417fa3fad3169">alloc_arg</a> (const std::vector&lt; int64_t &gt; &amp;shapes, const <a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> &amp;ctx, <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype)</td></tr>
<tr class="separator:a391e1c2ccc10a9edc19417fa3fad3169"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a17744ecbf554a996fe84dc5553564633"><td class="memItemLeft" align="right" valign="top">void&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a17744ecbf554a996fe84dc5553564633">alloc_aux</a> (const std::vector&lt; int64_t &gt; &amp;shapes, const <a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> &amp;ctx, <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype)</td></tr>
<tr class="separator:a17744ecbf554a996fe84dc5553564633"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table><table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-attribs"></a>
Public Attributes</h2></td></tr>
<tr class="memitem:a20c7b312f050f3a1cc606e924482a4c7"><td class="memItemLeft" align="right" valign="top">std::string&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a20c7b312f050f3a1cc606e924482a4c7">op</a></td></tr>
<tr class="separator:a20c7b312f050f3a1cc606e924482a4c7"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:abc2b3e8b645050faf1a4534de694e939"><td class="memItemLeft" align="right" valign="top">std::string&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#abc2b3e8b645050faf1a4534de694e939">name</a></td></tr>
<tr class="separator:abc2b3e8b645050faf1a4534de694e939"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a3489d59197ad9612dbdc23f963aa8d02"><td class="memItemLeft" align="right" valign="top"><a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a> *&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a3489d59197ad9612dbdc23f963aa8d02">tensor</a></td></tr>
<tr class="separator:a3489d59197ad9612dbdc23f963aa8d02"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a8364c3b8ae3800c775a23a8e90984c82"><td class="memItemLeft" align="right" valign="top">std::vector&lt; <a class="el" href="structmxnet_1_1ext_1_1NodeEntry.html">NodeEntry</a> &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a8364c3b8ae3800c775a23a8e90984c82">inputs</a></td></tr>
<tr class="separator:a8364c3b8ae3800c775a23a8e90984c82"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:aee95220e1a30197ca5fa726742f57183"><td class="memItemLeft" align="right" valign="top">std::vector&lt; <a class="el" href="structmxnet_1_1ext_1_1NodeEntry.html">NodeEntry</a> &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#aee95220e1a30197ca5fa726742f57183">outputs</a></td></tr>
<tr class="separator:aee95220e1a30197ca5fa726742f57183"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:afbe960c8b040871dfb9f8c8bdb529947"><td class="memItemLeft" align="right" valign="top">std::vector&lt; <a class="el" href="classmxnet_1_1ext_1_1Graph.html">Graph</a> * &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#afbe960c8b040871dfb9f8c8bdb529947">subgraphs</a></td></tr>
<tr class="separator:afbe960c8b040871dfb9f8c8bdb529947"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a3e1851c411ee0112de6ee3f1a90d2151"><td class="memItemLeft" align="right" valign="top">std::unordered_map&lt; std::string, std::string &gt;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html#a3e1851c411ee0112de6ee3f1a90d2151">attrs</a></td></tr>
<tr class="separator:a3e1851c411ee0112de6ee3f1a90d2151"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table>
<h2 class="groupheader">Constructor &amp; Destructor Documentation</h2>
<a id="a2c1ec7bd5c9132e2636e70a93a53e452"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a2c1ec7bd5c9132e2636e70a93a53e452">&#9670;&nbsp;</a></span>Node()</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">mxnet::ext::Node::Node </td>
<td>(</td>
<td class="paramname"></td><td>)</td>
<td></td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<h2 class="groupheader">Member Function Documentation</h2>
<a id="aac12981007cdb017f652d1242cf9ab05"></a>
<h2 class="memtitle"><span class="permalink"><a href="#aac12981007cdb017f652d1242cf9ab05">&#9670;&nbsp;</a></span>_setPassResource()</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">void mxnet::ext::Node::_setPassResource </td>
<td>(</td>
<td class="paramtype"><a class="el" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a> *&#160;</td>
<td class="paramname"><em>res_</em></td><td>)</td>
<td></td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="a391e1c2ccc10a9edc19417fa3fad3169"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a391e1c2ccc10a9edc19417fa3fad3169">&#9670;&nbsp;</a></span>alloc_arg()</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">void mxnet::ext::Node::alloc_arg </td>
<td>(</td>
<td class="paramtype">const std::vector&lt; int64_t &gt; &amp;&#160;</td>
<td class="paramname"><em>shapes</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype">const <a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> &amp;&#160;</td>
<td class="paramname"><em>ctx</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a>&#160;</td>
<td class="paramname"><em>dtype</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="a17744ecbf554a996fe84dc5553564633"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a17744ecbf554a996fe84dc5553564633">&#9670;&nbsp;</a></span>alloc_aux()</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">void mxnet::ext::Node::alloc_aux </td>
<td>(</td>
<td class="paramtype">const std::vector&lt; int64_t &gt; &amp;&#160;</td>
<td class="paramname"><em>shapes</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype">const <a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> &amp;&#160;</td>
<td class="paramname"><em>ctx</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a>&#160;</td>
<td class="paramname"><em>dtype</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<h2 class="groupheader">Member Data Documentation</h2>
<a id="a3e1851c411ee0112de6ee3f1a90d2151"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a3e1851c411ee0112de6ee3f1a90d2151">&#9670;&nbsp;</a></span>attrs</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::unordered_map&lt;std::string, std::string&gt; mxnet::ext::Node::attrs</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="a8364c3b8ae3800c775a23a8e90984c82"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a8364c3b8ae3800c775a23a8e90984c82">&#9670;&nbsp;</a></span>inputs</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::vector&lt;<a class="el" href="structmxnet_1_1ext_1_1NodeEntry.html">NodeEntry</a>&gt; mxnet::ext::Node::inputs</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="abc2b3e8b645050faf1a4534de694e939"></a>
<h2 class="memtitle"><span class="permalink"><a href="#abc2b3e8b645050faf1a4534de694e939">&#9670;&nbsp;</a></span>name</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::string mxnet::ext::Node::name</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="a20c7b312f050f3a1cc606e924482a4c7"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a20c7b312f050f3a1cc606e924482a4c7">&#9670;&nbsp;</a></span>op</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::string mxnet::ext::Node::op</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="aee95220e1a30197ca5fa726742f57183"></a>
<h2 class="memtitle"><span class="permalink"><a href="#aee95220e1a30197ca5fa726742f57183">&#9670;&nbsp;</a></span>outputs</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::vector&lt;<a class="el" href="structmxnet_1_1ext_1_1NodeEntry.html">NodeEntry</a>&gt; mxnet::ext::Node::outputs</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="afbe960c8b040871dfb9f8c8bdb529947"></a>
<h2 class="memtitle"><span class="permalink"><a href="#afbe960c8b040871dfb9f8c8bdb529947">&#9670;&nbsp;</a></span>subgraphs</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::vector&lt;<a class="el" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>*&gt; mxnet::ext::Node::subgraphs</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="a3489d59197ad9612dbdc23f963aa8d02"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a3489d59197ad9612dbdc23f963aa8d02">&#9670;&nbsp;</a></span>tensor</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname"><a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>* mxnet::ext::Node::tensor</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<hr/>The documentation for this class was generated from the following file:<ul>
<li>/work/mxnet/include/mxnet/<a class="el" href="lib__api_8h_source.html">lib_api.h</a></li>
</ul>
</div><!-- contents -->
<!-- start footer part -->
<hr class="footer"/><address class="footer"><small>
Generated on Thu Jan 5 2023 03:47:41 for mxnet by &#160;<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/>
</a> 1.8.17
</small></address>
</body>
</html>