| <!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&dn=gpl-2.0.txt GPL-v2 */ |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| /* @license-end */ |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| /* @license-end */</script> |
| <div id="main-nav"></div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_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> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> * Licensed to the Apache Software Foundation (ASF) under one</span></div> |
| <div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> * or more contributor license agreements. See the NOTICE file</span></div> |
| <div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> * distributed with this work for additional information</span></div> |
| <div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> * regarding copyright ownership. The ASF licenses this file</span></div> |
| <div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * to you under the Apache License, Version 2.0 (the</span></div> |
| <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> * "License"); you may not use this file except in compliance</span></div> |
| <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> * with the License. You may obtain a copy of the License at</span></div> |
| <div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span></div> |
| <div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="comment"> * Unless required by applicable law or agreed to in writing,</span></div> |
| <div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="comment"> * software distributed under the License is distributed on an</span></div> |
| <div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="comment"> * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span></div> |
| <div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="comment"> * KIND, either express or implied. See the License for the</span></div> |
| <div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="comment"> * specific language governing permissions and limitations</span></div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * under the License.</span></div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span>  </div> |
| <div class="line"><a name="l00026"></a><span class="lineno"> 26</span> <span class="preprocessor">#ifndef MXNET_CPP_OPERATOR_H_</span></div> |
| <div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#define MXNET_CPP_OPERATOR_H_</span></div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span>  </div> |
| <div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="preprocessor">#include <map></span></div> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#include <string></span></div> |
| <div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#include <vector></span></div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#include "<a class="code" href="cpp-package_2include_2mxnet-cpp_2base_8h.html">mxnet-cpp/base.h</a>"</span></div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#include "<a class="code" href="op__map_8h.html">mxnet-cpp/op_map.h</a>"</span></div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include "<a class="code" href="symbol_8h.html">mxnet-cpp/symbol.h</a>"</span></div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span>  </div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <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> <span class="keyword">namespace </span>cpp {</div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <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> <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>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00048"></a><span class="lineno"> 48</span>  <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& operator_name);</div> |
| <div class="line"><a name="l00049"></a><span class="lineno"> 49</span>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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>& rhs);</div> |
| <div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> T></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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a72e6940d1b4be12bbbba0bccc00ef739">SetParam</a>(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> T& value) {</div> |
| <div class="line"><a name="l00058"></a><span class="lineno"> 58</span>  std::string value_str;</div> |
| <div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  std::stringstream ss;</div> |
| <div class="line"><a name="l00060"></a><span class="lineno"> 60</span>  ss << value;</div> |
| <div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  ss >> value_str;</div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  </div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span>  params_[name] = value_str;</div> |
| <div class="line"><a name="l00064"></a><span class="lineno"> 64</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00065"></a><span class="lineno"> 65</span>  }</div> |
| <div class="line"><a name="l00072"></a><span class="lineno"> 72</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> T></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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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& value) {</div> |
| <div class="line"><a name="l00074"></a><span class="lineno"> 74</span>  std::string value_str;</div> |
| <div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  std::stringstream ss;</div> |
| <div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  ss << value;</div> |
| <div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  ss >> value_str;</div> |
| <div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  </div> |
| <div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  params_[arg_names_[pos]] = value_str;</div> |
| <div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00081"></a><span class="lineno"> 81</span>  }</div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#afb1cbbf2fba12676de09dcb64f486464">SetInput</a>(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1Symbol.html">Symbol</a>& symbol);</div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  <span class="keyword">template</span> <<span class="keywordtype">int</span> N = 0></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>  <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>& symbol) {</div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span>  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>  }</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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span>  }</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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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>& symbol) {</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  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>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  }</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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a6911a78584ab48e8f2e808552e678992">operator()</a>(<span class="keyword">const</span> std::vector<Symbol>& symbols) {</div> |
| <div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& s : symbols) {</div> |
| <div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  input_symbols_.push_back(s.GetHandle());</div> |
| <div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  }</div> |
| <div class="line"><a name="l00122"></a><span class="lineno"> 122</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  }</div> |
| <div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <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& name = <span class="stringliteral">""</span>);</div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  </div> |
| <div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#afb1cbbf2fba12676de09dcb64f486464">SetInput</a>(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> <a class="code" href="classmxnet_1_1cpp_1_1NDArray.html">NDArray</a>& ndarray);</div> |
| <div class="line"><a name="l00142"></a><span class="lineno"> 142</span>  <span class="keyword">template</span> <<span class="keywordtype">int</span> N = 0></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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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>& ndarray) {</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  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>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  }</div> |
| <div class="line"><a name="l00150"></a><span class="lineno"> 150</span>  <span class="keyword">template</span> <<span class="keyword">class </span>T, <span class="keyword">class</span>... Args, <span class="keywordtype">int</span> N = 0></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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a537e8d915b101757819a7a01da956a6b">PushInput</a>(<span class="keyword">const</span> T& t, Args... args) {</div> |
| <div class="line"><a name="l00152"></a><span class="lineno"> 152</span>  <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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a33b42e657f9a15ab4004b3b8226d71c6">PushInput</a><Args..., N + 1>(args...);</div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00155"></a><span class="lineno"> 155</span>  }</div> |
| <div class="line"><a name="l00159"></a><span class="lineno"> 159</span>  <span class="keyword">template</span> <<span class="keyword">class</span> T, <span class="keywordtype">int</span> N = 0></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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#afc7e1d919209cf861991bd41a0bc803f">PushInput</a>(<span class="keyword">const</span> T& t) {</div> |
| <div class="line"><a name="l00161"></a><span class="lineno"> 161</span>  <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>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  }</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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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>& ndarray) {</div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  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>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00172"></a><span class="lineno"> 172</span>  }</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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#a899126f9fb43626fa8a11d36add58038">operator()</a>(<span class="keyword">const</span> std::vector<NDArray>& ndarrays) {</div> |
| <div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& s : ndarrays) {</div> |
| <div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  input_ndarrays_.push_back(s.GetHandle());</div> |
| <div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  }</div> |
| <div class="line"><a name="l00182"></a><span class="lineno"> 182</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00183"></a><span class="lineno"> 183</span>  }</div> |
| <div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  <span class="keyword">template</span> <<span class="keyword">typename</span>... Args></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>  <a class="code" href="classmxnet_1_1cpp_1_1Operator.html">Operator</a>& <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>  <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>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00192"></a><span class="lineno"> 192</span>  }</div> |
| <div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  std::vector<NDArray> <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>  <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>& output);</div> |
| <div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1cpp_1_1Operator.html#abb488664e59eff22f62f28660fde54d1">Invoke</a>(std::vector<NDArray>& outputs);</div> |
| <div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  </div> |
| <div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  std::map<std::string, std::string> params_desc_;</div> |
| <div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  <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>  std::map<std::string, std::string> params_;</div> |
| <div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  std::vector<SymbolHandle> input_symbols_;</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  std::vector<NDArrayHandle> input_ndarrays_;</div> |
| <div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  std::vector<std::string> input_keys_;</div> |
| <div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  std::vector<std::string> arg_names_;</div> |
| <div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  <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>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1cpp_1_1OpMap.html">OpMap</a>*& op_map();</div> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span> };</div> |
| <div class="line"><a name="l00208"></a><span class="lineno"> 208</span> } <span class="comment">// namespace cpp</span></div> |
| <div class="line"><a name="l00209"></a><span class="lineno"> 209</span> } <span class="comment">// namespace mxnet</span></div> |
| <div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  </div> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span> <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 & SetInput(const std::string &name, const Symbol &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 &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 & SetParam(int pos, const T &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 & PushInput(const T &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 & operator=(const Operator &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 & operator()(const std::vector< Symbol > &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 & PushInput(const T &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 & operator()(const std::vector< NDArray > &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< NDArray > 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 & SetParam(const std::string &name, const T &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 &name="")</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 & operator()(const Symbol &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 & operator()(const NDArray &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 & 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 &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 & 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 & PushInput(const NDArray &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  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.17 |
| </small></address> |
| </body> |
| </html> |