Google Git
Sign in
apache / mxnet-site / d5c5a235eef72f64ccaf224b834a20ef3287469b / . / versions / master / api / cpp / docs / api / cpp-package_2include_2mxnet-cpp_2operator_8h_source.html
blob: 8d892e019961b393c3d1651fe6d3a846494edb52 [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/cpp-package/include/mxnet-cpp/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_c9bbcf2f7c4ed40b1b23631ec8abffa6.html">cpp-package</a></li><li class="navelem"><a class="el" href="dir_eb07344824255b98f70969533c25e736.html">include</a></li><li class="navelem"><a class="el" href="dir_0327d24fd86e9ce22567ad8f859e5290.html">mxnet-cpp</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="cpp-package_2include_2mxnet-cpp_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="l00026"></a><span class="lineno"> 26</span>&#160;<span class="preprocessor">#ifndef MXNET_CPP_OPERATOR_H_</span></div>
<div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160;<span class="preprocessor">#define MXNET_CPP_OPERATOR_H_</span></div>
<div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; </div>
<div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160;<span class="preprocessor">#include &lt;map&gt;</span></div>
<div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160;<span class="preprocessor">#include &lt;string&gt;</span></div>
<div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160;<span class="preprocessor">#include &lt;vector&gt;</span></div>
<div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="cpp-package_2include_2mxnet-cpp_2base_8h.html">mxnet-cpp/base.h</a>&quot;</span></div>
<div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="op__map_8h.html">mxnet-cpp/op_map.h</a>&quot;</span></div>
<div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="symbol_8h.html">mxnet-cpp/symbol.h</a>&quot;</span></div>
<div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; </div>
<div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160;<span class="keyword">namespace </span><a class="code" href="namespacemxnet.html">mxnet</a> {</div>
<div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160;<span class="keyword">namespace </span>cpp {</div>
<div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160;<span class="keyword">class </span>Mxnet;</div>
<div class="line"><a name="l00042"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html"> 42</a></span>&#160;<span class="keyword">class </span><a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a> {</div>
<div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; <span class="keyword">public</span>:</div>
<div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#ae6321f03d02b6d076d4154d353b1c4a5">Operator</a>(<span class="keyword">const</span> std::string&amp; operator_name);</div>
<div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#ae0057800586b01402bd99ae62e5614ef">operator=</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; rhs);</div>
<div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; <span class="keyword">template</span> &lt;<span class="keyword">typename</span> T&gt;</div>
<div class="line"><a name="l00057"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a72e6940d1b4be12bbbba0bccc00ef739"> 57</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a72e6940d1b4be12bbbba0bccc00ef739">SetParam</a>(<span class="keyword">const</span> std::string&amp; name, <span class="keyword">const</span> T&amp; value) {</div>
<div class="line"><a name="l00058"></a><span class="lineno"> 58</span>&#160; std::string value_str;</div>
<div class="line"><a name="l00059"></a><span class="lineno"> 59</span>&#160; std::stringstream ss;</div>
<div class="line"><a name="l00060"></a><span class="lineno"> 60</span>&#160; ss &lt;&lt; value;</div>
<div class="line"><a name="l00061"></a><span class="lineno"> 61</span>&#160; ss &gt;&gt; value_str;</div>
<div class="line"><a name="l00062"></a><span class="lineno"> 62</span>&#160; </div>
<div class="line"><a name="l00063"></a><span class="lineno"> 63</span>&#160; params_[name] = value_str;</div>
<div class="line"><a name="l00064"></a><span class="lineno"> 64</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00065"></a><span class="lineno"> 65</span>&#160; }</div>
<div class="line"><a name="l00072"></a><span class="lineno"> 72</span>&#160; <span class="keyword">template</span> &lt;<span class="keyword">typename</span> T&gt;</div>
<div class="line"><a name="l00073"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a280201370ff22876cb38afbe47861f42"> 73</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a280201370ff22876cb38afbe47861f42">SetParam</a>(<span class="keywordtype">int</span> pos, <span class="keyword">const</span> T&amp; value) {</div>
<div class="line"><a name="l00074"></a><span class="lineno"> 74</span>&#160; std::string value_str;</div>
<div class="line"><a name="l00075"></a><span class="lineno"> 75</span>&#160; std::stringstream ss;</div>
<div class="line"><a name="l00076"></a><span class="lineno"> 76</span>&#160; ss &lt;&lt; value;</div>
<div class="line"><a name="l00077"></a><span class="lineno"> 77</span>&#160; ss &gt;&gt; value_str;</div>
<div class="line"><a name="l00078"></a><span class="lineno"> 78</span>&#160; </div>
<div class="line"><a name="l00079"></a><span class="lineno"> 79</span>&#160; params_[arg_names_[pos]] = value_str;</div>
<div class="line"><a name="l00080"></a><span class="lineno"> 80</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00081"></a><span class="lineno"> 81</span>&#160; }</div>
<div class="line"><a name="l00088"></a><span class="lineno"> 88</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#afb1cbbf2fba12676de09dcb64f486464">SetInput</a>(<span class="keyword">const</span> std::string&amp; name, <span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1Symbol.html">Symbol</a>&amp; symbol);</div>
<div class="line"><a name="l00093"></a><span class="lineno"> 93</span>&#160; <span class="keyword">template</span> &lt;<span class="keywordtype">int</span> N = 0&gt;</div>
<div class="line"><a name="l00094"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a33b42e657f9a15ab4004b3b8226d71c6"> 94</a></span>&#160; <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a33b42e657f9a15ab4004b3b8226d71c6">PushInput</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1Symbol.html">Symbol</a>&amp; symbol) {</div>
<div class="line"><a name="l00095"></a><span class="lineno"> 95</span>&#160; input_symbols_.push_back(symbol.<a class="code" href="classmxnet_1_1cpp_1_1Symbol.html#a7a4d4a4cc1309356c81ed8cdd4919580">GetHandle</a>());</div>
<div class="line"><a name="l00096"></a><span class="lineno"> 96</span>&#160; }</div>
<div class="line"><a name="l00101"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#ac3c8a454dd7e27358d29f94f17713f18"> 101</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#ac3c8a454dd7e27358d29f94f17713f18">operator()</a>() {</div>
<div class="line"><a name="l00102"></a><span class="lineno"> 102</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160; }</div>
<div class="line"><a name="l00109"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a453b9eee13ca4e8acd3408de03aeafbe"> 109</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a453b9eee13ca4e8acd3408de03aeafbe">operator()</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1Symbol.html">Symbol</a>&amp; symbol) {</div>
<div class="line"><a name="l00110"></a><span class="lineno"> 110</span>&#160; input_symbols_.push_back(symbol.<a class="code" href="classmxnet_1_1cpp_1_1Symbol.html#a7a4d4a4cc1309356c81ed8cdd4919580">GetHandle</a>());</div>
<div class="line"><a name="l00111"></a><span class="lineno"> 111</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00112"></a><span class="lineno"> 112</span>&#160; }</div>
<div class="line"><a name="l00118"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a6911a78584ab48e8f2e808552e678992"> 118</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a6911a78584ab48e8f2e808552e678992">operator()</a>(<span class="keyword">const</span> std::vector&lt;Symbol&gt;&amp; symbols) {</div>
<div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160; <span class="keywordflow">for</span> (<span class="keyword">auto</span>&amp; s : symbols) {</div>
<div class="line"><a name="l00120"></a><span class="lineno"> 120</span>&#160; input_symbols_.push_back(s.GetHandle());</div>
<div class="line"><a name="l00121"></a><span class="lineno"> 121</span>&#160; }</div>
<div class="line"><a name="l00122"></a><span class="lineno"> 122</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00123"></a><span class="lineno"> 123</span>&#160; }</div>
<div class="line"><a name="l00129"></a><span class="lineno"> 129</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Symbol.html">Symbol</a> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a52943207a5d771a32f7e1a861e34d0e3">CreateSymbol</a>(<span class="keyword">const</span> std::string&amp; name = <span class="stringliteral">&quot;&quot;</span>);</div>
<div class="line"><a name="l00130"></a><span class="lineno"> 130</span>&#160; </div>
<div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#afb1cbbf2fba12676de09dcb64f486464">SetInput</a>(<span class="keyword">const</span> std::string&amp; name, <span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1NDArray.html">NDArray</a>&amp; ndarray);</div>
<div class="line"><a name="l00142"></a><span class="lineno"> 142</span>&#160; <span class="keyword">template</span> &lt;<span class="keywordtype">int</span> N = 0&gt;</div>
<div class="line"><a name="l00143"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#acf21af976feee8c84632c6810632acdb"> 143</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#acf21af976feee8c84632c6810632acdb">PushInput</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1NDArray.html">NDArray</a>&amp; ndarray) {</div>
<div class="line"><a name="l00144"></a><span class="lineno"> 144</span>&#160; input_ndarrays_.push_back(ndarray.<a class="code" href="classmxnet_1_1cpp_1_1NDArray.html#a4ee101e6e73626382af486003048cae0">GetHandle</a>());</div>
<div class="line"><a name="l00145"></a><span class="lineno"> 145</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00146"></a><span class="lineno"> 146</span>&#160; }</div>
<div class="line"><a name="l00150"></a><span class="lineno"> 150</span>&#160; <span class="keyword">template</span> &lt;<span class="keyword">class </span>T, <span class="keyword">class</span>... Args, <span class="keywordtype">int</span> N = 0&gt;</div>
<div class="line"><a name="l00151"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a537e8d915b101757819a7a01da956a6b"> 151</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a537e8d915b101757819a7a01da956a6b">PushInput</a>(<span class="keyword">const</span> T&amp; t, Args... args) {</div>
<div class="line"><a name="l00152"></a><span class="lineno"> 152</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a72e6940d1b4be12bbbba0bccc00ef739">SetParam</a>(N, t);</div>
<div class="line"><a name="l00153"></a><span class="lineno"> 153</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a33b42e657f9a15ab4004b3b8226d71c6">PushInput</a>&lt;Args..., N + 1&gt;(args...);</div>
<div class="line"><a name="l00154"></a><span class="lineno"> 154</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00155"></a><span class="lineno"> 155</span>&#160; }</div>
<div class="line"><a name="l00159"></a><span class="lineno"> 159</span>&#160; <span class="keyword">template</span> &lt;<span class="keyword">class</span> T, <span class="keywordtype">int</span> N = 0&gt;</div>
<div class="line"><a name="l00160"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#afc7e1d919209cf861991bd41a0bc803f"> 160</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#afc7e1d919209cf861991bd41a0bc803f">PushInput</a>(<span class="keyword">const</span> T&amp; t) {</div>
<div class="line"><a name="l00161"></a><span class="lineno"> 161</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a72e6940d1b4be12bbbba0bccc00ef739">SetParam</a>(N, t);</div>
<div class="line"><a name="l00162"></a><span class="lineno"> 162</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00163"></a><span class="lineno"> 163</span>&#160; }</div>
<div class="line"><a name="l00169"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a3116bcf3f735d37cd31a7916a5907b96"> 169</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a3116bcf3f735d37cd31a7916a5907b96">operator()</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1NDArray.html">NDArray</a>&amp; ndarray) {</div>
<div class="line"><a name="l00170"></a><span class="lineno"> 170</span>&#160; input_ndarrays_.push_back(ndarray.<a class="code" href="classmxnet_1_1cpp_1_1NDArray.html#a4ee101e6e73626382af486003048cae0">GetHandle</a>());</div>
<div class="line"><a name="l00171"></a><span class="lineno"> 171</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00172"></a><span class="lineno"> 172</span>&#160; }</div>
<div class="line"><a name="l00178"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#a899126f9fb43626fa8a11d36add58038"> 178</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a899126f9fb43626fa8a11d36add58038">operator()</a>(<span class="keyword">const</span> std::vector&lt;NDArray&gt;&amp; ndarrays) {</div>
<div class="line"><a name="l00179"></a><span class="lineno"> 179</span>&#160; <span class="keywordflow">for</span> (<span class="keyword">auto</span>&amp; s : ndarrays) {</div>
<div class="line"><a name="l00180"></a><span class="lineno"> 180</span>&#160; input_ndarrays_.push_back(s.GetHandle());</div>
<div class="line"><a name="l00181"></a><span class="lineno"> 181</span>&#160; }</div>
<div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160; }</div>
<div class="line"><a name="l00188"></a><span class="lineno"> 188</span>&#160; <span class="keyword">template</span> &lt;<span class="keyword">typename</span>... Args&gt;</div>
<div class="line"><a name="l00189"></a><span class="lineno"><a class="line" href="classmxnet_1_1cpp_1_1Operator.html#acfe0ed742dffdd849165081f444387f3"> 189</a></span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>&amp; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#acfe0ed742dffdd849165081f444387f3">operator()</a>(Args... args) {</div>
<div class="line"><a name="l00190"></a><span class="lineno"> 190</span>&#160; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a33b42e657f9a15ab4004b3b8226d71c6">PushInput</a>(args...);</div>
<div class="line"><a name="l00191"></a><span class="lineno"> 191</span>&#160; <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
<div class="line"><a name="l00192"></a><span class="lineno"> 192</span>&#160; }</div>
<div class="line"><a name="l00193"></a><span class="lineno"> 193</span>&#160; std::vector&lt;NDArray&gt; <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#abb488664e59eff22f62f28660fde54d1">Invoke</a>();</div>
<div class="line"><a name="l00194"></a><span class="lineno"> 194</span>&#160; <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#abb488664e59eff22f62f28660fde54d1">Invoke</a>(<a class="code" href="classmxnet_1_1cpp_1_1NDArray.html">NDArray</a>&amp; output);</div>
<div class="line"><a name="l00195"></a><span class="lineno"> 195</span>&#160; <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#abb488664e59eff22f62f28660fde54d1">Invoke</a>(std::vector&lt;NDArray&gt;&amp; outputs);</div>
<div class="line"><a name="l00196"></a><span class="lineno"> 196</span>&#160; </div>
<div class="line"><a name="l00197"></a><span class="lineno"> 197</span>&#160; <span class="keyword">private</span>:</div>
<div class="line"><a name="l00198"></a><span class="lineno"> 198</span>&#160; std::map&lt;std::string, std::string&gt; params_desc_;</div>
<div class="line"><a name="l00199"></a><span class="lineno"> 199</span>&#160; <span class="keywordtype">bool</span> variable_params_ = <span class="keyword">false</span>;</div>
<div class="line"><a name="l00200"></a><span class="lineno"> 200</span>&#160; std::map&lt;std::string, std::string&gt; params_;</div>
<div class="line"><a name="l00201"></a><span class="lineno"> 201</span>&#160; std::vector&lt;SymbolHandle&gt; input_symbols_;</div>
<div class="line"><a name="l00202"></a><span class="lineno"> 202</span>&#160; std::vector&lt;NDArrayHandle&gt; input_ndarrays_;</div>
<div class="line"><a name="l00203"></a><span class="lineno"> 203</span>&#160; std::vector&lt;std::string&gt; input_keys_;</div>
<div class="line"><a name="l00204"></a><span class="lineno"> 204</span>&#160; std::vector&lt;std::string&gt; arg_names_;</div>
<div class="line"><a name="l00205"></a><span class="lineno"> 205</span>&#160; <a class="code" href="include_2mxnet_2c__api_8h.html#aa2e96133f49cfdc1095da6cd08afbce0">AtomicSymbolCreator</a> handle_;</div>
<div class="line"><a name="l00206"></a><span class="lineno"> 206</span>&#160; <span class="keyword">static</span> <a class="code" href="classmxnet_1_1cpp_1_1OpMap.html">OpMap</a>*&amp; op_map();</div>
<div class="line"><a name="l00207"></a><span class="lineno"> 207</span>&#160;};</div>
<div class="line"><a name="l00208"></a><span class="lineno"> 208</span>&#160;} <span class="comment">// namespace cpp</span></div>
<div class="line"><a name="l00209"></a><span class="lineno"> 209</span>&#160;} <span class="comment">// namespace mxnet</span></div>
<div class="line"><a name="l00210"></a><span class="lineno"> 210</span>&#160; </div>
<div class="line"><a name="l00211"></a><span class="lineno"> 211</span>&#160;<span class="preprocessor">#endif // MXNET_CPP_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_1cpp_1_1Operator_html_afb1cbbf2fba12676de09dcb64f486464"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#afb1cbbf2fba12676de09dcb64f486464">mxnet::cpp::Operator::SetInput</a></div><div class="ttdeci">Operator &amp; SetInput(const std::string &amp;name, const Symbol &amp;symbol)</div><div class="ttdoc">add an input symbol</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a33b42e657f9a15ab4004b3b8226d71c6"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a33b42e657f9a15ab4004b3b8226d71c6">mxnet::cpp::Operator::PushInput</a></div><div class="ttdeci">void PushInput(const Symbol &amp;symbol)</div><div class="ttdoc">add an input symbol</div><div class="ttdef"><b>Definition:</b> operator.h:94</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a280201370ff22876cb38afbe47861f42"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a280201370ff22876cb38afbe47861f42">mxnet::cpp::Operator::SetParam</a></div><div class="ttdeci">Operator &amp; SetParam(int pos, const T &amp;value)</div><div class="ttdoc">set config parameters from positional inputs</div><div class="ttdef"><b>Definition:</b> operator.h:73</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a537e8d915b101757819a7a01da956a6b"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a537e8d915b101757819a7a01da956a6b">mxnet::cpp::Operator::PushInput</a></div><div class="ttdeci">Operator &amp; PushInput(const T &amp;t, Args... args)</div><div class="ttdoc">add positional inputs</div><div class="ttdef"><b>Definition:</b> operator.h:151</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_ae0057800586b01402bd99ae62e5614ef"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#ae0057800586b01402bd99ae62e5614ef">mxnet::cpp::Operator::operator=</a></div><div class="ttdeci">Operator &amp; operator=(const Operator &amp;rhs)</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a6911a78584ab48e8f2e808552e678992"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a6911a78584ab48e8f2e808552e678992">mxnet::cpp::Operator::operator()</a></div><div class="ttdeci">Operator &amp; operator()(const std::vector&lt; Symbol &gt; &amp;symbols)</div><div class="ttdoc">add a list of input symbols</div><div class="ttdef"><b>Definition:</b> operator.h:118</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html">mxnet::cpp::Operator</a></div><div class="ttdoc">Operator interface.</div><div class="ttdef"><b>Definition:</b> operator.h:42</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1NDArray_html"><div class="ttname"><a href="classmxnet_1_1cpp_1_1NDArray.html">mxnet::cpp::NDArray</a></div><div class="ttdoc">NDArray interface.</div><div class="ttdef"><b>Definition:</b> ndarray.h:122</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_afc7e1d919209cf861991bd41a0bc803f"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#afc7e1d919209cf861991bd41a0bc803f">mxnet::cpp::Operator::PushInput</a></div><div class="ttdeci">Operator &amp; PushInput(const T &amp;t)</div><div class="ttdoc">add the last positional input</div><div class="ttdef"><b>Definition:</b> operator.h:160</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a899126f9fb43626fa8a11d36add58038"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a899126f9fb43626fa8a11d36add58038">mxnet::cpp::Operator::operator()</a></div><div class="ttdeci">Operator &amp; operator()(const std::vector&lt; NDArray &gt; &amp;ndarrays)</div><div class="ttdoc">add a list of input ndarrays</div><div class="ttdef"><b>Definition:</b> operator.h:178</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1OpMap_html"><div class="ttname"><a href="classmxnet_1_1cpp_1_1OpMap.html">mxnet::cpp::OpMap</a></div><div class="ttdoc">OpMap instance holds a map of all the symbol creators so we can get symbol creators by name....</div><div class="ttdef"><b>Definition:</b> op_map.h:42</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_abb488664e59eff22f62f28660fde54d1"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#abb488664e59eff22f62f28660fde54d1">mxnet::cpp::Operator::Invoke</a></div><div class="ttdeci">std::vector&lt; NDArray &gt; Invoke()</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a72e6940d1b4be12bbbba0bccc00ef739"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a72e6940d1b4be12bbbba0bccc00ef739">mxnet::cpp::Operator::SetParam</a></div><div class="ttdeci">Operator &amp; SetParam(const std::string &amp;name, const T &amp;value)</div><div class="ttdoc">set config parameters</div><div class="ttdef"><b>Definition:</b> operator.h:57</div></div>
<div class="ttc" id="asymbol_8h_html"><div class="ttname"><a href="symbol_8h.html">symbol.h</a></div><div class="ttdoc">definition of symbol</div></div>
<div class="ttc" id="ainclude_2mxnet_2c__api_8h_html_aa2e96133f49cfdc1095da6cd08afbce0"><div class="ttname"><a href="include_2mxnet_2c__api_8h.html#aa2e96133f49cfdc1095da6cd08afbce0">AtomicSymbolCreator</a></div><div class="ttdeci">void * AtomicSymbolCreator</div><div class="ttdoc">handle to a function that takes param and creates symbol</div><div class="ttdef"><b>Definition:</b> c_api.h:78</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a52943207a5d771a32f7e1a861e34d0e3"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a52943207a5d771a32f7e1a861e34d0e3">mxnet::cpp::Operator::CreateSymbol</a></div><div class="ttdeci">Symbol CreateSymbol(const std::string &amp;name=&quot;&quot;)</div><div class="ttdoc">create a Symbol from the current operator</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a453b9eee13ca4e8acd3408de03aeafbe"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a453b9eee13ca4e8acd3408de03aeafbe">mxnet::cpp::Operator::operator()</a></div><div class="ttdeci">Operator &amp; operator()(const Symbol &amp;symbol)</div><div class="ttdoc">add input symbols</div><div class="ttdef"><b>Definition:</b> operator.h:109</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_a3116bcf3f735d37cd31a7916a5907b96"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#a3116bcf3f735d37cd31a7916a5907b96">mxnet::cpp::Operator::operator()</a></div><div class="ttdeci">Operator &amp; operator()(const NDArray &amp;ndarray)</div><div class="ttdoc">add input ndarrays</div><div class="ttdef"><b>Definition:</b> operator.h:169</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_acfe0ed742dffdd849165081f444387f3"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#acfe0ed742dffdd849165081f444387f3">mxnet::cpp::Operator::operator()</a></div><div class="ttdeci">Operator &amp; operator()(Args... args)</div><div class="ttdoc">add input ndarrays</div><div class="ttdef"><b>Definition:</b> operator.h:189</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_ae6321f03d02b6d076d4154d353b1c4a5"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#ae6321f03d02b6d076d4154d353b1c4a5">mxnet::cpp::Operator::Operator</a></div><div class="ttdeci">Operator(const std::string &amp;operator_name)</div><div class="ttdoc">Operator constructor.</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1NDArray_html_a4ee101e6e73626382af486003048cae0"><div class="ttname"><a href="classmxnet_1_1cpp_1_1NDArray.html#a4ee101e6e73626382af486003048cae0">mxnet::cpp::NDArray::GetHandle</a></div><div class="ttdeci">NDArrayHandle GetHandle() const</div><div class="ttdef"><b>Definition:</b> ndarray.h:475</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Symbol_html_a7a4d4a4cc1309356c81ed8cdd4919580"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Symbol.html#a7a4d4a4cc1309356c81ed8cdd4919580">mxnet::cpp::Symbol::GetHandle</a></div><div class="ttdeci">SymbolHandle GetHandle() const</div><div class="ttdef"><b>Definition:</b> symbol.h:142</div></div>
<div class="ttc" id="acpp-package_2include_2mxnet-cpp_2base_8h_html"><div class="ttname"><a href="cpp-package_2include_2mxnet-cpp_2base_8h.html">base.h</a></div><div class="ttdoc">base definitions for mxnetcpp</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_ac3c8a454dd7e27358d29f94f17713f18"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#ac3c8a454dd7e27358d29f94f17713f18">mxnet::cpp::Operator::operator()</a></div><div class="ttdeci">Operator &amp; operator()()</div><div class="ttdoc">add input symbols</div><div class="ttdef"><b>Definition:</b> operator.h:101</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Operator_html_acf21af976feee8c84632c6810632acdb"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Operator.html#acf21af976feee8c84632c6810632acdb">mxnet::cpp::Operator::PushInput</a></div><div class="ttdeci">Operator &amp; PushInput(const NDArray &amp;ndarray)</div><div class="ttdoc">add an input ndarray</div><div class="ttdef"><b>Definition:</b> operator.h:143</div></div>
<div class="ttc" id="aclassmxnet_1_1cpp_1_1Symbol_html"><div class="ttname"><a href="classmxnet_1_1cpp_1_1Symbol.html">mxnet::cpp::Symbol</a></div><div class="ttdoc">Symbol interface.</div><div class="ttdef"><b>Definition:</b> symbol.h:73</div></div>
<div class="ttc" id="aop__map_8h_html"><div class="ttname"><a href="op__map_8h.html">op_map.h</a></div><div class="ttdoc">definition of OpMap</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>