| <!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.6"/> |
| <title>mxnet: include/mxnet/operator.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/search.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(function() { searchBox.OnSelectItem(0); }); |
| </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 style="padding-left: 0.5em;"> |
| <div id="projectname">mxnet |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.6 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <div id="navrow1" class="tabs"> |
| <ul class="tablist"> |
| <li><a href="index.html"><span>Main Page</span></a></li> |
| <li><a href="namespaces.html"><span>Namespaces</span></a></li> |
| <li><a href="annotated.html"><span>Classes</span></a></li> |
| <li class="current"><a href="files.html"><span>Files</span></a></li> |
| <li> |
| <div id="MSearchBox" class="MSearchBoxInactive"> |
| <span class="left"> |
| <img id="MSearchSelect" src="search/mag_sel.png" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| alt=""/> |
| <input type="text" id="MSearchField" value="Search" accesskey="S" |
| onfocus="searchBox.OnSearchFieldFocus(true)" |
| onblur="searchBox.OnSearchFieldFocus(false)" |
| onkeyup="searchBox.OnSearchFieldChange(event)"/> |
| </span><span class="right"> |
| <a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a> |
| </span> |
| </div> |
| </li> |
| </ul> |
| </div> |
| <div id="navrow2" class="tabs2"> |
| <ul class="tablist"> |
| <li><a href="files.html"><span>File List</span></a></li> |
| <li><a href="globals.html"><span>File Members</span></a></li> |
| </ul> |
| </div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| <a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(0)"><span class="SelectionMark"> </span>All</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(1)"><span class="SelectionMark"> </span>Classes</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(2)"><span class="SelectionMark"> </span>Namespaces</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(3)"><span class="SelectionMark"> </span>Files</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(4)"><span class="SelectionMark"> </span>Functions</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(5)"><span class="SelectionMark"> </span>Variables</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(6)"><span class="SelectionMark"> </span>Typedefs</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(7)"><span class="SelectionMark"> </span>Enumerations</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(8)"><span class="SelectionMark"> </span>Enumerator</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(9)"><span class="SelectionMark"> </span>Friends</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(10)"><span class="SelectionMark"> </span>Macros</a></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_d44c64559bbebec7f509842c48db8b23.html">include</a></li><li class="navelem"><a class="el" href="dir_1143c7affb9ebd026cb6818dd282def7.html">mxnet</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">operator.h</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="operator_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> </div> |
| <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="preprocessor">#ifndef MXNET_OPERATOR_H_</span></div> |
| <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="preprocessor"></span><span class="preprocessor">#define MXNET_OPERATOR_H_</span></div> |
| <div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="preprocessor"></span></div> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="preprocessor">#include <dmlc/base.h></span></div> |
| <div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="preprocessor">#include <dmlc/json.h></span></div> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="preprocessor">#include <dmlc/logging.h></span></div> |
| <div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="preprocessor">#include <dmlc/registry.h></span></div> |
| <div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="preprocessor">#include <nnvm/node.h></span></div> |
| <div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="preprocessor">#include <vector></span></div> |
| <div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="preprocessor">#include <map></span></div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="preprocessor">#include <string></span></div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="preprocessor">#include <utility></span></div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span> <span class="preprocessor">#include "<a class="code" href="base_8h.html">./base.h</a>"</span></div> |
| <div class="line"><a name="l00020"></a><span class="lineno"> 20</span> <span class="preprocessor">#include "<a class="code" href="resource_8h.html">./resource.h</a>"</span></div> |
| <div class="line"><a name="l00021"></a><span class="lineno"> 21</span> </div> |
| <div class="line"><a name="l00022"></a><span class="lineno"> 22</span> <span class="keyword">namespace </span>mxnet {</div> |
| <div class="line"><a name="l00024"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398"> 24</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">OpReqType</a> {</div> |
| <div class="line"><a name="l00026"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26"> 26</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26">kNullOp</a>,</div> |
| <div class="line"><a name="l00028"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a"> 28</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a">kWriteTo</a>,</div> |
| <div class="line"><a name="l00034"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51"> 34</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51">kWriteInplace</a>,</div> |
| <div class="line"><a name="l00036"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383"> 36</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383">kAddTo</a></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> </div> |
| <div class="line"><a name="l00045"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html"> 45</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1OpContext.html">OpContext</a> {</div> |
| <div class="line"><a name="l00047"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#abd7405c065167e4e15af032b1e0a5c42"> 47</a></span>  <span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1OpContext.html#abd7405c065167e4e15af032b1e0a5c42">is_train</a>;</div> |
| <div class="line"><a name="l00049"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf"> 49</a></span>  <a class="code" href="structmxnet_1_1RunContext.html">RunContext</a> <a class="code" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf">run_ctx</a>;</div> |
| <div class="line"><a name="l00051"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef"> 51</a></span>  <a class="code" href="classmxnet_1_1engine_1_1CallbackOnComplete.html">engine::CallbackOnComplete</a> <a class="code" href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef">async_on_complete</a>;</div> |
| <div class="line"><a name="l00053"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b"> 53</a></span>  std::vector<Resource> <a class="code" href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b">requested</a>;</div> |
| <div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keyword">template</span><<span class="keyword">typename</span> xpu></div> |
| <div class="line"><a name="l00060"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#aa700c5c28e024b42d41e15f46615f1b8"> 60</a></span>  <span class="keyword">inline</span> mshadow::Stream<xpu>* <a class="code" href="structmxnet_1_1OpContext.html#aa700c5c28e024b42d41e15f46615f1b8">get_stream</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  <span class="keywordflow">return</span> <a class="code" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf">run_ctx</a>.<a class="code" href="structmxnet_1_1RunContext.html#a8bcd2fc6392f95a1fdc3d98f74dbfcb9">get_stream</a><xpu>();</div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  }</div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span> };</div> |
| <div class="line"><a name="l00064"></a><span class="lineno"> 64</span> </div> |
| <div class="line"><a name="l00077"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html"> 77</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1Operator.html">Operator</a> {</div> |
| <div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00080"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17a"> 80</a></span>  <span class="keyword">enum</span> <a class="code" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17a">ExecType</a> {</div> |
| <div class="line"><a name="l00082"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa8adac8fd6acb05ff169a2913ad05122c"> 82</a></span>  <a class="code" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa8adac8fd6acb05ff169a2913ad05122c">kSync</a>,</div> |
| <div class="line"><a name="l00087"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa6c403d828787dc2aafd6ee1e9cd02699"> 87</a></span>  <a class="code" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa6c403d828787dc2aafd6ee1e9cd02699">kAsync</a>,</div> |
| <div class="line"><a name="l00094"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aaebe5a0e5db6ee71f6905f9b34a1f05ad"> 94</a></span>  <a class="code" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aaebe5a0e5db6ee71f6905f9b34a1f05ad">kCrossDeviceCopy</a></div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span>  };</div> |
| <div class="line"><a name="l00097"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a4d34cc792d7271bd8b5b60eaa0b55767"> 97</a></span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1Operator.html#a4d34cc792d7271bd8b5b60eaa0b55767">~Operator</a>() {}</div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1Operator.html#a3dbe23d985df86da51eeb868439872dd">Forward</a>(<span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a> &ctx,</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  <span class="keyword">const</span> std::vector<TBlob> &in_data,</div> |
| <div class="line"><a name="l00111"></a><span class="lineno"> 111</span>  <span class="keyword">const</span> std::vector<OpReqType> &req,</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  <span class="keyword">const</span> std::vector<TBlob> &out_data,</div> |
| <div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  <span class="keyword">const</span> std::vector<TBlob> &aux_states) = 0;</div> |
| <div class="line"><a name="l00142"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#ad32d74b9bf56caf920b5e6379d2bb259"> 142</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1Operator.html#ad32d74b9bf56caf920b5e6379d2bb259">Backward</a>(<span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a> &ctx,</div> |
| <div class="line"><a name="l00143"></a><span class="lineno"> 143</span>  <span class="keyword">const</span> std::vector<TBlob> &out_grad,</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keyword">const</span> std::vector<TBlob> &in_data,</div> |
| <div class="line"><a name="l00145"></a><span class="lineno"> 145</span>  <span class="keyword">const</span> std::vector<TBlob> &out_data,</div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  <span class="keyword">const</span> std::vector<OpReqType> &req,</div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  <span class="keyword">const</span> std::vector<TBlob> &in_grad,</div> |
| <div class="line"><a name="l00148"></a><span class="lineno"> 148</span>  <span class="keyword">const</span> std::vector<TBlob> &aux_states) {</div> |
| <div class="line"><a name="l00149"></a><span class="lineno"> 149</span>  LOG(FATAL) << <span class="stringliteral">"Backward is not implemented"</span>;</div> |
| <div class="line"><a name="l00150"></a><span class="lineno"> 150</span>  }</div> |
| <div class="line"><a name="l00152"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a7bffb2be177edf8c29fd0148ebbd7ca5"> 152</a></span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17a">ExecType</a> <a class="code" href="classmxnet_1_1Operator.html#a7bffb2be177edf8c29fd0148ebbd7ca5">exec_type</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00153"></a><span class="lineno"> 153</span>  <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa8adac8fd6acb05ff169a2913ad05122c">kSync</a>;</div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  }</div> |
| <div class="line"><a name="l00155"></a><span class="lineno"> 155</span> };</div> |
| <div class="line"><a name="l00156"></a><span class="lineno"> 156</span> </div> |
| <div class="line"><a name="l00157"></a><span class="lineno"> 157</span> <span class="preprocessor">#if DMLC_USE_CXX11</span></div> |
| <div class="line"><a name="l00158"></a><span class="lineno"> 158</span> <span class="preprocessor"></span><span class="comment">// OperatorProperty allows C++11, while Operator do not rely on it.</span></div> |
| <div class="line"><a name="l00166"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html"> 166</a></span> <span class="comment"></span><span class="keyword">class </span><a class="code" href="classmxnet_1_1OperatorProperty.html">OperatorProperty</a> {</div> |
| <div class="line"><a name="l00167"></a><span class="lineno"> 167</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00171"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a610af8e9cb560104301ccd1efa2a96fd"> 171</a></span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a610af8e9cb560104301ccd1efa2a96fd">~OperatorProperty</a>() {}</div> |
| <div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a0fb32e0ab03a86517473db0f0630856c">Init</a>(<span class="keyword">const</span> std::vector<std::pair<std::string, std::string> >& kwargs) = 0;</div> |
| <div class="line"><a name="l00182"></a><span class="lineno"> 182</span>  <span class="keyword">virtual</span> std::map<std::string, std::string> <a class="code" href="classmxnet_1_1OperatorProperty.html#a704ae43181c8a88bfd0a47e8005f7cb8">GetParams</a>() <span class="keyword">const</span> = 0;</div> |
| <div class="line"><a name="l00187"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a24cc6a3763ff5f0881a2de0652810596"> 187</a></span>  <span class="keyword">virtual</span> std::vector<std::string> <a class="code" href="classmxnet_1_1OperatorProperty.html#a24cc6a3763ff5f0881a2de0652810596">ListArguments</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  <span class="keywordflow">return</span> {<span class="stringliteral">"data"</span>};</div> |
| <div class="line"><a name="l00189"></a><span class="lineno"> 189</span>  }</div> |
| <div class="line"><a name="l00194"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892"> 194</a></span>  <span class="keyword">virtual</span> std::vector<std::string> <a class="code" href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892">ListOutputs</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  <span class="keywordflow">return</span> {<span class="stringliteral">"output"</span>};</div> |
| <div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  }</div> |
| <div class="line"><a name="l00201"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a0e20dead67dad0cd76c147a997ef5094"> 201</a></span>  <span class="keyword">virtual</span> std::vector<std::string> <a class="code" href="classmxnet_1_1OperatorProperty.html#a0e20dead67dad0cd76c147a997ef5094">ListAuxiliaryStates</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  <span class="keywordflow">return</span> {};</div> |
| <div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  }</div> |
| <div class="line"><a name="l00205"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a49e25d85c38199338a9fa8150bc86df1"> 205</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a49e25d85c38199338a9fa8150bc86df1">NumOutputs</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00206"></a><span class="lineno"> 206</span>  <span class="keywordflow">return</span> this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892">ListOutputs</a>().size();</div> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  }</div> |
| <div class="line"><a name="l00220"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a6dc5ecd089ef01dcc154c02b20ad2f4b"> 220</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a6dc5ecd089ef01dcc154c02b20ad2f4b">NumVisibleOutputs</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00221"></a><span class="lineno"> 221</span>  <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a49e25d85c38199338a9fa8150bc86df1">NumOutputs</a>();</div> |
| <div class="line"><a name="l00222"></a><span class="lineno"> 222</span>  }</div> |
| <div class="line"><a name="l00240"></a><span class="lineno"> 240</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a543cbf36f0e4d8dc99109db71705a11c">InferShape</a>(std::vector<TShape> *in_shape,</div> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  std::vector<TShape> *out_shape,</div> |
| <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  std::vector<TShape> *aux_shape) <span class="keyword">const</span> = 0;</div> |
| <div class="line"><a name="l00260"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#ad5c9f53ea70f3eeb1bf27b983f420c0b"> 260</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#ad5c9f53ea70f3eeb1bf27b983f420c0b">InferType</a>(std::vector<int> *in_type,</div> |
| <div class="line"><a name="l00261"></a><span class="lineno"> 261</span>  std::vector<int> *out_type,</div> |
| <div class="line"><a name="l00262"></a><span class="lineno"> 262</span>  std::vector<int> *aux_type)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00263"></a><span class="lineno"> 263</span>  CHECK_LE(in_type->size(), this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a24cc6a3763ff5f0881a2de0652810596">ListArguments</a>().size());</div> |
| <div class="line"><a name="l00264"></a><span class="lineno"> 264</span>  <span class="keywordtype">int</span> n_in = this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a24cc6a3763ff5f0881a2de0652810596">ListArguments</a>().size();</div> |
| <div class="line"><a name="l00265"></a><span class="lineno"> 265</span>  <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> i = 0; i < in_type->size(); ++i) {</div> |
| <div class="line"><a name="l00266"></a><span class="lineno"> 266</span>  CHECK(in_type->at(i) == mshadow::default_type_flag ||</div> |
| <div class="line"><a name="l00267"></a><span class="lineno"> 267</span>  in_type->at(i) == -1) << <span class="stringliteral">"Unsupported data type "</span> << in_type->at(i);</div> |
| <div class="line"><a name="l00268"></a><span class="lineno"> 268</span>  }</div> |
| <div class="line"><a name="l00269"></a><span class="lineno"> 269</span>  in_type->clear();</div> |
| <div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < n_in; ++i ) in_type->push_back(mshadow::default_type_flag);</div> |
| <div class="line"><a name="l00271"></a><span class="lineno"> 271</span> </div> |
| <div class="line"><a name="l00272"></a><span class="lineno"> 272</span>  <span class="keywordtype">int</span> n_out = this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892">ListOutputs</a>().size();</div> |
| <div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  out_type->clear();</div> |
| <div class="line"><a name="l00274"></a><span class="lineno"> 274</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < n_out; ++i ) out_type->push_back(mshadow::default_type_flag);</div> |
| <div class="line"><a name="l00275"></a><span class="lineno"> 275</span> </div> |
| <div class="line"><a name="l00276"></a><span class="lineno"> 276</span>  <span class="keywordtype">int</span> n_aux = this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a0e20dead67dad0cd76c147a997ef5094">ListAuxiliaryStates</a>().size();</div> |
| <div class="line"><a name="l00277"></a><span class="lineno"> 277</span>  aux_type->clear();</div> |
| <div class="line"><a name="l00278"></a><span class="lineno"> 278</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < n_aux; ++i ) aux_type->push_back(mshadow::default_type_flag);</div> |
| <div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00280"></a><span class="lineno"> 280</span>  }</div> |
| <div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1OperatorProperty.html">OperatorProperty</a>* <a class="code" href="classmxnet_1_1OperatorProperty.html#ab38f2f877555ba34b548efd2575cac99">Copy</a>() <span class="keyword">const</span> = 0;</div> |
| <div class="line"><a name="l00289"></a><span class="lineno"> 289</span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1Operator.html">Operator</a>* <a class="code" href="classmxnet_1_1OperatorProperty.html#a546ab25ddaae590ca98705ae9abc7ac0">CreateOperator</a>(<a class="code" href="structmxnet_1_1Context.html">Context</a> ctx) <span class="keyword">const</span> = 0;</div> |
| <div class="line"><a name="l00297"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a7985004f99a931b4a6ff69df733bae72"> 297</a></span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1Operator.html">Operator</a>* <a class="code" href="classmxnet_1_1OperatorProperty.html#a7985004f99a931b4a6ff69df733bae72">CreateOperatorEx</a>(<a class="code" href="structmxnet_1_1Context.html">Context</a> ctx, std::vector<TShape> *in_shape,</div> |
| <div class="line"><a name="l00298"></a><span class="lineno"> 298</span>  std::vector<int> *in_type)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  std::vector<int> out_type, aux_type;</div> |
| <div class="line"><a name="l00300"></a><span class="lineno"> 300</span>  std::vector<TShape> out_shape, aux_shape;</div> |
| <div class="line"><a name="l00301"></a><span class="lineno"> 301</span>  out_type.resize(this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892">ListOutputs</a>().size());</div> |
| <div class="line"><a name="l00302"></a><span class="lineno"> 302</span>  out_shape.resize(this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892">ListOutputs</a>().size());</div> |
| <div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  aux_type.resize(this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a0e20dead67dad0cd76c147a997ef5094">ListAuxiliaryStates</a>().size());</div> |
| <div class="line"><a name="l00304"></a><span class="lineno"> 304</span>  aux_shape.resize(this-><a class="code" href="classmxnet_1_1OperatorProperty.html#a0e20dead67dad0cd76c147a997ef5094">ListAuxiliaryStates</a>().size());</div> |
| <div class="line"><a name="l00305"></a><span class="lineno"> 305</span>  CHECK(<a class="code" href="classmxnet_1_1OperatorProperty.html#ad5c9f53ea70f3eeb1bf27b983f420c0b">InferType</a>(in_type, &out_type, &aux_type));</div> |
| <div class="line"><a name="l00306"></a><span class="lineno"> 306</span>  CHECK(<a class="code" href="classmxnet_1_1OperatorProperty.html#a543cbf36f0e4d8dc99109db71705a11c">InferShape</a>(in_shape, &out_shape, &aux_shape));</div> |
| <div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a546ab25ddaae590ca98705ae9abc7ac0">CreateOperator</a>(ctx);</div> |
| <div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  }</div> |
| <div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  <span class="keyword">virtual</span> std::string <a class="code" href="classmxnet_1_1OperatorProperty.html#ac6b7df90c0285a1823b463c21e926698">TypeString</a>() <span class="keyword">const</span> = 0;</div> |
| <div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  <span class="comment">//--------------------------------------------------------</span></div> |
| <div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  <span class="comment">// All the below functions are optional to override.</span></div> |
| <div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  <span class="comment">//--------------------------------------------------------</span></div> |
| <div class="line"><a name="l00325"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a623210dd1b9495ca73f223e0b06f998c"> 325</a></span> <span class="comment"></span> <span class="keyword">virtual</span> std::vector<ResourceRequest> <a class="code" href="classmxnet_1_1OperatorProperty.html#a623210dd1b9495ca73f223e0b06f998c">ForwardResource</a>(</div> |
| <div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  <span class="keyword">const</span> std::vector<TShape> &in_shape)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00327"></a><span class="lineno"> 327</span>  <span class="keywordflow">return</span> std::vector<ResourceRequest>();</div> |
| <div class="line"><a name="l00328"></a><span class="lineno"> 328</span>  }</div> |
| <div class="line"><a name="l00336"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#ade7720a3e0ebba67d89978925a77609b"> 336</a></span>  <span class="keyword">virtual</span> std::vector<ResourceRequest> <a class="code" href="classmxnet_1_1OperatorProperty.html#ade7720a3e0ebba67d89978925a77609b">BackwardResource</a>(</div> |
| <div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  <span class="keyword">const</span> std::vector<TShape> &in_shape)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00338"></a><span class="lineno"> 338</span>  <span class="keywordflow">return</span> std::vector<ResourceRequest>();</div> |
| <div class="line"><a name="l00339"></a><span class="lineno"> 339</span>  }</div> |
| <div class="line"><a name="l00362"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#abf9e6a8d40750f3ee81fe30cbe3e2aae"> 362</a></span>  <span class="keyword">virtual</span> std::vector<int> <a class="code" href="classmxnet_1_1OperatorProperty.html#abf9e6a8d40750f3ee81fe30cbe3e2aae">DeclareBackwardDependency</a>(</div> |
| <div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  <span class="keyword">const</span> std::vector<int> &out_grad,</div> |
| <div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  <span class="keyword">const</span> std::vector<int> &in_data,</div> |
| <div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  <span class="keyword">const</span> std::vector<int> &out_data)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  <span class="comment">// By default requires to see all the things.</span></div> |
| <div class="line"><a name="l00367"></a><span class="lineno"> 367</span>  <span class="comment">// remember to override this function to get a better performance.</span></div> |
| <div class="line"><a name="l00368"></a><span class="lineno"> 368</span>  std::vector<int> ret = out_grad;</div> |
| <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  ret.insert(ret.end(), in_data.begin(), in_data.end());</div> |
| <div class="line"><a name="l00370"></a><span class="lineno"> 370</span>  ret.insert(ret.end(), out_data.begin(), out_data.end());</div> |
| <div class="line"><a name="l00371"></a><span class="lineno"> 371</span>  <span class="keywordflow">return</span> ret;</div> |
| <div class="line"><a name="l00372"></a><span class="lineno"> 372</span>  }</div> |
| <div class="line"><a name="l00394"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#aad7a3e87866515a2b7100e849a2fe966"> 394</a></span>  <span class="keyword">virtual</span> std::vector<std::pair<int, void*> > <a class="code" href="classmxnet_1_1OperatorProperty.html#aad7a3e87866515a2b7100e849a2fe966">ForwardInplaceOption</a>(</div> |
| <div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  <span class="keyword">const</span> std::vector<int> &in_data,</div> |
| <div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  <span class="keyword">const</span> std::vector<void*> &out_data)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  <span class="keywordflow">return</span> std::vector<std::pair<int, void*> >();</div> |
| <div class="line"><a name="l00398"></a><span class="lineno"> 398</span>  }</div> |
| <div class="line"><a name="l00425"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a67231a5a1525ec7480de98a79dce0194"> 425</a></span>  <span class="keyword">virtual</span> std::vector<std::pair<int, void*> > <a class="code" href="classmxnet_1_1OperatorProperty.html#a67231a5a1525ec7480de98a79dce0194">BackwardInplaceOption</a>(</div> |
| <div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  <span class="keyword">const</span> std::vector<int> &out_grad,</div> |
| <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  <span class="keyword">const</span> std::vector<int> &in_data,</div> |
| <div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="keyword">const</span> std::vector<int> &out_data,</div> |
| <div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  <span class="keyword">const</span> std::vector<void*> &in_grad)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00430"></a><span class="lineno"> 430</span>  <span class="keywordflow">return</span> std::vector<std::pair<int, void*> >();</div> |
| <div class="line"><a name="l00431"></a><span class="lineno"> 431</span>  }</div> |
| <div class="line"><a name="l00444"></a><span class="lineno"> 444</span>  <span class="keyword">template</span><<span class="keyword">typename</span> T></div> |
| <div class="line"><a name="l00445"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a9f488509e132094ce7e79125ce853042"> 445</a></span>  <span class="keyword">inline</span> std::vector<T> <a class="code" href="classmxnet_1_1OperatorProperty.html#a9f488509e132094ce7e79125ce853042">BackwardInputs</a>(<span class="keyword">const</span> std::vector<T> &out_grad,</div> |
| <div class="line"><a name="l00446"></a><span class="lineno"> 446</span>  <span class="keyword">const</span> std::vector<T> &in_data,</div> |
| <div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  <span class="keyword">const</span> std::vector<T> &out_data)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00448"></a><span class="lineno"> 448</span>  <span class="keywordtype">int</span> counter = 0;</div> |
| <div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  std::vector<int> out_grad_index(out_grad.size());</div> |
| <div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  std::vector<int> in_data_index(in_data.size());</div> |
| <div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  std::vector<int> out_data_index(out_data.size());</div> |
| <div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i < out_grad_index.size(); ++i) {</div> |
| <div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  out_grad_index[i] = counter++;</div> |
| <div class="line"><a name="l00454"></a><span class="lineno"> 454</span>  }</div> |
| <div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i < in_data_index.size(); ++i) {</div> |
| <div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  in_data_index[i] = counter++;</div> |
| <div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  }</div> |
| <div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i < out_data_index.size(); ++i) {</div> |
| <div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  out_data_index[i] = counter++;</div> |
| <div class="line"><a name="l00460"></a><span class="lineno"> 460</span>  }</div> |
| <div class="line"><a name="l00461"></a><span class="lineno"> 461</span>  std::vector<T> all_data;</div> |
| <div class="line"><a name="l00462"></a><span class="lineno"> 462</span>  all_data.insert(all_data.end(), out_grad.begin(), out_grad.end());</div> |
| <div class="line"><a name="l00463"></a><span class="lineno"> 463</span>  all_data.insert(all_data.end(), in_data.begin(), in_data.end());</div> |
| <div class="line"><a name="l00464"></a><span class="lineno"> 464</span>  all_data.insert(all_data.end(), out_data.begin(), out_data.end());</div> |
| <div class="line"><a name="l00465"></a><span class="lineno"> 465</span> </div> |
| <div class="line"><a name="l00466"></a><span class="lineno"> 466</span>  std::vector<int> ret_index = this-><a class="code" href="classmxnet_1_1OperatorProperty.html#abf9e6a8d40750f3ee81fe30cbe3e2aae">DeclareBackwardDependency</a>(</div> |
| <div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  out_grad_index, in_data_index, out_data_index);</div> |
| <div class="line"><a name="l00468"></a><span class="lineno"> 468</span> </div> |
| <div class="line"><a name="l00469"></a><span class="lineno"> 469</span>  std::vector<T> ret(ret_index.size());</div> |
| <div class="line"><a name="l00470"></a><span class="lineno"> 470</span>  <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i < ret_index.size(); ++i) {</div> |
| <div class="line"><a name="l00471"></a><span class="lineno"> 471</span>  ret[i] = all_data[ret_index[i]];</div> |
| <div class="line"><a name="l00472"></a><span class="lineno"> 472</span>  }</div> |
| <div class="line"><a name="l00473"></a><span class="lineno"> 473</span>  <span class="keywordflow">return</span> ret;</div> |
| <div class="line"><a name="l00474"></a><span class="lineno"> 474</span>  }</div> |
| <div class="line"><a name="l00480"></a><span class="lineno"> 480</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1OperatorProperty.html">OperatorProperty</a> *<a class="code" href="classmxnet_1_1OperatorProperty.html#a15cf1f905b08a008ff38556ef6a35dd0">Create</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* type_name);</div> |
| <div class="line"><a name="l00481"></a><span class="lineno"> 481</span> };</div> |
| <div class="line"><a name="l00482"></a><span class="lineno"> 482</span> </div> |
| <div class="line"><a name="l00484"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a6f5428fbf261a94eaac079ec24413815"> 484</a></span> <span class="keyword">typedef</span> std::function<OperatorProperty *()> <a class="code" href="namespacemxnet.html#a6f5428fbf261a94eaac079ec24413815">OperatorPropertyFactory</a>;</div> |
| <div class="line"><a name="l00488"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html"> 488</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1OperatorPropertyReg.html">OperatorPropertyReg</a></div> |
| <div class="line"><a name="l00489"></a><span class="lineno"> 489</span>  : <span class="keyword">public</span> dmlc::FunctionRegEntryBase<OperatorPropertyReg,</div> |
| <div class="line"><a name="l00490"></a><span class="lineno"> 490</span>  OperatorPropertyFactory> {</div> |
| <div class="line"><a name="l00503"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html#a0e3536ad5305fa1ef402c5385fb4a089"> 503</a></span>  <span class="keyword">inline</span> <a class="code" href="structmxnet_1_1OperatorPropertyReg.html">OperatorPropertyReg</a>& <a class="code" href="structmxnet_1_1OperatorPropertyReg.html#a0e3536ad5305fa1ef402c5385fb4a089">set_key_var_num_args</a>(<span class="keyword">const</span> std::string &key) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  this-><a class="code" href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c">key_var_num_args</a> = key;</div> |
| <div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00506"></a><span class="lineno"> 506</span>  }</div> |
| <div class="line"><a name="l00510"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html#ad959fdbefb6dd807ee1961a2fec7a038"> 510</a></span>  <span class="keyword">inline</span> <a class="code" href="structmxnet_1_1OperatorPropertyReg.html">OperatorPropertyReg</a>& <a class="code" href="structmxnet_1_1OperatorPropertyReg.html#ad959fdbefb6dd807ee1961a2fec7a038">check_name</a>() {</div> |
| <div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  <a class="code" href="classmxnet_1_1OperatorProperty.html">OperatorProperty</a> *p = this->body();</div> |
| <div class="line"><a name="l00512"></a><span class="lineno"> 512</span>  std::string type = p-><a class="code" href="classmxnet_1_1OperatorProperty.html#ac6b7df90c0285a1823b463c21e926698">TypeString</a>();</div> |
| <div class="line"><a name="l00513"></a><span class="lineno"> 513</span>  <span class="keyword">delete</span> p;</div> |
| <div class="line"><a name="l00514"></a><span class="lineno"> 514</span>  CHECK_EQ(this->name, type)</div> |
| <div class="line"><a name="l00515"></a><span class="lineno"> 515</span>  << <span class="stringliteral">"Register Name and TypeString mismatch, name=\""</span> << this->name << <span class="stringliteral">"\","</span></div> |
| <div class="line"><a name="l00516"></a><span class="lineno"> 516</span>  << <span class="stringliteral">" but TypeString=\""</span> << type <<<span class="stringliteral">"\""</span>;</div> |
| <div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00518"></a><span class="lineno"> 518</span>  }</div> |
| <div class="line"><a name="l00519"></a><span class="lineno"> 519</span> </div> |
| <div class="line"><a name="l00521"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c"> 521</a></span>  std::string <a class="code" href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c">key_var_num_args</a>;</div> |
| <div class="line"><a name="l00522"></a><span class="lineno"> 522</span> };</div> |
| <div class="line"><a name="l00523"></a><span class="lineno"> 523</span> </div> |
| <div class="line"><a name="l00524"></a><span class="lineno"> 524</span> <span class="comment">//---------------------------------------------------------------------------------</span></div> |
| <div class="line"><a name="l00525"></a><span class="lineno"> 525</span> <span class="comment">// The following part are API Registration of Operators</span></div> |
| <div class="line"><a name="l00526"></a><span class="lineno"> 526</span> <span class="comment">// See also MXNET_REGISTER_SIMPLE_OP in operator_util.h for registering simple ops.</span></div> |
| <div class="line"><a name="l00527"></a><span class="lineno"> 527</span> <span class="comment">//---------------------------------------------------------------------------------</span></div> |
| <div class="line"><a name="l00538"></a><span class="lineno"><a class="line" href="operator_8h.html#a926db042759256e94d20696c66f65709"> 538</a></span> <span class="comment"></span><span class="preprocessor">#define MXNET_REGISTER_OP_PROPERTY(name, OperatorPropertyType) \</span></div> |
| <div class="line"><a name="l00539"></a><span class="lineno"> 539</span> <span class="preprocessor"> DMLC_REGISTRY_REGISTER(::mxnet::OperatorPropertyReg, OperatorPropertyReg, name) \</span></div> |
| <div class="line"><a name="l00540"></a><span class="lineno"> 540</span> <span class="preprocessor"> .set_body([]() { return new OperatorPropertyType(); }) \</span></div> |
| <div class="line"><a name="l00541"></a><span class="lineno"> 541</span> <span class="preprocessor"> .set_return_type("NDArray-or-Symbol") \</span></div> |
| <div class="line"><a name="l00542"></a><span class="lineno"> 542</span> <span class="preprocessor"> .check_name()</span></div> |
| <div class="line"><a name="l00543"></a><span class="lineno"> 543</span> <span class="preprocessor"></span></div> |
| <div class="line"><a name="l00544"></a><span class="lineno"> 544</span> <span class="preprocessor">#endif // DMLC_USE_CXX11</span></div> |
| <div class="line"><a name="l00545"></a><span class="lineno"> 545</span> <span class="preprocessor"></span>} <span class="comment">// namespace mxnet</span></div> |
| <div class="line"><a name="l00546"></a><span class="lineno"> 546</span> <span class="preprocessor">#endif // MXNET_OPERATOR_H_</span></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_abf9e6a8d40750f3ee81fe30cbe3e2aae"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#abf9e6a8d40750f3ee81fe30cbe3e2aae">mxnet::OperatorProperty::DeclareBackwardDependency</a></div><div class="ttdeci">virtual std::vector< int > DeclareBackwardDependency(const std::vector< int > &out_grad, const std::vector< int > &in_data, const std::vector< int > &out_data) const </div><div class="ttdoc">Declare the input requirement of Backward pass. </div><div class="ttdef"><b>Definition:</b> operator.h:362</div></div> |
| <div class="ttc" id="structmxnet_1_1OperatorPropertyReg_html_ad959fdbefb6dd807ee1961a2fec7a038"><div class="ttname"><a href="structmxnet_1_1OperatorPropertyReg.html#ad959fdbefb6dd807ee1961a2fec7a038">mxnet::OperatorPropertyReg::check_name</a></div><div class="ttdeci">OperatorPropertyReg & check_name()</div><div class="ttdoc">Check if TypeString of the type matches the registered name. </div><div class="ttdef"><b>Definition:</b> operator.h:510</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26">mxnet::kNullOp</a></div><div class="ttdoc">no operation, do not write anything </div><div class="ttdef"><b>Definition:</b> operator.h:26</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a">mxnet::kWriteTo</a></div><div class="ttdoc">write gradient to provided space </div><div class="ttdef"><b>Definition:</b> operator.h:28</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a89a284fada5cc1801ca1c0390b8ae17aa8adac8fd6acb05ff169a2913ad05122c"><div class="ttname"><a href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa8adac8fd6acb05ff169a2913ad05122c">mxnet::Operator::kSync</a></div><div class="ttdoc">Forward/Backward are synchronize calls. </div><div class="ttdef"><b>Definition:</b> operator.h:82</div></div> |
| <div class="ttc" id="structmxnet_1_1RunContext_html_a8bcd2fc6392f95a1fdc3d98f74dbfcb9"><div class="ttname"><a href="structmxnet_1_1RunContext.html#a8bcd2fc6392f95a1fdc3d98f74dbfcb9">mxnet::RunContext::get_stream</a></div><div class="ttdeci">mshadow::Stream< xpu > * get_stream() const </div><div class="ttdoc">get mshadow stream from Context </div><div class="ttdef"><b>Definition:</b> base.h:224</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a3dbe23d985df86da51eeb868439872dd"><div class="ttname"><a href="classmxnet_1_1Operator.html#a3dbe23d985df86da51eeb868439872dd">mxnet::Operator::Forward</a></div><div class="ttdeci">virtual void Forward(const OpContext &ctx, const std::vector< TBlob > &in_data, const std::vector< OpReqType > &req, const std::vector< TBlob > &out_data, const std::vector< TBlob > &aux_states)=0</div><div class="ttdoc">perform a forward operation of Operator, save the output to TBlob. </div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_ad32d74b9bf56caf920b5e6379d2bb259"><div class="ttname"><a href="classmxnet_1_1Operator.html#ad32d74b9bf56caf920b5e6379d2bb259">mxnet::Operator::Backward</a></div><div class="ttdeci">virtual void Backward(const OpContext &ctx, const std::vector< TBlob > &out_grad, const std::vector< TBlob > &in_data, const std::vector< TBlob > &out_data, const std::vector< OpReqType > &req, const std::vector< TBlob > &in_grad, const std::vector< TBlob > &aux_states)</div><div class="ttdoc">Perform a Backward Operation, write gradient to the in_grad. </div><div class="ttdef"><b>Definition:</b> operator.h:142</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a4d34cc792d7271bd8b5b60eaa0b55767"><div class="ttname"><a href="classmxnet_1_1Operator.html#a4d34cc792d7271bd8b5b60eaa0b55767">mxnet::Operator::~Operator</a></div><div class="ttdeci">virtual ~Operator()</div><div class="ttdoc">destructor </div><div class="ttdef"><b>Definition:</b> operator.h:97</div></div> |
| <div class="ttc" id="namespacemxnet_html_a6f5428fbf261a94eaac079ec24413815"><div class="ttname"><a href="namespacemxnet.html#a6f5428fbf261a94eaac079ec24413815">mxnet::OperatorPropertyFactory</a></div><div class="ttdeci">std::function< OperatorProperty *()> OperatorPropertyFactory</div><div class="ttdoc">typedef the factory function of operator property </div><div class="ttdef"><b>Definition:</b> operator.h:484</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a9f488509e132094ce7e79125ce853042"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a9f488509e132094ce7e79125ce853042">mxnet::OperatorProperty::BackwardInputs</a></div><div class="ttdeci">std::vector< T > BackwardInputs(const std::vector< T > &out_grad, const std::vector< T > &in_data, const std::vector< T > &out_data) const </div><div class="ttdoc">Get Backward Input Dependency for generic types of data. Normally T can be pointer of Symbol::DataEnt...</div><div class="ttdef"><b>Definition:</b> operator.h:445</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_ade7720a3e0ebba67d89978925a77609b"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#ade7720a3e0ebba67d89978925a77609b">mxnet::OperatorProperty::BackwardResource</a></div><div class="ttdeci">virtual std::vector< ResourceRequest > BackwardResource(const std::vector< TShape > &in_shape) const </div><div class="ttdoc">Declare additional resource required in backward pass. These additional resources will be presented i...</div><div class="ttdef"><b>Definition:</b> operator.h:336</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a15cf1f905b08a008ff38556ef6a35dd0"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a15cf1f905b08a008ff38556ef6a35dd0">mxnet::OperatorProperty::Create</a></div><div class="ttdeci">static OperatorProperty * Create(const char *type_name)</div><div class="ttdoc">create OperatorProperty </div></div> |
| <div class="ttc" id="structmxnet_1_1RunContext_html"><div class="ttname"><a href="structmxnet_1_1RunContext.html">mxnet::RunContext</a></div><div class="ttdoc">execution time context. The information needed in runtime for actual execution. </div><div class="ttdef"><b>Definition:</b> base.h:213</div></div> |
| <div class="ttc" id="structmxnet_1_1OperatorPropertyReg_html_a0e3536ad5305fa1ef402c5385fb4a089"><div class="ttname"><a href="structmxnet_1_1OperatorPropertyReg.html#a0e3536ad5305fa1ef402c5385fb4a089">mxnet::OperatorPropertyReg::set_key_var_num_args</a></div><div class="ttdeci">OperatorPropertyReg & set_key_var_num_args(const std::string &key)</div><div class="ttdoc">Set key_var_num_args When this is set, the API caller is required to pass in a argument with key=key_...</div><div class="ttdef"><b>Definition:</b> operator.h:503</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a610af8e9cb560104301ccd1efa2a96fd"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a610af8e9cb560104301ccd1efa2a96fd">mxnet::OperatorProperty::~OperatorProperty</a></div><div class="ttdeci">virtual ~OperatorProperty()</div><div class="ttdoc">virtual destructor </div><div class="ttdef"><b>Definition:</b> operator.h:171</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a40ce0e3d4ac21412cfafafc5b1fd32ef"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef">mxnet::OpContext::async_on_complete</a></div><div class="ttdeci">engine::CallbackOnComplete async_on_complete</div><div class="ttdoc">the callback when operation completes, used by asynchronize ops </div><div class="ttdef"><b>Definition:</b> operator.h:51</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a1b37fb362748fa5027d4b5c925131892"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a1b37fb362748fa5027d4b5c925131892">mxnet::OperatorProperty::ListOutputs</a></div><div class="ttdeci">virtual std::vector< std::string > ListOutputs() const </div><div class="ttdoc">Get name of output values of Operator. </div><div class="ttdef"><b>Definition:</b> operator.h:194</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html"><div class="ttname"><a href="structmxnet_1_1OpContext.html">mxnet::OpContext</a></div><div class="ttdoc">All the possible information needed by Operator.Forward and Backward This is the superset of RunConte...</div><div class="ttdef"><b>Definition:</b> operator.h:45</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_abd7405c065167e4e15af032b1e0a5c42"><div class="ttname"><a href="structmxnet_1_1OpContext.html#abd7405c065167e4e15af032b1e0a5c42">mxnet::OpContext::is_train</a></div><div class="ttdeci">int is_train</div><div class="ttdoc">whether it is training phase </div><div class="ttdef"><b>Definition:</b> operator.h:47</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html"><div class="ttname"><a href="classmxnet_1_1Operator.html">mxnet::Operator</a></div><div class="ttdoc">Operator interface. Operator defines basic operation unit of optimized computation graph in mxnet...</div><div class="ttdef"><b>Definition:</b> operator.h:77</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a704ae43181c8a88bfd0a47e8005f7cb8"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a704ae43181c8a88bfd0a47e8005f7cb8">mxnet::OperatorProperty::GetParams</a></div><div class="ttdeci">virtual std::map< std::string, std::string > GetParams() const =0</div><div class="ttdoc">Get a map representation of internal parameters. This can be used by Init to recover the state of Ope...</div></div> |
| <div class="ttc" id="resource_8h_html"><div class="ttname"><a href="resource_8h.html">resource.h</a></div><div class="ttdoc">Global resource allocation handling. </div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a89a284fada5cc1801ca1c0390b8ae17aa6c403d828787dc2aafd6ee1e9cd02699"><div class="ttname"><a href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aa6c403d828787dc2aafd6ee1e9cd02699">mxnet::Operator::kAsync</a></div><div class="ttdoc">Forward/Backward are asynchronize, will call OpContext.async_on_complete when operation finishes...</div><div class="ttdef"><b>Definition:</b> operator.h:87</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_aad7a3e87866515a2b7100e849a2fe966"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#aad7a3e87866515a2b7100e849a2fe966">mxnet::OperatorProperty::ForwardInplaceOption</a></div><div class="ttdeci">virtual std::vector< std::pair< int, void * > > ForwardInplaceOption(const std::vector< int > &in_data, const std::vector< void * > &out_data) const </div><div class="ttdoc">Get possible forward inplace options. This function enables optimization to reuse memory of inputs in...</div><div class="ttdef"><b>Definition:</b> operator.h:394</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html">mxnet::OperatorProperty</a></div><div class="ttdoc">OperatorProperty is a object that stores all information about Operator. It also contains method to g...</div><div class="ttdef"><b>Definition:</b> operator.h:166</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a7985004f99a931b4a6ff69df733bae72"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a7985004f99a931b4a6ff69df733bae72">mxnet::OperatorProperty::CreateOperatorEx</a></div><div class="ttdeci">virtual Operator * CreateOperatorEx(Context ctx, std::vector< TShape > *in_shape, std::vector< int > *in_type) const </div><div class="ttdoc">Create a Operator on specific context and input shape/type. </div><div class="ttdef"><b>Definition:</b> operator.h:297</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a67231a5a1525ec7480de98a79dce0194"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a67231a5a1525ec7480de98a79dce0194">mxnet::OperatorProperty::BackwardInplaceOption</a></div><div class="ttdeci">virtual std::vector< std::pair< int, void * > > BackwardInplaceOption(const std::vector< int > &out_grad, const std::vector< int > &in_data, const std::vector< int > &out_data, const std::vector< void * > &in_grad) const </div><div class="ttdoc">Get possible backward inplace options. This function enables optimization to reuse memory of inputs i...</div><div class="ttdef"><b>Definition:</b> operator.h:425</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_ad5c9f53ea70f3eeb1bf27b983f420c0b"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#ad5c9f53ea70f3eeb1bf27b983f420c0b">mxnet::OperatorProperty::InferType</a></div><div class="ttdeci">virtual bool InferType(std::vector< int > *in_type, std::vector< int > *out_type, std::vector< int > *aux_type) const </div><div class="ttdoc">infer the data types of outputs and unknown input arguments </div><div class="ttdef"><b>Definition:</b> operator.h:260</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a0e20dead67dad0cd76c147a997ef5094"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a0e20dead67dad0cd76c147a997ef5094">mxnet::OperatorProperty::ListAuxiliaryStates</a></div><div class="ttdeci">virtual std::vector< std::string > ListAuxiliaryStates() const </div><div class="ttdoc">Get name of auxiliary states of Operator. </div><div class="ttdef"><b>Definition:</b> operator.h:201</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a24cc6a3763ff5f0881a2de0652810596"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a24cc6a3763ff5f0881a2de0652810596">mxnet::OperatorProperty::ListArguments</a></div><div class="ttdeci">virtual std::vector< std::string > ListArguments() const </div><div class="ttdoc">Get input arguments of the Operator. </div><div class="ttdef"><b>Definition:</b> operator.h:187</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51">mxnet::kWriteInplace</a></div><div class="ttdoc">perform an inplace write, Target shares memory with one of input arguments. This option only happen w...</div><div class="ttdef"><b>Definition:</b> operator.h:34</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">mxnet::OpReqType</a></div><div class="ttdeci">OpReqType</div><div class="ttdoc">operation request type to Forward and Backward </div><div class="ttdef"><b>Definition:</b> operator.h:24</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a89a284fada5cc1801ca1c0390b8ae17a"><div class="ttname"><a href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17a">mxnet::Operator::ExecType</a></div><div class="ttdeci">ExecType</div><div class="ttdoc">the execution type of the operator </div><div class="ttdef"><b>Definition:</b> operator.h:80</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a89a284fada5cc1801ca1c0390b8ae17aaebe5a0e5db6ee71f6905f9b34a1f05ad"><div class="ttname"><a href="classmxnet_1_1Operator.html#a89a284fada5cc1801ca1c0390b8ae17aaebe5a0e5db6ee71f6905f9b34a1f05ad">mxnet::Operator::kCrossDeviceCopy</a></div><div class="ttdoc">Cross device copy operation, this is a special operator That indicates copy across devices...</div><div class="ttdef"><b>Definition:</b> operator.h:94</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a833e95f95cf5e3af9c193b74b500fc5b"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b">mxnet::OpContext::requested</a></div><div class="ttdeci">std::vector< Resource > requested</div><div class="ttdoc">Resources requested by the operator. </div><div class="ttdef"><b>Definition:</b> operator.h:53</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a869d4c68c0028a175406f11332baf7cf"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf">mxnet::OpContext::run_ctx</a></div><div class="ttdeci">RunContext run_ctx</div><div class="ttdoc">RunContext related resources. </div><div class="ttdef"><b>Definition:</b> operator.h:49</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a0fb32e0ab03a86517473db0f0630856c"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a0fb32e0ab03a86517473db0f0630856c">mxnet::OperatorProperty::Init</a></div><div class="ttdeci">virtual void Init(const std::vector< std::pair< std::string, std::string > > &kwargs)=0</div><div class="ttdoc">Initialize the Operator by setting the parameters This function need to be called before all other fu...</div></div> |
| <div class="ttc" id="structmxnet_1_1OperatorPropertyReg_html_af3aee96c6a2289e7554d7503186a540c"><div class="ttname"><a href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c">mxnet::OperatorPropertyReg::key_var_num_args</a></div><div class="ttdeci">std::string key_var_num_args</div><div class="ttdoc">The key num_args name. </div><div class="ttdef"><b>Definition:</b> operator.h:521</div></div> |
| <div class="ttc" id="classmxnet_1_1engine_1_1CallbackOnComplete_html"><div class="ttname"><a href="classmxnet_1_1engine_1_1CallbackOnComplete.html">mxnet::engine::CallbackOnComplete</a></div><div class="ttdoc">OnComplete Callback to the engine, called by AsyncFn when action completes. </div><div class="ttdef"><b>Definition:</b> engine.h:37</div></div> |
| <div class="ttc" id="base_8h_html"><div class="ttname"><a href="base_8h.html">base.h</a></div><div class="ttdoc">configuation of mxnet as well as basic data structure. </div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a6dc5ecd089ef01dcc154c02b20ad2f4b"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a6dc5ecd089ef01dcc154c02b20ad2f4b">mxnet::OperatorProperty::NumVisibleOutputs</a></div><div class="ttdeci">virtual int NumVisibleOutputs() const </div><div class="ttdoc">get number of visible return values during Symbol creation. If NumVisibleOutputs() = k...</div><div class="ttdef"><b>Definition:</b> operator.h:220</div></div> |
| <div class="ttc" id="structmxnet_1_1OperatorPropertyReg_html"><div class="ttname"><a href="structmxnet_1_1OperatorPropertyReg.html">mxnet::OperatorPropertyReg</a></div><div class="ttdoc">Registry entry for OperatorProperty factory functions. </div><div class="ttdef"><b>Definition:</b> operator.h:488</div></div> |
| <div class="ttc" id="classmxnet_1_1Operator_html_a7bffb2be177edf8c29fd0148ebbd7ca5"><div class="ttname"><a href="classmxnet_1_1Operator.html#a7bffb2be177edf8c29fd0148ebbd7ca5">mxnet::Operator::exec_type</a></div><div class="ttdeci">virtual ExecType exec_type() const </div><div class="ttdef"><b>Definition:</b> operator.h:152</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383">mxnet::kAddTo</a></div><div class="ttdoc">add to the provided space </div><div class="ttdef"><b>Definition:</b> operator.h:36</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a543cbf36f0e4d8dc99109db71705a11c"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a543cbf36f0e4d8dc99109db71705a11c">mxnet::OperatorProperty::InferShape</a></div><div class="ttdeci">virtual bool InferShape(std::vector< TShape > *in_shape, std::vector< TShape > *out_shape, std::vector< TShape > *aux_shape) const =0</div><div class="ttdoc">infer the shapes of outputs and unknown input arguments </div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a623210dd1b9495ca73f223e0b06f998c"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a623210dd1b9495ca73f223e0b06f998c">mxnet::OperatorProperty::ForwardResource</a></div><div class="ttdeci">virtual std::vector< ResourceRequest > ForwardResource(const std::vector< TShape > &in_shape) const </div><div class="ttdoc">Declare additional resource required in forward pass. These additional resources will be presented in...</div><div class="ttdef"><b>Definition:</b> operator.h:325</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a546ab25ddaae590ca98705ae9abc7ac0"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a546ab25ddaae590ca98705ae9abc7ac0">mxnet::OperatorProperty::CreateOperator</a></div><div class="ttdeci">virtual Operator * CreateOperator(Context ctx) const =0</div><div class="ttdoc">Create a Operator on specific context. </div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_ab38f2f877555ba34b548efd2575cac99"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#ab38f2f877555ba34b548efd2575cac99">mxnet::OperatorProperty::Copy</a></div><div class="ttdeci">virtual OperatorProperty * Copy() const =0</div><div class="ttdoc">Copy this OperatorProperty. </div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_a49e25d85c38199338a9fa8150bc86df1"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a49e25d85c38199338a9fa8150bc86df1">mxnet::OperatorProperty::NumOutputs</a></div><div class="ttdeci">virtual int NumOutputs() const </div><div class="ttdef"><b>Definition:</b> operator.h:205</div></div> |
| <div class="ttc" id="structmxnet_1_1Context_html"><div class="ttname"><a href="structmxnet_1_1Context.html">mxnet::Context</a></div><div class="ttdoc">Context information about the execution environment. </div><div class="ttdef"><b>Definition:</b> base.h:116</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_aa700c5c28e024b42d41e15f46615f1b8"><div class="ttname"><a href="structmxnet_1_1OpContext.html#aa700c5c28e024b42d41e15f46615f1b8">mxnet::OpContext::get_stream</a></div><div class="ttdeci">mshadow::Stream< xpu > * get_stream() const </div><div class="ttdoc">get mshadow stream from Context </div><div class="ttdef"><b>Definition:</b> operator.h:60</div></div> |
| <div class="ttc" id="classmxnet_1_1OperatorProperty_html_ac6b7df90c0285a1823b463c21e926698"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#ac6b7df90c0285a1823b463c21e926698">mxnet::OperatorProperty::TypeString</a></div><div class="ttdeci">virtual std::string TypeString() const =0</div><div class="ttdoc">return the type string of the Operator subclasses override this function. </div></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Sun Aug 6 2017 07:14:31 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.6 |
| </small></address> |
| </body> |
| </html> |