| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.13"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>mxnet: /work/mxnet/include/mxnet/op_attr_types.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.13 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| </script> |
| <div id="main-nav"></div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_d44c64559bbebec7f509842c48db8b23.html">include</a></li><li class="navelem"><a class="el" href="dir_1143c7affb9ebd026cb6818dd282def7.html">mxnet</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">op_attr_types.h</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="include_2mxnet_2op__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="l00026"></a><span class="lineno"> 26</span> <span class="preprocessor">#ifndef MXNET_OP_ATTR_TYPES_H_</span></div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#define MXNET_OP_ATTR_TYPES_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 <<a class="code" href="tensor_8h.html">mshadow/tensor.h</a>></span></div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#include <<a class="code" href="3rdparty_2tvm_2nnvm_2include_2nnvm_2op__attr__types_8h.html">nnvm/op_attr_types.h</a>></span></div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span> </div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#include <vector></span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#include <functional></span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include <string></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="preprocessor">#include "<a class="code" href="include_2mxnet_2base_8h.html">./base.h</a>"</span></div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="preprocessor">#include "<a class="code" href="include_2mxnet_2ndarray_8h.html">./ndarray.h</a>"</span></div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="preprocessor">#include "<a class="code" href="engine_8h.html">./engine.h</a>"</span></div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="preprocessor">#include "<a class="code" href="resource_8h.html">./resource.h</a>"</span></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">namespace </span><a class="code" href="namespacemxnet.html">mxnet</a> {</div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span> </div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span> <span class="keyword">using</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a>;</div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span> </div><div class="line"><a name="l00046"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398"> 46</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">OpReqType</a> {</div><div class="line"><a name="l00048"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26"> 48</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a001b85096e07108e174fe4c406368a26">kNullOp</a>,</div><div class="line"><a name="l00050"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a"> 50</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a03b4c19f34d3add7deecd7ef6c54b42a">kWriteTo</a>,</div><div class="line"><a name="l00056"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51"> 56</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a640322face587ca12684378d0f053d51">kWriteInplace</a>,</div><div class="line"><a name="l00058"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383"> 58</a></span>  <a class="code" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398a1b850737dba4d2c42a2687d14c10f383">kAddTo</a></div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span> };</div><div class="line"><a name="l00060"></a><span class="lineno"> 60</span> </div><div class="line"><a name="l00067"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html"> 67</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1OpContext.html">OpContext</a> {</div><div class="line"><a name="l00069"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#adffa555a95b50893b5fe7a2930a71d62"> 69</a></span>  <span class="keywordtype">bool</span> <a class="code" href="structmxnet_1_1OpContext.html#adffa555a95b50893b5fe7a2930a71d62">need_grad</a>;</div><div class="line"><a name="l00071"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a2d550bd138ed08333bebed8dd876644f"> 71</a></span>  <span class="keywordtype">bool</span> <a class="code" href="structmxnet_1_1OpContext.html#a2d550bd138ed08333bebed8dd876644f">is_train</a>;</div><div class="line"><a name="l00073"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a869d4c68c0028a175406f11332baf7cf"> 73</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="l00075"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a40ce0e3d4ac21412cfafafc5b1fd32ef"> 75</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="l00077"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b"> 77</a></span>  std::vector<Resource> <a class="code" href="structmxnet_1_1OpContext.html#a833e95f95cf5e3af9c193b74b500fc5b">requested</a>;</div><div class="line"><a name="l00083"></a><span class="lineno"> 83</span>  <span class="keyword">template</span><<span class="keyword">typename</span> xpu></div><div class="line"><a name="l00084"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a0c48c2419d23551a626ae3de6ee1f6da"> 84</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<xpu></a>* <a class="code" href="structmxnet_1_1OpContext.html#a0c48c2419d23551a626ae3de6ee1f6da">get_stream</a>()<span class="keyword"> const </span>{</div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span>  <span class="keywordflow">return</span> run_ctx.<a class="code" href="structmxnet_1_1RunContext.html#a9901db00632774aef42081ceda0d23f1">get_stream</a><xpu>();</div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  }</div><div class="line"><a name="l00087"></a><span class="lineno"> 87</span> <span class="preprocessor">#if MXNET_USE_CUDA</span></div><div class="line"><a name="l00088"></a><span class="lineno"> 88</span> </div><div class="line"><a name="l00092"></a><span class="lineno"><a class="line" href="structmxnet_1_1OpContext.html#a1c05a1376bc046ff717827643af4974a"> 92</a></span>  <span class="keyword">inline</span> <a class="code" href="classmxnet_1_1SyncedGPUAuxStream.html">SyncedGPUAuxStream</a> <a class="code" href="structmxnet_1_1OpContext.html#a1c05a1376bc046ff717827643af4974a">get_gpu_aux_stream</a>()<span class="keyword"> const </span>{</div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span>  <span class="keywordflow">return</span> run_ctx.<a class="code" href="structmxnet_1_1RunContext.html#a5591cfe410da5ab004c4733a8a2fb0c7">get_gpu_aux_stream</a>();</div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span>  }</div><div class="line"><a name="l00095"></a><span class="lineno"> 95</span> <span class="preprocessor">#endif</span></div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span> };</div><div class="line"><a name="l00097"></a><span class="lineno"> 97</span> </div><div class="line"><a name="l00099"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad"> 99</a></span> <span class="keyword">enum class</span> <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ad">ExecType</a> {</div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span>  <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adae3bade32f12a5139b27ef40b3689b77c">kSync</a>,</div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>  <a class="code" href="namespacemxnet.html#a998b74220fab2b012cf8a179650e1b3ba9f2b960005d2a3a5f35ac32809d84db7">kAsync</a>,</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071ada3864860a073c64d74d3fc8e765fa6cad">kCrossDeviceCopy</a>,</div><div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <a class="code" href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adabfccd41d634d8095c956c29c9ee35257">kSubgraphExec</a>,</div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span> };</div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span> </div><div class="line"><a name="l00123"></a><span class="lineno"><a class="line" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b"> 123</a></span> <span class="keyword">enum class</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a> {</div><div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba7934d40c7c17315d892e8d5d745b1c66">kUndefined</a> = -1,</div><div class="line"><a name="l00125"></a><span class="lineno"> 125</span>  <span class="comment">// dispatch on FCompute or FStatefulCompute</span></div><div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba2157864769b54d1d0f8f9be63393e7d6">kFCompute</a>,</div><div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  <span class="comment">// dispatch on FComputeEx or FStatefulComputeEx, if available</span></div><div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba57f93219462690bc438372ecd92fedd5">kFComputeEx</a>,</div><div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <span class="comment">// dispatch on FCompute or FStatefulCompute, and performs storage fallback</span></div><div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2ba18bf63b4c196f79c9bca4ad8637e8fef">kFComputeFallback</a>,</div><div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  <span class="comment">// special dispatch mode for variables</span></div><div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2bafb751f6ee1af15e773242f2accd2dbff">kVariable</a>,</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> </div><div class="line"><a name="l00136"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a8e417d85458838afaba8b269f082226d"> 136</a></span> <span class="keyword">enum class</span> <a class="code" href="namespacemxnet.html#a8e417d85458838afaba8b269f082226d">QuantizeType</a> {</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  <span class="comment">// This operator doesn't support quantization</span></div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  <a class="code" href="namespacemxnet.html#a8e417d85458838afaba8b269f082226da35c3ace1970663a16e5c65baa5941b13">kNone</a> = 0,</div><div class="line"><a name="l00139"></a><span class="lineno"> 139</span>  <span class="comment">// This operator can get huge benefit from quantization, thus must be quantized</span></div><div class="line"><a name="l00140"></a><span class="lineno"> 140</span>  <a class="code" href="namespacemxnet.html#a8e417d85458838afaba8b269f082226da9d2303d7a748e1b486e5aeb1f2194610">kMust</a>,</div><div class="line"><a name="l00141"></a><span class="lineno"> 141</span>  <span class="comment">// This operator support quantization, but will be decided depending on the connection</span></div><div class="line"><a name="l00142"></a><span class="lineno"> 142</span>  <a class="code" href="namespacemxnet.html#a8e417d85458838afaba8b269f082226daaab0226a1be7a88fdffaa2442c55aa60">kSupport</a>,</div><div class="line"><a name="l00143"></a><span class="lineno"> 143</span> };</div><div class="line"><a name="l00144"></a><span class="lineno"> 144</span> </div><div class="line"><a name="l00149"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html"> 149</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> {</div><div class="line"><a name="l00150"></a><span class="lineno"> 150</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00151"></a><span class="lineno"> 151</span>  <span class="comment">/* \brief Create a OpStatePtr with state of type T.</span></div><div class="line"><a name="l00152"></a><span class="lineno"> 152</span> <span class="comment"> * \param args Arguments passed to T's constructor.</span></div><div class="line"><a name="l00153"></a><span class="lineno"> 153</span> <span class="comment"> */</span></div><div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  <span class="keyword">template</span><<span class="keyword">typename</span> T, <span class="keyword">typename</span>... Args></div><div class="line"><a name="l00155"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#a2484d61830b33181bf24c536ba91b790"> 155</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#a2484d61830b33181bf24c536ba91b790">Create</a>(Args&&... args) {</div><div class="line"><a name="l00156"></a><span class="lineno"> 156</span>  <a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> ret;</div><div class="line"><a name="l00157"></a><span class="lineno"> 157</span>  <span class="keyword">auto</span> state = <span class="keyword">new</span> T(std::forward<Args>(args)...);</div><div class="line"><a name="l00158"></a><span class="lineno"> 158</span>  <span class="keyword">auto</span> 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="l00159"></a><span class="lineno"> 159</span>  ret.ptr_.reset(</div><div class="line"><a name="l00160"></a><span class="lineno"> 160</span>  <span class="keyword">new</span> OpState(var, state),</div><div class="line"><a name="l00161"></a><span class="lineno"> 161</span>  [](OpState* p) {</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>([](<a class="code" href="structmxnet_1_1RunContext.html">RunContext</a> s) {}, <a class="code" href="structmxnet_1_1Context.html#a4bb231d70623fde216de51589347e513">Context::CPU</a>(), p->var);</div><div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  <span class="keyword">delete</span> <span class="keyword">reinterpret_cast<</span>T*<span class="keyword">></span>(p->state);</div><div class="line"><a name="l00164"></a><span class="lineno"> 164</span>  <span class="keyword">delete</span> p;</div><div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  });</div><div class="line"><a name="l00166"></a><span class="lineno"> 166</span> </div><div class="line"><a name="l00167"></a><span class="lineno"> 167</span>  <span class="keywordflow">return</span> ret;</div><div class="line"><a name="l00168"></a><span class="lineno"> 168</span>  }</div><div class="line"><a name="l00169"></a><span class="lineno"> 169</span>  <span class="comment">/* \brief Get engine variable associated with this state */</span></div><div class="line"><a name="l00170"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#af2082fb6224b78b2aac9e05f4b5365a4"> 170</a></span>  <a class="code" href="structmxnet_1_1engine_1_1Var.html">engine::VarHandle</a> <a class="code" href="classmxnet_1_1OpStatePtr.html#af2082fb6224b78b2aac9e05f4b5365a4">get_var</a>()<span class="keyword"> const </span>{</div><div class="line"><a name="l00171"></a><span class="lineno"> 171</span>  <span class="keywordflow">return</span> ptr_->var;</div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span>  }</div><div class="line"><a name="l00173"></a><span class="lineno"> 173</span>  <span class="comment">/* \brief Get state of type T */</span></div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span>  <span class="keyword">template</span><<span class="keyword">typename</span> T></div><div class="line"><a name="l00175"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#a6d720b8843340fa4d9a7ff7937f2e141"> 175</a></span>  T& <a class="code" href="classmxnet_1_1OpStatePtr.html#a6d720b8843340fa4d9a7ff7937f2e141">get_state</a>()<span class="keyword"> const </span>{</div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span>  <span class="keywordflow">return</span> *<span class="keyword">reinterpret_cast<</span>T*<span class="keyword">></span>(ptr_->state);</div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  }</div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span>  <span class="comment">/* \brief clear state */</span></div><div class="line"><a name="l00179"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#ae43213b3a65073a87bf489350a58d78e"> 179</a></span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1OpStatePtr.html#ae43213b3a65073a87bf489350a58d78e">reset</a>() {</div><div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  ptr_.reset();</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="comment">/* \brief checks whether the managed object is managed only by the current</span></div><div class="line"><a name="l00183"></a><span class="lineno"> 183</span> <span class="comment"> OpStatePtr instance */</span></div><div class="line"><a name="l00184"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#a369d56ad9ea92ded99e532a76a1e1b91"> 184</a></span>  <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1OpStatePtr.html#a369d56ad9ea92ded99e532a76a1e1b91">unique</a>()<span class="keyword"> const </span>{</div><div class="line"><a name="l00185"></a><span class="lineno"> 185</span>  <span class="keywordflow">return</span> ptr_.unique();</div><div class="line"><a name="l00186"></a><span class="lineno"> 186</span>  }</div><div class="line"><a name="l00187"></a><span class="lineno"> 187</span>  <span class="comment">/* \brief Whether state is empty */</span></div><div class="line"><a name="l00188"></a><span class="lineno"><a class="line" href="classmxnet_1_1OpStatePtr.html#ae233fba51851ddd1704de60c9c860ee3"> 188</a></span>  <span class="keyword">explicit</span> <span class="keyword">operator</span> bool()<span class="keyword"> const </span>{</div><div class="line"><a name="l00189"></a><span class="lineno"> 189</span>  <span class="keywordflow">return</span> ptr_ ? true : <span class="keyword">false</span>;</div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  }</div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span> </div><div class="line"><a name="l00192"></a><span class="lineno"> 192</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  <span class="comment">/* \brief state structure */</span></div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span>  <span class="keyword">struct </span>OpState {</div><div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  <a class="code" href="structmxnet_1_1engine_1_1Var.html">engine::VarHandle</a> var;</div><div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  <span class="keywordtype">void</span>* state;</div><div class="line"><a name="l00197"></a><span class="lineno"> 197</span> </div><div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  OpState(<a class="code" href="structmxnet_1_1engine_1_1Var.html">engine::VarHandle</a> var_, <span class="keywordtype">void</span>* state_) : var(var_), state(state_) {}</div><div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  OpState(<span class="keyword">const</span> OpState& other) = <span class="keyword">delete</span>;</div><div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  OpState& operator=(<span class="keyword">const</span> OpState& other) = <span class="keyword">delete</span>;</div><div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  };</div><div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  <span class="comment">/* \brief shared pointer to state */</span></div><div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  std::shared_ptr<OpState> ptr_;</div><div class="line"><a name="l00204"></a><span class="lineno"> 204</span> };</div><div class="line"><a name="l00205"></a><span class="lineno"> 205</span> </div><div class="line"><a name="l00218"></a><span class="lineno"> 218</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#ac7ea5e68a319e4f7d7c0b4ee91c2d20f">FCreateOpState</a> = std::function<<a class="code" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs,</div><div class="line"><a name="l00219"></a><span class="lineno"> 219</span>  <a class="code" href="structmxnet_1_1Context.html">Context</a> ctx,</div><div class="line"><a name="l00220"></a><span class="lineno"> 220</span>  <span class="keyword">const</span> <a class="code" href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a>& in_shape,</div><div class="line"><a name="l00221"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ac7ea5e68a319e4f7d7c0b4ee91c2d20f"> 221</a></span>  <span class="keyword">const</span> std::vector<int>& in_type)>;</div><div class="line"><a name="l00222"></a><span class="lineno"> 222</span> </div><div class="line"><a name="l00231"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ab11a4e6d5c8c47c99cfc3306db6538c8"> 231</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#ab11a4e6d5c8c47c99cfc3306db6538c8">THasDeterministicOutput</a> = bool;</div><div class="line"><a name="l00232"></a><span class="lineno"> 232</span> </div><div class="line"><a name="l00236"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a666f1055e5401ecec0278e8ba5790732"> 236</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="l00244"></a><span class="lineno"> 244</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="l00245"></a><span class="lineno"> 245</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div><div class="line"><a name="l00246"></a><span class="lineno"> 246</span>  <span class="keyword">const</span> std::vector<TBlob>& inputs,</div><div class="line"><a name="l00247"></a><span class="lineno"> 247</span>  <span class="keyword">const</span> std::vector<OpReqType>& req,</div><div class="line"><a name="l00248"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a9097a269dd76d833c014149ff3eaab9b"> 248</a></span>  <span class="keyword">const</span> std::vector<TBlob>& outputs)>;</div><div class="line"><a name="l00256"></a><span class="lineno"> 256</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="l00257"></a><span class="lineno"> 257</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div><div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  <span class="keyword">const</span> std::vector<NDArray>& inputs,</div><div class="line"><a name="l00259"></a><span class="lineno"> 259</span>  <span class="keyword">const</span> std::vector<OpReqType>& req,</div><div class="line"><a name="l00260"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ab12f1853179b1456152a563edff844a9"> 260</a></span>  <span class="keyword">const</span> std::vector<NDArray>& outputs)>;</div><div class="line"><a name="l00267"></a><span class="lineno"> 267</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#afaa687cbca9e992614be285711c1d72d">FResourceRequest</a> = std::function<</div><div class="line"><a name="l00268"></a><span class="lineno"><a class="line" href="namespacemxnet.html#afaa687cbca9e992614be285711c1d72d"> 268</a></span>  std::vector<ResourceRequest> (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& n)>;</div><div class="line"><a name="l00277"></a><span class="lineno"> 277</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a45b0a850c44c45609adc320b04c7ac7f">FResourceRequestEx</a> = std::function<</div><div class="line"><a name="l00278"></a><span class="lineno"> 278</span>  std::vector<ResourceRequest> (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& n,</div><div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  <span class="keyword">const</span> <span class="keywordtype">int</span> dev_mask,</div><div class="line"><a name="l00280"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a45b0a850c44c45609adc320b04c7ac7f"> 280</a></span>  <span class="keyword">const</span> <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a> dispatch_mode)>;</div><div class="line"><a name="l00286"></a><span class="lineno"> 286</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a0906e49a921265634672f555451075ed">FNDArrayFunction</a> = std::function<void (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a>& attrs,</div><div class="line"><a name="l00287"></a><span class="lineno"> 287</span>  <span class="keyword">const</span> std::vector<NDArray>& inputs,</div><div class="line"><a name="l00288"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a0906e49a921265634672f555451075ed"> 288</a></span>  std::vector<NDArray>* outputs)>;</div><div class="line"><a name="l00294"></a><span class="lineno"> 294</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#ad77e2048b7e8db91a12e05b477aac13e">FCompute</a> = std::function<void (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a>& attrs,</div><div class="line"><a name="l00295"></a><span class="lineno"> 295</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div><div class="line"><a name="l00296"></a><span class="lineno"> 296</span>  <span class="keyword">const</span> std::vector<TBlob>& inputs,</div><div class="line"><a name="l00297"></a><span class="lineno"> 297</span>  <span class="keyword">const</span> std::vector<OpReqType>& req,</div><div class="line"><a name="l00298"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ad77e2048b7e8db91a12e05b477aac13e"> 298</a></span>  <span class="keyword">const</span> std::vector<TBlob>& outputs)>;</div><div class="line"><a name="l00304"></a><span class="lineno"> 304</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a294d2660c065f8d229c27e65f0461e10">FComputeEx</a> = std::function<void (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a>& attrs,</div><div class="line"><a name="l00305"></a><span class="lineno"> 305</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1OpContext.html">OpContext</a>& ctx,</div><div class="line"><a name="l00306"></a><span class="lineno"> 306</span>  <span class="keyword">const</span> std::vector<NDArray>& inputs,</div><div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  <span class="keyword">const</span> std::vector<OpReqType>& req,</div><div class="line"><a name="l00308"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a294d2660c065f8d229c27e65f0461e10"> 308</a></span>  <span class="keyword">const</span> std::vector<NDArray>& outputs)>;</div><div class="line"><a name="l00309"></a><span class="lineno"> 309</span> </div><div class="line"><a name="l00316"></a><span class="lineno"> 316</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a0a1395f21139ba975a6268539df6a15d">FInferStorageType</a> = std::function<bool (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs,</div><div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  <span class="keyword">const</span> <span class="keywordtype">int</span> dev_mask,</div><div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  <a class="code" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a>* dispatch_mode,</div><div class="line"><a name="l00319"></a><span class="lineno"> 319</span>  std::vector<int>* in_attrs,</div><div class="line"><a name="l00320"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a0a1395f21139ba975a6268539df6a15d"> 320</a></span>  std::vector<int>* out_attrs)>;</div><div class="line"><a name="l00321"></a><span class="lineno"> 321</span> </div><div class="line"><a name="l00326"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ae5d610fd675a9b9c811901394ee07709"> 326</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#ae5d610fd675a9b9c811901394ee07709">FQuantizable</a> = std::function<QuantizeType (const NodeAttrs& attrs)>;</div><div class="line"><a name="l00327"></a><span class="lineno"> 327</span> </div><div class="line"><a name="l00332"></a><span class="lineno"><a class="line" href="namespacemxnet.html#aca9a7c38c0a9b100a35eba5f3b0ea108"> 332</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#aca9a7c38c0a9b100a35eba5f3b0ea108">FQuantizedOp</a> = std::function<nnvm::ObjectPtr (const NodeAttrs& attrs)>;</div><div class="line"><a name="l00333"></a><span class="lineno"> 333</span> </div><div class="line"><a name="l00340"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a7f7c271f6d03ee358eb9e66486e41045"> 340</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a7f7c271f6d03ee358eb9e66486e41045">FNeedRequantize</a> = std::function<bool (const NodeAttrs& attrs)>;</div><div class="line"><a name="l00341"></a><span class="lineno"> 341</span> </div><div class="line"><a name="l00347"></a><span class="lineno"> 347</span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a3cb7e17c82420da9bac723fee0bb044c">FAvoidQuantizeInput</a> = std::function<bool (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs,</div><div class="line"><a name="l00348"></a><span class="lineno"> 348</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span> index,</div><div class="line"><a name="l00349"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a3cb7e17c82420da9bac723fee0bb044c"> 349</a></span>  <span class="keyword">const</span> std::string quantize_granularity)>;</div><div class="line"><a name="l00350"></a><span class="lineno"> 350</span> </div><div class="line"><a name="l00356"></a><span class="lineno"><a class="line" href="namespacemxnet.html#abbd92d7ce22448f485be2946e79322bd"> 356</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#abbd92d7ce22448f485be2946e79322bd">FNeedCalibrateInput</a> = std::function<std::vector<int> (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div><div class="line"><a name="l00357"></a><span class="lineno"> 357</span> </div><div class="line"><a name="l00363"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a5874b9dfca8a1e2e7c72da5e3b6a9d0b"> 363</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a5874b9dfca8a1e2e7c72da5e3b6a9d0b">FNeedCalibrateOutput</a> = std::function<std::vector<int> (<span class="keyword">const</span> <a class="code" href="structnnvm_1_1NodeAttrs.html">NodeAttrs</a>& attrs)>;</div><div class="line"><a name="l00364"></a><span class="lineno"> 364</span> </div><div class="line"><a name="l00365"></a><span class="lineno"> 365</span> <span class="preprocessor">#if MXNET_USE_CUDA</span></div><div class="line"><a name="l00366"></a><span class="lineno"> 366</span> </div><div class="line"><a name="l00374"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a1f299d6a9a65bdf12162a5dcba62b710"> 374</a></span> <span class="keyword">using</span> <a class="code" href="namespacemxnet.html#a1f299d6a9a65bdf12162a5dcba62b710">FIsCUDAGraphsCompatible</a> = std::function<bool (const NodeAttrs& attrs, const bool is_train)>;</div><div class="line"><a name="l00375"></a><span class="lineno"> 375</span> </div><div class="line"><a name="l00376"></a><span class="lineno"> 376</span> <span class="preprocessor">#endif</span></div><div class="line"><a name="l00377"></a><span class="lineno"> 377</span> </div><div class="line"><a name="l00378"></a><span class="lineno"> 378</span> } <span class="comment">// namespace mxnet</span></div><div class="line"><a name="l00379"></a><span class="lineno"> 379</span> </div><div class="line"><a name="l00380"></a><span class="lineno"> 380</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&#39;s content is mutab...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:248</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_a6d720b8843340fa4d9a7ff7937f2e141"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#a6d720b8843340fa4d9a7ff7937f2e141">mxnet::OpStatePtr::get_state</a></div><div class="ttdeci">T & get_state() const</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:175</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:179</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_a369d56ad9ea92ded99e532a76a1e1b91"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#a369d56ad9ea92ded99e532a76a1e1b91">mxnet::OpStatePtr::unique</a></div><div class="ttdeci">bool unique() const</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:184</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 synchronous 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_ab11a4e6d5c8c47c99cfc3306db6538c8"><div class="ttname"><a href="namespacemxnet.html#ab11a4e6d5c8c47c99cfc3306db6538c8">mxnet::THasDeterministicOutput</a></div><div class="ttdeci">bool THasDeterministicOutput</div><div class="ttdoc">Whether the operator always produces the same output given the same input. This enables certain optim...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:231</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:48</div></div> |
| <div class="ttc" id="structnnvm_1_1NodeAttrs_html"><div class="ttname"><a href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a></div><div class="ttdoc">The attributes of the current operation node. Usually are additional parameters like axis...</div><div class="ttdef"><b>Definition:</b> node.h:119</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_a2484d61830b33181bf24c536ba91b790"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#a2484d61830b33181bf24c536ba91b790">mxnet::OpStatePtr::Create</a></div><div class="ttdeci">static OpStatePtr Create(Args &&... args)</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:155</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:50</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a1c05a1376bc046ff717827643af4974a"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a1c05a1376bc046ff717827643af4974a">mxnet::OpContext::get_gpu_aux_stream</a></div><div class="ttdeci">SyncedGPUAuxStream get_gpu_aux_stream() const</div><div class="ttdoc">get auxilary gpu stream auto-syncing object from Context </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:92</div></div> |
| <div class="ttc" id="namespacemxnet_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="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. An operator could register ResourceRequestEx, or ResourceRequest, or neither. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:268</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:260</div></div> |
| <div class="ttc" id="namespacemxnet_html_a1f299d6a9a65bdf12162a5dcba62b710"><div class="ttname"><a href="namespacemxnet.html#a1f299d6a9a65bdf12162a5dcba62b710">mxnet::FIsCUDAGraphsCompatible</a></div><div class="ttdeci">std::function< bool(const NodeAttrs &attrs, const bool is_train)> FIsCUDAGraphsCompatible</div><div class="ttdoc">Register a function to determine if the operator implementation is compatible with CUDA graphs...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:374</div></div> |
| <div class="ttc" id="structmxnet_1_1OpContext_html_a0c48c2419d23551a626ae3de6ee1f6da"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a0c48c2419d23551a626ae3de6ee1f6da">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:84</div></div> |
| <div class="ttc" id="namespacemxnet_html_a7f7c271f6d03ee358eb9e66486e41045"><div class="ttname"><a href="namespacemxnet.html#a7f7c271f6d03ee358eb9e66486e41045">mxnet::FNeedRequantize</a></div><div class="ttdeci">std::function< bool(const NodeAttrs &attrs)> FNeedRequantize</div><div class="ttdoc">Register a function to determine if the output of a quantized operator needs to be requantized...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:340</div></div> |
| <div class="ttc" id="namespacemxnet_html_a8e417d85458838afaba8b269f082226da9d2303d7a748e1b486e5aeb1f2194610"><div class="ttname"><a href="namespacemxnet.html#a8e417d85458838afaba8b269f082226da9d2303d7a748e1b486e5aeb1f2194610">mxnet::QuantizeType::kMust</a></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="structmxnet_1_1OpContext_html_a2d550bd138ed08333bebed8dd876644f"><div class="ttname"><a href="structmxnet_1_1OpContext.html#a2d550bd138ed08333bebed8dd876644f">mxnet::OpContext::is_train</a></div><div class="ttdeci">bool is_train</div><div class="ttdoc">whether it is training phase </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:71</div></div> |
| <div class="ttc" id="structmxnet_1_1RunContext_html_a5591cfe410da5ab004c4733a8a2fb0c7"><div class="ttname"><a href="structmxnet_1_1RunContext.html#a5591cfe410da5ab004c4733a8a2fb0c7">mxnet::RunContext::get_gpu_aux_stream</a></div><div class="ttdeci">SyncedGPUAuxStream get_gpu_aux_stream() const</div><div class="ttdoc">get an RAII object that transparently handles the syncing of the auxiliary stream. </div><div class="ttdef"><b>Definition:</b> base.h:379</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 it will 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:350</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:123</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="structmxnet_1_1engine_1_1Var_html"><div class="ttname"><a href="structmxnet_1_1engine_1_1Var.html">mxnet::engine::Var</a></div><div class="ttdoc">base class of engine variables. </div><div class="ttdef"><b>Definition:</b> engine.h:44</div></div> |
| <div class="ttc" id="classmxnet_1_1SyncedGPUAuxStream_html"><div class="ttname"><a href="classmxnet_1_1SyncedGPUAuxStream.html">mxnet::SyncedGPUAuxStream</a></div><div class="ttdoc">Provides automatic coordination of an auxilary stream with a primary one. This object, upon construction, prepares an aux stream for use by syncing it with enqueued primary-stream work. Object destruction will sync again so future primary-stream work will wait on enqueued aux-stream work. If MXNET_GPU_WORKER_NSTREAMS == 1, then this defaults simply: the primary stream will equal the aux stream and the syncs will be executed as nops. See ./src/operator/cudnn/cudnn_convolution-inl.h for a usage example. </div><div class="ttdef"><b>Definition:</b> base.h:315</div></div> |
| <div class="ttc" id="structmxnet_1_1RunContext_html_a9901db00632774aef42081ceda0d23f1"><div class="ttname"><a href="structmxnet_1_1RunContext.html#a9901db00632774aef42081ceda0d23f1">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:371</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:75</div></div> |
| <div class="ttc" id="tensor_8h_html"><div class="ttname"><a href="tensor_8h.html">tensor.h</a></div><div class="ttdoc">header file of tensor data structure and functions This lib requires explicit memory allocation and d...</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:67</div></div> |
| <div class="ttc" id="namespacemxnet_html_a8e417d85458838afaba8b269f082226da35c3ace1970663a16e5c65baa5941b13"><div class="ttname"><a href="namespacemxnet.html#a8e417d85458838afaba8b269f082226da35c3ace1970663a16e5c65baa5941b13">mxnet::QuantizeType::kNone</a></div></div> |
| <div class="ttc" id="namespacemxnet_html_a87e1c30aafac389774c6f4d905c0e5ea"><div class="ttname"><a href="namespacemxnet.html#a87e1c30aafac389774c6f4d905c0e5ea">mxnet::ShapeVector</a></div><div class="ttdeci">std::vector< mxnet::TShape > ShapeVector</div><div class="ttdoc">The result holder of shape of each NodeEntry in the graph. </div><div class="ttdef"><b>Definition:</b> tuple.h:820</div></div> |
| <div class="ttc" id="namespacemxnet_html_a8e417d85458838afaba8b269f082226d"><div class="ttname"><a href="namespacemxnet.html#a8e417d85458838afaba8b269f082226d">mxnet::QuantizeType</a></div><div class="ttdeci">QuantizeType</div><div class="ttdoc">the quantization type of the operator </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:136</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:236</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="structmxnet_1_1OpContext_html_adffa555a95b50893b5fe7a2930a71d62"><div class="ttname"><a href="structmxnet_1_1OpContext.html#adffa555a95b50893b5fe7a2930a71d62">mxnet::OpContext::need_grad</a></div><div class="ttdeci">bool need_grad</div><div class="ttdoc">whether there is a backward phase to compute gradients. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:69</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:288</div></div> |
| <div class="ttc" id="namespacemxnet_html_aca9a7c38c0a9b100a35eba5f3b0ea108"><div class="ttname"><a href="namespacemxnet.html#aca9a7c38c0a9b100a35eba5f3b0ea108">mxnet::FQuantizedOp</a></div><div class="ttdeci">std::function< nnvm::ObjectPtr(const NodeAttrs &attrs)> FQuantizedOp</div><div class="ttdoc">Register a quantized node creation function based on the attrs of the node. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:332</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_html_ac7ea5e68a319e4f7d7c0b4ee91c2d20f"><div class="ttname"><a href="namespacemxnet.html#ac7ea5e68a319e4f7d7c0b4ee91c2d20f">mxnet::FCreateOpState</a></div><div class="ttdeci">std::function< OpStatePtr(const NodeAttrs &attrs, Context ctx, const mxnet::ShapeVector &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:221</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, This option only happen when Target shares memory with one of input argumen...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:56</div></div> |
| <div class="ttc" id="namespacemxnet_html_ad7801fb2dd68294d4c7a9e289bc071adabfccd41d634d8095c956c29c9ee35257"><div class="ttname"><a href="namespacemxnet.html#ad7801fb2dd68294d4c7a9e289bc071adabfccd41d634d8095c956c29c9ee35257">mxnet::ExecType::kSubgraphExec</a></div><div class="ttdoc">A subgraph execution should happen in the main thread, instead of in the execution engine...</div></div> |
| <div class="ttc" id="include_2mxnet_2base_8h_html"><div class="ttname"><a href="include_2mxnet_2base_8h.html">base.h</a></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:46</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">Register an NDArray compute function for simple stateless forward only operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:308</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:77</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:73</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">Register a compute function for simple stateless forward only operator. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:298</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:73</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">Register a storage and dispatch mode inference function based on storage types of the inputs and outp...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:320</div></div> |
| <div class="ttc" id="namespacemxnet_html_a45b0a850c44c45609adc320b04c7ac7f"><div class="ttname"><a href="namespacemxnet.html#a45b0a850c44c45609adc320b04c7ac7f">mxnet::FResourceRequestEx</a></div><div class="ttdeci">std::function< std::vector< ResourceRequest >(const NodeAttrs &n, const int dev_mask, const DispatchMode dispatch_mode)> FResourceRequestEx</div><div class="ttdoc">The resource request from the operator. An operator could register ResourceRequestEx, or ResourceRequest, or neither. If an operator registers both ResourceRequestEx and ResourceRequest, ResourceRequest is ignored. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:280</div></div> |
| <div class="ttc" id="namespacemxnet_html_abbd92d7ce22448f485be2946e79322bd"><div class="ttname"><a href="namespacemxnet.html#abbd92d7ce22448f485be2946e79322bd">mxnet::FNeedCalibrateInput</a></div><div class="ttdeci">std::function< std::vector< int >(const NodeAttrs &attrs)> FNeedCalibrateInput</div><div class="ttdoc">Register a function to determine if the input of a quantized operator needs to be calibrated...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:356</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:58</div></div> |
| <div class="ttc" id="namespacemxnet_html_a3cb7e17c82420da9bac723fee0bb044c"><div class="ttname"><a href="namespacemxnet.html#a3cb7e17c82420da9bac723fee0bb044c">mxnet::FAvoidQuantizeInput</a></div><div class="ttdeci">std::function< bool(const NodeAttrs &attrs, const size_t index, const std::string quantize_granularity)> FAvoidQuantizeInput</div><div class="ttdoc">Register a function to determine if the input of a quantized operator needs to be quantized...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:349</div></div> |
| <div class="ttc" id="namespacemxnet_html_a8e417d85458838afaba8b269f082226daaab0226a1be7a88fdffaa2442c55aa60"><div class="ttname"><a href="namespacemxnet.html#a8e417d85458838afaba8b269f082226daaab0226a1be7a88fdffaa2442c55aa60">mxnet::QuantizeType::kSupport</a></div></div> |
| <div class="ttc" id="namespacemxnet_html_a5874b9dfca8a1e2e7c72da5e3b6a9d0b"><div class="ttname"><a href="namespacemxnet.html#a5874b9dfca8a1e2e7c72da5e3b6a9d0b">mxnet::FNeedCalibrateOutput</a></div><div class="ttdeci">std::function< std::vector< int >(const NodeAttrs &attrs)> FNeedCalibrateOutput</div><div class="ttdoc">Register a function to determine if the output of a quantized operator needs to be calibrated...</div><div class="ttdef"><b>Definition:</b> op_attr_types.h:363</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:99</div></div> |
| <div class="ttc" id="structmxnet_1_1Context_html"><div class="ttname"><a href="structmxnet_1_1Context.html">mxnet::Context</a></div><div class="ttdoc">Context information about the execution environment. </div><div class="ttdef"><b>Definition:</b> base.h:102</div></div> |
| <div class="ttc" id="include_2mxnet_2ndarray_8h_html"><div class="ttname"><a href="include_2mxnet_2ndarray_8h.html">ndarray.h</a></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="namespacemxnet_html_ae5d610fd675a9b9c811901394ee07709"><div class="ttname"><a href="namespacemxnet.html#ae5d610fd675a9b9c811901394ee07709">mxnet::FQuantizable</a></div><div class="ttdeci">std::function< QuantizeType(const NodeAttrs &attrs)> FQuantizable</div><div class="ttdoc">Register a quantized node creation function based on the attrs of the node. </div><div class="ttdef"><b>Definition:</b> op_attr_types.h:326</div></div> |
| <div class="ttc" id="3rdparty_2tvm_2nnvm_2include_2nnvm_2op__attr__types_8h_html"><div class="ttname"><a href="3rdparty_2tvm_2nnvm_2include_2nnvm_2op__attr__types_8h.html">op_attr_types.h</a></div><div class="ttdoc">Data structures that can appear in operator attributes. </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:149</div></div> |
| <div class="ttc" id="structmshadow_1_1Stream_html"><div class="ttname"><a href="structmshadow_1_1Stream.html">mshadow::Stream</a></div><div class="ttdoc">computaion stream structure, used for asynchronous computations </div><div class="ttdef"><b>Definition:</b> tensor.h:384</div></div> |
| <div class="ttc" id="classmxnet_1_1OpStatePtr_html_af2082fb6224b78b2aac9e05f4b5365a4"><div class="ttname"><a href="classmxnet_1_1OpStatePtr.html#af2082fb6224b78b2aac9e05f4b5365a4">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:170</div></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Mon Apr 19 2021 18:14:34 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.13 |
| </small></address> |
| </body> |
| </html> |