| <!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: mxnet::ext Namespace Reference</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="namespacemxnet.html">mxnet</a></li><li class="navelem"><a class="el" href="namespacemxnet_1_1ext.html">ext</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="summary"> |
| <a href="#nested-classes">Classes</a> | |
| <a href="#typedef-members">Typedefs</a> | |
| <a href="#enum-members">Enumerations</a> | |
| <a href="#func-members">Functions</a> </div> |
| <div class="headertitle"> |
| <div class="title">mxnet::ext Namespace Reference</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="nested-classes"></a> |
| Classes</h2></td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1CustomOp.html">CustomOp</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">Class to hold custom operator registration. <a href="classmxnet_1_1ext_1_1CustomOp.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1CustomOpSelector.html">CustomOpSelector</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1CustomPartitioner.html">CustomPartitioner</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">An abstract class for subgraph property. <a href="classmxnet_1_1ext_1_1CustomPartitioner.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1CustomPass.html">CustomPass</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">An abstract class for graph passes. <a href="classmxnet_1_1ext_1_1CustomPass.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">An abstract class for library authors creating stateful op custom library should override Forward and destructor, and has an option to implement Backward. <a href="classmxnet_1_1ext_1_1CustomStatefulOp.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html">CustomStatefulOpWrapper</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">StatefulOp wrapper class to pass to backend OpState. <a href="classmxnet_1_1ext_1_1CustomStatefulOpWrapper.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Graph.html">Graph</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structmxnet_1_1ext_1_1JsonVal.html">JsonVal</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">definition of JSON objects <a href="structmxnet_1_1ext_1_1JsonVal.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight"><a class="el" href="structmxnet_1_1Context.html" title="Context information about the execution environment. ">Context</a> info passing from MXNet <a class="el" href="structmxnet_1_1OpContext.html" title="All the possible information needed by Operator.Forward and Backward This is the superset of RunConte...">OpContext</a> dev_type is string repr of supported context, currently only "cpu" and "gpu" dev_id is the device index where the tensor locates. <a href="structmxnet_1_1ext_1_1MXContext.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1MXerrorMsgs.html">MXerrorMsgs</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structmxnet_1_1ext_1_1MXSparse.html">MXSparse</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">Tensor data structure used by custom operator. <a href="structmxnet_1_1ext_1_1MXTensor.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Node.html">Node</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct  </td><td class="memItemRight" valign="bottom"><a class="el" href="structmxnet_1_1ext_1_1NodeEntry.html">NodeEntry</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight">provide resource APIs memory allocation mechanism to Forward/Backward functions <a href="classmxnet_1_1ext_1_1OpResource.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1PassResource.html">PassResource</a></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:"><td class="memItemLeft" align="right" valign="top">class  </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1ext_1_1Registry.html">Registry</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight"><a class="el" href="classmxnet_1_1ext_1_1Registry.html" title="Registry class to registers things (ops, properties) Singleton class. ">Registry</a> class to registers things (ops, properties) Singleton class. <a href="classmxnet_1_1ext_1_1Registry.html#details">More...</a><br /></td></tr> |
| <tr class="separator:"><td class="memSeparator" colspan="2"> </td></tr> |
| </table><table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="typedef-members"></a> |
| Typedefs</h2></td></tr> |
| <tr class="memitem:a6047dc7436d917e950612a84fd912760"><td class="memItemLeft" align="right" valign="top">typedef void *(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a>) (void *, int)</td></tr> |
| <tr class="memdesc:a6047dc7436d917e950612a84fd912760"><td class="mdescLeft"> </td><td class="mdescRight">resource malloc function to allocate memory inside Forward/Backward functions <a href="#a6047dc7436d917e950612a84fd912760">More...</a><br /></td></tr> |
| <tr class="separator:a6047dc7436d917e950612a84fd912760"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a4128a869e7ed8d1ed7c04037d752300e"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a>) (void *, int, int, int, void **, int64_t **, int64_t **)</td></tr> |
| <tr class="memdesc:a4128a869e7ed8d1ed7c04037d752300e"><td class="mdescLeft"> </td><td class="mdescRight">sparse alloc function to allocate memory inside Forward/Backward functions <a href="#a4128a869e7ed8d1ed7c04037d752300e">More...</a><br /></td></tr> |
| <tr class="separator:a4128a869e7ed8d1ed7c04037d752300e"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a645677b0c72cbb9849069f61c3f45917"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a>) (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)</td></tr> |
| <tr class="memdesc:a645677b0c72cbb9849069f61c3f45917"><td class="mdescLeft"> </td><td class="mdescRight">resource malloc function to allocate ndarrays for graph passes <a href="#a645677b0c72cbb9849069f61c3f45917">More...</a><br /></td></tr> |
| <tr class="separator:a645677b0c72cbb9849069f61c3f45917"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:aebb5b6454e5e75b03cafa3ccebf6f752"><td class="memItemLeft" align="right" valign="top">typedef void * </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mx_stream_t</a></td></tr> |
| <tr class="memdesc:aebb5b6454e5e75b03cafa3ccebf6f752"><td class="mdescLeft"> </td><td class="mdescRight">GPU stream pointer, is void* when not compiled with CUDA. <a href="#aebb5b6454e5e75b03cafa3ccebf6f752">More...</a><br /></td></tr> |
| <tr class="separator:aebb5b6454e5e75b03cafa3ccebf6f752"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a8f6f2a2bef8d1c2ca135ed4137673a13"><td class="memItemLeft" align="right" valign="top">typedef void * </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mx_gpu_rand_t</a></td></tr> |
| <tr class="separator:a8f6f2a2bef8d1c2ca135ed4137673a13"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:acbae2c4272b5ac64651bffbf8f000dd8"><td class="memItemLeft" align="right" valign="top">typedef std::mt19937 </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8">mx_cpu_rand_t</a></td></tr> |
| <tr class="separator:acbae2c4272b5ac64651bffbf8f000dd8"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:afa2624f2a3ccf55f7dd53578896b4603"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a>) (const std::unordered_map< std::string, std::string > &attributes, std::vector< <a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a> > *inputs, std::vector< <a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a> > *outputs, const <a class="el" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a> &res)</td></tr> |
| <tr class="memdesc:afa2624f2a3ccf55f7dd53578896b4603"><td class="mdescLeft"> </td><td class="mdescRight">Custom <a class="el" href="classmxnet_1_1Operator.html" title="Operator interface. Operator defines basic operation unit of optimized computation graph in mxnet...">Operator</a> function templates. <a href="#afa2624f2a3ccf55f7dd53578896b4603">More...</a><br /></td></tr> |
| <tr class="separator:afa2624f2a3ccf55f7dd53578896b4603"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a963277b3eebf164f5103a44ae9acdbfc"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a>) (const std::unordered_map< std::string, std::string > &attributes, int *num_inputs, int *num_outputs)</td></tr> |
| <tr class="separator:a963277b3eebf164f5103a44ae9acdbfc"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a6d159349f953d943811338e60b331869"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a>) (const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *in_types, std::vector< int > *out_types)</td></tr> |
| <tr class="separator:a6d159349f953d943811338e60b331869"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:af05bbe0926a318bb3674c8db6d18039a"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a>) (const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *in_storage_types, std::vector< int > *out_storage_types)</td></tr> |
| <tr class="separator:af05bbe0926a318bb3674c8db6d18039a"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a0bf4c32f0c8f5da90619bf0b1f567ef9"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a>) (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)</td></tr> |
| <tr class="separator:a0bf4c32f0c8f5da90619bf0b1f567ef9"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:abe5578d0e77b2548b8d384ba6df66484"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a>) (const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *input_indices)</td></tr> |
| <tr class="separator:abe5578d0e77b2548b8d384ba6df66484"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a559767743debbc2b39b203185ca67923"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">createOpState_t</a>) (const std::unordered_map< std::string, std::string > &attributes, const <a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> &ctx, const std::vector< std::vector< unsigned int > > &in_shapes, const std::vector< int > in_types, <a class="el" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a> **)</td></tr> |
| <tr class="separator:a559767743debbc2b39b203185ca67923"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a08f1c47db9eeec676f4b003ed01a622d"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a>) (<a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, const std::unordered_map< std::string, std::string > &options)</td></tr> |
| <tr class="memdesc:a08f1c47db9eeec676f4b003ed01a622d"><td class="mdescLeft"> </td><td class="mdescRight">Custom Pass Create function template. <a href="#a08f1c47db9eeec676f4b003ed01a622d">More...</a><br /></td></tr> |
| <tr class="separator:a08f1c47db9eeec676f4b003ed01a622d"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a497cf9a31d84c5ab6e7d41f5cf4789ab"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a>) (const <a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, std::vector< int > *ids, const std::unordered_map< std::string, std::string > &options)</td></tr> |
| <tr class="memdesc:a497cf9a31d84c5ab6e7d41f5cf4789ab"><td class="mdescLeft"> </td><td class="mdescRight">Custom Subgraph Create function template. <a href="#a497cf9a31d84c5ab6e7d41f5cf4789ab">More...</a><br /></td></tr> |
| <tr class="separator:a497cf9a31d84c5ab6e7d41f5cf4789ab"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:afab0ffe74ed0a370ee02a8af00084659"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a>) (const <a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, <a class="el" href="classmxnet_1_1ext_1_1CustomOpSelector.html">CustomOpSelector</a> **sel_inst, const std::unordered_map< std::string, std::string > &options)</td></tr> |
| <tr class="separator:afab0ffe74ed0a370ee02a8af00084659"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a18a96fba494d035a7bc60beb30a07fae"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a>) (const <a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *subgraph, int subgraph_id, bool *accept, const std::unordered_map< std::string, std::string > &options, std::unordered_map< std::string, std::string > *attrs)</td></tr> |
| <tr class="separator:a18a96fba494d035a7bc60beb30a07fae"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a6fb7af175432173dec6120674a1f07a4"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a6fb7af175432173dec6120674a1f07a4">opRegSize_t</a>) (void)</td></tr> |
| <tr class="separator:a6fb7af175432173dec6120674a1f07a4"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a10b027e352f8fd55d9639398807614bd"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a10b027e352f8fd55d9639398807614bd">opRegGet_t</a>) (int idx, const char **name, int *isSGop, const char ***forward_ctx, <a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a> **forward_fp, int *forward_count, const char ***backward_ctx, <a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a> **backward_fp, int *backward_count, const char ***create_op_ctx, <a class="el" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a> **create_op_fp, int *create_op_count, <a class="el" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a> *parse, <a class="el" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a> *type, <a class="el" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a> *stype, <a class="el" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a> *shape, <a class="el" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a> *mutate)</td></tr> |
| <tr class="separator:a10b027e352f8fd55d9639398807614bd"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:aff50c6302fa056352a7838a7c0fba79c"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#aff50c6302fa056352a7838a7c0fba79c">opCallFree_t</a>) (void *ptr)</td></tr> |
| <tr class="separator:aff50c6302fa056352a7838a7c0fba79c"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ac3c274938c9da1e44e9897639af5c0d4"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ac3c274938c9da1e44e9897639af5c0d4">opCallParseAttrs_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a> parseAttrs, const char *const *keys, const char *const *vals, int num, int *num_in, int *num_out)</td></tr> |
| <tr class="separator:ac3c274938c9da1e44e9897639af5c0d4"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ade42ac66b3d7c1e063d0f71eda6394a3"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ade42ac66b3d7c1e063d0f71eda6394a3">opCallInferShape_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a> 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)</td></tr> |
| <tr class="separator:ade42ac66b3d7c1e063d0f71eda6394a3"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a08d9994f1542d34fce1eeabda3a8ad85"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a08d9994f1542d34fce1eeabda3a8ad85">opCallInferType_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a> inferType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</td></tr> |
| <tr class="separator:a08d9994f1542d34fce1eeabda3a8ad85"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ae5f9488dd6bdbea94a72c42382c2b6c8"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ae5f9488dd6bdbea94a72c42382c2b6c8">opCallInferSType_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a> inferSType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</td></tr> |
| <tr class="separator:ae5f9488dd6bdbea94a72c42382c2b6c8"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:af447080bf199456379fd14d68804c1c7"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#af447080bf199456379fd14d68804c1c7">opCallFComp_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a> 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, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, void *cpu_alloc, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc, void *gpu_alloc, void *cuda_stream, <a class="el" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> 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)</td></tr> |
| <tr class="separator:af447080bf199456379fd14d68804c1c7"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ab867c91f53d13cb949e9c9a579fa3da5"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ab867c91f53d13cb949e9c9a579fa3da5">opCallMutateInputs_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a> mutate, const char *const *keys, const char *const *vals, int num, int **mutate_indices, int *indices_size)</td></tr> |
| <tr class="separator:ab867c91f53d13cb949e9c9a579fa3da5"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:adf4bd71f1334318c5258e9421709b03c"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#adf4bd71f1334318c5258e9421709b03c">opCallCreateOpState_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">createOpState_t</a> 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)</td></tr> |
| <tr class="separator:adf4bd71f1334318c5258e9421709b03c"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ae77b4c4a8ed7275fd755b458533c7a70"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ae77b4c4a8ed7275fd755b458533c7a70">opCallDestroyOpState_t</a>) (void *state_op)</td></tr> |
| <tr class="separator:ae77b4c4a8ed7275fd755b458533c7a70"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ada6444201d1204384d1287d91a4c3fc7"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ada6444201d1204384d1287d91a4c3fc7">opCallFStatefulComp_t</a>) (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, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, void *cpu_alloc, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc, void *gpu_alloc, void *stream, <a class="el" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> 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)</td></tr> |
| <tr class="separator:ada6444201d1204384d1287d91a4c3fc7"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a8a4d41b35b45f3644497d9a8f717b6a3"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a8a4d41b35b45f3644497d9a8f717b6a3">partRegSize_t</a>) (void)</td></tr> |
| <tr class="separator:a8a4d41b35b45f3644497d9a8f717b6a3"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a0b6d87bb4c06678d0be1dd3ad5916a1b"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a0b6d87bb4c06678d0be1dd3ad5916a1b">partRegGetCount_t</a>) (int idx, const char **name)</td></tr> |
| <tr class="separator:a0b6d87bb4c06678d0be1dd3ad5916a1b"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ae628507b542734ee29ddf2f499fbdb05"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ae628507b542734ee29ddf2f499fbdb05">partRegGet_t</a>) (int part_idx, int stg_idx, const char **strategy, <a class="el" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> *supportedOps, <a class="el" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> *createSelector, <a class="el" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> *reviewSubgraph, const char **op_name)</td></tr> |
| <tr class="separator:ae628507b542734ee29ddf2f499fbdb05"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a03342e626369680bc842ec763588a91e"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a03342e626369680bc842ec763588a91e">partCallSupportedOps_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> supportedOps, const char *json, int num_ids, int *ids, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</td></tr> |
| <tr class="separator:a03342e626369680bc842ec763588a91e"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ad62142c24acef23ecd62e7fd53000349"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ad62142c24acef23ecd62e7fd53000349">partCallCreateSelector_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> createSelector, const char *json, void **selector, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</td></tr> |
| <tr class="separator:ad62142c24acef23ecd62e7fd53000349"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a462103a47b8f387ee5a31393c0a47c1a"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a462103a47b8f387ee5a31393c0a47c1a">partCallSelect_t</a>) (void *sel_inst, int nodeID, int *selected)</td></tr> |
| <tr class="separator:a462103a47b8f387ee5a31393c0a47c1a"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ad2fed961bb0a655fb91fc6ee4c26162a"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ad2fed961bb0a655fb91fc6ee4c26162a">partCallSelectInput_t</a>) (void *sel_inst, int nodeID, int input_nodeID, int *selected)</td></tr> |
| <tr class="separator:ad2fed961bb0a655fb91fc6ee4c26162a"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a5b53f5861183811ba8fc2cdf9dbee5f6"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a5b53f5861183811ba8fc2cdf9dbee5f6">partCallSelectOutput_t</a>) (void *sel_inst, int nodeID, int output_nodeID, int *selected)</td></tr> |
| <tr class="separator:a5b53f5861183811ba8fc2cdf9dbee5f6"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a7ad34e214897e21a5c10bc57ddf1b51a"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a7ad34e214897e21a5c10bc57ddf1b51a">partCallFilter_t</a>) (void *sel_inst, int *candidates, int num_candidates, int **keep, int *num_keep)</td></tr> |
| <tr class="separator:a7ad34e214897e21a5c10bc57ddf1b51a"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a32fc355e422990910299065269f10459"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a32fc355e422990910299065269f10459">partCallReset_t</a>) (void *sel_inst)</td></tr> |
| <tr class="separator:a32fc355e422990910299065269f10459"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:aefb1c0f85ebec11dd8e4f8720392ce4a"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#aefb1c0f85ebec11dd8e4f8720392ce4a">partCallReviewSubgraph_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> 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)</td></tr> |
| <tr class="separator:aefb1c0f85ebec11dd8e4f8720392ce4a"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a6f47c7152edff58ce548870e7c2de0ab"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a6f47c7152edff58ce548870e7c2de0ab">passRegSize_t</a>) (void)</td></tr> |
| <tr class="separator:a6f47c7152edff58ce548870e7c2de0ab"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a83e21d3dbcb68574f93c95db0a83ce31"><td class="memItemLeft" align="right" valign="top">typedef void(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a83e21d3dbcb68574f93c95db0a83ce31">passRegGet_t</a>) (int pass_idx, <a class="el" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> *graphPass, const char **pass_name)</td></tr> |
| <tr class="separator:a83e21d3dbcb68574f93c95db0a83ce31"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a4d96003e8adf76b07e993b83f7cef5bd"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a4d96003e8adf76b07e993b83f7cef5bd">passCallGraphPass_t</a>) (<a class="el" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> 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, <a class="el" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc, const void *nd_alloc)</td></tr> |
| <tr class="separator:a4d96003e8adf76b07e993b83f7cef5bd"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a9089010cd352a7e5a36fc3ade38dd221"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a9089010cd352a7e5a36fc3ade38dd221">initialize_t</a>) (int version)</td></tr> |
| <tr class="separator:a9089010cd352a7e5a36fc3ade38dd221"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a69ba2ade8bd893eb8461a21cd8f60eb5"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a69ba2ade8bd893eb8461a21cd8f60eb5">opVersion_t</a>) ()</td></tr> |
| <tr class="separator:a69ba2ade8bd893eb8461a21cd8f60eb5"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a82c4b41a9a9a1160f1043d10dc9fcf50"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a82c4b41a9a9a1160f1043d10dc9fcf50">msgSize_t</a>) (void)</td></tr> |
| <tr class="separator:a82c4b41a9a9a1160f1043d10dc9fcf50"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:aa7339fd6ce1c23dd296015f450463cc4"><td class="memItemLeft" align="right" valign="top">typedef int(* </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#aa7339fd6ce1c23dd296015f450463cc4">msgGet_t</a>) (int idx, const char **msg)</td></tr> |
| <tr class="separator:aa7339fd6ce1c23dd296015f450463cc4"><td class="memSeparator" colspan="2"> </td></tr> |
| </table><table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="enum-members"></a> |
| Enumerations</h2></td></tr> |
| <tr class="memitem:af63051445af41291367314796a35ef9f"><td class="memItemLeft" align="right" valign="top">enum  </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">MXDType</a> { <br /> |
|   <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704">kFloat32</a> = 0, |
| <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884">kFloat64</a> = 1, |
| <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103">kFloat16</a> = 2, |
| <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2">kUint8</a> = 3, |
| <br /> |
|   <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8">kInt32</a> = 4, |
| <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6">kInt8</a> = 5, |
| <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869">kInt64</a> = 6, |
| <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048">kUNSET</a> = 100 |
| <br /> |
| }<tr class="memdesc:af63051445af41291367314796a35ef9f"><td class="mdescLeft"> </td><td class="mdescRight">Tensor data type, consistent with mshadow data type. <a href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">More...</a><br /></td></tr> |
| </td></tr> |
| <tr class="separator:af63051445af41291367314796a35ef9f"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a390d7b0a8597ac789c1439cb87948166"><td class="memItemLeft" align="right" valign="top">enum  </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">MXStorageType</a> { <a class="el" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e">kDefaultStorage</a> = 0, |
| <a class="el" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a">kRowSparseStorage</a> = 1, |
| <a class="el" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de">kCSRStorage</a> = 2 |
| }</td></tr> |
| <tr class="separator:a390d7b0a8597ac789c1439cb87948166"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a2a16b18af2cdbff088a3c76d62b39b40"><td class="memItemLeft" align="right" valign="top">enum  </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a> { <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670">MX_FAIL</a> = 0, |
| <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306">MX_SUCCESS</a> = 1 |
| }</td></tr> |
| <tr class="separator:a2a16b18af2cdbff088a3c76d62b39b40"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a3bef475e5c2518a5fa6202ac47ce07d1"><td class="memItemLeft" align="right" valign="top">enum  </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">JsonType</a> { <br /> |
|   <a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9">ERR</a>, |
| <a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9">STR</a>, |
| <a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1">NUM</a>, |
| <a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901">LIST</a>, |
| <br /> |
|   <a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241">MAP</a> |
| <br /> |
| }<tr class="memdesc:a3bef475e5c2518a5fa6202ac47ce07d1"><td class="mdescLeft"> </td><td class="mdescRight">Json utility to parse serialized subgraph symbol. <a href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">More...</a><br /></td></tr> |
| </td></tr> |
| <tr class="separator:a3bef475e5c2518a5fa6202ac47ce07d1"><td class="memSeparator" colspan="2"> </td></tr> |
| </table><table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a> |
| Functions</h2></td></tr> |
| <tr class="memitem:ac19085fc1abe2e392d1305d9ac3bc015"><td class="memItemLeft" align="right" valign="top">std::string </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#ac19085fc1abe2e392d1305d9ac3bc015">getShapeAt</a> (const std::string &shape, unsigned index)</td></tr> |
| <tr class="separator:ac19085fc1abe2e392d1305d9ac3bc015"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a8b389f7b678d499fd8d5ba79f089c95e"><td class="memItemLeft" align="right" valign="top">std::string </td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemxnet_1_1ext.html#a8b389f7b678d499fd8d5ba79f089c95e">getDtypeAt</a> (const std::string &dtype, unsigned index)</td></tr> |
| <tr class="separator:a8b389f7b678d499fd8d5ba79f089c95e"><td class="memSeparator" colspan="2"> </td></tr> |
| </table> |
| <h2 class="groupheader">Typedef Documentation</h2> |
| <a id="a559767743debbc2b39b203185ca67923"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a559767743debbc2b39b203185ca67923">◆ </a></span>createOpState_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::createOpState_t) (const std::unordered_map< std::string, std::string > &attributes, const <a class="el" href="structmxnet_1_1ext_1_1MXContext.html">MXContext</a> &ctx, const std::vector< std::vector< unsigned int > > &in_shapes, const std::vector< int > in_types, <a class="el" href="classmxnet_1_1ext_1_1CustomStatefulOp.html">CustomStatefulOp</a> **)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="afab0ffe74ed0a370ee02a8af00084659"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#afab0ffe74ed0a370ee02a8af00084659">◆ </a></span>createSelector_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::createSelector_t) (const <a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, <a class="el" href="classmxnet_1_1ext_1_1CustomOpSelector.html">CustomOpSelector</a> **sel_inst, const std::unordered_map< std::string, std::string > &options)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="afa2624f2a3ccf55f7dd53578896b4603"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#afa2624f2a3ccf55f7dd53578896b4603">◆ </a></span>fcomp_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::fcomp_t) (const std::unordered_map< std::string, std::string > &attributes, std::vector< <a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a> > *inputs, std::vector< <a class="el" href="structmxnet_1_1ext_1_1MXTensor.html">MXTensor</a> > *outputs, const <a class="el" href="classmxnet_1_1ext_1_1OpResource.html">OpResource</a> &res)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Custom <a class="el" href="classmxnet_1_1Operator.html" title="Operator interface. Operator defines basic operation unit of optimized computation graph in mxnet...">Operator</a> function templates. </p> |
| |
| </div> |
| </div> |
| <a id="a08f1c47db9eeec676f4b003ed01a622d"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a08f1c47db9eeec676f4b003ed01a622d">◆ </a></span>graphPass_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::graphPass_t) (<a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, const std::unordered_map< std::string, std::string > &options)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Custom Pass Create function template. </p> |
| |
| </div> |
| </div> |
| <a id="a0bf4c32f0c8f5da90619bf0b1f567ef9"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a0bf4c32f0c8f5da90619bf0b1f567ef9">◆ </a></span>inferShape_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="af05bbe0926a318bb3674c8db6d18039a"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af05bbe0926a318bb3674c8db6d18039a">◆ </a></span>inferSType_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::inferSType_t) (const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *in_storage_types, std::vector< int > *out_storage_types)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a6d159349f953d943811338e60b331869"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a6d159349f953d943811338e60b331869">◆ </a></span>inferType_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::inferType_t) (const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *in_types, std::vector< int > *out_types)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a9089010cd352a7e5a36fc3ade38dd221"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a9089010cd352a7e5a36fc3ade38dd221">◆ </a></span>initialize_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::initialize_t) (int version)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="aa7339fd6ce1c23dd296015f450463cc4"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#aa7339fd6ce1c23dd296015f450463cc4">◆ </a></span>msgGet_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::msgGet_t) (int idx, const char **msg)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a82c4b41a9a9a1160f1043d10dc9fcf50"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a82c4b41a9a9a1160f1043d10dc9fcf50">◆ </a></span>msgSize_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::msgSize_t) (void)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="abe5578d0e77b2548b8d384ba6df66484"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#abe5578d0e77b2548b8d384ba6df66484">◆ </a></span>mutateInputs_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::mutateInputs_t) (const std::unordered_map< std::string, std::string > &attributes, std::vector< int > *input_indices)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="acbae2c4272b5ac64651bffbf8f000dd8"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#acbae2c4272b5ac64651bffbf8f000dd8">◆ </a></span>mx_cpu_rand_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef std::mt19937 <a class="el" href="namespacemxnet_1_1ext.html#acbae2c4272b5ac64651bffbf8f000dd8">mxnet::ext::mx_cpu_rand_t</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a8f6f2a2bef8d1c2ca135ed4137673a13"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a8f6f2a2bef8d1c2ca135ed4137673a13">◆ </a></span>mx_gpu_rand_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void* <a class="el" href="namespacemxnet_1_1ext.html#a8f6f2a2bef8d1c2ca135ed4137673a13">mxnet::ext::mx_gpu_rand_t</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="aebb5b6454e5e75b03cafa3ccebf6f752"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#aebb5b6454e5e75b03cafa3ccebf6f752">◆ </a></span>mx_stream_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void* <a class="el" href="namespacemxnet_1_1ext.html#aebb5b6454e5e75b03cafa3ccebf6f752">mxnet::ext::mx_stream_t</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>GPU stream pointer, is void* when not compiled with CUDA. </p> |
| |
| </div> |
| </div> |
| <a id="a645677b0c72cbb9849069f61c3f45917"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a645677b0c72cbb9849069f61c3f45917">◆ </a></span>nd_malloc_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>resource malloc function to allocate ndarrays for graph passes </p> |
| |
| </div> |
| </div> |
| <a id="adf4bd71f1334318c5258e9421709b03c"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#adf4bd71f1334318c5258e9421709b03c">◆ </a></span>opCallCreateOpState_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallCreateOpState_t) (<a class="el" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">createOpState_t</a> 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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ae77b4c4a8ed7275fd755b458533c7a70"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ae77b4c4a8ed7275fd755b458533c7a70">◆ </a></span>opCallDestroyOpState_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallDestroyOpState_t) (void *state_op)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="af447080bf199456379fd14d68804c1c7"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af447080bf199456379fd14d68804c1c7">◆ </a></span>opCallFComp_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallFComp_t) (<a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">fcomp_t</a> 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, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, void *cpu_alloc, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc, void *gpu_alloc, void *cuda_stream, <a class="el" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> 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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="aff50c6302fa056352a7838a7c0fba79c"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#aff50c6302fa056352a7838a7c0fba79c">◆ </a></span>opCallFree_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallFree_t) (void *ptr)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ada6444201d1204384d1287d91a4c3fc7"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ada6444201d1204384d1287d91a4c3fc7">◆ </a></span>opCallFStatefulComp_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::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, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> cpu_malloc, void *cpu_alloc, <a class="el" href="namespacemxnet_1_1ext.html#a6047dc7436d917e950612a84fd912760">xpu_malloc_t</a> gpu_malloc, void *gpu_alloc, void *stream, <a class="el" href="namespacemxnet_1_1ext.html#a4128a869e7ed8d1ed7c04037d752300e">sparse_malloc_t</a> 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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ade42ac66b3d7c1e063d0f71eda6394a3"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ade42ac66b3d7c1e063d0f71eda6394a3">◆ </a></span>opCallInferShape_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallInferShape_t) (<a class="el" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">inferShape_t</a> 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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ae5f9488dd6bdbea94a72c42382c2b6c8"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ae5f9488dd6bdbea94a72c42382c2b6c8">◆ </a></span>opCallInferSType_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallInferSType_t) (<a class="el" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">inferSType_t</a> inferSType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a08d9994f1542d34fce1eeabda3a8ad85"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a08d9994f1542d34fce1eeabda3a8ad85">◆ </a></span>opCallInferType_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallInferType_t) (<a class="el" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">inferType_t</a> inferType, const char *const *keys, const char *const *vals, int num, int *intypes, int num_in, int *outtypes, int num_out)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ab867c91f53d13cb949e9c9a579fa3da5"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ab867c91f53d13cb949e9c9a579fa3da5">◆ </a></span>opCallMutateInputs_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallMutateInputs_t) (<a class="el" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mutateInputs_t</a> mutate, const char *const *keys, const char *const *vals, int num, int **mutate_indices, int *indices_size)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ac3c274938c9da1e44e9897639af5c0d4"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ac3c274938c9da1e44e9897639af5c0d4">◆ </a></span>opCallParseAttrs_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opCallParseAttrs_t) (<a class="el" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">parseAttrs_t</a> parseAttrs, const char *const *keys, const char *const *vals, int num, int *num_in, int *num_out)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a10b027e352f8fd55d9639398807614bd"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a10b027e352f8fd55d9639398807614bd">◆ </a></span>opRegGet_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opRegGet_t) (int idx, const char **name, int *isSGop, const char ***forward_ctx, <a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a> **forward_fp, int *forward_count, const char ***backward_ctx, <a class="el" href="namespacemxnet_1_1ext.html#afa2624f2a3ccf55f7dd53578896b4603">mxnet::ext::fcomp_t</a> **backward_fp, int *backward_count, const char ***create_op_ctx, <a class="el" href="namespacemxnet_1_1ext.html#a559767743debbc2b39b203185ca67923">mxnet::ext::createOpState_t</a> **create_op_fp, int *create_op_count, <a class="el" href="namespacemxnet_1_1ext.html#a963277b3eebf164f5103a44ae9acdbfc">mxnet::ext::parseAttrs_t</a> *parse, <a class="el" href="namespacemxnet_1_1ext.html#a6d159349f953d943811338e60b331869">mxnet::ext::inferType_t</a> *type, <a class="el" href="namespacemxnet_1_1ext.html#af05bbe0926a318bb3674c8db6d18039a">mxnet::ext::inferSType_t</a> *stype, <a class="el" href="namespacemxnet_1_1ext.html#a0bf4c32f0c8f5da90619bf0b1f567ef9">mxnet::ext::inferShape_t</a> *shape, <a class="el" href="namespacemxnet_1_1ext.html#abe5578d0e77b2548b8d384ba6df66484">mxnet::ext::mutateInputs_t</a> *mutate)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a6fb7af175432173dec6120674a1f07a4"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a6fb7af175432173dec6120674a1f07a4">◆ </a></span>opRegSize_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opRegSize_t) (void)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a69ba2ade8bd893eb8461a21cd8f60eb5"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a69ba2ade8bd893eb8461a21cd8f60eb5">◆ </a></span>opVersion_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::opVersion_t) ()</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a963277b3eebf164f5103a44ae9acdbfc"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a963277b3eebf164f5103a44ae9acdbfc">◆ </a></span>parseAttrs_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::parseAttrs_t) (const std::unordered_map< std::string, std::string > &attributes, int *num_inputs, int *num_outputs)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ad62142c24acef23ecd62e7fd53000349"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ad62142c24acef23ecd62e7fd53000349">◆ </a></span>partCallCreateSelector_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::partCallCreateSelector_t) (<a class="el" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> createSelector, const char *json, void **selector, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a7ad34e214897e21a5c10bc57ddf1b51a"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a7ad34e214897e21a5c10bc57ddf1b51a">◆ </a></span>partCallFilter_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::partCallFilter_t) (void *sel_inst, int *candidates, int num_candidates, int **keep, int *num_keep)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a32fc355e422990910299065269f10459"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a32fc355e422990910299065269f10459">◆ </a></span>partCallReset_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::partCallReset_t) (void *sel_inst)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="aefb1c0f85ebec11dd8e4f8720392ce4a"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#aefb1c0f85ebec11dd8e4f8720392ce4a">◆ </a></span>partCallReviewSubgraph_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::partCallReviewSubgraph_t) (<a class="el" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> 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)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a462103a47b8f387ee5a31393c0a47c1a"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a462103a47b8f387ee5a31393c0a47c1a">◆ </a></span>partCallSelect_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::partCallSelect_t) (void *sel_inst, int nodeID, int *selected)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ad2fed961bb0a655fb91fc6ee4c26162a"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ad2fed961bb0a655fb91fc6ee4c26162a">◆ </a></span>partCallSelectInput_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::partCallSelectInput_t) (void *sel_inst, int nodeID, int input_nodeID, int *selected)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a5b53f5861183811ba8fc2cdf9dbee5f6"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a5b53f5861183811ba8fc2cdf9dbee5f6">◆ </a></span>partCallSelectOutput_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::partCallSelectOutput_t) (void *sel_inst, int nodeID, int output_nodeID, int *selected)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a03342e626369680bc842ec763588a91e"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a03342e626369680bc842ec763588a91e">◆ </a></span>partCallSupportedOps_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::partCallSupportedOps_t) (<a class="el" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> supportedOps, const char *json, int num_ids, int *ids, const char *const *opt_keys, const char *const *opt_vals, int num_opts)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ae628507b542734ee29ddf2f499fbdb05"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ae628507b542734ee29ddf2f499fbdb05">◆ </a></span>partRegGet_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::partRegGet_t) (int part_idx, int stg_idx, const char **strategy, <a class="el" href="namespacemxnet_1_1ext.html#a497cf9a31d84c5ab6e7d41f5cf4789ab">supportedOps_t</a> *supportedOps, <a class="el" href="namespacemxnet_1_1ext.html#afab0ffe74ed0a370ee02a8af00084659">createSelector_t</a> *createSelector, <a class="el" href="namespacemxnet_1_1ext.html#a18a96fba494d035a7bc60beb30a07fae">reviewSubgraph_t</a> *reviewSubgraph, const char **op_name)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a0b6d87bb4c06678d0be1dd3ad5916a1b"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a0b6d87bb4c06678d0be1dd3ad5916a1b">◆ </a></span>partRegGetCount_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::partRegGetCount_t) (int idx, const char **name)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a8a4d41b35b45f3644497d9a8f717b6a3"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a8a4d41b35b45f3644497d9a8f717b6a3">◆ </a></span>partRegSize_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::partRegSize_t) (void)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a4d96003e8adf76b07e993b83f7cef5bd"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a4d96003e8adf76b07e993b83f7cef5bd">◆ </a></span>passCallGraphPass_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::passCallGraphPass_t) (<a class="el" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> 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, <a class="el" href="namespacemxnet_1_1ext.html#a645677b0c72cbb9849069f61c3f45917">nd_malloc_t</a> nd_malloc, const void *nd_alloc)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a83e21d3dbcb68574f93c95db0a83ce31"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a83e21d3dbcb68574f93c95db0a83ce31">◆ </a></span>passRegGet_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::passRegGet_t) (int pass_idx, <a class="el" href="namespacemxnet_1_1ext.html#a08f1c47db9eeec676f4b003ed01a622d">graphPass_t</a> *graphPass, const char **pass_name)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a6f47c7152edff58ce548870e7c2de0ab"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a6f47c7152edff58ce548870e7c2de0ab">◆ </a></span>passRegSize_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef int(* mxnet::ext::passRegSize_t) (void)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a18a96fba494d035a7bc60beb30a07fae"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a18a96fba494d035a7bc60beb30a07fae">◆ </a></span>reviewSubgraph_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::reviewSubgraph_t) (const <a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *subgraph, int subgraph_id, bool *accept, const std::unordered_map< std::string, std::string > &options, std::unordered_map< std::string, std::string > *attrs)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a4128a869e7ed8d1ed7c04037d752300e"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a4128a869e7ed8d1ed7c04037d752300e">◆ </a></span>sparse_malloc_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void(* mxnet::ext::sparse_malloc_t) (void *, int, int, int, void **, int64_t **, int64_t **)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>sparse alloc function to allocate memory inside Forward/Backward functions </p> |
| |
| </div> |
| </div> |
| <a id="a497cf9a31d84c5ab6e7d41f5cf4789ab"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a497cf9a31d84c5ab6e7d41f5cf4789ab">◆ </a></span>supportedOps_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">MXReturnValue</a>(* mxnet::ext::supportedOps_t) (const <a class="el" href="classmxnet_1_1ext_1_1Graph.html">mxnet::ext::Graph</a> *graph, std::vector< int > *ids, const std::unordered_map< std::string, std::string > &options)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Custom Subgraph Create function template. </p> |
| |
| </div> |
| </div> |
| <a id="a6047dc7436d917e950612a84fd912760"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a6047dc7436d917e950612a84fd912760">◆ </a></span>xpu_malloc_t</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">typedef void*(* mxnet::ext::xpu_malloc_t) (void *, int)</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>resource malloc function to allocate memory inside Forward/Backward functions </p> |
| |
| </div> |
| </div> |
| <h2 class="groupheader">Enumeration Type Documentation</h2> |
| <a id="a3bef475e5c2518a5fa6202ac47ce07d1"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a3bef475e5c2518a5fa6202ac47ce07d1">◆ </a></span>JsonType</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">enum <a class="el" href="namespacemxnet_1_1ext.html#a3bef475e5c2518a5fa6202ac47ce07d1">mxnet::ext::JsonType</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Json utility to parse serialized subgraph symbol. </p> |
| <p>Types of JSON objects </p> |
| <table class="fieldtable"> |
| <tr><th colspan="2">Enumerator</th></tr><tr><td class="fieldname"><a id="a3bef475e5c2518a5fa6202ac47ce07d1a46668b3ec5c8b0e51345e8f6e52e63c9"></a>ERR </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a3bef475e5c2518a5fa6202ac47ce07d1a83a5d030292beb0d1789db72e30601f9"></a>STR </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a3bef475e5c2518a5fa6202ac47ce07d1aec218a2fc415a5dfe0586b1cd41eace1"></a>NUM </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a3bef475e5c2518a5fa6202ac47ce07d1aa546c63a655963d12ece2aa35b798901"></a>LIST </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a3bef475e5c2518a5fa6202ac47ce07d1a8e3a4528b909298a732acf81acbef241"></a>MAP </td><td class="fielddoc"></td></tr> |
| </table> |
| |
| </div> |
| </div> |
| <a id="af63051445af41291367314796a35ef9f"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af63051445af41291367314796a35ef9f">◆ </a></span>MXDType</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">enum <a class="el" href="namespacemxnet_1_1ext.html#af63051445af41291367314796a35ef9f">mxnet::ext::MXDType</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Tensor data type, consistent with mshadow data type. </p> |
| <table class="fieldtable"> |
| <tr><th colspan="2">Enumerator</th></tr><tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9fabb2b1b3c7a227b6059fc89b9e86e8704"></a>kFloat32 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9facdbc87b2e662a8bfb834346d1aec6884"></a>kFloat64 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9fabe9d3b09e5b6a6a6ce40c6e6727a0103"></a>kFloat16 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9faace6878fbc1feaa61866f2cde4b92fd2"></a>kUint8 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9faba0176d183c843a93e73f244e8eb08b8"></a>kInt32 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9fa40cebe2fb035da2e0d71a4fbb38e68c6"></a>kInt8 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9faee6141c5d0a7dba64fc8cdacfb6e5869"></a>kInt64 </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="af63051445af41291367314796a35ef9faaac2c2fc5ad0820d77233be7e7935048"></a>kUNSET </td><td class="fielddoc"></td></tr> |
| </table> |
| |
| </div> |
| </div> |
| <a id="a2a16b18af2cdbff088a3c76d62b39b40"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a2a16b18af2cdbff088a3c76d62b39b40">◆ </a></span>MXReturnValue</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">enum <a class="el" href="namespacemxnet_1_1ext.html#a2a16b18af2cdbff088a3c76d62b39b40">mxnet::ext::MXReturnValue</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| <table class="fieldtable"> |
| <tr><th colspan="2">Enumerator</th></tr><tr><td class="fieldname"><a id="a2a16b18af2cdbff088a3c76d62b39b40aeb26a79748e9601cca81a57a5a236670"></a>MX_FAIL </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a2a16b18af2cdbff088a3c76d62b39b40ae5a7150c47b7ca0ca6b0791d0ad14306"></a>MX_SUCCESS </td><td class="fielddoc"></td></tr> |
| </table> |
| |
| </div> |
| </div> |
| <a id="a390d7b0a8597ac789c1439cb87948166"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a390d7b0a8597ac789c1439cb87948166">◆ </a></span>MXStorageType</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">enum <a class="el" href="namespacemxnet_1_1ext.html#a390d7b0a8597ac789c1439cb87948166">mxnet::ext::MXStorageType</a></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| <table class="fieldtable"> |
| <tr><th colspan="2">Enumerator</th></tr><tr><td class="fieldname"><a id="a390d7b0a8597ac789c1439cb87948166aae97a392afb8883280a194eff9bc105e"></a>kDefaultStorage </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a390d7b0a8597ac789c1439cb87948166acf413c3686f4621684b24f62832df56a"></a>kRowSparseStorage </td><td class="fielddoc"></td></tr> |
| <tr><td class="fieldname"><a id="a390d7b0a8597ac789c1439cb87948166aa6581cebefe05ab00873ac5597aae3de"></a>kCSRStorage </td><td class="fielddoc"></td></tr> |
| </table> |
| |
| </div> |
| </div> |
| <h2 class="groupheader">Function Documentation</h2> |
| <a id="a8b389f7b678d499fd8d5ba79f089c95e"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a8b389f7b678d499fd8d5ba79f089c95e">◆ </a></span>getDtypeAt()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">std::string mxnet::ext::getDtypeAt </td> |
| <td>(</td> |
| <td class="paramtype">const std::string & </td> |
| <td class="paramname"><em>dtype</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">unsigned </td> |
| <td class="paramname"><em>index</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="ac19085fc1abe2e392d1305d9ac3bc015"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ac19085fc1abe2e392d1305d9ac3bc015">◆ </a></span>getShapeAt()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">std::string mxnet::ext::getShapeAt </td> |
| <td>(</td> |
| <td class="paramtype">const std::string & </td> |
| <td class="paramname"><em>shape</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">unsigned </td> |
| <td class="paramname"><em>index</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| </div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Jan 5 2023 00:58:44 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> |