Google Git
Sign in
apache / mxnet-site / d5c5a235eef72f64ccaf224b834a20ef3287469b / . / versions / 1.9.1 / api / cpp / docs / api / 3rdparty_2tvm_2nnvm_2include_2nnvm_2op__attr__types_8h_source.html
blob: 9ef3cebf371f030ff90c5bc5ab585af7e7afd2f8 [file] [log] [blame]
<!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: /work/mxnet/3rdparty/tvm/nnvm/include/nnvm/op_attr_types.h Source File</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="dir_8cab8f464681f7cc51cee77e79a434cd.html">3rdparty</a></li><li class="navelem"><a class="el" href="dir_d40daf7ef9783ad6669561b9a39523a5.html">tvm</a></li><li class="navelem"><a class="el" href="dir_59ab2d7405f48271a817356d16f4868c.html">nnvm</a></li><li class="navelem"><a class="el" href="dir_564b1b3930cee8f0cd6786ede046ff35.html">include</a></li><li class="navelem"><a class="el" href="dir_aadb892b7c99db7e2c3fc68a6c967545.html">nnvm</a></li> </ul>
</div>
</div><!-- top -->
<div class="header">
<div class="headertitle">
<div class="title">op_attr_types.h</div> </div>
</div><!--header-->
<div class="contents">
<a href="3rdparty_2tvm_2nnvm_2include_2nnvm_2op__attr__types_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span>&#160;<span class="comment">/*</span></div><div class="line"><a name="l00002"></a><span class="lineno"> 2</span>&#160;<span class="comment"> * Licensed to the Apache Software Foundation (ASF) under one</span></div><div class="line"><a name="l00003"></a><span class="lineno"> 3</span>&#160;<span class="comment"> * or more contributor license agreements. See the NOTICE file</span></div><div class="line"><a name="l00004"></a><span class="lineno"> 4</span>&#160;<span class="comment"> * distributed with this work for additional information</span></div><div class="line"><a name="l00005"></a><span class="lineno"> 5</span>&#160;<span class="comment"> * regarding copyright ownership. The ASF licenses this file</span></div><div class="line"><a name="l00006"></a><span class="lineno"> 6</span>&#160;<span class="comment"> * to you under the Apache License, Version 2.0 (the</span></div><div class="line"><a name="l00007"></a><span class="lineno"> 7</span>&#160;<span class="comment"> * &quot;License&quot;); you may not use this file except in compliance</span></div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span>&#160;<span class="comment"> * with the License. You may obtain a copy of the License at</span></div><div class="line"><a name="l00009"></a><span class="lineno"> 9</span>&#160;<span class="comment"> *</span></div><div class="line"><a name="l00010"></a><span class="lineno"> 10</span>&#160;<span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span></div><div class="line"><a name="l00011"></a><span class="lineno"> 11</span>&#160;<span class="comment"> *</span></div><div class="line"><a name="l00012"></a><span class="lineno"> 12</span>&#160;<span class="comment"> * Unless required by applicable law or agreed to in writing,</span></div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span>&#160;<span class="comment"> * software distributed under the License is distributed on an</span></div><div class="line"><a name="l00014"></a><span class="lineno"> 14</span>&#160;<span class="comment"> * &quot;AS IS&quot; BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span>&#160;<span class="comment"> * KIND, either express or implied. See the License for the</span></div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160;<span class="comment"> * specific language governing permissions and limitations</span></div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160;<span class="comment"> * under the License.</span></div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span>&#160;<span class="comment"> */</span></div><div class="line"><a name="l00019"></a><span class="lineno"> 19</span>&#160;</div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160;<span class="preprocessor">#ifndef NNVM_OP_ATTR_TYPES_H_</span></div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160;<span class="preprocessor">#define NNVM_OP_ATTR_TYPES_H_</span></div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160;</div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160;<span class="preprocessor">#include &lt;vector&gt;</span></div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160;<span class="preprocessor">#include &lt;string&gt;</span></div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160;<span class="preprocessor">#include &lt;utility&gt;</span></div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160;<span class="preprocessor">#include &lt;functional&gt;</span></div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160;<span class="preprocessor">#include &lt;unordered_map&gt;</span></div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2base_8h.html">base.h</a>&quot;</span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2node_8h.html">node.h</a>&quot;</span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2tuple_8h.html">tuple.h</a>&quot;</span></div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="layout_8h.html">layout.h</a>&quot;</span></div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160;</div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160;<span class="keyword">namespace </span><a class="code" href="namespacennvm.html">nnvm</a> {</div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160;</div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160;<span class="comment">// These types are optional attributes in each operator.</span></div><div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160;<span class="comment">// Each attribute can be required by some passes.</span></div><div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160;</div><div class="line"><a name="l00051"></a><span class="lineno"><a class="line" href="namespacennvm.html#a5b811999ef6b9e8a789e2c359642738a"> 51</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a5b811999ef6b9e8a789e2c359642738a">FListInputNames</a> = std::function&lt;std::vector&lt;std::string&gt; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160;</div><div class="line"><a name="l00063"></a><span class="lineno"><a class="line" href="namespacennvm.html#a475e6d8a1f6f0f63977e0b2e2a8b0ef7"> 63</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a475e6d8a1f6f0f63977e0b2e2a8b0ef7">FNumVisibleOutputs</a> = std::function&lt;uint32_t (const NodeAttrs&amp; attrs)&gt;;</div><div class="line"><a name="l00064"></a><span class="lineno"> 64</span>&#160;</div><div class="line"><a name="l00074"></a><span class="lineno"><a class="line" href="namespacennvm.html#af4188d4a9d1daecdecb9b0136b9ff3e7"> 74</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#af4188d4a9d1daecdecb9b0136b9ff3e7">FListOutputNames</a> = std::function&lt;std::vector&lt;std::string&gt; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00075"></a><span class="lineno"> 75</span>&#160;</div><div class="line"><a name="l00084"></a><span class="lineno"><a class="line" href="namespacennvm.html#a12f43958d8171a8f2ab11880192b7125"> 84</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a12f43958d8171a8f2ab11880192b7125">FMutateInputs</a> = std::function&lt;std::vector&lt;uint32_t&gt; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span>&#160;</div><div class="line"><a name="l00091"></a><span class="lineno"> 91</span>&#160;<span class="keyword">template</span>&lt;<span class="keyword">typename</span> AttrType&gt;</div><div class="line"><a name="l00092"></a><span class="lineno"> 92</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a39a12a38eb8f3ac625170e8be2099d57">FInferNodeEntryAttr</a> = std::function&lt;bool (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs,</div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span>&#160; std::vector&lt;AttrType&gt; *in_attrs,</div><div class="line"><a name="l00094"></a><span class="lineno"><a class="line" href="namespacennvm.html#a39a12a38eb8f3ac625170e8be2099d57"> 94</a></span>&#160; std::vector&lt;AttrType&gt; *out_attrs)&gt;;</div><div class="line"><a name="l00095"></a><span class="lineno"> 95</span>&#160;</div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#ac04d7b8e69e50b67769cb496c15b5fce">FGetAttrDict</a> = std::function&lt;</div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>&#160; std::unordered_map&lt;std::string, std::string&gt;</div><div class="line"><a name="l00105"></a><span class="lineno"><a class="line" href="namespacennvm.html#ac04d7b8e69e50b67769cb496c15b5fce"> 105</a></span>&#160; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160;</div><div class="line"><a name="l00117"></a><span class="lineno"><a class="line" href="namespacennvm.html#ae80cd6cc2432e36350ad84335d5e5308"> 117</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#ae80cd6cc2432e36350ad84335d5e5308">FInferShape</a> = <a class="code" href="namespacennvm.html#a39a12a38eb8f3ac625170e8be2099d57">FInferNodeEntryAttr&lt;TShape&gt;</a>;</div><div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160;</div><div class="line"><a name="l00126"></a><span class="lineno"><a class="line" href="namespacennvm.html#a6539f5c3067c4c3cd7dbbdd6b6a6f165"> 126</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a6539f5c3067c4c3cd7dbbdd6b6a6f165">FInferType</a> = <a class="code" href="namespacennvm.html#a39a12a38eb8f3ac625170e8be2099d57">FInferNodeEntryAttr&lt;int&gt;</a>;</div><div class="line"><a name="l00127"></a><span class="lineno"> 127</span>&#160;</div><div class="line"><a name="l00136"></a><span class="lineno"><a class="line" href="namespacennvm.html#ac317e6034c26f9cb453e91cd38e35df9"> 136</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#ac317e6034c26f9cb453e91cd38e35df9">TIsBackward</a> = bool;</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160;</div><div class="line"><a name="l00147"></a><span class="lineno"><a class="line" href="namespacennvm.html#ad27d6c45aee15d0dab91419c63f8f900"> 147</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#ad27d6c45aee15d0dab91419c63f8f900">TIsGhost</a> = bool;</div><div class="line"><a name="l00148"></a><span class="lineno"> 148</span>&#160;</div><div class="line"><a name="l00158"></a><span class="lineno"> 158</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a7d4bdafc4a28981d4d9b5fc8207fc04c">FInplaceOption</a> = std::function&lt;</div><div class="line"><a name="l00159"></a><span class="lineno"><a class="line" href="namespacennvm.html#a7d4bdafc4a28981d4d9b5fc8207fc04c"> 159</a></span>&#160; std::vector&lt;std::pair&lt;int, int&gt; &gt; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00160"></a><span class="lineno"> 160</span>&#160;</div><div class="line"><a name="l00171"></a><span class="lineno"><a class="line" href="namespacennvm.html#a39ef67828f4ddb6c7aea2e3b1b685d90"> 171</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a39ef67828f4ddb6c7aea2e3b1b685d90">FInplaceIdentity</a> = std::function&lt;std::vector&lt;bool&gt; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span>&#160;</div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#ab951c7d30fcc4f17ffddbfd082d24792">FIgnoreInputs</a> = std::function&lt;</div><div class="line"><a name="l00183"></a><span class="lineno"><a class="line" href="namespacennvm.html#ab951c7d30fcc4f17ffddbfd082d24792"> 183</a></span>&#160; std::vector&lt;uint32_t&gt; (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00184"></a><span class="lineno"> 184</span>&#160;</div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#ad12fee9c95a28ac9441c38d66cde559d">FGradient</a> = std::function&lt;std::vector&lt;NodeEntry&gt;(</div><div class="line"><a name="l00195"></a><span class="lineno"> 195</span>&#160; <span class="keyword">const</span> <a class="code" href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">ObjectPtr</a>&amp; nodeptr,</div><div class="line"><a name="l00196"></a><span class="lineno"><a class="line" href="namespacennvm.html#ad12fee9c95a28ac9441c38d66cde559d"> 196</a></span>&#160; <span class="keyword">const</span> std::vector&lt;NodeEntry&gt;&amp; out_grads)&gt;;</div><div class="line"><a name="l00197"></a><span class="lineno"> 197</span>&#160;</div><div class="line"><a name="l00205"></a><span class="lineno"> 205</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#aa7c5989aee30cef7813c484d12fc922d">FSetInputVarAttrOnCompose</a> = std::function&lt;void(</div><div class="line"><a name="l00206"></a><span class="lineno"> 206</span>&#160; <span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs,</div><div class="line"><a name="l00207"></a><span class="lineno"> 207</span>&#160; <a class="code" href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">ObjectPtr</a> var,</div><div class="line"><a name="l00208"></a><span class="lineno"><a class="line" href="namespacennvm.html#aa7c5989aee30cef7813c484d12fc922d"> 208</a></span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> index)&gt;;</div><div class="line"><a name="l00209"></a><span class="lineno"> 209</span>&#160;</div><div class="line"><a name="l00229"></a><span class="lineno"> 229</span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#a2df0ffb0a105133c97d5fbc9d782c618">FCorrectLayout</a> = std::function&lt;bool(</div><div class="line"><a name="l00230"></a><span class="lineno"> 230</span>&#160; <span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs,</div><div class="line"><a name="l00231"></a><span class="lineno"> 231</span>&#160; std::vector&lt;Layout&gt; *ilayouts,</div><div class="line"><a name="l00232"></a><span class="lineno"> 232</span>&#160; <span class="keyword">const</span> std::vector&lt;Layout&gt; *last_ilayouts,</div><div class="line"><a name="l00233"></a><span class="lineno"><a class="line" href="namespacennvm.html#a2df0ffb0a105133c97d5fbc9d782c618"> 233</a></span>&#160; std::vector&lt;Layout&gt; *olayouts)&gt;;</div><div class="line"><a name="l00234"></a><span class="lineno"> 234</span>&#160;</div><div class="line"><a name="l00245"></a><span class="lineno"><a class="line" href="namespacennvm.html#af0d636002e2d607e2692cd0899c416b6"> 245</a></span>&#160;<span class="keyword">using</span> <a class="code" href="namespacennvm.html#af0d636002e2d607e2692cd0899c416b6">FInputGraph</a> = std::function&lt;std::vector&lt;uint32_t&gt;(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>&amp; attrs)&gt;;</div><div class="line"><a name="l00246"></a><span class="lineno"> 246</span>&#160;</div><div class="line"><a name="l00247"></a><span class="lineno"> 247</span>&#160;} <span class="comment">// namespace nnvm</span></div><div class="line"><a name="l00248"></a><span class="lineno"> 248</span>&#160;</div><div class="line"><a name="l00249"></a><span class="lineno"> 249</span>&#160;<span class="preprocessor">#endif // NNVM_OP_ATTR_TYPES_H_</span></div><div class="ttc" id="namespacennvm_html_ac04d7b8e69e50b67769cb496c15b5fce"><div class="ttname"><a href="namespacennvm.html#ac04d7b8e69e50b67769cb496c15b5fce">nnvm::FGetAttrDict</a></div><div class="ttdeci">std::function&lt; std::unordered_map&lt; std::string, std::string &gt;(const NodeAttrs &amp;attrs)&gt; FGetAttrDict</div><div class="ttdoc">Get attribute dictionary from node. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:105</div></div>
<div class="ttc" id="namespacennvm_html"><div class="ttname"><a href="namespacennvm.html">nnvm</a></div><div class="ttdef"><b>Definition:</b> base.h:35</div></div>
<div class="ttc" id="namespacennvm_html_ad12fee9c95a28ac9441c38d66cde559d"><div class="ttname"><a href="namespacennvm.html#ad12fee9c95a28ac9441c38d66cde559d">nnvm::FGradient</a></div><div class="ttdeci">std::function&lt; std::vector&lt; NodeEntry &gt;(const ObjectPtr &amp;nodeptr, const std::vector&lt; NodeEntry &gt; &amp;out_grads)&gt; FGradient</div><div class="ttdoc">Get the gradient node of the op node This function generates the backward graph of the node...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:196</div></div>
<div class="ttc" id="namespacennvm_html_a5b811999ef6b9e8a789e2c359642738a"><div class="ttname"><a href="namespacennvm.html#a5b811999ef6b9e8a789e2c359642738a">nnvm::FListInputNames</a></div><div class="ttdeci">std::function&lt; std::vector&lt; std::string &gt;(const NodeAttrs &amp;attrs)&gt; FListInputNames</div><div class="ttdoc">Return list of input arguments names of each operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:51</div></div>
<div class="ttc" id="namespacennvm_html_a7d4bdafc4a28981d4d9b5fc8207fc04c"><div class="ttname"><a href="namespacennvm.html#a7d4bdafc4a28981d4d9b5fc8207fc04c">nnvm::FInplaceOption</a></div><div class="ttdeci">std::function&lt; std::vector&lt; std::pair&lt; int, int &gt; &gt;(const NodeAttrs &amp;attrs)&gt; FInplaceOption</div><div class="ttdoc">Get possible inplace options. This function enables optimization to reuse memory of inputs in output...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:159</div></div>
<div class="ttc" id="structnnvm_1_1NodeAttrs_html"><div class="ttname"><a href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a></div><div class="ttdoc">The attributes of the current operation node. Usually are additional parameters like axis...</div><div class="ttdef"><b>Definition:</b> node.h:119</div></div>
<div class="ttc" id="namespacennvm_html_ad27d6c45aee15d0dab91419c63f8f900"><div class="ttname"><a href="namespacennvm.html#ad27d6c45aee15d0dab91419c63f8f900">nnvm::TIsGhost</a></div><div class="ttdeci">bool TIsGhost</div><div class="ttdoc">Whether this op is a ghost node. If TIsGhost is true: </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:147</div></div>
<div class="ttc" id="namespacennvm_html_ae80cd6cc2432e36350ad84335d5e5308"><div class="ttname"><a href="namespacennvm.html#ae80cd6cc2432e36350ad84335d5e5308">nnvm::FInferShape</a></div><div class="ttdeci">FInferNodeEntryAttr&lt; TShape &gt; FInferShape</div><div class="ttdoc">Shape inference function. Update the shapes given the input shape information. TShape.ndim() == 0 means the shape is still unknown. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:117</div></div>
<div class="ttc" id="3rdparty_2tvm_2nnvm_2include_2nnvm_2node_8h_html"><div class="ttname"><a href="3rdparty_2tvm_2nnvm_2include_2nnvm_2node_8h.html">node.h</a></div><div class="ttdoc">Graph node data structure. </div></div>
<div class="ttc" id="namespacennvm_html_a12f43958d8171a8f2ab11880192b7125"><div class="ttname"><a href="namespacennvm.html#a12f43958d8171a8f2ab11880192b7125">nnvm::FMutateInputs</a></div><div class="ttdeci">std::function&lt; std::vector&lt; uint32_t &gt;(const NodeAttrs &amp;attrs)&gt; FMutateInputs</div><div class="ttdoc">Check whether operator will mutate k-th input. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:84</div></div>
<div class="ttc" id="namespacennvm_html_af4188d4a9d1daecdecb9b0136b9ff3e7"><div class="ttname"><a href="namespacennvm.html#af4188d4a9d1daecdecb9b0136b9ff3e7">nnvm::FListOutputNames</a></div><div class="ttdeci">std::function&lt; std::vector&lt; std::string &gt;(const NodeAttrs &amp;attrs)&gt; FListOutputNames</div><div class="ttdoc">Return list of output arguments names of each operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:74</div></div>
<div class="ttc" id="3rdparty_2tvm_2nnvm_2include_2nnvm_2tuple_8h_html"><div class="ttname"><a href="3rdparty_2tvm_2nnvm_2include_2nnvm_2tuple_8h.html">tuple.h</a></div><div class="ttdoc">Data structure Tuple and TShape to store dynamic sized shapes. </div></div>
<div class="ttc" id="namespacennvm_html_a39ef67828f4ddb6c7aea2e3b1b685d90"><div class="ttname"><a href="namespacennvm.html#a39ef67828f4ddb6c7aea2e3b1b685d90">nnvm::FInplaceIdentity</a></div><div class="ttdeci">std::function&lt; std::vector&lt; bool &gt;(const NodeAttrs &amp;attrs)&gt; FInplaceIdentity</div><div class="ttdoc">Get if the inplace option is an identity This function enables inplace optimization even when input r...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:171</div></div>
<div class="ttc" id="namespacennvm_html_ab951c7d30fcc4f17ffddbfd082d24792"><div class="ttname"><a href="namespacennvm.html#ab951c7d30fcc4f17ffddbfd082d24792">nnvm::FIgnoreInputs</a></div><div class="ttdeci">std::function&lt; std::vector&lt; uint32_t &gt;(const NodeAttrs &amp;attrs)&gt; FIgnoreInputs</div><div class="ttdoc">Get list of inputs in the op whose content are actually not used by the operator These are dummy inpu...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:183</div></div>
<div class="ttc" id="namespacennvm_html_a39a12a38eb8f3ac625170e8be2099d57"><div class="ttname"><a href="namespacennvm.html#a39a12a38eb8f3ac625170e8be2099d57">nnvm::FInferNodeEntryAttr</a></div><div class="ttdeci">std::function&lt; bool(const NodeAttrs &amp;attrs, std::vector&lt; AttrType &gt; *in_attrs, std::vector&lt; AttrType &gt; *out_attrs)&gt; FInferNodeEntryAttr</div><div class="ttdoc">Inference function of certain type. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:94</div></div>
<div class="ttc" id="namespacennvm_html_a6539f5c3067c4c3cd7dbbdd6b6a6f165"><div class="ttname"><a href="namespacennvm.html#a6539f5c3067c4c3cd7dbbdd6b6a6f165">nnvm::FInferType</a></div><div class="ttdeci">FInferNodeEntryAttr&lt; int &gt; FInferType</div><div class="ttdoc">Type inference function. Update the type given the known type information. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:126</div></div>
<div class="ttc" id="namespacennvm_html_af0d636002e2d607e2692cd0899c416b6"><div class="ttname"><a href="namespacennvm.html#af0d636002e2d607e2692cd0899c416b6">nnvm::FInputGraph</a></div><div class="ttdeci">std::function&lt; std::vector&lt; uint32_t &gt;(const NodeAttrs &amp;attrs)&gt; FInputGraph</div><div class="ttdoc">Get a list of inputs that represent graphs instead of data. Normally, input symbols are considered as...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:245</div></div>
<div class="ttc" id="namespacennvm_html_a475e6d8a1f6f0f63977e0b2e2a8b0ef7"><div class="ttname"><a href="namespacennvm.html#a475e6d8a1f6f0f63977e0b2e2a8b0ef7">nnvm::FNumVisibleOutputs</a></div><div class="ttdeci">std::function&lt; uint32_t(const NodeAttrs &amp;attrs)&gt; FNumVisibleOutputs</div><div class="ttdoc">Return number of visible outputs by the user. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:63</div></div>
<div class="ttc" id="namespacennvm_html_a2df0ffb0a105133c97d5fbc9d782c618"><div class="ttname"><a href="namespacennvm.html#a2df0ffb0a105133c97d5fbc9d782c618">nnvm::FCorrectLayout</a></div><div class="ttdeci">std::function&lt; bool(const NodeAttrs &amp;attrs, std::vector&lt; Layout &gt; *ilayouts, const std::vector&lt; Layout &gt; *last_ilayouts, std::vector&lt; Layout &gt; *olayouts)&gt; FCorrectLayout</div><div class="ttdoc">Infer &amp; correct function of node layout. See Layout for layout convention. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:233</div></div>
<div class="ttc" id="namespacennvm_html_ac317e6034c26f9cb453e91cd38e35df9"><div class="ttname"><a href="namespacennvm.html#ac317e6034c26f9cb453e91cd38e35df9">nnvm::TIsBackward</a></div><div class="ttdeci">bool TIsBackward</div><div class="ttdoc">Whether this op is an explicit backward operator, If TIsBackward is true: </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:136</div></div>
<div class="ttc" id="layout_8h_html"><div class="ttname"><a href="layout_8h.html">layout.h</a></div><div class="ttdoc">Layout expression. The layout is composed of upper cases, lower cases and numbers, where upper case indicates a (super-)dimension and the corresponding lower case with factor size indicates the split (sub-)dimension. For example, NCHW16c can describe a 5-D tensor of [batch_size, channel, height, width, channel_block]. Here sub-dimension channel_block=16 is the split of super-dimension C (channel). </div></div>
<div class="ttc" id="namespacennvm_html_aa7c5989aee30cef7813c484d12fc922d"><div class="ttname"><a href="namespacennvm.html#aa7c5989aee30cef7813c484d12fc922d">nnvm::FSetInputVarAttrOnCompose</a></div><div class="ttdeci">std::function&lt; void(const NodeAttrs &amp;attrs, ObjectPtr var, const int index)&gt; FSetInputVarAttrOnCompose</div><div class="ttdoc">Set the attributes of input variable. Usually used for setting initialization or weight decay...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:208</div></div>
<div class="ttc" id="namespacennvm_html_a573897937ff3b88f0014feabb8e44a92"><div class="ttname"><a href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">nnvm::ObjectPtr</a></div><div class="ttdeci">std::shared_ptr&lt; Node &gt; ObjectPtr</div><div class="ttdoc">we always used ObjectPtr for a reference pointer to the node, so this alias can be changed in case...</div><div class="ttdef"><b>Definition:</b> node.h:48</div></div>
<div class="ttc" id="3rdparty_2tvm_2nnvm_2include_2nnvm_2base_8h_html"><div class="ttname"><a href="3rdparty_2tvm_2nnvm_2include_2nnvm_2base_8h.html">base.h</a></div><div class="ttdoc">Configuration of nnvm as well as basic data structure. </div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->
<hr class="footer"/><address class="footer"><small>
Generated on Thu Jan 5 2023 00:58:42 for mxnet by &#160;<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/>
</a> 1.8.13
</small></address>
</body>
</html>