blob: 42273d8da5babd97b950dc805919399bab11164a [file] [log] [blame]
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<meta name="generator" content="Doxygen 1.8.17"/>
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<title>mxnet: /work/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/searchdata.js"></script>
<script type="text/javascript" src="search/search.js"></script>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
</head>
<body>
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
<div id="titlearea">
<table cellspacing="0" cellpadding="0">
<tbody>
<tr style="height: 56px;">
<td id="projectalign" style="padding-left: 0.5em;">
<div id="projectname">mxnet
</div>
</td>
</tr>
</tbody>
</table>
</div>
<!-- end header part -->
<!-- Generated by Doxygen 1.8.17 -->
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
var searchBox = new SearchBox("searchBox", "search",false,'Search');
/* @license-end */
</script>
<script type="text/javascript" src="menudata.js"></script>
<script type="text/javascript" src="menu.js"></script>
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
$(function() {
initMenu('',true,false,'search.php','Search');
$(document).ready(function() { init_search(); });
});
/* @license-end */</script>
<div id="main-nav"></div>
<!-- window showing the filter options -->
<div id="MSearchSelectWindow"
onmouseover="return searchBox.OnSearchSelectShow()"
onmouseout="return searchBox.OnSearchSelectHide()"
onkeydown="return searchBox.OnSearchSelectKey(event)">
</div>
<!-- iframe showing the search results (closed by default) -->
<div id="MSearchResultsWindow">
<iframe src="javascript:void(0)" frameborder="0"
name="MSearchResults" id="MSearchResults">
</iframe>
</div>
<div id="nav-path" class="navpath">
<ul>
<li class="navelem"><a class="el" href="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="include_2mxnet_2operator_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span>&#160;<span class="comment">/*</span></div>
<div class="line"><a name="l00002"></a><span class="lineno"> 2</span>&#160;<span class="comment"> * Licensed to the Apache Software Foundation (ASF) under one</span></div>
<div class="line"><a name="l00003"></a><span class="lineno"> 3</span>&#160;<span class="comment"> * or more contributor license agreements. See the NOTICE file</span></div>
<div class="line"><a name="l00004"></a><span class="lineno"> 4</span>&#160;<span class="comment"> * distributed with this work for additional information</span></div>
<div class="line"><a name="l00005"></a><span class="lineno"> 5</span>&#160;<span class="comment"> * regarding copyright ownership. The ASF licenses this file</span></div>
<div class="line"><a name="l00006"></a><span class="lineno"> 6</span>&#160;<span class="comment"> * to you under the Apache License, Version 2.0 (the</span></div>
<div class="line"><a name="l00007"></a><span class="lineno"> 7</span>&#160;<span class="comment"> * &quot;License&quot;); you may not use this file except in compliance</span></div>
<div class="line"><a name="l00008"></a><span class="lineno"> 8</span>&#160;<span class="comment"> * with the License. You may obtain a copy of the License at</span></div>
<div class="line"><a name="l00009"></a><span class="lineno"> 9</span>&#160;<span class="comment"> *</span></div>
<div class="line"><a name="l00010"></a><span class="lineno"> 10</span>&#160;<span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span></div>
<div class="line"><a name="l00011"></a><span class="lineno"> 11</span>&#160;<span class="comment"> *</span></div>
<div class="line"><a name="l00012"></a><span class="lineno"> 12</span>&#160;<span class="comment"> * Unless required by applicable law or agreed to in writing,</span></div>
<div class="line"><a name="l00013"></a><span class="lineno"> 13</span>&#160;<span class="comment"> * software distributed under the License is distributed on an</span></div>
<div class="line"><a name="l00014"></a><span class="lineno"> 14</span>&#160;<span class="comment"> * &quot;AS IS&quot; BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span></div>
<div class="line"><a name="l00015"></a><span class="lineno"> 15</span>&#160;<span class="comment"> * KIND, either express or implied. See the License for the</span></div>
<div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160;<span class="comment"> * specific language governing permissions and limitations</span></div>
<div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160;<span class="comment"> * under the License.</span></div>
<div class="line"><a name="l00018"></a><span class="lineno"> 18</span>&#160;<span class="comment"> */</span></div>
<div class="line"><a name="l00019"></a><span class="lineno"> 19</span>&#160; </div>
<div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160;<span class="preprocessor">#ifndef MXNET_OPERATOR_H_</span></div>
<div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160;<span class="preprocessor">#define MXNET_OPERATOR_H_</span></div>
<div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; </div>
<div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="3rdparty_2dmlc-core_2include_2dmlc_2base_8h.html">dmlc/base.h</a>&gt;</span></div>
<div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="json_8h.html">dmlc/json.h</a>&gt;</span></div>
<div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160;<span class="preprocessor">#include &lt;dmlc/logging.h&gt;</span></div>
<div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="3rdparty_2dmlc-core_2include_2dmlc_2registry_8h.html">dmlc/registry.h</a>&gt;</span></div>
<div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160;<span class="preprocessor">#include &lt;<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2node_8h.html">nnvm/node.h</a>&gt;</span></div>
<div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160;<span class="preprocessor">#include &lt;vector&gt;</span></div>
<div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160;<span class="preprocessor">#include &lt;map&gt;</span></div>
<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160;<span class="preprocessor">#include &lt;string&gt;</span></div>
<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160;<span class="preprocessor">#include &lt;utility&gt;</span></div>
<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="3rdparty_2dmlc-core_2include_2dmlc_2base_8h.html">./base.h</a>&quot;</span></div>
<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="resource_8h.html">./resource.h</a>&quot;</span></div>
<div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="include_2mxnet_2op__attr__types_8h.html">./op_attr_types.h</a>&quot;</span></div>
<div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; </div>
<div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160;<span class="keyword">namespace </span><a class="code" href="namespacemxnet.html">mxnet</a> {</div>
<div class="line"><a name="l00054"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html"> 54</a></span>&#160;<span class="keyword">class </span><a class="code" href="classmxnet_1_1Operator.html">Operator</a> {</div>
<div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; <span class="keyword">public</span>:</div>
<div class="line"><a name="l00057"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#a4d34cc792d7271bd8b5b60eaa0b55767"> 57</a></span>&#160; <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1Operator.html#a4d34cc792d7271bd8b5b60eaa0b55767">~Operator</a>() {}</div>
<div class="line"><a name="l00069"></a><span class="lineno"> 69</span>&#160; <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>&amp; ctx,</div>
<div class="line"><a name="l00070"></a><span class="lineno"> 70</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; in_data,</div>
<div class="line"><a name="l00071"></a><span class="lineno"> 71</span>&#160; <span class="keyword">const</span> std::vector&lt;OpReqType&gt;&amp; req,</div>
<div class="line"><a name="l00072"></a><span class="lineno"> 72</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; out_data,</div>
<div class="line"><a name="l00073"></a><span class="lineno"> 73</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; aux_states) = 0;</div>
<div class="line"><a name="l00102"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#ad32d74b9bf56caf920b5e6379d2bb259"> 102</a></span>&#160; <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>&amp; ctx,</div>
<div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; out_grad,</div>
<div class="line"><a name="l00104"></a><span class="lineno"> 104</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; in_data,</div>
<div class="line"><a name="l00105"></a><span class="lineno"> 105</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; out_data,</div>
<div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160; <span class="keyword">const</span> std::vector&lt;OpReqType&gt;&amp; req,</div>
<div class="line"><a name="l00107"></a><span class="lineno"> 107</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; in_grad,</div>
<div class="line"><a name="l00108"></a><span class="lineno"> 108</span>&#160; <span class="keyword">const</span> std::vector&lt;TBlob&gt;&amp; aux_states) {</div>
<div class="line"><a name="l00109"></a><span class="lineno"> 109</span>&#160; LOG(FATAL) &lt;&lt; <span class="stringliteral">&quot;Backward is not implemented&quot;</span>;</div>
<div class="line"><a name="l00110"></a><span class="lineno"> 110</span>&#160; }</div>
<div class="line"><a name="l00112"></a><span class="lineno"><a class="line" href="classmxnet_1_1Operator.html#aa6b4c247a914b4659c773965248019c7"> 112</a></span>&#160; <span class="keyword">virtual</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad">ExecType</a> <a class="code" href="classmxnet_1_1Operator.html#aa6b4c247a914b4659c773965248019c7">exec_type</a>() <span class="comment">// NOLINT(*) exec_type has been moved to OperatorProperty</span></div>
<div class="line"><a name="l00113"></a><span class="lineno"> 113</span>&#160; const final { <span class="comment">// NOLINT(*) exec_type has been moved to OperatorProperty</span></div>
<div class="line"><a name="l00114"></a><span class="lineno"> 114</span>&#160; <span class="keywordflow">return</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c">ExecType::kSync</a>;</div>
<div class="line"><a name="l00115"></a><span class="lineno"> 115</span>&#160; }</div>
<div class="line"><a name="l00116"></a><span class="lineno"> 116</span>&#160;};</div>
<div class="line"><a name="l00117"></a><span class="lineno"> 117</span>&#160; </div>
<div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160;<span class="preprocessor">#if DMLC_USE_CXX11</span></div>
<div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160;<span class="comment">// OperatorProperty allows C++11, while Operator do not rely on it.</span></div>
<div class="line"><a name="l00127"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html"> 127</a></span>&#160;<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="l00128"></a><span class="lineno"> 128</span>&#160; <span class="keyword">public</span>:</div>
<div class="line"><a name="l00132"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a610af8e9cb560104301ccd1efa2a96fd"> 132</a></span>&#160; <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a610af8e9cb560104301ccd1efa2a96fd">~OperatorProperty</a>() {}</div>
<div class="line"><a name="l00138"></a><span class="lineno"> 138</span>&#160; <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&lt;std::pair&lt;std::string, std::string&gt; &gt;&amp; kwargs) = 0;</div>
<div class="line"><a name="l00143"></a><span class="lineno"> 143</span>&#160; <span class="keyword">virtual</span> std::map&lt;std::string, std::string&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a704ae43181c8a88bfd0a47e8005f7cb8">GetParams</a>() <span class="keyword">const</span> = 0;</div>
<div class="line"><a name="l00148"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#ad01be3a84423800af7e818b45afc748f"> 148</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;std::string&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#ad01be3a84423800af7e818b45afc748f">ListArguments</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00149"></a><span class="lineno"> 149</span>&#160; <span class="keywordflow">return</span> {<span class="stringliteral">&quot;data&quot;</span>};</div>
<div class="line"><a name="l00150"></a><span class="lineno"> 150</span>&#160; }</div>
<div class="line"><a name="l00155"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7"> 155</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;std::string&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7">ListOutputs</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00156"></a><span class="lineno"> 156</span>&#160; <span class="keywordflow">return</span> {<span class="stringliteral">&quot;output&quot;</span>};</div>
<div class="line"><a name="l00157"></a><span class="lineno"> 157</span>&#160; }</div>
<div class="line"><a name="l00162"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#aa6d3a955ffa452a368a9dd4d33ce5a25"> 162</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;std::string&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#aa6d3a955ffa452a368a9dd4d33ce5a25">ListAuxiliaryStates</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00163"></a><span class="lineno"> 163</span>&#160; <span class="keywordflow">return</span> {};</div>
<div class="line"><a name="l00164"></a><span class="lineno"> 164</span>&#160; }</div>
<div class="line"><a name="l00166"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a9fc6e9f04f52b1c05f62fb7357bb95c4"> 166</a></span>&#160; <span class="keyword">virtual</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a9fc6e9f04f52b1c05f62fb7357bb95c4">NumOutputs</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00167"></a><span class="lineno"> 167</span>&#160; <span class="keywordflow">return</span> this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7">ListOutputs</a>().size();</div>
<div class="line"><a name="l00168"></a><span class="lineno"> 168</span>&#160; }</div>
<div class="line"><a name="l00181"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a006f0e2c9b3f934e1f73b242b7e8b062"> 181</a></span>&#160; <span class="keyword">virtual</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a006f0e2c9b3f934e1f73b242b7e8b062">NumVisibleOutputs</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160; <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a9fc6e9f04f52b1c05f62fb7357bb95c4">NumOutputs</a>();</div>
<div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160; }</div>
<div class="line"><a name="l00201"></a><span class="lineno"> 201</span>&#160; <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a4be455d921fcb9e61ffafc567a31dc39">InferShape</a>(<a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>* in_shape,</div>
<div class="line"><a name="l00202"></a><span class="lineno"> 202</span>&#160; <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>* out_shape,</div>
<div class="line"><a name="l00203"></a><span class="lineno"> 203</span>&#160; <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>* aux_shape) <span class="keyword">const</span> = 0;</div>
<div class="line"><a name="l00221"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a819b81e14761e9e8b9c888a63c7df1fa"> 221</a></span>&#160; <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a819b81e14761e9e8b9c888a63c7df1fa">InferType</a>(std::vector&lt;int&gt;* in_type,</div>
<div class="line"><a name="l00222"></a><span class="lineno"> 222</span>&#160; std::vector&lt;int&gt;* out_type,</div>
<div class="line"><a name="l00223"></a><span class="lineno"> 223</span>&#160; std::vector&lt;int&gt;* aux_type)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00224"></a><span class="lineno"> 224</span>&#160; CHECK_LE(in_type-&gt;size(), this-&gt;ListArguments().size());</div>
<div class="line"><a name="l00225"></a><span class="lineno"> 225</span>&#160; <span class="keywordtype">int</span> n_in = this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#ad01be3a84423800af7e818b45afc748f">ListArguments</a>().size();</div>
<div class="line"><a name="l00226"></a><span class="lineno"> 226</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">unsigned</span> i = 0; i &lt; in_type-&gt;size(); ++i) {</div>
<div class="line"><a name="l00227"></a><span class="lineno"> 227</span>&#160; CHECK(in_type-&gt;at(i) == <a class="code" href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">mshadow::default_type_flag</a> || in_type-&gt;at(i) == -1)</div>
<div class="line"><a name="l00228"></a><span class="lineno"> 228</span>&#160; &lt;&lt; <span class="stringliteral">&quot;Unsupported data type &quot;</span> &lt;&lt; in_type-&gt;at(i);</div>
<div class="line"><a name="l00229"></a><span class="lineno"> 229</span>&#160; }</div>
<div class="line"><a name="l00230"></a><span class="lineno"> 230</span>&#160; in_type-&gt;clear();</div>
<div class="line"><a name="l00231"></a><span class="lineno"> 231</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i &lt; n_in; ++i)</div>
<div class="line"><a name="l00232"></a><span class="lineno"> 232</span>&#160; in_type-&gt;push_back(<a class="code" href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">mshadow::default_type_flag</a>);</div>
<div class="line"><a name="l00233"></a><span class="lineno"> 233</span>&#160; </div>
<div class="line"><a name="l00234"></a><span class="lineno"> 234</span>&#160; <span class="keywordtype">int</span> n_out = this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7">ListOutputs</a>().size();</div>
<div class="line"><a name="l00235"></a><span class="lineno"> 235</span>&#160; out_type-&gt;clear();</div>
<div class="line"><a name="l00236"></a><span class="lineno"> 236</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i &lt; n_out; ++i)</div>
<div class="line"><a name="l00237"></a><span class="lineno"> 237</span>&#160; out_type-&gt;push_back(<a class="code" href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">mshadow::default_type_flag</a>);</div>
<div class="line"><a name="l00238"></a><span class="lineno"> 238</span>&#160; </div>
<div class="line"><a name="l00239"></a><span class="lineno"> 239</span>&#160; <span class="keywordtype">int</span> n_aux = this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#aa6d3a955ffa452a368a9dd4d33ce5a25">ListAuxiliaryStates</a>().size();</div>
<div class="line"><a name="l00240"></a><span class="lineno"> 240</span>&#160; aux_type-&gt;clear();</div>
<div class="line"><a name="l00241"></a><span class="lineno"> 241</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i &lt; n_aux; ++i)</div>
<div class="line"><a name="l00242"></a><span class="lineno"> 242</span>&#160; aux_type-&gt;push_back(<a class="code" href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">mshadow::default_type_flag</a>);</div>
<div class="line"><a name="l00243"></a><span class="lineno"> 243</span>&#160; <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
<div class="line"><a name="l00244"></a><span class="lineno"> 244</span>&#160; }</div>
<div class="line"><a name="l00249"></a><span class="lineno"> 249</span>&#160; <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="l00253"></a><span class="lineno"> 253</span>&#160; <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="l00261"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a1e7e3e95e522a48e9064858bf084be05"> 261</a></span>&#160; <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1Operator.html">Operator</a>* <a class="code" href="classmxnet_1_1OperatorProperty.html#a1e7e3e95e522a48e9064858bf084be05">CreateOperatorEx</a>(<a class="code" href="structmxnet_1_1Context.html">Context</a> ctx,</div>
<div class="line"><a name="l00262"></a><span class="lineno"> 262</span>&#160; <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>* in_shape,</div>
<div class="line"><a name="l00263"></a><span class="lineno"> 263</span>&#160; std::vector&lt;int&gt;* in_type)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00264"></a><span class="lineno"> 264</span>&#160; std::vector&lt;int&gt; out_type, aux_type;</div>
<div class="line"><a name="l00265"></a><span class="lineno"> 265</span>&#160; <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a> out_shape, aux_shape;</div>
<div class="line"><a name="l00266"></a><span class="lineno"> 266</span>&#160; out_type.resize(this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7">ListOutputs</a>().size());</div>
<div class="line"><a name="l00267"></a><span class="lineno"> 267</span>&#160; out_shape.resize(this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7">ListOutputs</a>().size());</div>
<div class="line"><a name="l00268"></a><span class="lineno"> 268</span>&#160; aux_type.resize(this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#aa6d3a955ffa452a368a9dd4d33ce5a25">ListAuxiliaryStates</a>().size());</div>
<div class="line"><a name="l00269"></a><span class="lineno"> 269</span>&#160; aux_shape.resize(this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#aa6d3a955ffa452a368a9dd4d33ce5a25">ListAuxiliaryStates</a>().size());</div>
<div class="line"><a name="l00270"></a><span class="lineno"> 270</span>&#160; CHECK(<a class="code" href="classmxnet_1_1OperatorProperty.html#a819b81e14761e9e8b9c888a63c7df1fa">InferType</a>(in_type, &amp;out_type, &amp;aux_type));</div>
<div class="line"><a name="l00271"></a><span class="lineno"> 271</span>&#160; CHECK(<a class="code" href="classmxnet_1_1OperatorProperty.html#a4be455d921fcb9e61ffafc567a31dc39">InferShape</a>(in_shape, &amp;out_shape, &amp;aux_shape));</div>
<div class="line"><a name="l00272"></a><span class="lineno"> 272</span>&#160; <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1OperatorProperty.html#a546ab25ddaae590ca98705ae9abc7ac0">CreateOperator</a>(ctx);</div>
<div class="line"><a name="l00273"></a><span class="lineno"> 273</span>&#160; }</div>
<div class="line"><a name="l00279"></a><span class="lineno"> 279</span>&#160; <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="l00280"></a><span class="lineno"> 280</span>&#160; <span class="comment">//--------------------------------------------------------</span></div>
<div class="line"><a name="l00281"></a><span class="lineno"> 281</span>&#160; <span class="comment">// All the below functions are optional to override.</span></div>
<div class="line"><a name="l00282"></a><span class="lineno"> 282</span>&#160; <span class="comment">//--------------------------------------------------------</span></div>
<div class="line"><a name="l00290"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a203b8aa89f254f8e7a91b83d3ed18908"> 290</a></span>&#160;<span class="comment"></span> <span class="keyword">virtual</span> std::vector&lt;ResourceRequest&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a203b8aa89f254f8e7a91b83d3ed18908">ForwardResource</a>(<span class="keyword">const</span> <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>&amp; in_shape)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00291"></a><span class="lineno"> 291</span>&#160; <span class="keywordflow">return</span> std::vector&lt;ResourceRequest&gt;();</div>
<div class="line"><a name="l00292"></a><span class="lineno"> 292</span>&#160; }</div>
<div class="line"><a name="l00300"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a84d08b437e69ea9c402725d7f9165ead"> 300</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;ResourceRequest&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a84d08b437e69ea9c402725d7f9165ead">BackwardResource</a>(<span class="keyword">const</span> <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>&amp; in_shape)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00301"></a><span class="lineno"> 301</span>&#160; <span class="keywordflow">return</span> std::vector&lt;ResourceRequest&gt;();</div>
<div class="line"><a name="l00302"></a><span class="lineno"> 302</span>&#160; }</div>
<div class="line"><a name="l00325"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a8aa205ea6ce962bfa3fcbe73a8f3db5d"> 325</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;int&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a8aa205ea6ce962bfa3fcbe73a8f3db5d">DeclareBackwardDependency</a>(<span class="keyword">const</span> std::vector&lt;int&gt;&amp; out_grad,</div>
<div class="line"><a name="l00326"></a><span class="lineno"> 326</span>&#160; <span class="keyword">const</span> std::vector&lt;int&gt;&amp; in_data,</div>
<div class="line"><a name="l00327"></a><span class="lineno"> 327</span>&#160; <span class="keyword">const</span> std::vector&lt;int&gt;&amp; out_data)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00328"></a><span class="lineno"> 328</span>&#160; <span class="comment">// By default requires to see all the things.</span></div>
<div class="line"><a name="l00329"></a><span class="lineno"> 329</span>&#160; <span class="comment">// remember to override this function to get a better performance.</span></div>
<div class="line"><a name="l00330"></a><span class="lineno"> 330</span>&#160; std::vector&lt;int&gt; ret = out_grad;</div>
<div class="line"><a name="l00331"></a><span class="lineno"> 331</span>&#160; ret.insert(ret.end(), in_data.begin(), in_data.end());</div>
<div class="line"><a name="l00332"></a><span class="lineno"> 332</span>&#160; ret.insert(ret.end(), out_data.begin(), out_data.end());</div>
<div class="line"><a name="l00333"></a><span class="lineno"> 333</span>&#160; <span class="keywordflow">return</span> ret;</div>
<div class="line"><a name="l00334"></a><span class="lineno"> 334</span>&#160; }</div>
<div class="line"><a name="l00356"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a91afbc64b53766d2d19236e99d966902"> 356</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;std::pair&lt;int, void*&gt; &gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a91afbc64b53766d2d19236e99d966902">ForwardInplaceOption</a>(</div>
<div class="line"><a name="l00357"></a><span class="lineno"> 357</span>&#160; <span class="keyword">const</span> std::vector&lt;int&gt;&amp; in_data,</div>
<div class="line"><a name="l00358"></a><span class="lineno"> 358</span>&#160; <span class="keyword">const</span> std::vector&lt;void*&gt;&amp; out_data)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00359"></a><span class="lineno"> 359</span>&#160; <span class="keywordflow">return</span> std::vector&lt;std::pair&lt;int, void*&gt; &gt;();</div>
<div class="line"><a name="l00360"></a><span class="lineno"> 360</span>&#160; }</div>
<div class="line"><a name="l00387"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a2c3c0366154cd257a3b76bbcf924ef80"> 387</a></span>&#160; <span class="keyword">virtual</span> std::vector&lt;std::pair&lt;int, void*&gt; &gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a2c3c0366154cd257a3b76bbcf924ef80">BackwardInplaceOption</a>(</div>
<div class="line"><a name="l00388"></a><span class="lineno"> 388</span>&#160; <span class="keyword">const</span> std::vector&lt;int&gt;&amp; out_grad,</div>
<div class="line"><a name="l00389"></a><span class="lineno"> 389</span>&#160; <span class="keyword">const</span> std::vector&lt;int&gt;&amp; in_data,</div>
<div class="line"><a name="l00390"></a><span class="lineno"> 390</span>&#160; <span class="keyword">const</span> std::vector&lt;int&gt;&amp; out_data,</div>
<div class="line"><a name="l00391"></a><span class="lineno"> 391</span>&#160; <span class="keyword">const</span> std::vector&lt;void*&gt;&amp; in_grad)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00392"></a><span class="lineno"> 392</span>&#160; <span class="keywordflow">return</span> std::vector&lt;std::pair&lt;int, void*&gt; &gt;();</div>
<div class="line"><a name="l00393"></a><span class="lineno"> 393</span>&#160; }</div>
<div class="line"><a name="l00406"></a><span class="lineno"> 406</span>&#160; <span class="keyword">template</span> &lt;<span class="keyword">typename</span> T&gt;</div>
<div class="line"><a name="l00407"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a6c5c8df10748a7eee873d4b7eb8d684b"> 407</a></span>&#160; <span class="keyword">inline</span> std::vector&lt;T&gt; <a class="code" href="classmxnet_1_1OperatorProperty.html#a6c5c8df10748a7eee873d4b7eb8d684b">BackwardInputs</a>(<span class="keyword">const</span> std::vector&lt;T&gt;&amp; out_grad,</div>
<div class="line"><a name="l00408"></a><span class="lineno"> 408</span>&#160; <span class="keyword">const</span> std::vector&lt;T&gt;&amp; in_data,</div>
<div class="line"><a name="l00409"></a><span class="lineno"> 409</span>&#160; <span class="keyword">const</span> std::vector&lt;T&gt;&amp; out_data)<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00410"></a><span class="lineno"> 410</span>&#160; <span class="keywordtype">int</span> counter = 0;</div>
<div class="line"><a name="l00411"></a><span class="lineno"> 411</span>&#160; std::vector&lt;int&gt; out_grad_index(out_grad.size());</div>
<div class="line"><a name="l00412"></a><span class="lineno"> 412</span>&#160; std::vector&lt;int&gt; in_data_index(in_data.size());</div>
<div class="line"><a name="l00413"></a><span class="lineno"> 413</span>&#160; std::vector&lt;int&gt; out_data_index(out_data.size());</div>
<div class="line"><a name="l00414"></a><span class="lineno"> 414</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; out_grad_index.size(); ++i) {</div>
<div class="line"><a name="l00415"></a><span class="lineno"> 415</span>&#160; out_grad_index[i] = counter++;</div>
<div class="line"><a name="l00416"></a><span class="lineno"> 416</span>&#160; }</div>
<div class="line"><a name="l00417"></a><span class="lineno"> 417</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; in_data_index.size(); ++i) {</div>
<div class="line"><a name="l00418"></a><span class="lineno"> 418</span>&#160; in_data_index[i] = counter++;</div>
<div class="line"><a name="l00419"></a><span class="lineno"> 419</span>&#160; }</div>
<div class="line"><a name="l00420"></a><span class="lineno"> 420</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; out_data_index.size(); ++i) {</div>
<div class="line"><a name="l00421"></a><span class="lineno"> 421</span>&#160; out_data_index[i] = counter++;</div>
<div class="line"><a name="l00422"></a><span class="lineno"> 422</span>&#160; }</div>
<div class="line"><a name="l00423"></a><span class="lineno"> 423</span>&#160; std::vector&lt;T&gt; all_data;</div>
<div class="line"><a name="l00424"></a><span class="lineno"> 424</span>&#160; all_data.insert(all_data.end(), out_grad.begin(), out_grad.end());</div>
<div class="line"><a name="l00425"></a><span class="lineno"> 425</span>&#160; all_data.insert(all_data.end(), in_data.begin(), in_data.end());</div>
<div class="line"><a name="l00426"></a><span class="lineno"> 426</span>&#160; all_data.insert(all_data.end(), out_data.begin(), out_data.end());</div>
<div class="line"><a name="l00427"></a><span class="lineno"> 427</span>&#160; </div>
<div class="line"><a name="l00428"></a><span class="lineno"> 428</span>&#160; std::vector&lt;int&gt; ret_index =</div>
<div class="line"><a name="l00429"></a><span class="lineno"> 429</span>&#160; this-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#a8aa205ea6ce962bfa3fcbe73a8f3db5d">DeclareBackwardDependency</a>(out_grad_index, in_data_index, out_data_index);</div>
<div class="line"><a name="l00430"></a><span class="lineno"> 430</span>&#160; </div>
<div class="line"><a name="l00431"></a><span class="lineno"> 431</span>&#160; std::vector&lt;T&gt; ret(ret_index.size());</div>
<div class="line"><a name="l00432"></a><span class="lineno"> 432</span>&#160; <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; ret_index.size(); ++i) {</div>
<div class="line"><a name="l00433"></a><span class="lineno"> 433</span>&#160; ret[i] = all_data[ret_index[i]];</div>
<div class="line"><a name="l00434"></a><span class="lineno"> 434</span>&#160; }</div>
<div class="line"><a name="l00435"></a><span class="lineno"> 435</span>&#160; <span class="keywordflow">return</span> ret;</div>
<div class="line"><a name="l00436"></a><span class="lineno"> 436</span>&#160; }</div>
<div class="line"><a name="l00442"></a><span class="lineno"> 442</span>&#160; <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>* <a class="code" href="namespacedmlc.html#a1e731f276d3a3efa8e17f46ccf35f828">type_name</a>);</div>
<div class="line"><a name="l00444"></a><span class="lineno"><a class="line" href="classmxnet_1_1OperatorProperty.html#a149c52bd285e4fa78109a535765a9b7c"> 444</a></span>&#160; <span class="keyword">virtual</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad">ExecType</a> <a class="code" href="classmxnet_1_1OperatorProperty.html#a149c52bd285e4fa78109a535765a9b7c">exec_type</a>()<span class="keyword"> const </span>{</div>
<div class="line"><a name="l00445"></a><span class="lineno"> 445</span>&#160; <span class="keywordflow">return</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c">ExecType::kSync</a>;</div>
<div class="line"><a name="l00446"></a><span class="lineno"> 446</span>&#160; }</div>
<div class="line"><a name="l00447"></a><span class="lineno"> 447</span>&#160;};</div>
<div class="line"><a name="l00448"></a><span class="lineno"> 448</span>&#160; </div>
<div class="line"><a name="l00450"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ae4a8cd5223645e84d61318401dd9a4e0"> 450</a></span>&#160;<span class="keyword">typedef</span> std::function&lt;<a class="code" href="classmxnet_1_1OperatorProperty.html">OperatorProperty</a>*()&gt; <a class="code" href="namespacemxnet.html#ae4a8cd5223645e84d61318401dd9a4e0">OperatorPropertyFactory</a>;</div>
<div class="line"><a name="l00454"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html"> 454</a></span>&#160;<span class="keyword">struct </span><a class="code" href="structmxnet_1_1OperatorPropertyReg.html">OperatorPropertyReg</a></div>
<div class="line"><a name="l00455"></a><span class="lineno"> 455</span>&#160; : <span class="keyword">public</span> <a class="code" href="classdmlc_1_1FunctionRegEntryBase.html">dmlc::FunctionRegEntryBase</a>&lt;OperatorPropertyReg, OperatorPropertyFactory&gt; {</div>
<div class="line"><a name="l00468"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html#a0e3536ad5305fa1ef402c5385fb4a089"> 468</a></span>&#160; <span class="keyword">inline</span> <a class="code" href="structmxnet_1_1OperatorPropertyReg.html">OperatorPropertyReg</a>&amp; <a class="code" href="structmxnet_1_1OperatorPropertyReg.html#a0e3536ad5305fa1ef402c5385fb4a089">set_key_var_num_args</a>(<span class="keyword">const</span> std::string&amp; key) { <span class="comment">// NOLINT(*)</span></div>
<div class="line"><a name="l00469"></a><span class="lineno"> 469</span>&#160; this-&gt;<a class="code" href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c">key_var_num_args</a> = key;</div>
<div class="line"><a name="l00470"></a><span class="lineno"> 470</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00471"></a><span class="lineno"> 471</span>&#160; }</div>
<div class="line"><a name="l00475"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html#ad959fdbefb6dd807ee1961a2fec7a038"> 475</a></span>&#160; <span class="keyword">inline</span> <a class="code" href="structmxnet_1_1OperatorPropertyReg.html">OperatorPropertyReg</a>&amp; <a class="code" href="structmxnet_1_1OperatorPropertyReg.html#ad959fdbefb6dd807ee1961a2fec7a038">check_name</a>() {</div>
<div class="line"><a name="l00476"></a><span class="lineno"> 476</span>&#160; <a class="code" href="classmxnet_1_1OperatorProperty.html">OperatorProperty</a>* p = this-&gt;<a class="code" href="classdmlc_1_1FunctionRegEntryBase.html#ac593e3ad3722e750b01b110ae01dd5ee">body</a>();</div>
<div class="line"><a name="l00477"></a><span class="lineno"> 477</span>&#160; std::string type = p-&gt;<a class="code" href="classmxnet_1_1OperatorProperty.html#ac6b7df90c0285a1823b463c21e926698">TypeString</a>();</div>
<div class="line"><a name="l00478"></a><span class="lineno"> 478</span>&#160; <span class="keyword">delete</span> p;</div>
<div class="line"><a name="l00479"></a><span class="lineno"> 479</span>&#160; CHECK_EQ(this-&gt;<a class="code" href="classdmlc_1_1FunctionRegEntryBase.html#af2acc06f141492afe85d4fff777d7f31">name</a>, type) &lt;&lt; <span class="stringliteral">&quot;Register Name and TypeString mismatch, name=\&quot;&quot;</span> &lt;&lt; this-&gt;<a class="code" href="classdmlc_1_1FunctionRegEntryBase.html#af2acc06f141492afe85d4fff777d7f31">name</a></div>
<div class="line"><a name="l00480"></a><span class="lineno"> 480</span>&#160; &lt;&lt; <span class="stringliteral">&quot;\&quot;,&quot;</span></div>
<div class="line"><a name="l00481"></a><span class="lineno"> 481</span>&#160; &lt;&lt; <span class="stringliteral">&quot; but TypeString=\&quot;&quot;</span> &lt;&lt; type &lt;&lt; <span class="stringliteral">&quot;\&quot;&quot;</span>;</div>
<div class="line"><a name="l00482"></a><span class="lineno"> 482</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00483"></a><span class="lineno"> 483</span>&#160; }</div>
<div class="line"><a name="l00484"></a><span class="lineno"> 484</span>&#160; </div>
<div class="line"><a name="l00486"></a><span class="lineno"><a class="line" href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c"> 486</a></span>&#160; std::string <a class="code" href="structmxnet_1_1OperatorPropertyReg.html#af3aee96c6a2289e7554d7503186a540c">key_var_num_args</a>;</div>
<div class="line"><a name="l00487"></a><span class="lineno"> 487</span>&#160;};</div>
<div class="line"><a name="l00488"></a><span class="lineno"> 488</span>&#160; </div>
<div class="line"><a name="l00489"></a><span class="lineno"> 489</span>&#160;<span class="comment">//---------------------------------------------------------------------------------</span></div>
<div class="line"><a name="l00490"></a><span class="lineno"> 490</span>&#160;<span class="comment">// The following part are API Registration of Operators</span></div>
<div class="line"><a name="l00491"></a><span class="lineno"> 491</span>&#160;<span class="comment">// See also MXNET_REGISTER_SIMPLE_OP in operator_util.h for registering simple ops.</span></div>
<div class="line"><a name="l00492"></a><span class="lineno"> 492</span>&#160;<span class="comment">//---------------------------------------------------------------------------------</span></div>
<div class="line"><a name="l00503"></a><span class="lineno"><a class="line" href="include_2mxnet_2operator_8h.html#a926db042759256e94d20696c66f65709"> 503</a></span>&#160;<span class="comment"></span><span class="preprocessor">#define MXNET_REGISTER_OP_PROPERTY(name, OperatorPropertyType) \</span></div>
<div class="line"><a name="l00504"></a><span class="lineno"> 504</span>&#160;<span class="preprocessor"> DMLC_REGISTRY_REGISTER(::mxnet::OperatorPropertyReg, OperatorPropertyReg, name) \</span></div>
<div class="line"><a name="l00505"></a><span class="lineno"> 505</span>&#160;<span class="preprocessor"> .set_body([]() { return new OperatorPropertyType(); }) \</span></div>
<div class="line"><a name="l00506"></a><span class="lineno"> 506</span>&#160;<span class="preprocessor"> .set_return_type(&quot;NDArray-or-Symbol&quot;) \</span></div>
<div class="line"><a name="l00507"></a><span class="lineno"> 507</span>&#160;<span class="preprocessor"> .check_name()</span></div>
<div class="line"><a name="l00508"></a><span class="lineno"> 508</span>&#160; </div>
<div class="line"><a name="l00509"></a><span class="lineno"> 509</span>&#160;<span class="preprocessor">#endif // DMLC_USE_CXX11</span></div>
<div class="line"><a name="l00510"></a><span class="lineno"> 510</span>&#160;} <span class="comment">// namespace mxnet</span></div>
<div class="line"><a name="l00511"></a><span class="lineno"> 511</span>&#160;<span class="preprocessor">#endif // MXNET_OPERATOR_H_</span></div>
</div><!-- fragment --></div><!-- contents -->
<div class="ttc" id="anamespacemxnet_html"><div class="ttname"><a href="namespacemxnet.html">mxnet</a></div><div class="ttdoc">namespace of mxnet</div><div class="ttdef"><b>Definition:</b> api_registry.h:33</div></div>
<div class="ttc" id="aclassmxnet_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&lt; std::string, std::string &gt; 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="anamespacemshadow_html_a143ad1a73ba518ef0aad29d36f2569ee"><div class="ttname"><a href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">mshadow::default_type_flag</a></div><div class="ttdeci">const int default_type_flag</div><div class="ttdoc">type enum value for default real type</div><div class="ttdef"><b>Definition:</b> base.h:492</div></div>
<div class="ttc" id="aclassdmlc_1_1FunctionRegEntryBase_html"><div class="ttname"><a href="classdmlc_1_1FunctionRegEntryBase.html">dmlc::FunctionRegEntryBase</a></div><div class="ttdoc">Common base class for function registry.</div><div class="ttdef"><b>Definition:</b> registry.h:151</div></div>
<div class="ttc" id="aclassmxnet_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:127</div></div>
<div class="ttc" id="astructmxnet_1_1OperatorPropertyReg_html_ad959fdbefb6dd807ee1961a2fec7a038"><div class="ttname"><a href="structmxnet_1_1OperatorPropertyReg.html#ad959fdbefb6dd807ee1961a2fec7a038">mxnet::OperatorPropertyReg::check_name</a></div><div class="ttdeci">OperatorPropertyReg &amp; 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:475</div></div>
<div class="ttc" id="aclassmxnet_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&lt; std::pair&lt; std::string, std::string &gt; &gt; &amp;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="aclassmxnet_1_1OperatorProperty_html_a819b81e14761e9e8b9c888a63c7df1fa"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a819b81e14761e9e8b9c888a63c7df1fa">mxnet::OperatorProperty::InferType</a></div><div class="ttdeci">virtual bool InferType(std::vector&lt; int &gt; *in_type, std::vector&lt; int &gt; *out_type, std::vector&lt; int &gt; *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:221</div></div>
<div class="ttc" id="aclassmxnet_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:54</div></div>
<div class="ttc" id="aclassmxnet_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="aclassmxnet_1_1OperatorProperty_html_a149c52bd285e4fa78109a535765a9b7c"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a149c52bd285e4fa78109a535765a9b7c">mxnet::OperatorProperty::exec_type</a></div><div class="ttdeci">virtual ExecType exec_type() const</div><div class="ttdef"><b>Definition:</b> operator.h:444</div></div>
<div class="ttc" id="a3rdparty_2dmlc-core_2include_2dmlc_2base_8h_html"><div class="ttname"><a href="3rdparty_2dmlc-core_2include_2dmlc_2base_8h.html">base.h</a></div><div class="ttdoc">defines configuration macros</div></div>
<div class="ttc" id="aclassmxnet_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 class="ttc" id="aclassmxnet_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:132</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a4be455d921fcb9e61ffafc567a31dc39"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a4be455d921fcb9e61ffafc567a31dc39">mxnet::OperatorProperty::InferShape</a></div><div class="ttdeci">virtual bool InferShape(mxnet::ShapeVector *in_shape, mxnet::ShapeVector *out_shape, mxnet::ShapeVector *aux_shape) const =0</div><div class="ttdoc">infer the shapes of outputs and unknown input arguments</div></div>
<div class="ttc" id="astructmxnet_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. This is the superset of RunContext....</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:66</div></div>
<div class="ttc" id="astructmxnet_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:454</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a9fc6e9f04f52b1c05f62fb7357bb95c4"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a9fc6e9f04f52b1c05f62fb7357bb95c4">mxnet::OperatorProperty::NumOutputs</a></div><div class="ttdeci">virtual int NumOutputs() const</div><div class="ttdef"><b>Definition:</b> operator.h:166</div></div>
<div class="ttc" id="astructmxnet_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 &amp; set_key_var_num_args(const std::string &amp;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:468</div></div>
<div class="ttc" id="aclassmxnet_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 &amp;ctx, const std::vector&lt; TBlob &gt; &amp;in_data, const std::vector&lt; OpReqType &gt; &amp;req, const std::vector&lt; TBlob &gt; &amp;out_data, const std::vector&lt; TBlob &gt; &amp;aux_states)=0</div><div class="ttdoc">perform a forward operation of Operator, save the output to TBlob.</div></div>
<div class="ttc" id="anamespacedmlc_html_a1e731f276d3a3efa8e17f46ccf35f828"><div class="ttname"><a href="namespacedmlc.html#a1e731f276d3a3efa8e17f46ccf35f828">dmlc::type_name</a></div><div class="ttdeci">std::string type_name()</div><div class="ttdoc">the string representation of type name</div><div class="ttdef"><b>Definition:</b> type_traits.h:101</div></div>
<div class="ttc" id="anamespacemxnet_html_ae4a8cd5223645e84d61318401dd9a4e0"><div class="ttname"><a href="namespacemxnet.html#ae4a8cd5223645e84d61318401dd9a4e0">mxnet::OperatorPropertyFactory</a></div><div class="ttdeci">std::function&lt; OperatorProperty *()&gt; OperatorPropertyFactory</div><div class="ttdoc">typedef the factory function of operator property</div><div class="ttdef"><b>Definition:</b> operator.h:450</div></div>
<div class="ttc" id="aclassmxnet_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="aclassmxnet_1_1OperatorProperty_html_a8aa205ea6ce962bfa3fcbe73a8f3db5d"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a8aa205ea6ce962bfa3fcbe73a8f3db5d">mxnet::OperatorProperty::DeclareBackwardDependency</a></div><div class="ttdeci">virtual std::vector&lt; int &gt; DeclareBackwardDependency(const std::vector&lt; int &gt; &amp;out_grad, const std::vector&lt; int &gt; &amp;in_data, const std::vector&lt; int &gt; &amp;out_data) const</div><div class="ttdoc">Declare the input requirement of Backward pass.</div><div class="ttdef"><b>Definition:</b> operator.h:325</div></div>
<div class="ttc" id="aresource_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="anamespacemxnet_html_ad7801fb2dd68294d4c7a9e289bc071ad"><div class="ttname"><a href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad">mxnet::ExecType</a></div><div class="ttdeci">ExecType</div><div class="ttdoc">the execution type of the operator</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:98</div></div>
<div class="ttc" id="anamespacemxnet_html_ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c"><div class="ttname"><a href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c">mxnet::ExecType::kSync</a></div><div class="ttdeci">@ kSync</div><div class="ttdoc">Forward/Backward are synchronous calls.</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_ab9e52badb93fe965ce961803f2101aa7"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#ab9e52badb93fe965ce961803f2101aa7">mxnet::OperatorProperty::ListOutputs</a></div><div class="ttdeci">virtual std::vector&lt; std::string &gt; ListOutputs() const</div><div class="ttdoc">Get name of output values of Operator.</div><div class="ttdef"><b>Definition:</b> operator.h:155</div></div>
<div class="ttc" id="astructmxnet_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:90</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a2c3c0366154cd257a3b76bbcf924ef80"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a2c3c0366154cd257a3b76bbcf924ef80">mxnet::OperatorProperty::BackwardInplaceOption</a></div><div class="ttdeci">virtual std::vector&lt; std::pair&lt; int, void * &gt; &gt; BackwardInplaceOption(const std::vector&lt; int &gt; &amp;out_grad, const std::vector&lt; int &gt; &amp;in_data, const std::vector&lt; int &gt; &amp;out_data, const std::vector&lt; void * &gt; &amp;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:387</div></div>
<div class="ttc" id="ainclude_2mxnet_2op__attr__types_8h_html"><div class="ttname"><a href="include_2mxnet_2op__attr__types_8h.html">op_attr_types.h</a></div><div class="ttdoc">Additional operator attributes beside the ones provided by NNVM.</div></div>
<div class="ttc" id="astructmxnet_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:486</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a91afbc64b53766d2d19236e99d966902"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a91afbc64b53766d2d19236e99d966902">mxnet::OperatorProperty::ForwardInplaceOption</a></div><div class="ttdeci">virtual std::vector&lt; std::pair&lt; int, void * &gt; &gt; ForwardInplaceOption(const std::vector&lt; int &gt; &amp;in_data, const std::vector&lt; void * &gt; &amp;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:356</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_ad01be3a84423800af7e818b45afc748f"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#ad01be3a84423800af7e818b45afc748f">mxnet::OperatorProperty::ListArguments</a></div><div class="ttdeci">virtual std::vector&lt; std::string &gt; ListArguments() const</div><div class="ttdoc">Get input arguments of the Operator.</div><div class="ttdef"><b>Definition:</b> operator.h:148</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a006f0e2c9b3f934e1f73b242b7e8b062"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a006f0e2c9b3f934e1f73b242b7e8b062">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:181</div></div>
<div class="ttc" id="aclassmxnet_1_1Operator_html_aa6b4c247a914b4659c773965248019c7"><div class="ttname"><a href="classmxnet_1_1Operator.html#aa6b4c247a914b4659c773965248019c7">mxnet::Operator::exec_type</a></div><div class="ttdeci">virtual ExecType exec_type() const final</div><div class="ttdef"><b>Definition:</b> operator.h:112</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a84d08b437e69ea9c402725d7f9165ead"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a84d08b437e69ea9c402725d7f9165ead">mxnet::OperatorProperty::BackwardResource</a></div><div class="ttdeci">virtual std::vector&lt; ResourceRequest &gt; BackwardResource(const mxnet::ShapeVector &amp;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:300</div></div>
<div class="ttc" id="anamespacemxnet_html_a87e1c30aafac389774c6f4d905c0e5ea"><div class="ttname"><a href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a></div><div class="ttdeci">std::vector&lt; mxnet::TShape &gt; ShapeVector</div><div class="ttdoc">The result holder of shape of each NodeEntry in the graph.</div><div class="ttdef"><b>Definition:</b> tuple.h:830</div></div>
<div class="ttc" id="ajson_8h_html"><div class="ttname"><a href="json_8h.html">json.h</a></div><div class="ttdoc">Lightweight JSON Reader/Writer that read save into C++ data structs. This includes STL composites and...</div></div>
<div class="ttc" id="aclassdmlc_1_1FunctionRegEntryBase_html_ac593e3ad3722e750b01b110ae01dd5ee"><div class="ttname"><a href="classdmlc_1_1FunctionRegEntryBase.html#ac593e3ad3722e750b01b110ae01dd5ee">dmlc::FunctionRegEntryBase&lt; OperatorPropertyReg, OperatorPropertyFactory &gt;::body</a></div><div class="ttdeci">OperatorPropertyFactory body</div><div class="ttdoc">Function body to create ProductType.</div><div class="ttdef"><b>Definition:</b> registry.h:160</div></div>
<div class="ttc" id="a3rdparty_2dmlc-core_2include_2dmlc_2registry_8h_html"><div class="ttname"><a href="3rdparty_2dmlc-core_2include_2dmlc_2registry_8h.html">registry.h</a></div><div class="ttdoc">Registry utility that helps to build registry singletons.</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="aclassmxnet_1_1OperatorProperty_html_a1e7e3e95e522a48e9064858bf084be05"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a1e7e3e95e522a48e9064858bf084be05">mxnet::OperatorProperty::CreateOperatorEx</a></div><div class="ttdeci">virtual Operator * CreateOperatorEx(Context ctx, mxnet::ShapeVector *in_shape, std::vector&lt; int &gt; *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:261</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a6c5c8df10748a7eee873d4b7eb8d684b"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a6c5c8df10748a7eee873d4b7eb8d684b">mxnet::OperatorProperty::BackwardInputs</a></div><div class="ttdeci">std::vector&lt; T &gt; BackwardInputs(const std::vector&lt; T &gt; &amp;out_grad, const std::vector&lt; T &gt; &amp;in_data, const std::vector&lt; T &gt; &amp;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:407</div></div>
<div class="ttc" id="aclassmxnet_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:57</div></div>
<div class="ttc" id="aclassmxnet_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="aclassmxnet_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 &amp;ctx, const std::vector&lt; TBlob &gt; &amp;out_grad, const std::vector&lt; TBlob &gt; &amp;in_data, const std::vector&lt; TBlob &gt; &amp;out_data, const std::vector&lt; OpReqType &gt; &amp;req, const std::vector&lt; TBlob &gt; &amp;in_grad, const std::vector&lt; TBlob &gt; &amp;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:102</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_a203b8aa89f254f8e7a91b83d3ed18908"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#a203b8aa89f254f8e7a91b83d3ed18908">mxnet::OperatorProperty::ForwardResource</a></div><div class="ttdeci">virtual std::vector&lt; ResourceRequest &gt; ForwardResource(const mxnet::ShapeVector &amp;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:290</div></div>
<div class="ttc" id="aclassdmlc_1_1FunctionRegEntryBase_html_af2acc06f141492afe85d4fff777d7f31"><div class="ttname"><a href="classdmlc_1_1FunctionRegEntryBase.html#af2acc06f141492afe85d4fff777d7f31">dmlc::FunctionRegEntryBase&lt; OperatorPropertyReg, OperatorPropertyFactory &gt;::name</a></div><div class="ttdeci">std::string name</div><div class="ttdoc">name of the entry</div><div class="ttdef"><b>Definition:</b> registry.h:154</div></div>
<div class="ttc" id="aclassmxnet_1_1OperatorProperty_html_aa6d3a955ffa452a368a9dd4d33ce5a25"><div class="ttname"><a href="classmxnet_1_1OperatorProperty.html#aa6d3a955ffa452a368a9dd4d33ce5a25">mxnet::OperatorProperty::ListAuxiliaryStates</a></div><div class="ttdeci">virtual std::vector&lt; std::string &gt; ListAuxiliaryStates() const</div><div class="ttdoc">Get name of auxiliary states of Operator.</div><div class="ttdef"><b>Definition:</b> operator.h:162</div></div>
<!-- start footer part -->
<hr class="footer"/><address class="footer"><small>
Generated on Thu Jan 5 2023 03:47:39 for mxnet by &#160;<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/>
</a> 1.8.17
</small></address>
</body>
</html>