| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.13"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>mxnet: /work/mxnet/include/mxnet/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.13 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| </script> |
| <div id="main-nav"></div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_d44c64559bbebec7f509842c48db8b23.html">include</a></li><li class="navelem"><a class="el" href="dir_1143c7affb9ebd026cb6818dd282def7.html">mxnet</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">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>* <span class="keyword">get</span>();</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* <span class="keyword">get</span>(<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">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">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">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> CPU();</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> GPU();</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> CPU(<span class="keywordtype">int</span> dev_id);</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> GPU(<span class="keywordtype">int</span> dev_id);</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> *data{<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* indptr = <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> <span class="keyword">set</span>(<span class="keywordtype">void</span> *data_ptr, <span class="keyword">const</span> int64_t* dims, <span class="keywordtype">int</span> ndims, <span class="keywordtype">void</span> *idx,</div><div class="line"><a name="l00314"></a><span class="lineno"> 314</span>  int64_t num_idx, <span class="keywordtype">void</span> *idx_ptr = <span class="keyword">nullptr</span>, int64_t num_idx_ptr = 0);</div><div class="line"><a name="l00315"></a><span class="lineno"> 315</span> };</div><div class="line"><a name="l00316"></a><span class="lineno"> 316</span> </div><div class="line"><a name="l00320"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html"> 320</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="l00321"></a><span class="lineno"> 321</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>();</div><div class="line"><a name="l00322"></a><span class="lineno"> 322</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">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="l00323"></a><span class="lineno"> 323</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>(<span class="keywordtype">void</span> *data_ptr, std::vector<int64_t> shape, <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> dtype,</div><div class="line"><a name="l00324"></a><span class="lineno"> 324</span>  <span class="keywordtype">size_t</span> vID, <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> mx_ctx, <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> stype = <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e">kDefaultStorage</a>);</div><div class="line"><a name="l00325"></a><span class="lineno"> 325</span> </div><div class="line"><a name="l00327"></a><span class="lineno"> 327</span>  <span class="keywordtype">void</span> setTensor(<span class="keywordtype">void</span> *dptr, <a class="code" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> type, <span class="keyword">const</span> int64_t* dims, <span class="keywordtype">int</span> ndims,</div><div class="line"><a name="l00328"></a><span class="lineno"> 328</span>  <span class="keywordtype">size_t</span> vID, <a class="code" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> mx_ctx, <a class="code" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> storage_type);</div><div class="line"><a name="l00329"></a><span class="lineno"> 329</span> </div><div class="line"><a name="l00331"></a><span class="lineno"> 331</span>  <span class="keywordtype">void</span> setDLTensor();</div><div class="line"><a name="l00332"></a><span class="lineno"> 332</span> </div><div class="line"><a name="l00334"></a><span class="lineno"> 334</span>  <span class="keyword">template</span><<span class="keyword">typename</span> data_type></div><div class="line"><a name="l00335"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a4b16f5086857cf3dd983f9ef2b96c204"> 335</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="l00336"></a><span class="lineno"> 336</span>  <span class="keywordflow">return</span> <span class="keyword">reinterpret_cast<</span>data_type*<span class="keyword">></span>(data_ptr);</div><div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  }</div><div class="line"><a name="l00338"></a><span class="lineno"> 338</span> </div><div class="line"><a name="l00340"></a><span class="lineno"> 340</span>  int64_t <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a416e02951e24afc2792816f54de16320">size</a>() <span class="keyword">const</span>;</div><div class="line"><a name="l00341"></a><span class="lineno"> 341</span> </div><div class="line"><a name="l00343"></a><span class="lineno"> 343</span>  <span class="keywordtype">bool</span> isSame(<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="l00344"></a><span class="lineno"> 344</span> </div><div class="line"><a name="l00345"></a><span class="lineno"> 345</span>  <span class="comment">// For dense, data_ptr points to 1D flattened tensor data</span></div><div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  <span class="comment">// For sparse, data_ptr points to MXSparse</span></div><div class="line"><a name="l00347"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a3d1d1a007585a61bc8e53ebaa2e61480"> 347</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="l00348"></a><span class="lineno"> 348</span> </div><div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  <span class="comment">// shape is in [2,3,4] format to represent high-dim tensor</span></div><div class="line"><a name="l00350"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#ae2d186fe466cd35bc4b77e038106b80d"> 350</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="l00351"></a><span class="lineno"> 351</span> </div><div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  <span class="comment">// type can only be MXDType enum types</span></div><div class="line"><a name="l00353"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#ad306b093c8348887640e51e040c0748f"> 353</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="l00354"></a><span class="lineno"> 354</span> </div><div class="line"><a name="l00355"></a><span class="lineno"> 355</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="l00356"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a7d51476c0d999dc545aaa4e729d2f841"> 356</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="l00357"></a><span class="lineno"> 357</span> </div><div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  <span class="comment">// context of MXTensor representing which device the tensor data is located</span></div><div class="line"><a name="l00359"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a634c4e6805c4928fcbbf859306ce1fee"> 359</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="l00360"></a><span class="lineno"> 360</span> </div><div class="line"><a name="l00361"></a><span class="lineno"> 361</span>  <span class="comment">// corresponding DLTensor repr of MXTensor</span></div><div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  <span class="comment">// easy way to reuse functions taking DLTensor</span></div><div class="line"><a name="l00363"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#af7369ec802b04819c641334e521d7dae"> 363</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="l00364"></a><span class="lineno"> 364</span> </div><div class="line"><a name="l00365"></a><span class="lineno"> 365</span>  <span class="comment">// storage type</span></div><div class="line"><a name="l00366"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1MXTensor.html#a38959dfb4fd816f407c76a7285639758"> 366</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="l00367"></a><span class="lineno"> 367</span> };</div><div class="line"><a name="l00368"></a><span class="lineno"> 368</span> </div><div class="line"><a name="l00370"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760"> 370</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="l00372"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e"> 372</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="l00374"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917"> 374</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, <span class="keyword">const</span> int64_t* shapes, <span class="keywordtype">int</span> num_shapes,</div><div class="line"><a name="l00375"></a><span class="lineno"> 375</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* dev_str, <span class="keywordtype">int</span> dev_id, <span class="keywordtype">int</span> dtype, <span class="keyword">const</span> <span class="keywordtype">char</span>* name,</div><div class="line"><a name="l00376"></a><span class="lineno"> 376</span>  <span class="keywordtype">int</span> isArg, <span class="keywordtype">void</span>** data);</div><div class="line"><a name="l00378"></a><span class="lineno"> 378</span> <span class="preprocessor">#if defined(__NVCC__)</span></div><div class="line"><a name="l00379"></a><span class="lineno"> 379</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="l00380"></a><span class="lineno"> 380</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="l00381"></a><span class="lineno"> 381</span> <span class="preprocessor">#else</span></div><div class="line"><a name="l00382"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752"> 382</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="l00383"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13"> 383</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="l00384"></a><span class="lineno"> 384</span> <span class="preprocessor">#endif</span></div><div class="line"><a name="l00385"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8"> 385</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="l00386"></a><span class="lineno"> 386</span> </div><div class="line"><a name="l00388"></a><span class="lineno"> 388</span> <span class="comment">/* Each thread should generate random number unique sequence out of different states */</span></div><div class="line"><a name="l00389"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a3bab156d26b9eface536d0c5d03d9b52"> 389</a></span> <span class="preprocessor">#define MX_NUM_CPU_RANDOM_STATES 1024</span></div><div class="line"><a name="l00390"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aa0ac2ed4c5f2ee26d5c748aee108e3cd"> 390</a></span> <span class="preprocessor">#define MX_NUM_GPU_RANDOM_STATES 32768</span></div><div class="line"><a name="l00391"></a><span class="lineno"> 391</span> </div><div class="line"><a name="l00392"></a><span class="lineno"> 392</span> <span class="comment">/* \brief Class to help allocate new args/aux params in graph passes */</span></div><div class="line"><a name="l00393"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1PassResource.html"> 393</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="l00394"></a><span class="lineno"> 394</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  <a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>(std::unordered_map<std::string, MXTensor>* new_args,</div><div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  std::unordered_map<std::string, MXTensor>* new_aux,</div><div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc, <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc);</div><div class="line"><a name="l00398"></a><span class="lineno"> 398</span> </div><div class="line"><a name="l00399"></a><span class="lineno"> 399</span>  <span class="comment">// allocate new arg param, adds to args map, returns newly allocated tensor</span></div><div class="line"><a name="l00400"></a><span class="lineno"> 400</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>* alloc_arg(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> std::vector<int64_t>& shapes,</div><div class="line"><a name="l00401"></a><span class="lineno"> 401</span>  <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) <span class="keyword">const</span>;</div><div class="line"><a name="l00402"></a><span class="lineno"> 402</span> </div><div class="line"><a name="l00403"></a><span class="lineno"> 403</span>  <span class="comment">// allocate new aux param, adds to aux map, returns newly allocated tensor</span></div><div class="line"><a name="l00404"></a><span class="lineno"> 404</span>  <a class="code" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a>* alloc_aux(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> std::vector<int64_t>& shapes,</div><div class="line"><a name="l00405"></a><span class="lineno"> 405</span>  <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) <span class="keyword">const</span>;</div><div class="line"><a name="l00406"></a><span class="lineno"> 406</span> </div><div class="line"><a name="l00407"></a><span class="lineno"> 407</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00408"></a><span class="lineno"> 408</span>  std::unordered_map<std::string, MXTensor>* new_args_;</div><div class="line"><a name="l00409"></a><span class="lineno"> 409</span>  std::unordered_map<std::string, MXTensor>* new_aux_;</div><div class="line"><a name="l00410"></a><span class="lineno"> 410</span>  <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc_;</div><div class="line"><a name="l00411"></a><span class="lineno"> 411</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc_;</div><div class="line"><a name="l00412"></a><span class="lineno"> 412</span> };</div><div class="line"><a name="l00413"></a><span class="lineno"> 413</span> </div><div class="line"><a name="l00417"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1OpResource.html"> 417</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="l00418"></a><span class="lineno"> 418</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  <a class="code" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a>(<a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc_fp, <span class="keywordtype">void</span>* cpu_alloc_fp,</div><div class="line"><a name="l00420"></a><span class="lineno"> 420</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc_fp, <span class="keywordtype">void</span>* gpu_alloc_fp, <span class="keywordtype">void</span>* stream,</div><div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc_fp, <span class="keywordtype">void</span>* sparse_alloc_fp,</div><div class="line"><a name="l00422"></a><span class="lineno"> 422</span>  <span class="keywordtype">void</span>* rng_cpu_states, <span class="keywordtype">void</span>* rng_gpu_states);</div><div class="line"><a name="l00423"></a><span class="lineno"> 423</span> </div><div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  <span class="keywordtype">void</span>* alloc_cpu(<span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a416e02951e24afc2792816f54de16320">size</a>) <span class="keyword">const</span>;</div><div class="line"><a name="l00426"></a><span class="lineno"> 426</span> </div><div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="keywordtype">void</span>* alloc_gpu(<span class="keywordtype">int</span> size) <span class="keyword">const</span>;</div><div class="line"><a name="l00429"></a><span class="lineno"> 429</span> </div><div class="line"><a name="l00431"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1OpResource.html#a712e78a6c50beca023bc20064688bd1f"> 431</a></span>  <span class="keyword">inline</span> mx_stream_t <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="l00432"></a><span class="lineno"> 432</span>  <span class="keywordflow">return</span> <span class="keyword">static_cast<</span>mx_stream_t<span class="keyword">></span>(cuda_stream);</div><div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  }</div><div class="line"><a name="l00434"></a><span class="lineno"> 434</span> </div><div class="line"><a name="l00436"></a><span class="lineno"> 436</span>  <span class="keywordtype">void</span> alloc_sparse(<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="l00437"></a><span class="lineno"> 437</span> </div><div class="line"><a name="l00439"></a><span class="lineno"> 439</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="l00440"></a><span class="lineno"> 440</span>  mx_cpu_rand_t* get_cpu_rand_states() <span class="keyword">const</span>;</div><div class="line"><a name="l00441"></a><span class="lineno"> 441</span> </div><div class="line"><a name="l00443"></a><span class="lineno"> 443</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="l00444"></a><span class="lineno"> 444</span>  <span class="comment">/* Note that if you are using cpu build, it will return a nullptr */</span></div><div class="line"><a name="l00445"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1OpResource.html#a3c75f8ca600930d970e3ea20cd624884"> 445</a></span>  <span class="keyword">inline</span> mx_gpu_rand_t* <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="l00446"></a><span class="lineno"> 446</span>  <span class="keywordflow">return</span> <span class="keyword">static_cast<</span>mx_gpu_rand_t*<span class="keyword">></span>(rand_gpu_states);</div><div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  }</div><div class="line"><a name="l00448"></a><span class="lineno"> 448</span> </div><div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, gpu_malloc;</div><div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  <span class="keywordtype">void</span> *cpu_alloc, *gpu_alloc;</div><div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keywordtype">void</span> *cuda_stream;</div><div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc;</div><div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  <span class="keywordtype">void</span> *sparse_alloc;</div><div class="line"><a name="l00461"></a><span class="lineno"> 461</span>  <span class="keywordtype">void</span> *rand_cpu_states, *rand_gpu_states;</div><div class="line"><a name="l00462"></a><span class="lineno"> 462</span> };</div><div class="line"><a name="l00463"></a><span class="lineno"> 463</span> </div><div class="line"><a name="l00465"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ab5f044e03093ffae2626b35fa17f610d"> 465</a></span> <span class="preprocessor">#define MX_STR_SUBGRAPH_SYM_JSON "subgraph_sym_json"</span></div><div class="line"><a name="l00466"></a><span class="lineno"> 466</span> </div><div class="line"><a name="l00467"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a73b8e3f85f6f0aba1c0fd0917773b24f"> 467</a></span> <span class="preprocessor">#define MX_STR_DTYPE "__ext_dtype__"</span></div><div class="line"><a name="l00468"></a><span class="lineno"> 468</span> </div><div class="line"><a name="l00469"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ab031cafe8f2e358de3c1cadc57a346f3"> 469</a></span> <span class="preprocessor">#define MX_STR_SHAPE "__ext_shape__"</span></div><div class="line"><a name="l00470"></a><span class="lineno"> 470</span> </div><div class="line"><a name="l00471"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a633f26064a8c0d69aa833175a66daa2e"> 471</a></span> <span class="preprocessor">#define MX_STR_EXTRA_INPUTS "__ext_extra_inputs__"</span></div><div class="line"><a name="l00472"></a><span class="lineno"> 472</span> </div><div class="line"><a name="l00473"></a><span class="lineno"> 473</span> <span class="comment">/* \brief get shape value from list of shapes string</span></div><div class="line"><a name="l00474"></a><span class="lineno"> 474</span> <span class="comment"> *</span></div><div class="line"><a name="l00475"></a><span class="lineno"> 475</span> <span class="comment"> * Examples:</span></div><div class="line"><a name="l00476"></a><span class="lineno"> 476</span> <span class="comment"> *</span></div><div class="line"><a name="l00477"></a><span class="lineno"> 477</span> <span class="comment"> * getShapeAt("[[1]]", 0) returns "[1]"</span></div><div class="line"><a name="l00478"></a><span class="lineno"> 478</span> <span class="comment"> * getShapeAt("[[1],[2,3]]", 1) returns "[2,3]"</span></div><div class="line"><a name="l00479"></a><span class="lineno"> 479</span> <span class="comment"> */</span></div><div class="line"><a name="l00480"></a><span class="lineno"> 480</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="l00481"></a><span class="lineno"> 481</span> </div><div class="line"><a name="l00482"></a><span class="lineno"> 482</span> <span class="comment">/* \brief get dtype value from list of dtypes string</span></div><div class="line"><a name="l00483"></a><span class="lineno"> 483</span> <span class="comment"> *</span></div><div class="line"><a name="l00484"></a><span class="lineno"> 484</span> <span class="comment"> * Examples:</span></div><div class="line"><a name="l00485"></a><span class="lineno"> 485</span> <span class="comment"> *</span></div><div class="line"><a name="l00486"></a><span class="lineno"> 486</span> <span class="comment"> * getDtypeAt("[1]", 0) returns "1"</span></div><div class="line"><a name="l00487"></a><span class="lineno"> 487</span> <span class="comment"> * getDtypeAt("[1,2]", 1) returns "2" </span></div><div class="line"><a name="l00488"></a><span class="lineno"> 488</span> <span class="comment"> */</span></div><div class="line"><a name="l00489"></a><span class="lineno"> 489</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="l00490"></a><span class="lineno"> 490</span> </div><div class="line"><a name="l00495"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9"> 495</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="l00496"></a><span class="lineno"> 496</span> </div><div class="line"><a name="l00498"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html"> 498</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="l00499"></a><span class="lineno"> 499</span>  <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>(); <span class="comment">// default constructor</span></div><div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  <span class="comment">// construct a JSON object by type</span></div><div class="line"><a name="l00501"></a><span class="lineno"> 501</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>(<a class="code" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">JsonType</a> t);</div><div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  <span class="comment">// construct a string JSON object</span></div><div class="line"><a name="l00503"></a><span class="lineno"> 503</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>(std::string s);</div><div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  <span class="comment">// construct a number JSON object</span></div><div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  <span class="keyword">explicit</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>(<span class="keywordtype">int</span> n);</div><div class="line"><a name="l00506"></a><span class="lineno"> 506</span>  <span class="comment">// complex constructor</span></div><div class="line"><a name="l00507"></a><span class="lineno"> 507</span>  <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">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="l00508"></a><span class="lineno"> 508</span>  <span class="keywordtype">bool</span> operator<(<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="l00509"></a><span class="lineno"> 509</span> </div><div class="line"><a name="l00510"></a><span class="lineno"> 510</span>  <span class="comment">// convert JSON object back to JSON-compatible string</span></div><div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  std::string dump() <span class="keyword">const</span>;</div><div class="line"><a name="l00512"></a><span class="lineno"> 512</span> </div><div class="line"><a name="l00513"></a><span class="lineno"> 513</span>  <span class="comment">// convert JSON-compatible string to JSON object</span></div><div class="line"><a name="l00514"></a><span class="lineno"> 514</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> parse(<span class="keyword">const</span> std::string& json);</div><div class="line"><a name="l00515"></a><span class="lineno"> 515</span> </div><div class="line"><a name="l00516"></a><span class="lineno"> 516</span>  <span class="comment">// parse a string JSON object</span></div><div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> parse_string(<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="l00518"></a><span class="lineno"> 518</span> </div><div class="line"><a name="l00519"></a><span class="lineno"> 519</span>  <span class="comment">// parse a number JSON object</span></div><div class="line"><a name="l00520"></a><span class="lineno"> 520</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> parse_num(<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="l00521"></a><span class="lineno"> 521</span> </div><div class="line"><a name="l00522"></a><span class="lineno"> 522</span>  <span class="comment">// parse a list of JSON objects</span></div><div class="line"><a name="l00523"></a><span class="lineno"> 523</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> parse_list(<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="l00524"></a><span class="lineno"> 524</span> </div><div class="line"><a name="l00525"></a><span class="lineno"> 525</span>  <span class="comment">// parse a map of JSON objects</span></div><div class="line"><a name="l00526"></a><span class="lineno"> 526</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> parse_map(<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="l00527"></a><span class="lineno"> 527</span> </div><div class="line"><a name="l00528"></a><span class="lineno"> 528</span>  <span class="comment">// generic parse function</span></div><div class="line"><a name="l00529"></a><span class="lineno"> 529</span>  <span class="keyword">static</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> parse(<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="l00530"></a><span class="lineno"> 530</span> </div><div class="line"><a name="l00531"></a><span class="lineno"> 531</span>  <span class="comment">// debug function to convert data structure to a debugstring</span></div><div class="line"><a name="l00532"></a><span class="lineno"> 532</span>  std::string toString() <span class="keyword">const</span>;</div><div class="line"><a name="l00533"></a><span class="lineno"> 533</span> </div><div class="line"><a name="l00534"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#a9f826ae984fc3f102c1bb6761f3f23ee"> 534</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="l00535"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#a1d9e77f9eb4ab27cdcffb85495a760ae"> 535</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="l00536"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#af5a3b01d85dcfa4765838c1b38f414e9"> 536</a></span>  std::string <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#af5a3b01d85dcfa4765838c1b38f414e9">str</a>;</div><div class="line"><a name="l00537"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#af5d2babd1cb84d40ad07585cac9c5970"> 537</a></span>  std::vector<JsonVal> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html#af5d2babd1cb84d40ad07585cac9c5970">list</a>;</div><div class="line"><a name="l00538"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1JsonVal.html#aae3712b54e6f015df3b75b2827da1adc"> 538</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="l00539"></a><span class="lineno"> 539</span> };</div><div class="line"><a name="l00540"></a><span class="lineno"> 540</span> </div><div class="line"><a name="l00544"></a><span class="lineno"> 544</span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>;</div><div class="line"><a name="l00545"></a><span class="lineno"> 545</span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>;</div><div class="line"><a name="l00546"></a><span class="lineno"> 546</span> </div><div class="line"><a name="l00547"></a><span class="lineno"> 547</span> <span class="comment">// Representation of an input/output to a node</span></div><div class="line"><a name="l00548"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1NodeEntry.html"> 548</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="l00549"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1NodeEntry.html#a9ccf74d370d36c4ed911e94f84a431be"> 549</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="l00550"></a><span class="lineno"><a class="line" href="structmxnet_1_1ext_1_1NodeEntry.html#ac3d191e6775d212446933fc94ca21624"> 550</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="l00551"></a><span class="lineno"> 551</span> };</div><div class="line"><a name="l00552"></a><span class="lineno"> 552</span> </div><div class="line"><a name="l00553"></a><span class="lineno"> 553</span> <span class="comment">// Representation of a node in the graph</span></div><div class="line"><a name="l00554"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html"> 554</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="l00555"></a><span class="lineno"> 555</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00556"></a><span class="lineno"> 556</span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>();</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">// internally set passResource to enable tensor allocation for graph passes</span></div><div class="line"><a name="l00559"></a><span class="lineno"> 559</span>  <span class="keywordtype">void</span> _setPassResource(<a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res_);</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">/* \brief allocate an arg tensor for this node */</span></div><div class="line"><a name="l00562"></a><span class="lineno"> 562</span>  <span class="keywordtype">void</span> alloc_arg(<span class="keyword">const</span> std::vector<int64_t>& shapes,</div><div class="line"><a name="l00563"></a><span class="lineno"> 563</span>  <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="l00564"></a><span class="lineno"> 564</span> </div><div class="line"><a name="l00565"></a><span class="lineno"> 565</span>  <span class="comment">/* \brief allocate an aux tensor for this node */</span></div><div class="line"><a name="l00566"></a><span class="lineno"> 566</span>  <span class="keywordtype">void</span> alloc_aux(<span class="keyword">const</span> std::vector<int64_t>& shapes,</div><div class="line"><a name="l00567"></a><span class="lineno"> 567</span>  <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="l00568"></a><span class="lineno"> 568</span> </div><div class="line"><a name="l00569"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a20c7b312f050f3a1cc606e924482a4c7"> 569</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="l00570"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#abc2b3e8b645050faf1a4534de694e939"> 570</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="l00571"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a3489d59197ad9612dbdc23f963aa8d02"> 571</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="l00572"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a8364c3b8ae3800c775a23a8e90984c82"> 572</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="l00573"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#aee95220e1a30197ca5fa726742f57183"> 573</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="l00574"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#afbe960c8b040871dfb9f8c8bdb529947"> 574</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="l00575"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Node.html#a3e1851c411ee0112de6ee3f1a90d2151"> 575</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="l00576"></a><span class="lineno"> 576</span> </div><div class="line"><a name="l00577"></a><span class="lineno"> 577</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00578"></a><span class="lineno"> 578</span>  <a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res;</div><div class="line"><a name="l00579"></a><span class="lineno"> 579</span> };</div><div class="line"><a name="l00580"></a><span class="lineno"> 580</span> </div><div class="line"><a name="l00581"></a><span class="lineno"> 581</span> <span class="comment">// Representation of the graph</span></div><div class="line"><a name="l00582"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html"> 582</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="l00583"></a><span class="lineno"> 583</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00584"></a><span class="lineno"> 584</span>  <a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>();</div><div class="line"><a name="l00585"></a><span class="lineno"> 585</span> </div><div class="line"><a name="l00586"></a><span class="lineno"> 586</span>  <span class="comment">/* \brief deleted nodes when deleting the graph */</span></div><div class="line"><a name="l00587"></a><span class="lineno"> 587</span>  ~<a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>();</div><div class="line"><a name="l00588"></a><span class="lineno"> 588</span> </div><div class="line"><a name="l00589"></a><span class="lineno"> 589</span>  <span class="comment">/* \brief create a graph object from an unparsed string */</span></div><div class="line"><a name="l00590"></a><span class="lineno"> 590</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>* fromString(<span class="keyword">const</span> std::string& json);</div><div class="line"><a name="l00591"></a><span class="lineno"> 591</span> </div><div class="line"><a name="l00592"></a><span class="lineno"> 592</span>  <span class="comment">/* \brief create a graph object from a parsed JSON object */</span></div><div class="line"><a name="l00593"></a><span class="lineno"> 593</span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">Graph</a>* fromJson(<a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> val);</div><div class="line"><a name="l00594"></a><span class="lineno"> 594</span> </div><div class="line"><a name="l00595"></a><span class="lineno"> 595</span>  <span class="comment">/* \brief convert graph object back to JSON object */</span></div><div class="line"><a name="l00596"></a><span class="lineno"> 596</span>  <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a> toJson() <span class="keyword">const</span>;</div><div class="line"><a name="l00597"></a><span class="lineno"> 597</span> </div><div class="line"><a name="l00598"></a><span class="lineno"> 598</span>  <span class="comment">/* \brief convert graph object to JSON string */</span></div><div class="line"><a name="l00599"></a><span class="lineno"> 599</span>  std::string toString() <span class="keyword">const</span>;</div><div class="line"><a name="l00600"></a><span class="lineno"> 600</span> </div><div class="line"><a name="l00601"></a><span class="lineno"> 601</span>  <span class="comment">/* \brief visits a node "n" */</span></div><div class="line"><a name="l00602"></a><span class="lineno"> 602</span>  <span class="keywordtype">void</span> _dfs_util(<a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* n, std::unordered_set<Node*>* to_visit,</div><div class="line"><a name="l00603"></a><span class="lineno"> 603</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="l00604"></a><span class="lineno"> 604</span> </div><div class="line"><a name="l00605"></a><span class="lineno"> 605</span>  <span class="comment">/* \brief post-order DFS graph traversal */</span></div><div class="line"><a name="l00606"></a><span class="lineno"> 606</span>  <span class="keywordtype">void</span> DFS(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="l00607"></a><span class="lineno"> 607</span> </div><div class="line"><a name="l00608"></a><span class="lineno"> 608</span>  <span class="comment">/* \brief sort graph nodes in topological order */</span></div><div class="line"><a name="l00609"></a><span class="lineno"> 609</span>  std::vector<Node*> topological_sort() <span class="keyword">const</span>;</div><div class="line"><a name="l00610"></a><span class="lineno"> 610</span> </div><div class="line"><a name="l00611"></a><span class="lineno"> 611</span>  <span class="comment">/* \brief print out graph details */</span></div><div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  <span class="keywordtype">void</span> print(<span class="keywordtype">int</span> indent = 0) <span class="keyword">const</span>;</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 add a new node to this graph */</span></div><div class="line"><a name="l00615"></a><span class="lineno"> 615</span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* addNode(<span class="keyword">const</span> std::string& name, <span class="keyword">const</span> std::string& op);</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 get node at index in graph */</span></div><div class="line"><a name="l00618"></a><span class="lineno"> 618</span>  <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* getNode(<span class="keywordtype">size_t</span> idx);</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 get const node at index in const graph */</span></div><div class="line"><a name="l00621"></a><span class="lineno"> 621</span>  <span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Node.html">Node</a>* getNode(<span class="keywordtype">size_t</span> idx) <span class="keyword">const</span>;</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 get attribute on graph */</span></div><div class="line"><a name="l00624"></a><span class="lineno"> 624</span>  <span class="keyword">const</span> <a class="code" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a>& getAttr(<span class="keyword">const</span> std::string& key) <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 get number of nodes in the graph */</span></div><div class="line"><a name="l00627"></a><span class="lineno"> 627</span>  <span class="keywordtype">size_t</span> <a class="code" href="classmxnet_1_1ext_1_1MXerrorMsgs.html#a416e02951e24afc2792816f54de16320">size</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">// internally set passResource to enable tensor allocation for graph passes</span></div><div class="line"><a name="l00630"></a><span class="lineno"> 630</span>  <span class="keywordtype">void</span> _setPassResource(<a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res_);</div><div class="line"><a name="l00631"></a><span class="lineno"> 631</span> </div><div class="line"><a name="l00632"></a><span class="lineno"> 632</span>  <span class="comment">// internally set arg/aux params when available</span></div><div class="line"><a name="l00633"></a><span class="lineno"> 633</span>  <span class="keywordtype">void</span> _setParams(std::unordered_map<std::string, mxnet::ext::MXTensor>* args,</div><div class="line"><a name="l00634"></a><span class="lineno"> 634</span>  std::unordered_map<std::string, mxnet::ext::MXTensor>* aux);</div><div class="line"><a name="l00635"></a><span class="lineno"> 635</span> </div><div class="line"><a name="l00636"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html#a44c1ad9a070013bdf4390b0894d9294d"> 636</a></span>  std::vector<Node*> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#a44c1ad9a070013bdf4390b0894d9294d">inputs</a>;</div><div class="line"><a name="l00637"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html#afda2391d5b2ae16121ab1c1bf5eb3ae3"> 637</a></span>  std::vector<NodeEntry> <a class="code" href="classmxnet_1_1ext_1_1Graph.html#afda2391d5b2ae16121ab1c1bf5eb3ae3">outputs</a>;</div><div class="line"><a name="l00638"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Graph.html#a7997e82fd30ac0817c0555fc0969e87f"> 638</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="l00639"></a><span class="lineno"> 639</span> </div><div class="line"><a name="l00640"></a><span class="lineno"> 640</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00641"></a><span class="lineno"> 641</span>  std::vector<Node*> nodes;</div><div class="line"><a name="l00642"></a><span class="lineno"> 642</span>  <a class="code" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a>* res;</div><div class="line"><a name="l00643"></a><span class="lineno"> 643</span> };</div><div class="line"><a name="l00644"></a><span class="lineno"> 644</span> </div><div class="line"><a name="l00645"></a><span class="lineno"> 645</span> <span class="comment">/* \brief An abstract class for library authors creating custom</span></div><div class="line"><a name="l00646"></a><span class="lineno"> 646</span> <span class="comment"> * partitioners. Optional, can just implement supportedOps instead</span></div><div class="line"><a name="l00647"></a><span class="lineno"> 647</span> <span class="comment"> */</span></div><div class="line"><a name="l00648"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOpSelector.html"> 648</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="l00649"></a><span class="lineno"> 649</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00650"></a><span class="lineno"> 650</span>  <span class="comment">/* \brief Select a node to include in subgraph, return true to include node</span></div><div class="line"><a name="l00651"></a><span class="lineno"> 651</span> <span class="comment"> * nodeID - index of node in graph</span></div><div class="line"><a name="l00652"></a><span class="lineno"> 652</span> <span class="comment"> */</span></div><div class="line"><a name="l00653"></a><span class="lineno"> 653</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> Select(<span class="keywordtype">int</span> nodeID) = 0;</div><div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  <span class="comment">/* \brief Select an input node from current node to include in subgraph</span></div><div class="line"><a name="l00655"></a><span class="lineno"> 655</span> <span class="comment"> * return true to include node</span></div><div class="line"><a name="l00656"></a><span class="lineno"> 656</span> <span class="comment"> * nodeID - index of node in graph</span></div><div class="line"><a name="l00657"></a><span class="lineno"> 657</span> <span class="comment"> * input_nodeID - index of input node in graph</span></div><div class="line"><a name="l00658"></a><span class="lineno"> 658</span> <span class="comment"> */</span></div><div class="line"><a name="l00659"></a><span class="lineno"> 659</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> SelectInput(<span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> input_nodeID) = 0;</div><div class="line"><a name="l00660"></a><span class="lineno"> 660</span>  <span class="comment">/* \brief Select an output node from current node to include in subgraph</span></div><div class="line"><a name="l00661"></a><span class="lineno"> 661</span> <span class="comment"> * return true to include node</span></div><div class="line"><a name="l00662"></a><span class="lineno"> 662</span> <span class="comment"> * nodeID - index of node in graph</span></div><div class="line"><a name="l00663"></a><span class="lineno"> 663</span> <span class="comment"> * output_nodeID - index of output node in graph</span></div><div class="line"><a name="l00664"></a><span class="lineno"> 664</span> <span class="comment"> */</span></div><div class="line"><a name="l00665"></a><span class="lineno"> 665</span>  <span class="keyword">virtual</span> <span class="keywordtype">bool</span> SelectOutput(<span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> output_nodeID) = 0;</div><div class="line"><a name="l00666"></a><span class="lineno"> 666</span>  <span class="comment">/* \brief Review nodes to include in subgraph</span></div><div class="line"><a name="l00667"></a><span class="lineno"> 667</span> <span class="comment"> * return set of candidate nodes to keep in subgraph</span></div><div class="line"><a name="l00668"></a><span class="lineno"> 668</span> <span class="comment"> * candidates - indices of nodes to include in subgraph</span></div><div class="line"><a name="l00669"></a><span class="lineno"> 669</span> <span class="comment"> * keep - indices of nodes to keep in subgraph</span></div><div class="line"><a name="l00670"></a><span class="lineno"> 670</span> <span class="comment"> */</span></div><div class="line"><a name="l00671"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a2e39f0a634eee550cb79b225d1beac02"> 671</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,</div><div class="line"><a name="l00672"></a><span class="lineno"> 672</span>  std::vector<int>* keep) {</div><div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  keep->insert(keep->end(), candidates.begin(), candidates.end());</div><div class="line"><a name="l00674"></a><span class="lineno"> 674</span>  }</div><div class="line"><a name="l00675"></a><span class="lineno"> 675</span>  <span class="comment">/* \brief Reset any selector state, called after growing subgraph, before filter</span></div><div class="line"><a name="l00676"></a><span class="lineno"> 676</span> <span class="comment"> * Called after finished calling SelectInput/SelectOutput and growing subgraph</span></div><div class="line"><a name="l00677"></a><span class="lineno"> 677</span> <span class="comment"> */</span></div><div class="line"><a name="l00678"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOpSelector.html#a76533a016df5e8c3e39a722f2bf48f13"> 678</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="l00679"></a><span class="lineno"> 679</span> };</div><div class="line"><a name="l00680"></a><span class="lineno"> 680</span> </div><div class="line"><a name="l00686"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html"> 686</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="l00687"></a><span class="lineno"> 687</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00688"></a><span class="lineno"> 688</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>();</div><div class="line"><a name="l00689"></a><span class="lineno"> 689</span>  <span class="keyword">virtual</span> ~<a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>();</div><div class="line"><a name="l00690"></a><span class="lineno"> 690</span> </div><div class="line"><a name="l00691"></a><span class="lineno"> 691</span>  <span class="keyword">template</span><<span class="keyword">class </span>A, <span class="keyword">typename</span> ...Ts></div><div class="line"><a name="l00692"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#aa48c6b2f073ef04c5d69060fd1aedf1d"> 692</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#aa48c6b2f073ef04c5d69060fd1aedf1d">create</a>(Ts...args) {</div><div class="line"><a name="l00693"></a><span class="lineno"> 693</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="l00694"></a><span class="lineno"> 694</span>  op->created = <span class="keyword">true</span>;</div><div class="line"><a name="l00695"></a><span class="lineno"> 695</span>  <span class="keywordflow">return</span> op;</div><div class="line"><a name="l00696"></a><span class="lineno"> 696</span>  }</div><div class="line"><a name="l00697"></a><span class="lineno"> 697</span> </div><div class="line"><a name="l00698"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a46fb6fe57019f1ef4d4a616c7ef7297f"> 698</a></span>  <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#a46fb6fe57019f1ef4d4a616c7ef7297f">wasCreated</a>() { <span class="keywordflow">return</span> created; }</div><div class="line"><a name="l00699"></a><span class="lineno"> 699</span> </div><div class="line"><a name="l00700"></a><span class="lineno"> 700</span>  <span class="keyword">virtual</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> Forward(std::vector<MXTensor>* inputs,</div><div class="line"><a name="l00701"></a><span class="lineno"> 701</span>  std::vector<MXTensor>* outputs,</div><div class="line"><a name="l00702"></a><span class="lineno"> 702</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="l00703"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#ad8758d40952666f93d644f704b686da6"> 703</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="l00704"></a><span class="lineno"> 704</span>  std::vector<MXTensor>* outputs,</div><div class="line"><a name="l00705"></a><span class="lineno"> 705</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="l00706"></a><span class="lineno"> 706</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="l00707"></a><span class="lineno"> 707</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670">MX_FAIL</a>;</div><div class="line"><a name="l00708"></a><span class="lineno"> 708</span>  }</div><div class="line"><a name="l00709"></a><span class="lineno"> 709</span> </div><div class="line"><a name="l00710"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOp.html#aca7db0bccdf4840d5cd70ff73f7624da"> 710</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="l00711"></a><span class="lineno"> 711</span> </div><div class="line"><a name="l00712"></a><span class="lineno"> 712</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00713"></a><span class="lineno"> 713</span>  <span class="keywordtype">bool</span> created;</div><div class="line"><a name="l00714"></a><span class="lineno"> 714</span> };</div><div class="line"><a name="l00715"></a><span class="lineno"> 715</span> </div><div class="line"><a name="l00717"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603"> 717</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,</div><div class="line"><a name="l00718"></a><span class="lineno"> 718</span>  std::string>& attributes,</div><div class="line"><a name="l00719"></a><span class="lineno"> 719</span>  std::vector<MXTensor>* inputs,</div><div class="line"><a name="l00720"></a><span class="lineno"> 720</span>  std::vector<MXTensor>* outputs,</div><div class="line"><a name="l00721"></a><span class="lineno"> 721</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="l00722"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc"> 722</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>)(<span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00723"></a><span class="lineno"> 723</span>  std::string>& attributes,</div><div class="line"><a name="l00724"></a><span class="lineno"> 724</span>  <span class="keywordtype">int</span>* num_inputs, <span class="keywordtype">int</span>* num_outputs);</div><div class="line"><a name="l00725"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869"> 725</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,</div><div class="line"><a name="l00726"></a><span class="lineno"> 726</span>  std::string>& attributes,</div><div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  std::vector<int>* in_types,</div><div class="line"><a name="l00728"></a><span class="lineno"> 728</span>  std::vector<int>* out_types);</div><div class="line"><a name="l00729"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a"> 729</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>)(<span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00730"></a><span class="lineno"> 730</span>  std::string>& attributes,</div><div class="line"><a name="l00731"></a><span class="lineno"> 731</span>  std::vector<int>* in_storage_types,</div><div class="line"><a name="l00732"></a><span class="lineno"> 732</span>  std::vector<int>* out_storage_types);</div><div class="line"><a name="l00733"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9"> 733</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>)(<span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  std::string>& attributes,</div><div class="line"><a name="l00735"></a><span class="lineno"> 735</span>  std::vector<std::vector<unsigned int> >* in_shapes,</div><div class="line"><a name="l00736"></a><span class="lineno"> 736</span>  std::vector<std::vector<unsigned int> >* out_shapes);</div><div class="line"><a name="l00737"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484"> 737</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>)(<span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00738"></a><span class="lineno"> 738</span>  std::string>& attributes,</div><div class="line"><a name="l00739"></a><span class="lineno"> 739</span>  std::vector<int>* input_indices);</div><div class="line"><a name="l00740"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923"> 740</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>)(<span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00741"></a><span class="lineno"> 741</span>  std::string>& attributes,</div><div class="line"><a name="l00742"></a><span class="lineno"> 742</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="l00743"></a><span class="lineno"> 743</span>  <span class="keyword">const</span> std::vector<std::vector<unsigned int> >& in_shapes,</div><div class="line"><a name="l00744"></a><span class="lineno"> 744</span>  <span class="keyword">const</span> std::vector<int> in_types,</div><div class="line"><a name="l00745"></a><span class="lineno"> 745</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>**);</div><div class="line"><a name="l00746"></a><span class="lineno"> 746</span> </div><div class="line"><a name="l00750"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html"> 750</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="l00751"></a><span class="lineno"> 751</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* op_name);</div><div class="line"><a name="l00753"></a><span class="lineno"> 753</span> </div><div class="line"><a name="l00754"></a><span class="lineno"> 754</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setForward(<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="l00755"></a><span class="lineno"> 755</span> </div><div class="line"><a name="l00756"></a><span class="lineno"> 756</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setBackward(<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="l00757"></a><span class="lineno"> 757</span> </div><div class="line"><a name="l00758"></a><span class="lineno"> 758</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setParseAttrs(<a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a> func);</div><div class="line"><a name="l00759"></a><span class="lineno"> 759</span> </div><div class="line"><a name="l00760"></a><span class="lineno"> 760</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setInferType(<a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a> func);</div><div class="line"><a name="l00761"></a><span class="lineno"> 761</span> </div><div class="line"><a name="l00762"></a><span class="lineno"> 762</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setInferSType(<a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a> func);</div><div class="line"><a name="l00763"></a><span class="lineno"> 763</span> </div><div class="line"><a name="l00764"></a><span class="lineno"> 764</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setInferShape(<a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a> func);</div><div class="line"><a name="l00765"></a><span class="lineno"> 765</span> </div><div class="line"><a name="l00766"></a><span class="lineno"> 766</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setMutateInputs(<a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a> func);</div><div class="line"><a name="l00767"></a><span class="lineno"> 767</span> </div><div class="line"><a name="l00768"></a><span class="lineno"> 768</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setCreateOpState(<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="l00769"></a><span class="lineno"> 769</span> </div><div class="line"><a name="l00770"></a><span class="lineno"> 770</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a>& setIsSubgraphOp();</div><div class="line"><a name="l00771"></a><span class="lineno"> 771</span> </div><div class="line"><a name="l00772"></a><span class="lineno"> 772</span>  <span class="keywordtype">void</span> mapToVector();</div><div class="line"><a name="l00773"></a><span class="lineno"> 773</span> </div><div class="line"><a name="l00775"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#aa1de9b3397a13c3633350307593837f3"> 775</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="l00776"></a><span class="lineno"> 776</span> </div><div class="line"><a name="l00778"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a2978383f7d86dafd36cff9eeba16c8be"> 778</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="l00779"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a07eb2f161653eed3b4332355cb050137"> 779</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="l00780"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a96338af40de6ff68da43ad5649dab18d"> 780</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="l00781"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#adb7dd13f57912db3f560c2baf223d1d7"> 781</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="l00782"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#ae25800af6998af47fe82a31474314bba"> 782</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="l00783"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a2a205d1c793a296d3a312a575e9eab76"> 783</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="l00784"></a><span class="lineno"> 784</span> </div><div class="line"><a name="l00786"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#acb5a3d13564e147807e0006c6f75eada"> 786</a></span>  std::vector<const char*> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#acb5a3d13564e147807e0006c6f75eada">forward_ctx_cstr</a>, backward_ctx_cstr, create_op_ctx_cstr;</div><div class="line"><a name="l00787"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#a34340edf7aaefd434e9be4d9e7ce03dd"> 787</a></span>  std::vector<fcomp_t> <a class="code" href="classmxnet_1_1ext_1_1CustomOp.html#a34340edf7aaefd434e9be4d9e7ce03dd">forward_fp</a>, backward_fp;</div><div class="line"><a name="l00788"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomOp.html#ad2d93199d25ceace68343f8c516bf63e"> 788</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="l00789"></a><span class="lineno"> 789</span> </div><div class="line"><a name="l00790"></a><span class="lineno"> 790</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00791"></a><span class="lineno"> 791</span>  <span class="keywordtype">void</span> raiseDuplicateContextError();</div><div class="line"><a name="l00792"></a><span class="lineno"> 792</span> </div><div class="line"><a name="l00794"></a><span class="lineno"> 794</span>  std::unordered_map<const char*, fcomp_t> forward_ctx_map, backward_ctx_map;</div><div class="line"><a name="l00795"></a><span class="lineno"> 795</span>  std::unordered_map<const char*, createOpState_t> create_op_ctx_map;</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> </div><div class="line"><a name="l00799"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d"> 799</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="l00800"></a><span class="lineno"> 800</span>  <span class="keyword">const</span> std::unordered_map<std::string, std::string>& options);</div><div class="line"><a name="l00801"></a><span class="lineno"> 801</span> </div><div class="line"><a name="l00805"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPass.html"> 805</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="l00806"></a><span class="lineno"> 806</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00807"></a><span class="lineno"> 807</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html">CustomPass</a>();</div><div class="line"><a name="l00808"></a><span class="lineno"> 808</span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html">CustomPass</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* pass_name);</div><div class="line"><a name="l00809"></a><span class="lineno"> 809</span> </div><div class="line"><a name="l00810"></a><span class="lineno"> 810</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPass.html">CustomPass</a>& setBody(<a class="code" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> fn);</div><div class="line"><a name="l00811"></a><span class="lineno"> 811</span> </div><div class="line"><a name="l00813"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPass.html#aec569fc21f800cb803d47508cd495fcb"> 813</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="l00815"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPass.html#a6dd29f51d329906280090109a634bb6a"> 815</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="l00816"></a><span class="lineno"> 816</span> };</div><div class="line"><a name="l00817"></a><span class="lineno"> 817</span> </div><div class="line"><a name="l00819"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab"> 819</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>)(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, std::vector<int>* ids,</div><div class="line"><a name="l00820"></a><span class="lineno"> 820</span>  <span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00821"></a><span class="lineno"> 821</span>  std::string>& options);</div><div class="line"><a name="l00822"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659"> 822</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>)(<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="l00823"></a><span class="lineno"> 823</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomOpSelector.html">CustomOpSelector</a>** sel_inst,</div><div class="line"><a name="l00824"></a><span class="lineno"> 824</span>  <span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00825"></a><span class="lineno"> 825</span>  std::string>& options);</div><div class="line"><a name="l00826"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae"> 826</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>)(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *subgraph, <span class="keywordtype">int</span> subgraph_id,</div><div class="line"><a name="l00827"></a><span class="lineno"> 827</span>  <span class="keywordtype">bool</span>* accept,</div><div class="line"><a name="l00828"></a><span class="lineno"> 828</span>  <span class="keyword">const</span> std::unordered_map<std::string,</div><div class="line"><a name="l00829"></a><span class="lineno"> 829</span>  std::string>& options,</div><div class="line"><a name="l00830"></a><span class="lineno"> 830</span>  std::unordered_map<std::string,</div><div class="line"><a name="l00831"></a><span class="lineno"> 831</span>  std::string>* attrs);</div><div class="line"><a name="l00832"></a><span class="lineno"> 832</span> </div><div class="line"><a name="l00836"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html"> 836</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="l00837"></a><span class="lineno"> 837</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00838"></a><span class="lineno"> 838</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>();</div><div class="line"><a name="l00839"></a><span class="lineno"> 839</span> </div><div class="line"><a name="l00840"></a><span class="lineno"> 840</span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>(<span class="keyword">const</span> <span class="keywordtype">char</span>* backend_name);</div><div class="line"><a name="l00841"></a><span class="lineno"> 841</span> </div><div class="line"><a name="l00842"></a><span class="lineno"> 842</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& addStrategy(<span class="keyword">const</span> <span class="keywordtype">char</span>* prop_name,</div><div class="line"><a name="l00843"></a><span class="lineno"> 843</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* sg_name);</div><div class="line"><a name="l00844"></a><span class="lineno"> 844</span> </div><div class="line"><a name="l00845"></a><span class="lineno"> 845</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& setSupportedOps(<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="l00846"></a><span class="lineno"> 846</span> </div><div class="line"><a name="l00847"></a><span class="lineno"> 847</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& setCreateSelector(<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="l00848"></a><span class="lineno"> 848</span> </div><div class="line"><a name="l00849"></a><span class="lineno"> 849</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a>& setReviewSubgraph(<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="l00850"></a><span class="lineno"> 850</span> </div><div class="line"><a name="l00851"></a><span class="lineno"> 851</span>  <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> getSupportedOps(<span class="keywordtype">int</span> stg_id);</div><div class="line"><a name="l00852"></a><span class="lineno"> 852</span> </div><div class="line"><a name="l00853"></a><span class="lineno"> 853</span>  <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> getCreateSelector(<span class="keywordtype">int</span> stg_id);</div><div class="line"><a name="l00854"></a><span class="lineno"> 854</span> </div><div class="line"><a name="l00855"></a><span class="lineno"> 855</span>  <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> getReviewSubgraph(<span class="keywordtype">int</span> stg_id);</div><div class="line"><a name="l00856"></a><span class="lineno"> 856</span> </div><div class="line"><a name="l00858"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a1b620d0a37e0104eb60db211930ed5a5"> 858</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="l00859"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#ab25d24f86446a2f6d3aa4da85f869472"> 859</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="l00860"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#af96c64b969175dccd80943ef69207250"> 860</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="l00861"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#a76e111c49dd7b6e3cfe550739f319611"> 861</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="l00863"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#abaadec312533d0a24f1faeba6db7411b"> 863</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="l00865"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomPartitioner.html#aeaad9d379d75938aa12b908c1a7a9af6"> 865</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="l00866"></a><span class="lineno"> 866</span> };</div><div class="line"><a name="l00867"></a><span class="lineno"> 867</span> </div><div class="line"><a name="l00872"></a><span class="lineno"> 872</span> <span class="keyword">template</span> <<span class="keyword">class</span> T></div><div class="line"><a name="l00873"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html"> 873</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="l00874"></a><span class="lineno"> 874</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l00879"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#a34d70208a11534185f4288284430acb8"> 879</a></span>  <span class="keyword">static</span> <a class="code" href="classmxnet_1_1ext_1_1Registry.html">Registry</a>* <span class="keyword">get</span>() <a class="code" href="lib__api_8h.html#acb30ec605ba828f085c28799b89383ad">PRIVATE_SYMBOL</a> {</div><div class="line"><a name="l00880"></a><span class="lineno"> 880</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="l00881"></a><span class="lineno"> 881</span>  <span class="keywordflow">return</span> &inst;</div><div class="line"><a name="l00882"></a><span class="lineno"> 882</span>  }</div><div class="line"><a name="l00887"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#a32f3e2d387f02dc4edc8c0c51095af28"> 887</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="l00888"></a><span class="lineno"> 888</span>  T *entry = <span class="keyword">new</span> T(name);</div><div class="line"><a name="l00889"></a><span class="lineno"> 889</span>  entries.push_back(entry);</div><div class="line"><a name="l00890"></a><span class="lineno"> 890</span>  <span class="keywordflow">return</span> *entry;</div><div class="line"><a name="l00891"></a><span class="lineno"> 891</span>  }</div><div class="line"><a name="l00892"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#a1a7613063268d9ff754bd3f53cdb35d6"> 892</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="l00893"></a><span class="lineno"> 893</span>  <span class="keywordflow">return</span> entries.size();</div><div class="line"><a name="l00894"></a><span class="lineno"> 894</span>  }</div><div class="line"><a name="l00895"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1Registry.html#aac245796efea7dbb18a86ff2be034453"> 895</a></span>  T& <span class="keyword">get</span>(<span class="keywordtype">int</span> idx) {</div><div class="line"><a name="l00896"></a><span class="lineno"> 896</span>  <span class="keywordflow">return</span> *(entries.at(idx));</div><div class="line"><a name="l00897"></a><span class="lineno"> 897</span>  }</div><div class="line"><a name="l00898"></a><span class="lineno"> 898</span> </div><div class="line"><a name="l00899"></a><span class="lineno"> 899</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l00901"></a><span class="lineno"> 901</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>  ~<a class="code" href="classmxnet_1_1ext_1_1Registry.html">Registry</a>() {}</div><div class="line"><a name="l00905"></a><span class="lineno"> 905</span>  std::vector<T*> entries;</div><div class="line"><a name="l00906"></a><span class="lineno"> 906</span> };</div><div class="line"><a name="l00907"></a><span class="lineno"> 907</span> </div><div class="line"><a name="l00913"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac745a1b1f09e730abf98f51d2e513103"> 913</a></span> <span class="preprocessor">#define MX_STR_CONCAT_(__a, __b) __a ## __b</span></div><div class="line"><a name="l00914"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad900a5506445a60d8e5ebd5079b0324f"> 914</a></span> <span class="preprocessor">#define MX_STR_CONCAT(__a, __b) MX_STR_CONCAT_(__a, __b)</span></div><div class="line"><a name="l00915"></a><span class="lineno"> 915</span> </div><div class="line"><a name="l00917"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a8a5ee382299049479eaadc1f87d9dffb"> 917</a></span> <span class="preprocessor">#define MX_STRINGIFY(x) #x</span></div><div class="line"><a name="l00918"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac281c11d1b738cf6f4d44d1d6d2a41f5"> 918</a></span> <span class="preprocessor">#define MX_TOSTRING(x) MX_STRINGIFY(x)</span></div><div class="line"><a name="l00919"></a><span class="lineno"> 919</span> </div><div class="line"><a name="l00921"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a412e972cba005076b3475796cd1ffbf6"> 921</a></span> <span class="preprocessor">#define MX_REGISTER_NAME_(Name) MXNet ## _CustomOp ## _ ## Name</span></div><div class="line"><a name="l00922"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aff6fd8259cfbba7e57a637c862dac3ca"> 922</a></span> <span class="preprocessor">#define MX_REGISTER_DEF_(Name) mxnet::ext::CustomOp MX_REGISTER_NAME_(Name)</span></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="lib__api_8h.html#a94176fcbf0773036ac1e43e4b4dde51f"> 924</a></span> <span class="preprocessor">#define MX_REGISTER_PROP_NAME_(Name) MXNet ## _CustomSubProp ## _ ## Name</span></div><div class="line"><a name="l00925"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2e2f91e3c9f64ceffdfdb55f36354e91"> 925</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="l00926"></a><span class="lineno"> 926</span> </div><div class="line"><a name="l00927"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a35d87ac322fe13c4c98a97d4c44c95e2"> 927</a></span> <span class="preprocessor">#define MX_REGISTER_PASS_NAME_(Name) MXNet ## _CustomPass ## _ ## Name</span></div><div class="line"><a name="l00928"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a4a6ec114cb6f848ec8170c9df3475d59"> 928</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="l00929"></a><span class="lineno"> 929</span> </div><div class="line"><a name="l00931"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abb165f6d1b3c2f0d4edfb5667a8b3638"> 931</a></span> <span class="preprocessor">#define REGISTER_OP(Name) MX_STR_CONCAT(MX_REGISTER_DEF_(Name), __COUNTER__) = \</span></div><div class="line"><a name="l00932"></a><span class="lineno"> 932</span> <span class="preprocessor"> mxnet::ext::Registry<mxnet::ext::CustomOp>::get()->add(MX_TOSTRING(Name))</span></div><div class="line"><a name="l00933"></a><span class="lineno"> 933</span> </div><div class="line"><a name="l00934"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a93f481366195b39d8e4ea3bbec2a1b04"> 934</a></span> <span class="preprocessor">#define REGISTER_PARTITIONER(Name) \</span></div><div class="line"><a name="l00935"></a><span class="lineno"> 935</span> <span class="preprocessor"> MX_STR_CONCAT(MX_REGISTER_PROP_DEF_(Name), __COUNTER__) = \</span></div><div class="line"><a name="l00936"></a><span class="lineno"> 936</span> <span class="preprocessor"> mxnet::ext::Registry<mxnet::ext::CustomPartitioner>::get()->add(MX_TOSTRING(Name))</span></div><div class="line"><a name="l00937"></a><span class="lineno"> 937</span> </div><div class="line"><a name="l00938"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ab9d14ed170244c036e1ff8a3531e89c6"> 938</a></span> <span class="preprocessor">#define REGISTER_PASS(Name) \</span></div><div class="line"><a name="l00939"></a><span class="lineno"> 939</span> <span class="preprocessor"> MX_STR_CONCAT(MX_REGISTER_PASS_DEF_(Name), __COUNTER__) = \</span></div><div class="line"><a name="l00940"></a><span class="lineno"> 940</span> <span class="preprocessor"> mxnet::ext::Registry<mxnet::ext::CustomPass>::get()->add(MX_TOSTRING(Name))</span></div><div class="line"><a name="l00941"></a><span class="lineno"> 941</span> </div><div class="line"><a name="l00942"></a><span class="lineno"> 942</span> <span class="comment">/* -------------- BELOW ARE CTYPE FUNCTIONS PROTOTYPES --------------- */</span></div><div class="line"><a name="l00943"></a><span class="lineno"> 943</span> </div><div class="line"><a name="l00949"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac39728fb75d241b62d4ed9261b7ddca3"> 949</a></span> <span class="preprocessor">#define MXLIB_OPREGSIZE_STR "_opRegSize"</span></div><div class="line"><a name="l00950"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6fb7af175432173dec6120674a1f07a4"> 950</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="l00951"></a><span class="lineno"> 951</span> </div><div class="line"><a name="l00952"></a><span class="lineno"><a class="line" href="lib__api_8h.html#addceae596034e042d887954af8499f06"> 952</a></span> <span class="preprocessor">#define MXLIB_OPREGGET_STR "_opRegGet"</span></div><div class="line"><a name="l00953"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a10b027e352f8fd55d9639398807614bd"> 953</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, <span class="keyword">const</span> <span class="keywordtype">char</span>** name, <span class="keywordtype">int</span> *isSGop,</div><div class="line"><a name="l00954"></a><span class="lineno"> 954</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** forward_ctx, <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** forward_fp,</div><div class="line"><a name="l00955"></a><span class="lineno"> 955</span>  <span class="keywordtype">int</span>* forward_count, <span class="keyword">const</span> <span class="keywordtype">char</span>*** backward_ctx,</div><div class="line"><a name="l00956"></a><span class="lineno"> 956</span>  <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** backward_fp, <span class="keywordtype">int</span>* backward_count,</div><div class="line"><a name="l00957"></a><span class="lineno"> 957</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** create_op_ctx, <a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a>** create_op_fp,</div><div class="line"><a name="l00958"></a><span class="lineno"> 958</span>  <span class="keywordtype">int</span>* create_op_count, <a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a>* parse,</div><div class="line"><a name="l00959"></a><span class="lineno"> 959</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a>* type, <a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a>* stype,</div><div class="line"><a name="l00960"></a><span class="lineno"> 960</span>  <a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a>* shape, <a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a>* mutate);</div><div class="line"><a name="l00961"></a><span class="lineno"> 961</span> </div><div class="line"><a name="l00962"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aa9581ed05ac3aefc194698100cc54e5a"> 962</a></span> <span class="preprocessor">#define MXLIB_OPCALLFREE_STR "_opCallFree"</span></div><div class="line"><a name="l00963"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aff50c6302fa056352a7838a7c0fba79c"> 963</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="l00964"></a><span class="lineno"> 964</span> </div><div class="line"><a name="l00965"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a6ff60ae5ae2081aac9715da91f38845e"> 965</a></span> <span class="preprocessor">#define MXLIB_OPCALLPARSEATTRS_STR "_opCallParseAttrs"</span></div><div class="line"><a name="l00966"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ac3c274938c9da1e44e9897639af5c0d4"> 966</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l00967"></a><span class="lineno"> 967</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l00968"></a><span class="lineno"> 968</span>  <span class="keywordtype">int</span>* num_in, <span class="keywordtype">int</span>* num_out);</div><div class="line"><a name="l00969"></a><span class="lineno"> 969</span> </div><div class="line"><a name="l00970"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ac21fd9c191ceab0d17561f4523ad6d23"> 970</a></span> <span class="preprocessor">#define MXLIB_OPCALLINFERSHAPE_STR "_opCallInferShape"</span></div><div class="line"><a name="l00971"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ade42ac66b3d7c1e063d0f71eda6394a3"> 971</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l00972"></a><span class="lineno"> 972</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l00973"></a><span class="lineno"> 973</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes, <span class="keywordtype">int</span>* indims, <span class="keywordtype">int</span> num_in,</div><div class="line"><a name="l00974"></a><span class="lineno"> 974</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** mod_inshapes, <span class="keywordtype">int</span>** mod_indims,</div><div class="line"><a name="l00975"></a><span class="lineno"> 975</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** outshapes, <span class="keywordtype">int</span>** outdims, <span class="keywordtype">int</span> num_out);</div><div class="line"><a name="l00976"></a><span class="lineno"> 976</span> </div><div class="line"><a name="l00977"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a7dfca2084c7df461862458f77a8f6244"> 977</a></span> <span class="preprocessor">#define MXLIB_OPCALLINFERTYPE_STR "_opCallInferType"</span></div><div class="line"><a name="l00978"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a08d9994f1542d34fce1eeabda3a8ad85"> 978</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l00979"></a><span class="lineno"> 979</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l00980"></a><span class="lineno"> 980</span>  <span class="keywordtype">int</span>* intypes, <span class="keywordtype">int</span> num_in, <span class="keywordtype">int</span>* outtypes, <span class="keywordtype">int</span> num_out);</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#a95494016bf876dcb3ec4ae0b3c5f6941"> 982</a></span> <span class="preprocessor">#define MXLIB_OPCALLINFERSTYPE_STR "_opCallInferSType"</span></div><div class="line"><a name="l00983"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ae5f9488dd6bdbea94a72c42382c2b6c8"> 983</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l00984"></a><span class="lineno"> 984</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l00985"></a><span class="lineno"> 985</span>  <span class="keywordtype">int</span>* intypes, <span class="keywordtype">int</span> num_in, <span class="keywordtype">int</span>* outtypes, <span class="keywordtype">int</span> num_out);</div><div class="line"><a name="l00986"></a><span class="lineno"> 986</span> </div><div class="line"><a name="l00987"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a906b409889769dcb9ed23c68becf93e3"> 987</a></span> <span class="preprocessor">#define MXLIB_OPCALLFCOMP_STR "_opCallFCompute"</span></div><div class="line"><a name="l00988"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#af447080bf199456379fd14d68804c1c7"> 988</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l00989"></a><span class="lineno"> 989</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l00990"></a><span class="lineno"> 990</span>  <span class="keyword">const</span> int64_t** inshapes, <span class="keywordtype">int</span>* indims,</div><div class="line"><a name="l00991"></a><span class="lineno"> 991</span>  <span class="keywordtype">void</span>** indata, <span class="keywordtype">int</span>* intypes,</div><div class="line"><a name="l00992"></a><span class="lineno"> 992</span>  <span class="keywordtype">size_t</span>* inIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type,</div><div class="line"><a name="l00993"></a><span class="lineno"> 993</span>  <span class="keywordtype">int</span>* indev_id, <span class="keywordtype">int</span> num_in,</div><div class="line"><a name="l00994"></a><span class="lineno"> 994</span>  <span class="keyword">const</span> int64_t** outshapes, <span class="keywordtype">int</span>* outdims,</div><div class="line"><a name="l00995"></a><span class="lineno"> 995</span>  <span class="keywordtype">void</span>** outdata, <span class="keywordtype">int</span>* outtypes,</div><div class="line"><a name="l00996"></a><span class="lineno"> 996</span>  <span class="keywordtype">size_t</span>* outIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div><div class="line"><a name="l00997"></a><span class="lineno"> 997</span>  <span class="keywordtype">int</span>* outdev_id, <span class="keywordtype">int</span> num_out,</div><div class="line"><a name="l00998"></a><span class="lineno"> 998</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, <span class="keywordtype">void</span>* cpu_alloc,</div><div class="line"><a name="l00999"></a><span class="lineno"> 999</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc, <span class="keywordtype">void</span>* gpu_alloc, <span class="keywordtype">void</span>* cuda_stream,</div><div class="line"><a name="l01000"></a><span class="lineno"> 1000</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc, <span class="keywordtype">void</span>* sparse_alloc,</div><div class="line"><a name="l01001"></a><span class="lineno"> 1001</span>  <span class="keywordtype">int</span>* instypes, <span class="keywordtype">int</span>* outstypes,</div><div class="line"><a name="l01002"></a><span class="lineno"> 1002</span>  <span class="keywordtype">void</span>** in_indices, <span class="keywordtype">void</span>** out_indices,</div><div class="line"><a name="l01003"></a><span class="lineno"> 1003</span>  <span class="keywordtype">void</span>** in_indptr, <span class="keywordtype">void</span>** out_indptr,</div><div class="line"><a name="l01004"></a><span class="lineno"> 1004</span>  int64_t* in_indices_shapes, int64_t* out_indices_shapes,</div><div class="line"><a name="l01005"></a><span class="lineno"> 1005</span>  int64_t* in_indptr_shapes, int64_t* out_indptr_shapes,</div><div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>  <span class="keywordtype">void</span>* rng_cpu_states, <span class="keywordtype">void</span>* rng_gpu_states);</div><div class="line"><a name="l01007"></a><span class="lineno"> 1007</span> </div><div class="line"><a name="l01008"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abf2577efc9a9eed8d27d6ca0af98abe1"> 1008</a></span> <span class="preprocessor">#define MXLIB_OPCALLMUTATEINPUTS_STR "_opCallMutateInputs"</span></div><div class="line"><a name="l01009"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ab867c91f53d13cb949e9c9a579fa3da5"> 1009</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01010"></a><span class="lineno"> 1010</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l01011"></a><span class="lineno"> 1011</span>  <span class="keywordtype">int</span>** mutate_indices, <span class="keywordtype">int</span>* indices_size);</div><div class="line"><a name="l01012"></a><span class="lineno"> 1012</span> </div><div class="line"><a name="l01013"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a950d395e618360e1d9c4c5b1fedb018d"> 1013</a></span> <span class="preprocessor">#define MXLIB_OPCALLCREATEOPSTATE_STR "_opCallCreateOpState"</span></div><div class="line"><a name="l01014"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#adf4bd71f1334318c5258e9421709b03c"> 1014</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, <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, <span class="keywordtype">int</span> num, <span class="keyword">const</span> <span class="keywordtype">char</span>* dev_type,</div><div class="line"><a name="l01016"></a><span class="lineno"> 1016</span>  <span class="keywordtype">int</span> dev_id, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes, <span class="keywordtype">int</span>* indims,</div><div class="line"><a name="l01017"></a><span class="lineno"> 1017</span>  <span class="keywordtype">int</span> num_in, <span class="keyword">const</span> <span class="keywordtype">int</span>* intypes, <span class="keywordtype">void</span>** state_op);</div><div class="line"><a name="l01018"></a><span class="lineno"> 1018</span> </div><div class="line"><a name="l01019"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a1724ab7121cfa78fda47b146eed0e9a9"> 1019</a></span> <span class="preprocessor">#define MXLIB_OPCALLDESTROYOPSTATE_STR "_opCallDestroyOpState"</span></div><div class="line"><a name="l01020"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70"> 1020</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="l01021"></a><span class="lineno"> 1021</span> </div><div class="line"><a name="l01022"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad10be40f94357f8913f99288ccae5bea"> 1022</a></span> <span class="preprocessor">#define MXLIB_OPCALLFSTATEFULCOMP_STR "_opCallFStatefulCompute"</span></div><div class="line"><a name="l01023"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ada6444201d1204384d1287d91a4c3fc7"> 1023</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, <span class="keywordtype">void</span>* state_op,</div><div class="line"><a name="l01024"></a><span class="lineno"> 1024</span>  <span class="keyword">const</span> int64_t** inshapes, <span class="keywordtype">int</span>* indims,</div><div class="line"><a name="l01025"></a><span class="lineno"> 1025</span>  <span class="keywordtype">void</span>** indata, <span class="keywordtype">int</span>* intypes,</div><div class="line"><a name="l01026"></a><span class="lineno"> 1026</span>  <span class="keywordtype">size_t</span>* inIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type,</div><div class="line"><a name="l01027"></a><span class="lineno"> 1027</span>  <span class="keywordtype">int</span>* indev_id, <span class="keywordtype">int</span> num_in,</div><div class="line"><a name="l01028"></a><span class="lineno"> 1028</span>  <span class="keyword">const</span> int64_t** outshapes, <span class="keywordtype">int</span>* outdims,</div><div class="line"><a name="l01029"></a><span class="lineno"> 1029</span>  <span class="keywordtype">void</span>** outdata, <span class="keywordtype">int</span>* outtypes,</div><div class="line"><a name="l01030"></a><span class="lineno"> 1030</span>  <span class="keywordtype">size_t</span>* outIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div><div class="line"><a name="l01031"></a><span class="lineno"> 1031</span>  <span class="keywordtype">int</span>* outdev_id, <span class="keywordtype">int</span> num_out,</div><div class="line"><a name="l01032"></a><span class="lineno"> 1032</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, <span class="keywordtype">void</span>* cpu_alloc,</div><div class="line"><a name="l01033"></a><span class="lineno"> 1033</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc, <span class="keywordtype">void</span>* gpu_alloc, <span class="keywordtype">void</span>* stream,</div><div class="line"><a name="l01034"></a><span class="lineno"> 1034</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> sparse_malloc, <span class="keywordtype">void</span>* sparse_alloc,</div><div class="line"><a name="l01035"></a><span class="lineno"> 1035</span>  <span class="keywordtype">int</span>* instypes, <span class="keywordtype">int</span>* outstypes,</div><div class="line"><a name="l01036"></a><span class="lineno"> 1036</span>  <span class="keywordtype">void</span>** in_indices, <span class="keywordtype">void</span>** out_indices,</div><div class="line"><a name="l01037"></a><span class="lineno"> 1037</span>  <span class="keywordtype">void</span>** in_indptr, <span class="keywordtype">void</span>** out_indptr,</div><div class="line"><a name="l01038"></a><span class="lineno"> 1038</span>  int64_t* in_indices_shapes, int64_t* out_indices_shapes,</div><div class="line"><a name="l01039"></a><span class="lineno"> 1039</span>  int64_t* in_indptr_shapes, int64_t* out_indptr_shapes,</div><div class="line"><a name="l01040"></a><span class="lineno"> 1040</span>  <span class="keywordtype">void</span>* rng_cpu_states, <span class="keywordtype">void</span>* rng_gpu_states);</div><div class="line"><a name="l01041"></a><span class="lineno"> 1041</span> </div><div class="line"><a name="l01042"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a16de6fee0cb7e09e2e3c0885b44d646d"> 1042</a></span> <span class="preprocessor">#define MXLIB_PARTREGSIZE_STR "_partRegSize"</span></div><div class="line"><a name="l01043"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a8a4d41b35b45f3644497d9a8f717b6a3"> 1043</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="l01044"></a><span class="lineno"> 1044</span> </div><div class="line"><a name="l01045"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a96ebb8de1ce9d8f8a493e262b477f0b7"> 1045</a></span> <span class="preprocessor">#define MXLIB_PARTREGGETCOUNT_STR "_partRegGetCount"</span></div><div class="line"><a name="l01046"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a0b6d87bb4c06678d0be1dd3ad5916a1b"> 1046</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="l01047"></a><span class="lineno"> 1047</span> </div><div class="line"><a name="l01048"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a76dda0c1c4d6eafc6ac80a72527ec556"> 1048</a></span> <span class="preprocessor">#define MXLIB_PARTREGGET_STR "_partRegGet"</span></div><div class="line"><a name="l01049"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ae628507b542734ee29ddf2f499fbdb05"> 1049</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, <span class="keywordtype">int</span> stg_idx, <span class="keyword">const</span> <span class="keywordtype">char</span>** strategy,</div><div class="line"><a name="l01050"></a><span class="lineno"> 1050</span>  <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a>* supportedOps, <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a>* createSelector,</div><div class="line"><a name="l01051"></a><span class="lineno"> 1051</span>  <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a>* reviewSubgraph, <span class="keyword">const</span> <span class="keywordtype">char</span>** op_name);</div><div class="line"><a name="l01052"></a><span class="lineno"> 1052</span> </div><div class="line"><a name="l01053"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a7e8e351e67da4f1822042c0d01e8bbf4"> 1053</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSUPPORTEDOPS_STR "_partCallSupportedOps"</span></div><div class="line"><a name="l01054"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a03342e626369680bc842ec763588a91e"> 1054</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01055"></a><span class="lineno"> 1055</span>  <span class="keywordtype">int</span> num_ids, <span class="keywordtype">int</span> *ids, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01056"></a><span class="lineno"> 1056</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts);</div><div class="line"><a name="l01057"></a><span class="lineno"> 1057</span> </div><div class="line"><a name="l01058"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a2136a3986fc3a229cb426f41d0dfe3c8"> 1058</a></span> <span class="preprocessor">#define MXLIB_PARTCALLCREATESELECTOR_STR "_partCallCreateSelector"</span></div><div class="line"><a name="l01059"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ad62142c24acef23ecd62e7fd53000349"> 1059</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01060"></a><span class="lineno"> 1060</span>  <span class="keywordtype">void</span>** selector, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01061"></a><span class="lineno"> 1061</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts);</div><div class="line"><a name="l01062"></a><span class="lineno"> 1062</span> </div><div class="line"><a name="l01063"></a><span class="lineno"><a class="line" href="lib__api_8h.html#acca6909f1ef0b0a4e7dc4815dd75255a"> 1063</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSELECT_STR "_partCallSelect"</span></div><div class="line"><a name="l01064"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a462103a47b8f387ee5a31393c0a47c1a"> 1064</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="l01065"></a><span class="lineno"> 1065</span> </div><div class="line"><a name="l01066"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a84b9b257f62fcfa8e55af189fe7d88cb"> 1066</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSELECTINPUT_STR "_partCallSelectInput"</span></div><div class="line"><a name="l01067"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#ad2fed961bb0a655fb91fc6ee4c26162a"> 1067</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,</div><div class="line"><a name="l01068"></a><span class="lineno"> 1068</span>  <span class="keywordtype">int</span>* selected);</div><div class="line"><a name="l01069"></a><span class="lineno"> 1069</span> </div><div class="line"><a name="l01070"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abf23d7995f4c8617b136c6c311405b35"> 1070</a></span> <span class="preprocessor">#define MXLIB_PARTCALLSELECTOUTPUT_STR "_partCallSelectOutput"</span></div><div class="line"><a name="l01071"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a5b53f5861183811ba8fc2cdf9dbee5f6"> 1071</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, <span class="keywordtype">int</span> nodeID, <span class="keywordtype">int</span> output_nodeID,</div><div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>  <span class="keywordtype">int</span>* selected);</div><div class="line"><a name="l01073"></a><span class="lineno"> 1073</span> </div><div class="line"><a name="l01074"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad43d2f7d1ca8f2b248630ef95b3eb5e9"> 1074</a></span> <span class="preprocessor">#define MXLIB_PARTCALLFILTER_STR "_partCallFilter"</span></div><div class="line"><a name="l01075"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a7ad34e214897e21a5c10bc57ddf1b51a"> 1075</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, <span class="keywordtype">int</span>* candidates, <span class="keywordtype">int</span> num_candidates,</div><div class="line"><a name="l01076"></a><span class="lineno"> 1076</span>  <span class="keywordtype">int</span>** keep, <span class="keywordtype">int</span>* num_keep);</div><div class="line"><a name="l01077"></a><span class="lineno"> 1077</span> </div><div class="line"><a name="l01078"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ae39d5c24d804a90a045483fe5d87b2c2"> 1078</a></span> <span class="preprocessor">#define MXLIB_PARTCALLRESET_STR "_partCallReset"</span></div><div class="line"><a name="l01079"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a32fc355e422990910299065269f10459"> 1079</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="l01080"></a><span class="lineno"> 1080</span> </div><div class="line"><a name="l01081"></a><span class="lineno"><a class="line" href="lib__api_8h.html#aebf42394c4714072d81f4612a9d425fe"> 1081</a></span> <span class="preprocessor">#define MXLIB_PARTCALLREVIEWSUBGRAPH_STR "_partCallReviewSubgraph"</span></div><div class="line"><a name="l01082"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aefb1c0f85ebec11dd8e4f8720392ce4a"> 1082</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01083"></a><span class="lineno"> 1083</span>  <span class="keywordtype">int</span> subgraph_id, <span class="keywordtype">int</span> *accept, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01084"></a><span class="lineno"> 1084</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts,</div><div class="line"><a name="l01085"></a><span class="lineno"> 1085</span>  <span class="keywordtype">char</span>*** attr_keys, <span class="keywordtype">char</span>*** attr_vals, <span class="keywordtype">int</span> *num_attrs,</div><div class="line"><a name="l01086"></a><span class="lineno"> 1086</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names, <span class="keywordtype">int</span> num_args,</div><div class="line"><a name="l01087"></a><span class="lineno"> 1087</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div><div class="line"><a name="l01088"></a><span class="lineno"> 1088</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div><div class="line"><a name="l01089"></a><span class="lineno"> 1089</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div><div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div><div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names, <span class="keywordtype">int</span> num_aux,</div><div class="line"><a name="l01092"></a><span class="lineno"> 1092</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div><div class="line"><a name="l01093"></a><span class="lineno"> 1093</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div><div class="line"><a name="l01094"></a><span class="lineno"> 1094</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div><div class="line"><a name="l01095"></a><span class="lineno"> 1095</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id);</div><div class="line"><a name="l01096"></a><span class="lineno"> 1096</span> </div><div class="line"><a name="l01097"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a6ec8627b92e8212d9e3368619e8361a4"> 1097</a></span> <span class="preprocessor">#define MXLIB_PASSREGSIZE_STR "_passRegSize"</span></div><div class="line"><a name="l01098"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a6f47c7152edff58ce548870e7c2de0ab"> 1098</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="l01099"></a><span class="lineno"> 1099</span> </div><div class="line"><a name="l01100"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a303c810b43b30121a5f24ed880d53d09"> 1100</a></span> <span class="preprocessor">#define MXLIB_PASSREGGET_STR "_passRegGet"</span></div><div class="line"><a name="l01101"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a83e21d3dbcb68574f93c95db0a83ce31"> 1101</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="l01102"></a><span class="lineno"> 1102</span> </div><div class="line"><a name="l01103"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a5ebc06cae18b2b72e41789d816734540"> 1103</a></span> <span class="preprocessor">#define MXLIB_PASSCALLGRAPHPASS_STR "_passCallGraphPass"</span></div><div class="line"><a name="l01104"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a4d96003e8adf76b07e993b83f7cef5bd"> 1104</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *in_graph,</div><div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>  <span class="keywordtype">char</span>** out_graph, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01106"></a><span class="lineno"> 1106</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts,</div><div class="line"><a name="l01107"></a><span class="lineno"> 1107</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* pass_name, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names,</div><div class="line"><a name="l01108"></a><span class="lineno"> 1108</span>  <span class="keywordtype">int</span> num_args, <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data,</div><div class="line"><a name="l01109"></a><span class="lineno"> 1109</span>  <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes, <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims,</div><div class="line"><a name="l01110"></a><span class="lineno"> 1110</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types, <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs,</div><div class="line"><a name="l01111"></a><span class="lineno"> 1111</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type, <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div><div class="line"><a name="l01112"></a><span class="lineno"> 1112</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names, <span class="keywordtype">int</span> num_aux,</div><div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div><div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div><div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div><div class="line"><a name="l01116"></a><span class="lineno"> 1116</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id, <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc,</div><div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc);</div><div class="line"><a name="l01118"></a><span class="lineno"> 1118</span> </div><div class="line"><a name="l01119"></a><span class="lineno"><a class="line" href="lib__api_8h.html#ad5405711bcd25d56b2fc2a3fd827ea3c"> 1119</a></span> <span class="preprocessor">#define MXLIB_INITIALIZE_STR "initialize"</span></div><div class="line"><a name="l01120"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a9089010cd352a7e5a36fc3ade38dd221"> 1120</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="l01121"></a><span class="lineno"> 1121</span> </div><div class="line"><a name="l01122"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a13594d09f1287481f1b7b6b695d6c1d8"> 1122</a></span> <span class="preprocessor">#define MXLIB_OPVERSION_STR "_opVersion"</span></div><div class="line"><a name="l01123"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a69ba2ade8bd893eb8461a21cd8f60eb5"> 1123</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="l01124"></a><span class="lineno"> 1124</span> </div><div class="line"><a name="l01125"></a><span class="lineno"><a class="line" href="lib__api_8h.html#abbae445517e023b3f8f81b1999171ae5"> 1125</a></span> <span class="preprocessor">#define MXLIB_MSGSIZE_STR "_msgSize"</span></div><div class="line"><a name="l01126"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#a82c4b41a9a9a1160f1043d10dc9fcf50"> 1126</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="l01127"></a><span class="lineno"> 1127</span> </div><div class="line"><a name="l01128"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a12e8d84f980811cd9a90e5e4506d2011"> 1128</a></span> <span class="preprocessor">#define MXLIB_MSGGET_STR "_msgGet"</span></div><div class="line"><a name="l01129"></a><span class="lineno"><a class="line" href="namespacemxnet_1_1ext.html#aa7339fd6ce1c23dd296015f450463cc4"> 1129</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="l01130"></a><span class="lineno"> 1130</span> </div><div class="line"><a name="l01132"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html"> 1132</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="l01133"></a><span class="lineno"> 1133</span>  <span class="keyword">public</span>:</div><div class="line"><a name="l01134"></a><span class="lineno"> 1134</span>  ~<a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html">CustomStatefulOpWrapper</a>();</div><div class="line"><a name="l01135"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#aad23e82ea3f745ac3a92c07b6afed1d2"> 1135</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="l01136"></a><span class="lineno"> 1136</span>  : instance(inst), destroy_(destroy) {}</div><div class="line"><a name="l01137"></a><span class="lineno"><a class="line" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#ace82ec005741e89957ba7fdfe433f654"> 1137</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>() { <span class="keywordflow">return</span> instance; }</div><div class="line"><a name="l01138"></a><span class="lineno"> 1138</span>  <span class="keyword">private</span>:</div><div class="line"><a name="l01139"></a><span class="lineno"> 1139</span>  <a class="code" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a>* instance;</div><div class="line"><a name="l01140"></a><span class="lineno"> 1140</span>  <a class="code" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70">opCallDestroyOpState_t</a> destroy_;</div><div class="line"><a name="l01141"></a><span class="lineno"> 1141</span> };</div><div class="line"><a name="l01142"></a><span class="lineno"> 1142</span> </div><div class="line"><a name="l01143"></a><span class="lineno"> 1143</span> <span class="preprocessor">#if defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__)</span></div><div class="line"><a name="l01144"></a><span class="lineno"> 1144</span> <span class="preprocessor">#define MX_INT_RET __declspec(dllexport) int __cdecl</span></div><div class="line"><a name="l01145"></a><span class="lineno"> 1145</span> <span class="preprocessor">#define MX_VOID_RET __declspec(dllexport) void __cdecl</span></div><div class="line"><a name="l01146"></a><span class="lineno"> 1146</span> <span class="preprocessor">#else</span></div><div class="line"><a name="l01147"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a561ade7c8fd77c6882e937514f5b1e05"> 1147</a></span> <span class="preprocessor">#define MX_INT_RET int</span></div><div class="line"><a name="l01148"></a><span class="lineno"><a class="line" href="lib__api_8h.html#a140a8dac396de476f9ed0c6a093cf4fa"> 1148</a></span> <span class="preprocessor">#define MX_VOID_RET void</span></div><div class="line"><a name="l01149"></a><span class="lineno"> 1149</span> <span class="preprocessor">#endif</span></div><div class="line"><a name="l01150"></a><span class="lineno"> 1150</span> </div><div class="line"><a name="l01151"></a><span class="lineno"> 1151</span> } <span class="comment">// namespace ext</span></div><div class="line"><a name="l01152"></a><span class="lineno"> 1152</span> } <span class="comment">// namespace mxnet</span></div><div class="line"><a name="l01153"></a><span class="lineno"> 1153</span> </div><div class="line"><a name="l01154"></a><span class="lineno"> 1154</span> <span class="keyword">extern</span> <span class="stringliteral">"C"</span> {</div><div class="line"><a name="l01156"></a><span class="lineno"> 1156</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="l01157"></a><span class="lineno"> 1157</span> </div><div class="line"><a name="l01159"></a><span class="lineno"> 1159</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="l01160"></a><span class="lineno"> 1160</span> </div><div class="line"><a name="l01162"></a><span class="lineno"> 1162</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, <span class="keyword">const</span> <span class="keywordtype">char</span>** name, <span class="keywordtype">int</span> *isSGop,</div><div class="line"><a name="l01163"></a><span class="lineno"> 1163</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** forward_ctx, <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** forward_fp,</div><div class="line"><a name="l01164"></a><span class="lineno"> 1164</span>  <span class="keywordtype">int</span>* forward_count, <span class="keyword">const</span> <span class="keywordtype">char</span>*** backward_ctx,</div><div class="line"><a name="l01165"></a><span class="lineno"> 1165</span>  <a class="code" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a>** backward_fp, <span class="keywordtype">int</span>* backward_count,</div><div class="line"><a name="l01166"></a><span class="lineno"> 1166</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>*** create_op_ctx, <a class="code" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a>** create_op_fp,</div><div class="line"><a name="l01167"></a><span class="lineno"> 1167</span>  <span class="keywordtype">int</span>* create_op_count, <a class="code" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a>* parse,</div><div class="line"><a name="l01168"></a><span class="lineno"> 1168</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a>* type, <a class="code" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a>* stype,</div><div class="line"><a name="l01169"></a><span class="lineno"> 1169</span>  <a class="code" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a>* shape, <a class="code" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a>* mutate);</div><div class="line"><a name="l01170"></a><span class="lineno"> 1170</span> </div><div class="line"><a name="l01172"></a><span class="lineno"> 1172</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="l01173"></a><span class="lineno"> 1173</span> </div><div class="line"><a name="l01175"></a><span class="lineno"> 1175</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01176"></a><span class="lineno"> 1176</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l01177"></a><span class="lineno"> 1177</span>  <span class="keywordtype">int</span>* num_in, <span class="keywordtype">int</span>* num_out);</div><div class="line"><a name="l01178"></a><span class="lineno"> 1178</span> </div><div class="line"><a name="l01180"></a><span class="lineno"> 1180</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01181"></a><span class="lineno"> 1181</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l01182"></a><span class="lineno"> 1182</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes, <span class="keywordtype">int</span>* indims, <span class="keywordtype">int</span> num_in,</div><div class="line"><a name="l01183"></a><span class="lineno"> 1183</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** mod_inshapes, <span class="keywordtype">int</span>** mod_indims,</div><div class="line"><a name="l01184"></a><span class="lineno"> 1184</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>*** outshapes, <span class="keywordtype">int</span>** outdims, <span class="keywordtype">int</span> num_out);</div><div class="line"><a name="l01185"></a><span class="lineno"> 1185</span> </div><div class="line"><a name="l01187"></a><span class="lineno"> 1187</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01188"></a><span class="lineno"> 1188</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l01189"></a><span class="lineno"> 1189</span>  <span class="keywordtype">int</span>* intypes, <span class="keywordtype">int</span> num_in, <span class="keywordtype">int</span>* outtypes, <span class="keywordtype">int</span> num_out);</div><div class="line"><a name="l01190"></a><span class="lineno"> 1190</span> </div><div class="line"><a name="l01192"></a><span class="lineno"> 1192</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01193"></a><span class="lineno"> 1193</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l01194"></a><span class="lineno"> 1194</span>  <span class="keywordtype">int</span>* instypes, <span class="keywordtype">int</span> num_in, <span class="keywordtype">int</span>* outstypes, <span class="keywordtype">int</span> num_out);</div><div class="line"><a name="l01195"></a><span class="lineno"> 1195</span> </div><div class="line"><a name="l01197"></a><span class="lineno"> 1197</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01198"></a><span class="lineno"> 1198</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals,</div><div class="line"><a name="l01199"></a><span class="lineno"> 1199</span>  <span class="keywordtype">int</span> num, <span class="keyword">const</span> int64_t** inshapes, <span class="keywordtype">int</span>* indims, <span class="keywordtype">void</span>** indata,</div><div class="line"><a name="l01200"></a><span class="lineno"> 1200</span>  <span class="keywordtype">int</span>* intypes, <span class="keywordtype">size_t</span>* inIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type, <span class="keywordtype">int</span>* indev_id,</div><div class="line"><a name="l01201"></a><span class="lineno"> 1201</span>  <span class="keywordtype">int</span> num_in, <span class="keyword">const</span> int64_t** outshapes, <span class="keywordtype">int</span>* outdims, <span class="keywordtype">void</span>** outdata,</div><div class="line"><a name="l01202"></a><span class="lineno"> 1202</span>  <span class="keywordtype">int</span>* outtypes, <span class="keywordtype">size_t</span>* outIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div><div class="line"><a name="l01203"></a><span class="lineno"> 1203</span>  <span class="keywordtype">int</span>* outdev_id, <span class="keywordtype">int</span> num_out, <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> cpu_malloc,</div><div class="line"><a name="l01204"></a><span class="lineno"> 1204</span>  <span class="keywordtype">void</span>* cpu_alloc,</div><div class="line"><a name="l01205"></a><span class="lineno"> 1205</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> gpu_malloc, <span class="keywordtype">void</span>* gpu_alloc,</div><div class="line"><a name="l01206"></a><span class="lineno"> 1206</span>  <span class="keywordtype">void</span>* cuda_stream,</div><div class="line"><a name="l01207"></a><span class="lineno"> 1207</span>  <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">mxnet::ext::sparse_malloc_t</a> sparse_malloc, <span class="keywordtype">void</span>* sparse_alloc,</div><div class="line"><a name="l01208"></a><span class="lineno"> 1208</span>  <span class="keywordtype">int</span>* instypes, <span class="keywordtype">int</span>* outstypes, <span class="keywordtype">void</span>** in_indices, <span class="keywordtype">void</span>** out_indices,</div><div class="line"><a name="l01209"></a><span class="lineno"> 1209</span>  <span class="keywordtype">void</span>** in_indptr, <span class="keywordtype">void</span>** out_indptr,</div><div class="line"><a name="l01210"></a><span class="lineno"> 1210</span>  int64_t* in_indices_shapes, int64_t* out_indices_shapes,</div><div class="line"><a name="l01211"></a><span class="lineno"> 1211</span>  int64_t* in_indptr_shapes, int64_t* out_indptr_shapes,</div><div class="line"><a name="l01212"></a><span class="lineno"> 1212</span>  <span class="keywordtype">void</span>* rng_cpu_states, <span class="keywordtype">void</span>* rng_gpu_states);</div><div class="line"><a name="l01213"></a><span class="lineno"> 1213</span> </div><div class="line"><a name="l01215"></a><span class="lineno"> 1215</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01216"></a><span class="lineno"> 1216</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num,</div><div class="line"><a name="l01217"></a><span class="lineno"> 1217</span>  <span class="keywordtype">int</span>** mutate_indices, <span class="keywordtype">int</span>* indices_size);</div><div class="line"><a name="l01218"></a><span class="lineno"> 1218</span> </div><div class="line"><a name="l01220"></a><span class="lineno"> 1220</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, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* keys,</div><div class="line"><a name="l01221"></a><span class="lineno"> 1221</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* vals, <span class="keywordtype">int</span> num, <span class="keyword">const</span> <span class="keywordtype">char</span>* dev_type,</div><div class="line"><a name="l01222"></a><span class="lineno"> 1222</span>  <span class="keywordtype">int</span> dev_id, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span>** inshapes, <span class="keywordtype">int</span>* indims,</div><div class="line"><a name="l01223"></a><span class="lineno"> 1223</span>  <span class="keywordtype">int</span> num_in, <span class="keyword">const</span> <span class="keywordtype">int</span>* intypes, <span class="keywordtype">void</span>** state_op);</div><div class="line"><a name="l01224"></a><span class="lineno"> 1224</span> </div><div class="line"><a name="l01226"></a><span class="lineno"> 1226</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="l01227"></a><span class="lineno"> 1227</span> </div><div class="line"><a name="l01229"></a><span class="lineno"> 1229</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, <span class="keywordtype">void</span>* state_op, <span class="keyword">const</span> int64_t** inshapes,</div><div class="line"><a name="l01230"></a><span class="lineno"> 1230</span>  <span class="keywordtype">int</span>* indims, <span class="keywordtype">void</span>** indata, <span class="keywordtype">int</span>* intypes, <span class="keywordtype">size_t</span>* inIDs,</div><div class="line"><a name="l01231"></a><span class="lineno"> 1231</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** indev_type, <span class="keywordtype">int</span>* indev_id, <span class="keywordtype">int</span> num_in,</div><div class="line"><a name="l01232"></a><span class="lineno"> 1232</span>  <span class="keyword">const</span> int64_t** outshapes, <span class="keywordtype">int</span>* outdims, <span class="keywordtype">void</span>** outdata,</div><div class="line"><a name="l01233"></a><span class="lineno"> 1233</span>  <span class="keywordtype">int</span>* outtypes, <span class="keywordtype">size_t</span>* outIDs, <span class="keyword">const</span> <span class="keywordtype">char</span>** outdev_type,</div><div class="line"><a name="l01234"></a><span class="lineno"> 1234</span>  <span class="keywordtype">int</span>* outdev_id, <span class="keywordtype">int</span> num_out,</div><div class="line"><a name="l01235"></a><span class="lineno"> 1235</span>  <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> cpu_malloc,</div><div class="line"><a name="l01236"></a><span class="lineno"> 1236</span>  <span class="keywordtype">void</span>* cpu_alloc, <a class="code" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">mxnet::ext::xpu_malloc_t</a> gpu_malloc,</div><div class="line"><a name="l01237"></a><span class="lineno"> 1237</span>  <span class="keywordtype">void</span>* gpu_alloc,</div><div class="line"><a name="l01238"></a><span class="lineno"> 1238</span>  <span class="keywordtype">void</span>* stream, <a class="code" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">mxnet::ext::sparse_malloc_t</a> sparse_malloc,</div><div class="line"><a name="l01239"></a><span class="lineno"> 1239</span>  <span class="keywordtype">void</span>* sparse_alloc, <span class="keywordtype">int</span>* instypes, <span class="keywordtype">int</span>* outstypes,</div><div class="line"><a name="l01240"></a><span class="lineno"> 1240</span>  <span class="keywordtype">void</span>** in_indices, <span class="keywordtype">void</span>** out_indices, <span class="keywordtype">void</span>** in_indptr,</div><div class="line"><a name="l01241"></a><span class="lineno"> 1241</span>  <span class="keywordtype">void</span>** out_indptr, int64_t* in_indices_shapes,</div><div class="line"><a name="l01242"></a><span class="lineno"> 1242</span>  int64_t* out_indices_shapes, int64_t* in_indptr_shapes,</div><div class="line"><a name="l01243"></a><span class="lineno"> 1243</span>  int64_t* out_indptr_shapes,</div><div class="line"><a name="l01244"></a><span class="lineno"> 1244</span>  <span class="keywordtype">void</span>* rng_cpu_states, <span class="keywordtype">void</span>* rng_gpu_states);</div><div class="line"><a name="l01245"></a><span class="lineno"> 1245</span> </div><div class="line"><a name="l01247"></a><span class="lineno"> 1247</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="l01248"></a><span class="lineno"> 1248</span> </div><div class="line"><a name="l01249"></a><span class="lineno"> 1249</span>  <span class="comment">/* returns number of strategies registered for partitioner</span></div><div class="line"><a name="l01250"></a><span class="lineno"> 1250</span> <span class="comment"> * at specified index */</span></div><div class="line"><a name="l01251"></a><span class="lineno"> 1251</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="l01252"></a><span class="lineno"> 1252</span> </div><div class="line"><a name="l01254"></a><span class="lineno"> 1254</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, <span class="keywordtype">int</span> stg_idx, <span class="keyword">const</span> <span class="keywordtype">char</span>** strategy,</div><div class="line"><a name="l01255"></a><span class="lineno"> 1255</span>  <a class="code" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">mxnet::ext::supportedOps_t</a>* supportedOps,</div><div class="line"><a name="l01256"></a><span class="lineno"> 1256</span>  <a class="code" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">mxnet::ext::createSelector_t</a>* createSelector,</div><div class="line"><a name="l01257"></a><span class="lineno"> 1257</span>  <a class="code" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">mxnet::ext::reviewSubgraph_t</a>* reviewSubgraph, <span class="keyword">const</span> <span class="keywordtype">char</span>** op_name);</div><div class="line"><a name="l01258"></a><span class="lineno"> 1258</span> </div><div class="line"><a name="l01260"></a><span class="lineno"> 1260</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01261"></a><span class="lineno"> 1261</span>  <span class="keywordtype">int</span> num_ids, <span class="keywordtype">int</span> *ids, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01262"></a><span class="lineno"> 1262</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts);</div><div class="line"><a name="l01263"></a><span class="lineno"> 1263</span> </div><div class="line"><a name="l01265"></a><span class="lineno"> 1265</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01266"></a><span class="lineno"> 1266</span>  <span class="keywordtype">void</span>** selector, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01267"></a><span class="lineno"> 1267</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts);</div><div class="line"><a name="l01268"></a><span class="lineno"> 1268</span> </div><div class="line"><a name="l01270"></a><span class="lineno"> 1270</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="l01271"></a><span class="lineno"> 1271</span> </div><div class="line"><a name="l01273"></a><span class="lineno"> 1273</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,</div><div class="line"><a name="l01274"></a><span class="lineno"> 1274</span>  <span class="keywordtype">int</span> input_nodeID, <span class="keywordtype">int</span>* selected);</div><div class="line"><a name="l01275"></a><span class="lineno"> 1275</span> </div><div class="line"><a name="l01277"></a><span class="lineno"> 1277</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,</div><div class="line"><a name="l01278"></a><span class="lineno"> 1278</span>  <span class="keywordtype">int</span> output_nodeID, <span class="keywordtype">int</span>* selected);</div><div class="line"><a name="l01279"></a><span class="lineno"> 1279</span> </div><div class="line"><a name="l01281"></a><span class="lineno"> 1281</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, <span class="keywordtype">int</span>* candidates, <span class="keywordtype">int</span> num_candidates,</div><div class="line"><a name="l01282"></a><span class="lineno"> 1282</span>  <span class="keywordtype">int</span>** keep, <span class="keywordtype">int</span>* num_keep);</div><div class="line"><a name="l01283"></a><span class="lineno"> 1283</span> </div><div class="line"><a name="l01285"></a><span class="lineno"> 1285</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="l01286"></a><span class="lineno"> 1286</span> </div><div class="line"><a name="l01288"></a><span class="lineno"> 1288</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01289"></a><span class="lineno"> 1289</span>  <span class="keywordtype">int</span> subgraph_id, <span class="keywordtype">int</span> *accept, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01290"></a><span class="lineno"> 1290</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts,</div><div class="line"><a name="l01291"></a><span class="lineno"> 1291</span>  <span class="keywordtype">char</span>*** attr_keys, <span class="keywordtype">char</span>*** attr_vals, <span class="keywordtype">int</span> *num_attrs,</div><div class="line"><a name="l01292"></a><span class="lineno"> 1292</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names, <span class="keywordtype">int</span> num_args,</div><div class="line"><a name="l01293"></a><span class="lineno"> 1293</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div><div class="line"><a name="l01294"></a><span class="lineno"> 1294</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div><div class="line"><a name="l01295"></a><span class="lineno"> 1295</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div><div class="line"><a name="l01296"></a><span class="lineno"> 1296</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id,</div><div class="line"><a name="l01297"></a><span class="lineno"> 1297</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names, <span class="keywordtype">int</span> num_aux,</div><div class="line"><a name="l01298"></a><span class="lineno"> 1298</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div><div class="line"><a name="l01299"></a><span class="lineno"> 1299</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div><div class="line"><a name="l01300"></a><span class="lineno"> 1300</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div><div class="line"><a name="l01301"></a><span class="lineno"> 1301</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id);</div><div class="line"><a name="l01302"></a><span class="lineno"> 1302</span> </div><div class="line"><a name="l01304"></a><span class="lineno"> 1304</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="l01305"></a><span class="lineno"> 1305</span> </div><div class="line"><a name="l01307"></a><span class="lineno"> 1307</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,</div><div class="line"><a name="l01308"></a><span class="lineno"> 1308</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>** pass_name);</div><div class="line"><a name="l01309"></a><span class="lineno"> 1309</span> </div><div class="line"><a name="l01311"></a><span class="lineno"> 1311</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, <span class="keyword">const</span> <span class="keywordtype">char</span> *json,</div><div class="line"><a name="l01312"></a><span class="lineno"> 1312</span>  <span class="keywordtype">char</span>** out_graph, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_keys,</div><div class="line"><a name="l01313"></a><span class="lineno"> 1313</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* opt_vals, <span class="keywordtype">int</span> num_opts,</div><div class="line"><a name="l01314"></a><span class="lineno"> 1314</span>  <span class="keyword">const</span> <span class="keywordtype">char</span>* pass_name, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_names, <span class="keywordtype">int</span> num_args,</div><div class="line"><a name="l01315"></a><span class="lineno"> 1315</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* arg_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* arg_shapes,</div><div class="line"><a name="l01316"></a><span class="lineno"> 1316</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_types,</div><div class="line"><a name="l01317"></a><span class="lineno"> 1317</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* arg_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* arg_dev_type,</div><div class="line"><a name="l01318"></a><span class="lineno"> 1318</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* arg_dev_id, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_names, <span class="keywordtype">int</span> num_aux,</div><div class="line"><a name="l01319"></a><span class="lineno"> 1319</span>  <span class="keywordtype">void</span>* <span class="keyword">const</span>* aux_data, <span class="keyword">const</span> int64_t* <span class="keyword">const</span>* aux_shapes,</div><div class="line"><a name="l01320"></a><span class="lineno"> 1320</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dims, <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_types,</div><div class="line"><a name="l01321"></a><span class="lineno"> 1321</span>  <span class="keyword">const</span> <span class="keywordtype">size_t</span>* aux_IDs, <span class="keyword">const</span> <span class="keywordtype">char</span>* <span class="keyword">const</span>* aux_dev_type,</div><div class="line"><a name="l01322"></a><span class="lineno"> 1322</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>* aux_dev_id, <a class="code" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">mxnet::ext::nd_malloc_t</a> nd_malloc,</div><div class="line"><a name="l01323"></a><span class="lineno"> 1323</span>  <span class="keyword">const</span> <span class="keywordtype">void</span>* nd_alloc);</div><div class="line"><a name="l01324"></a><span class="lineno"> 1324</span> </div><div class="line"><a name="l01332"></a><span class="lineno"> 1332</span> <span class="preprocessor">#if defined(_WIN32) || defined(_WIN64) || defined(__WINDOWS__)</span></div><div class="line"><a name="l01333"></a><span class="lineno"> 1333</span>  __declspec(dllexport) <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">mxnet::ext::MXReturnValue</a> __cdecl</div><div class="line"><a name="l01334"></a><span class="lineno"> 1334</span> <span class="preprocessor">#else</span></div><div class="line"><a name="l01335"></a><span class="lineno"> 1335</span>  <a class="code" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">mxnet::ext::MXReturnValue</a></div><div class="line"><a name="l01336"></a><span class="lineno"> 1336</span> <span class="preprocessor">#endif</span></div><div class="line"><a name="l01337"></a><span class="lineno"> 1337</span>  <a class="code" href="lib__api_8h.html#a3b4abe8a1f483ab51bb4274be2f193ac">initialize</a>(<span class="keywordtype">int</span> version);</div><div class="line"><a name="l01338"></a><span class="lineno"> 1338</span> </div><div class="line"><a name="l01339"></a><span class="lineno"> 1339</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="l01340"></a><span class="lineno"> 1340</span> </div><div class="line"><a name="l01342"></a><span class="lineno"> 1342</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="l01343"></a><span class="lineno"> 1343</span> } <span class="comment">// extern "C"</span></div><div class="line"><a name="l01344"></a><span class="lineno"> 1344</span> </div><div class="line"><a name="l01345"></a><span class="lineno"> 1345</span> <span class="preprocessor">#endif // MXNET_LIB_API_H_</span></div><div class="ttc" id="classmxnet_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:836</div></div> |
| <div class="ttc" id="namespacemxnet_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:1123</div></div> |
| <div class="ttc" id="namespacemxnet_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:1054</div></div> |
| <div class="ttc" id="classmxnet_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:858</div></div> |
| <div class="ttc" id="namespacemxnet_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:733</div></div> |
| <div class="ttc" id="lib__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="namespacemxnet_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="namespacemxnet_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:966</div></div> |
| <div class="ttc" id="lib__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="classmxnet_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:703</div></div> |
| <div class="ttc" id="namespacemxnet_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:1129</div></div> |
| <div class="ttc" id="classmxnet_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:775</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048">mxnet::ext::kUNSET</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:257</div></div> |
| <div class="ttc" id="classmxnet_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:571</div></div> |
| <div class="ttc" id="classmxnet_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:671</div></div> |
| <div class="ttc" id="classmxnet_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:1135</div></div> |
| <div class="ttc" id="classmxnet_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:417</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6">mxnet::ext::kInt8</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:255</div></div> |
| <div class="ttc" id="namespacemxnet_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:1079</div></div> |
| <div class="ttc" id="classmxnet_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:863</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573a71b2f851779936d824bf2c235ed94a8d"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a71b2f851779936d824bf2c235ed94a8d">kDLOpenCL</a></div><div class="ttdoc">OpenCL devices. </div><div class="ttdef"><b>Definition:</b> lib_api.h:112</div></div> |
| <div class="ttc" id="classmxnet_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:573</div></div> |
| <div class="ttc" id="lib__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="lib__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="classmxnet_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:1132</div></div> |
| <div class="ttc" id="lib__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="classmxnet_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:887</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573a36ced4bf0304d68b5c633ddba5633e66"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a36ced4bf0304d68b5c633ddba5633e66">kDLMetal</a></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="structmxnet_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="namespacemxnet_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:1104</div></div> |
| <div class="ttc" id="namespacemxnet_html"><div class="ttname"><a href="namespacemxnet.html">mxnet</a></div><div class="ttdoc">namespace of mxnet </div><div class="ttdef"><b>Definition:</b> api_registry.h:33</div></div> |
| <div class="ttc" id="lib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:145</div></div> |
| <div class="ttc" id="namespacemxnet_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:1067</div></div> |
| <div class="ttc" id="lib__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:1148</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573adeeb5e6f50f4309676e49f7a12512171"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573adeeb5e6f50f4309676e49f7a12512171">kDLROCM</a></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="structDLContext_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="classmxnet_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:574</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680">kDLGPU</a></div><div class="ttdoc">CUDA GPU device. </div><div class="ttdef"><b>Definition:</b> lib_api.h:105</div></div> |
| <div class="ttc" id="structmxnet_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:549</div></div> |
| <div class="ttc" id="classmxnet_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:710</div></div> |
| <div class="ttc" id="namespacemxnet_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:1049</div></div> |
| <div class="ttc" id="lib__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="structmxnet_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="structmxnet_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="namespacemxnet_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="lib__api_8h_html_a49db572e18d84c47738fc746568fb573aea5cdddfa3ac11130048ad8e2a012363"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573aea5cdddfa3ac11130048ad8e2a012363">kDLExtDev</a></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="lib__api_8h_html_a49db572e18d84c47738fc746568fb573a61cf710df30b59b43a5e34e37ecd1973"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a61cf710df30b59b43a5e34e37ecd1973">kDLCPUPinned</a></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="classmxnet_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:780</div></div> |
| <div class="ttc" id="structmxnet_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:359</div></div> |
| <div class="ttc" id="lib__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="lib__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="namespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241">mxnet::ext::MAP</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:495</div></div> |
| <div class="ttc" id="namespacemxnet_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:799</div></div> |
| <div class="ttc" id="lib__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="lib__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="lib__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="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8">mxnet::ext::kInt32</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:254</div></div> |
| <div class="ttc" id="structmxnet_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:535</div></div> |
| <div class="ttc" id="structmxnet_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:320</div></div> |
| <div class="ttc" id="lib__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="classmxnet_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:575</div></div> |
| <div class="ttc" id="namespacemxnet_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:370</div></div> |
| <div class="ttc" id="namespacemxnet_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:971</div></div> |
| <div class="ttc" id="lib__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="lib__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="lib__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="lib__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="namespacemxnet_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:740</div></div> |
| <div class="ttc" id="namespacemxnet_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="namespacemxnet_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:1071</div></div> |
| <div class="ttc" id="classmxnet_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:570</div></div> |
| <div class="ttc" id="lib__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="classmxnet_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:1137</div></div> |
| <div class="ttc" id="classmxnet_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:860</div></div> |
| <div class="ttc" id="classmxnet_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:572</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2">mxnet::ext::kUint8</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:253</div></div> |
| <div class="ttc" id="classmxnet_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:815</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869">mxnet::ext::kInt64</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:256</div></div> |
| <div class="ttc" id="structmxnet_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:536</div></div> |
| <div class="ttc" id="namespacemxnet_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:822</div></div> |
| <div class="ttc" id="structmxnet_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:353</div></div> |
| <div class="ttc" id="namespacemxnet_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:1009</div></div> |
| <div class="ttc" id="lib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:143</div></div> |
| <div class="ttc" id="classmxnet_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:779</div></div> |
| <div class="ttc" id="classmxnet_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:638</div></div> |
| <div class="ttc" id="namespacemxnet_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:1064</div></div> |
| <div class="ttc" id="structmxnet_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:550</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704">mxnet::ext::kFloat32</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:250</div></div> |
| <div class="ttc" id="classmxnet_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="classmxnet_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:750</div></div> |
| <div class="ttc" id="lib__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="classmxnet_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:582</div></div> |
| <div class="ttc" id="namespacemxnet_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:953</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573a31aefcffd60630580d74bf055c2c3ba0"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573a31aefcffd60630580d74bf055c2c3ba0">kDLVulkan</a></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="lib__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="classmxnet_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:781</div></div> |
| <div class="ttc" id="lib__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="structmxnet_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="namespacemxnet_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:729</div></div> |
| <div class="ttc" id="classmxnet_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:865</div></div> |
| <div class="ttc" id="namespacemxnet_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:826</div></div> |
| <div class="ttc" id="dlpack_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="lib__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="namespacemxnet_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:383</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573af063bf37ea24c1839d3a78703c2468e7"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573af063bf37ea24c1839d3a78703c2468e7">kDLVPI</a></div><div class="ttdoc">Verilog simulator buffer. </div><div class="ttdef"><b>Definition:</b> lib_api.h:118</div></div> |
| <div class="ttc" id="classmxnet_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:431</div></div> |
| <div class="ttc" id="classmxnet_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:788</div></div> |
| <div class="ttc" id="namespacemxnet_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:717</div></div> |
| <div class="ttc" id="lib__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="namespacemxnet_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:495</div></div> |
| <div class="ttc" id="lib__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="lib__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="namespacemxnet_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:983</div></div> |
| <div class="ttc" id="lib__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="lib__api_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a"><div class="ttname"><a href="lib__api_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:144</div></div> |
| <div class="ttc" id="namespacemxnet_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:1098</div></div> |
| <div class="ttc" id="structmxnet_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:498</div></div> |
| <div class="ttc" id="structmxnet_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:548</div></div> |
| <div class="ttc" id="lib__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="namespacemxnet_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:1082</div></div> |
| <div class="ttc" id="namespacemxnet_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="classmxnet_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:783</div></div> |
| <div class="ttc" id="namespacemxnet_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:725</div></div> |
| <div class="ttc" id="namespacemxnet_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:963</div></div> |
| <div class="ttc" id="namespacemxnet_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:1023</div></div> |
| <div class="ttc" id="structmxnet_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:356</div></div> |
| <div class="ttc" id="structmxnet_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:537</div></div> |
| <div class="ttc" id="namespacemxnet_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:819</div></div> |
| <div class="ttc" id="lib__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:1147</div></div> |
| <div class="ttc" id="classmxnet_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:813</div></div> |
| <div class="ttc" id="classmxnet_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:778</div></div> |
| <div class="ttc" id="classmxnet_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:393</div></div> |
| <div class="ttc" id="structmxnet_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:335</div></div> |
| <div class="ttc" id="namespacemxnet_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:722</div></div> |
| <div class="ttc" id="classmxnet_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:782</div></div> |
| <div class="ttc" id="classmxnet_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:698</div></div> |
| <div class="ttc" id="structmxnet_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, currently only "cpu" and "gpu" dev_id is the device index where the tensor locates. </div><div class="ttdef"><b>Definition:</b> lib_api.h:277</div></div> |
| <div class="ttc" id="namespacemxnet_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:385</div></div> |
| <div class="ttc" id="lib__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="lib__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="namespacemxnet_1_1ext_html_a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670">mxnet::ext::MX_FAIL</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:291</div></div> |
| <div class="ttc" id="classmxnet_1_1ext_1_1CustomStatefulOp_html_aa48c6b2f073ef04c5d69060fd1aedf1d"><div class="ttname"><a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#aa48c6b2f073ef04c5d69060fd1aedf1d">mxnet::ext::CustomStatefulOp::create</a></div><div class="ttdeci">static CustomStatefulOp * create(Ts...args)</div><div class="ttdef"><b>Definition:</b> lib_api.h:692</div></div> |
| <div class="ttc" id="namespacemxnet_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:382</div></div> |
| <div class="ttc" id="classmxnet_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:686</div></div> |
| <div class="ttc" id="namespacemxnet_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:1059</div></div> |
| <div class="ttc" id="classmxnet_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:859</div></div> |
| <div class="ttc" id="classmxnet_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:892</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306">mxnet::ext::MX_SUCCESS</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:292</div></div> |
| <div class="ttc" id="classmxnet_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:786</div></div> |
| <div class="ttc" id="lib__api_8h_html_a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e"><div class="ttname"><a href="lib__api_8h.html#a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e">kDLCPU</a></div><div class="ttdoc">CPU device. </div><div class="ttdef"><b>Definition:</b> lib_api.h:103</div></div> |
| <div class="ttc" id="namespacemxnet_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:1020</div></div> |
| <div class="ttc" id="namespacemxnet_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:1043</div></div> |
| <div class="ttc" id="classmxnet_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:554</div></div> |
| <div class="ttc" id="namespacemxnet_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:1101</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901">mxnet::ext::LIST</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:495</div></div> |
| <div class="ttc" id="namespacemxnet_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:372</div></div> |
| <div class="ttc" id="lib__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="namespacemxnet_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:1075</div></div> |
| <div class="ttc" id="classmxnet_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:637</div></div> |
| <div class="ttc" id="structmxnet_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="structmxnet_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="classmxnet_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:636</div></div> |
| <div class="ttc" id="structmxnet_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:363</div></div> |
| <div class="ttc" id="lib__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="namespacemxnet_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:978</div></div> |
| <div class="ttc" id="structmxnet_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="namespacemxnet_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:1014</div></div> |
| <div class="ttc" id="namespacemxnet_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="namespacemxnet_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:1046</div></div> |
| <div class="ttc" id="classmxnet_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:787</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1">mxnet::ext::NUM</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:495</div></div> |
| <div class="ttc" id="lib__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="namespacemxnet_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:1120</div></div> |
| <div class="ttc" id="structmxnet_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:538</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de">mxnet::ext::kCSRStorage</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:269</div></div> |
| <div class="ttc" id="namespacemxnet_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:950</div></div> |
| <div class="ttc" id="classmxnet_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:805</div></div> |
| <div class="ttc" id="structmxnet_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:350</div></div> |
| <div class="ttc" id="classmxnet_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:873</div></div> |
| <div class="ttc" id="structmxnet_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:534</div></div> |
| <div class="ttc" id="classmxnet_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="namespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a">mxnet::ext::kRowSparseStorage</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:267</div></div> |
| <div class="ttc" id="structDLDataType_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="namespacemxnet_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:374</div></div> |
| <div class="ttc" id="structDLTensor_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="lib__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="classmxnet_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:648</div></div> |
| <div class="ttc" id="classmxnet_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="classmxnet_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:678</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e">mxnet::ext::kDefaultStorage</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:265</div></div> |
| <div class="ttc" id="classmxnet_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:569</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884">mxnet::ext::kFloat64</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:251</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103"><div class="ttname"><a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103">mxnet::ext::kFloat16</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:252</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9">mxnet::ext::STR</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:495</div></div> |
| <div class="ttc" id="namespacemxnet_1_1ext_html_a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9"><div class="ttname"><a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9">mxnet::ext::ERR</a></div><div class="ttdef"><b>Definition:</b> lib_api.h:495</div></div> |
| <div class="ttc" id="namespacemxnet_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:737</div></div> |
| <div class="ttc" id="namespacemxnet_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:1126</div></div> |
| <div class="ttc" id="classmxnet_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:445</div></div> |
| <div class="ttc" id="classmxnet_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:861</div></div> |
| <div class="ttc" id="structmxnet_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:366</div></div> |
| <div class="ttc" id="namespacemxnet_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:988</div></div> |
| <div class="ttc" id="lib__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="lib__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="structmxnet_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:347</div></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Jan 5 2023 00:58:42 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.13 |
| </small></address> |
| </body> |
| </html> |
