| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.17"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>mxnet: /work/mxnet/include/mxnet/lib_api.h Source File</title> |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="jquery.js"></script> |
| <script type="text/javascript" src="dynsections.js"></script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="search/searchdata.js"></script> |
| <script type="text/javascript" src="search/search.js"></script> |
| <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| <div id="titlearea"> |
| <table cellspacing="0" cellpadding="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| <td id="projectalign" style="padding-left: 0.5em;"> |
| <div id="projectname">mxnet |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.17 --> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| /* @license-end */ |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| /* @license-end */</script> |
| <div id="main-nav"></div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_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">lib_api.h</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="lib__api_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="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#ifndef MXNET_LIB_API_H_</span></div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#define MXNET_LIB_API_H_</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 <stdint.h></span></div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#include <stdlib.h></span></div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#include <string.h></span></div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="preprocessor">#include <vector></span></div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="preprocessor">#include <map></span></div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="preprocessor">#include <unordered_set></span></div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span> <span class="preprocessor">#include <unordered_map></span></div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="preprocessor">#include <string></span></div> |
| <div class="line"><a name="l00042"></a><span class="lineno"> 42</span> <span class="preprocessor">#include <iostream></span></div> |
| <div class="line"><a name="l00043"></a><span class="lineno"> 43</span> <span class="preprocessor">#include <utility></span></div> |
| <div class="line"><a name="l00044"></a><span class="lineno"> 44</span> <span class="preprocessor">#include <stdexcept></span></div> |
| <div class="line"><a name="l00045"></a><span class="lineno"> 45</span> <span class="preprocessor">#include <functional></span></div> |
| <div class="line"><a name="l00046"></a><span class="lineno"> 46</span> <span class="preprocessor">#include <random></span></div> |
| <div class="line"><a name="l00047"></a><span class="lineno"> 47</span> <span class="preprocessor">#include <sstream></span></div> |
| <div class="line"><a name="l00048"></a><span class="lineno"> 48</span>  </div> |
| <div class="line"><a name="l00049"></a><span class="lineno"> 49</span> <span class="preprocessor">#if defined(__NVCC__)</span></div> |
| <div class="line"><a name="l00050"></a><span class="lineno"> 50</span> <span class="preprocessor">#include <cuda_runtime.h></span></div> |
| <div class="line"><a name="l00051"></a><span class="lineno"> 51</span> <span class="preprocessor">#include <curand_kernel.h></span></div> |
| <div class="line"><a name="l00052"></a><span class="lineno"> 52</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00053"></a><span class="lineno"> 53</span>  </div> |
| <div class="line"><a name="l00054"></a><span class="lineno"> 54</span> <span class="comment">/* Make sure to update the version number everytime you make changes */</span></div> |
| <div class="line"><a name="l00055"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aae1f2ee7f9331ffbd79db3708fd193ed"> 55</a></span> <span class="preprocessor">#define MX_LIBRARY_VERSION 11</span></div> |
| <div class="line"><a name="l00056"></a><span class="lineno"> 56</span>  </div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span> <span class="preprocessor">#if defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__)</span></div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span> <span class="preprocessor">#define PRIVATE_SYMBOL</span></div> |
| <div class="line"><a name="l00064"></a><span class="lineno"> 64</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00065"></a><span class="lineno"><a class="line" href="lib__api_8h.html#acb30ec605ba828f085c28799b89383ad"> 65</a></span> <span class="preprocessor">#define PRIVATE_SYMBOL __attribute__((visibility("hidden")))</span></div> |
| <div class="line"><a name="l00066"></a><span class="lineno"> 66</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  </div> |
| <div class="line"><a name="l00068"></a><span class="lineno"> 68</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00069"></a><span class="lineno"> 69</span> <span class="comment"> * Import from DLPack https://github.com/dmlc/dlpack/blob/master/include/dlpack/dlpack.h</span></div> |
| <div class="line"><a name="l00070"></a><span class="lineno"> 70</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00071"></a><span class="lineno"> 71</span> <span class="preprocessor">#ifndef DLPACK_VERSION</span></div> |
| <div class="line"><a name="l00072"></a><span class="lineno"> 72</span> <span class="preprocessor">#ifdef __cplusplus</span></div> |
| <div class="line"><a name="l00073"></a><span class="lineno"> 73</span> <span class="preprocessor">#define DLPACK_EXTERN_C extern "C"</span></div> |
| <div class="line"><a name="l00074"></a><span class="lineno"> 74</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00075"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a673d5372271f4182a2fb5315c2e335b4"> 75</a></span> <span class="preprocessor">#define DLPACK_EXTERN_C</span></div> |
| <div class="line"><a name="l00076"></a><span class="lineno"> 76</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  </div> |
| <div class="line"><a name="l00079"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a493a8fe63f179b98d4e7a1728659a808"> 79</a></span> <span class="preprocessor">#define DLPACK_VERSION 020</span></div> |
| <div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  </div> |
| <div class="line"><a name="l00082"></a><span class="lineno"> 82</span> <span class="preprocessor">#ifdef _WIN32</span></div> |
| <div class="line"><a name="l00083"></a><span class="lineno"> 83</span> <span class="preprocessor">#ifdef DLPACK_EXPORTS</span></div> |
| <div class="line"><a name="l00084"></a><span class="lineno"> 84</span> <span class="preprocessor">#define DLPACK_DLL __declspec(dllexport)</span></div> |
| <div class="line"><a name="l00085"></a><span class="lineno"> 85</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00086"></a><span class="lineno"> 86</span> <span class="preprocessor">#define DLPACK_DLL __declspec(dllimport)</span></div> |
| <div class="line"><a name="l00087"></a><span class="lineno"> 87</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00089"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad57157a7cb94036d48d0444fd14ab6cf"> 89</a></span> <span class="preprocessor">#define DLPACK_DLL</span></div> |
| <div class="line"><a name="l00090"></a><span class="lineno"> 90</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00091"></a><span class="lineno"> 91</span>  </div> |
| <div class="line"><a name="l00092"></a><span class="lineno"> 92</span> <span class="preprocessor">#include <stdint.h></span></div> |
| <div class="line"><a name="l00093"></a><span class="lineno"> 93</span> <span class="preprocessor">#include <stddef.h></span></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">#ifdef __cplusplus</span></div> |
| <div class="line"><a name="l00096"></a><span class="lineno"> 96</span> <span class="keyword">extern</span> <span class="stringliteral">"C"</span> {</div> |
| <div class="line"><a name="l00097"></a><span class="lineno"> 97</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  </div> |
| <div class="line"><a name="l00101"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573"> 101</a></span> <span class="keyword">typedef</span> <span class="keyword">enum</span> {</div> |
| <div class="line"><a name="l00103"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e"> 103</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e">kDLCPU</a> = 1,</div> |
| <div class="line"><a name="l00105"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680"> 105</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680">kDLGPU</a> = 2,</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a61cf710df30b59b43a5e34e37ecd1973"> 110</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a61cf710df30b59b43a5e34e37ecd1973">kDLCPUPinned</a> = 3,</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a71b2f851779936d824bf2c235ed94a8d"> 112</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a71b2f851779936d824bf2c235ed94a8d">kDLOpenCL</a> = 4,</div> |
| <div class="line"><a name="l00114"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a31aefcffd60630580d74bf055c2c3ba0"> 114</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a31aefcffd60630580d74bf055c2c3ba0">kDLVulkan</a> = 7,</div> |
| <div class="line"><a name="l00116"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a36ced4bf0304d68b5c633ddba5633e66"> 116</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a36ced4bf0304d68b5c633ddba5633e66">kDLMetal</a> = 8,</div> |
| <div class="line"><a name="l00118"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573af063bf37ea24c1839d3a78703c2468e7"> 118</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573af063bf37ea24c1839d3a78703c2468e7">kDLVPI</a> = 9,</div> |
| <div class="line"><a name="l00120"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573adeeb5e6f50f4309676e49f7a12512171"> 120</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573adeeb5e6f50f4309676e49f7a12512171">kDLROCM</a> = 10,</div> |
| <div class="line"><a name="l00126"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573aea5cdddfa3ac11130048ad8e2a012363"> 126</a></span>  <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573aea5cdddfa3ac11130048ad8e2a012363">kDLExtDev</a> = 12,</div> |
| <div class="line"><a name="l00127"></a><span class="lineno"> 127</span> } <a class="code" href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573">DLDeviceType</a>;</div> |
| <div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  </div> |
| <div class="line"><a name="l00132"></a><span class="lineno"> 132</span> <span class="keyword">typedef</span> <span class="keyword">struct </span>{</div> |
| <div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  <a class="code" href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573">DLDeviceType</a> device_type;</div> |
| <div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  <span class="keywordtype">int</span> device_id;</div> |
| <div class="line"><a name="l00137"></a><span class="lineno"> 137</span> } <a class="code" href="structDLContext.html">DLContext</a>;</div> |
| <div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  </div> |
| <div class="line"><a name="l00142"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50"> 142</a></span> <span class="keyword">typedef</span> <span class="keyword">enum</span> {</div> |
| <div class="line"><a name="l00143"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e"> 143</a></span>  <a class="code" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a> = 0U,</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a"> 144</a></span>  <a class="code" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a> = 1U,</div> |
| <div class="line"><a name="l00145"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0"> 145</a></span>  <a class="code" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a> = 2U,</div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span> } <a class="code" href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50">DLDataTypeCode</a>;</div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  </div> |
| <div class="line"><a name="l00156"></a><span class="lineno"> 156</span> <span class="keyword">typedef</span> <span class="keyword">struct </span>{</div> |
| <div class="line"><a name="l00162"></a><span class="lineno"> 162</span>  uint8_t code;</div> |
| <div class="line"><a name="l00166"></a><span class="lineno"> 166</span>  uint8_t bits;</div> |
| <div class="line"><a name="l00168"></a><span class="lineno"> 168</span>  uint16_t lanes;</div> |
| <div class="line"><a name="l00169"></a><span class="lineno"> 169</span> } <a class="code" href="structDLDataType.html">DLDataType</a>;</div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  </div> |
| <div class="line"><a name="l00174"></a><span class="lineno"> 174</span> <span class="keyword">typedef</span> <span class="keyword">struct </span>{</div> |
| <div class="line"><a name="l00194"></a><span class="lineno"> 194</span>  <span class="keywordtype">void</span>* data;</div> |
| <div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  <a class="code" href="structDLContext.html">DLContext</a> ctx;</div> |
| <div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  <span class="keywordtype">int</span> ndim;</div> |
| <div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  <a class="code" href="structDLDataType.html">DLDataType</a> dtype;</div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  int64_t* shape;</div> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  int64_t* strides;</div> |
| <div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  uint64_t byte_offset;</div> |
| <div class="line"><a name="l00210"></a><span class="lineno"> 210</span> } <a class="code" href="structDLTensor.html">DLTensor</a>;</div> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span> <span class="preprocessor">#ifdef __cplusplus</span></div> |
| <div class="line"><a name="l00212"></a><span class="lineno"> 212</span> } <span class="comment">// DLPACK_EXTERN_C</span></div> |
| <div class="line"><a name="l00213"></a><span class="lineno"> 213</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00214"></a><span class="lineno"> 214</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  </div> |
| <div class="line"><a name="l00216"></a><span class="lineno"> 216</span> <span class="keyword">namespace </span><a class="code" href="namespacemxnet.html">mxnet</a> {</div> |
| <div class="line"><a name="l00217"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html"> 217</a></span> <span class="keyword">namespace </span>ext {</div> |
| <div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  </div> |
| <div class="line"><a name="l00219"></a><span class="lineno"> 219</span> <span class="comment">/* \brief Class to store error messages from extensions to pass to MXNet */</span></div> |
| <div class="line"><a name="l00220"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1MXerrorMsgs.html"> 220</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html">MXerrorMsgs</a> {</div> |
| <div class="line"><a name="l00221"></a><span class="lineno"> 221</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00222"></a><span class="lineno"> 222</span>  <span class="comment">/* \brief get singleton pointer to class */</span></div> |
| <div class="line"><a name="l00223"></a><span class="lineno"> 223</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html">MXerrorMsgs</a>* <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a6da046f33093c9274b12f8bf6cdba61e">get</a>();</div> |
| <div class="line"><a name="l00224"></a><span class="lineno"> 224</span>  </div> |
| <div class="line"><a name="l00225"></a><span class="lineno"> 225</span>  <span class="comment">/* \brief add a new error message */</span></div> |
| <div class="line"><a name="l00226"></a><span class="lineno"> 226</span>  std::stringstream& <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#ab794a5b63f8679e704539a9c0bb0d946">add</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* file, <span class="keywordtype">int</span> line);</div> |
| <div class="line"><a name="l00227"></a><span class="lineno"> 227</span>  </div> |
| <div class="line"><a name="l00228"></a><span class="lineno"> 228</span>  <span class="comment">/* \brief return number of error messages */</span></div> |
| <div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a416e02951e24afc2792816f54de16320">size</a>();</div> |
| <div class="line"><a name="l00230"></a><span class="lineno"> 230</span>  </div> |
| <div class="line"><a name="l00231"></a><span class="lineno"> 231</span>  <span class="comment">/* \brief get error message at index */</span></div> |
| <div class="line"><a name="l00232"></a><span class="lineno"> 232</span>  <span class="keyword">const</span> std::string* <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a6da046f33093c9274b12f8bf6cdba61e">get</a>(<span class="keywordtype">int</span> idx);</div> |
| <div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  </div> |
| <div class="line"><a name="l00234"></a><span class="lineno"> 234</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00236"></a><span class="lineno"> 236</span>  <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html">MXerrorMsgs</a>() {}</div> |
| <div class="line"><a name="l00238"></a><span class="lineno"> 238</span>  ~<a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html">MXerrorMsgs</a>() {}</div> |
| <div class="line"><a name="l00240"></a><span class="lineno"> 240</span>  std::vector<std::stringstream> messages;</div> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span> };</div> |
| <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  </div> |
| <div class="line"><a name="l00243"></a><span class="lineno"> 243</span> <span class="comment">// Add a new error message, example: MX_ERROR_MSG << "my error msg";</span></div> |
| <div class="line"><a name="l00244"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a481a7e516c02ab43e5b0e0016cb7997f"> 244</a></span> <span class="preprocessor">#define MX_ERROR_MSG mxnet::ext::MXerrorMsgs::get()->add(__FILE__, __LINE__)</span></div> |
| <div class="line"><a name="l00245"></a><span class="lineno"> 245</span>  </div> |
| <div class="line"><a name="l00249"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f"> 249</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> {</div> |
| <div class="line"><a name="l00250"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704"> 250</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704">kFloat32</a> = 0,</div> |
| <div class="line"><a name="l00251"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884"> 251</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884">kFloat64</a> = 1,</div> |
| <div class="line"><a name="l00252"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103"> 252</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103">kFloat16</a> = 2,</div> |
| <div class="line"><a name="l00253"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2"> 253</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2">kUint8</a> = 3,</div> |
| <div class="line"><a name="l00254"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8"> 254</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8">kInt32</a> = 4,</div> |
| <div class="line"><a name="l00255"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6"> 255</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6">kInt8</a> = 5,</div> |
| <div class="line"><a name="l00256"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869"> 256</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869">kInt64</a> = 6,</div> |
| <div class="line"><a name="l00257"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048"> 257</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048">kUNSET</a> = 100,</div> |
| <div class="line"><a name="l00258"></a><span class="lineno"> 258</span> };</div> |
| <div class="line"><a name="l00259"></a><span class="lineno"> 259</span>  </div> |
| <div class="line"><a name="l00260"></a><span class="lineno"> 260</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00261"></a><span class="lineno"> 261</span> <span class="comment"> * MXTensor storage type.</span></div> |
| <div class="line"><a name="l00262"></a><span class="lineno"> 262</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00263"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166"> 263</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> {</div> |
| <div class="line"><a name="l00264"></a><span class="lineno"> 264</span>  <span class="comment">// dense</span></div> |
| <div class="line"><a name="l00265"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e"> 265</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e">kDefaultStorage</a> = 0,</div> |
| <div class="line"><a name="l00266"></a><span class="lineno"> 266</span>  <span class="comment">// row sparse</span></div> |
| <div class="line"><a name="l00267"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a"> 267</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a">kRowSparseStorage</a> = 1,</div> |
| <div class="line"><a name="l00268"></a><span class="lineno"> 268</span>  <span class="comment">// csr</span></div> |
| <div class="line"><a name="l00269"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de"> 269</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de">kCSRStorage</a> = 2,</div> |
| <div class="line"><a name="l00270"></a><span class="lineno"> 270</span> };</div> |
| <div class="line"><a name="l00271"></a><span class="lineno"> 271</span>  </div> |
| <div class="line"><a name="l00277"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXContext.html"> 277</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> {</div> |
| <div class="line"><a name="l00278"></a><span class="lineno"> 278</span>  <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#af6287fbdda27736f4406bf2ec3d69756">MXContext</a>();</div> |
| <div class="line"><a name="l00279"></a><span class="lineno"> 279</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#af6287fbdda27736f4406bf2ec3d69756">MXContext</a>(std::string dev_type_, <span class="keywordtype">int</span> dev_id_);</div> |
| <div class="line"><a name="l00280"></a><span class="lineno"> 280</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#af6287fbdda27736f4406bf2ec3d69756">MXContext</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* dev_type_, <span class="keywordtype">int</span> dev_id_);</div> |
| <div class="line"><a name="l00281"></a><span class="lineno"> 281</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a5ff62e51ebb2a5c898281b73a0c45c7a">CPU</a>();</div> |
| <div class="line"><a name="l00282"></a><span class="lineno"> 282</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a0930c97f53cdd8fb98cb1bec71c23e68">GPU</a>();</div> |
| <div class="line"><a name="l00283"></a><span class="lineno"> 283</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a5ff62e51ebb2a5c898281b73a0c45c7a">CPU</a>(<span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a40b89dc89df67507bc0d7f94b700f5a2">dev_id</a>);</div> |
| <div class="line"><a name="l00284"></a><span class="lineno"> 284</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a0930c97f53cdd8fb98cb1bec71c23e68">GPU</a>(<span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a40b89dc89df67507bc0d7f94b700f5a2">dev_id</a>);</div> |
| <div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  </div> |
| <div class="line"><a name="l00286"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXContext.html#a79b8b2861199ad3d8890b2086cdb8582"> 286</a></span>  std::string <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a79b8b2861199ad3d8890b2086cdb8582">dev_type</a>;</div> |
| <div class="line"><a name="l00287"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXContext.html#a40b89dc89df67507bc0d7f94b700f5a2"> 287</a></span>  <span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html#a40b89dc89df67507bc0d7f94b700f5a2">dev_id</a>;</div> |
| <div class="line"><a name="l00288"></a><span class="lineno"> 288</span> };</div> |
| <div class="line"><a name="l00289"></a><span class="lineno"> 289</span>  </div> |
| <div class="line"><a name="l00290"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40"> 290</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> {</div> |
| <div class="line"><a name="l00291"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670"> 291</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670">MX_FAIL</a> = 0,</div> |
| <div class="line"><a name="l00292"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306"> 292</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306">MX_SUCCESS</a> = 1,</div> |
| <div class="line"><a name="l00293"></a><span class="lineno"> 293</span> };</div> |
| <div class="line"><a name="l00294"></a><span class="lineno"> 294</span>  </div> |
| <div class="line"><a name="l00295"></a><span class="lineno"> 295</span> <span class="comment">// For sparse tensors, read/write the data from NDarray via pointers.</span></div> |
| <div class="line"><a name="l00296"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html"> 296</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1ext_1_1MXSparse.html">MXSparse</a> {</div> |
| <div class="line"><a name="l00297"></a><span class="lineno"> 297</span>  <span class="comment">// Pointer to data.</span></div> |
| <div class="line"><a name="l00298"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html#aa90461a780707c2a867589aa16a09a14"> 298</a></span>  <span class="keywordtype">void</span>* <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#aa90461a780707c2a867589aa16a09a14">data</a>{<span class="keyword">nullptr</span>};</div> |
| <div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  <span class="comment">// length of (non-zero) data.</span></div> |
| <div class="line"><a name="l00300"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html#a7abe4f71d2754fbf7ed00736c37ffe43"> 300</a></span>  int64_t <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#a7abe4f71d2754fbf7ed00736c37ffe43">data_len</a>;</div> |
| <div class="line"><a name="l00301"></a><span class="lineno"> 301</span>  </div> |
| <div class="line"><a name="l00302"></a><span class="lineno"> 302</span>  <span class="comment">// To store aux data for sparse.</span></div> |
| <div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  <span class="comment">// For CSR, indices stores the col index of non-zero elements.</span></div> |
| <div class="line"><a name="l00304"></a><span class="lineno"> 304</span>  <span class="comment">// For row sparse, indices store row index of rows which have non-zero elements.</span></div> |
| <div class="line"><a name="l00305"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html#a711d78a6b1533b30b0d5bd4af03bec21"> 305</a></span>  int64_t* <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#a711d78a6b1533b30b0d5bd4af03bec21">indices</a>;</div> |
| <div class="line"><a name="l00306"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html#a5df6670c7736ef03a71aedf9c044371f"> 306</a></span>  int64_t <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#a5df6670c7736ef03a71aedf9c044371f">indices_len</a>;</div> |
| <div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  </div> |
| <div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  <span class="comment">// For CSR, indptr gives the start and end index of data for each row.</span></div> |
| <div class="line"><a name="l00309"></a><span class="lineno"> 309</span>  <span class="comment">// For row sparse, indptr is not used.</span></div> |
| <div class="line"><a name="l00310"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html#a738c782852852154eb6fc61b8e9b34b0"> 310</a></span>  int64_t* <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#a738c782852852154eb6fc61b8e9b34b0">indptr</a> = <span class="keyword">nullptr</span>;</div> |
| <div class="line"><a name="l00311"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXSparse.html#a3a02a02338f173c6735ba5b8b1eb1b5c"> 311</a></span>  int64_t <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#a3a02a02338f173c6735ba5b8b1eb1b5c">indptr_len</a>;</div> |
| <div class="line"><a name="l00312"></a><span class="lineno"> 312</span>  </div> |
| <div class="line"><a name="l00313"></a><span class="lineno"> 313</span>  <span class="keywordtype">void</span> <a class="code" href="structmxnet_1_1ext_1_1MXSparse.html#abb2fa268862b67ad08c6b946130609a5">set</a>(<span class="keywordtype">void</span>* data_ptr,</div> |
| <div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  <span class="keyword">const</span> int64_t* dims,</div> |
| <div class="line"><a name="l00315"></a><span class="lineno"> 315</span>  <span class="keywordtype">int</span> ndims,</div> |
| <div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  <span class="keywordtype">void</span>* idx,</div> |
| <div class="line"><a name="l00317"></a><span class="lineno"> 317</span>  int64_t num_idx,</div> |
| <div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  <span class="keywordtype">void</span>* idx_ptr = <span class="keyword">nullptr</span>,</div> |
| <div class="line"><a name="l00319"></a><span class="lineno"> 319</span>  int64_t num_idx_ptr = 0);</div> |
| <div class="line"><a name="l00320"></a><span class="lineno"> 320</span> };</div> |
| <div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  </div> |
| <div class="line"><a name="l00325"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html"> 325</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a> {</div> |
| <div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#accf665442e338c1f09832dde3752fc02">MXTensor</a>();</div> |
| <div class="line"><a name="l00327"></a><span class="lineno"> 327</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#accf665442e338c1f09832dde3752fc02">MXTensor</a>(<span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>& oth);</div> |
| <div class="line"><a name="l00328"></a><span class="lineno"> 328</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#accf665442e338c1f09832dde3752fc02">MXTensor</a>(<span class="keywordtype">void</span>* <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a3d1d1a007585a61bc8e53ebaa2e61480">data_ptr</a>,</div> |
| <div class="line"><a name="l00329"></a><span class="lineno"> 329</span>  std::vector<int64_t> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#ae2d186fe466cd35bc4b77e038106b80d">shape</a>,</div> |
| <div class="line"><a name="l00330"></a><span class="lineno"> 330</span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#ad306b093c8348887640e51e040c0748f">dtype</a>,</div> |
| <div class="line"><a name="l00331"></a><span class="lineno"> 331</span>  <span class="keywordtype">size_t</span> vID,</div> |
| <div class="line"><a name="l00332"></a><span class="lineno"> 332</span>  <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> mx_ctx,</div> |
| <div class="line"><a name="l00333"></a><span class="lineno"> 333</span>  <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a38959dfb4fd816f407c76a7285639758">stype</a> = <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e">kDefaultStorage</a>);</div> |
| <div class="line"><a name="l00334"></a><span class="lineno"> 334</span>  </div> |
| <div class="line"><a name="l00336"></a><span class="lineno"> 336</span>  <span class="keywordtype">void</span> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a764cc32d43d66be2095065b9525e47c2">setTensor</a>(<span class="keywordtype">void</span>* dptr,</div> |
| <div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> type,</div> |
| <div class="line"><a name="l00338"></a><span class="lineno"> 338</span>  <span class="keyword">const</span> int64_t* dims,</div> |
| <div class="line"><a name="l00339"></a><span class="lineno"> 339</span>  <span class="keywordtype">int</span> ndims,</div> |
| <div class="line"><a name="l00340"></a><span class="lineno"> 340</span>  <span class="keywordtype">size_t</span> vID,</div> |
| <div class="line"><a name="l00341"></a><span class="lineno"> 341</span>  <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> mx_ctx,</div> |
| <div class="line"><a name="l00342"></a><span class="lineno"> 342</span>  <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> storage_type);</div> |
| <div class="line"><a name="l00343"></a><span class="lineno"> 343</span>  </div> |
| <div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  <span class="keywordtype">void</span> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#aead603e2824c32eeb027bee2d2b4a640">setDLTensor</a>();</div> |
| <div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  </div> |
| <div class="line"><a name="l00348"></a><span class="lineno"> 348</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> data_type></div> |
| <div class="line"><a name="l00349"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a4b16f5086857cf3dd983f9ef2b96c204"> 349</a></span>  <span class="keyword">inline</span> data_type* <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a4b16f5086857cf3dd983f9ef2b96c204">data</a>() {</div> |
| <div class="line"><a name="l00350"></a><span class="lineno"> 350</span>  <span class="keywordflow">return</span> <span class="keyword">reinterpret_cast<</span>data_type*<span class="keyword">></span>(<a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a3d1d1a007585a61bc8e53ebaa2e61480">data_ptr</a>);</div> |
| <div class="line"><a name="l00351"></a><span class="lineno"> 351</span>  }</div> |
| <div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  </div> |
| <div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  int64_t <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a67f147927ed642c1adec68471856a703">size</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00355"></a><span class="lineno"> 355</span>  </div> |
| <div class="line"><a name="l00357"></a><span class="lineno"> 357</span>  <span class="keywordtype">bool</span> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a8934bedd9d9643ee1e8c69bcae65c224">isSame</a>(<span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>& oth) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  </div> |
| <div class="line"><a name="l00359"></a><span class="lineno"> 359</span>  <span class="comment">// For dense, data_ptr points to 1D flattened tensor data</span></div> |
| <div class="line"><a name="l00360"></a><span class="lineno"> 360</span>  <span class="comment">// For sparse, data_ptr points to MXSparse</span></div> |
| <div class="line"><a name="l00361"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a3d1d1a007585a61bc8e53ebaa2e61480"> 361</a></span>  <span class="keywordtype">void</span>* <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a3d1d1a007585a61bc8e53ebaa2e61480">data_ptr</a>;</div> |
| <div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  </div> |
| <div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  <span class="comment">// shape is in [2,3,4] format to represent high-dim tensor</span></div> |
| <div class="line"><a name="l00364"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#ae2d186fe466cd35bc4b77e038106b80d"> 364</a></span>  std::vector<int64_t> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#ae2d186fe466cd35bc4b77e038106b80d">shape</a>;</div> |
| <div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  </div> |
| <div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  <span class="comment">// type can only be MXDType enum types</span></div> |
| <div class="line"><a name="l00367"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#ad306b093c8348887640e51e040c0748f"> 367</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#ad306b093c8348887640e51e040c0748f">dtype</a>;</div> |
| <div class="line"><a name="l00368"></a><span class="lineno"> 368</span>  </div> |
| <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  <span class="comment">// version number updated if the tensor has changed since the last use by custom op</span></div> |
| <div class="line"><a name="l00370"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a7d51476c0d999dc545aaa4e729d2f841"> 370</a></span>  <span class="keywordtype">size_t</span> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a7d51476c0d999dc545aaa4e729d2f841">verID</a>;</div> |
| <div class="line"><a name="l00371"></a><span class="lineno"> 371</span>  </div> |
| <div class="line"><a name="l00372"></a><span class="lineno"> 372</span>  <span class="comment">// context of MXTensor representing which device the tensor data is located</span></div> |
| <div class="line"><a name="l00373"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a634c4e6805c4928fcbbf859306ce1fee"> 373</a></span>  <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a634c4e6805c4928fcbbf859306ce1fee">ctx</a>;</div> |
| <div class="line"><a name="l00374"></a><span class="lineno"> 374</span>  </div> |
| <div class="line"><a name="l00375"></a><span class="lineno"> 375</span>  <span class="comment">// corresponding DLTensor repr of MXTensor</span></div> |
| <div class="line"><a name="l00376"></a><span class="lineno"> 376</span>  <span class="comment">// easy way to reuse functions taking DLTensor</span></div> |
| <div class="line"><a name="l00377"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#af7369ec802b04819c641334e521d7dae"> 377</a></span>  <a class="code" href="structDLTensor.html">DLTensor</a> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#af7369ec802b04819c641334e521d7dae">dltensor</a>;</div> |
| <div class="line"><a name="l00378"></a><span class="lineno"> 378</span>  </div> |
| <div class="line"><a name="l00379"></a><span class="lineno"> 379</span>  <span class="comment">// storage type</span></div> |
| <div class="line"><a name="l00380"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a38959dfb4fd816f407c76a7285639758"> 380</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html#a38959dfb4fd816f407c76a7285639758">stype</a>;</div> |
| <div class="line"><a name="l00381"></a><span class="lineno"> 381</span> };</div> |
| <div class="line"><a name="l00382"></a><span class="lineno"> 382</span>  </div> |
| <div class="line"><a name="l00384"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760"> 384</a></span> <span class="keyword">typedef</span> <span class="keywordtype">void</span>* (*xpu_malloc_t)(<span class="keywordtype">void</span>*, int);</div> |
| <div class="line"><a name="l00386"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e"> 386</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a>)(<span class="keywordtype">void</span>*, int, int, int, <span class="keywordtype">void</span>**, int64_t**, int64_t**);</div> |
| <div class="line"><a name="l00388"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917"> 388</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a>)(<span class="keyword">const</span> <span class="keywordtype">void</span>* _ndarray_alloc,</div> |
| <div class="line"><a name="l00389"></a><span class="lineno"> 389</span>  <span class="keyword">const</span> int64_t* shapes,</div> |
| <div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  <span class="keywordtype">int</span> num_shapes,</div> |
| <div class="line"><a name="l00391"></a><span class="lineno"> 391</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* dev_str,</div> |
| <div class="line"><a name="l00392"></a><span class="lineno"> 392</span>  <span class="keywordtype">int</span> dev_id,</div> |
| <div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  <span class="keywordtype">int</span> dtype,</div> |
| <div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* name,</div> |
| <div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  <span class="keywordtype">int</span> isArg,</div> |
| <div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  <span class="keywordtype">void</span>** data);</div> |
| <div class="line"><a name="l00398"></a><span class="lineno"> 398</span> <span class="preprocessor">#if defined(__NVCC__)</span></div> |
| <div class="line"><a name="l00399"></a><span class="lineno"> 399</span> <span class="keyword">typedef</span> cudaStream_t <a class="code" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mx_stream_t</a>;</div> |
| <div class="line"><a name="l00400"></a><span class="lineno"> 400</span> <span class="keyword">typedef</span> curandStatePhilox4_32_10_t <a class="code" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mx_gpu_rand_t</a>;</div> |
| <div class="line"><a name="l00401"></a><span class="lineno"> 401</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00402"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752"> 402</a></span> <span class="keyword">typedef</span> <span class="keywordtype">void</span>* <a class="code" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mx_stream_t</a>;</div> |
| <div class="line"><a name="l00403"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13"> 403</a></span> <span class="keyword">typedef</span> <span class="keywordtype">void</span>* <a class="code" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mx_gpu_rand_t</a>;</div> |
| <div class="line"><a name="l00404"></a><span class="lineno"> 404</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00405"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8"> 405</a></span> <span class="keyword">typedef</span> std::mt19937 <a class="code" href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8">mx_cpu_rand_t</a>;</div> |
| <div class="line"><a name="l00406"></a><span class="lineno"> 406</span>  </div> |
| <div class="line"><a name="l00408"></a><span class="lineno"> 408</span> <span class="comment">/* Each thread should generate random number unique sequence out of different states */</span></div> |
| <div class="line"><a name="l00409"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a3bab156d26b9eface536d0c5d03d9b52"> 409</a></span> <span class="preprocessor">#define MX_NUM_CPU_RANDOM_STATES 1024</span></div> |
| <div class="line"><a name="l00410"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aa0ac2ed4c5f2ee26d5c748aee108e3cd"> 410</a></span> <span class="preprocessor">#define MX_NUM_GPU_RANDOM_STATES 32768</span></div> |
| <div class="line"><a name="l00411"></a><span class="lineno"> 411</span>  </div> |
| <div class="line"><a name="l00412"></a><span class="lineno"> 412</span> <span class="comment">/* \brief Class to help allocate new args/aux params in graph passes */</span></div> |
| <div class="line"><a name="l00413"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1PassResource.html"> 413</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a> {</div> |
| <div class="line"><a name="l00414"></a><span class="lineno"> 414</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  <a class="code" href="classmxnet_1_1ext_1_1PassResource.html#af7ee64a64fbfa3d1f068ebd4976911d5">PassResource</a>(std::unordered_map<std::string, MXTensor>* new_args,</div> |
| <div class="line"><a name="l00416"></a><span class="lineno"> 416</span>  std::unordered_map<std::string, MXTensor>* new_aux,</div> |
| <div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc,</div> |
| <div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc);</div> |
| <div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  </div> |
| <div class="line"><a name="l00420"></a><span class="lineno"> 420</span>  <span class="comment">// allocate new arg param, adds to args map, returns newly allocated tensor</span></div> |
| <div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>* <a class="code" href="classmxnet_1_1ext_1_1PassResource.html#a9db8c01ef60ebd075caf275ab499a05b">alloc_arg</a>(<span class="keyword">const</span> std::string& name,</div> |
| <div class="line"><a name="l00422"></a><span class="lineno"> 422</span>  <span class="keyword">const</span> std::vector<int64_t>& shapes,</div> |
| <div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a>& ctx,</div> |
| <div class="line"><a name="l00424"></a><span class="lineno"> 424</span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  </div> |
| <div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  <span class="comment">// allocate new aux param, adds to aux map, returns newly allocated tensor</span></div> |
| <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>* <a class="code" href="classmxnet_1_1ext_1_1PassResource.html#aeff45b0c181dc59ce73a4802d2c77a0e">alloc_aux</a>(<span class="keyword">const</span> std::string& name,</div> |
| <div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="keyword">const</span> std::vector<int64_t>& shapes,</div> |
| <div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a>& ctx,</div> |
| <div class="line"><a name="l00430"></a><span class="lineno"> 430</span>  <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00431"></a><span class="lineno"> 431</span>  </div> |
| <div class="line"><a name="l00432"></a><span class="lineno"> 432</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  std::unordered_map<std::string, MXTensor>* new_args_;</div> |
| <div class="line"><a name="l00434"></a><span class="lineno"> 434</span>  std::unordered_map<std::string, MXTensor>* new_aux_;</div> |
| <div class="line"><a name="l00435"></a><span class="lineno"> 435</span>  <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc_;</div> |
| <div class="line"><a name="l00436"></a><span class="lineno"> 436</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc_;</div> |
| <div class="line"><a name="l00437"></a><span class="lineno"> 437</span> };</div> |
| <div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  </div> |
| <div class="line"><a name="l00442"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1OpResource.html"> 442</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a> {</div> |
| <div class="line"><a name="l00443"></a><span class="lineno"> 443</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00444"></a><span class="lineno"> 444</span>  <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#a4e9eb54d2481fcb97231fec20290bd5d">OpResource</a>(<a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc_fp,</div> |
| <div class="line"><a name="l00445"></a><span class="lineno"> 445</span>  <span class="keywordtype">void</span>* cpu_alloc_fp,</div> |
| <div class="line"><a name="l00446"></a><span class="lineno"> 446</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc_fp,</div> |
| <div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  <span class="keywordtype">void</span>* gpu_alloc_fp,</div> |
| <div class="line"><a name="l00448"></a><span class="lineno"> 448</span>  <span class="keywordtype">void</span>* stream,</div> |
| <div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc_fp,</div> |
| <div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  <span class="keywordtype">void</span>* sparse_alloc_fp,</div> |
| <div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  <span class="keywordtype">void</span>* rng_cpu_states,</div> |
| <div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  <span class="keywordtype">void</span>* rng_gpu_states);</div> |
| <div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  </div> |
| <div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keywordtype">void</span>* <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#aca8a01d001adef1481fa5506b76d117b">alloc_cpu</a>(<span class="keywordtype">int</span> size) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  </div> |
| <div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  <span class="keywordtype">void</span>* <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#a9f2d4d97f7d2eb3c492c8f8fcfb98100">alloc_gpu</a>(<span class="keywordtype">int</span> size) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  </div> |
| <div class="line"><a name="l00461"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1OpResource.html#a712e78a6c50beca023bc20064688bd1f"> 461</a></span>  <span class="keyword">inline</span> <a class="code" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mx_stream_t</a> <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#a712e78a6c50beca023bc20064688bd1f">get_cuda_stream</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00462"></a><span class="lineno"> 462</span>  <span class="keywordflow">return</span> <span class="keyword">static_cast<</span><a class="code" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mx_stream_t</a><span class="keyword">></span>(cuda_stream);</div> |
| <div class="line"><a name="l00463"></a><span class="lineno"> 463</span>  }</div> |
| <div class="line"><a name="l00464"></a><span class="lineno"> 464</span>  </div> |
| <div class="line"><a name="l00466"></a><span class="lineno"> 466</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#a037945db6e2934e1f4acefe2c8337740">alloc_sparse</a>(<a class="code" href="structmxnet_1_1ext_1_1MXSparse.html">MXSparse</a>* sparse, <span class="keywordtype">int</span> index, <span class="keywordtype">int</span> indices_len, <span class="keywordtype">int</span> indptr_len = 0) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  </div> |
| <div class="line"><a name="l00469"></a><span class="lineno"> 469</span>  <span class="comment">/* Access each state by states[id], but this id should be <= MX_NUM_CPU_RANDOM_STATES */</span></div> |
| <div class="line"><a name="l00470"></a><span class="lineno"> 470</span>  <a class="code" href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8">mx_cpu_rand_t</a>* <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#acd1f86bb22908ed80946488bb99c5020">get_cpu_rand_states</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00471"></a><span class="lineno"> 471</span>  </div> |
| <div class="line"><a name="l00473"></a><span class="lineno"> 473</span>  <span class="comment">/* Access each state by states[id], but this id should be <= MX_NUM_GPU_RANDOM_STATES */</span></div> |
| <div class="line"><a name="l00474"></a><span class="lineno"> 474</span>  <span class="comment">/* Note that if you are using cpu build, it will return a nullptr */</span></div> |
| <div class="line"><a name="l00475"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1OpResource.html#a3c75f8ca600930d970e3ea20cd624884"> 475</a></span>  <span class="keyword">inline</span> <a class="code" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mx_gpu_rand_t</a>* <a class="code" href="classmxnet_1_1ext_1_1OpResource.html#a3c75f8ca600930d970e3ea20cd624884">get_gpu_rand_states</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00476"></a><span class="lineno"> 476</span>  <span class="keywordflow">return</span> <span class="keyword">static_cast<</span><a class="code" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mx_gpu_rand_t</a>*<span class="keyword">></span>(rand_gpu_states);</div> |
| <div class="line"><a name="l00477"></a><span class="lineno"> 477</span>  }</div> |
| <div class="line"><a name="l00478"></a><span class="lineno"> 478</span>  </div> |
| <div class="line"><a name="l00479"></a><span class="lineno"> 479</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00481"></a><span class="lineno"> 481</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, gpu_malloc;</div> |
| <div class="line"><a name="l00483"></a><span class="lineno"> 483</span>  <span class="keywordtype">void</span> *cpu_alloc, *gpu_alloc;</div> |
| <div class="line"><a name="l00485"></a><span class="lineno"> 485</span>  <span class="keywordtype">void</span>* cuda_stream;</div> |
| <div class="line"><a name="l00487"></a><span class="lineno"> 487</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc;</div> |
| <div class="line"><a name="l00489"></a><span class="lineno"> 489</span>  <span class="keywordtype">void</span>* sparse_alloc;</div> |
| <div class="line"><a name="l00491"></a><span class="lineno"> 491</span>  <span class="keywordtype">void</span> *rand_cpu_states, *rand_gpu_states;</div> |
| <div class="line"><a name="l00492"></a><span class="lineno"> 492</span> };</div> |
| <div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  </div> |
| <div class="line"><a name="l00495"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ab5f044e03093ffae2626b35fa17f610d"> 495</a></span> <span class="preprocessor">#define MX_STR_SUBGRAPH_SYM_JSON "subgraph_sym_json"</span></div> |
| <div class="line"><a name="l00496"></a><span class="lineno"> 496</span>  </div> |
| <div class="line"><a name="l00497"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a73b8e3f85f6f0aba1c0fd0917773b24f"> 497</a></span> <span class="preprocessor">#define MX_STR_DTYPE "__ext_dtype__"</span></div> |
| <div class="line"><a name="l00498"></a><span class="lineno"> 498</span>  </div> |
| <div class="line"><a name="l00499"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ab031cafe8f2e358de3c1cadc57a346f3"> 499</a></span> <span class="preprocessor">#define MX_STR_SHAPE "__ext_shape__"</span></div> |
| <div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  </div> |
| <div class="line"><a name="l00501"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a633f26064a8c0d69aa833175a66daa2e"> 501</a></span> <span class="preprocessor">#define MX_STR_EXTRA_INPUTS "__ext_extra_inputs__"</span></div> |
| <div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  </div> |
| <div class="line"><a name="l00503"></a><span class="lineno"> 503</span> <span class="comment">/* \brief get shape value from list of shapes string</span></div> |
| <div class="line"><a name="l00504"></a><span class="lineno"> 504</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00505"></a><span class="lineno"> 505</span> <span class="comment"> * Examples:</span></div> |
| <div class="line"><a name="l00506"></a><span class="lineno"> 506</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00507"></a><span class="lineno"> 507</span> <span class="comment"> * getShapeAt("[[1]]", 0) returns "[1]"</span></div> |
| <div class="line"><a name="l00508"></a><span class="lineno"> 508</span> <span class="comment"> * getShapeAt("[[1],[2,3]]", 1) returns "[2,3]"</span></div> |
| <div class="line"><a name="l00509"></a><span class="lineno"> 509</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00510"></a><span class="lineno"> 510</span> std::string <a class="code" href="namespacemxnet_1_1ext.html#ac19085fc1abe2e392d1305d9ac3bc015">getShapeAt</a>(<span class="keyword">const</span> std::string& shape, <span class="keywordtype">unsigned</span> index);</div> |
| <div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  </div> |
| <div class="line"><a name="l00512"></a><span class="lineno"> 512</span> <span class="comment">/* \brief get dtype value from list of dtypes string</span></div> |
| <div class="line"><a name="l00513"></a><span class="lineno"> 513</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00514"></a><span class="lineno"> 514</span> <span class="comment"> * Examples:</span></div> |
| <div class="line"><a name="l00515"></a><span class="lineno"> 515</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00516"></a><span class="lineno"> 516</span> <span class="comment"> * getDtypeAt("[1]", 0) returns "1"</span></div> |
| <div class="line"><a name="l00517"></a><span class="lineno"> 517</span> <span class="comment"> * getDtypeAt("[1,2]", 1) returns "2"</span></div> |
| <div class="line"><a name="l00518"></a><span class="lineno"> 518</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00519"></a><span class="lineno"> 519</span> std::string <a class="code" href="namespacemxnet_1_1ext.html#a8b389f7b678d499fd8d5ba79f089c95e">getDtypeAt</a>(<span class="keyword">const</span> std::string& dtype, <span class="keywordtype">unsigned</span> index);</div> |
| <div class="line"><a name="l00520"></a><span class="lineno"> 520</span>  </div> |
| <div class="line"><a name="l00525"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9"> 525</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">JsonType</a> { <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9">ERR</a>, <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9">STR</a>, <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1">NUM</a>, <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901">LIST</a>, <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241">MAP</a> };</div> |
| <div class="line"><a name="l00526"></a><span class="lineno"> 526</span>  </div> |
| <div class="line"><a name="l00528"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html"> 528</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> {</div> |
| <div class="line"><a name="l00529"></a><span class="lineno"> 529</span>  <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a886ffa05148d30a14f97c07382ae72d0">JsonVal</a>(); <span class="comment">// default constructor</span></div> |
| <div class="line"><a name="l00530"></a><span class="lineno"> 530</span>  <span class="comment">// construct a JSON object by type</span></div> |
| <div class="line"><a name="l00531"></a><span class="lineno"> 531</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a886ffa05148d30a14f97c07382ae72d0">JsonVal</a>(<a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">JsonType</a> t);</div> |
| <div class="line"><a name="l00532"></a><span class="lineno"> 532</span>  <span class="comment">// construct a string JSON object</span></div> |
| <div class="line"><a name="l00533"></a><span class="lineno"> 533</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a886ffa05148d30a14f97c07382ae72d0">JsonVal</a>(std::string s);</div> |
| <div class="line"><a name="l00534"></a><span class="lineno"> 534</span>  <span class="comment">// construct a number JSON object</span></div> |
| <div class="line"><a name="l00535"></a><span class="lineno"> 535</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a886ffa05148d30a14f97c07382ae72d0">JsonVal</a>(<span class="keywordtype">int</span> n);</div> |
| <div class="line"><a name="l00536"></a><span class="lineno"> 536</span>  <span class="comment">// complex constructor</span></div> |
| <div class="line"><a name="l00537"></a><span class="lineno"> 537</span>  <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a886ffa05148d30a14f97c07382ae72d0">JsonVal</a>(<a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">JsonType</a> t, <span class="keywordtype">int</span> n, std::string s);</div> |
| <div class="line"><a name="l00538"></a><span class="lineno"> 538</span>  <span class="keywordtype">bool</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a9d1096da2919bfc1861d70c3f2079339">operator<</a>(<span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>& o) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00539"></a><span class="lineno"> 539</span>  </div> |
| <div class="line"><a name="l00540"></a><span class="lineno"> 540</span>  <span class="comment">// convert JSON object back to JSON-compatible string</span></div> |
| <div class="line"><a name="l00541"></a><span class="lineno"> 541</span>  std::string <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#afadf4694fc70c782d4b3b63cd2d1e1e8">dump</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00542"></a><span class="lineno"> 542</span>  </div> |
| <div class="line"><a name="l00543"></a><span class="lineno"> 543</span>  <span class="comment">// convert JSON-compatible string to JSON object</span></div> |
| <div class="line"><a name="l00544"></a><span class="lineno"> 544</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#ae3918edf21e047e5d381db65e2497008">parse</a>(<span class="keyword">const</span> std::string& json);</div> |
| <div class="line"><a name="l00545"></a><span class="lineno"> 545</span>  </div> |
| <div class="line"><a name="l00546"></a><span class="lineno"> 546</span>  <span class="comment">// parse a string JSON object</span></div> |
| <div class="line"><a name="l00547"></a><span class="lineno"> 547</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#ac5005c18501c8da80e170f88e1eaa454">parse_string</a>(<span class="keyword">const</span> std::string& json, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>* idx);</div> |
| <div class="line"><a name="l00548"></a><span class="lineno"> 548</span>  </div> |
| <div class="line"><a name="l00549"></a><span class="lineno"> 549</span>  <span class="comment">// parse a number JSON object</span></div> |
| <div class="line"><a name="l00550"></a><span class="lineno"> 550</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a08ba0f08fe57589b83315d1f0cefde37">parse_num</a>(<span class="keyword">const</span> std::string& json, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>* idx);</div> |
| <div class="line"><a name="l00551"></a><span class="lineno"> 551</span>  </div> |
| <div class="line"><a name="l00552"></a><span class="lineno"> 552</span>  <span class="comment">// parse a list of JSON objects</span></div> |
| <div class="line"><a name="l00553"></a><span class="lineno"> 553</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a65a765a25fdbaa5ac44cd42c8a848c44">parse_list</a>(<span class="keyword">const</span> std::string& json, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>* idx);</div> |
| <div class="line"><a name="l00554"></a><span class="lineno"> 554</span>  </div> |
| <div class="line"><a name="l00555"></a><span class="lineno"> 555</span>  <span class="comment">// parse a map of JSON objects</span></div> |
| <div class="line"><a name="l00556"></a><span class="lineno"> 556</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#aaf2854a4075aa8cd6ef138112f1676b6">parse_map</a>(<span class="keyword">const</span> std::string& json, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>* idx);</div> |
| <div class="line"><a name="l00557"></a><span class="lineno"> 557</span>  </div> |
| <div class="line"><a name="l00558"></a><span class="lineno"> 558</span>  <span class="comment">// generic parse function</span></div> |
| <div class="line"><a name="l00559"></a><span class="lineno"> 559</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#ae3918edf21e047e5d381db65e2497008">parse</a>(<span class="keyword">const</span> std::string& json, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>* idx);</div> |
| <div class="line"><a name="l00560"></a><span class="lineno"> 560</span>  </div> |
| <div class="line"><a name="l00561"></a><span class="lineno"> 561</span>  <span class="comment">// debug function to convert data structure to a debugstring</span></div> |
| <div class="line"><a name="l00562"></a><span class="lineno"> 562</span>  std::string <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a7345d89ba6e4c628641cf36c4123145d">toString</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00563"></a><span class="lineno"> 563</span>  </div> |
| <div class="line"><a name="l00564"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#a9f826ae984fc3f102c1bb6761f3f23ee"> 564</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">JsonType</a> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a9f826ae984fc3f102c1bb6761f3f23ee">type</a>;</div> |
| <div class="line"><a name="l00565"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#a1d9e77f9eb4ab27cdcffb85495a760ae"> 565</a></span>  <span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#a1d9e77f9eb4ab27cdcffb85495a760ae">num</a>;</div> |
| <div class="line"><a name="l00566"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#af5a3b01d85dcfa4765838c1b38f414e9"> 566</a></span>  std::string <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#af5a3b01d85dcfa4765838c1b38f414e9">str</a>;</div> |
| <div class="line"><a name="l00567"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#af5d2babd1cb84d40ad07585cac9c5970"> 567</a></span>  std::vector<JsonVal> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#af5d2babd1cb84d40ad07585cac9c5970">list</a>;</div> |
| <div class="line"><a name="l00568"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#aae3712b54e6f015df3b75b2827da1adc"> 568</a></span>  std::map<JsonVal, JsonVal> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#aae3712b54e6f015df3b75b2827da1adc">map</a>;</div> |
| <div class="line"><a name="l00569"></a><span class="lineno"> 569</span> };</div> |
| <div class="line"><a name="l00570"></a><span class="lineno"> 570</span>  </div> |
| <div class="line"><a name="l00574"></a><span class="lineno"> 574</span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>;</div> |
| <div class="line"><a name="l00575"></a><span class="lineno"> 575</span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>;</div> |
| <div class="line"><a name="l00576"></a><span class="lineno"> 576</span>  </div> |
| <div class="line"><a name="l00577"></a><span class="lineno"> 577</span> <span class="comment">// Representation of an input/output to a node</span></div> |
| <div class="line"><a name="l00578"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1NodeEntry.html"> 578</a></span> <span class="keyword">struct </span><a class="code" href="structmxnet_1_1ext_1_1NodeEntry.html">NodeEntry</a> {</div> |
| <div class="line"><a name="l00579"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1NodeEntry.html#a9ccf74d370d36c4ed911e94f84a431be"> 579</a></span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* <a class="code" href="structmxnet_1_1ext_1_1NodeEntry.html#a9ccf74d370d36c4ed911e94f84a431be">node</a>; <span class="comment">// other node thats producing/consuming inputs/outputs</span></div> |
| <div class="line"><a name="l00580"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1NodeEntry.html#ac3d191e6775d212446933fc94ca21624"> 580</a></span>  <span class="keywordtype">int</span> <a class="code" href="structmxnet_1_1ext_1_1NodeEntry.html#ac3d191e6775d212446933fc94ca21624">entry</a>; <span class="comment">// entry index from other node (ie. output index from producing node)</span></div> |
| <div class="line"><a name="l00581"></a><span class="lineno"> 581</span> };</div> |
| <div class="line"><a name="l00582"></a><span class="lineno"> 582</span>  </div> |
| <div class="line"><a name="l00583"></a><span class="lineno"> 583</span> <span class="comment">// Representation of a node in the graph</span></div> |
| <div class="line"><a name="l00584"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html"> 584</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a> {</div> |
| <div class="line"><a name="l00585"></a><span class="lineno"> 585</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00586"></a><span class="lineno"> 586</span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html#a2c1ec7bd5c9132e2636e70a93a53e452">Node</a>();</div> |
| <div class="line"><a name="l00587"></a><span class="lineno"> 587</span>  </div> |
| <div class="line"><a name="l00588"></a><span class="lineno"> 588</span>  <span class="comment">// internally set passResource to enable tensor allocation for graph passes</span></div> |
| <div class="line"><a name="l00589"></a><span class="lineno"> 589</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Node.html#aac12981007cdb017f652d1242cf9ab05">_setPassResource</a>(<a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res_);</div> |
| <div class="line"><a name="l00590"></a><span class="lineno"> 590</span>  </div> |
| <div class="line"><a name="l00591"></a><span class="lineno"> 591</span>  <span class="comment">/* \brief allocate an arg tensor for this node */</span></div> |
| <div class="line"><a name="l00592"></a><span class="lineno"> 592</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Node.html#a391e1c2ccc10a9edc19417fa3fad3169">alloc_arg</a>(<span class="keyword">const</span> std::vector<int64_t>& shapes, <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a>& ctx, <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype);</div> |
| <div class="line"><a name="l00593"></a><span class="lineno"> 593</span>  </div> |
| <div class="line"><a name="l00594"></a><span class="lineno"> 594</span>  <span class="comment">/* \brief allocate an aux tensor for this node */</span></div> |
| <div class="line"><a name="l00595"></a><span class="lineno"> 595</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Node.html#a17744ecbf554a996fe84dc5553564633">alloc_aux</a>(<span class="keyword">const</span> std::vector<int64_t>& shapes, <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a>& ctx, <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype);</div> |
| <div class="line"><a name="l00596"></a><span class="lineno"> 596</span>  </div> |
| <div class="line"><a name="l00597"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a20c7b312f050f3a1cc606e924482a4c7"> 597</a></span>  std::string <a class="code" href="classmxnet_1_1ext_1_1Node.html#a20c7b312f050f3a1cc606e924482a4c7">op</a>; <span class="comment">// operator name (ie. Convolution)</span></div> |
| <div class="line"><a name="l00598"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#abc2b3e8b645050faf1a4534de694e939"> 598</a></span>  std::string <a class="code" href="classmxnet_1_1ext_1_1Node.html#abc2b3e8b645050faf1a4534de694e939">name</a>; <span class="comment">// unique node name (ie. conv_0 or conv_1)</span></div> |
| <div class="line"><a name="l00599"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a3489d59197ad9612dbdc23f963aa8d02"> 599</a></span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>* <a class="code" href="classmxnet_1_1ext_1_1Node.html#a3489d59197ad9612dbdc23f963aa8d02">tensor</a>; <span class="comment">// tensor data for input nodes</span></div> |
| <div class="line"><a name="l00600"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a8364c3b8ae3800c775a23a8e90984c82"> 600</a></span>  std::vector<NodeEntry> <a class="code" href="classmxnet_1_1ext_1_1Node.html#a8364c3b8ae3800c775a23a8e90984c82">inputs</a>; <span class="comment">// set of inputs to the node</span></div> |
| <div class="line"><a name="l00601"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#aee95220e1a30197ca5fa726742f57183"> 601</a></span>  std::vector<NodeEntry> <a class="code" href="classmxnet_1_1ext_1_1Node.html#aee95220e1a30197ca5fa726742f57183">outputs</a>; <span class="comment">// set of outputs from the node</span></div> |
| <div class="line"><a name="l00602"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#afbe960c8b040871dfb9f8c8bdb529947"> 602</a></span>  std::vector<Graph*> <a class="code" href="classmxnet_1_1ext_1_1Node.html#afbe960c8b040871dfb9f8c8bdb529947">subgraphs</a>; <span class="comment">// set of subgraphs within this node</span></div> |
| <div class="line"><a name="l00603"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a3e1851c411ee0112de6ee3f1a90d2151"> 603</a></span>  std::unordered_map<std::string, std::string> <a class="code" href="classmxnet_1_1ext_1_1Node.html#a3e1851c411ee0112de6ee3f1a90d2151">attrs</a>; <span class="comment">// node attributes</span></div> |
| <div class="line"><a name="l00604"></a><span class="lineno"> 604</span>  </div> |
| <div class="line"><a name="l00605"></a><span class="lineno"> 605</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00606"></a><span class="lineno"> 606</span>  <a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res;</div> |
| <div class="line"><a name="l00607"></a><span class="lineno"> 607</span> };</div> |
| <div class="line"><a name="l00608"></a><span class="lineno"> 608</span>  </div> |
| <div class="line"><a name="l00609"></a><span class="lineno"> 609</span> <span class="comment">// Representation of the graph</span></div> |
| <div class="line"><a name="l00610"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html"> 610</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a> {</div> |
| <div class="line"><a name="l00611"></a><span class="lineno"> 611</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  <a class="code" href="classmxnet_1_1ext_1_1Graph.html#ab49ade50779a6d387b0073ad04d6dafa">Graph</a>();</div> |
| <div class="line"><a name="l00613"></a><span class="lineno"> 613</span>  </div> |
| <div class="line"><a name="l00614"></a><span class="lineno"> 614</span>  <span class="comment">/* \brief deleted nodes when deleting the graph */</span></div> |
| <div class="line"><a name="l00615"></a><span class="lineno"> 615</span>  <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a4e00a65e67fee5acdc7122513642af0b">~Graph</a>();</div> |
| <div class="line"><a name="l00616"></a><span class="lineno"> 616</span>  </div> |
| <div class="line"><a name="l00617"></a><span class="lineno"> 617</span>  <span class="comment">/* \brief create a graph object from an unparsed string */</span></div> |
| <div class="line"><a name="l00618"></a><span class="lineno"> 618</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>* <a class="code" href="classmxnet_1_1ext_1_1Graph.html#aa5bd268a502dce80ffce037252fa0443">fromString</a>(<span class="keyword">const</span> std::string& json);</div> |
| <div class="line"><a name="l00619"></a><span class="lineno"> 619</span>  </div> |
| <div class="line"><a name="l00620"></a><span class="lineno"> 620</span>  <span class="comment">/* \brief create a graph object from a parsed JSON object */</span></div> |
| <div class="line"><a name="l00621"></a><span class="lineno"> 621</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>* <a class="code" href="classmxnet_1_1ext_1_1Graph.html#aad7a2b18c34573c8ded3f4d8069df3c9">fromJson</a>(<a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> val);</div> |
| <div class="line"><a name="l00622"></a><span class="lineno"> 622</span>  </div> |
| <div class="line"><a name="l00623"></a><span class="lineno"> 623</span>  <span class="comment">/* \brief convert graph object back to JSON object */</span></div> |
| <div class="line"><a name="l00624"></a><span class="lineno"> 624</span>  <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#ae45b21c0986b8986bcd9ecd5e2b3580e">toJson</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00625"></a><span class="lineno"> 625</span>  </div> |
| <div class="line"><a name="l00626"></a><span class="lineno"> 626</span>  <span class="comment">/* \brief convert graph object to JSON string */</span></div> |
| <div class="line"><a name="l00627"></a><span class="lineno"> 627</span>  std::string <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a1907581693bae41486523f199fecacf5">toString</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00628"></a><span class="lineno"> 628</span>  </div> |
| <div class="line"><a name="l00629"></a><span class="lineno"> 629</span>  <span class="comment">/* \brief visits a node "n" */</span></div> |
| <div class="line"><a name="l00630"></a><span class="lineno"> 630</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#af827c73b729d3e8cb8672b10107312fd">_dfs_util</a>(<a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* n,</div> |
| <div class="line"><a name="l00631"></a><span class="lineno"> 631</span>  std::unordered_set<Node*>* to_visit,</div> |
| <div class="line"><a name="l00632"></a><span class="lineno"> 632</span>  std::function<<span class="keywordtype">void</span>(<a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>*)> handler) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00633"></a><span class="lineno"> 633</span>  </div> |
| <div class="line"><a name="l00634"></a><span class="lineno"> 634</span>  <span class="comment">/* \brief post-order DFS graph traversal */</span></div> |
| <div class="line"><a name="l00635"></a><span class="lineno"> 635</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a995c0b2b6d1d520852a664a31c44c984">DFS</a>(std::function<<span class="keywordtype">void</span>(<a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>*)> handler) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00636"></a><span class="lineno"> 636</span>  </div> |
| <div class="line"><a name="l00637"></a><span class="lineno"> 637</span>  <span class="comment">/* \brief sort graph nodes in topological order */</span></div> |
| <div class="line"><a name="l00638"></a><span class="lineno"> 638</span>  std::vector<Node*> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#ad2fb294e9396e263c79199054f9b5520">topological_sort</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00639"></a><span class="lineno"> 639</span>  </div> |
| <div class="line"><a name="l00640"></a><span class="lineno"> 640</span>  <span class="comment">/* \brief print out graph details */</span></div> |
| <div class="line"><a name="l00641"></a><span class="lineno"> 641</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a0e92d22b68b63878afb266cdd2091c39">print</a>(<span class="keywordtype">int</span> indent = 0) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00642"></a><span class="lineno"> 642</span>  </div> |
| <div class="line"><a name="l00643"></a><span class="lineno"> 643</span>  <span class="comment">/* \brief add a new node to this graph */</span></div> |
| <div class="line"><a name="l00644"></a><span class="lineno"> 644</span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* <a class="code" href="classmxnet_1_1ext_1_1Graph.html#af6ab4915225627081794e2d02b60b223">addNode</a>(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> std::string& op);</div> |
| <div class="line"><a name="l00645"></a><span class="lineno"> 645</span>  </div> |
| <div class="line"><a name="l00646"></a><span class="lineno"> 646</span>  <span class="comment">/* \brief get node at index in graph */</span></div> |
| <div class="line"><a name="l00647"></a><span class="lineno"> 647</span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* <a class="code" href="classmxnet_1_1ext_1_1Graph.html#ae5aabe1376e697faf14c30561c8533cd">getNode</a>(<span class="keywordtype">size_t</span> idx);</div> |
| <div class="line"><a name="l00648"></a><span class="lineno"> 648</span>  </div> |
| <div class="line"><a name="l00649"></a><span class="lineno"> 649</span>  <span class="comment">/* \brief get const node at index in const graph */</span></div> |
| <div class="line"><a name="l00650"></a><span class="lineno"> 650</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* <a class="code" href="classmxnet_1_1ext_1_1Graph.html#ae5aabe1376e697faf14c30561c8533cd">getNode</a>(<span class="keywordtype">size_t</span> idx) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00651"></a><span class="lineno"> 651</span>  </div> |
| <div class="line"><a name="l00652"></a><span class="lineno"> 652</span>  <span class="comment">/* \brief get attribute on graph */</span></div> |
| <div class="line"><a name="l00653"></a><span class="lineno"> 653</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>& <a class="code" href="classmxnet_1_1ext_1_1Graph.html#aa5ee4f0b935a06fe8d947aee14f10597">getAttr</a>(<span class="keyword">const</span> std::string& key) <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  </div> |
| <div class="line"><a name="l00655"></a><span class="lineno"> 655</span>  <span class="comment">/* \brief get number of nodes in the graph */</span></div> |
| <div class="line"><a name="l00656"></a><span class="lineno"> 656</span>  <span class="keywordtype">size_t</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a6aeb4f0d430f6599a887af467ca05d61">size</a>() <span class="keyword">const</span>;</div> |
| <div class="line"><a name="l00657"></a><span class="lineno"> 657</span>  </div> |
| <div class="line"><a name="l00658"></a><span class="lineno"> 658</span>  <span class="comment">// internally set passResource to enable tensor allocation for graph passes</span></div> |
| <div class="line"><a name="l00659"></a><span class="lineno"> 659</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a4158fd61578679630e5eba3a6d889ca3">_setPassResource</a>(<a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res_);</div> |
| <div class="line"><a name="l00660"></a><span class="lineno"> 660</span>  </div> |
| <div class="line"><a name="l00661"></a><span class="lineno"> 661</span>  <span class="comment">// internally set arg/aux params when available</span></div> |
| <div class="line"><a name="l00662"></a><span class="lineno"> 662</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a9233c8bb6aa95b7f8d03de17960723b6">_setParams</a>(std::unordered_map<std::string, mxnet::ext::MXTensor>* args,</div> |
| <div class="line"><a name="l00663"></a><span class="lineno"> 663</span>  std::unordered_map<std::string, mxnet::ext::MXTensor>* aux);</div> |
| <div class="line"><a name="l00664"></a><span class="lineno"> 664</span>  </div> |
| <div class="line"><a name="l00665"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html#a44c1ad9a070013bdf4390b0894d9294d"> 665</a></span>  std::vector<Node*> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a44c1ad9a070013bdf4390b0894d9294d">inputs</a>;</div> |
| <div class="line"><a name="l00666"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html#afda2391d5b2ae16121ab1c1bf5eb3ae3"> 666</a></span>  std::vector<NodeEntry> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#afda2391d5b2ae16121ab1c1bf5eb3ae3">outputs</a>;</div> |
| <div class="line"><a name="l00667"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html#a7997e82fd30ac0817c0555fc0969e87f"> 667</a></span>  std::map<std::string, JsonVal> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a7997e82fd30ac0817c0555fc0969e87f">attrs</a>;</div> |
| <div class="line"><a name="l00668"></a><span class="lineno"> 668</span>  </div> |
| <div class="line"><a name="l00669"></a><span class="lineno"> 669</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00670"></a><span class="lineno"> 670</span>  std::vector<Node*> nodes;</div> |
| <div class="line"><a name="l00671"></a><span class="lineno"> 671</span>  <a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res;</div> |
| <div class="line"><a name="l00672"></a><span class="lineno"> 672</span> };</div> |
| <div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  </div> |
| <div class="line"><a name="l00674"></a><span class="lineno"> 674</span> <span class="comment">/* \brief An abstract class for library authors creating custom</span></div> |
| <div class="line"><a name="l00675"></a><span class="lineno"> 675</span> <span class="comment"> * partitioners. Optional, can just implement supportedOps instead</span></div> |
| <div class="line"><a name="l00676"></a><span class="lineno"> 676</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00677"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOpSelector.html"> 677</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html">CustomOpSelector</a> {</div> |
| <div class="line"><a name="l00678"></a><span class="lineno"> 678</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00679"></a><span class="lineno"> 679</span>  <span class="comment">/* \brief Select a node to include in subgraph, return true to include node</span></div> |
| <div class="line"><a name="l00680"></a><span class="lineno"> 680</span> <span class="comment"> * nodeID - index of node in graph</span></div> |
| <div class="line"><a name="l00681"></a><span class="lineno"> 681</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00682"></a><span class="lineno"> 682</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a1ebe634fe8f38266d1e85624d1707e55">Select</a>(<span class="keywordtype">int</span> nodeID) = 0;</div> |
| <div class="line"><a name="l00683"></a><span class="lineno"> 683</span>  <span class="comment">/* \brief Select an input node from current node to include in subgraph</span></div> |
| <div class="line"><a name="l00684"></a><span class="lineno"> 684</span> <span class="comment"> * return true to include node</span></div> |
| <div class="line"><a name="l00685"></a><span class="lineno"> 685</span> <span class="comment"> * nodeID - index of node in graph</span></div> |
| <div class="line"><a name="l00686"></a><span class="lineno"> 686</span> <span class="comment"> * input_nodeID - index of input node in graph</span></div> |
| <div class="line"><a name="l00687"></a><span class="lineno"> 687</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00688"></a><span class="lineno"> 688</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html#aa4b8875b76531510de9bf672034fd96d">SelectInput</a>(<span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> input_nodeID) = 0;</div> |
| <div class="line"><a name="l00689"></a><span class="lineno"> 689</span>  <span class="comment">/* \brief Select an output node from current node to include in subgraph</span></div> |
| <div class="line"><a name="l00690"></a><span class="lineno"> 690</span> <span class="comment"> * return true to include node</span></div> |
| <div class="line"><a name="l00691"></a><span class="lineno"> 691</span> <span class="comment"> * nodeID - index of node in graph</span></div> |
| <div class="line"><a name="l00692"></a><span class="lineno"> 692</span> <span class="comment"> * output_nodeID - index of output node in graph</span></div> |
| <div class="line"><a name="l00693"></a><span class="lineno"> 693</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00694"></a><span class="lineno"> 694</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a42727340ba85aa1aaf8493e0322978e3">SelectOutput</a>(<span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> output_nodeID) = 0;</div> |
| <div class="line"><a name="l00695"></a><span class="lineno"> 695</span>  <span class="comment">/* \brief Review nodes to include in subgraph</span></div> |
| <div class="line"><a name="l00696"></a><span class="lineno"> 696</span> <span class="comment"> * return set of candidate nodes to keep in subgraph</span></div> |
| <div class="line"><a name="l00697"></a><span class="lineno"> 697</span> <span class="comment"> * candidates - indices of nodes to include in subgraph</span></div> |
| <div class="line"><a name="l00698"></a><span class="lineno"> 698</span> <span class="comment"> * keep - indices of nodes to keep in subgraph</span></div> |
| <div class="line"><a name="l00699"></a><span class="lineno"> 699</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00700"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a2e39f0a634eee550cb79b225d1beac02"> 700</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a2e39f0a634eee550cb79b225d1beac02">Filter</a>(<span class="keyword">const</span> std::vector<int>& candidates, std::vector<int>* keep) {</div> |
| <div class="line"><a name="l00701"></a><span class="lineno"> 701</span>  keep->insert(keep->end(), candidates.begin(), candidates.end());</div> |
| <div class="line"><a name="l00702"></a><span class="lineno"> 702</span>  }</div> |
| <div class="line"><a name="l00703"></a><span class="lineno"> 703</span>  <span class="comment">/* \brief Reset any selector state, called after growing subgraph, before filter</span></div> |
| <div class="line"><a name="l00704"></a><span class="lineno"> 704</span> <span class="comment"> * Called after finished calling SelectInput/SelectOutput and growing subgraph</span></div> |
| <div class="line"><a name="l00705"></a><span class="lineno"> 705</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00706"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a76533a016df5e8c3e39a722f2bf48f13"> 706</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a76533a016df5e8c3e39a722f2bf48f13">Reset</a>() {}</div> |
| <div class="line"><a name="l00707"></a><span class="lineno"> 707</span> };</div> |
| <div class="line"><a name="l00708"></a><span class="lineno"> 708</span>  </div> |
| <div class="line"><a name="l00714"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html"> 714</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a> {</div> |
| <div class="line"><a name="l00715"></a><span class="lineno"> 715</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00716"></a><span class="lineno"> 716</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#adacc6e8d18cc2428ace58c96111e6f39">CustomStatefulOp</a>();</div> |
| <div class="line"><a name="l00717"></a><span class="lineno"> 717</span>  <span class="keyword">virtual</span> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#acbf1b61e0f1cb13ba9df92cf2774ba7b">~CustomStatefulOp</a>();</div> |
| <div class="line"><a name="l00718"></a><span class="lineno"> 718</span>  </div> |
| <div class="line"><a name="l00719"></a><span class="lineno"> 719</span>  <span class="keyword">template</span> <<span class="keyword">class </span>A, <span class="keyword">typename</span>... Ts></div> |
| <div class="line"><a name="l00720"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a42d0e539870e3c27db8eb88606765f5d"> 720</a></span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>* <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a42d0e539870e3c27db8eb88606765f5d">create</a>(Ts... args) {</div> |
| <div class="line"><a name="l00721"></a><span class="lineno"> 721</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>* op = <span class="keyword">new</span> A(args...);</div> |
| <div class="line"><a name="l00722"></a><span class="lineno"> 722</span>  op->created = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00723"></a><span class="lineno"> 723</span>  <span class="keywordflow">return</span> op;</div> |
| <div class="line"><a name="l00724"></a><span class="lineno"> 724</span>  }</div> |
| <div class="line"><a name="l00725"></a><span class="lineno"> 725</span>  </div> |
| <div class="line"><a name="l00726"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a46fb6fe57019f1ef4d4a616c7ef7297f"> 726</a></span>  <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a46fb6fe57019f1ef4d4a616c7ef7297f">wasCreated</a>() {</div> |
| <div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  <span class="keywordflow">return</span> created;</div> |
| <div class="line"><a name="l00728"></a><span class="lineno"> 728</span>  }</div> |
| <div class="line"><a name="l00729"></a><span class="lineno"> 729</span>  </div> |
| <div class="line"><a name="l00730"></a><span class="lineno"> 730</span>  <span class="keyword">virtual</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a96fa3699ce4d0324480cc0185d49cf30">Forward</a>(std::vector<MXTensor>* inputs,</div> |
| <div class="line"><a name="l00731"></a><span class="lineno"> 731</span>  std::vector<MXTensor>* outputs,</div> |
| <div class="line"><a name="l00732"></a><span class="lineno"> 732</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a>& op_res) = 0;</div> |
| <div class="line"><a name="l00733"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#ad8758d40952666f93d644f704b686da6"> 733</a></span>  <span class="keyword">virtual</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#ad8758d40952666f93d644f704b686da6">Backward</a>(std::vector<MXTensor>* inputs,</div> |
| <div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  std::vector<MXTensor>* outputs,</div> |
| <div class="line"><a name="l00735"></a><span class="lineno"> 735</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a>& op_res) {</div> |
| <div class="line"><a name="l00736"></a><span class="lineno"> 736</span>  <a class="code" href="lib__api_8h.html#a481a7e516c02ab43e5b0e0016cb7997f">MX_ERROR_MSG</a> << <span class="stringliteral">"Error! Operator does not support backward"</span> << std::endl;</div> |
| <div class="line"><a name="l00737"></a><span class="lineno"> 737</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670">MX_FAIL</a>;</div> |
| <div class="line"><a name="l00738"></a><span class="lineno"> 738</span>  }</div> |
| <div class="line"><a name="l00739"></a><span class="lineno"> 739</span>  </div> |
| <div class="line"><a name="l00740"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#aca7db0bccdf4840d5cd70ff73f7624da"> 740</a></span>  <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#aca7db0bccdf4840d5cd70ff73f7624da">ignore_warn</a>;</div> |
| <div class="line"><a name="l00741"></a><span class="lineno"> 741</span>  </div> |
| <div class="line"><a name="l00742"></a><span class="lineno"> 742</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00743"></a><span class="lineno"> 743</span>  <span class="keywordtype">bool</span> created;</div> |
| <div class="line"><a name="l00744"></a><span class="lineno"> 744</span> };</div> |
| <div class="line"><a name="l00745"></a><span class="lineno"> 745</span>  </div> |
| <div class="line"><a name="l00747"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603"> 747</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a>)(<span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00748"></a><span class="lineno"> 748</span>  std::vector<MXTensor>* inputs,</div> |
| <div class="line"><a name="l00749"></a><span class="lineno"> 749</span>  std::vector<MXTensor>* outputs,</div> |
| <div class="line"><a name="l00750"></a><span class="lineno"> 750</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a>& res);</div> |
| <div class="line"><a name="l00751"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc"> 751</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a>)(</div> |
| <div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00753"></a><span class="lineno"> 753</span>  <span class="keywordtype">int</span>* num_inputs,</div> |
| <div class="line"><a name="l00754"></a><span class="lineno"> 754</span>  <span class="keywordtype">int</span>* num_outputs);</div> |
| <div class="line"><a name="l00755"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869"> 755</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a>)(<span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00756"></a><span class="lineno"> 756</span>  std::vector<int>* in_types,</div> |
| <div class="line"><a name="l00757"></a><span class="lineno"> 757</span>  std::vector<int>* out_types);</div> |
| <div class="line"><a name="l00758"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a"> 758</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a>)(</div> |
| <div class="line"><a name="l00759"></a><span class="lineno"> 759</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00760"></a><span class="lineno"> 760</span>  std::vector<int>* in_storage_types,</div> |
| <div class="line"><a name="l00761"></a><span class="lineno"> 761</span>  std::vector<int>* out_storage_types);</div> |
| <div class="line"><a name="l00762"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9"> 762</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a>)(</div> |
| <div class="line"><a name="l00763"></a><span class="lineno"> 763</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00764"></a><span class="lineno"> 764</span>  std::vector<std::vector<unsigned int> >* in_shapes,</div> |
| <div class="line"><a name="l00765"></a><span class="lineno"> 765</span>  std::vector<std::vector<unsigned int> >* out_shapes);</div> |
| <div class="line"><a name="l00766"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484"> 766</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a>)(</div> |
| <div class="line"><a name="l00767"></a><span class="lineno"> 767</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00768"></a><span class="lineno"> 768</span>  std::vector<int>* input_indices);</div> |
| <div class="line"><a name="l00769"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923"> 769</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">createOpState_t</a>)(</div> |
| <div class="line"><a name="l00770"></a><span class="lineno"> 770</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& attributes,</div> |
| <div class="line"><a name="l00771"></a><span class="lineno"> 771</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a>& ctx,</div> |
| <div class="line"><a name="l00772"></a><span class="lineno"> 772</span>  <span class="keyword">const</span> std::vector<std::vector<unsigned int> >& in_shapes,</div> |
| <div class="line"><a name="l00773"></a><span class="lineno"> 773</span>  <span class="keyword">const</span> std::vector<int> in_types,</div> |
| <div class="line"><a name="l00774"></a><span class="lineno"> 774</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>**);</div> |
| <div class="line"><a name="l00775"></a><span class="lineno"> 775</span>  </div> |
| <div class="line"><a name="l00779"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html"> 779</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a> {</div> |
| <div class="line"><a name="l00780"></a><span class="lineno"> 780</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00781"></a><span class="lineno"> 781</span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a49edc0a4c130a3f811d67410e07f1970">CustomOp</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* op_name);</div> |
| <div class="line"><a name="l00782"></a><span class="lineno"> 782</span>  </div> |
| <div class="line"><a name="l00783"></a><span class="lineno"> 783</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#ac1d121b14423cb590b611bbbe866f3c7">setForward</a>(<a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a> fcomp, <span class="keyword">const</span> <span class="keywordtype">char</span>* ctx);</div> |
| <div class="line"><a name="l00784"></a><span class="lineno"> 784</span>  </div> |
| <div class="line"><a name="l00785"></a><span class="lineno"> 785</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a48b8906bb9fb74daf9db7c9725dfe549">setBackward</a>(<a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a> fgrad, <span class="keyword">const</span> <span class="keywordtype">char</span>* ctx);</div> |
| <div class="line"><a name="l00786"></a><span class="lineno"> 786</span>  </div> |
| <div class="line"><a name="l00787"></a><span class="lineno"> 787</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#ad0ff7bf816398b0eee6d8464705df7de">setParseAttrs</a>(<a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a> func);</div> |
| <div class="line"><a name="l00788"></a><span class="lineno"> 788</span>  </div> |
| <div class="line"><a name="l00789"></a><span class="lineno"> 789</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#ad79465455155119203c88a45547f7b95">setInferType</a>(<a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a> func);</div> |
| <div class="line"><a name="l00790"></a><span class="lineno"> 790</span>  </div> |
| <div class="line"><a name="l00791"></a><span class="lineno"> 791</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#acc7b8856d1a69b0b95ac0e2556800d4e">setInferSType</a>(<a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a> func);</div> |
| <div class="line"><a name="l00792"></a><span class="lineno"> 792</span>  </div> |
| <div class="line"><a name="l00793"></a><span class="lineno"> 793</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#af70aca6c1872bc2750b5ccfb72655298">setInferShape</a>(<a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a> func);</div> |
| <div class="line"><a name="l00794"></a><span class="lineno"> 794</span>  </div> |
| <div class="line"><a name="l00795"></a><span class="lineno"> 795</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#aa4f66a1473ea311e3adcd0f7c259677c">setMutateInputs</a>(<a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a> func);</div> |
| <div class="line"><a name="l00796"></a><span class="lineno"> 796</span>  </div> |
| <div class="line"><a name="l00797"></a><span class="lineno"> 797</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a61b8b8d52e15d830875b643851a6af9c">setCreateOpState</a>(<a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">createOpState_t</a> func, <span class="keyword">const</span> <span class="keywordtype">char</span>* ctx);</div> |
| <div class="line"><a name="l00798"></a><span class="lineno"> 798</span>  </div> |
| <div class="line"><a name="l00799"></a><span class="lineno"> 799</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a2065f7e6313a774f9b1881f3b038d13d">setIsSubgraphOp</a>();</div> |
| <div class="line"><a name="l00800"></a><span class="lineno"> 800</span>  </div> |
| <div class="line"><a name="l00801"></a><span class="lineno"> 801</span>  <span class="keywordtype">void</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#ae69f99666f2e874f57c41fa3e823491b">mapToVector</a>();</div> |
| <div class="line"><a name="l00802"></a><span class="lineno"> 802</span>  </div> |
| <div class="line"><a name="l00804"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#aa1de9b3397a13c3633350307593837f3"> 804</a></span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#aa1de9b3397a13c3633350307593837f3">name</a>;</div> |
| <div class="line"><a name="l00805"></a><span class="lineno"> 805</span>  </div> |
| <div class="line"><a name="l00807"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a2978383f7d86dafd36cff9eeba16c8be"> 807</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a2978383f7d86dafd36cff9eeba16c8be">parse_attrs</a>;</div> |
| <div class="line"><a name="l00808"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a07eb2f161653eed3b4332355cb050137"> 808</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a07eb2f161653eed3b4332355cb050137">infer_type</a>;</div> |
| <div class="line"><a name="l00809"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a96338af40de6ff68da43ad5649dab18d"> 809</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a96338af40de6ff68da43ad5649dab18d">infer_storage_type</a>;</div> |
| <div class="line"><a name="l00810"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#adb7dd13f57912db3f560c2baf223d1d7"> 810</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#adb7dd13f57912db3f560c2baf223d1d7">infer_shape</a>;</div> |
| <div class="line"><a name="l00811"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#ae25800af6998af47fe82a31474314bba"> 811</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#ae25800af6998af47fe82a31474314bba">mutate_inputs</a>;</div> |
| <div class="line"><a name="l00812"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a2a205d1c793a296d3a312a575e9eab76"> 812</a></span>  <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a2a205d1c793a296d3a312a575e9eab76">isSGop</a>;</div> |
| <div class="line"><a name="l00813"></a><span class="lineno"> 813</span>  </div> |
| <div class="line"><a name="l00815"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#acb5a3d13564e147807e0006c6f75eada"> 815</a></span>  std::vector<const char*> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#acb5a3d13564e147807e0006c6f75eada">forward_ctx_cstr</a>, <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a6ce1906895cb2b88cae60e3e1137c8bd">backward_ctx_cstr</a>, <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a7a1b71e81d17df4f68e664a5d931866d">create_op_ctx_cstr</a>;</div> |
| <div class="line"><a name="l00816"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a34340edf7aaefd434e9be4d9e7ce03dd"> 816</a></span>  std::vector<fcomp_t> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a34340edf7aaefd434e9be4d9e7ce03dd">forward_fp</a>, <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a55c4fbc10f64e0a6f02bae3a020b4e38">backward_fp</a>;</div> |
| <div class="line"><a name="l00817"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#ad2d93199d25ceace68343f8c516bf63e"> 817</a></span>  std::vector<createOpState_t> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#ad2d93199d25ceace68343f8c516bf63e">create_op_fp</a>;</div> |
| <div class="line"><a name="l00818"></a><span class="lineno"> 818</span>  </div> |
| <div class="line"><a name="l00819"></a><span class="lineno"> 819</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00820"></a><span class="lineno"> 820</span>  <span class="keywordtype">void</span> raiseDuplicateContextError();</div> |
| <div class="line"><a name="l00821"></a><span class="lineno"> 821</span>  </div> |
| <div class="line"><a name="l00823"></a><span class="lineno"> 823</span>  std::unordered_map<const char*, fcomp_t> forward_ctx_map, backward_ctx_map;</div> |
| <div class="line"><a name="l00824"></a><span class="lineno"> 824</span>  std::unordered_map<const char*, createOpState_t> create_op_ctx_map;</div> |
| <div class="line"><a name="l00825"></a><span class="lineno"> 825</span> };</div> |
| <div class="line"><a name="l00826"></a><span class="lineno"> 826</span>  </div> |
| <div class="line"><a name="l00828"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d"> 828</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a>)(<a class="code" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a>* graph,</div> |
| <div class="line"><a name="l00829"></a><span class="lineno"> 829</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& options);</div> |
| <div class="line"><a name="l00830"></a><span class="lineno"> 830</span>  </div> |
| <div class="line"><a name="l00834"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPass.html"> 834</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1CustomPass.html">CustomPass</a> {</div> |
| <div class="line"><a name="l00835"></a><span class="lineno"> 835</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00836"></a><span class="lineno"> 836</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html#a38420e01eb27a91bc7b11da785b7a652">CustomPass</a>();</div> |
| <div class="line"><a name="l00837"></a><span class="lineno"> 837</span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html#a38420e01eb27a91bc7b11da785b7a652">CustomPass</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* pass_name);</div> |
| <div class="line"><a name="l00838"></a><span class="lineno"> 838</span>  </div> |
| <div class="line"><a name="l00839"></a><span class="lineno"> 839</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html">CustomPass</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html#a7fbdda42ff6b530c2625d435a8691651">setBody</a>(<a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> fn);</div> |
| <div class="line"><a name="l00840"></a><span class="lineno"> 840</span>  </div> |
| <div class="line"><a name="l00842"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPass.html#aec569fc21f800cb803d47508cd495fcb"> 842</a></span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html#aec569fc21f800cb803d47508cd495fcb">name</a>;</div> |
| <div class="line"><a name="l00844"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPass.html#a6dd29f51d329906280090109a634bb6a"> 844</a></span>  <a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html#a6dd29f51d329906280090109a634bb6a">pass</a>;</div> |
| <div class="line"><a name="l00845"></a><span class="lineno"> 845</span> };</div> |
| <div class="line"><a name="l00846"></a><span class="lineno"> 846</span>  </div> |
| <div class="line"><a name="l00848"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab"> 848</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a>)(</div> |
| <div class="line"><a name="l00849"></a><span class="lineno"> 849</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a>* graph,</div> |
| <div class="line"><a name="l00850"></a><span class="lineno"> 850</span>  std::vector<int>* ids,</div> |
| <div class="line"><a name="l00851"></a><span class="lineno"> 851</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& options);</div> |
| <div class="line"><a name="l00852"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659"> 852</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a>)(</div> |
| <div class="line"><a name="l00853"></a><span class="lineno"> 853</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a>* graph,</div> |
| <div class="line"><a name="l00854"></a><span class="lineno"> 854</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html">CustomOpSelector</a>** sel_inst,</div> |
| <div class="line"><a name="l00855"></a><span class="lineno"> 855</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& options);</div> |
| <div class="line"><a name="l00856"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae"> 856</a></span> <span class="keyword">typedef</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> (*<a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a>)(</div> |
| <div class="line"><a name="l00857"></a><span class="lineno"> 857</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a>* subgraph,</div> |
| <div class="line"><a name="l00858"></a><span class="lineno"> 858</span>  <span class="keywordtype">int</span> subgraph_id,</div> |
| <div class="line"><a name="l00859"></a><span class="lineno"> 859</span>  <span class="keywordtype">bool</span>* accept,</div> |
| <div class="line"><a name="l00860"></a><span class="lineno"> 860</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& options,</div> |
| <div class="line"><a name="l00861"></a><span class="lineno"> 861</span>  std::unordered_map<std::string, std::string>* attrs);</div> |
| <div class="line"><a name="l00862"></a><span class="lineno"> 862</span>  </div> |
| <div class="line"><a name="l00866"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html"> 866</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a> {</div> |
| <div class="line"><a name="l00867"></a><span class="lineno"> 867</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00868"></a><span class="lineno"> 868</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a99f30444690e140720d54d69cf42ce0e">CustomPartitioner</a>();</div> |
| <div class="line"><a name="l00869"></a><span class="lineno"> 869</span>  </div> |
| <div class="line"><a name="l00870"></a><span class="lineno"> 870</span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a99f30444690e140720d54d69cf42ce0e">CustomPartitioner</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* backend_name);</div> |
| <div class="line"><a name="l00871"></a><span class="lineno"> 871</span>  </div> |
| <div class="line"><a name="l00872"></a><span class="lineno"> 872</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a228dcaf825c6fcb89d8310fec774f4cd">addStrategy</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* prop_name, <span class="keyword">const</span> <span class="keywordtype">char</span>* sg_name);</div> |
| <div class="line"><a name="l00873"></a><span class="lineno"> 873</span>  </div> |
| <div class="line"><a name="l00874"></a><span class="lineno"> 874</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a35eadaeb5efe27f440026294f62bfc42">setSupportedOps</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* prop_name, <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> fn);</div> |
| <div class="line"><a name="l00875"></a><span class="lineno"> 875</span>  </div> |
| <div class="line"><a name="l00876"></a><span class="lineno"> 876</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a4ac609b75f121c4b70b3ba21405deb85">setCreateSelector</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* prop_name, <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> fn);</div> |
| <div class="line"><a name="l00877"></a><span class="lineno"> 877</span>  </div> |
| <div class="line"><a name="l00878"></a><span class="lineno"> 878</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#aa345494550891b4c7481b4123eef9586">setReviewSubgraph</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* prop_name, <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> fn);</div> |
| <div class="line"><a name="l00879"></a><span class="lineno"> 879</span>  </div> |
| <div class="line"><a name="l00880"></a><span class="lineno"> 880</span>  <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a588e1f76a13f07ea1a96db28c0ddb15a">getSupportedOps</a>(<span class="keywordtype">int</span> stg_id);</div> |
| <div class="line"><a name="l00881"></a><span class="lineno"> 881</span>  </div> |
| <div class="line"><a name="l00882"></a><span class="lineno"> 882</span>  <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a5a0891b0fffc7891d263b4850d96560b">getCreateSelector</a>(<span class="keywordtype">int</span> stg_id);</div> |
| <div class="line"><a name="l00883"></a><span class="lineno"> 883</span>  </div> |
| <div class="line"><a name="l00884"></a><span class="lineno"> 884</span>  <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#ab3865614538d9fa49295cd24cb6284c3">getReviewSubgraph</a>(<span class="keywordtype">int</span> stg_id);</div> |
| <div class="line"><a name="l00885"></a><span class="lineno"> 885</span>  </div> |
| <div class="line"><a name="l00887"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a1b620d0a37e0104eb60db211930ed5a5"> 887</a></span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a1b620d0a37e0104eb60db211930ed5a5">name</a>;</div> |
| <div class="line"><a name="l00888"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#ab25d24f86446a2f6d3aa4da85f869472"> 888</a></span>  std::map<std::string, supportedOps_t> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#ab25d24f86446a2f6d3aa4da85f869472">supported_map</a>;</div> |
| <div class="line"><a name="l00889"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#af96c64b969175dccd80943ef69207250"> 889</a></span>  std::map<std::string, createSelector_t> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#af96c64b969175dccd80943ef69207250">selector_map</a>;</div> |
| <div class="line"><a name="l00890"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a76e111c49dd7b6e3cfe550739f319611"> 890</a></span>  std::map<std::string, reviewSubgraph_t> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a76e111c49dd7b6e3cfe550739f319611">review_map</a>;</div> |
| <div class="line"><a name="l00892"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#abaadec312533d0a24f1faeba6db7411b"> 892</a></span>  std::vector<const char*> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#abaadec312533d0a24f1faeba6db7411b">strategies</a>;</div> |
| <div class="line"><a name="l00894"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#aeaad9d379d75938aa12b908c1a7a9af6"> 894</a></span>  std::vector<const char*> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html#aeaad9d379d75938aa12b908c1a7a9af6">op_names</a>;</div> |
| <div class="line"><a name="l00895"></a><span class="lineno"> 895</span> };</div> |
| <div class="line"><a name="l00896"></a><span class="lineno"> 896</span>  </div> |
| <div class="line"><a name="l00901"></a><span class="lineno"> 901</span> <span class="keyword">template</span> <<span class="keyword">class</span> T></div> |
| <div class="line"><a name="l00902"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html"> 902</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Registry.html">Registry</a> {</div> |
| <div class="line"><a name="l00903"></a><span class="lineno"> 903</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00908"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#a34d70208a11534185f4288284430acb8"> 908</a></span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Registry.html">Registry</a>* <a class="code" href="classmxnet_1_1ext_1_1Registry.html#a34d70208a11534185f4288284430acb8">get</a>() <a class="code" href="lib__api_8h.html#acb30ec605ba828f085c28799b89383ad">PRIVATE_SYMBOL</a> {</div> |
| <div class="line"><a name="l00909"></a><span class="lineno"> 909</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Registry.html">Registry</a> inst;</div> |
| <div class="line"><a name="l00910"></a><span class="lineno"> 910</span>  <span class="keywordflow">return</span> &inst;</div> |
| <div class="line"><a name="l00911"></a><span class="lineno"> 911</span>  }</div> |
| <div class="line"><a name="l00916"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#a32f3e2d387f02dc4edc8c0c51095af28"> 916</a></span>  T& <a class="code" href="classmxnet_1_1ext_1_1Registry.html#a32f3e2d387f02dc4edc8c0c51095af28">add</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* name) {</div> |
| <div class="line"><a name="l00917"></a><span class="lineno"> 917</span>  T* entry = <span class="keyword">new</span> T(name);</div> |
| <div class="line"><a name="l00918"></a><span class="lineno"> 918</span>  entries.push_back(entry);</div> |
| <div class="line"><a name="l00919"></a><span class="lineno"> 919</span>  <span class="keywordflow">return</span> *entry;</div> |
| <div class="line"><a name="l00920"></a><span class="lineno"> 920</span>  }</div> |
| <div class="line"><a name="l00921"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#a1a7613063268d9ff754bd3f53cdb35d6"> 921</a></span>  <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1ext_1_1Registry.html#a1a7613063268d9ff754bd3f53cdb35d6">size</a>() {</div> |
| <div class="line"><a name="l00922"></a><span class="lineno"> 922</span>  <span class="keywordflow">return</span> entries.size();</div> |
| <div class="line"><a name="l00923"></a><span class="lineno"> 923</span>  }</div> |
| <div class="line"><a name="l00924"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#aac245796efea7dbb18a86ff2be034453"> 924</a></span>  T& <a class="code" href="classmxnet_1_1ext_1_1Registry.html#aac245796efea7dbb18a86ff2be034453">get</a>(<span class="keywordtype">int</span> idx) {</div> |
| <div class="line"><a name="l00925"></a><span class="lineno"> 925</span>  <span class="keywordflow">return</span> *(entries.at(idx));</div> |
| <div class="line"><a name="l00926"></a><span class="lineno"> 926</span>  }</div> |
| <div class="line"><a name="l00927"></a><span class="lineno"> 927</span>  </div> |
| <div class="line"><a name="l00928"></a><span class="lineno"> 928</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00930"></a><span class="lineno"> 930</span>  <a class="code" href="classmxnet_1_1ext_1_1Registry.html">Registry</a>() {}</div> |
| <div class="line"><a name="l00932"></a><span class="lineno"> 932</span>  ~Registry() {}</div> |
| <div class="line"><a name="l00934"></a><span class="lineno"> 934</span>  std::vector<T*> entries;</div> |
| <div class="line"><a name="l00935"></a><span class="lineno"> 935</span> };</div> |
| <div class="line"><a name="l00936"></a><span class="lineno"> 936</span>  </div> |
| <div class="line"><a name="l00942"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac745a1b1f09e730abf98f51d2e513103"> 942</a></span> <span class="preprocessor">#define MX_STR_CONCAT_(__a, __b) __a##__b</span></div> |
| <div class="line"><a name="l00943"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad900a5506445a60d8e5ebd5079b0324f"> 943</a></span> <span class="preprocessor">#define MX_STR_CONCAT(__a, __b) MX_STR_CONCAT_(__a, __b)</span></div> |
| <div class="line"><a name="l00944"></a><span class="lineno"> 944</span>  </div> |
| <div class="line"><a name="l00946"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a8a5ee382299049479eaadc1f87d9dffb"> 946</a></span> <span class="preprocessor">#define MX_STRINGIFY(x) #x</span></div> |
| <div class="line"><a name="l00947"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac281c11d1b738cf6f4d44d1d6d2a41f5"> 947</a></span> <span class="preprocessor">#define MX_TOSTRING(x) MX_STRINGIFY(x)</span></div> |
| <div class="line"><a name="l00948"></a><span class="lineno"> 948</span>  </div> |
| <div class="line"><a name="l00950"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a412e972cba005076b3475796cd1ffbf6"> 950</a></span> <span class="preprocessor">#define MX_REGISTER_NAME_(Name) MXNet##_CustomOp##_##Name</span></div> |
| <div class="line"><a name="l00951"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aff6fd8259cfbba7e57a637c862dac3ca"> 951</a></span> <span class="preprocessor">#define MX_REGISTER_DEF_(Name) mxnet::ext::CustomOp MX_REGISTER_NAME_(Name)</span></div> |
| <div class="line"><a name="l00952"></a><span class="lineno"> 952</span>  </div> |
| <div class="line"><a name="l00953"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a94176fcbf0773036ac1e43e4b4dde51f"> 953</a></span> <span class="preprocessor">#define MX_REGISTER_PROP_NAME_(Name) MXNet##_CustomSubProp##_##Name</span></div> |
| <div class="line"><a name="l00954"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2e2f91e3c9f64ceffdfdb55f36354e91"> 954</a></span> <span class="preprocessor">#define MX_REGISTER_PROP_DEF_(Name) mxnet::ext::CustomPartitioner MX_REGISTER_PROP_NAME_(Name)</span></div> |
| <div class="line"><a name="l00955"></a><span class="lineno"> 955</span>  </div> |
| <div class="line"><a name="l00956"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a35d87ac322fe13c4c98a97d4c44c95e2"> 956</a></span> <span class="preprocessor">#define MX_REGISTER_PASS_NAME_(Name) MXNet##_CustomPass##_##Name</span></div> |
| <div class="line"><a name="l00957"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a4a6ec114cb6f848ec8170c9df3475d59"> 957</a></span> <span class="preprocessor">#define MX_REGISTER_PASS_DEF_(Name) mxnet::ext::CustomPass MX_REGISTER_PASS_NAME_(Name)</span></div> |
| <div class="line"><a name="l00958"></a><span class="lineno"> 958</span>  </div> |
| <div class="line"><a name="l00960"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abb165f6d1b3c2f0d4edfb5667a8b3638"> 960</a></span> <span class="preprocessor">#define REGISTER_OP(Name) \</span></div> |
| <div class="line"><a name="l00961"></a><span class="lineno"> 961</span> <span class="preprocessor"> MX_STR_CONCAT(MX_REGISTER_DEF_(Name), __COUNTER__) = \</span></div> |
| <div class="line"><a name="l00962"></a><span class="lineno"> 962</span> <span class="preprocessor"> mxnet::ext::Registry<mxnet::ext::CustomOp>::get()->add(MX_TOSTRING(Name))</span></div> |
| <div class="line"><a name="l00963"></a><span class="lineno"> 963</span>  </div> |
| <div class="line"><a name="l00964"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a93f481366195b39d8e4ea3bbec2a1b04"> 964</a></span> <span class="preprocessor">#define REGISTER_PARTITIONER(Name) \</span></div> |
| <div class="line"><a name="l00965"></a><span class="lineno"> 965</span> <span class="preprocessor"> MX_STR_CONCAT(MX_REGISTER_PROP_DEF_(Name), __COUNTER__) = \</span></div> |
| <div class="line"><a name="l00966"></a><span class="lineno"> 966</span> <span class="preprocessor"> mxnet::ext::Registry<mxnet::ext::CustomPartitioner>::get()->add(MX_TOSTRING(Name))</span></div> |
| <div class="line"><a name="l00967"></a><span class="lineno"> 967</span>  </div> |
| <div class="line"><a name="l00968"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ab9d14ed170244c036e1ff8a3531e89c6"> 968</a></span> <span class="preprocessor">#define REGISTER_PASS(Name) \</span></div> |
| <div class="line"><a name="l00969"></a><span class="lineno"> 969</span> <span class="preprocessor"> MX_STR_CONCAT(MX_REGISTER_PASS_DEF_(Name), __COUNTER__) = \</span></div> |
| <div class="line"><a name="l00970"></a><span class="lineno"> 970</span> <span class="preprocessor"> mxnet::ext::Registry<mxnet::ext::CustomPass>::get()->add(MX_TOSTRING(Name))</span></div> |
| <div class="line"><a name="l00971"></a><span class="lineno"> 971</span>  </div> |
| <div class="line"><a name="l00972"></a><span class="lineno"> 972</span> <span class="comment">/* -------------- BELOW ARE CTYPE FUNCTIONS PROTOTYPES --------------- */</span></div> |
| <div class="line"><a name="l00973"></a><span class="lineno"> 973</span>  </div> |
| <div class="line"><a name="l00979"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac39728fb75d241b62d4ed9261b7ddca3"> 979</a></span> <span class="preprocessor">#define MXLIB_OPREGSIZE_STR "_opRegSize"</span></div> |
| <div class="line"><a name="l00980"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6fb7af175432173dec6120674a1f07a4"> 980</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a6fb7af175432173dec6120674a1f07a4">opRegSize_t</a>)(void);</div> |
| <div class="line"><a name="l00981"></a><span class="lineno"> 981</span>  </div> |
| <div class="line"><a name="l00982"></a><span class="lineno"><a class="line" href="lib__api_8h.html#addceae596034e042d887954af8499f06"> 982</a></span> <span class="preprocessor">#define MXLIB_OPREGGET_STR "_opRegGet"</span></div> |
| <div class="line"><a name="l00983"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a10b027e352f8fd55d9639398807614bd"> 983</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a10b027e352f8fd55d9639398807614bd">opRegGet_t</a>)(<span class="keywordtype">int</span> idx,</div> |
| <div class="line"><a name="l00984"></a><span class="lineno"> 984</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** name,</div> |
| <div class="line"><a name="l00985"></a><span class="lineno"> 985</span>  <span class="keywordtype">int</span>* isSGop,</div> |
| <div class="line"><a name="l00986"></a><span class="lineno"> 986</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** forward_ctx,</div> |
| <div class="line"><a name="l00987"></a><span class="lineno"> 987</span>  <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** forward_fp,</div> |
| <div class="line"><a name="l00988"></a><span class="lineno"> 988</span>  <span class="keywordtype">int</span>* forward_count,</div> |
| <div class="line"><a name="l00989"></a><span class="lineno"> 989</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** backward_ctx,</div> |
| <div class="line"><a name="l00990"></a><span class="lineno"> 990</span>  <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** backward_fp,</div> |
| <div class="line"><a name="l00991"></a><span class="lineno"> 991</span>  <span class="keywordtype">int</span>* backward_count,</div> |
| <div class="line"><a name="l00992"></a><span class="lineno"> 992</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** create_op_ctx,</div> |
| <div class="line"><a name="l00993"></a><span class="lineno"> 993</span>  <a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a>** create_op_fp,</div> |
| <div class="line"><a name="l00994"></a><span class="lineno"> 994</span>  <span class="keywordtype">int</span>* create_op_count,</div> |
| <div class="line"><a name="l00995"></a><span class="lineno"> 995</span>  <a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a>* parse,</div> |
| <div class="line"><a name="l00996"></a><span class="lineno"> 996</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a>* type,</div> |
| <div class="line"><a name="l00997"></a><span class="lineno"> 997</span>  <a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a>* stype,</div> |
| <div class="line"><a name="l00998"></a><span class="lineno"> 998</span>  <a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a>* shape,</div> |
| <div class="line"><a name="l00999"></a><span class="lineno"> 999</span>  <a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a>* mutate);</div> |
| <div class="line"><a name="l01000"></a><span class="lineno"> 1000</span>  </div> |
| <div class="line"><a name="l01001"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aa9581ed05ac3aefc194698100cc54e5a"> 1001</a></span> <span class="preprocessor">#define MXLIB_OPCALLFREE_STR "_opCallFree"</span></div> |
| <div class="line"><a name="l01002"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aff50c6302fa056352a7838a7c0fba79c"> 1002</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#aff50c6302fa056352a7838a7c0fba79c">opCallFree_t</a>)(<span class="keywordtype">void</span>* ptr);</div> |
| <div class="line"><a name="l01003"></a><span class="lineno"> 1003</span>  </div> |
| <div class="line"><a name="l01004"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a6ff60ae5ae2081aac9715da91f38845e"> 1004</a></span> <span class="preprocessor">#define MXLIB_OPCALLPARSEATTRS_STR "_opCallParseAttrs"</span></div> |
| <div class="line"><a name="l01005"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ac3c274938c9da1e44e9897639af5c0d4"> 1005</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ac3c274938c9da1e44e9897639af5c0d4">opCallParseAttrs_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a> parseAttrs,</div> |
| <div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01007"></a><span class="lineno"> 1007</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01008"></a><span class="lineno"> 1008</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01009"></a><span class="lineno"> 1009</span>  <span class="keywordtype">int</span>* num_in,</div> |
| <div class="line"><a name="l01010"></a><span class="lineno"> 1010</span>  <span class="keywordtype">int</span>* num_out);</div> |
| <div class="line"><a name="l01011"></a><span class="lineno"> 1011</span>  </div> |
| <div class="line"><a name="l01012"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac21fd9c191ceab0d17561f4523ad6d23"> 1012</a></span> <span class="preprocessor">#define MXLIB_OPCALLINFERSHAPE_STR "_opCallInferShape"</span></div> |
| <div class="line"><a name="l01013"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ade42ac66b3d7c1e063d0f71eda6394a3"> 1013</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ade42ac66b3d7c1e063d0f71eda6394a3">opCallInferShape_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a> inferShape,</div> |
| <div class="line"><a name="l01014"></a><span class="lineno"> 1014</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01015"></a><span class="lineno"> 1015</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01016"></a><span class="lineno"> 1016</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01017"></a><span class="lineno"> 1017</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes,</div> |
| <div class="line"><a name="l01018"></a><span class="lineno"> 1018</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01019"></a><span class="lineno"> 1019</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01020"></a><span class="lineno"> 1020</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** mod_inshapes,</div> |
| <div class="line"><a name="l01021"></a><span class="lineno"> 1021</span>  <span class="keywordtype">int</span>** mod_indims,</div> |
| <div class="line"><a name="l01022"></a><span class="lineno"> 1022</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** outshapes,</div> |
| <div class="line"><a name="l01023"></a><span class="lineno"> 1023</span>  <span class="keywordtype">int</span>** outdims,</div> |
| <div class="line"><a name="l01024"></a><span class="lineno"> 1024</span>  <span class="keywordtype">int</span> num_out);</div> |
| <div class="line"><a name="l01025"></a><span class="lineno"> 1025</span>  </div> |
| <div class="line"><a name="l01026"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a7dfca2084c7df461862458f77a8f6244"> 1026</a></span> <span class="preprocessor">#define MXLIB_OPCALLINFERTYPE_STR "_opCallInferType"</span></div> |
| <div class="line"><a name="l01027"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a08d9994f1542d34fce1eeabda3a8ad85"> 1027</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a08d9994f1542d34fce1eeabda3a8ad85">opCallInferType_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a> inferType,</div> |
| <div class="line"><a name="l01028"></a><span class="lineno"> 1028</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01029"></a><span class="lineno"> 1029</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01030"></a><span class="lineno"> 1030</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01031"></a><span class="lineno"> 1031</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01032"></a><span class="lineno"> 1032</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01033"></a><span class="lineno"> 1033</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01034"></a><span class="lineno"> 1034</span>  <span class="keywordtype">int</span> num_out);</div> |
| <div class="line"><a name="l01035"></a><span class="lineno"> 1035</span>  </div> |
| <div class="line"><a name="l01036"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a95494016bf876dcb3ec4ae0b3c5f6941"> 1036</a></span> <span class="preprocessor">#define MXLIB_OPCALLINFERSTYPE_STR "_opCallInferSType"</span></div> |
| <div class="line"><a name="l01037"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ae5f9488dd6bdbea94a72c42382c2b6c8"> 1037</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ae5f9488dd6bdbea94a72c42382c2b6c8">opCallInferSType_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a> inferSType,</div> |
| <div class="line"><a name="l01038"></a><span class="lineno"> 1038</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01039"></a><span class="lineno"> 1039</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01040"></a><span class="lineno"> 1040</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01041"></a><span class="lineno"> 1041</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01042"></a><span class="lineno"> 1042</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01043"></a><span class="lineno"> 1043</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01044"></a><span class="lineno"> 1044</span>  <span class="keywordtype">int</span> num_out);</div> |
| <div class="line"><a name="l01045"></a><span class="lineno"> 1045</span>  </div> |
| <div class="line"><a name="l01046"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a906b409889769dcb9ed23c68becf93e3"> 1046</a></span> <span class="preprocessor">#define MXLIB_OPCALLFCOMP_STR "_opCallFCompute"</span></div> |
| <div class="line"><a name="l01047"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af447080bf199456379fd14d68804c1c7"> 1047</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#af447080bf199456379fd14d68804c1c7">opCallFComp_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a> fcomp,</div> |
| <div class="line"><a name="l01048"></a><span class="lineno"> 1048</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01049"></a><span class="lineno"> 1049</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01050"></a><span class="lineno"> 1050</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01051"></a><span class="lineno"> 1051</span>  <span class="keyword">const</span> int64_t** inshapes,</div> |
| <div class="line"><a name="l01052"></a><span class="lineno"> 1052</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01053"></a><span class="lineno"> 1053</span>  <span class="keywordtype">void</span>** indata,</div> |
| <div class="line"><a name="l01054"></a><span class="lineno"> 1054</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01055"></a><span class="lineno"> 1055</span>  <span class="keywordtype">size_t</span>* inIDs,</div> |
| <div class="line"><a name="l01056"></a><span class="lineno"> 1056</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type,</div> |
| <div class="line"><a name="l01057"></a><span class="lineno"> 1057</span>  <span class="keywordtype">int</span>* indev_id,</div> |
| <div class="line"><a name="l01058"></a><span class="lineno"> 1058</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01059"></a><span class="lineno"> 1059</span>  <span class="keyword">const</span> int64_t** outshapes,</div> |
| <div class="line"><a name="l01060"></a><span class="lineno"> 1060</span>  <span class="keywordtype">int</span>* outdims,</div> |
| <div class="line"><a name="l01061"></a><span class="lineno"> 1061</span>  <span class="keywordtype">void</span>** outdata,</div> |
| <div class="line"><a name="l01062"></a><span class="lineno"> 1062</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01063"></a><span class="lineno"> 1063</span>  <span class="keywordtype">size_t</span>* outIDs,</div> |
| <div class="line"><a name="l01064"></a><span class="lineno"> 1064</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div> |
| <div class="line"><a name="l01065"></a><span class="lineno"> 1065</span>  <span class="keywordtype">int</span>* outdev_id,</div> |
| <div class="line"><a name="l01066"></a><span class="lineno"> 1066</span>  <span class="keywordtype">int</span> num_out,</div> |
| <div class="line"><a name="l01067"></a><span class="lineno"> 1067</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc,</div> |
| <div class="line"><a name="l01068"></a><span class="lineno"> 1068</span>  <span class="keywordtype">void</span>* cpu_alloc,</div> |
| <div class="line"><a name="l01069"></a><span class="lineno"> 1069</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc,</div> |
| <div class="line"><a name="l01070"></a><span class="lineno"> 1070</span>  <span class="keywordtype">void</span>* gpu_alloc,</div> |
| <div class="line"><a name="l01071"></a><span class="lineno"> 1071</span>  <span class="keywordtype">void</span>* cuda_stream,</div> |
| <div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc,</div> |
| <div class="line"><a name="l01073"></a><span class="lineno"> 1073</span>  <span class="keywordtype">void</span>* sparse_alloc,</div> |
| <div class="line"><a name="l01074"></a><span class="lineno"> 1074</span>  <span class="keywordtype">int</span>* instypes,</div> |
| <div class="line"><a name="l01075"></a><span class="lineno"> 1075</span>  <span class="keywordtype">int</span>* outstypes,</div> |
| <div class="line"><a name="l01076"></a><span class="lineno"> 1076</span>  <span class="keywordtype">void</span>** in_indices,</div> |
| <div class="line"><a name="l01077"></a><span class="lineno"> 1077</span>  <span class="keywordtype">void</span>** out_indices,</div> |
| <div class="line"><a name="l01078"></a><span class="lineno"> 1078</span>  <span class="keywordtype">void</span>** in_indptr,</div> |
| <div class="line"><a name="l01079"></a><span class="lineno"> 1079</span>  <span class="keywordtype">void</span>** out_indptr,</div> |
| <div class="line"><a name="l01080"></a><span class="lineno"> 1080</span>  int64_t* in_indices_shapes,</div> |
| <div class="line"><a name="l01081"></a><span class="lineno"> 1081</span>  int64_t* out_indices_shapes,</div> |
| <div class="line"><a name="l01082"></a><span class="lineno"> 1082</span>  int64_t* in_indptr_shapes,</div> |
| <div class="line"><a name="l01083"></a><span class="lineno"> 1083</span>  int64_t* out_indptr_shapes,</div> |
| <div class="line"><a name="l01084"></a><span class="lineno"> 1084</span>  <span class="keywordtype">void</span>* rng_cpu_states,</div> |
| <div class="line"><a name="l01085"></a><span class="lineno"> 1085</span>  <span class="keywordtype">void</span>* rng_gpu_states);</div> |
| <div class="line"><a name="l01086"></a><span class="lineno"> 1086</span>  </div> |
| <div class="line"><a name="l01087"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abf2577efc9a9eed8d27d6ca0af98abe1"> 1087</a></span> <span class="preprocessor">#define MXLIB_OPCALLMUTATEINPUTS_STR "_opCallMutateInputs"</span></div> |
| <div class="line"><a name="l01088"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ab867c91f53d13cb949e9c9a579fa3da5"> 1088</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ab867c91f53d13cb949e9c9a579fa3da5">opCallMutateInputs_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a> mutate,</div> |
| <div class="line"><a name="l01089"></a><span class="lineno"> 1089</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01092"></a><span class="lineno"> 1092</span>  <span class="keywordtype">int</span>** mutate_indices,</div> |
| <div class="line"><a name="l01093"></a><span class="lineno"> 1093</span>  <span class="keywordtype">int</span>* indices_size);</div> |
| <div class="line"><a name="l01094"></a><span class="lineno"> 1094</span>  </div> |
| <div class="line"><a name="l01095"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a950d395e618360e1d9c4c5b1fedb018d"> 1095</a></span> <span class="preprocessor">#define MXLIB_OPCALLCREATEOPSTATE_STR "_opCallCreateOpState"</span></div> |
| <div class="line"><a name="l01096"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#adf4bd71f1334318c5258e9421709b03c"> 1096</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#adf4bd71f1334318c5258e9421709b03c">opCallCreateOpState_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">createOpState_t</a> create_op,</div> |
| <div class="line"><a name="l01097"></a><span class="lineno"> 1097</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01098"></a><span class="lineno"> 1098</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01099"></a><span class="lineno"> 1099</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01100"></a><span class="lineno"> 1100</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* dev_type,</div> |
| <div class="line"><a name="l01101"></a><span class="lineno"> 1101</span>  <span class="keywordtype">int</span> dev_id,</div> |
| <div class="line"><a name="l01102"></a><span class="lineno"> 1102</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes,</div> |
| <div class="line"><a name="l01103"></a><span class="lineno"> 1103</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01104"></a><span class="lineno"> 1104</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01106"></a><span class="lineno"> 1106</span>  <span class="keywordtype">void</span>** state_op);</div> |
| <div class="line"><a name="l01107"></a><span class="lineno"> 1107</span>  </div> |
| <div class="line"><a name="l01108"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a1724ab7121cfa78fda47b146eed0e9a9"> 1108</a></span> <span class="preprocessor">#define MXLIB_OPCALLDESTROYOPSTATE_STR "_opCallDestroyOpState"</span></div> |
| <div class="line"><a name="l01109"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70"> 1109</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70">opCallDestroyOpState_t</a>)(<span class="keywordtype">void</span>* state_op);</div> |
| <div class="line"><a name="l01110"></a><span class="lineno"> 1110</span>  </div> |
| <div class="line"><a name="l01111"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad10be40f94357f8913f99288ccae5bea"> 1111</a></span> <span class="preprocessor">#define MXLIB_OPCALLFSTATEFULCOMP_STR "_opCallFStatefulCompute"</span></div> |
| <div class="line"><a name="l01112"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ada6444201d1204384d1287d91a4c3fc7"> 1112</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ada6444201d1204384d1287d91a4c3fc7">opCallFStatefulComp_t</a>)(<span class="keywordtype">int</span> is_forward,</div> |
| <div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>  <span class="keywordtype">void</span>* state_op,</div> |
| <div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>  <span class="keyword">const</span> int64_t** inshapes,</div> |
| <div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01116"></a><span class="lineno"> 1116</span>  <span class="keywordtype">void</span>** indata,</div> |
| <div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01118"></a><span class="lineno"> 1118</span>  <span class="keywordtype">size_t</span>* inIDs,</div> |
| <div class="line"><a name="l01119"></a><span class="lineno"> 1119</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type,</div> |
| <div class="line"><a name="l01120"></a><span class="lineno"> 1120</span>  <span class="keywordtype">int</span>* indev_id,</div> |
| <div class="line"><a name="l01121"></a><span class="lineno"> 1121</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01122"></a><span class="lineno"> 1122</span>  <span class="keyword">const</span> int64_t** outshapes,</div> |
| <div class="line"><a name="l01123"></a><span class="lineno"> 1123</span>  <span class="keywordtype">int</span>* outdims,</div> |
| <div class="line"><a name="l01124"></a><span class="lineno"> 1124</span>  <span class="keywordtype">void</span>** outdata,</div> |
| <div class="line"><a name="l01125"></a><span class="lineno"> 1125</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01126"></a><span class="lineno"> 1126</span>  <span class="keywordtype">size_t</span>* outIDs,</div> |
| <div class="line"><a name="l01127"></a><span class="lineno"> 1127</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div> |
| <div class="line"><a name="l01128"></a><span class="lineno"> 1128</span>  <span class="keywordtype">int</span>* outdev_id,</div> |
| <div class="line"><a name="l01129"></a><span class="lineno"> 1129</span>  <span class="keywordtype">int</span> num_out,</div> |
| <div class="line"><a name="l01130"></a><span class="lineno"> 1130</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc,</div> |
| <div class="line"><a name="l01131"></a><span class="lineno"> 1131</span>  <span class="keywordtype">void</span>* cpu_alloc,</div> |
| <div class="line"><a name="l01132"></a><span class="lineno"> 1132</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc,</div> |
| <div class="line"><a name="l01133"></a><span class="lineno"> 1133</span>  <span class="keywordtype">void</span>* gpu_alloc,</div> |
| <div class="line"><a name="l01134"></a><span class="lineno"> 1134</span>  <span class="keywordtype">void</span>* stream,</div> |
| <div class="line"><a name="l01135"></a><span class="lineno"> 1135</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc,</div> |
| <div class="line"><a name="l01136"></a><span class="lineno"> 1136</span>  <span class="keywordtype">void</span>* sparse_alloc,</div> |
| <div class="line"><a name="l01137"></a><span class="lineno"> 1137</span>  <span class="keywordtype">int</span>* instypes,</div> |
| <div class="line"><a name="l01138"></a><span class="lineno"> 1138</span>  <span class="keywordtype">int</span>* outstypes,</div> |
| <div class="line"><a name="l01139"></a><span class="lineno"> 1139</span>  <span class="keywordtype">void</span>** in_indices,</div> |
| <div class="line"><a name="l01140"></a><span class="lineno"> 1140</span>  <span class="keywordtype">void</span>** out_indices,</div> |
| <div class="line"><a name="l01141"></a><span class="lineno"> 1141</span>  <span class="keywordtype">void</span>** in_indptr,</div> |
| <div class="line"><a name="l01142"></a><span class="lineno"> 1142</span>  <span class="keywordtype">void</span>** out_indptr,</div> |
| <div class="line"><a name="l01143"></a><span class="lineno"> 1143</span>  int64_t* in_indices_shapes,</div> |
| <div class="line"><a name="l01144"></a><span class="lineno"> 1144</span>  int64_t* out_indices_shapes,</div> |
| <div class="line"><a name="l01145"></a><span class="lineno"> 1145</span>  int64_t* in_indptr_shapes,</div> |
| <div class="line"><a name="l01146"></a><span class="lineno"> 1146</span>  int64_t* out_indptr_shapes,</div> |
| <div class="line"><a name="l01147"></a><span class="lineno"> 1147</span>  <span class="keywordtype">void</span>* rng_cpu_states,</div> |
| <div class="line"><a name="l01148"></a><span class="lineno"> 1148</span>  <span class="keywordtype">void</span>* rng_gpu_states);</div> |
| <div class="line"><a name="l01149"></a><span class="lineno"> 1149</span>  </div> |
| <div class="line"><a name="l01150"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a16de6fee0cb7e09e2e3c0885b44d646d"> 1150</a></span> <span class="preprocessor">#define MXLIB_PARTREGSIZE_STR "_partRegSize"</span></div> |
| <div class="line"><a name="l01151"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a8a4d41b35b45f3644497d9a8f717b6a3"> 1151</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a8a4d41b35b45f3644497d9a8f717b6a3">partRegSize_t</a>)(void);</div> |
| <div class="line"><a name="l01152"></a><span class="lineno"> 1152</span>  </div> |
| <div class="line"><a name="l01153"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a96ebb8de1ce9d8f8a493e262b477f0b7"> 1153</a></span> <span class="preprocessor">#define MXLIB_PARTREGGETCOUNT_STR "_partRegGetCount"</span></div> |
| <div class="line"><a name="l01154"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a0b6d87bb4c06678d0be1dd3ad5916a1b"> 1154</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a0b6d87bb4c06678d0be1dd3ad5916a1b">partRegGetCount_t</a>)(<span class="keywordtype">int</span> idx, <span class="keyword">const</span> <span class="keywordtype">char</span>** name);</div> |
| <div class="line"><a name="l01155"></a><span class="lineno"> 1155</span>  </div> |
| <div class="line"><a name="l01156"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a76dda0c1c4d6eafc6ac80a72527ec556"> 1156</a></span> <span class="preprocessor">#define MXLIB_PARTREGGET_STR "_partRegGet"</span></div> |
| <div class="line"><a name="l01157"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ae628507b542734ee29ddf2f499fbdb05"> 1157</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#ae628507b542734ee29ddf2f499fbdb05">partRegGet_t</a>)(<span class="keywordtype">int</span> part_idx,</div> |
| <div class="line"><a name="l01158"></a><span class="lineno"> 1158</span>  <span class="keywordtype">int</span> stg_idx,</div> |
| <div class="line"><a name="l01159"></a><span class="lineno"> 1159</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** strategy,</div> |
| <div class="line"><a name="l01160"></a><span class="lineno"> 1160</span>  <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a>* supportedOps,</div> |
| <div class="line"><a name="l01161"></a><span class="lineno"> 1161</span>  <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a>* createSelector,</div> |
| <div class="line"><a name="l01162"></a><span class="lineno"> 1162</span>  <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a>* reviewSubgraph,</div> |
| <div class="line"><a name="l01163"></a><span class="lineno"> 1163</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** op_name);</div> |
| <div class="line"><a name="l01164"></a><span class="lineno"> 1164</span>  </div> |
| <div class="line"><a name="l01165"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a7e8e351e67da4f1822042c0d01e8bbf4"> 1165</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSUPPORTEDOPS_STR "_partCallSupportedOps"</span></div> |
| <div class="line"><a name="l01166"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a03342e626369680bc842ec763588a91e"> 1166</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a03342e626369680bc842ec763588a91e">partCallSupportedOps_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> supportedOps,</div> |
| <div class="line"><a name="l01167"></a><span class="lineno"> 1167</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01168"></a><span class="lineno"> 1168</span>  <span class="keywordtype">int</span> num_ids,</div> |
| <div class="line"><a name="l01169"></a><span class="lineno"> 1169</span>  <span class="keywordtype">int</span>* ids,</div> |
| <div class="line"><a name="l01170"></a><span class="lineno"> 1170</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01171"></a><span class="lineno"> 1171</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01172"></a><span class="lineno"> 1172</span>  <span class="keywordtype">int</span> num_opts);</div> |
| <div class="line"><a name="l01173"></a><span class="lineno"> 1173</span>  </div> |
| <div class="line"><a name="l01174"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2136a3986fc3a229cb426f41d0dfe3c8"> 1174</a></span> <span class="preprocessor">#define MXLIB_PARTCALLCREATESELECTOR_STR "_partCallCreateSelector"</span></div> |
| <div class="line"><a name="l01175"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ad62142c24acef23ecd62e7fd53000349"> 1175</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#ad62142c24acef23ecd62e7fd53000349">partCallCreateSelector_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> createSelector,</div> |
| <div class="line"><a name="l01176"></a><span class="lineno"> 1176</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01177"></a><span class="lineno"> 1177</span>  <span class="keywordtype">void</span>** selector,</div> |
| <div class="line"><a name="l01178"></a><span class="lineno"> 1178</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01179"></a><span class="lineno"> 1179</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01180"></a><span class="lineno"> 1180</span>  <span class="keywordtype">int</span> num_opts);</div> |
| <div class="line"><a name="l01181"></a><span class="lineno"> 1181</span>  </div> |
| <div class="line"><a name="l01182"></a><span class="lineno"><a class="line" href="lib__api_8h.html#acca6909f1ef0b0a4e7dc4815dd75255a"> 1182</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSELECT_STR "_partCallSelect"</span></div> |
| <div class="line"><a name="l01183"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a462103a47b8f387ee5a31393c0a47c1a"> 1183</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a462103a47b8f387ee5a31393c0a47c1a">partCallSelect_t</a>)(<span class="keywordtype">void</span>* sel_inst, <span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span>* selected);</div> |
| <div class="line"><a name="l01184"></a><span class="lineno"> 1184</span>  </div> |
| <div class="line"><a name="l01185"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a84b9b257f62fcfa8e55af189fe7d88cb"> 1185</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSELECTINPUT_STR "_partCallSelectInput"</span></div> |
| <div class="line"><a name="l01186"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ad2fed961bb0a655fb91fc6ee4c26162a"> 1186</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#ad2fed961bb0a655fb91fc6ee4c26162a">partCallSelectInput_t</a>)(<span class="keywordtype">void</span>* sel_inst, <span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> input_nodeID, <span class="keywordtype">int</span>* selected);</div> |
| <div class="line"><a name="l01187"></a><span class="lineno"> 1187</span>  </div> |
| <div class="line"><a name="l01188"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abf23d7995f4c8617b136c6c311405b35"> 1188</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSELECTOUTPUT_STR "_partCallSelectOutput"</span></div> |
| <div class="line"><a name="l01189"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a5b53f5861183811ba8fc2cdf9dbee5f6"> 1189</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a5b53f5861183811ba8fc2cdf9dbee5f6">partCallSelectOutput_t</a>)(<span class="keywordtype">void</span>* sel_inst,</div> |
| <div class="line"><a name="l01190"></a><span class="lineno"> 1190</span>  <span class="keywordtype">int</span> nodeID,</div> |
| <div class="line"><a name="l01191"></a><span class="lineno"> 1191</span>  <span class="keywordtype">int</span> output_nodeID,</div> |
| <div class="line"><a name="l01192"></a><span class="lineno"> 1192</span>  <span class="keywordtype">int</span>* selected);</div> |
| <div class="line"><a name="l01193"></a><span class="lineno"> 1193</span>  </div> |
| <div class="line"><a name="l01194"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad43d2f7d1ca8f2b248630ef95b3eb5e9"> 1194</a></span> <span class="preprocessor">#define MXLIB_PARTCALLFILTER_STR "_partCallFilter"</span></div> |
| <div class="line"><a name="l01195"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a7ad34e214897e21a5c10bc57ddf1b51a"> 1195</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a7ad34e214897e21a5c10bc57ddf1b51a">partCallFilter_t</a>)(<span class="keywordtype">void</span>* sel_inst,</div> |
| <div class="line"><a name="l01196"></a><span class="lineno"> 1196</span>  <span class="keywordtype">int</span>* candidates,</div> |
| <div class="line"><a name="l01197"></a><span class="lineno"> 1197</span>  <span class="keywordtype">int</span> num_candidates,</div> |
| <div class="line"><a name="l01198"></a><span class="lineno"> 1198</span>  <span class="keywordtype">int</span>** keep,</div> |
| <div class="line"><a name="l01199"></a><span class="lineno"> 1199</span>  <span class="keywordtype">int</span>* num_keep);</div> |
| <div class="line"><a name="l01200"></a><span class="lineno"> 1200</span>  </div> |
| <div class="line"><a name="l01201"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ae39d5c24d804a90a045483fe5d87b2c2"> 1201</a></span> <span class="preprocessor">#define MXLIB_PARTCALLRESET_STR "_partCallReset"</span></div> |
| <div class="line"><a name="l01202"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a32fc355e422990910299065269f10459"> 1202</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a32fc355e422990910299065269f10459">partCallReset_t</a>)(<span class="keywordtype">void</span>* sel_inst);</div> |
| <div class="line"><a name="l01203"></a><span class="lineno"> 1203</span>  </div> |
| <div class="line"><a name="l01204"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aebf42394c4714072d81f4612a9d425fe"> 1204</a></span> <span class="preprocessor">#define MXLIB_PARTCALLREVIEWSUBGRAPH_STR "_partCallReviewSubgraph"</span></div> |
| <div class="line"><a name="l01205"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aefb1c0f85ebec11dd8e4f8720392ce4a"> 1205</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#aefb1c0f85ebec11dd8e4f8720392ce4a">partCallReviewSubgraph_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> reviewSubgraph,</div> |
| <div class="line"><a name="l01206"></a><span class="lineno"> 1206</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01207"></a><span class="lineno"> 1207</span>  <span class="keywordtype">int</span> subgraph_id,</div> |
| <div class="line"><a name="l01208"></a><span class="lineno"> 1208</span>  <span class="keywordtype">int</span>* accept,</div> |
| <div class="line"><a name="l01209"></a><span class="lineno"> 1209</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01210"></a><span class="lineno"> 1210</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01211"></a><span class="lineno"> 1211</span>  <span class="keywordtype">int</span> num_opts,</div> |
| <div class="line"><a name="l01212"></a><span class="lineno"> 1212</span>  <span class="keywordtype">char</span>*** attr_keys,</div> |
| <div class="line"><a name="l01213"></a><span class="lineno"> 1213</span>  <span class="keywordtype">char</span>*** attr_vals,</div> |
| <div class="line"><a name="l01214"></a><span class="lineno"> 1214</span>  <span class="keywordtype">int</span>* num_attrs,</div> |
| <div class="line"><a name="l01215"></a><span class="lineno"> 1215</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names,</div> |
| <div class="line"><a name="l01216"></a><span class="lineno"> 1216</span>  <span class="keywordtype">int</span> num_args,</div> |
| <div class="line"><a name="l01217"></a><span class="lineno"> 1217</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data,</div> |
| <div class="line"><a name="l01218"></a><span class="lineno"> 1218</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div> |
| <div class="line"><a name="l01219"></a><span class="lineno"> 1219</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims,</div> |
| <div class="line"><a name="l01220"></a><span class="lineno"> 1220</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div> |
| <div class="line"><a name="l01221"></a><span class="lineno"> 1221</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs,</div> |
| <div class="line"><a name="l01222"></a><span class="lineno"> 1222</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div> |
| <div class="line"><a name="l01223"></a><span class="lineno"> 1223</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div> |
| <div class="line"><a name="l01224"></a><span class="lineno"> 1224</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names,</div> |
| <div class="line"><a name="l01225"></a><span class="lineno"> 1225</span>  <span class="keywordtype">int</span> num_aux,</div> |
| <div class="line"><a name="l01226"></a><span class="lineno"> 1226</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data,</div> |
| <div class="line"><a name="l01227"></a><span class="lineno"> 1227</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div> |
| <div class="line"><a name="l01228"></a><span class="lineno"> 1228</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims,</div> |
| <div class="line"><a name="l01229"></a><span class="lineno"> 1229</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div> |
| <div class="line"><a name="l01230"></a><span class="lineno"> 1230</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs,</div> |
| <div class="line"><a name="l01231"></a><span class="lineno"> 1231</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div> |
| <div class="line"><a name="l01232"></a><span class="lineno"> 1232</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id);</div> |
| <div class="line"><a name="l01233"></a><span class="lineno"> 1233</span>  </div> |
| <div class="line"><a name="l01234"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a6ec8627b92e8212d9e3368619e8361a4"> 1234</a></span> <span class="preprocessor">#define MXLIB_PASSREGSIZE_STR "_passRegSize"</span></div> |
| <div class="line"><a name="l01235"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6f47c7152edff58ce548870e7c2de0ab"> 1235</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a6f47c7152edff58ce548870e7c2de0ab">passRegSize_t</a>)(void);</div> |
| <div class="line"><a name="l01236"></a><span class="lineno"> 1236</span>  </div> |
| <div class="line"><a name="l01237"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a303c810b43b30121a5f24ed880d53d09"> 1237</a></span> <span class="preprocessor">#define MXLIB_PASSREGGET_STR "_passRegGet"</span></div> |
| <div class="line"><a name="l01238"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a83e21d3dbcb68574f93c95db0a83ce31"> 1238</a></span> <span class="keyword">typedef</span> void (*<a class="code" href="namespacemxnet_1_1ext.html#a83e21d3dbcb68574f93c95db0a83ce31">passRegGet_t</a>)(<span class="keywordtype">int</span> pass_idx, <a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a>* graphPass, <span class="keyword">const</span> <span class="keywordtype">char</span>** pass_name);</div> |
| <div class="line"><a name="l01239"></a><span class="lineno"> 1239</span>  </div> |
| <div class="line"><a name="l01240"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a5ebc06cae18b2b72e41789d816734540"> 1240</a></span> <span class="preprocessor">#define MXLIB_PASSCALLGRAPHPASS_STR "_passCallGraphPass"</span></div> |
| <div class="line"><a name="l01241"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a4d96003e8adf76b07e993b83f7cef5bd"> 1241</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a4d96003e8adf76b07e993b83f7cef5bd">passCallGraphPass_t</a>)(<a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> graphPass,</div> |
| <div class="line"><a name="l01242"></a><span class="lineno"> 1242</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* in_graph,</div> |
| <div class="line"><a name="l01243"></a><span class="lineno"> 1243</span>  <span class="keywordtype">char</span>** out_graph,</div> |
| <div class="line"><a name="l01244"></a><span class="lineno"> 1244</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01245"></a><span class="lineno"> 1245</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01246"></a><span class="lineno"> 1246</span>  <span class="keywordtype">int</span> num_opts,</div> |
| <div class="line"><a name="l01247"></a><span class="lineno"> 1247</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* pass_name,</div> |
| <div class="line"><a name="l01248"></a><span class="lineno"> 1248</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names,</div> |
| <div class="line"><a name="l01249"></a><span class="lineno"> 1249</span>  <span class="keywordtype">int</span> num_args,</div> |
| <div class="line"><a name="l01250"></a><span class="lineno"> 1250</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data,</div> |
| <div class="line"><a name="l01251"></a><span class="lineno"> 1251</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div> |
| <div class="line"><a name="l01252"></a><span class="lineno"> 1252</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims,</div> |
| <div class="line"><a name="l01253"></a><span class="lineno"> 1253</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div> |
| <div class="line"><a name="l01254"></a><span class="lineno"> 1254</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs,</div> |
| <div class="line"><a name="l01255"></a><span class="lineno"> 1255</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div> |
| <div class="line"><a name="l01256"></a><span class="lineno"> 1256</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div> |
| <div class="line"><a name="l01257"></a><span class="lineno"> 1257</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names,</div> |
| <div class="line"><a name="l01258"></a><span class="lineno"> 1258</span>  <span class="keywordtype">int</span> num_aux,</div> |
| <div class="line"><a name="l01259"></a><span class="lineno"> 1259</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data,</div> |
| <div class="line"><a name="l01260"></a><span class="lineno"> 1260</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div> |
| <div class="line"><a name="l01261"></a><span class="lineno"> 1261</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims,</div> |
| <div class="line"><a name="l01262"></a><span class="lineno"> 1262</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div> |
| <div class="line"><a name="l01263"></a><span class="lineno"> 1263</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs,</div> |
| <div class="line"><a name="l01264"></a><span class="lineno"> 1264</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div> |
| <div class="line"><a name="l01265"></a><span class="lineno"> 1265</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id,</div> |
| <div class="line"><a name="l01266"></a><span class="lineno"> 1266</span>  <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc,</div> |
| <div class="line"><a name="l01267"></a><span class="lineno"> 1267</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc);</div> |
| <div class="line"><a name="l01268"></a><span class="lineno"> 1268</span>  </div> |
| <div class="line"><a name="l01269"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad5405711bcd25d56b2fc2a3fd827ea3c"> 1269</a></span> <span class="preprocessor">#define MXLIB_INITIALIZE_STR "initialize"</span></div> |
| <div class="line"><a name="l01270"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a9089010cd352a7e5a36fc3ade38dd221"> 1270</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a9089010cd352a7e5a36fc3ade38dd221">initialize_t</a>)(<span class="keywordtype">int</span> version);</div> |
| <div class="line"><a name="l01271"></a><span class="lineno"> 1271</span>  </div> |
| <div class="line"><a name="l01272"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a13594d09f1287481f1b7b6b695d6c1d8"> 1272</a></span> <span class="preprocessor">#define MXLIB_OPVERSION_STR "_opVersion"</span></div> |
| <div class="line"><a name="l01273"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a69ba2ade8bd893eb8461a21cd8f60eb5"> 1273</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a69ba2ade8bd893eb8461a21cd8f60eb5">opVersion_t</a>)();</div> |
| <div class="line"><a name="l01274"></a><span class="lineno"> 1274</span>  </div> |
| <div class="line"><a name="l01275"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abbae445517e023b3f8f81b1999171ae5"> 1275</a></span> <span class="preprocessor">#define MXLIB_MSGSIZE_STR "_msgSize"</span></div> |
| <div class="line"><a name="l01276"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a82c4b41a9a9a1160f1043d10dc9fcf50"> 1276</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#a82c4b41a9a9a1160f1043d10dc9fcf50">msgSize_t</a>)(void);</div> |
| <div class="line"><a name="l01277"></a><span class="lineno"> 1277</span>  </div> |
| <div class="line"><a name="l01278"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a12e8d84f980811cd9a90e5e4506d2011"> 1278</a></span> <span class="preprocessor">#define MXLIB_MSGGET_STR "_msgGet"</span></div> |
| <div class="line"><a name="l01279"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aa7339fd6ce1c23dd296015f450463cc4"> 1279</a></span> <span class="keyword">typedef</span> int (*<a class="code" href="namespacemxnet_1_1ext.html#aa7339fd6ce1c23dd296015f450463cc4">msgGet_t</a>)(<span class="keywordtype">int</span> idx, <span class="keyword">const</span> <span class="keywordtype">char</span>** msg);</div> |
| <div class="line"><a name="l01280"></a><span class="lineno"> 1280</span>  </div> |
| <div class="line"><a name="l01282"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html"> 1282</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html">CustomStatefulOpWrapper</a> {</div> |
| <div class="line"><a name="l01283"></a><span class="lineno"> 1283</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l01284"></a><span class="lineno"> 1284</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#a670e9614e832a6d520e90d56d82c5890">~CustomStatefulOpWrapper</a>();</div> |
| <div class="line"><a name="l01285"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#aad23e82ea3f745ac3a92c07b6afed1d2"> 1285</a></span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#aad23e82ea3f745ac3a92c07b6afed1d2">CustomStatefulOpWrapper</a>(<a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>* inst, <a class="code" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70">opCallDestroyOpState_t</a> destroy)</div> |
| <div class="line"><a name="l01286"></a><span class="lineno"> 1286</span>  : instance(inst), destroy_(destroy) {}</div> |
| <div class="line"><a name="l01287"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#ace82ec005741e89957ba7fdfe433f654"> 1287</a></span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>* <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#ace82ec005741e89957ba7fdfe433f654">get_instance</a>() {</div> |
| <div class="line"><a name="l01288"></a><span class="lineno"> 1288</span>  <span class="keywordflow">return</span> instance;</div> |
| <div class="line"><a name="l01289"></a><span class="lineno"> 1289</span>  }</div> |
| <div class="line"><a name="l01290"></a><span class="lineno"> 1290</span>  </div> |
| <div class="line"><a name="l01291"></a><span class="lineno"> 1291</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l01292"></a><span class="lineno"> 1292</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>* instance;</div> |
| <div class="line"><a name="l01293"></a><span class="lineno"> 1293</span>  <a class="code" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70">opCallDestroyOpState_t</a> destroy_;</div> |
| <div class="line"><a name="l01294"></a><span class="lineno"> 1294</span> };</div> |
| <div class="line"><a name="l01295"></a><span class="lineno"> 1295</span>  </div> |
| <div class="line"><a name="l01296"></a><span class="lineno"> 1296</span> <span class="preprocessor">#if defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__)</span></div> |
| <div class="line"><a name="l01297"></a><span class="lineno"> 1297</span> <span class="preprocessor">#define MX_INT_RET __declspec(dllexport) int __cdecl</span></div> |
| <div class="line"><a name="l01298"></a><span class="lineno"> 1298</span> <span class="preprocessor">#define MX_VOID_RET __declspec(dllexport) void __cdecl</span></div> |
| <div class="line"><a name="l01299"></a><span class="lineno"> 1299</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l01300"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05"> 1300</a></span> <span class="preprocessor">#define MX_INT_RET int</span></div> |
| <div class="line"><a name="l01301"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa"> 1301</a></span> <span class="preprocessor">#define MX_VOID_RET void</span></div> |
| <div class="line"><a name="l01302"></a><span class="lineno"> 1302</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l01303"></a><span class="lineno"> 1303</span>  </div> |
| <div class="line"><a name="l01304"></a><span class="lineno"> 1304</span> } <span class="comment">// namespace ext</span></div> |
| <div class="line"><a name="l01305"></a><span class="lineno"> 1305</span> } <span class="comment">// namespace mxnet</span></div> |
| <div class="line"><a name="l01306"></a><span class="lineno"> 1306</span>  </div> |
| <div class="line"><a name="l01307"></a><span class="lineno"> 1307</span> <span class="keyword">extern</span> <span class="stringliteral">"C"</span> {</div> |
| <div class="line"><a name="l01309"></a><span class="lineno"> 1309</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#ade85fac9a80edcdf9e6062cb079cd325">_opVersion</a>();</div> |
| <div class="line"><a name="l01310"></a><span class="lineno"> 1310</span>  </div> |
| <div class="line"><a name="l01312"></a><span class="lineno"> 1312</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a6efacb1a30200f9f1fb05f77783959de">_opRegSize</a>();</div> |
| <div class="line"><a name="l01313"></a><span class="lineno"> 1313</span>  </div> |
| <div class="line"><a name="l01315"></a><span class="lineno"> 1315</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#a5b0657e0605ca60b79241a4324139f63">_opRegGet</a>(<span class="keywordtype">int</span> idx,</div> |
| <div class="line"><a name="l01316"></a><span class="lineno"> 1316</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** name,</div> |
| <div class="line"><a name="l01317"></a><span class="lineno"> 1317</span>  <span class="keywordtype">int</span>* isSGop,</div> |
| <div class="line"><a name="l01318"></a><span class="lineno"> 1318</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** forward_ctx,</div> |
| <div class="line"><a name="l01319"></a><span class="lineno"> 1319</span>  <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** forward_fp,</div> |
| <div class="line"><a name="l01320"></a><span class="lineno"> 1320</span>  <span class="keywordtype">int</span>* forward_count,</div> |
| <div class="line"><a name="l01321"></a><span class="lineno"> 1321</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** backward_ctx,</div> |
| <div class="line"><a name="l01322"></a><span class="lineno"> 1322</span>  <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** backward_fp,</div> |
| <div class="line"><a name="l01323"></a><span class="lineno"> 1323</span>  <span class="keywordtype">int</span>* backward_count,</div> |
| <div class="line"><a name="l01324"></a><span class="lineno"> 1324</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** create_op_ctx,</div> |
| <div class="line"><a name="l01325"></a><span class="lineno"> 1325</span>  <a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a>** create_op_fp,</div> |
| <div class="line"><a name="l01326"></a><span class="lineno"> 1326</span>  <span class="keywordtype">int</span>* create_op_count,</div> |
| <div class="line"><a name="l01327"></a><span class="lineno"> 1327</span>  <a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a>* parse,</div> |
| <div class="line"><a name="l01328"></a><span class="lineno"> 1328</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a>* type,</div> |
| <div class="line"><a name="l01329"></a><span class="lineno"> 1329</span>  <a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a>* stype,</div> |
| <div class="line"><a name="l01330"></a><span class="lineno"> 1330</span>  <a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a>* shape,</div> |
| <div class="line"><a name="l01331"></a><span class="lineno"> 1331</span>  <a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a>* mutate);</div> |
| <div class="line"><a name="l01332"></a><span class="lineno"> 1332</span>  </div> |
| <div class="line"><a name="l01334"></a><span class="lineno"> 1334</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#a6e74b9b5a033075b0a934561a518a2cb">_opCallFree</a>(<span class="keywordtype">void</span>* ptr);</div> |
| <div class="line"><a name="l01335"></a><span class="lineno"> 1335</span>  </div> |
| <div class="line"><a name="l01337"></a><span class="lineno"> 1337</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#ac88d6741ca655523db73fc18a8e06828">_opCallParseAttrs</a>(<a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a> parseAttrs,</div> |
| <div class="line"><a name="l01338"></a><span class="lineno"> 1338</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01339"></a><span class="lineno"> 1339</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01340"></a><span class="lineno"> 1340</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01341"></a><span class="lineno"> 1341</span>  <span class="keywordtype">int</span>* num_in,</div> |
| <div class="line"><a name="l01342"></a><span class="lineno"> 1342</span>  <span class="keywordtype">int</span>* num_out);</div> |
| <div class="line"><a name="l01343"></a><span class="lineno"> 1343</span>  </div> |
| <div class="line"><a name="l01345"></a><span class="lineno"> 1345</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a2ba7371715fcd75f05c0ba0f0c8b78b4">_opCallInferShape</a>(<a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a> inferShape,</div> |
| <div class="line"><a name="l01346"></a><span class="lineno"> 1346</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01347"></a><span class="lineno"> 1347</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01348"></a><span class="lineno"> 1348</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01349"></a><span class="lineno"> 1349</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes,</div> |
| <div class="line"><a name="l01350"></a><span class="lineno"> 1350</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01351"></a><span class="lineno"> 1351</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01352"></a><span class="lineno"> 1352</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** mod_inshapes,</div> |
| <div class="line"><a name="l01353"></a><span class="lineno"> 1353</span>  <span class="keywordtype">int</span>** mod_indims,</div> |
| <div class="line"><a name="l01354"></a><span class="lineno"> 1354</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** outshapes,</div> |
| <div class="line"><a name="l01355"></a><span class="lineno"> 1355</span>  <span class="keywordtype">int</span>** outdims,</div> |
| <div class="line"><a name="l01356"></a><span class="lineno"> 1356</span>  <span class="keywordtype">int</span> num_out);</div> |
| <div class="line"><a name="l01357"></a><span class="lineno"> 1357</span>  </div> |
| <div class="line"><a name="l01359"></a><span class="lineno"> 1359</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#adb3450125394a91102043bcc28fe8e39">_opCallInferType</a>(<a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a> inferType,</div> |
| <div class="line"><a name="l01360"></a><span class="lineno"> 1360</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01361"></a><span class="lineno"> 1361</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01362"></a><span class="lineno"> 1362</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01363"></a><span class="lineno"> 1363</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01364"></a><span class="lineno"> 1364</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01365"></a><span class="lineno"> 1365</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01366"></a><span class="lineno"> 1366</span>  <span class="keywordtype">int</span> num_out);</div> |
| <div class="line"><a name="l01367"></a><span class="lineno"> 1367</span>  </div> |
| <div class="line"><a name="l01369"></a><span class="lineno"> 1369</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a67702f536b0652fab92a46a4c522f0a8">_opCallInferSType</a>(<a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a> inferSType,</div> |
| <div class="line"><a name="l01370"></a><span class="lineno"> 1370</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01371"></a><span class="lineno"> 1371</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01372"></a><span class="lineno"> 1372</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01373"></a><span class="lineno"> 1373</span>  <span class="keywordtype">int</span>* instypes,</div> |
| <div class="line"><a name="l01374"></a><span class="lineno"> 1374</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01375"></a><span class="lineno"> 1375</span>  <span class="keywordtype">int</span>* outstypes,</div> |
| <div class="line"><a name="l01376"></a><span class="lineno"> 1376</span>  <span class="keywordtype">int</span> num_out);</div> |
| <div class="line"><a name="l01377"></a><span class="lineno"> 1377</span>  </div> |
| <div class="line"><a name="l01379"></a><span class="lineno"> 1379</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a948af234ca1c3b0c8b78e8db2f32c310">_opCallFCompute</a>(<a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a> fcomp,</div> |
| <div class="line"><a name="l01380"></a><span class="lineno"> 1380</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01381"></a><span class="lineno"> 1381</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01382"></a><span class="lineno"> 1382</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01383"></a><span class="lineno"> 1383</span>  <span class="keyword">const</span> int64_t** inshapes,</div> |
| <div class="line"><a name="l01384"></a><span class="lineno"> 1384</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01385"></a><span class="lineno"> 1385</span>  <span class="keywordtype">void</span>** indata,</div> |
| <div class="line"><a name="l01386"></a><span class="lineno"> 1386</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01387"></a><span class="lineno"> 1387</span>  <span class="keywordtype">size_t</span>* inIDs,</div> |
| <div class="line"><a name="l01388"></a><span class="lineno"> 1388</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type,</div> |
| <div class="line"><a name="l01389"></a><span class="lineno"> 1389</span>  <span class="keywordtype">int</span>* indev_id,</div> |
| <div class="line"><a name="l01390"></a><span class="lineno"> 1390</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01391"></a><span class="lineno"> 1391</span>  <span class="keyword">const</span> int64_t** outshapes,</div> |
| <div class="line"><a name="l01392"></a><span class="lineno"> 1392</span>  <span class="keywordtype">int</span>* outdims,</div> |
| <div class="line"><a name="l01393"></a><span class="lineno"> 1393</span>  <span class="keywordtype">void</span>** outdata,</div> |
| <div class="line"><a name="l01394"></a><span class="lineno"> 1394</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01395"></a><span class="lineno"> 1395</span>  <span class="keywordtype">size_t</span>* outIDs,</div> |
| <div class="line"><a name="l01396"></a><span class="lineno"> 1396</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div> |
| <div class="line"><a name="l01397"></a><span class="lineno"> 1397</span>  <span class="keywordtype">int</span>* outdev_id,</div> |
| <div class="line"><a name="l01398"></a><span class="lineno"> 1398</span>  <span class="keywordtype">int</span> num_out,</div> |
| <div class="line"><a name="l01399"></a><span class="lineno"> 1399</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> cpu_malloc,</div> |
| <div class="line"><a name="l01400"></a><span class="lineno"> 1400</span>  <span class="keywordtype">void</span>* cpu_alloc,</div> |
| <div class="line"><a name="l01401"></a><span class="lineno"> 1401</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> gpu_malloc,</div> |
| <div class="line"><a name="l01402"></a><span class="lineno"> 1402</span>  <span class="keywordtype">void</span>* gpu_alloc,</div> |
| <div class="line"><a name="l01403"></a><span class="lineno"> 1403</span>  <span class="keywordtype">void</span>* cuda_stream,</div> |
| <div class="line"><a name="l01404"></a><span class="lineno"> 1404</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">mxnet::ext::sparse_malloc_t</a> sparse_malloc,</div> |
| <div class="line"><a name="l01405"></a><span class="lineno"> 1405</span>  <span class="keywordtype">void</span>* sparse_alloc,</div> |
| <div class="line"><a name="l01406"></a><span class="lineno"> 1406</span>  <span class="keywordtype">int</span>* instypes,</div> |
| <div class="line"><a name="l01407"></a><span class="lineno"> 1407</span>  <span class="keywordtype">int</span>* outstypes,</div> |
| <div class="line"><a name="l01408"></a><span class="lineno"> 1408</span>  <span class="keywordtype">void</span>** in_indices,</div> |
| <div class="line"><a name="l01409"></a><span class="lineno"> 1409</span>  <span class="keywordtype">void</span>** out_indices,</div> |
| <div class="line"><a name="l01410"></a><span class="lineno"> 1410</span>  <span class="keywordtype">void</span>** in_indptr,</div> |
| <div class="line"><a name="l01411"></a><span class="lineno"> 1411</span>  <span class="keywordtype">void</span>** out_indptr,</div> |
| <div class="line"><a name="l01412"></a><span class="lineno"> 1412</span>  int64_t* in_indices_shapes,</div> |
| <div class="line"><a name="l01413"></a><span class="lineno"> 1413</span>  int64_t* out_indices_shapes,</div> |
| <div class="line"><a name="l01414"></a><span class="lineno"> 1414</span>  int64_t* in_indptr_shapes,</div> |
| <div class="line"><a name="l01415"></a><span class="lineno"> 1415</span>  int64_t* out_indptr_shapes,</div> |
| <div class="line"><a name="l01416"></a><span class="lineno"> 1416</span>  <span class="keywordtype">void</span>* rng_cpu_states,</div> |
| <div class="line"><a name="l01417"></a><span class="lineno"> 1417</span>  <span class="keywordtype">void</span>* rng_gpu_states);</div> |
| <div class="line"><a name="l01418"></a><span class="lineno"> 1418</span>  </div> |
| <div class="line"><a name="l01420"></a><span class="lineno"> 1420</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#aa14e95577fdb7526da526ae1ec236d48">_opCallMutateInputs</a>(<a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a> mutate,</div> |
| <div class="line"><a name="l01421"></a><span class="lineno"> 1421</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01422"></a><span class="lineno"> 1422</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01423"></a><span class="lineno"> 1423</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01424"></a><span class="lineno"> 1424</span>  <span class="keywordtype">int</span>** mutate_indices,</div> |
| <div class="line"><a name="l01425"></a><span class="lineno"> 1425</span>  <span class="keywordtype">int</span>* indices_size);</div> |
| <div class="line"><a name="l01426"></a><span class="lineno"> 1426</span>  </div> |
| <div class="line"><a name="l01428"></a><span class="lineno"> 1428</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#af358af16fb117cd0675234835a4e7c08">_opCallCreateOpState</a>(<a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a> create_op,</div> |
| <div class="line"><a name="l01429"></a><span class="lineno"> 1429</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div> |
| <div class="line"><a name="l01430"></a><span class="lineno"> 1430</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div> |
| <div class="line"><a name="l01431"></a><span class="lineno"> 1431</span>  <span class="keywordtype">int</span> num,</div> |
| <div class="line"><a name="l01432"></a><span class="lineno"> 1432</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* dev_type,</div> |
| <div class="line"><a name="l01433"></a><span class="lineno"> 1433</span>  <span class="keywordtype">int</span> dev_id,</div> |
| <div class="line"><a name="l01434"></a><span class="lineno"> 1434</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes,</div> |
| <div class="line"><a name="l01435"></a><span class="lineno"> 1435</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01436"></a><span class="lineno"> 1436</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01437"></a><span class="lineno"> 1437</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01438"></a><span class="lineno"> 1438</span>  <span class="keywordtype">void</span>** state_op);</div> |
| <div class="line"><a name="l01439"></a><span class="lineno"> 1439</span>  </div> |
| <div class="line"><a name="l01441"></a><span class="lineno"> 1441</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#a0cc29cb55b56ebd7e288b3f5946b5bce">_opCallDestroyOpState</a>(<span class="keywordtype">void</span>* state_op);</div> |
| <div class="line"><a name="l01442"></a><span class="lineno"> 1442</span>  </div> |
| <div class="line"><a name="l01444"></a><span class="lineno"> 1444</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#af0214213e9b925b0750a33f20ff2a1f9">_opCallFStatefulCompute</a>(<span class="keywordtype">int</span> is_forward,</div> |
| <div class="line"><a name="l01445"></a><span class="lineno"> 1445</span>  <span class="keywordtype">void</span>* state_op,</div> |
| <div class="line"><a name="l01446"></a><span class="lineno"> 1446</span>  <span class="keyword">const</span> int64_t** inshapes,</div> |
| <div class="line"><a name="l01447"></a><span class="lineno"> 1447</span>  <span class="keywordtype">int</span>* indims,</div> |
| <div class="line"><a name="l01448"></a><span class="lineno"> 1448</span>  <span class="keywordtype">void</span>** indata,</div> |
| <div class="line"><a name="l01449"></a><span class="lineno"> 1449</span>  <span class="keywordtype">int</span>* intypes,</div> |
| <div class="line"><a name="l01450"></a><span class="lineno"> 1450</span>  <span class="keywordtype">size_t</span>* inIDs,</div> |
| <div class="line"><a name="l01451"></a><span class="lineno"> 1451</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type,</div> |
| <div class="line"><a name="l01452"></a><span class="lineno"> 1452</span>  <span class="keywordtype">int</span>* indev_id,</div> |
| <div class="line"><a name="l01453"></a><span class="lineno"> 1453</span>  <span class="keywordtype">int</span> num_in,</div> |
| <div class="line"><a name="l01454"></a><span class="lineno"> 1454</span>  <span class="keyword">const</span> int64_t** outshapes,</div> |
| <div class="line"><a name="l01455"></a><span class="lineno"> 1455</span>  <span class="keywordtype">int</span>* outdims,</div> |
| <div class="line"><a name="l01456"></a><span class="lineno"> 1456</span>  <span class="keywordtype">void</span>** outdata,</div> |
| <div class="line"><a name="l01457"></a><span class="lineno"> 1457</span>  <span class="keywordtype">int</span>* outtypes,</div> |
| <div class="line"><a name="l01458"></a><span class="lineno"> 1458</span>  <span class="keywordtype">size_t</span>* outIDs,</div> |
| <div class="line"><a name="l01459"></a><span class="lineno"> 1459</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div> |
| <div class="line"><a name="l01460"></a><span class="lineno"> 1460</span>  <span class="keywordtype">int</span>* outdev_id,</div> |
| <div class="line"><a name="l01461"></a><span class="lineno"> 1461</span>  <span class="keywordtype">int</span> num_out,</div> |
| <div class="line"><a name="l01462"></a><span class="lineno"> 1462</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> cpu_malloc,</div> |
| <div class="line"><a name="l01463"></a><span class="lineno"> 1463</span>  <span class="keywordtype">void</span>* cpu_alloc,</div> |
| <div class="line"><a name="l01464"></a><span class="lineno"> 1464</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> gpu_malloc,</div> |
| <div class="line"><a name="l01465"></a><span class="lineno"> 1465</span>  <span class="keywordtype">void</span>* gpu_alloc,</div> |
| <div class="line"><a name="l01466"></a><span class="lineno"> 1466</span>  <span class="keywordtype">void</span>* stream,</div> |
| <div class="line"><a name="l01467"></a><span class="lineno"> 1467</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">mxnet::ext::sparse_malloc_t</a> sparse_malloc,</div> |
| <div class="line"><a name="l01468"></a><span class="lineno"> 1468</span>  <span class="keywordtype">void</span>* sparse_alloc,</div> |
| <div class="line"><a name="l01469"></a><span class="lineno"> 1469</span>  <span class="keywordtype">int</span>* instypes,</div> |
| <div class="line"><a name="l01470"></a><span class="lineno"> 1470</span>  <span class="keywordtype">int</span>* outstypes,</div> |
| <div class="line"><a name="l01471"></a><span class="lineno"> 1471</span>  <span class="keywordtype">void</span>** in_indices,</div> |
| <div class="line"><a name="l01472"></a><span class="lineno"> 1472</span>  <span class="keywordtype">void</span>** out_indices,</div> |
| <div class="line"><a name="l01473"></a><span class="lineno"> 1473</span>  <span class="keywordtype">void</span>** in_indptr,</div> |
| <div class="line"><a name="l01474"></a><span class="lineno"> 1474</span>  <span class="keywordtype">void</span>** out_indptr,</div> |
| <div class="line"><a name="l01475"></a><span class="lineno"> 1475</span>  int64_t* in_indices_shapes,</div> |
| <div class="line"><a name="l01476"></a><span class="lineno"> 1476</span>  int64_t* out_indices_shapes,</div> |
| <div class="line"><a name="l01477"></a><span class="lineno"> 1477</span>  int64_t* in_indptr_shapes,</div> |
| <div class="line"><a name="l01478"></a><span class="lineno"> 1478</span>  int64_t* out_indptr_shapes,</div> |
| <div class="line"><a name="l01479"></a><span class="lineno"> 1479</span>  <span class="keywordtype">void</span>* rng_cpu_states,</div> |
| <div class="line"><a name="l01480"></a><span class="lineno"> 1480</span>  <span class="keywordtype">void</span>* rng_gpu_states);</div> |
| <div class="line"><a name="l01481"></a><span class="lineno"> 1481</span>  </div> |
| <div class="line"><a name="l01483"></a><span class="lineno"> 1483</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a8088989b079b01b821c7b7d0b960dfaa">_partRegSize</a>();</div> |
| <div class="line"><a name="l01484"></a><span class="lineno"> 1484</span>  </div> |
| <div class="line"><a name="l01485"></a><span class="lineno"> 1485</span> <span class="comment">/* returns number of strategies registered for partitioner</span></div> |
| <div class="line"><a name="l01486"></a><span class="lineno"> 1486</span> <span class="comment"> * at specified index */</span></div> |
| <div class="line"><a name="l01487"></a><span class="lineno"> 1487</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#aa8a5a6598b9b674dcd015eab7c25bc38">_partRegGetCount</a>(<span class="keywordtype">int</span> idx, <span class="keyword">const</span> <span class="keywordtype">char</span>** name);</div> |
| <div class="line"><a name="l01488"></a><span class="lineno"> 1488</span>  </div> |
| <div class="line"><a name="l01490"></a><span class="lineno"> 1490</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#ae62af668ca56e5f29a4428681f7d0914">_partRegGet</a>(<span class="keywordtype">int</span> part_idx,</div> |
| <div class="line"><a name="l01491"></a><span class="lineno"> 1491</span>  <span class="keywordtype">int</span> stg_idx,</div> |
| <div class="line"><a name="l01492"></a><span class="lineno"> 1492</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** strategy,</div> |
| <div class="line"><a name="l01493"></a><span class="lineno"> 1493</span>  <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">mxnet::ext::supportedOps_t</a>* supportedOps,</div> |
| <div class="line"><a name="l01494"></a><span class="lineno"> 1494</span>  <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">mxnet::ext::createSelector_t</a>* createSelector,</div> |
| <div class="line"><a name="l01495"></a><span class="lineno"> 1495</span>  <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">mxnet::ext::reviewSubgraph_t</a>* reviewSubgraph,</div> |
| <div class="line"><a name="l01496"></a><span class="lineno"> 1496</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** op_name);</div> |
| <div class="line"><a name="l01497"></a><span class="lineno"> 1497</span>  </div> |
| <div class="line"><a name="l01499"></a><span class="lineno"> 1499</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#ae845d0155a42d87b0714a4ee3e6240f5">_partCallSupportedOps</a>(<a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">mxnet::ext::supportedOps_t</a> supportedOps,</div> |
| <div class="line"><a name="l01500"></a><span class="lineno"> 1500</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01501"></a><span class="lineno"> 1501</span>  <span class="keywordtype">int</span> num_ids,</div> |
| <div class="line"><a name="l01502"></a><span class="lineno"> 1502</span>  <span class="keywordtype">int</span>* ids,</div> |
| <div class="line"><a name="l01503"></a><span class="lineno"> 1503</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01504"></a><span class="lineno"> 1504</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01505"></a><span class="lineno"> 1505</span>  <span class="keywordtype">int</span> num_opts);</div> |
| <div class="line"><a name="l01506"></a><span class="lineno"> 1506</span>  </div> |
| <div class="line"><a name="l01508"></a><span class="lineno"> 1508</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a70e75073813f5a358d365715205ea001">_partCallCreateSelector</a>(<a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">mxnet::ext::createSelector_t</a> createSelector,</div> |
| <div class="line"><a name="l01509"></a><span class="lineno"> 1509</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01510"></a><span class="lineno"> 1510</span>  <span class="keywordtype">void</span>** selector,</div> |
| <div class="line"><a name="l01511"></a><span class="lineno"> 1511</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01512"></a><span class="lineno"> 1512</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01513"></a><span class="lineno"> 1513</span>  <span class="keywordtype">int</span> num_opts);</div> |
| <div class="line"><a name="l01514"></a><span class="lineno"> 1514</span>  </div> |
| <div class="line"><a name="l01516"></a><span class="lineno"> 1516</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#acc5fcd7471b3ee96cf6939b1f2799a61">_partCallSelect</a>(<span class="keywordtype">void</span>* sel_inst, <span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span>* selected);</div> |
| <div class="line"><a name="l01517"></a><span class="lineno"> 1517</span>  </div> |
| <div class="line"><a name="l01519"></a><span class="lineno"> 1519</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#a876d77c3d311c7006f7b23578f85678a">_partCallSelectInput</a>(<span class="keywordtype">void</span>* sel_inst, <span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> input_nodeID, <span class="keywordtype">int</span>* selected);</div> |
| <div class="line"><a name="l01520"></a><span class="lineno"> 1520</span>  </div> |
| <div class="line"><a name="l01522"></a><span class="lineno"> 1522</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#af623e317a08ae2d7822e4e81d4dcb1a4">_partCallSelectOutput</a>(<span class="keywordtype">void</span>* sel_inst, <span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> output_nodeID, <span class="keywordtype">int</span>* selected);</div> |
| <div class="line"><a name="l01523"></a><span class="lineno"> 1523</span>  </div> |
| <div class="line"><a name="l01525"></a><span class="lineno"> 1525</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#a670530b58e8ddc61982798fd788c3edf">_partCallFilter</a>(<span class="keywordtype">void</span>* sel_inst,</div> |
| <div class="line"><a name="l01526"></a><span class="lineno"> 1526</span>  <span class="keywordtype">int</span>* candidates,</div> |
| <div class="line"><a name="l01527"></a><span class="lineno"> 1527</span>  <span class="keywordtype">int</span> num_candidates,</div> |
| <div class="line"><a name="l01528"></a><span class="lineno"> 1528</span>  <span class="keywordtype">int</span>** keep,</div> |
| <div class="line"><a name="l01529"></a><span class="lineno"> 1529</span>  <span class="keywordtype">int</span>* num_keep);</div> |
| <div class="line"><a name="l01530"></a><span class="lineno"> 1530</span>  </div> |
| <div class="line"><a name="l01532"></a><span class="lineno"> 1532</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#a8002368ce25c4f689b48c511b092373b">_partCallReset</a>(<span class="keywordtype">void</span>* sel_inst);</div> |
| <div class="line"><a name="l01533"></a><span class="lineno"> 1533</span>  </div> |
| <div class="line"><a name="l01535"></a><span class="lineno"> 1535</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a8354cc31577d1c0308f40e8639bc3f6f">_partCallReviewSubgraph</a>(<a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">mxnet::ext::reviewSubgraph_t</a> reviewSubgraph,</div> |
| <div class="line"><a name="l01536"></a><span class="lineno"> 1536</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01537"></a><span class="lineno"> 1537</span>  <span class="keywordtype">int</span> subgraph_id,</div> |
| <div class="line"><a name="l01538"></a><span class="lineno"> 1538</span>  <span class="keywordtype">int</span>* accept,</div> |
| <div class="line"><a name="l01539"></a><span class="lineno"> 1539</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01540"></a><span class="lineno"> 1540</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01541"></a><span class="lineno"> 1541</span>  <span class="keywordtype">int</span> num_opts,</div> |
| <div class="line"><a name="l01542"></a><span class="lineno"> 1542</span>  <span class="keywordtype">char</span>*** attr_keys,</div> |
| <div class="line"><a name="l01543"></a><span class="lineno"> 1543</span>  <span class="keywordtype">char</span>*** attr_vals,</div> |
| <div class="line"><a name="l01544"></a><span class="lineno"> 1544</span>  <span class="keywordtype">int</span>* num_attrs,</div> |
| <div class="line"><a name="l01545"></a><span class="lineno"> 1545</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names,</div> |
| <div class="line"><a name="l01546"></a><span class="lineno"> 1546</span>  <span class="keywordtype">int</span> num_args,</div> |
| <div class="line"><a name="l01547"></a><span class="lineno"> 1547</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data,</div> |
| <div class="line"><a name="l01548"></a><span class="lineno"> 1548</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div> |
| <div class="line"><a name="l01549"></a><span class="lineno"> 1549</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims,</div> |
| <div class="line"><a name="l01550"></a><span class="lineno"> 1550</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div> |
| <div class="line"><a name="l01551"></a><span class="lineno"> 1551</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs,</div> |
| <div class="line"><a name="l01552"></a><span class="lineno"> 1552</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div> |
| <div class="line"><a name="l01553"></a><span class="lineno"> 1553</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div> |
| <div class="line"><a name="l01554"></a><span class="lineno"> 1554</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names,</div> |
| <div class="line"><a name="l01555"></a><span class="lineno"> 1555</span>  <span class="keywordtype">int</span> num_aux,</div> |
| <div class="line"><a name="l01556"></a><span class="lineno"> 1556</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data,</div> |
| <div class="line"><a name="l01557"></a><span class="lineno"> 1557</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div> |
| <div class="line"><a name="l01558"></a><span class="lineno"> 1558</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims,</div> |
| <div class="line"><a name="l01559"></a><span class="lineno"> 1559</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div> |
| <div class="line"><a name="l01560"></a><span class="lineno"> 1560</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs,</div> |
| <div class="line"><a name="l01561"></a><span class="lineno"> 1561</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div> |
| <div class="line"><a name="l01562"></a><span class="lineno"> 1562</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id);</div> |
| <div class="line"><a name="l01563"></a><span class="lineno"> 1563</span>  </div> |
| <div class="line"><a name="l01565"></a><span class="lineno"> 1565</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a97a361685f2eb83a744f7fef61d8f26b">_passRegSize</a>();</div> |
| <div class="line"><a name="l01566"></a><span class="lineno"> 1566</span>  </div> |
| <div class="line"><a name="l01568"></a><span class="lineno"> 1568</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#ac47e22b68597148e7989b3e1ec6497e0">_passRegGet</a>(<span class="keywordtype">int</span> pass_idx, <a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">mxnet::ext::graphPass_t</a>* graphPass, <span class="keyword">const</span> <span class="keywordtype">char</span>** pass_name);</div> |
| <div class="line"><a name="l01569"></a><span class="lineno"> 1569</span>  </div> |
| <div class="line"><a name="l01571"></a><span class="lineno"> 1571</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#a37f969a158ff72a111f84a466aca5119">_passCallGraphPass</a>(<a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">mxnet::ext::graphPass_t</a> graphPass,</div> |
| <div class="line"><a name="l01572"></a><span class="lineno"> 1572</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* json,</div> |
| <div class="line"><a name="l01573"></a><span class="lineno"> 1573</span>  <span class="keywordtype">char</span>** out_graph,</div> |
| <div class="line"><a name="l01574"></a><span class="lineno"> 1574</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div> |
| <div class="line"><a name="l01575"></a><span class="lineno"> 1575</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals,</div> |
| <div class="line"><a name="l01576"></a><span class="lineno"> 1576</span>  <span class="keywordtype">int</span> num_opts,</div> |
| <div class="line"><a name="l01577"></a><span class="lineno"> 1577</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* pass_name,</div> |
| <div class="line"><a name="l01578"></a><span class="lineno"> 1578</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names,</div> |
| <div class="line"><a name="l01579"></a><span class="lineno"> 1579</span>  <span class="keywordtype">int</span> num_args,</div> |
| <div class="line"><a name="l01580"></a><span class="lineno"> 1580</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data,</div> |
| <div class="line"><a name="l01581"></a><span class="lineno"> 1581</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div> |
| <div class="line"><a name="l01582"></a><span class="lineno"> 1582</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims,</div> |
| <div class="line"><a name="l01583"></a><span class="lineno"> 1583</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div> |
| <div class="line"><a name="l01584"></a><span class="lineno"> 1584</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs,</div> |
| <div class="line"><a name="l01585"></a><span class="lineno"> 1585</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div> |
| <div class="line"><a name="l01586"></a><span class="lineno"> 1586</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div> |
| <div class="line"><a name="l01587"></a><span class="lineno"> 1587</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names,</div> |
| <div class="line"><a name="l01588"></a><span class="lineno"> 1588</span>  <span class="keywordtype">int</span> num_aux,</div> |
| <div class="line"><a name="l01589"></a><span class="lineno"> 1589</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data,</div> |
| <div class="line"><a name="l01590"></a><span class="lineno"> 1590</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div> |
| <div class="line"><a name="l01591"></a><span class="lineno"> 1591</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims,</div> |
| <div class="line"><a name="l01592"></a><span class="lineno"> 1592</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div> |
| <div class="line"><a name="l01593"></a><span class="lineno"> 1593</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs,</div> |
| <div class="line"><a name="l01594"></a><span class="lineno"> 1594</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div> |
| <div class="line"><a name="l01595"></a><span class="lineno"> 1595</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id,</div> |
| <div class="line"><a name="l01596"></a><span class="lineno"> 1596</span>  <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">mxnet::ext::nd_malloc_t</a> nd_malloc,</div> |
| <div class="line"><a name="l01597"></a><span class="lineno"> 1597</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc);</div> |
| <div class="line"><a name="l01598"></a><span class="lineno"> 1598</span>  </div> |
| <div class="line"><a name="l01606"></a><span class="lineno"> 1606</span> <span class="preprocessor">#if defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__)</span></div> |
| <div class="line"><a name="l01607"></a><span class="lineno"> 1607</span> __declspec(dllexport) <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">mxnet::ext::MXReturnValue</a> __cdecl</div> |
| <div class="line"><a name="l01608"></a><span class="lineno"> 1608</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l01609"></a><span class="lineno"> 1609</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">mxnet::ext::MXReturnValue</a></div> |
| <div class="line"><a name="l01610"></a><span class="lineno"> 1610</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l01611"></a><span class="lineno"> 1611</span>  <a class="code" href="lib__api_8h.html#a3b4abe8a1f483ab51bb4274be2f193ac">initialize</a>(<span class="keywordtype">int</span> version);</div> |
| <div class="line"><a name="l01612"></a><span class="lineno"> 1612</span>  </div> |
| <div class="line"><a name="l01613"></a><span class="lineno"> 1613</span> <a class="code" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a> <a class="code" href="lib__api_8h.html#ab3a2f05b845f641f30f417756dc64fa0">_msgSize</a>();</div> |
| <div class="line"><a name="l01614"></a><span class="lineno"> 1614</span>  </div> |
| <div class="line"><a name="l01616"></a><span class="lineno"> 1616</span> <a class="code" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a> <a class="code" href="lib__api_8h.html#aed3b9c90c89e5a3cf303b51c4ffa1938">_msgGet</a>(<span class="keywordtype">int</span> idx, <span class="keyword">const</span> <span class="keywordtype">char</span>** msg);</div> |
| <div class="line"><a name="l01617"></a><span class="lineno"> 1617</span> } <span class="comment">// extern "C"</span></div> |
| <div class="line"><a name="l01618"></a><span class="lineno"> 1618</span>  </div> |
| <div class="line"><a name="l01619"></a><span class="lineno"> 1619</span> <span class="preprocessor">#endif // MXNET_LIB_API_H_</span></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a96338af40de6ff68da43ad5649dab18d"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a96338af40de6ff68da43ad5649dab18d">mxnet::ext::CustomOp::infer_storage_type</a></div><div class="ttdeci">inferSType_t infer_storage_type</div><div class="ttdef"><b>Definition:</b> lib_api.h:809</div></div> |
| <div class="ttc" id="anamespacemxnet_html"><div class="ttname"><a href="namespacemxnet.html">mxnet</a></div><div class="ttdoc">namespace of mxnet</div><div class="ttdef"><b>Definition:</b> api_registry.h:33</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a462103a47b8f387ee5a31393c0a47c1a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a462103a47b8f387ee5a31393c0a47c1a">mxnet::ext::partCallSelect_t</a></div><div class="ttdeci">void(* partCallSelect_t)(void *sel_inst, int nodeID, int *selected)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1183</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a4e00a65e67fee5acdc7122513642af0b"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a4e00a65e67fee5acdc7122513642af0b">mxnet::ext::Graph::~Graph</a></div><div class="ttdeci">~Graph()</div></div> |
| <div class="ttc" id="alib__api_8h_html_acc5fcd7471b3ee96cf6939b1f2799a61"><div class="ttname"><a href="lib__api_8h.html#acc5fcd7471b3ee96cf6939b1f2799a61">_partCallSelect</a></div><div class="ttdeci">MX_VOID_RET _partCallSelect(void *sel_inst, int nodeID, int *selected)</div><div class="ttdoc">returns status of calling select function from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a2a205d1c793a296d3a312a575e9eab76"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a2a205d1c793a296d3a312a575e9eab76">mxnet::ext::CustomOp::isSGop</a></div><div class="ttdeci">bool isSGop</div><div class="ttdef"><b>Definition:</b> lib_api.h:812</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a8f6f2a2bef8d1c2ca135ed4137673a13"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mxnet::ext::mx_gpu_rand_t</a></div><div class="ttdeci">void * mx_gpu_rand_t</div><div class="ttdef"><b>Definition:</b> lib_api.h:403</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">mxnet::ext::JsonType</a></div><div class="ttdeci">JsonType</div><div class="ttdoc">Json utility to parse serialized subgraph symbol.</div><div class="ttdef"><b>Definition:</b> lib_api.h:525</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_ad306b093c8348887640e51e040c0748f"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#ad306b093c8348887640e51e040c0748f">mxnet::ext::MXTensor::dtype</a></div><div class="ttdeci">MXDType dtype</div><div class="ttdef"><b>Definition:</b> lib_api.h:367</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_ae3918edf21e047e5d381db65e2497008"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#ae3918edf21e047e5d381db65e2497008">mxnet::ext::JsonVal::parse</a></div><div class="ttdeci">static JsonVal parse(const std::string &json)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_a711d78a6b1533b30b0d5bd4af03bec21"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#a711d78a6b1533b30b0d5bd4af03bec21">mxnet::ext::MXSparse::indices</a></div><div class="ttdeci">int64_t * indices</div><div class="ttdef"><b>Definition:</b> lib_api.h:305</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_afa2624f2a3ccf55f7dd53578896b4603"><div class="ttname"><a href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a></div><div class="ttdeci">MXReturnValue(* fcomp_t)(const std::unordered_map< std::string, std::string > &attributes, std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &res)</div><div class="ttdoc">Custom Operator function templates.</div><div class="ttdef"><b>Definition:</b> lib_api.h:747</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a3d1d1a007585a61bc8e53ebaa2e61480"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a3d1d1a007585a61bc8e53ebaa2e61480">mxnet::ext::MXTensor::data_ptr</a></div><div class="ttdeci">void * data_ptr</div><div class="ttdef"><b>Definition:</b> lib_api.h:361</div></div> |
| <div class="ttc" id="astructDLDataType_html"><div class="ttname"><a href="structDLDataType.html">DLDataType</a></div><div class="ttdoc">The data type the tensor can hold.</div><div class="ttdef"><b>Definition:</b> dlpack.h:94</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a48b8906bb9fb74daf9db7c9725dfe549"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a48b8906bb9fb74daf9db7c9725dfe549">mxnet::ext::CustomOp::setBackward</a></div><div class="ttdeci">CustomOp & setBackward(fcomp_t fgrad, const char *ctx)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a6f47c7152edff58ce548870e7c2de0ab"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a6f47c7152edff58ce548870e7c2de0ab">mxnet::ext::passRegSize_t</a></div><div class="ttdeci">int(* passRegSize_t)(void)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1235</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ae5f9488dd6bdbea94a72c42382c2b6c8"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ae5f9488dd6bdbea94a72c42382c2b6c8">mxnet::ext::opCallInferSType_t</a></div><div class="ttdeci">int(* opCallInferSType_t)(inferSType_t inferSType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1037</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573a71b2f851779936d824bf2c235ed94a8d"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a71b2f851779936d824bf2c235ed94a8d">kDLOpenCL</a></div><div class="ttdeci">@ kDLOpenCL</div><div class="ttdoc">OpenCL devices.</div><div class="ttdef"><b>Definition:</b> lib_api.h:112</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a20c7b312f050f3a1cc606e924482a4c7"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a20c7b312f050f3a1cc606e924482a4c7">mxnet::ext::Node::op</a></div><div class="ttdeci">std::string op</div><div class="ttdef"><b>Definition:</b> lib_api.h:597</div></div> |
| <div class="ttc" id="alib__api_8h_html_a70e75073813f5a358d365715205ea001"><div class="ttname"><a href="lib__api_8h.html#a70e75073813f5a358d365715205ea001">_partCallCreateSelector</a></div><div class="ttdeci">MX_INT_RET _partCallCreateSelector(mxnet::ext::createSelector_t createSelector, const char *json, void **selector, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</div><div class="ttdoc">returns status of calling create selector function from library</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1NodeEntry_html"><div class="ttname"><a href="structmxnet_1_1ext_1_1NodeEntry.html">mxnet::ext::NodeEntry</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:578</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_a9f2d4d97f7d2eb3c492c8f8fcfb98100"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#a9f2d4d97f7d2eb3c492c8f8fcfb98100">mxnet::ext::OpResource::alloc_gpu</a></div><div class="ttdeci">void * alloc_gpu(int size) const</div><div class="ttdoc">allocate gpu memory controlled by MXNet</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_ad8758d40952666f93d644f704b686da6"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#ad8758d40952666f93d644f704b686da6">mxnet::ext::CustomStatefulOp::Backward</a></div><div class="ttdeci">virtual MXReturnValue Backward(std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &op_res)</div><div class="ttdef"><b>Definition:</b> lib_api.h:733</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a886ffa05148d30a14f97c07382ae72d0"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a886ffa05148d30a14f97c07382ae72d0">mxnet::ext::JsonVal::JsonVal</a></div><div class="ttdeci">JsonVal()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:610</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html">mxnet::ext::CustomPartitioner</a></div><div class="ttdoc">An abstract class for subgraph property.</div><div class="ttdef"><b>Definition:</b> lib_api.h:866</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_ae2d186fe466cd35bc4b77e038106b80d"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#ae2d186fe466cd35bc4b77e038106b80d">mxnet::ext::MXTensor::shape</a></div><div class="ttdeci">std::vector< int64_t > shape</div><div class="ttdef"><b>Definition:</b> lib_api.h:364</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_aa5ee4f0b935a06fe8d947aee14f10597"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#aa5ee4f0b935a06fe8d947aee14f10597">mxnet::ext::Graph::getAttr</a></div><div class="ttdeci">const JsonVal & getAttr(const std::string &key) const</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a8b389f7b678d499fd8d5ba79f089c95e"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a8b389f7b678d499fd8d5ba79f089c95e">mxnet::ext::getDtypeAt</a></div><div class="ttdeci">std::string getDtypeAt(const std::string &dtype, unsigned index)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPass_html_a7fbdda42ff6b530c2625d435a8691651"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPass.html#a7fbdda42ff6b530c2625d435a8691651">mxnet::ext::CustomPass::setBody</a></div><div class="ttdeci">CustomPass & setBody(graphPass_t fn)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a2065f7e6313a774f9b1881f3b038d13d"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a2065f7e6313a774f9b1881f3b038d13d">mxnet::ext::CustomOp::setIsSubgraphOp</a></div><div class="ttdeci">CustomOp & setIsSubgraphOp()</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_aff50c6302fa056352a7838a7c0fba79c"><div class="ttname"><a href="namespacemxnet_1_1ext.html#aff50c6302fa056352a7838a7c0fba79c">mxnet::ext::opCallFree_t</a></div><div class="ttdeci">int(* opCallFree_t)(void *ptr)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1002</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a18a96fba494d035a7bc60beb30a07fae"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">mxnet::ext::reviewSubgraph_t</a></div><div class="ttdeci">MXReturnValue(* reviewSubgraph_t)(const mxnet::ext::Graph *subgraph, int subgraph_id, bool *accept, const std::unordered_map< std::string, std::string > &options, std::unordered_map< std::string, std::string > *attrs)</div><div class="ttdef"><b>Definition:</b> lib_api.h:856</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXContext_html_a5ff62e51ebb2a5c898281b73a0c45c7a"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXContext.html#a5ff62e51ebb2a5c898281b73a0c45c7a">mxnet::ext::MXContext::CPU</a></div><div class="ttdeci">static MXContext CPU()</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html">mxnet::ext::MXTensor</a></div><div class="ttdoc">Tensor data structure used by custom operator.</div><div class="ttdef"><b>Definition:</b> lib_api.h:325</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_ae25800af6998af47fe82a31474314bba"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#ae25800af6998af47fe82a31474314bba">mxnet::ext::CustomOp::mutate_inputs</a></div><div class="ttdeci">mutateInputs_t mutate_inputs</div><div class="ttdef"><b>Definition:</b> lib_api.h:811</div></div> |
| <div class="ttc" id="alib__api_8h_html_af358af16fb117cd0675234835a4e7c08"><div class="ttname"><a href="lib__api_8h.html#af358af16fb117cd0675234835a4e7c08">_opCallCreateOpState</a></div><div class="ttdeci">MX_INT_RET _opCallCreateOpState(mxnet::ext::createOpState_t create_op, const char *const *keys, const char *const *vals, int num, const char *dev_type, int dev_id, unsigned int **inshapes, int *indims, int num_in, const int *intypes, void **state_op)</div><div class="ttdoc">returns status of calling createStatefulOp function for operator from library</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_aa90461a780707c2a867589aa16a09a14"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#aa90461a780707c2a867589aa16a09a14">mxnet::ext::MXSparse::data</a></div><div class="ttdeci">void * data</div><div class="ttdef"><b>Definition:</b> lib_api.h:298</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_acbae2c4272b5ac64651bffbf8f000dd8"><div class="ttname"><a href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8">mxnet::ext::mx_cpu_rand_t</a></div><div class="ttdeci">std::mt19937 mx_cpu_rand_t</div><div class="ttdef"><b>Definition:</b> lib_api.h:405</div></div> |
| <div class="ttc" id="alib__api_8h_html_aa8a5a6598b9b674dcd015eab7c25bc38"><div class="ttname"><a href="lib__api_8h.html#aa8a5a6598b9b674dcd015eab7c25bc38">_partRegGetCount</a></div><div class="ttdeci">MX_INT_RET _partRegGetCount(int idx, const char **name)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_aa5bd268a502dce80ffce037252fa0443"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#aa5bd268a502dce80ffce037252fa0443">mxnet::ext::Graph::fromString</a></div><div class="ttdeci">static Graph * fromString(const std::string &json)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPass_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPass.html">mxnet::ext::CustomPass</a></div><div class="ttdoc">An abstract class for graph passes.</div><div class="ttdef"><b>Definition:</b> lib_api.h:834</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de">mxnet::ext::kCSRStorage</a></div><div class="ttdeci">@ kCSRStorage</div><div class="ttdef"><b>Definition:</b> lib_api.h:269</div></div> |
| <div class="ttc" id="alib__api_8h_html_a5b0657e0605ca60b79241a4324139f63"><div class="ttname"><a href="lib__api_8h.html#a5b0657e0605ca60b79241a4324139f63">_opRegGet</a></div><div class="ttdeci">MX_VOID_RET _opRegGet(int idx, const char **name, int *isSGop, const char ***forward_ctx, mxnet::ext::fcomp_t **forward_fp, int *forward_count, const char ***backward_ctx, mxnet::ext::fcomp_t **backward_fp, int *backward_count, const char ***create_op_ctx, mxnet::ext::createOpState_t **create_op_fp, int *create_op_count, mxnet::ext::parseAttrs_t *parse, mxnet::ext::inferType_t *type, mxnet::ext::inferSType_t *stype, mxnet::ext::inferShape_t *shape, mxnet::ext::mutateInputs_t *mutate)</div><div class="ttdoc">returns operator registration at specified index</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a8364c3b8ae3800c775a23a8e90984c82"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a8364c3b8ae3800c775a23a8e90984c82">mxnet::ext::Node::inputs</a></div><div class="ttdeci">std::vector< NodeEntry > inputs</div><div class="ttdef"><b>Definition:</b> lib_api.h:600</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_aefb1c0f85ebec11dd8e4f8720392ce4a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#aefb1c0f85ebec11dd8e4f8720392ce4a">mxnet::ext::partCallReviewSubgraph_t</a></div><div class="ttdeci">int(* partCallReviewSubgraph_t)(reviewSubgraph_t reviewSubgraph, const char *json, int subgraph_id, int *accept, const char *const *opt_keys, const char *const *opt_vals, int num_opts, char ***attr_keys, char ***attr_vals, int *num_attrs, const char *const *arg_names, int num_args, void *const *arg_data, const int64_t *const *arg_shapes, const int *arg_dims, const int *arg_types, const size_t *arg_IDs, const char *const *arg_dev_type, const int *arg_dev_id, const char *const *aux_names, int num_aux, void *const *aux_data, const int64_t *const *aux_shapes, const int *aux_dims, const int *aux_types, const size_t *aux_IDs, const char *const *aux_dev_type, const int *aux_dev_id)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1205</div></div> |
| <div class="ttc" id="alib__api_8h_html_a948af234ca1c3b0c8b78e8db2f32c310"><div class="ttname"><a href="lib__api_8h.html#a948af234ca1c3b0c8b78e8db2f32c310">_opCallFCompute</a></div><div class="ttdeci">MX_INT_RET _opCallFCompute(mxnet::ext::fcomp_t fcomp, const char *const *keys, const char *const *vals, int num, const int64_t **inshapes, int *indims, void **indata, int *intypes, size_t *inIDs, const char **indev_type, int *indev_id, int num_in, const int64_t **outshapes, int *outdims, void **outdata, int *outtypes, size_t *outIDs, const char **outdev_type, int *outdev_id, int num_out, mxnet::ext::xpu_malloc_t cpu_malloc, void *cpu_alloc, mxnet::ext::xpu_malloc_t gpu_malloc, void *gpu_alloc, void *cuda_stream, mxnet::ext::sparse_malloc_t sparse_malloc, void *sparse_alloc, int *instypes, int *outstypes, void **in_indices, void **out_indices, void **in_indptr, void **out_indptr, int64_t *in_indices_shapes, int64_t *out_indices_shapes, int64_t *in_indptr_shapes, int64_t *out_indptr_shapes, void *rng_cpu_states, void *rng_gpu_states)</div><div class="ttdoc">returns status of calling Forward/Backward function for operator from library</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a6d159349f953d943811338e60b331869"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a></div><div class="ttdeci">MXReturnValue(* inferType_t)(const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *in_types, std::vector< int > *out_types)</div><div class="ttdef"><b>Definition:</b> lib_api.h:755</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOpSelector_html_a42727340ba85aa1aaf8493e0322978e3"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOpSelector.html#a42727340ba85aa1aaf8493e0322978e3">mxnet::ext::CustomOpSelector::SelectOutput</a></div><div class="ttdeci">virtual bool SelectOutput(int nodeID, int output_nodeID)=0</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOpSelector_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOpSelector.html">mxnet::ext::CustomOpSelector</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:677</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPass_html_a38420e01eb27a91bc7b11da785b7a652"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPass.html#a38420e01eb27a91bc7b11da785b7a652">mxnet::ext::CustomPass::CustomPass</a></div><div class="ttdeci">CustomPass()</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_a3a02a02338f173c6735ba5b8b1eb1b5c"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#a3a02a02338f173c6735ba5b8b1eb1b5c">mxnet::ext::MXSparse::indptr_len</a></div><div class="ttdeci">int64_t indptr_len</div><div class="ttdef"><b>Definition:</b> lib_api.h:311</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573adeeb5e6f50f4309676e49f7a12512171"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573adeeb5e6f50f4309676e49f7a12512171">kDLROCM</a></div><div class="ttdeci">@ kDLROCM</div><div class="ttdoc">ROCm GPUs for AMD GPUs.</div><div class="ttdef"><b>Definition:</b> lib_api.h:120</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a497cf9a31d84c5ab6e7d41f5cf4789ab"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">mxnet::ext::supportedOps_t</a></div><div class="ttdeci">MXReturnValue(* supportedOps_t)(const mxnet::ext::Graph *graph, std::vector< int > *ids, const std::unordered_map< std::string, std::string > &options)</div><div class="ttdoc">Custom Subgraph Create function template.</div><div class="ttdef"><b>Definition:</b> lib_api.h:848</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_aac12981007cdb017f652d1242cf9ab05"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#aac12981007cdb017f652d1242cf9ab05">mxnet::ext::Node::_setPassResource</a></div><div class="ttdeci">void _setPassResource(PassResource *res_)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ada6444201d1204384d1287d91a4c3fc7"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ada6444201d1204384d1287d91a4c3fc7">mxnet::ext::opCallFStatefulComp_t</a></div><div class="ttdeci">int(* opCallFStatefulComp_t)(int is_forward, void *state_op, const int64_t **inshapes, int *indims, void **indata, int *intypes, size_t *inIDs, const char **indev_type, int *indev_id, int num_in, const int64_t **outshapes, int *outdims, void **outdata, int *outtypes, size_t *outIDs, const char **outdev_type, int *outdev_id, int num_out, xpu_malloc_t cpu_malloc, void *cpu_alloc, xpu_malloc_t gpu_malloc, void *gpu_alloc, void *stream, sparse_malloc_t sparse_malloc, void *sparse_alloc, int *instypes, int *outstypes, void **in_indices, void **out_indices, void **in_indptr, void **out_indptr, int64_t *in_indices_shapes, int64_t *out_indices_shapes, int64_t *in_indptr_shapes, int64_t *out_indptr_shapes, void *rng_cpu_states, void *rng_gpu_states)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1112</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306">mxnet::ext::MX_SUCCESS</a></div><div class="ttdeci">@ MX_SUCCESS</div><div class="ttdef"><b>Definition:</b> lib_api.h:292</div></div> |
| <div class="ttc" id="astructDLContext_html"><div class="ttname"><a href="structDLContext.html">DLContext</a></div><div class="ttdoc">A Device context for Tensor and operator.</div><div class="ttdef"><b>Definition:</b> dlpack.h:69</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a963277b3eebf164f5103a44ae9acdbfc"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a></div><div class="ttdeci">MXReturnValue(* parseAttrs_t)(const std::unordered_map< std::string, std::string > &attributes, int *num_inputs, int *num_outputs)</div><div class="ttdef"><b>Definition:</b> lib_api.h:751</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ad62142c24acef23ecd62e7fd53000349"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ad62142c24acef23ecd62e7fd53000349">mxnet::ext::partCallCreateSelector_t</a></div><div class="ttdeci">int(* partCallCreateSelector_t)(createSelector_t createSelector, const char *json, void **selector, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1175</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1PassResource_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1PassResource.html">mxnet::ext::PassResource</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:413</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a7ad34e214897e21a5c10bc57ddf1b51a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a7ad34e214897e21a5c10bc57ddf1b51a">mxnet::ext::partCallFilter_t</a></div><div class="ttdeci">void(* partCallFilter_t)(void *sel_inst, int *candidates, int num_candidates, int **keep, int *num_keep)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1195</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a17744ecbf554a996fe84dc5553564633"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a17744ecbf554a996fe84dc5553564633">mxnet::ext::Node::alloc_aux</a></div><div class="ttdeci">void alloc_aux(const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_aebb5b6454e5e75b03cafa3ccebf6f752"><div class="ttname"><a href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mxnet::ext::mx_stream_t</a></div><div class="ttdeci">void * mx_stream_t</div><div class="ttdoc">GPU stream pointer, is void* when not compiled with CUDA.</div><div class="ttdef"><b>Definition:</b> lib_api.h:402</div></div> |
| <div class="ttc" id="alib__api_8h_html_ae62af668ca56e5f29a4428681f7d0914"><div class="ttname"><a href="lib__api_8h.html#ae62af668ca56e5f29a4428681f7d0914">_partRegGet</a></div><div class="ttdeci">MX_VOID_RET _partRegGet(int part_idx, int stg_idx, const char **strategy, mxnet::ext::supportedOps_t *supportedOps, mxnet::ext::createSelector_t *createSelector, mxnet::ext::reviewSubgraph_t *reviewSubgraph, const char **op_name)</div><div class="ttdoc">returns partitioner registration at specified index</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html">mxnet::ext::JsonVal</a></div><div class="ttdoc">definition of JSON objects</div><div class="ttdef"><b>Definition:</b> lib_api.h:528</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2">mxnet::ext::kUint8</a></div><div class="ttdeci">@ kUint8</div><div class="ttdef"><b>Definition:</b> lib_api.h:253</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPass_html_a6dd29f51d329906280090109a634bb6a"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPass.html#a6dd29f51d329906280090109a634bb6a">mxnet::ext::CustomPass::pass</a></div><div class="ttdeci">graphPass_t pass</div><div class="ttdoc">pass function</div><div class="ttdef"><b>Definition:</b> lib_api.h:844</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_adf4bd71f1334318c5258e9421709b03c"><div class="ttname"><a href="namespacemxnet_1_1ext.html#adf4bd71f1334318c5258e9421709b03c">mxnet::ext::opCallCreateOpState_t</a></div><div class="ttdeci">int(* opCallCreateOpState_t)(createOpState_t create_op, const char *const *keys, const char *const *vals, int num, const char *dev_type, int dev_id, unsigned int **inshapes, int *indims, int num_in, const int *intypes, void **state_op)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1096</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_a738c782852852154eb6fc61b8e9b34b0"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#a738c782852852154eb6fc61b8e9b34b0">mxnet::ext::MXSparse::indptr</a></div><div class="ttdeci">int64_t * indptr</div><div class="ttdef"><b>Definition:</b> lib_api.h:310</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a9d1096da2919bfc1861d70c3f2079339"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a9d1096da2919bfc1861d70c3f2079339">mxnet::ext::JsonVal::operator<</a></div><div class="ttdeci">bool operator<(const JsonVal &o) const</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOpWrapper_html_ace82ec005741e89957ba7fdfe433f654"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#ace82ec005741e89957ba7fdfe433f654">mxnet::ext::CustomStatefulOpWrapper::get_instance</a></div><div class="ttdeci">CustomStatefulOp * get_instance()</div><div class="ttdef"><b>Definition:</b> lib_api.h:1287</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901">mxnet::ext::LIST</a></div><div class="ttdeci">@ LIST</div><div class="ttdef"><b>Definition:</b> lib_api.h:525</div></div> |
| <div class="ttc" id="alib__api_8h_html_adb3450125394a91102043bcc28fe8e39"><div class="ttname"><a href="lib__api_8h.html#adb3450125394a91102043bcc28fe8e39">_opCallInferType</a></div><div class="ttdeci">MX_INT_RET _opCallInferType(mxnet::ext::inferType_t inferType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</div><div class="ttdoc">returns status of calling inferType function for operator from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_af827c73b729d3e8cb8672b10107312fd"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#af827c73b729d3e8cb8672b10107312fd">mxnet::ext::Graph::_dfs_util</a></div><div class="ttdeci">void _dfs_util(Node *n, std::unordered_set< Node * > *to_visit, std::function< void(Node *)> handler) const</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_a712e78a6c50beca023bc20064688bd1f"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#a712e78a6c50beca023bc20064688bd1f">mxnet::ext::OpResource::get_cuda_stream</a></div><div class="ttdeci">mx_stream_t get_cuda_stream() const</div><div class="ttdoc">return the cuda stream object with correct type</div><div class="ttdef"><b>Definition:</b> lib_api.h:461</div></div> |
| <div class="ttc" id="adlpack_8h_html_a49db572e18d84c47738fc746568fb573"><div class="ttname"><a href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573">DLDeviceType</a></div><div class="ttdeci">DLDeviceType</div><div class="ttdoc">The device type in DLContext.</div><div class="ttdef"><b>Definition:</b> dlpack.h:38</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_af5d2babd1cb84d40ad07585cac9c5970"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#af5d2babd1cb84d40ad07585cac9c5970">mxnet::ext::JsonVal::list</a></div><div class="ttdeci">std::vector< JsonVal > list</div><div class="ttdef"><b>Definition:</b> lib_api.h:567</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ae77b4c4a8ed7275fd755b458533c7a70"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70">mxnet::ext::opCallDestroyOpState_t</a></div><div class="ttdeci">int(* opCallDestroyOpState_t)(void *state_op)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1109</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_a96fa3699ce4d0324480cc0185d49cf30"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a96fa3699ce4d0324480cc0185d49cf30">mxnet::ext::CustomStatefulOp::Forward</a></div><div class="ttdeci">virtual MXReturnValue Forward(std::vector< MXTensor > *inputs, std::vector< MXTensor > *outputs, const OpResource &op_res)=0</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1PassResource_html_af7ee64a64fbfa3d1f068ebd4976911d5"><div class="ttname"><a href="classmxnet_1_1ext_1_1PassResource.html#af7ee64a64fbfa3d1f068ebd4976911d5">mxnet::ext::PassResource::PassResource</a></div><div class="ttdeci">PassResource(std::unordered_map< std::string, MXTensor > *new_args, std::unordered_map< std::string, MXTensor > *new_aux, nd_malloc_t nd_malloc, const void *nd_alloc)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a8a4d41b35b45f3644497d9a8f717b6a3"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a8a4d41b35b45f3644497d9a8f717b6a3">mxnet::ext::partRegSize_t</a></div><div class="ttdeci">int(* partRegSize_t)(void)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1151</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a55c4fbc10f64e0a6f02bae3a020b4e38"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a55c4fbc10f64e0a6f02bae3a020b4e38">mxnet::ext::CustomOp::backward_fp</a></div><div class="ttdeci">std::vector< fcomp_t > backward_fp</div><div class="ttdef"><b>Definition:</b> lib_api.h:816</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a0e92d22b68b63878afb266cdd2091c39"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a0e92d22b68b63878afb266cdd2091c39">mxnet::ext::Graph::print</a></div><div class="ttdeci">void print(int indent=0) const</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXContext_html_a40b89dc89df67507bc0d7f94b700f5a2"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXContext.html#a40b89dc89df67507bc0d7f94b700f5a2">mxnet::ext::MXContext::dev_id</a></div><div class="ttdeci">int dev_id</div><div class="ttdef"><b>Definition:</b> lib_api.h:287</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a6fb7af175432173dec6120674a1f07a4"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a6fb7af175432173dec6120674a1f07a4">mxnet::ext::opRegSize_t</a></div><div class="ttdeci">int(* opRegSize_t)(void)</div><div class="ttdef"><b>Definition:</b> lib_api.h:980</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a1907581693bae41486523f199fecacf5"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a1907581693bae41486523f199fecacf5">mxnet::ext::Graph::toString</a></div><div class="ttdeci">std::string toString() const</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a83e21d3dbcb68574f93c95db0a83ce31"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a83e21d3dbcb68574f93c95db0a83ce31">mxnet::ext::passRegGet_t</a></div><div class="ttdeci">void(* passRegGet_t)(int pass_idx, graphPass_t *graphPass, const char **pass_name)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1238</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a4128a869e7ed8d1ed7c04037d752300e"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">mxnet::ext::sparse_malloc_t</a></div><div class="ttdeci">void(* sparse_malloc_t)(void *, int, int, int, void **, int64_t **, int64_t **)</div><div class="ttdoc">sparse alloc function to allocate memory inside Forward/Backward functions</div><div class="ttdef"><b>Definition:</b> lib_api.h:386</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a07eb2f161653eed3b4332355cb050137"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a07eb2f161653eed3b4332355cb050137">mxnet::ext::CustomOp::infer_type</a></div><div class="ttdeci">inferType_t infer_type</div><div class="ttdef"><b>Definition:</b> lib_api.h:808</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_aa345494550891b4c7481b4123eef9586"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#aa345494550891b4c7481b4123eef9586">mxnet::ext::CustomPartitioner::setReviewSubgraph</a></div><div class="ttdeci">CustomPartitioner & setReviewSubgraph(const char *prop_name, reviewSubgraph_t fn)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOpSelector_html_aa4b8875b76531510de9bf672034fd96d"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOpSelector.html#aa4b8875b76531510de9bf672034fd96d">mxnet::ext::CustomOpSelector::SelectInput</a></div><div class="ttdeci">virtual bool SelectInput(int nodeID, int input_nodeID)=0</div></div> |
| <div class="ttc" id="alib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a></div><div class="ttdeci">@ kDLUInt</div><div class="ttdef"><b>Definition:</b> lib_api.h:144</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a645677b0c72cbb9849069f61c3f45917"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">mxnet::ext::nd_malloc_t</a></div><div class="ttdeci">void(* nd_malloc_t)(const void *_ndarray_alloc, const int64_t *shapes, int num_shapes, const char *dev_str, int dev_id, int dtype, const char *name, int isArg, void **data)</div><div class="ttdoc">resource malloc function to allocate ndarrays for graph passes</div><div class="ttdef"><b>Definition:</b> lib_api.h:388</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a8934bedd9d9643ee1e8c69bcae65c224"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a8934bedd9d9643ee1e8c69bcae65c224">mxnet::ext::MXTensor::isSame</a></div><div class="ttdeci">bool isSame(const MXTensor &oth) const</div><div class="ttdoc">helper function to compare two MXTensors</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_aca8a01d001adef1481fa5506b76d117b"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#aca8a01d001adef1481fa5506b76d117b">mxnet::ext::OpResource::alloc_cpu</a></div><div class="ttdeci">void * alloc_cpu(int size) const</div><div class="ttdoc">allocate cpu memory controlled by MXNet</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a08d9994f1542d34fce1eeabda3a8ad85"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a08d9994f1542d34fce1eeabda3a8ad85">mxnet::ext::opCallInferType_t</a></div><div class="ttdeci">int(* opCallInferType_t)(inferType_t inferType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1027</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a61b8b8d52e15d830875b643851a6af9c"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a61b8b8d52e15d830875b643851a6af9c">mxnet::ext::CustomOp::setCreateOpState</a></div><div class="ttdeci">CustomOp & setCreateOpState(createOpState_t func, const char *ctx)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a3489d59197ad9612dbdc23f963aa8d02"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a3489d59197ad9612dbdc23f963aa8d02">mxnet::ext::Node::tensor</a></div><div class="ttdeci">MXTensor * tensor</div><div class="ttdef"><b>Definition:</b> lib_api.h:599</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a35eadaeb5efe27f440026294f62bfc42"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a35eadaeb5efe27f440026294f62bfc42">mxnet::ext::CustomPartitioner::setSupportedOps</a></div><div class="ttdeci">CustomPartitioner & setSupportedOps(const char *prop_name, supportedOps_t fn)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a7997e82fd30ac0817c0555fc0969e87f"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a7997e82fd30ac0817c0555fc0969e87f">mxnet::ext::Graph::attrs</a></div><div class="ttdeci">std::map< std::string, JsonVal > attrs</div><div class="ttdef"><b>Definition:</b> lib_api.h:667</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a0b6d87bb4c06678d0be1dd3ad5916a1b"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a0b6d87bb4c06678d0be1dd3ad5916a1b">mxnet::ext::partRegGetCount_t</a></div><div class="ttdeci">int(* partRegGetCount_t)(int idx, const char **name)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1154</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573a36ced4bf0304d68b5c633ddba5633e66"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a36ced4bf0304d68b5c633ddba5633e66">kDLMetal</a></div><div class="ttdeci">@ kDLMetal</div><div class="ttdoc">Metal for Apple GPU.</div><div class="ttdef"><b>Definition:</b> lib_api.h:116</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_acb5a3d13564e147807e0006c6f75eada"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#acb5a3d13564e147807e0006c6f75eada">mxnet::ext::CustomOp::forward_ctx_cstr</a></div><div class="ttdeci">std::vector< const char * > forward_ctx_cstr</div><div class="ttdoc">vector repr of ctx map to be easily loaded from c_api</div><div class="ttdef"><b>Definition:</b> lib_api.h:815</div></div> |
| <div class="ttc" id="alib__api_8h_html_a8354cc31577d1c0308f40e8639bc3f6f"><div class="ttname"><a href="lib__api_8h.html#a8354cc31577d1c0308f40e8639bc3f6f">_partCallReviewSubgraph</a></div><div class="ttdeci">MX_INT_RET _partCallReviewSubgraph(mxnet::ext::reviewSubgraph_t reviewSubgraph, const char *json, int subgraph_id, int *accept, const char *const *opt_keys, const char *const *opt_vals, int num_opts, char ***attr_keys, char ***attr_vals, int *num_attrs, const char *const *arg_names, int num_args, void *const *arg_data, const int64_t *const *arg_shapes, const int *arg_dims, const int *arg_types, const size_t *arg_IDs, const char *const *arg_dev_type, const int *arg_dev_id, const char *const *aux_names, int num_aux, void *const *aux_data, const int64_t *const *aux_shapes, const int *aux_dims, const int *aux_types, const size_t *aux_IDs, const char *const *aux_dev_type, const int *aux_dev_id)</div><div class="ttdoc">returns status of calling review subgraph function from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_ab3865614538d9fa49295cd24cb6284c3"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#ab3865614538d9fa49295cd24cb6284c3">mxnet::ext::CustomPartitioner::getReviewSubgraph</a></div><div class="ttdeci">reviewSubgraph_t getReviewSubgraph(int stg_id)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9f"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">mxnet::ext::MXDType</a></div><div class="ttdeci">MXDType</div><div class="ttdoc">Tensor data type, consistent with mshadow data type.</div><div class="ttdef"><b>Definition:</b> lib_api.h:249</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_adacc6e8d18cc2428ace58c96111e6f39"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#adacc6e8d18cc2428ace58c96111e6f39">mxnet::ext::CustomStatefulOp::CustomStatefulOp</a></div><div class="ttdeci">CustomStatefulOp()</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a9089010cd352a7e5a36fc3ade38dd221"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a9089010cd352a7e5a36fc3ade38dd221">mxnet::ext::initialize_t</a></div><div class="ttdeci">int(* initialize_t)(int version)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1270</div></div> |
| <div class="ttc" id="alib__api_8h_html_a0cc29cb55b56ebd7e288b3f5946b5bce"><div class="ttname"><a href="lib__api_8h.html#a0cc29cb55b56ebd7e288b3f5946b5bce">_opCallDestroyOpState</a></div><div class="ttdeci">MX_VOID_RET _opCallDestroyOpState(void *state_op)</div><div class="ttdoc">returns status of deleting StatefulOp instance for operator from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a4158fd61578679630e5eba3a6d889ca3"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a4158fd61578679630e5eba3a6d889ca3">mxnet::ext::Graph::_setPassResource</a></div><div class="ttdeci">void _setPassResource(PassResource *res_)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a1b620d0a37e0104eb60db211930ed5a5"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a1b620d0a37e0104eb60db211930ed5a5">mxnet::ext::CustomPartitioner::name</a></div><div class="ttdeci">const char * name</div><div class="ttdoc">partitioner name</div><div class="ttdef"><b>Definition:</b> lib_api.h:887</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_abe5578d0e77b2548b8d384ba6df66484"><div class="ttname"><a href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a></div><div class="ttdeci">MXReturnValue(* mutateInputs_t)(const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *input_indices)</div><div class="ttdef"><b>Definition:</b> lib_api.h:766</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670">mxnet::ext::MX_FAIL</a></div><div class="ttdeci">@ MX_FAIL</div><div class="ttdef"><b>Definition:</b> lib_api.h:291</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1NodeEntry_html_ac3d191e6775d212446933fc94ca21624"><div class="ttname"><a href="structmxnet_1_1ext_1_1NodeEntry.html#ac3d191e6775d212446933fc94ca21624">mxnet::ext::NodeEntry::entry</a></div><div class="ttdeci">int entry</div><div class="ttdef"><b>Definition:</b> lib_api.h:580</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_aaf2854a4075aa8cd6ef138112f1676b6"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#aaf2854a4075aa8cd6ef138112f1676b6">mxnet::ext::JsonVal::parse_map</a></div><div class="ttdeci">static JsonVal parse_map(const std::string &json, unsigned int *idx)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Registry_html_aac245796efea7dbb18a86ff2be034453"><div class="ttname"><a href="classmxnet_1_1ext_1_1Registry.html#aac245796efea7dbb18a86ff2be034453">mxnet::ext::Registry::get</a></div><div class="ttdeci">T & get(int idx)</div><div class="ttdef"><b>Definition:</b> lib_api.h:924</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_aead603e2824c32eeb027bee2d2b4a640"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#aead603e2824c32eeb027bee2d2b4a640">mxnet::ext::MXTensor::setDLTensor</a></div><div class="ttdeci">void setDLTensor()</div><div class="ttdoc">populate DLTensor fields</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a99f30444690e140720d54d69cf42ce0e"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a99f30444690e140720d54d69cf42ce0e">mxnet::ext::CustomPartitioner::CustomPartitioner</a></div><div class="ttdeci">CustomPartitioner()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOpSelector_html_a76533a016df5e8c3e39a722f2bf48f13"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOpSelector.html#a76533a016df5e8c3e39a722f2bf48f13">mxnet::ext::CustomOpSelector::Reset</a></div><div class="ttdeci">virtual void Reset()</div><div class="ttdef"><b>Definition:</b> lib_api.h:706</div></div> |
| <div class="ttc" id="alib__api_8h_html_ae845d0155a42d87b0714a4ee3e6240f5"><div class="ttname"><a href="lib__api_8h.html#ae845d0155a42d87b0714a4ee3e6240f5">_partCallSupportedOps</a></div><div class="ttdeci">MX_INT_RET _partCallSupportedOps(mxnet::ext::supportedOps_t supportedOps, const char *json, int num_ids, int *ids, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</div><div class="ttdoc">returns status of calling supported ops function from library</div></div> |
| <div class="ttc" id="alib__api_8h_html_ac88d6741ca655523db73fc18a8e06828"><div class="ttname"><a href="lib__api_8h.html#ac88d6741ca655523db73fc18a8e06828">_opCallParseAttrs</a></div><div class="ttdeci">MX_INT_RET _opCallParseAttrs(mxnet::ext::parseAttrs_t parseAttrs, const char *const *keys, const char *const *vals, int num, int *num_in, int *num_out)</div><div class="ttdoc">returns status of calling parse attributes function for operator from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_aeaad9d379d75938aa12b908c1a7a9af6"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#aeaad9d379d75938aa12b908c1a7a9af6">mxnet::ext::CustomPartitioner::op_names</a></div><div class="ttdeci">std::vector< const char * > op_names</div><div class="ttdoc">subgraph operator name</div><div class="ttdef"><b>Definition:</b> lib_api.h:894</div></div> |
| <div class="ttc" id="alib__api_8h_html_a481a7e516c02ab43e5b0e0016cb7997f"><div class="ttname"><a href="lib__api_8h.html#a481a7e516c02ab43e5b0e0016cb7997f">MX_ERROR_MSG</a></div><div class="ttdeci">#define MX_ERROR_MSG</div><div class="ttdef"><b>Definition:</b> lib_api.h:244</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPass_html_aec569fc21f800cb803d47508cd495fcb"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPass.html#aec569fc21f800cb803d47508cd495fcb">mxnet::ext::CustomPass::name</a></div><div class="ttdeci">const char * name</div><div class="ttdoc">pass name</div><div class="ttdef"><b>Definition:</b> lib_api.h:842</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a3e1851c411ee0112de6ee3f1a90d2151"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a3e1851c411ee0112de6ee3f1a90d2151">mxnet::ext::Node::attrs</a></div><div class="ttdeci">std::unordered_map< std::string, std::string > attrs</div><div class="ttdef"><b>Definition:</b> lib_api.h:603</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103">mxnet::ext::kFloat16</a></div><div class="ttdeci">@ kFloat16</div><div class="ttdef"><b>Definition:</b> lib_api.h:252</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_aa1de9b3397a13c3633350307593837f3"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#aa1de9b3397a13c3633350307593837f3">mxnet::ext::CustomOp::name</a></div><div class="ttdeci">const char * name</div><div class="ttdoc">operator name</div><div class="ttdef"><b>Definition:</b> lib_api.h:804</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html">mxnet::ext::MXSparse</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:296</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXContext_html"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXContext.html">mxnet::ext::MXContext</a></div><div class="ttdoc">Context info passing from MXNet OpContext dev_type is string repr of supported context,...</div><div class="ttdef"><b>Definition:</b> lib_api.h:277</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Registry_html_a34d70208a11534185f4288284430acb8"><div class="ttname"><a href="classmxnet_1_1ext_1_1Registry.html#a34d70208a11534185f4288284430acb8">mxnet::ext::Registry::get</a></div><div class="ttdeci">static Registry * get() PRIVATE_SYMBOL</div><div class="ttdoc">get singleton pointer to class</div><div class="ttdef"><b>Definition:</b> lib_api.h:908</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a7d51476c0d999dc545aaa4e729d2f841"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a7d51476c0d999dc545aaa4e729d2f841">mxnet::ext::MXTensor::verID</a></div><div class="ttdeci">size_t verID</div><div class="ttdef"><b>Definition:</b> lib_api.h:370</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9">mxnet::ext::STR</a></div><div class="ttdeci">@ STR</div><div class="ttdef"><b>Definition:</b> lib_api.h:525</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9">mxnet::ext::ERR</a></div><div class="ttdeci">@ ERR</div><div class="ttdef"><b>Definition:</b> lib_api.h:525</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a588e1f76a13f07ea1a96db28c0ddb15a"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a588e1f76a13f07ea1a96db28c0ddb15a">mxnet::ext::CustomPartitioner::getSupportedOps</a></div><div class="ttdeci">supportedOps_t getSupportedOps(int stg_id)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a7345d89ba6e4c628641cf36c4123145d"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a7345d89ba6e4c628641cf36c4123145d">mxnet::ext::JsonVal::toString</a></div><div class="ttdeci">std::string toString() const</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573af063bf37ea24c1839d3a78703c2468e7"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573af063bf37ea24c1839d3a78703c2468e7">kDLVPI</a></div><div class="ttdeci">@ kDLVPI</div><div class="ttdoc">Verilog simulator buffer.</div><div class="ttdef"><b>Definition:</b> lib_api.h:118</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_aae3712b54e6f015df3b75b2827da1adc"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#aae3712b54e6f015df3b75b2827da1adc">mxnet::ext::JsonVal::map</a></div><div class="ttdeci">std::map< JsonVal, JsonVal > map</div><div class="ttdef"><b>Definition:</b> lib_api.h:568</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a764cc32d43d66be2095065b9525e47c2"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a764cc32d43d66be2095065b9525e47c2">mxnet::ext::MXTensor::setTensor</a></div><div class="ttdeci">void setTensor(void *dptr, MXDType type, const int64_t *dims, int ndims, size_t vID, MXContext mx_ctx, MXStorageType storage_type)</div><div class="ttdoc">populate internal tensor fields</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a03342e626369680bc842ec763588a91e"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a03342e626369680bc842ec763588a91e">mxnet::ext::partCallSupportedOps_t</a></div><div class="ttdeci">int(* partCallSupportedOps_t)(supportedOps_t supportedOps, const char *json, int num_ids, int *ids, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1166</div></div> |
| <div class="ttc" id="alib__api_8h_html_af0214213e9b925b0750a33f20ff2a1f9"><div class="ttname"><a href="lib__api_8h.html#af0214213e9b925b0750a33f20ff2a1f9">_opCallFStatefulCompute</a></div><div class="ttdeci">MX_INT_RET _opCallFStatefulCompute(int is_forward, void *state_op, const int64_t **inshapes, int *indims, void **indata, int *intypes, size_t *inIDs, const char **indev_type, int *indev_id, int num_in, const int64_t **outshapes, int *outdims, void **outdata, int *outtypes, size_t *outIDs, const char **outdev_type, int *outdev_id, int num_out, mxnet::ext::xpu_malloc_t cpu_malloc, void *cpu_alloc, mxnet::ext::xpu_malloc_t gpu_malloc, void *gpu_alloc, void *stream, mxnet::ext::sparse_malloc_t sparse_malloc, void *sparse_alloc, int *instypes, int *outstypes, void **in_indices, void **out_indices, void **in_indptr, void **out_indptr, int64_t *in_indices_shapes, int64_t *out_indices_shapes, int64_t *in_indptr_shapes, int64_t *out_indptr_shapes, void *rng_cpu_states, void *rng_gpu_states)</div><div class="ttdoc">returns status of calling Stateful Forward/Backward for operator from library</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_afadf4694fc70c782d4b3b63cd2d1e1e8"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#afadf4694fc70c782d4b3b63cd2d1e1e8">mxnet::ext::JsonVal::dump</a></div><div class="ttdeci">std::string dump() const</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1PassResource_html_aeff45b0c181dc59ce73a4802d2c77a0e"><div class="ttname"><a href="classmxnet_1_1ext_1_1PassResource.html#aeff45b0c181dc59ce73a4802d2c77a0e">mxnet::ext::PassResource::alloc_aux</a></div><div class="ttdeci">MXTensor * alloc_aux(const std::string &name, const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype) const</div></div> |
| <div class="ttc" id="alib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a></div><div class="ttdeci">@ kDLInt</div><div class="ttdef"><b>Definition:</b> lib_api.h:143</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a9f826ae984fc3f102c1bb6761f3f23ee"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a9f826ae984fc3f102c1bb6761f3f23ee">mxnet::ext::JsonVal::type</a></div><div class="ttdeci">JsonType type</div><div class="ttdef"><b>Definition:</b> lib_api.h:564</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_ab49ade50779a6d387b0073ad04d6dafa"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#ab49ade50779a6d387b0073ad04d6dafa">mxnet::ext::Graph::Graph</a></div><div class="ttdeci">Graph()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a9233c8bb6aa95b7f8d03de17960723b6"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a9233c8bb6aa95b7f8d03de17960723b6">mxnet::ext::Graph::_setParams</a></div><div class="ttdeci">void _setParams(std::unordered_map< std::string, mxnet::ext::MXTensor > *args, std::unordered_map< std::string, mxnet::ext::MXTensor > *aux)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ac19085fc1abe2e392d1305d9ac3bc015"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ac19085fc1abe2e392d1305d9ac3bc015">mxnet::ext::getShapeAt</a></div><div class="ttdeci">std::string getShapeAt(const std::string &shape, unsigned index)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a08ba0f08fe57589b83315d1f0cefde37"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a08ba0f08fe57589b83315d1f0cefde37">mxnet::ext::JsonVal::parse_num</a></div><div class="ttdeci">static JsonVal parse_num(const std::string &json, unsigned int *idx)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a82c4b41a9a9a1160f1043d10dc9fcf50"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a82c4b41a9a9a1160f1043d10dc9fcf50">mxnet::ext::msgSize_t</a></div><div class="ttdeci">int(* msgSize_t)(void)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1276</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ac3c274938c9da1e44e9897639af5c0d4"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ac3c274938c9da1e44e9897639af5c0d4">mxnet::ext::opCallParseAttrs_t</a></div><div class="ttdeci">int(* opCallParseAttrs_t)(parseAttrs_t parseAttrs, const char *const *keys, const char *const *vals, int num, int *num_in, int *num_out)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1005</div></div> |
| <div class="ttc" id="alib__api_8h_html_acb30ec605ba828f085c28799b89383ad"><div class="ttname"><a href="lib__api_8h.html#acb30ec605ba828f085c28799b89383ad">PRIVATE_SYMBOL</a></div><div class="ttdeci">#define PRIVATE_SYMBOL</div><div class="ttdoc">For loading multiple custom op libraries in Linux, exporting same symbol multiple times may lead to u...</div><div class="ttdef"><b>Definition:</b> lib_api.h:65</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1NodeEntry_html_a9ccf74d370d36c4ed911e94f84a431be"><div class="ttname"><a href="structmxnet_1_1ext_1_1NodeEntry.html#a9ccf74d370d36c4ed911e94f84a431be">mxnet::ext::NodeEntry::node</a></div><div class="ttdeci">Node * node</div><div class="ttdef"><b>Definition:</b> lib_api.h:579</div></div> |
| <div class="ttc" id="alib__api_8h_html_a8088989b079b01b821c7b7d0b960dfaa"><div class="ttname"><a href="lib__api_8h.html#a8088989b079b01b821c7b7d0b960dfaa">_partRegSize</a></div><div class="ttdeci">MX_INT_RET _partRegSize()</div><div class="ttdoc">returns number of partitioners registered in this library</div></div> |
| <div class="ttc" id="alib__api_8h_html_a67702f536b0652fab92a46a4c522f0a8"><div class="ttname"><a href="lib__api_8h.html#a67702f536b0652fab92a46a4c522f0a8">_opCallInferSType</a></div><div class="ttdeci">MX_INT_RET _opCallInferSType(mxnet::ext::inferSType_t inferSType, const char *const *keys, const char *const *vals, int num, int *instypes, int num_in, int *outstypes, int num_out)</div><div class="ttdoc">returns status of calling inferSType function for operator from library</div></div> |
| <div class="ttc" id="alib__api_8h_html_a6e74b9b5a033075b0a934561a518a2cb"><div class="ttname"><a href="lib__api_8h.html#a6e74b9b5a033075b0a934561a518a2cb">_opCallFree</a></div><div class="ttdeci">MX_VOID_RET _opCallFree(void *ptr)</div><div class="ttdoc">calls free from the external library for library allocated arrays</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_afbe960c8b040871dfb9f8c8bdb529947"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#afbe960c8b040871dfb9f8c8bdb529947">mxnet::ext::Node::subgraphs</a></div><div class="ttdeci">std::vector< Graph * > subgraphs</div><div class="ttdef"><b>Definition:</b> lib_api.h:602</div></div> |
| <div class="ttc" id="alib__api_8h_html_ac47e22b68597148e7989b3e1ec6497e0"><div class="ttname"><a href="lib__api_8h.html#ac47e22b68597148e7989b3e1ec6497e0">_passRegGet</a></div><div class="ttdeci">MX_VOID_RET _passRegGet(int pass_idx, mxnet::ext::graphPass_t *graphPass, const char **pass_name)</div><div class="ttdoc">returns pass registration at specified index</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af447080bf199456379fd14d68804c1c7"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af447080bf199456379fd14d68804c1c7">mxnet::ext::opCallFComp_t</a></div><div class="ttdeci">int(* opCallFComp_t)(fcomp_t fcomp, const char *const *keys, const char *const *vals, int num, const int64_t **inshapes, int *indims, void **indata, int *intypes, size_t *inIDs, const char **indev_type, int *indev_id, int num_in, const int64_t **outshapes, int *outdims, void **outdata, int *outtypes, size_t *outIDs, const char **outdev_type, int *outdev_id, int num_out, xpu_malloc_t cpu_malloc, void *cpu_alloc, xpu_malloc_t gpu_malloc, void *gpu_alloc, void *cuda_stream, sparse_malloc_t sparse_malloc, void *sparse_alloc, int *instypes, int *outstypes, void **in_indices, void **out_indices, void **in_indptr, void **out_indptr, int64_t *in_indices_shapes, int64_t *out_indices_shapes, int64_t *in_indptr_shapes, int64_t *out_indptr_shapes, void *rng_cpu_states, void *rng_gpu_states)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1047</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html">mxnet::ext::OpResource</a></div><div class="ttdoc">provide resource APIs memory allocation mechanism to Forward/Backward functions</div><div class="ttdef"><b>Definition:</b> lib_api.h:442</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a32fc355e422990910299065269f10459"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a32fc355e422990910299065269f10459">mxnet::ext::partCallReset_t</a></div><div class="ttdeci">void(* partCallReset_t)(void *sel_inst)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1202</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1MXerrorMsgs_html_a416e02951e24afc2792816f54de16320"><div class="ttname"><a href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a416e02951e24afc2792816f54de16320">mxnet::ext::MXerrorMsgs::size</a></div><div class="ttdeci">int size()</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXContext_html_a79b8b2861199ad3d8890b2086cdb8582"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXContext.html#a79b8b2861199ad3d8890b2086cdb8582">mxnet::ext::MXContext::dev_type</a></div><div class="ttdeci">std::string dev_type</div><div class="ttdef"><b>Definition:</b> lib_api.h:286</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_ae5aabe1376e697faf14c30561c8533cd"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#ae5aabe1376e697faf14c30561c8533cd">mxnet::ext::Graph::getNode</a></div><div class="ttdeci">Node * getNode(size_t idx)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html">mxnet::ext::Node</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:584</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1PassResource_html_a9db8c01ef60ebd075caf275ab499a05b"><div class="ttname"><a href="classmxnet_1_1ext_1_1PassResource.html#a9db8c01ef60ebd075caf275ab499a05b">mxnet::ext::PassResource::alloc_arg</a></div><div class="ttdeci">MXTensor * alloc_arg(const std::string &name, const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype) const</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a7a1b71e81d17df4f68e664a5d931866d"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a7a1b71e81d17df4f68e664a5d931866d">mxnet::ext::CustomOp::create_op_ctx_cstr</a></div><div class="ttdeci">std::vector< const char * > create_op_ctx_cstr</div><div class="ttdef"><b>Definition:</b> lib_api.h:815</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a69ba2ade8bd893eb8461a21cd8f60eb5"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a69ba2ade8bd893eb8461a21cd8f60eb5">mxnet::ext::opVersion_t</a></div><div class="ttdeci">int(* opVersion_t)()</div><div class="ttdef"><b>Definition:</b> lib_api.h:1273</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048">mxnet::ext::kUNSET</a></div><div class="ttdeci">@ kUNSET</div><div class="ttdef"><b>Definition:</b> lib_api.h:257</div></div> |
| <div class="ttc" id="alib__api_8h_html_a3b4abe8a1f483ab51bb4274be2f193ac"><div class="ttname"><a href="lib__api_8h.html#a3b4abe8a1f483ab51bb4274be2f193ac">initialize</a></div><div class="ttdeci">mxnet::ext::MXReturnValue initialize(int version)</div><div class="ttdoc">Checks if the MXNet version is supported by the library. If supported, initializes the library.</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704">mxnet::ext::kFloat32</a></div><div class="ttdeci">@ kFloat32</div><div class="ttdef"><b>Definition:</b> lib_api.h:250</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_ae45b21c0986b8986bcd9ecd5e2b3580e"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#ae45b21c0986b8986bcd9ecd5e2b3580e">mxnet::ext::Graph::toJson</a></div><div class="ttdeci">JsonVal toJson() const</div></div> |
| <div class="ttc" id="alib__api_8h_html_aa14e95577fdb7526da526ae1ec236d48"><div class="ttname"><a href="lib__api_8h.html#aa14e95577fdb7526da526ae1ec236d48">_opCallMutateInputs</a></div><div class="ttdeci">MX_INT_RET _opCallMutateInputs(mxnet::ext::mutateInputs_t mutate, const char *const *keys, const char *const *vals, int num, int **mutate_indices, int *indices_size)</div><div class="ttdoc">returns status of calling mutateInputs function for operator from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1MXerrorMsgs_html_a6da046f33093c9274b12f8bf6cdba61e"><div class="ttname"><a href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a6da046f33093c9274b12f8bf6cdba61e">mxnet::ext::MXerrorMsgs::get</a></div><div class="ttdeci">static MXerrorMsgs * get()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_ad0ff7bf816398b0eee6d8464705df7de"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#ad0ff7bf816398b0eee6d8464705df7de">mxnet::ext::CustomOp::setParseAttrs</a></div><div class="ttdeci">CustomOp & setParseAttrs(parseAttrs_t func)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_abb2fa268862b67ad08c6b946130609a5"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#abb2fa268862b67ad08c6b946130609a5">mxnet::ext::MXSparse::set</a></div><div class="ttdeci">void set(void *data_ptr, const int64_t *dims, int ndims, void *idx, int64_t num_idx, void *idx_ptr=nullptr, int64_t num_idx_ptr=0)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_af70aca6c1872bc2750b5ccfb72655298"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#af70aca6c1872bc2750b5ccfb72655298">mxnet::ext::CustomOp::setInferShape</a></div><div class="ttdeci">CustomOp & setInferShape(inferShape_t func)</div></div> |
| <div class="ttc" id="alib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50">DLDataTypeCode</a></div><div class="ttdeci">DLDataTypeCode</div><div class="ttdoc">The type code options DLDataType.</div><div class="ttdef"><b>Definition:</b> lib_api.h:142</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html">mxnet::ext::CustomOp</a></div><div class="ttdoc">Class to hold custom operator registration.</div><div class="ttdef"><b>Definition:</b> lib_api.h:779</div></div> |
| <div class="ttc" id="alib__api_8h_html_af623e317a08ae2d7822e4e81d4dcb1a4"><div class="ttname"><a href="lib__api_8h.html#af623e317a08ae2d7822e4e81d4dcb1a4">_partCallSelectOutput</a></div><div class="ttdeci">MX_VOID_RET _partCallSelectOutput(void *sel_inst, int nodeID, int output_nodeID, int *selected)</div><div class="ttdoc">returns status of calling select output function from library</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a">mxnet::ext::kRowSparseStorage</a></div><div class="ttdeci">@ kRowSparseStorage</div><div class="ttdef"><b>Definition:</b> lib_api.h:267</div></div> |
| <div class="ttc" id="alib__api_8h_html_a876d77c3d311c7006f7b23578f85678a"><div class="ttname"><a href="lib__api_8h.html#a876d77c3d311c7006f7b23578f85678a">_partCallSelectInput</a></div><div class="ttdeci">MX_VOID_RET _partCallSelectInput(void *sel_inst, int nodeID, int input_nodeID, int *selected)</div><div class="ttdoc">returns status of calling select input function from library</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a0bf4c32f0c8f5da90619bf0b1f567ef9"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a></div><div class="ttdeci">MXReturnValue(* inferShape_t)(const std::unordered_map< std::string, std::string > &attributes, std::vector< std::vector< unsigned int > > *in_shapes, std::vector< std::vector< unsigned int > > *out_shapes)</div><div class="ttdef"><b>Definition:</b> lib_api.h:762</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_af6ab4915225627081794e2d02b60b223"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#af6ab4915225627081794e2d02b60b223">mxnet::ext::Graph::addNode</a></div><div class="ttdeci">Node * addNode(const std::string &name, const std::string &op)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1MXerrorMsgs_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1MXerrorMsgs.html">mxnet::ext::MXerrorMsgs</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:220</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a4d96003e8adf76b07e993b83f7cef5bd"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a4d96003e8adf76b07e993b83f7cef5bd">mxnet::ext::passCallGraphPass_t</a></div><div class="ttdeci">int(* passCallGraphPass_t)(graphPass_t graphPass, const char *in_graph, char **out_graph, const char *const *opt_keys, const char *const *opt_vals, int num_opts, const char *pass_name, const char *const *arg_names, int num_args, void *const *arg_data, const int64_t *const *arg_shapes, const int *arg_dims, const int *arg_types, const size_t *arg_IDs, const char *const *arg_dev_type, const int *arg_dev_id, const char *const *aux_names, int num_aux, void *const *aux_data, const int64_t *const *aux_shapes, const int *aux_dims, const int *aux_types, const size_t *aux_IDs, const char *const *aux_dev_type, const int *aux_dev_id, nd_malloc_t nd_malloc, const void *nd_alloc)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1241</div></div> |
| <div class="ttc" id="alib__api_8h_html_a140a8dac396de476f9ed0c6a093cf4fa"><div class="ttname"><a href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa">MX_VOID_RET</a></div><div class="ttdeci">#define MX_VOID_RET</div><div class="ttdef"><b>Definition:</b> lib_api.h:1301</div></div> |
| <div class="ttc" id="alib__api_8h_html_a670530b58e8ddc61982798fd788c3edf"><div class="ttname"><a href="lib__api_8h.html#a670530b58e8ddc61982798fd788c3edf">_partCallFilter</a></div><div class="ttdeci">MX_VOID_RET _partCallFilter(void *sel_inst, int *candidates, int num_candidates, int **keep, int *num_keep)</div><div class="ttdoc">returns status of calling filter function from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_a42d0e539870e3c27db8eb88606765f5d"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a42d0e539870e3c27db8eb88606765f5d">mxnet::ext::CustomStatefulOp::create</a></div><div class="ttdeci">static CustomStatefulOp * create(Ts... args)</div><div class="ttdef"><b>Definition:</b> lib_api.h:720</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_aa7339fd6ce1c23dd296015f450463cc4"><div class="ttname"><a href="namespacemxnet_1_1ext.html#aa7339fd6ce1c23dd296015f450463cc4">mxnet::ext::msgGet_t</a></div><div class="ttdeci">int(* msgGet_t)(int idx, const char **msg)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1279</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884">mxnet::ext::kFloat64</a></div><div class="ttdeci">@ kFloat64</div><div class="ttdef"><b>Definition:</b> lib_api.h:251</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_acd1f86bb22908ed80946488bb99c5020"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#acd1f86bb22908ed80946488bb99c5020">mxnet::ext::OpResource::get_cpu_rand_states</a></div><div class="ttdeci">mx_cpu_rand_t * get_cpu_rand_states() const</div><div class="ttdoc">get pointer to initialized and seeded random number states located on CPU</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a2c1ec7bd5c9132e2636e70a93a53e452"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a2c1ec7bd5c9132e2636e70a93a53e452">mxnet::ext::Node::Node</a></div><div class="ttdeci">Node()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Registry_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1Registry.html">mxnet::ext::Registry</a></div><div class="ttdoc">Registry class to registers things (ops, properties) Singleton class.</div><div class="ttdef"><b>Definition:</b> lib_api.h:902</div></div> |
| <div class="ttc" id="alib__api_8h_html_a2ba7371715fcd75f05c0ba0f0c8b78b4"><div class="ttname"><a href="lib__api_8h.html#a2ba7371715fcd75f05c0ba0f0c8b78b4">_opCallInferShape</a></div><div class="ttdeci">MX_INT_RET _opCallInferShape(mxnet::ext::inferShape_t inferShape, const char *const *keys, const char *const *vals, int num, unsigned int **inshapes, int *indims, int num_in, unsigned int ***mod_inshapes, int **mod_indims, unsigned int ***outshapes, int **outdims, int num_out)</div><div class="ttdoc">returns status of calling inferShape function for operator from library</div></div> |
| <div class="ttc" id="alib__api_8h_html_a6efacb1a30200f9f1fb05f77783959de"><div class="ttname"><a href="lib__api_8h.html#a6efacb1a30200f9f1fb05f77783959de">_opRegSize</a></div><div class="ttdeci">MX_INT_RET _opRegSize()</div><div class="ttdoc">returns number of ops registered in this library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_ac1d121b14423cb590b611bbbe866f3c7"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#ac1d121b14423cb590b611bbbe866f3c7">mxnet::ext::CustomOp::setForward</a></div><div class="ttdeci">CustomOp & setForward(fcomp_t fcomp, const char *ctx)</div></div> |
| <div class="ttc" id="alib__api_8h_html_a97a361685f2eb83a744f7fef61d8f26b"><div class="ttname"><a href="lib__api_8h.html#a97a361685f2eb83a744f7fef61d8f26b">_passRegSize</a></div><div class="ttdeci">MX_INT_RET _passRegSize()</div><div class="ttdoc">returns number of graph passes registered in this library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a49edc0a4c130a3f811d67410e07f1970"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a49edc0a4c130a3f811d67410e07f1970">mxnet::ext::CustomOp::CustomOp</a></div><div class="ttdeci">CustomOp(const char *op_name)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a4b16f5086857cf3dd983f9ef2b96c204"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a4b16f5086857cf3dd983f9ef2b96c204">mxnet::ext::MXTensor::data</a></div><div class="ttdeci">data_type * data()</div><div class="ttdoc">helper function to cast data pointer</div><div class="ttdef"><b>Definition:</b> lib_api.h:349</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_a391e1c2ccc10a9edc19417fa3fad3169"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#a391e1c2ccc10a9edc19417fa3fad3169">mxnet::ext::Node::alloc_arg</a></div><div class="ttdeci">void alloc_arg(const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_a3c75f8ca600930d970e3ea20cd624884"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#a3c75f8ca600930d970e3ea20cd624884">mxnet::ext::OpResource::get_gpu_rand_states</a></div><div class="ttdeci">mx_gpu_rand_t * get_gpu_rand_states() const</div><div class="ttdoc">get pointer to initialized and seeded random number states located on GPU</div><div class="ttdef"><b>Definition:</b> lib_api.h:475</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_accf665442e338c1f09832dde3752fc02"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#accf665442e338c1f09832dde3752fc02">mxnet::ext::MXTensor::MXTensor</a></div><div class="ttdeci">MXTensor()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_afda2391d5b2ae16121ab1c1bf5eb3ae3"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#afda2391d5b2ae16121ab1c1bf5eb3ae3">mxnet::ext::Graph::outputs</a></div><div class="ttdeci">std::vector< NodeEntry > outputs</div><div class="ttdef"><b>Definition:</b> lib_api.h:666</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e">kDLCPU</a></div><div class="ttdeci">@ kDLCPU</div><div class="ttdoc">CPU device.</div><div class="ttdef"><b>Definition:</b> lib_api.h:103</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_ad2d93199d25ceace68343f8c516bf63e"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#ad2d93199d25ceace68343f8c516bf63e">mxnet::ext::CustomOp::create_op_fp</a></div><div class="ttdeci">std::vector< createOpState_t > create_op_fp</div><div class="ttdef"><b>Definition:</b> lib_api.h:817</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680">kDLGPU</a></div><div class="ttdeci">@ kDLGPU</div><div class="ttdoc">CUDA GPU device.</div><div class="ttdef"><b>Definition:</b> lib_api.h:105</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_a037945db6e2934e1f4acefe2c8337740"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#a037945db6e2934e1f4acefe2c8337740">mxnet::ext::OpResource::alloc_sparse</a></div><div class="ttdeci">void alloc_sparse(MXSparse *sparse, int index, int indices_len, int indptr_len=0) const</div><div class="ttdoc">allocate sparse memory controlled by MXNet</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOpSelector_html_a1ebe634fe8f38266d1e85624d1707e55"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOpSelector.html#a1ebe634fe8f38266d1e85624d1707e55">mxnet::ext::CustomOpSelector::Select</a></div><div class="ttdeci">virtual bool Select(int nodeID)=0</div></div> |
| <div class="ttc" id="alib__api_8h_html_a37f969a158ff72a111f84a466aca5119"><div class="ttname"><a href="lib__api_8h.html#a37f969a158ff72a111f84a466aca5119">_passCallGraphPass</a></div><div class="ttdeci">MX_INT_RET _passCallGraphPass(mxnet::ext::graphPass_t graphPass, const char *json, char **out_graph, const char *const *opt_keys, const char *const *opt_vals, int num_opts, const char *pass_name, const char *const *arg_names, int num_args, void *const *arg_data, const int64_t *const *arg_shapes, const int *arg_dims, const int *arg_types, const size_t *arg_IDs, const char *const *arg_dev_type, const int *arg_dev_id, const char *const *aux_names, int num_aux, void *const *aux_data, const int64_t *const *aux_shapes, const int *aux_dims, const int *aux_types, const size_t *aux_IDs, const char *const *aux_dev_type, const int *aux_dev_id, mxnet::ext::nd_malloc_t nd_malloc, const void *nd_alloc)</div><div class="ttdoc">returns status of calling graph pass function from library</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a6ce1906895cb2b88cae60e3e1137c8bd"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a6ce1906895cb2b88cae60e3e1137c8bd">mxnet::ext::CustomOp::backward_ctx_cstr</a></div><div class="ttdeci">std::vector< const char * > backward_ctx_cstr</div><div class="ttdef"><b>Definition:</b> lib_api.h:815</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ad2fed961bb0a655fb91fc6ee4c26162a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ad2fed961bb0a655fb91fc6ee4c26162a">mxnet::ext::partCallSelectInput_t</a></div><div class="ttdeci">void(* partCallSelectInput_t)(void *sel_inst, int nodeID, int input_nodeID, int *selected)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1186</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a228dcaf825c6fcb89d8310fec774f4cd"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a228dcaf825c6fcb89d8310fec774f4cd">mxnet::ext::CustomPartitioner::addStrategy</a></div><div class="ttdeci">CustomPartitioner & addStrategy(const char *prop_name, const char *sg_name)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a08f1c47db9eeec676f4b003ed01a622d"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">mxnet::ext::graphPass_t</a></div><div class="ttdeci">MXReturnValue(* graphPass_t)(mxnet::ext::Graph *graph, const std::unordered_map< std::string, std::string > &options)</div><div class="ttdoc">Custom Pass Create function template.</div><div class="ttdef"><b>Definition:</b> lib_api.h:828</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOpSelector_html_a2e39f0a634eee550cb79b225d1beac02"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOpSelector.html#a2e39f0a634eee550cb79b225d1beac02">mxnet::ext::CustomOpSelector::Filter</a></div><div class="ttdeci">virtual void Filter(const std::vector< int > &candidates, std::vector< int > *keep)</div><div class="ttdef"><b>Definition:</b> lib_api.h:700</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXContext_html_af6287fbdda27736f4406bf2ec3d69756"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXContext.html#af6287fbdda27736f4406bf2ec3d69756">mxnet::ext::MXContext::MXContext</a></div><div class="ttdeci">MXContext()</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a38959dfb4fd816f407c76a7285639758"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a38959dfb4fd816f407c76a7285639758">mxnet::ext::MXTensor::stype</a></div><div class="ttdeci">MXStorageType stype</div><div class="ttdef"><b>Definition:</b> lib_api.h:380</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOpWrapper_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html">mxnet::ext::CustomStatefulOpWrapper</a></div><div class="ttdoc">StatefulOp wrapper class to pass to backend OpState.</div><div class="ttdef"><b>Definition:</b> lib_api.h:1282</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ae628507b542734ee29ddf2f499fbdb05"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ae628507b542734ee29ddf2f499fbdb05">mxnet::ext::partRegGet_t</a></div><div class="ttdeci">void(* partRegGet_t)(int part_idx, int stg_idx, const char **strategy, supportedOps_t *supportedOps, createSelector_t *createSelector, reviewSubgraph_t *reviewSubgraph, const char **op_name)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1157</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a6047dc7436d917e950612a84fd912760"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a></div><div class="ttdeci">void *(* xpu_malloc_t)(void *, int)</div><div class="ttdoc">resource malloc function to allocate memory inside Forward/Backward functions</div><div class="ttdef"><b>Definition:</b> lib_api.h:384</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_af7369ec802b04819c641334e521d7dae"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#af7369ec802b04819c641334e521d7dae">mxnet::ext::MXTensor::dltensor</a></div><div class="ttdeci">DLTensor dltensor</div><div class="ttdef"><b>Definition:</b> lib_api.h:377</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_a46fb6fe57019f1ef4d4a616c7ef7297f"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a46fb6fe57019f1ef4d4a616c7ef7297f">mxnet::ext::CustomStatefulOp::wasCreated</a></div><div class="ttdeci">bool wasCreated()</div><div class="ttdef"><b>Definition:</b> lib_api.h:726</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a76e111c49dd7b6e3cfe550739f319611"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a76e111c49dd7b6e3cfe550739f319611">mxnet::ext::CustomPartitioner::review_map</a></div><div class="ttdeci">std::map< std::string, reviewSubgraph_t > review_map</div><div class="ttdef"><b>Definition:</b> lib_api.h:890</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_abaadec312533d0a24f1faeba6db7411b"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#abaadec312533d0a24f1faeba6db7411b">mxnet::ext::CustomPartitioner::strategies</a></div><div class="ttdeci">std::vector< const char * > strategies</div><div class="ttdoc">strategy names</div><div class="ttdef"><b>Definition:</b> lib_api.h:892</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">mxnet::ext::MXStorageType</a></div><div class="ttdeci">MXStorageType</div><div class="ttdef"><b>Definition:</b> lib_api.h:263</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_adb7dd13f57912db3f560c2baf223d1d7"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#adb7dd13f57912db3f560c2baf223d1d7">mxnet::ext::CustomOp::infer_shape</a></div><div class="ttdeci">inferShape_t infer_shape</div><div class="ttdef"><b>Definition:</b> lib_api.h:810</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a2a16b18af2cdbff088a3c76d62b39b40"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">mxnet::ext::MXReturnValue</a></div><div class="ttdeci">MXReturnValue</div><div class="ttdef"><b>Definition:</b> lib_api.h:290</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Registry_html_a32f3e2d387f02dc4edc8c0c51095af28"><div class="ttname"><a href="classmxnet_1_1ext_1_1Registry.html#a32f3e2d387f02dc4edc8c0c51095af28">mxnet::ext::Registry::add</a></div><div class="ttdeci">T & add(const char *name)</div><div class="ttdoc">add a new entry</div><div class="ttdef"><b>Definition:</b> lib_api.h:916</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_aa4f66a1473ea311e3adcd0f7c259677c"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#aa4f66a1473ea311e3adcd0f7c259677c">mxnet::ext::CustomOp::setMutateInputs</a></div><div class="ttdeci">CustomOp & setMutateInputs(mutateInputs_t func)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e">mxnet::ext::kDefaultStorage</a></div><div class="ttdeci">@ kDefaultStorage</div><div class="ttdef"><b>Definition:</b> lib_api.h:265</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a5b53f5861183811ba8fc2cdf9dbee5f6"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a5b53f5861183811ba8fc2cdf9dbee5f6">mxnet::ext::partCallSelectOutput_t</a></div><div class="ttdeci">void(* partCallSelectOutput_t)(void *sel_inst, int nodeID, int output_nodeID, int *selected)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1189</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a4ac609b75f121c4b70b3ba21405deb85"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a4ac609b75f121c4b70b3ba21405deb85">mxnet::ext::CustomPartitioner::setCreateSelector</a></div><div class="ttdeci">CustomPartitioner & setCreateSelector(const char *prop_name, createSelector_t fn)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a995c0b2b6d1d520852a664a31c44c984"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a995c0b2b6d1d520852a664a31c44c984">mxnet::ext::Graph::DFS</a></div><div class="ttdeci">void DFS(std::function< void(Node *)> handler) const</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_ac5005c18501c8da80e170f88e1eaa454"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#ac5005c18501c8da80e170f88e1eaa454">mxnet::ext::JsonVal::parse_string</a></div><div class="ttdeci">static JsonVal parse_string(const std::string &json, unsigned int *idx)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_aca7db0bccdf4840d5cd70ff73f7624da"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#aca7db0bccdf4840d5cd70ff73f7624da">mxnet::ext::CustomStatefulOp::ignore_warn</a></div><div class="ttdeci">bool ignore_warn</div><div class="ttdef"><b>Definition:</b> lib_api.h:740</div></div> |
| <div class="ttc" id="alib__api_8h_html_a8002368ce25c4f689b48c511b092373b"><div class="ttname"><a href="lib__api_8h.html#a8002368ce25c4f689b48c511b092373b">_partCallReset</a></div><div class="ttdeci">MX_VOID_RET _partCallReset(void *sel_inst)</div><div class="ttdoc">returns status of calling reset selector function from library</div></div> |
| <div class="ttc" id="alib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a></div><div class="ttdeci">@ kDLFloat</div><div class="ttdef"><b>Definition:</b> lib_api.h:145</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_aad7a2b18c34573c8ded3f4d8069df3c9"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#aad7a2b18c34573c8ded3f4d8069df3c9">mxnet::ext::Graph::fromJson</a></div><div class="ttdeci">static Graph * fromJson(JsonVal val)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a634c4e6805c4928fcbbf859306ce1fee"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a634c4e6805c4928fcbbf859306ce1fee">mxnet::ext::MXTensor::ctx</a></div><div class="ttdeci">MXContext ctx</div><div class="ttdef"><b>Definition:</b> lib_api.h:373</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1MXerrorMsgs_html_ab794a5b63f8679e704539a9c0bb0d946"><div class="ttname"><a href="classmxnet_1_1ext_1_1MXerrorMsgs.html#ab794a5b63f8679e704539a9c0bb0d946">mxnet::ext::MXerrorMsgs::add</a></div><div class="ttdeci">std::stringstream & add(const char *file, int line)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXContext_html_a0930c97f53cdd8fb98cb1bec71c23e68"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXContext.html#a0930c97f53cdd8fb98cb1bec71c23e68">mxnet::ext::MXContext::GPU</a></div><div class="ttdeci">static MXContext GPU()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a2978383f7d86dafd36cff9eeba16c8be"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a2978383f7d86dafd36cff9eeba16c8be">mxnet::ext::CustomOp::parse_attrs</a></div><div class="ttdeci">parseAttrs_t parse_attrs</div><div class="ttdoc">operator functions</div><div class="ttdef"><b>Definition:</b> lib_api.h:807</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_abc2b3e8b645050faf1a4534de694e939"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#abc2b3e8b645050faf1a4534de694e939">mxnet::ext::Node::name</a></div><div class="ttdeci">std::string name</div><div class="ttdef"><b>Definition:</b> lib_api.h:598</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXTensor_html_a67f147927ed642c1adec68471856a703"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXTensor.html#a67f147927ed642c1adec68471856a703">mxnet::ext::MXTensor::size</a></div><div class="ttdeci">int64_t size() const</div><div class="ttdoc">helper function to get data size</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_a7abe4f71d2754fbf7ed00736c37ffe43"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#a7abe4f71d2754fbf7ed00736c37ffe43">mxnet::ext::MXSparse::data_len</a></div><div class="ttdeci">int64_t data_len</div><div class="ttdef"><b>Definition:</b> lib_api.h:300</div></div> |
| <div class="ttc" id="alib__api_8h_html_a561ade7c8fd77c6882e937514f5b1e05"><div class="ttname"><a href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05">MX_INT_RET</a></div><div class="ttdeci">#define MX_INT_RET</div><div class="ttdef"><b>Definition:</b> lib_api.h:1300</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html">mxnet::ext::CustomStatefulOp</a></div><div class="ttdoc">An abstract class for library authors creating stateful op custom library should override Forward and...</div><div class="ttdef"><b>Definition:</b> lib_api.h:714</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_ad2fb294e9396e263c79199054f9b5520"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#ad2fb294e9396e263c79199054f9b5520">mxnet::ext::Graph::topological_sort</a></div><div class="ttdeci">std::vector< Node * > topological_sort() const</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573">DLDeviceType</a></div><div class="ttdeci">DLDeviceType</div><div class="ttdoc">The device type in DLContext.</div><div class="ttdef"><b>Definition:</b> lib_api.h:101</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_acc7b8856d1a69b0b95ac0e2556800d4e"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#acc7b8856d1a69b0b95ac0e2556800d4e">mxnet::ext::CustomOp::setInferSType</a></div><div class="ttdeci">CustomOp & setInferSType(inferSType_t func)</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a1d9e77f9eb4ab27cdcffb85495a760ae"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a1d9e77f9eb4ab27cdcffb85495a760ae">mxnet::ext::JsonVal::num</a></div><div class="ttdeci">int num</div><div class="ttdef"><b>Definition:</b> lib_api.h:565</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ade42ac66b3d7c1e063d0f71eda6394a3"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ade42ac66b3d7c1e063d0f71eda6394a3">mxnet::ext::opCallInferShape_t</a></div><div class="ttdeci">int(* opCallInferShape_t)(inferShape_t inferShape, const char *const *keys, const char *const *vals, int num, unsigned int **inshapes, int *indims, int num_in, unsigned int ***mod_inshapes, int **mod_indims, unsigned int ***outshapes, int **outdims, int num_out)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1013</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOpWrapper_html_a670e9614e832a6d520e90d56d82c5890"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#a670e9614e832a6d520e90d56d82c5890">mxnet::ext::CustomStatefulOpWrapper::~CustomStatefulOpWrapper</a></div><div class="ttdeci">~CustomStatefulOpWrapper()</div></div> |
| <div class="ttc" id="astructDLTensor_html"><div class="ttname"><a href="structDLTensor.html">DLTensor</a></div><div class="ttdoc">Plain C Tensor object, does not manage memory.</div><div class="ttdef"><b>Definition:</b> dlpack.h:112</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_af96c64b969175dccd80943ef69207250"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#af96c64b969175dccd80943ef69207250">mxnet::ext::CustomPartitioner::selector_map</a></div><div class="ttdeci">std::map< std::string, createSelector_t > selector_map</div><div class="ttdef"><b>Definition:</b> lib_api.h:889</div></div> |
| <div class="ttc" id="alib__api_8h_html_aed3b9c90c89e5a3cf303b51c4ffa1938"><div class="ttname"><a href="lib__api_8h.html#aed3b9c90c89e5a3cf303b51c4ffa1938">_msgGet</a></div><div class="ttdeci">MX_VOID_RET _msgGet(int idx, const char **msg)</div><div class="ttdoc">returns operator registration at specified index</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_afab0ffe74ed0a370ee02a8af00084659"><div class="ttname"><a href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">mxnet::ext::createSelector_t</a></div><div class="ttdeci">MXReturnValue(* createSelector_t)(const mxnet::ext::Graph *graph, CustomOpSelector **sel_inst, const std::unordered_map< std::string, std::string > &options)</div><div class="ttdef"><b>Definition:</b> lib_api.h:852</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a559767743debbc2b39b203185ca67923"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a></div><div class="ttdeci">MXReturnValue(* createOpState_t)(const std::unordered_map< std::string, std::string > &attributes, const MXContext &ctx, const std::vector< std::vector< unsigned int > > &in_shapes, const std::vector< int > in_types, CustomStatefulOp **)</div><div class="ttdef"><b>Definition:</b> lib_api.h:769</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1OpResource_html_a4e9eb54d2481fcb97231fec20290bd5d"><div class="ttname"><a href="classmxnet_1_1ext_1_1OpResource.html#a4e9eb54d2481fcb97231fec20290bd5d">mxnet::ext::OpResource::OpResource</a></div><div class="ttdeci">OpResource(xpu_malloc_t cpu_malloc_fp, void *cpu_alloc_fp, xpu_malloc_t gpu_malloc_fp, void *gpu_alloc_fp, void *stream, sparse_malloc_t sparse_malloc_fp, void *sparse_alloc_fp, void *rng_cpu_states, void *rng_gpu_states)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a6aeb4f0d430f6599a887af467ca05d61"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a6aeb4f0d430f6599a887af467ca05d61">mxnet::ext::Graph::size</a></div><div class="ttdeci">size_t size() const</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOp_html_acbf1b61e0f1cb13ba9df92cf2774ba7b"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#acbf1b61e0f1cb13ba9df92cf2774ba7b">mxnet::ext::CustomStatefulOp::~CustomStatefulOp</a></div><div class="ttdeci">virtual ~CustomStatefulOp()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_ab25d24f86446a2f6d3aa4da85f869472"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#ab25d24f86446a2f6d3aa4da85f869472">mxnet::ext::CustomPartitioner::supported_map</a></div><div class="ttdeci">std::map< std::string, supportedOps_t > supported_map</div><div class="ttdef"><b>Definition:</b> lib_api.h:888</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573aea5cdddfa3ac11130048ad8e2a012363"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573aea5cdddfa3ac11130048ad8e2a012363">kDLExtDev</a></div><div class="ttdeci">@ kDLExtDev</div><div class="ttdoc">Reserved extension device type, used for quickly test extension device The semantics can differ depen...</div><div class="ttdef"><b>Definition:</b> lib_api.h:126</div></div> |
| <div class="ttc" id="alib__api_8h_html_ab3a2f05b845f641f30f417756dc64fa0"><div class="ttname"><a href="lib__api_8h.html#ab3a2f05b845f641f30f417756dc64fa0">_msgSize</a></div><div class="ttdeci">MX_INT_RET _msgSize()</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_a34340edf7aaefd434e9be4d9e7ce03dd"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#a34340edf7aaefd434e9be4d9e7ce03dd">mxnet::ext::CustomOp::forward_fp</a></div><div class="ttdeci">std::vector< fcomp_t > forward_fp</div><div class="ttdef"><b>Definition:</b> lib_api.h:816</div></div> |
| <div class="ttc" id="alib__api_8h_html_ade85fac9a80edcdf9e6062cb079cd325"><div class="ttname"><a href="lib__api_8h.html#ade85fac9a80edcdf9e6062cb079cd325">_opVersion</a></div><div class="ttdeci">MX_INT_RET _opVersion()</div><div class="ttdoc">returns MXNet library version</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869">mxnet::ext::kInt64</a></div><div class="ttdeci">@ kInt64</div><div class="ttdef"><b>Definition:</b> lib_api.h:256</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a10b027e352f8fd55d9639398807614bd"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a10b027e352f8fd55d9639398807614bd">mxnet::ext::opRegGet_t</a></div><div class="ttdeci">int(* opRegGet_t)(int idx, const char **name, int *isSGop, const char ***forward_ctx, mxnet::ext::fcomp_t **forward_fp, int *forward_count, const char ***backward_ctx, mxnet::ext::fcomp_t **backward_fp, int *backward_count, const char ***create_op_ctx, mxnet::ext::createOpState_t **create_op_fp, int *create_op_count, mxnet::ext::parseAttrs_t *parse, mxnet::ext::inferType_t *type, mxnet::ext::inferSType_t *stype, mxnet::ext::inferShape_t *shape, mxnet::ext::mutateInputs_t *mutate)</div><div class="ttdef"><b>Definition:</b> lib_api.h:983</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1MXSparse_html_a5df6670c7736ef03a71aedf9c044371f"><div class="ttname"><a href="structmxnet_1_1ext_1_1MXSparse.html#a5df6670c7736ef03a71aedf9c044371f">mxnet::ext::MXSparse::indices_len</a></div><div class="ttdeci">int64_t indices_len</div><div class="ttdef"><b>Definition:</b> lib_api.h:306</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573a61cf710df30b59b43a5e34e37ecd1973"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a61cf710df30b59b43a5e34e37ecd1973">kDLCPUPinned</a></div><div class="ttdeci">@ kDLCPUPinned</div><div class="ttdoc">Pinned CUDA GPU device by cudaMallocHost.</div><div class="ttdef"><b>Definition:</b> lib_api.h:110</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomStatefulOpWrapper_html_aad23e82ea3f745ac3a92c07b6afed1d2"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#aad23e82ea3f745ac3a92c07b6afed1d2">mxnet::ext::CustomStatefulOpWrapper::CustomStatefulOpWrapper</a></div><div class="ttdeci">CustomStatefulOpWrapper(CustomStatefulOp *inst, opCallDestroyOpState_t destroy)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1285</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1">mxnet::ext::NUM</a></div><div class="ttdeci">@ NUM</div><div class="ttdef"><b>Definition:</b> lib_api.h:525</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8">mxnet::ext::kInt32</a></div><div class="ttdeci">@ kInt32</div><div class="ttdef"><b>Definition:</b> lib_api.h:254</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Registry_html_a1a7613063268d9ff754bd3f53cdb35d6"><div class="ttname"><a href="classmxnet_1_1ext_1_1Registry.html#a1a7613063268d9ff754bd3f53cdb35d6">mxnet::ext::Registry::size</a></div><div class="ttdeci">int size()</div><div class="ttdef"><b>Definition:</b> lib_api.h:921</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241">mxnet::ext::MAP</a></div><div class="ttdeci">@ MAP</div><div class="ttdef"><b>Definition:</b> lib_api.h:525</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Graph_html_a44c1ad9a070013bdf4390b0894d9294d"><div class="ttname"><a href="classmxnet_1_1ext_1_1Graph.html#a44c1ad9a070013bdf4390b0894d9294d">mxnet::ext::Graph::inputs</a></div><div class="ttdeci">std::vector< Node * > inputs</div><div class="ttdef"><b>Definition:</b> lib_api.h:665</div></div> |
| <div class="ttc" id="alib__api_8h_html_a49db572e18d84c47738fc746568fb573a31aefcffd60630580d74bf055c2c3ba0"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a31aefcffd60630580d74bf055c2c3ba0">kDLVulkan</a></div><div class="ttdeci">@ kDLVulkan</div><div class="ttdoc">Vulkan buffer for next generation graphics.</div><div class="ttdef"><b>Definition:</b> lib_api.h:114</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomPartitioner_html_a5a0891b0fffc7891d263b4850d96560b"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomPartitioner.html#a5a0891b0fffc7891d263b4850d96560b">mxnet::ext::CustomPartitioner::getCreateSelector</a></div><div class="ttdeci">createSelector_t getCreateSelector(int stg_id)</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_ad79465455155119203c88a45547f7b95"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#ad79465455155119203c88a45547f7b95">mxnet::ext::CustomOp::setInferType</a></div><div class="ttdeci">CustomOp & setInferType(inferType_t func)</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_ab867c91f53d13cb949e9c9a579fa3da5"><div class="ttname"><a href="namespacemxnet_1_1ext.html#ab867c91f53d13cb949e9c9a579fa3da5">mxnet::ext::opCallMutateInputs_t</a></div><div class="ttdeci">int(* opCallMutateInputs_t)(mutateInputs_t mutate, const char *const *keys, const char *const *vals, int num, int **mutate_indices, int *indices_size)</div><div class="ttdef"><b>Definition:</b> lib_api.h:1088</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1Node_html_aee95220e1a30197ca5fa726742f57183"><div class="ttname"><a href="classmxnet_1_1ext_1_1Node.html#aee95220e1a30197ca5fa726742f57183">mxnet::ext::Node::outputs</a></div><div class="ttdeci">std::vector< NodeEntry > outputs</div><div class="ttdef"><b>Definition:</b> lib_api.h:601</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_af5a3b01d85dcfa4765838c1b38f414e9"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#af5a3b01d85dcfa4765838c1b38f414e9">mxnet::ext::JsonVal::str</a></div><div class="ttdeci">std::string str</div><div class="ttdef"><b>Definition:</b> lib_api.h:566</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6">mxnet::ext::kInt8</a></div><div class="ttdeci">@ kInt8</div><div class="ttdef"><b>Definition:</b> lib_api.h:255</div></div> |
| <div class="ttc" id="aclassmxnet_1_1ext_1_1CustomOp_html_ae69f99666f2e874f57c41fa3e823491b"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomOp.html#ae69f99666f2e874f57c41fa3e823491b">mxnet::ext::CustomOp::mapToVector</a></div><div class="ttdeci">void mapToVector()</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1ext_html_af05bbe0926a318bb3674c8db6d18039a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a></div><div class="ttdeci">MXReturnValue(* inferSType_t)(const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *in_storage_types, std::vector< int > *out_storage_types)</div><div class="ttdef"><b>Definition:</b> lib_api.h:758</div></div> |
| <div class="ttc" id="astructmxnet_1_1ext_1_1JsonVal_html_a65a765a25fdbaa5ac44cd42c8a848c44"><div class="ttname"><a href="structmxnet_1_1ext_1_1JsonVal.html#a65a765a25fdbaa5ac44cd42c8a848c44">mxnet::ext::JsonVal::parse_list</a></div><div class="ttdeci">static JsonVal parse_list(const std::string &json, unsigned int *idx)</div></div> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Jan 5 2023 03:47:39 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.17 |
| </small></address> |
| </body> |
| </html> |
