| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta content="text/xhtml;charset=utf-8" http-equiv="Content-Type"/> |
| <meta content="IE=9" http-equiv="X-UA-Compatible"/> |
| <meta content="Doxygen 1.8.6" name="generator"/> |
| <title>mxnet: include/mxnet/op_attr_types.h Source File</title> |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <script src="jquery.js" type="text/javascript"></script> |
| <script src="dynsections.js" type="text/javascript"></script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script src="search/search.js" type="text/javascript"></script> |
| <script type="text/javascript"> |
| $(document).ready(function() { searchBox.OnSelectItem(0); }); |
| </script> |
| <link href="doxygen.css" rel="stylesheet" type="text/css"/> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| <div id="titlearea"> |
| <table cellpadding="0" cellspacing="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| <td style="padding-left: 0.5em;"> |
| <div id="projectname">mxnet |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.6 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <div class="tabs" id="navrow1"> |
| <ul class="tablist"> |
| <li><a href="index.html"><span>Main Page</span></a></li> |
| <li><a href="namespaces.html"><span>Namespaces</span></a></li> |
| <li><a href="annotated.html"><span>Classes</span></a></li> |
| <li class="current"><a href="files.html"><span>Files</span></a></li> |
| <li> |
| <div class="MSearchBoxInactive" id="MSearchBox"> |
| <span class="left"> |
| <img alt="" id="MSearchSelect" onmouseout="return searchBox.OnSearchSelectHide()" onmouseover="return searchBox.OnSearchSelectShow()" src="search/mag_sel.png"/> |
| <input accesskey="S" id="MSearchField" onblur="searchBox.OnSearchFieldFocus(false)" onfocus="searchBox.OnSearchFieldFocus(true)" onkeyup="searchBox.OnSearchFieldChange(event)" type="text" value="Search"/> |
| </span><span class="right"> |
| <a href="javascript:searchBox.CloseResultsWindow()" id="MSearchClose"><img alt="" border="0" id="MSearchCloseImg" src="search/close.png"/></a> |
| </span> |
| </div> |
| </li> |
| </ul> |
| </div> |
| <div class="tabs2" id="navrow2"> |
| <ul class="tablist"> |
| <li><a href="files.html"><span>File List</span></a></li> |
| <li><a href="globals.html"><span>File Members</span></a></li> |
| </ul> |
| </div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" onkeydown="return searchBox.OnSearchSelectKey(event)" onmouseout="return searchBox.OnSearchSelectHide()" onmouseover="return searchBox.OnSearchSelectShow()"> |
| <a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(0)"><span class="SelectionMark"> </span>All</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(1)"><span class="SelectionMark"> </span>Classes</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(2)"><span class="SelectionMark"> </span>Namespaces</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(3)"><span class="SelectionMark"> </span>Files</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(4)"><span class="SelectionMark"> </span>Functions</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(5)"><span class="SelectionMark"> </span>Variables</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(6)"><span class="SelectionMark"> </span>Typedefs</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(7)"><span class="SelectionMark"> </span>Enumerations</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(8)"><span class="SelectionMark"> </span>Enumerator</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(9)"><span class="SelectionMark"> </span>Friends</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(10)"><span class="SelectionMark"> </span>Macros</a></div> |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe frameborder="0" id="MSearchResults" name="MSearchResults" src="javascript:void(0)"> |
| </iframe> |
| </div> |
| <div class="navpath" id="nav-path"> |
| <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">op_attr_types.h</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="op__attr__types_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> * Licensed to the Apache Software Foundation (ASF) under one</span></div> |
| <div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> * or more contributor license agreements. See the NOTICE file</span></div> |
| <div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> * distributed with this work for additional information</span></div> |
| <div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> * regarding copyright ownership. The ASF licenses this file</span></div> |
| <div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * to you under the Apache License, Version 2.0 (the</span></div> |
| <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> * "License"); you may not use this file except in compliance</span></div> |
| <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> * with the License. You may obtain a copy of the License at</span></div> |
| <div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span></div> |
| <div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="comment"> * Unless required by applicable law or agreed to in writing,</span></div> |
| <div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="comment"> * software distributed under the License is distributed on an</span></div> |
| <div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="comment"> * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span></div> |
| <div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="comment"> * KIND, either express or implied. See the License for the</span></div> |
| <div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="comment"> * specific language governing permissions and limitations</span></div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * under the License.</span></div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span> </div> |
| <div class="line"><a name="l00025"></a><span class="lineno"> 25</span> <span class="preprocessor">#ifndef MXNET_OP_ATTR_TYPES_H_</span></div> |
| <div class="line"><a name="l00026"></a><span class="lineno"> 26</span> <span class="preprocessor"></span><span class="preprocessor">#define MXNET_OP_ATTR_TYPES_H_</span></div> |
| <div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor"></span></div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span> <span class="preprocessor">#include <mshadow/tensor.h></span></div> |
| <div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="preprocessor">#include <nnvm/op_attr_types.h></span></div> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</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 <functional></span></div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span> </div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include "<a class="code" href="base_8h.html">./base.h</a>"</span></div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#include "<a class="code" href="ndarray_8h.html">./ndarray.h</a>"</span></div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#include "<a class="code" href="engine_8h.html">./engine.h</a>"</span></div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="preprocessor">#include "<a class="code" href="resource_8h.html">./resource.h</a>"</span></div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span> </div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="keyword">namespace </span>mxnet {</div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span> </div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="keyword">using</span> nnvm::NodeAttrs;</div> |
| <div class="line"><a name="l00042"></a><span class="lineno"> 42</span> </div> |
| <div class="line"><a name="l00044"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398"> 44</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">OpReqType</a> {</div> |
| <div class="line"><a name="l00046"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26"> 46</a></span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26">kNullOp</a>,</div> |
| <div class="line"><a name="l00048"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a"> 48</a></span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a">kWriteTo</a>,</div> |
| <div class="line"><a name="l00054"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51"> 54</a></span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51">kWriteInplace</a>,</div> |
| <div class="line"><a name="l00056"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383"> 56</a></span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383">kAddTo</a></div> |
| <div class="line"><a name="l00057"></a><span class="lineno"> 57</span> };</div> |
| <div class="line"><a name="l00058"></a><span class="lineno"> 58</span> </div> |
| <div class="line"><a name="l00065"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html"> 65</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1OpContext.html">OpContext</a> {</div> |
| <div class="line"><a name="l00067"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#abd7405c065167e4e15af032b1e0a5c42"> 67</a></span> <span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1OpContext.html#abd7405c065167e4e15af032b1e0a5c42">is_train</a>;</div> |
| <div class="line"><a name="l00069"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf"> 69</a></span> <a class="code" href="structmxnet_1_1RunContext.html">RunContext</a> <a class="code" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf">run_ctx</a>;</div> |
| <div class="line"><a name="l00071"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef"> 71</a></span> <a class="code" href="classmxnet_1_1engine_1_1CallbackOnComplete.html">engine::CallbackOnComplete</a> <a class="code" href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef">async_on_complete</a>;</div> |
| <div class="line"><a name="l00073"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b"> 73</a></span> std::vector<Resource> <a class="code" href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b">requested</a>;</div> |
| <div class="line"><a name="l00079"></a><span class="lineno"> 79</span> <span class="keyword">template</span><<span class="keyword">typename</span> xpu></div> |
| <div class="line"><a name="l00080"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#aa700c5c28e024b42d41e15f46615f1b8"> 80</a></span> <span class="keyword">inline</span> mshadow::Stream<xpu>* <a class="code" href="structmxnet_1_1OpContext.html#aa700c5c28e024b42d41e15f46615f1b8">get_stream</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00081"></a><span class="lineno"> 81</span> <span class="keywordflow">return</span> <a class="code" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf">run_ctx</a>.<a class="code" href="structmxnet_1_1RunContext.html#a8bcd2fc6392f95a1fdc3d98f74dbfcb9">get_stream</a><xpu>();</div> |
| <div class="line"><a name="l00082"></a><span class="lineno"> 82</span> }</div> |
| <div class="line"><a name="l00083"></a><span class="lineno"> 83</span> };</div> |
| <div class="line"><a name="l00084"></a><span class="lineno"> 84</span> </div> |
| <div class="line"><a name="l00086"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad"> 86</a></span> <span class="keyword">enum class</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad">ExecType</a> {</div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c">kSync</a>,</div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span> <a class="code" href="namespacemxnet.html#a998b74220fab2b012cf8a179650e1b3ba9f2b960005d2a3a5f35ac32809d84db7">kAsync</a>,</div> |
| <div class="line"><a name="l00095"></a><span class="lineno"> 95</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adafcf37e733f29d6aa14fb34fa737a809a">kLocal</a>,</div> |
| <div class="line"><a name="l00102"></a><span class="lineno"> 102</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ada3864860a073c64d74d3fc8e765fa6cad">kCrossDeviceCopy</a></div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span> };</div> |
| <div class="line"><a name="l00104"></a><span class="lineno"> 104</span> </div> |
| <div class="line"><a name="l00106"></a><span class="lineno"><a class="line" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b"> 106</a></span> <span class="keyword">enum class</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a> {</div> |
| <div class="line"><a name="l00107"></a><span class="lineno"> 107</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba7934d40c7c17315d892e8d5d745b1c66">kUndefined</a> = -1,</div> |
| <div class="line"><a name="l00108"></a><span class="lineno"> 108</span> <span class="comment">// dispatch on FCompute or FStatefulCompute</span></div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba2157864769b54d1d0f8f9be63393e7d6">kFCompute</a>,</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span> <span class="comment">// dispatch on FComputeEx or FStatefulComputeEx, if available</span></div> |
| <div class="line"><a name="l00111"></a><span class="lineno"> 111</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba57f93219462690bc438372ecd92fedd5">kFComputeEx</a>,</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span> <span class="comment">// dispatch on FCompute or FStatefulCompute, and performs storage fallback</span></div> |
| <div class="line"><a name="l00113"></a><span class="lineno"> 113</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba18bf63b4c196f79c9bca4ad8637e8fef">kFComputeFallback</a>,</div> |
| <div class="line"><a name="l00114"></a><span class="lineno"> 114</span> <span class="comment">// special dispatch mode for variables</span></div> |
| <div class="line"><a name="l00115"></a><span class="lineno"> 115</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2bafb751f6ee1af15e773242f2accd2dbff">kVariable</a>,</div> |
| <div class="line"><a name="l00116"></a><span class="lineno"> 116</span> };</div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span> </div> |
| <div class="line"><a name="l00122"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html"> 122</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> {</div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span> <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00124"></a><span class="lineno"> 124</span> <span class="comment">/* \brief Create a OpStatePtr with state of type T.</span></div> |
| <div class="line"><a name="l00125"></a><span class="lineno"> 125</span> <span class="comment"> * \param args Arguments passed to T's constructor.</span></div> |
| <div class="line"><a name="l00126"></a><span class="lineno"> 126</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00127"></a><span class="lineno"> 127</span> <span class="keyword">template</span><<span class="keyword">typename</span> T, <span class="keyword">typename</span>... Args></div> |
| <div class="line"><a name="l00128"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#ae0f821c7efce61a11730ef8b93070e3e"> 128</a></span> <span class="keyword">static</span> <a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> <a class="code" href="classmxnet_1_1OpStatePtr.html#ae0f821c7efce61a11730ef8b93070e3e">Create</a>(Args&&... args) {</div> |
| <div class="line"><a name="l00129"></a><span class="lineno"> 129</span> <a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> ret;</div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span> ret.ptr_ = std::make_shared<OpState>();</div> |
| <div class="line"><a name="l00131"></a><span class="lineno"> 131</span> ret.ptr_->var_ = <a class="code" href="classmxnet_1_1Engine.html#ae0a23da15ef63d9479c7468e1f2f825f">Engine::Get</a>()-><a class="code" href="classmxnet_1_1Engine.html#a6e141b188f018d5d933ab99868631d5e">NewVariable</a>();</div> |
| <div class="line"><a name="l00132"></a><span class="lineno"> 132</span> ret.ptr_->state_.construct<T>(std::forward<Args>(args)...);</div> |
| <div class="line"><a name="l00133"></a><span class="lineno"> 133</span> </div> |
| <div class="line"><a name="l00134"></a><span class="lineno"> 134</span> <span class="keywordflow">return</span> ret;</div> |
| <div class="line"><a name="l00135"></a><span class="lineno"> 135</span> }</div> |
| <div class="line"><a name="l00136"></a><span class="lineno"> 136</span> <span class="comment">/* \brief Get engine variable associated with this state */</span></div> |
| <div class="line"><a name="l00137"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#a77a1636269466569c0b6e32941ed7264"> 137</a></span> <a class="code" href="namespacemxnet_1_1engine.html#a9d36c4f33eae8531586dc2edf83ae7cf">engine::VarHandle</a> <a class="code" href="classmxnet_1_1OpStatePtr.html#a77a1636269466569c0b6e32941ed7264">get_var</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00138"></a><span class="lineno"> 138</span> <span class="keywordflow">return</span> ptr_->var_;</div> |
| <div class="line"><a name="l00139"></a><span class="lineno"> 139</span> }</div> |
| <div class="line"><a name="l00140"></a><span class="lineno"> 140</span> <span class="comment">/* \brief Get state of type T */</span></div> |
| <div class="line"><a name="l00141"></a><span class="lineno"> 141</span> <span class="keyword">template</span><<span class="keyword">typename</span> T></div> |
| <div class="line"><a name="l00142"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#aa2ea1d7ae06008faf44a92c009b2b26e"> 142</a></span> T& <a class="code" href="classmxnet_1_1OpStatePtr.html#aa2ea1d7ae06008faf44a92c009b2b26e">get_state</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00143"></a><span class="lineno"> 143</span> <span class="keywordflow">return</span> dmlc::get<T>(ptr_->state_);</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span> }</div> |
| <div class="line"><a name="l00145"></a><span class="lineno"> 145</span> <span class="comment">/* \brief clear state */</span></div> |
| <div class="line"><a name="l00146"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#ae43213b3a65073a87bf489350a58d78e"> 146</a></span> <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1OpStatePtr.html#ae43213b3a65073a87bf489350a58d78e">reset</a>() {</div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span> ptr_.reset();</div> |
| <div class="line"><a name="l00148"></a><span class="lineno"> 148</span> }</div> |
| <div class="line"><a name="l00149"></a><span class="lineno"> 149</span> <span class="comment">/* \brief Whether state is empty */</span></div> |
| <div class="line"><a name="l00150"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#a6f407d1c8d461dc474cd1c50aa53e3f8"> 150</a></span> <span class="keyword">explicit</span> <span class="keyword">operator</span> bool()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00151"></a><span class="lineno"> 151</span> <span class="keywordflow">return</span> ptr_ ? <span class="keyword">true</span> : <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00152"></a><span class="lineno"> 152</span> }</div> |
| <div class="line"><a name="l00153"></a><span class="lineno"> 153</span> </div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span> <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00155"></a><span class="lineno"> 155</span> <span class="comment">/* \brief state structure */</span></div> |
| <div class="line"><a name="l00156"></a><span class="lineno"> 156</span> <span class="keyword">struct </span>OpState {</div> |
| <div class="line"><a name="l00157"></a><span class="lineno"> 157</span> OpState() {}</div> |
| <div class="line"><a name="l00158"></a><span class="lineno"> 158</span> OpState(<span class="keyword">const</span> OpState& other) = <span class="keyword">delete</span>;</div> |
| <div class="line"><a name="l00159"></a><span class="lineno"> 159</span> OpState& operator=(<span class="keyword">const</span> OpState& other) = <span class="keyword">delete</span>;</div> |
| <div class="line"><a name="l00160"></a><span class="lineno"> 160</span> </div> |
| <div class="line"><a name="l00161"></a><span class="lineno"> 161</span> ~OpState() {</div> |
| <div class="line"><a name="l00162"></a><span class="lineno"> 162</span> <a class="code" href="classmxnet_1_1Engine.html#ae0a23da15ef63d9479c7468e1f2f825f">Engine::Get</a>()-><a class="code" href="classmxnet_1_1Engine.html#a738e5192dab345ab0ec9888b095903cf">DeleteVariable</a>([](RunContext s) {}, <a class="code" href="structmxnet_1_1Context.html#a4bb231d70623fde216de51589347e513">Context::CPU</a>(), var_);</div> |
| <div class="line"><a name="l00163"></a><span class="lineno"> 163</span> }</div> |
| <div class="line"><a name="l00164"></a><span class="lineno"> 164</span> </div> |
| <div class="line"><a name="l00165"></a><span class="lineno"> 165</span> <a class="code" href="namespacemxnet_1_1engine.html#a9d36c4f33eae8531586dc2edf83ae7cf">engine::VarHandle</a> var_;</div> |
| <div class="line"><a name="l00166"></a><span class="lineno"> 166</span> dmlc::any state_;</div> |
| <div class="line"><a name="l00167"></a><span class="lineno"> 167</span> };</div> |
| <div class="line"><a name="l00168"></a><span class="lineno"> 168</span> <span class="comment">/* \brief shared pointer to state */</span></div> |
| <div class="line"><a name="l00169"></a><span class="lineno"> 169</span> std::shared_ptr<OpState> ptr_;</div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span> };</div> |
| <div class="line"><a name="l00171"></a><span class="lineno"> 171</span> </div> |
| <div class="line"><a name="l00184"></a><span class="lineno"> 184</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#acb83431b596f805fabcc4ba400c62e85">FCreateOpState</a> = std::function<OpStatePtr (<span class="keyword">const</span> NodeAttrs& attrs,</div> |
| <div class="line"><a name="l00185"></a><span class="lineno"> 185</span> Context ctx,</div> |
| <div class="line"><a name="l00186"></a><span class="lineno"> 186</span> <span class="keyword">const</span> std::vector<TShape>& in_shape,</div> |
| <div class="line"><a name="l00187"></a><span class="lineno"><a class="line" href="namespacemxnet.html#acb83431b596f805fabcc4ba400c62e85"> 187</a></span> <span class="keyword">const</span> std::vector<int>& in_type)>;</div> |
| <div class="line"><a name="l00191"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a666f1055e5401ecec0278e8ba5790732"> 191</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a666f1055e5401ecec0278e8ba5790732">FExecType</a> = std::function<ExecType (const NodeAttrs& attrs)>;</div> |
| <div class="line"><a name="l00199"></a><span class="lineno"> 199</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a9097a269dd76d833c014149ff3eaab9b">FStatefulCompute</a> = std::function<void (<span class="keyword">const</span> <a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a>& state,</div> |
| <div class="line"><a name="l00200"></a><span class="lineno"> 200</span> <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div> |
| <div class="line"><a name="l00201"></a><span class="lineno"> 201</span> <span class="keyword">const</span> std::vector<TBlob>& inputs,</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span> <span class="keyword">const</span> std::vector<OpReqType>& req,</div> |
| <div class="line"><a name="l00203"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a9097a269dd76d833c014149ff3eaab9b"> 203</a></span> <span class="keyword">const</span> std::vector<TBlob>& outputs)>;</div> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#ab12f1853179b1456152a563edff844a9">FStatefulComputeEx</a> = std::function<void (<span class="keyword">const</span> <a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a>& state,</div> |
| <div class="line"><a name="l00212"></a><span class="lineno"> 212</span> <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div> |
| <div class="line"><a name="l00213"></a><span class="lineno"> 213</span> <span class="keyword">const</span> std::vector<NDArray>& inputs,</div> |
| <div class="line"><a name="l00214"></a><span class="lineno"> 214</span> <span class="keyword">const</span> std::vector<OpReqType>& req,</div> |
| <div class="line"><a name="l00215"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ab12f1853179b1456152a563edff844a9"> 215</a></span> <span class="keyword">const</span> std::vector<NDArray>& outputs)>;</div> |
| <div class="line"><a name="l00221"></a><span class="lineno"> 221</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#afaa687cbca9e992614be285711c1d72d">FResourceRequest</a> = std::function<</div> |
| <div class="line"><a name="l00222"></a><span class="lineno"><a class="line" href="namespacemxnet.html#afaa687cbca9e992614be285711c1d72d"> 222</a></span> std::vector<ResourceRequest> (<span class="keyword">const</span> NodeAttrs& n)>;</div> |
| <div class="line"><a name="l00228"></a><span class="lineno"> 228</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a0906e49a921265634672f555451075ed">FNDArrayFunction</a> = std::function<void (<span class="keyword">const</span> nnvm::NodeAttrs& attrs,</div> |
| <div class="line"><a name="l00229"></a><span class="lineno"> 229</span> <span class="keyword">const</span> std::vector<NDArray>& inputs,</div> |
| <div class="line"><a name="l00230"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a0906e49a921265634672f555451075ed"> 230</a></span> std::vector<NDArray>* outputs)>;</div> |
| <div class="line"><a name="l00236"></a><span class="lineno"> 236</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#ad77e2048b7e8db91a12e05b477aac13e">FCompute</a> = std::function<void (<span class="keyword">const</span> nnvm::NodeAttrs& attrs,</div> |
| <div class="line"><a name="l00237"></a><span class="lineno"> 237</span> <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div> |
| <div class="line"><a name="l00238"></a><span class="lineno"> 238</span> <span class="keyword">const</span> std::vector<TBlob>& inputs,</div> |
| <div class="line"><a name="l00239"></a><span class="lineno"> 239</span> <span class="keyword">const</span> std::vector<OpReqType>& req,</div> |
| <div class="line"><a name="l00240"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ad77e2048b7e8db91a12e05b477aac13e"> 240</a></span> <span class="keyword">const</span> std::vector<TBlob>& outputs)>;</div> |
| <div class="line"><a name="l00247"></a><span class="lineno"> 247</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a294d2660c065f8d229c27e65f0461e10">FComputeEx</a> = std::function<void (<span class="keyword">const</span> nnvm::NodeAttrs& attrs,</div> |
| <div class="line"><a name="l00248"></a><span class="lineno"> 248</span> <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div> |
| <div class="line"><a name="l00249"></a><span class="lineno"> 249</span> <span class="keyword">const</span> std::vector<NDArray>& inputs,</div> |
| <div class="line"><a name="l00250"></a><span class="lineno"> 250</span> <span class="keyword">const</span> std::vector<OpReqType>& req,</div> |
| <div class="line"><a name="l00251"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a294d2660c065f8d229c27e65f0461e10"> 251</a></span> <span class="keyword">const</span> std::vector<NDArray>& outputs)>;</div> |
| <div class="line"><a name="l00252"></a><span class="lineno"> 252</span> </div> |
| <div class="line"><a name="l00259"></a><span class="lineno"> 259</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a0a1395f21139ba975a6268539df6a15d">FInferStorageType</a> = std::function<bool (<span class="keyword">const</span> NodeAttrs& attrs,</div> |
| <div class="line"><a name="l00260"></a><span class="lineno"> 260</span> <span class="keyword">const</span> <span class="keywordtype">int</span> dev_mask,</div> |
| <div class="line"><a name="l00261"></a><span class="lineno"> 261</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a>* dispatch_mode,</div> |
| <div class="line"><a name="l00262"></a><span class="lineno"> 262</span> std::vector<int>* in_attrs,</div> |
| <div class="line"><a name="l00263"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a0a1395f21139ba975a6268539df6a15d"> 263</a></span> std::vector<int>* out_attrs)>;</div> |
| <div class="line"><a name="l00264"></a><span class="lineno"> 264</span> </div> |
| <div class="line"><a name="l00265"></a><span class="lineno"> 265</span> } <span class="comment">// namespace mxnet</span></div> |
| <div class="line"><a name="l00266"></a><span class="lineno"> 266</span> </div> |
| <div class="line"><a name="l00267"></a><span class="lineno"> 267</span> <span class="preprocessor">#endif // MXNET_OP_ATTR_TYPES_H_</span></div> |
| <div class="ttc" id="namespacemxnet_html_a9097a269dd76d833c014149ff3eaab9b"><div class="ttname"><a href="namespacemxnet.html#a9097a269dd76d833c014149ff3eaab9b">mxnet::FStatefulCompute</a></div><div class="ttdeci">std::function< void(const OpStatePtr &state, const OpContext &ctx, const std::vector< TBlob > &inputs, const std::vector< OpReqType > &req, const std::vector< TBlob > &outputs)> FStatefulCompute</div><div class="ttdoc">Resiger a compute function for stateful operator. OpStatePtr is a pointer type, it's content is mutab...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:203</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_ae43213b3a65073a87bf489350a58d78e"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#ae43213b3a65073a87bf489350a58d78e">mxnet::OpStatePtr::reset</a></div><div class="ttdeci">void reset()</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:146</div></div> |
| <div class="ttc" id="namespacemxnet_html_ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c"><div class="ttname"><a href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c">mxnet::ExecType::kSync</a></div><div class="ttdoc">Forward/Backward are synchronize calls. </div></div> |
| <div class="ttc" id="namespacemxnet_html_aad5fec2b4bceabbf4b884460cc2e0a2bafb751f6ee1af15e773242f2accd2dbff"><div class="ttname"><a href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2bafb751f6ee1af15e773242f2accd2dbff">mxnet::DispatchMode::kVariable</a></div></div> |
| <div class="ttc" id="engine_8h_html"><div class="ttname"><a href="engine_8h.html">engine.h</a></div><div class="ttdoc">Engine that schedules all the operations according to dependency. </div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26">mxnet::kNullOp</a></div><div class="ttdoc">no operation, do not write anything </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:46</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a">mxnet::kWriteTo</a></div><div class="ttdoc">write gradient to provided space </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:48</div></div> |
| <div class="ttc" id="namespacemxnet_html_aad5fec2b4bceabbf4b884460cc2e0a2ba57f93219462690bc438372ecd92fedd5"><div class="ttname"><a href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba57f93219462690bc438372ecd92fedd5">mxnet::DispatchMode::kFComputeEx</a></div></div> |
| <div class="ttc" id="namespacemxnet_html_afaa687cbca9e992614be285711c1d72d"><div class="ttname"><a href="namespacemxnet.html#afaa687cbca9e992614be285711c1d72d">mxnet::FResourceRequest</a></div><div class="ttdeci">std::function< std::vector< ResourceRequest >(const NodeAttrs &n)> FResourceRequest</div><div class="ttdoc">The resource request from the operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:222</div></div> |
| <div class="ttc" id="namespacemxnet_html_ab12f1853179b1456152a563edff844a9"><div class="ttname"><a href="namespacemxnet.html#ab12f1853179b1456152a563edff844a9">mxnet::FStatefulComputeEx</a></div><div class="ttdeci">std::function< void(const OpStatePtr &state, const OpContext &ctx, const std::vector< NDArray > &inputs, const std::vector< OpReqType > &req, const std::vector< NDArray > &outputs)> FStatefulComputeEx</div><div class="ttdoc">Resiger a compute function for stateful operator using NDArray interface. OpStatePtr is a pointer typ...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:215</div></div> |
| <div class="ttc" id="structmxnet_1_1RunContext_html_a8bcd2fc6392f95a1fdc3d98f74dbfcb9"><div class="ttname"><a href="structmxnet_1_1RunContext.html#a8bcd2fc6392f95a1fdc3d98f74dbfcb9">mxnet::RunContext::get_stream</a></div><div class="ttdeci">mshadow::Stream< xpu > * get_stream() const </div><div class="ttdoc">get mshadow stream from Context </div><div class="ttdef"><b>Definition:</b> base.h:251</div></div> |
| <div class="ttc" id="namespacemxnet_html_a998b74220fab2b012cf8a179650e1b3ba9f2b960005d2a3a5f35ac32809d84db7"><div class="ttname"><a href="namespacemxnet.html#a998b74220fab2b012cf8a179650e1b3ba9f2b960005d2a3a5f35ac32809d84db7">mxnet::FnProperty::kAsync</a></div><div class="ttdoc">Asynchronous function call. </div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_a77a1636269466569c0b6e32941ed7264"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#a77a1636269466569c0b6e32941ed7264">mxnet::OpStatePtr::get_var</a></div><div class="ttdeci">engine::VarHandle get_var() const </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:137</div></div> |
| <div class="ttc" id="namespacemxnet_html_ad7801fb2dd68294d4c7a9e289bc071ada3864860a073c64d74d3fc8e765fa6cad"><div class="ttname"><a href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ada3864860a073c64d74d3fc8e765fa6cad">mxnet::ExecType::kCrossDeviceCopy</a></div><div class="ttdoc">Cross device copy operation, this is a special operator That indicates copy across devices...</div></div> |
| <div class="ttc" id="structmxnet_1_1RunContext_html"><div class="ttname"><a href="structmxnet_1_1RunContext.html">mxnet::RunContext</a></div><div class="ttdoc">execution time context. The information needed in runtime for actual execution. </div><div class="ttdef"><b>Definition:</b> base.h:238</div></div> |
| <div class="ttc" id="namespacemxnet_html_aad5fec2b4bceabbf4b884460cc2e0a2b"><div class="ttname"><a href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">mxnet::DispatchMode</a></div><div class="ttdeci">DispatchMode</div><div class="ttdoc">the dispatch mode of the operator </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:106</div></div> |
| <div class="ttc" id="namespacemxnet_html_aad5fec2b4bceabbf4b884460cc2e0a2ba2157864769b54d1d0f8f9be63393e7d6"><div class="ttname"><a href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba2157864769b54d1d0f8f9be63393e7d6">mxnet::DispatchMode::kFCompute</a></div></div> |
| <div class="ttc" id="namespacemxnet_html_aad5fec2b4bceabbf4b884460cc2e0a2ba18bf63b4c196f79c9bca4ad8637e8fef"><div class="ttname"><a href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba18bf63b4c196f79c9bca4ad8637e8fef">mxnet::DispatchMode::kFComputeFallback</a></div></div> |
| <div class="ttc" id="namespacemxnet_html_ad7801fb2dd68294d4c7a9e289bc071adafcf37e733f29d6aa14fb34fa737a809a"><div class="ttname"><a href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adafcf37e733f29d6aa14fb34fa737a809a">mxnet::ExecType::kLocal</a></div><div class="ttdoc">Run this operator on the scheduling thread without pushing to engine. </div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_aa2ea1d7ae06008faf44a92c009b2b26e"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#aa2ea1d7ae06008faf44a92c009b2b26e">mxnet::OpStatePtr::get_state</a></div><div class="ttdeci">T & get_state() const </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:142</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a40ce0e3d4ac21412cfafafc5b1fd32ef"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef">mxnet::OpContext::async_on_complete</a></div><div class="ttdeci">engine::CallbackOnComplete async_on_complete</div><div class="ttdoc">the callback when operation completes, used by asynchronize ops </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:71</div></div> |
| <div class="ttc" id="namespacemxnet_html_acb83431b596f805fabcc4ba400c62e85"><div class="ttname"><a href="namespacemxnet.html#acb83431b596f805fabcc4ba400c62e85">mxnet::FCreateOpState</a></div><div class="ttdeci">std::function< OpStatePtr(const NodeAttrs &attrs, Context ctx, const std::vector< TShape > &in_shape, const std::vector< int > &in_type)> FCreateOpState</div><div class="ttdoc">Create a Layer style, forward/backward operator. This is easy to write code that contains state...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:187</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html"><div class="ttname"><a href="structmxnet_1_1OpContext.html">mxnet::OpContext</a></div><div class="ttdoc">All the possible information needed by Operator.Forward and Backward This is the superset of RunConte...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:65</div></div> |
| <div class="ttc" id="ndarray_8h_html"><div class="ttname"><a href="ndarray_8h.html">ndarray.h</a></div><div class="ttdoc">NDArray interface that handles array arithematics. </div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_abd7405c065167e4e15af032b1e0a5c42"><div class="ttname"><a href="structmxnet_1_1OpContext.html#abd7405c065167e4e15af032b1e0a5c42">mxnet::OpContext::is_train</a></div><div class="ttdeci">int is_train</div><div class="ttdoc">whether it is training phase </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:67</div></div> |
| <div class="ttc" id="namespacemxnet_html_a666f1055e5401ecec0278e8ba5790732"><div class="ttname"><a href="namespacemxnet.html#a666f1055e5401ecec0278e8ba5790732">mxnet::FExecType</a></div><div class="ttdeci">std::function< ExecType(const NodeAttrs &attrs)> FExecType</div><div class="ttdoc">Execution mode of this operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:191</div></div> |
| <div class="ttc" id="classmxnet_1_1Engine_html_a6e141b188f018d5d933ab99868631d5e"><div class="ttname"><a href="classmxnet_1_1Engine.html#a6e141b188f018d5d933ab99868631d5e">mxnet::Engine::NewVariable</a></div><div class="ttdeci">virtual VarHandle NewVariable()=0</div><div class="ttdoc">Allocate a new variable, the variable can then be used to schedule the operation concurrently via dep...</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_ae0f821c7efce61a11730ef8b93070e3e"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#ae0f821c7efce61a11730ef8b93070e3e">mxnet::OpStatePtr::Create</a></div><div class="ttdeci">static OpStatePtr Create(Args &&...args)</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:128</div></div> |
| <div class="ttc" id="namespacemxnet_html_a0906e49a921265634672f555451075ed"><div class="ttname"><a href="namespacemxnet.html#a0906e49a921265634672f555451075ed">mxnet::FNDArrayFunction</a></div><div class="ttdeci">std::function< void(const nnvm::NodeAttrs &attrs, const std::vector< NDArray > &inputs, std::vector< NDArray > *outputs)> FNDArrayFunction</div><div class="ttdoc">Register an operator called as a NDArray function. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:230</div></div> |
| <div class="ttc" id="resource_8h_html"><div class="ttname"><a href="resource_8h.html">resource.h</a></div><div class="ttdoc">Global resource allocation handling. </div></div> |
| <div class="ttc" id="classmxnet_1_1Engine_html_a738e5192dab345ab0ec9888b095903cf"><div class="ttname"><a href="classmxnet_1_1Engine.html#a738e5192dab345ab0ec9888b095903cf">mxnet::Engine::DeleteVariable</a></div><div class="ttdeci">virtual void DeleteVariable(SyncFn delete_fn, Context exec_ctx, VarHandle var)=0</div><div class="ttdoc">Schedule the deletion of a variable. </div></div> |
| <div class="ttc" id="namespacemxnet_1_1engine_html_a9d36c4f33eae8531586dc2edf83ae7cf"><div class="ttname"><a href="namespacemxnet_1_1engine.html#a9d36c4f33eae8531586dc2edf83ae7cf">mxnet::engine::VarHandle</a></div><div class="ttdeci">Var * VarHandle</div><div class="ttdoc">Variable pointer type, usually hold by user used to specify dependencies. </div><div class="ttdef"><b>Definition:</b> engine.h:46</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51">mxnet::kWriteInplace</a></div><div class="ttdoc">perform an inplace write, Target shares memory with one of input arguments. This option only happen w...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:54</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">mxnet::OpReqType</a></div><div class="ttdeci">OpReqType</div><div class="ttdoc">operation request type to Forward and Backward </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:44</div></div> |
| <div class="ttc" id="namespacemxnet_html_a294d2660c065f8d229c27e65f0461e10"><div class="ttname"><a href="namespacemxnet.html#a294d2660c065f8d229c27e65f0461e10">mxnet::FComputeEx</a></div><div class="ttdeci">std::function< void(const nnvm::NodeAttrs &attrs, const OpContext &ctx, const std::vector< NDArray > &inputs, const std::vector< OpReqType > &req, const std::vector< NDArray > &outputs)> FComputeEx</div><div class="ttdoc">Resiger an NDArray compute function for simple stateless forward only operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:251</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a833e95f95cf5e3af9c193b74b500fc5b"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b">mxnet::OpContext::requested</a></div><div class="ttdeci">std::vector< Resource > requested</div><div class="ttdoc">Resources requested by the operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:73</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a869d4c68c0028a175406f11332baf7cf"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf">mxnet::OpContext::run_ctx</a></div><div class="ttdeci">RunContext run_ctx</div><div class="ttdoc">RunContext related resources. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:69</div></div> |
| <div class="ttc" id="structmxnet_1_1Context_html_a4bb231d70623fde216de51589347e513"><div class="ttname"><a href="structmxnet_1_1Context.html#a4bb231d70623fde216de51589347e513">mxnet::Context::CPU</a></div><div class="ttdeci">static Context CPU(int32_t dev_id=0)</div></div> |
| <div class="ttc" id="namespacemxnet_html_ad77e2048b7e8db91a12e05b477aac13e"><div class="ttname"><a href="namespacemxnet.html#ad77e2048b7e8db91a12e05b477aac13e">mxnet::FCompute</a></div><div class="ttdeci">std::function< void(const nnvm::NodeAttrs &attrs, const OpContext &ctx, const std::vector< TBlob > &inputs, const std::vector< OpReqType > &req, const std::vector< TBlob > &outputs)> FCompute</div><div class="ttdoc">Resiger a compute function for simple stateless forward only operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:240</div></div> |
| <div class="ttc" id="classmxnet_1_1engine_1_1CallbackOnComplete_html"><div class="ttname"><a href="classmxnet_1_1engine_1_1CallbackOnComplete.html">mxnet::engine::CallbackOnComplete</a></div><div class="ttdoc">OnComplete Callback to the engine, called by AsyncFn when action completes. </div><div class="ttdef"><b>Definition:</b> engine.h:55</div></div> |
| <div class="ttc" id="base_8h_html"><div class="ttname"><a href="base_8h.html">base.h</a></div><div class="ttdoc">configuation of mxnet as well as basic data structure. </div></div> |
| <div class="ttc" id="classmxnet_1_1Engine_html_ae0a23da15ef63d9479c7468e1f2f825f"><div class="ttname"><a href="classmxnet_1_1Engine.html#ae0a23da15ef63d9479c7468e1f2f825f">mxnet::Engine::Get</a></div><div class="ttdeci">static Engine * Get()</div></div> |
| <div class="ttc" id="namespacemxnet_html_a0a1395f21139ba975a6268539df6a15d"><div class="ttname"><a href="namespacemxnet.html#a0a1395f21139ba975a6268539df6a15d">mxnet::FInferStorageType</a></div><div class="ttdeci">std::function< bool(const NodeAttrs &attrs, const int dev_mask, DispatchMode *dispatch_mode, std::vector< int > *in_attrs, std::vector< int > *out_attrs)> FInferStorageType</div><div class="ttdoc">Resiger a storage and dispatch mode inference function based on storage types of the inputs and outpu...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:263</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383"><div class="ttname"><a href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383">mxnet::kAddTo</a></div><div class="ttdoc">add to the provided space </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:56</div></div> |
| <div class="ttc" id="namespacemxnet_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:86</div></div> |
| <div class="ttc" id="namespacemxnet_html_aad5fec2b4bceabbf4b884460cc2e0a2ba7934d40c7c17315d892e8d5d745b1c66"><div class="ttname"><a href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba7934d40c7c17315d892e8d5d745b1c66">mxnet::DispatchMode::kUndefined</a></div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_aa700c5c28e024b42d41e15f46615f1b8"><div class="ttname"><a href="structmxnet_1_1OpContext.html#aa700c5c28e024b42d41e15f46615f1b8">mxnet::OpContext::get_stream</a></div><div class="ttdeci">mshadow::Stream< xpu > * get_stream() const </div><div class="ttdoc">get mshadow stream from Context </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:80</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html">mxnet::OpStatePtr</a></div><div class="ttdoc">Operator state. This is a pointer type, its content is mutable even if OpStatePtr is const...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:122</div></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Sat Oct 28 2017 08:21:17 for mxnet by <a href="http://www.doxygen.org/index.html"> |
| <img alt="doxygen" class="footer" src="doxygen.png"/> |
| </a> 1.8.6 |
| </small></address> |
| </body> |
| </html> |
