| <!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: /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.17 --> |
| <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_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> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <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> <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> <span class="comment"> * distributed with this work for additional information</span></div> |
| <div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <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> <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> <span class="comment"> * "License"); you may not use this file except in compliance</span></div> |
| <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <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> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <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> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <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> <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> <span class="comment"> * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span></div> |
| <div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <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> <span class="comment"> * specific language governing permissions and limitations</span></div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * under the License.</span></div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span>  </div> |
| <div class="line"><a name="l00024"></a><span class="lineno"> 24</span> <span class="preprocessor">#ifndef NNVM_OP_ATTR_TYPES_H_</span></div> |
| <div class="line"><a name="l00025"></a><span class="lineno"> 25</span> <span class="preprocessor">#define NNVM_OP_ATTR_TYPES_H_</span></div> |
| <div class="line"><a name="l00026"></a><span class="lineno"> 26</span>  </div> |
| <div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#include <functional></span></div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span> <span class="preprocessor">#include <string></span></div> |
| <div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="preprocessor">#include <unordered_map></span></div> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#include <utility></span></div> |
| <div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#include <vector></span></div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span>  </div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#include "<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2base_8h.html">base.h</a>"</span></div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include "<a class="code" href="layout_8h.html">layout.h</a>"</span></div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#include "<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2node_8h.html">node.h</a>"</span></div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#include "<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2tuple_8h.html">tuple.h</a>"</span></div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span>  </div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="keyword">namespace </span><a class="code" href="namespacennvm.html">nnvm</a> {</div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span>  </div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span> <span class="comment">// These types are optional attributes in each operator.</span></div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="comment">// Each attribute can be required by some passes.</span></div> |
| <div class="line"><a name="l00042"></a><span class="lineno"> 42</span>  </div> |
| <div class="line"><a name="l00052"></a><span class="lineno"><a class="line" href="namespacennvm.html#a60262824f44e38531e7697eb1b3f4cc6"> 52</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a60262824f44e38531e7697eb1b3f4cc6">FListInputNames</a> = std::function<std::vector<std::string>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00053"></a><span class="lineno"> 53</span>  </div> |
| <div class="line"><a name="l00064"></a><span class="lineno"><a class="line" href="namespacennvm.html#a87bd9af947545be600c74c00a142c7fe"> 64</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a87bd9af947545be600c74c00a142c7fe">FNumVisibleOutputs</a> = std::function<uint32_t(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00065"></a><span class="lineno"> 65</span>  </div> |
| <div class="line"><a name="l00075"></a><span class="lineno"><a class="line" href="namespacennvm.html#a80d346f5583090eaf293fffc6028ff60"> 75</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a80d346f5583090eaf293fffc6028ff60">FListOutputNames</a> = std::function<std::vector<std::string>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  </div> |
| <div class="line"><a name="l00085"></a><span class="lineno"><a class="line" href="namespacennvm.html#a9a8f8438feb93e69317a79594a7873b9"> 85</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a9a8f8438feb93e69317a79594a7873b9">FMutateInputs</a> = std::function<std::vector<uint32_t>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  </div> |
| <div class="line"><a name="l00092"></a><span class="lineno"> 92</span> <span class="keyword">template</span> <<span class="keyword">typename</span> AttrType></div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a4e51cf5506d5c8a9a70be11fa6a9d1e1">FInferNodeEntryAttr</a> = std::function<bool(</div> |
| <div class="line"><a name="l00094"></a><span class="lineno"><a class="line" href="namespacennvm.html#a4e51cf5506d5c8a9a70be11fa6a9d1e1"> 94</a></span>  <span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs, std::vector<AttrType>* in_attrs, std::vector<AttrType>* out_attrs)>;</div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span>  </div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#abaa4e70445e45a1fd816d16f9633a68b">FGetAttrDict</a> =</div> |
| <div class="line"><a name="l00104"></a><span class="lineno"><a class="line" href="namespacennvm.html#abaa4e70445e45a1fd816d16f9633a68b"> 104</a></span>  std::function<std::unordered_map<std::string, std::string>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  </div> |
| <div class="line"><a name="l00116"></a><span class="lineno"><a class="line" href="namespacennvm.html#ae80cd6cc2432e36350ad84335d5e5308"> 116</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#ae80cd6cc2432e36350ad84335d5e5308">FInferShape</a> = <a class="code" href="namespacennvm.html#a4e51cf5506d5c8a9a70be11fa6a9d1e1">FInferNodeEntryAttr<TShape></a>;</div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  </div> |
| <div class="line"><a name="l00125"></a><span class="lineno"><a class="line" href="namespacennvm.html#a6539f5c3067c4c3cd7dbbdd6b6a6f165"> 125</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a6539f5c3067c4c3cd7dbbdd6b6a6f165">FInferType</a> = <a class="code" href="namespacennvm.html#a4e51cf5506d5c8a9a70be11fa6a9d1e1">FInferNodeEntryAttr<int></a>;</div> |
| <div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  </div> |
| <div class="line"><a name="l00135"></a><span class="lineno"><a class="line" href="namespacennvm.html#ac317e6034c26f9cb453e91cd38e35df9"> 135</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#ac317e6034c26f9cb453e91cd38e35df9">TIsBackward</a> = bool;</div> |
| <div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  </div> |
| <div class="line"><a name="l00146"></a><span class="lineno"><a class="line" href="namespacennvm.html#ad27d6c45aee15d0dab91419c63f8f900"> 146</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#ad27d6c45aee15d0dab91419c63f8f900">TIsGhost</a> = bool;</div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  </div> |
| <div class="line"><a name="l00157"></a><span class="lineno"><a class="line" href="namespacennvm.html#abf4dd9163262b22cdac032714207e533"> 157</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#abf4dd9163262b22cdac032714207e533">FInplaceOption</a> = std::function<std::vector<std::pair<int, int> >(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00158"></a><span class="lineno"> 158</span>  </div> |
| <div class="line"><a name="l00169"></a><span class="lineno"><a class="line" href="namespacennvm.html#a802f95398ea93b2bec4d239ce5b86b00"> 169</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a802f95398ea93b2bec4d239ce5b86b00">FInplaceIdentity</a> = std::function<std::vector<bool>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  </div> |
| <div class="line"><a name="l00180"></a><span class="lineno"><a class="line" href="namespacennvm.html#af328934d094e7ff06f96b103350175f4"> 180</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#af328934d094e7ff06f96b103350175f4">FIgnoreInputs</a> = std::function<std::vector<uint32_t>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  </div> |
| <div class="line"><a name="l00191"></a><span class="lineno"> 191</span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a1a68754797b1a4db29122527592915ff">FGradient</a> = std::function<std::vector<NodeEntry>(<span class="keyword">const</span> <a class="code" href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">ObjectPtr</a>& nodeptr,</div> |
| <div class="line"><a name="l00192"></a><span class="lineno"><a class="line" href="namespacennvm.html#a1a68754797b1a4db29122527592915ff"> 192</a></span>  <span class="keyword">const</span> std::vector<NodeEntry>& out_grads)>;</div> |
| <div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  </div> |
| <div class="line"><a name="l00201"></a><span class="lineno"> 201</span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a90a58a8b530e412ea5d6f055fa760527">FSetInputVarAttrOnCompose</a> =</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"><a class="line" href="namespacennvm.html#a90a58a8b530e412ea5d6f055fa760527"> 202</a></span>  std::function<void(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs, <a class="code" href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">ObjectPtr</a> var, <span class="keyword">const</span> <span class="keywordtype">int</span> index)>;</div> |
| <div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  </div> |
| <div class="line"><a name="l00223"></a><span class="lineno"> 223</span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#a19b81be31144872d26056290092763a0">FCorrectLayout</a> =</div> |
| <div class="line"><a name="l00224"></a><span class="lineno"> 224</span>  std::function<bool(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs, std::vector<Layout>* ilayouts,</div> |
| <div class="line"><a name="l00225"></a><span class="lineno"><a class="line" href="namespacennvm.html#a19b81be31144872d26056290092763a0"> 225</a></span>  <span class="keyword">const</span> std::vector<Layout>* last_ilayouts, std::vector<Layout>* olayouts)>;</div> |
| <div class="line"><a name="l00226"></a><span class="lineno"> 226</span>  </div> |
| <div class="line"><a name="l00237"></a><span class="lineno"><a class="line" href="namespacennvm.html#af0d636002e2d607e2692cd0899c416b6"> 237</a></span> <span class="keyword">using</span> <a class="code" href="namespacennvm.html#af0d636002e2d607e2692cd0899c416b6">FInputGraph</a> = std::function<std::vector<uint32_t>(<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div> |
| <div class="line"><a name="l00238"></a><span class="lineno"> 238</span>  </div> |
| <div class="line"><a name="l00239"></a><span class="lineno"> 239</span> } <span class="comment">// namespace nnvm</span></div> |
| <div class="line"><a name="l00240"></a><span class="lineno"> 240</span>  </div> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span> <span class="preprocessor">#endif // NNVM_OP_ATTR_TYPES_H_</span></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <div class="ttc" id="anamespacennvm_html_a90a58a8b530e412ea5d6f055fa760527"><div class="ttname"><a href="namespacennvm.html#a90a58a8b530e412ea5d6f055fa760527">nnvm::FSetInputVarAttrOnCompose</a></div><div class="ttdeci">std::function< void(const NodeAttrs &attrs, ObjectPtr var, const int index)> 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:202</div></div> |
| <div class="ttc" id="anamespacennvm_html_a87bd9af947545be600c74c00a142c7fe"><div class="ttname"><a href="namespacennvm.html#a87bd9af947545be600c74c00a142c7fe">nnvm::FNumVisibleOutputs</a></div><div class="ttdeci">std::function< uint32_t(const NodeAttrs &attrs)> FNumVisibleOutputs</div><div class="ttdoc">Return number of visible outputs by the user.</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:64</div></div> |
| <div class="ttc" id="anamespacennvm_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:135</div></div> |
| <div class="ttc" id="anamespacennvm_html_ae80cd6cc2432e36350ad84335d5e5308"><div class="ttname"><a href="namespacennvm.html#ae80cd6cc2432e36350ad84335d5e5308">nnvm::FInferShape</a></div><div class="ttdeci">FInferNodeEntryAttr< TShape > FInferShape</div><div class="ttdoc">Shape inference function. Update the shapes given the input shape information. TShape....</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:116</div></div> |
| <div class="ttc" id="anamespacennvm_html_a1a68754797b1a4db29122527592915ff"><div class="ttname"><a href="namespacennvm.html#a1a68754797b1a4db29122527592915ff">nnvm::FGradient</a></div><div class="ttdeci">std::function< std::vector< NodeEntry >(const ObjectPtr &nodeptr, const std::vector< NodeEntry > &out_grads)> 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:192</div></div> |
| <div class="ttc" id="a3rdparty_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="anamespacennvm_html_a80d346f5583090eaf293fffc6028ff60"><div class="ttname"><a href="namespacennvm.html#a80d346f5583090eaf293fffc6028ff60">nnvm::FListOutputNames</a></div><div class="ttdeci">std::function< std::vector< std::string >(const NodeAttrs &attrs)> 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:75</div></div> |
| <div class="ttc" id="a3rdparty_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 class="ttc" id="anamespacennvm_html_a9a8f8438feb93e69317a79594a7873b9"><div class="ttname"><a href="namespacennvm.html#a9a8f8438feb93e69317a79594a7873b9">nnvm::FMutateInputs</a></div><div class="ttdeci">std::function< std::vector< uint32_t >(const NodeAttrs &attrs)> FMutateInputs</div><div class="ttdoc">Check whether operator will mutate k-th input.</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:85</div></div> |
| <div class="ttc" id="anamespacennvm_html_af328934d094e7ff06f96b103350175f4"><div class="ttname"><a href="namespacennvm.html#af328934d094e7ff06f96b103350175f4">nnvm::FIgnoreInputs</a></div><div class="ttdeci">std::function< std::vector< uint32_t >(const NodeAttrs &attrs)> 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:180</div></div> |
| <div class="ttc" id="anamespacennvm_html_a802f95398ea93b2bec4d239ce5b86b00"><div class="ttname"><a href="namespacennvm.html#a802f95398ea93b2bec4d239ce5b86b00">nnvm::FInplaceIdentity</a></div><div class="ttdeci">std::function< std::vector< bool >(const NodeAttrs &attrs)> 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:169</div></div> |
| <div class="ttc" id="astructnnvm_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:107</div></div> |
| <div class="ttc" id="anamespacennvm_html_a573897937ff3b88f0014feabb8e44a92"><div class="ttname"><a href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">nnvm::ObjectPtr</a></div><div class="ttdeci">std::shared_ptr< Node > 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:49</div></div> |
| <div class="ttc" id="anamespacennvm_html_a60262824f44e38531e7697eb1b3f4cc6"><div class="ttname"><a href="namespacennvm.html#a60262824f44e38531e7697eb1b3f4cc6">nnvm::FListInputNames</a></div><div class="ttdeci">std::function< std::vector< std::string >(const NodeAttrs &attrs)> 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:52</div></div> |
| <div class="ttc" id="anamespacennvm_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:146</div></div> |
| <div class="ttc" id="anamespacennvm_html_af0d636002e2d607e2692cd0899c416b6"><div class="ttname"><a href="namespacennvm.html#af0d636002e2d607e2692cd0899c416b6">nnvm::FInputGraph</a></div><div class="ttdeci">std::function< std::vector< uint32_t >(const NodeAttrs &attrs)> 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:237</div></div> |
| <div class="ttc" id="anamespacennvm_html_abf4dd9163262b22cdac032714207e533"><div class="ttname"><a href="namespacennvm.html#abf4dd9163262b22cdac032714207e533">nnvm::FInplaceOption</a></div><div class="ttdeci">std::function< std::vector< std::pair< int, int > >(const NodeAttrs &attrs)> 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:157</div></div> |
| <div class="ttc" id="alayout_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,...</div></div> |
| <div class="ttc" id="anamespacennvm_html_abaa4e70445e45a1fd816d16f9633a68b"><div class="ttname"><a href="namespacennvm.html#abaa4e70445e45a1fd816d16f9633a68b">nnvm::FGetAttrDict</a></div><div class="ttdeci">std::function< std::unordered_map< std::string, std::string >(const NodeAttrs &attrs)> FGetAttrDict</div><div class="ttdoc">Get attribute dictionary from node.</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:104</div></div> |
| <div class="ttc" id="anamespacennvm_html_a6539f5c3067c4c3cd7dbbdd6b6a6f165"><div class="ttname"><a href="namespacennvm.html#a6539f5c3067c4c3cd7dbbdd6b6a6f165">nnvm::FInferType</a></div><div class="ttdeci">FInferNodeEntryAttr< int > 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:125</div></div> |
| <div class="ttc" id="a3rdparty_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="anamespacennvm_html_a4e51cf5506d5c8a9a70be11fa6a9d1e1"><div class="ttname"><a href="namespacennvm.html#a4e51cf5506d5c8a9a70be11fa6a9d1e1">nnvm::FInferNodeEntryAttr</a></div><div class="ttdeci">std::function< bool(const NodeAttrs &attrs, std::vector< AttrType > *in_attrs, std::vector< AttrType > *out_attrs)> 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="anamespacennvm_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="anamespacennvm_html_a19b81be31144872d26056290092763a0"><div class="ttname"><a href="namespacennvm.html#a19b81be31144872d26056290092763a0">nnvm::FCorrectLayout</a></div><div class="ttdeci">std::function< bool(const NodeAttrs &attrs, std::vector< Layout > *ilayouts, const std::vector< Layout > *last_ilayouts, std::vector< Layout > *olayouts)> FCorrectLayout</div><div class="ttdoc">Infer & correct function of node layout. See Layout for layout convention.</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:225</div></div> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Sat Nov 5 2022 01:16:57 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.17 |
| </small></address> |
| </body> |
| </html> |