| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.17"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>mxnet: mxnet::Imperative Class 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.17 --> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| /* @license-end */ |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| /* @license-end */</script> |
| <div id="main-nav"></div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="namespacemxnet.html">mxnet</a></li><li class="navelem"><a class="el" href="classmxnet_1_1Imperative.html">Imperative</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="summary"> |
| <a href="#nested-classes">Classes</a> | |
| <a href="#pub-methods">Public Member Functions</a> | |
| <a href="#pub-static-methods">Static Public Member Functions</a> | |
| <a href="#friends">Friends</a> | |
| <a href="classmxnet_1_1Imperative-members.html">List of all members</a> </div> |
| <div class="headertitle"> |
| <div class="title">mxnet::Imperative Class Reference</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| |
| <p>runtime functions for <a class="el" href="classmxnet_1_1NDArray.html" title="ndarray interface">NDArray</a> |
| <a href="classmxnet_1_1Imperative.html#details">More...</a></p> |
| |
| <p><code>#include <<a class="el" href="imperative_8h_source.html">imperative.h</a>></code></p> |
| <div class="dynheader"> |
| Collaboration diagram for mxnet::Imperative:</div> |
| <div class="dyncontent"> |
| <div class="center"><img src="classmxnet_1_1Imperative__coll__graph.png" border="0" usemap="#mxnet_1_1Imperative_coll__map" alt="Collaboration graph"/></div> |
| <!-- MAP 0 --> |
| </div> |
| <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_1Imperative_1_1AGInfo.html">AGInfo</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_1Imperative_1_1DCInfo.html">DCInfo</a></td></tr> |
| <tr class="memdesc:"><td class="mdescLeft"> </td><td class="mdescRight"><a class="el" href="classmxnet_1_1Imperative_1_1DCInfo.html" title="DCInfo datastructure to enable deferred computation.">DCInfo</a> datastructure to enable deferred computation. <a href="classmxnet_1_1Imperative_1_1DCInfo.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="pub-methods"></a> |
| Public Member Functions</h2></td></tr> |
| <tr class="memitem:af6024ec51f5fcd73935994f7b43f62ef"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#af6024ec51f5fcd73935994f7b43f62ef">is_training</a> () const</td></tr> |
| <tr class="memdesc:af6024ec51f5fcd73935994f7b43f62ef"><td class="mdescLeft"> </td><td class="mdescRight">whether operator recording is on. <a href="classmxnet_1_1Imperative.html#af6024ec51f5fcd73935994f7b43f62ef">More...</a><br /></td></tr> |
| <tr class="separator:af6024ec51f5fcd73935994f7b43f62ef"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a2c1495dbdc4439fd534307fa981b77f0"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a2c1495dbdc4439fd534307fa981b77f0">set_is_training</a> (bool is_train)</td></tr> |
| <tr class="memdesc:a2c1495dbdc4439fd534307fa981b77f0"><td class="mdescLeft"> </td><td class="mdescRight">turn on or turn off operator recording for autograd. <a href="classmxnet_1_1Imperative.html#a2c1495dbdc4439fd534307fa981b77f0">More...</a><br /></td></tr> |
| <tr class="separator:a2c1495dbdc4439fd534307fa981b77f0"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:acbbd0eafd83cd96a40cce3d01147b1a3"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#acbbd0eafd83cd96a40cce3d01147b1a3">is_recording</a> () const</td></tr> |
| <tr class="memdesc:acbbd0eafd83cd96a40cce3d01147b1a3"><td class="mdescLeft"> </td><td class="mdescRight">whether operator recording is on. <a href="classmxnet_1_1Imperative.html#acbbd0eafd83cd96a40cce3d01147b1a3">More...</a><br /></td></tr> |
| <tr class="separator:acbbd0eafd83cd96a40cce3d01147b1a3"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:abc9227372b3d4f4b5ec8ed33ba6e68ad"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#abc9227372b3d4f4b5ec8ed33ba6e68ad">set_is_recording</a> (bool <a class="el" href="classmxnet_1_1Imperative.html#acbbd0eafd83cd96a40cce3d01147b1a3">is_recording</a>)</td></tr> |
| <tr class="memdesc:abc9227372b3d4f4b5ec8ed33ba6e68ad"><td class="mdescLeft"> </td><td class="mdescRight">turn on or turn off operator recording for autograd. <a href="classmxnet_1_1Imperative.html#abc9227372b3d4f4b5ec8ed33ba6e68ad">More...</a><br /></td></tr> |
| <tr class="separator:abc9227372b3d4f4b5ec8ed33ba6e68ad"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a350f6f69b794358586436f58992a3e38"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a350f6f69b794358586436f58992a3e38">is_deferred_compute</a> () const</td></tr> |
| <tr class="memdesc:a350f6f69b794358586436f58992a3e38"><td class="mdescLeft"> </td><td class="mdescRight">whether deferred compute mode is on. <a href="classmxnet_1_1Imperative.html#a350f6f69b794358586436f58992a3e38">More...</a><br /></td></tr> |
| <tr class="separator:a350f6f69b794358586436f58992a3e38"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a20d520b1969ab70928e6e84898d632c4"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a20d520b1969ab70928e6e84898d632c4">set_is_deferred_compute</a> (bool <a class="el" href="classmxnet_1_1Imperative.html#a350f6f69b794358586436f58992a3e38">is_deferred_compute</a>)</td></tr> |
| <tr class="memdesc:a20d520b1969ab70928e6e84898d632c4"><td class="mdescLeft"> </td><td class="mdescRight">turn on or turn off operator recording for autograd. <a href="classmxnet_1_1Imperative.html#a20d520b1969ab70928e6e84898d632c4">More...</a><br /></td></tr> |
| <tr class="separator:a20d520b1969ab70928e6e84898d632c4"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:af65cc2bc2b45b328d99ba443211b8be7"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#af65cc2bc2b45b328d99ba443211b8be7">is_np_shape</a> () const</td></tr> |
| <tr class="memdesc:af65cc2bc2b45b328d99ba443211b8be7"><td class="mdescLeft"> </td><td class="mdescRight">return current numpy compatibility status, GlobalOn(2), ThreadLocalOn(1), Off(0). <a href="classmxnet_1_1Imperative.html#af65cc2bc2b45b328d99ba443211b8be7">More...</a><br /></td></tr> |
| <tr class="separator:af65cc2bc2b45b328d99ba443211b8be7"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:afc403b976f76aae9dcfc3790d9190832"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#afc403b976f76aae9dcfc3790d9190832">set_is_np_shape</a> (int <a class="el" href="classmxnet_1_1Imperative.html#af65cc2bc2b45b328d99ba443211b8be7">is_np_shape</a>)</td></tr> |
| <tr class="memdesc:afc403b976f76aae9dcfc3790d9190832"><td class="mdescLeft"> </td><td class="mdescRight">specify numpy compatibility off, thread local on or global on. <a href="classmxnet_1_1Imperative.html#afc403b976f76aae9dcfc3790d9190832">More...</a><br /></td></tr> |
| <tr class="separator:afc403b976f76aae9dcfc3790d9190832"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a63ff2923f4b9daa8c35d1a683d17949f"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a63ff2923f4b9daa8c35d1a683d17949f">is_np_default_dtype</a> () const</td></tr> |
| <tr class="memdesc:a63ff2923f4b9daa8c35d1a683d17949f"><td class="mdescLeft"> </td><td class="mdescRight">return current numpy default dtype compatibility status. <a href="classmxnet_1_1Imperative.html#a63ff2923f4b9daa8c35d1a683d17949f">More...</a><br /></td></tr> |
| <tr class="separator:a63ff2923f4b9daa8c35d1a683d17949f"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:afb64b62624152e84ca54d9371268ab40"><td class="memItemLeft" align="right" valign="top">bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#afb64b62624152e84ca54d9371268ab40">set_is_np_default_dtype</a> (bool <a class="el" href="classmxnet_1_1Imperative.html#a63ff2923f4b9daa8c35d1a683d17949f">is_np_default_dtype</a>)</td></tr> |
| <tr class="memdesc:afb64b62624152e84ca54d9371268ab40"><td class="mdescLeft"> </td><td class="mdescRight">specify numpy default dtype off or global on. <a href="classmxnet_1_1Imperative.html#afb64b62624152e84ca54d9371268ab40">More...</a><br /></td></tr> |
| <tr class="separator:afb64b62624152e84ca54d9371268ab40"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a70d6fd0885c9b17d2cacf5122a74d6f6"><td class="memItemLeft" align="right" valign="top"><a class="el" href="namespacemxnet.html#abdd24c24bd41ef5d0eff7628845e7a36">OptConstraint</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a70d6fd0885c9b17d2cacf5122a74d6f6">get_opt_constraints</a> () const</td></tr> |
| <tr class="memdesc:a70d6fd0885c9b17d2cacf5122a74d6f6"><td class="mdescLeft"> </td><td class="mdescRight">return current optimization constraints. <a href="classmxnet_1_1Imperative.html#a70d6fd0885c9b17d2cacf5122a74d6f6">More...</a><br /></td></tr> |
| <tr class="separator:a70d6fd0885c9b17d2cacf5122a74d6f6"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a76a9e1537fff16c402a7cc6062d4d1b0"><td class="memItemLeft" align="right" valign="top"><a class="el" href="namespacemxnet.html#abdd24c24bd41ef5d0eff7628845e7a36">OptConstraint</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a76a9e1537fff16c402a7cc6062d4d1b0">set_opt_constraints</a> (<a class="el" href="namespacemxnet.html#abdd24c24bd41ef5d0eff7628845e7a36">OptConstraint</a> constraints)</td></tr> |
| <tr class="memdesc:a76a9e1537fff16c402a7cc6062d4d1b0"><td class="mdescLeft"> </td><td class="mdescRight">set optimization constraints. <a href="classmxnet_1_1Imperative.html#a76a9e1537fff16c402a7cc6062d4d1b0">More...</a><br /></td></tr> |
| <tr class="separator:a76a9e1537fff16c402a7cc6062d4d1b0"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a2b0f55c31882412c44468d8ba1a97fd1"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a2b0f55c31882412c44468d8ba1a97fd1">RecordOp</a> (<a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> &&attrs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &inputs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &outputs, const <a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> &state=<a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a>(), std::vector< bool > *p_save_inputs=nullptr, std::vector< bool > *p_save_outputs=nullptr)</td></tr> |
| <tr class="memdesc:a2b0f55c31882412c44468d8ba1a97fd1"><td class="mdescLeft"> </td><td class="mdescRight">to record operator, return corresponding node. <a href="classmxnet_1_1Imperative.html#a2b0f55c31882412c44468d8ba1a97fd1">More...</a><br /></td></tr> |
| <tr class="separator:a2b0f55c31882412c44468d8ba1a97fd1"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a2efa644fae290825385c50e2f8feb74d"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a2efa644fae290825385c50e2f8feb74d">RecordDeferredCompute</a> (<a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> &&attrs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &inputs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &outputs)</td></tr> |
| <tr class="memdesc:a2efa644fae290825385c50e2f8feb74d"><td class="mdescLeft"> </td><td class="mdescRight">to record operator, return corresponding node. <a href="classmxnet_1_1Imperative.html#a2efa644fae290825385c50e2f8feb74d">More...</a><br /></td></tr> |
| <tr class="separator:a2efa644fae290825385c50e2f8feb74d"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a0f901368c813c144b8ec6aacac7567db"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classnnvm_1_1Symbol.html">nnvm::Symbol</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a0f901368c813c144b8ec6aacac7567db">GetDeferredComputeSymbol</a> (const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &outputs)</td></tr> |
| <tr class="memdesc:a0f901368c813c144b8ec6aacac7567db"><td class="mdescLeft"> </td><td class="mdescRight">obtain symbol representation of deferred compute session. <a href="classmxnet_1_1Imperative.html#a0f901368c813c144b8ec6aacac7567db">More...</a><br /></td></tr> |
| <tr class="separator:a0f901368c813c144b8ec6aacac7567db"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a16947c6577a274d7bd8dbc3b0aac0107"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a16947c6577a274d7bd8dbc3b0aac0107">SetDeferredComputeVariable</a> (<a class="el" href="classmxnet_1_1NDArrayHandle.html">NDArrayHandle</a> *arrays, <a class="el" href="include_2mxnet_2c__api_8h.html#a301e9a7722b326126defdd39ccd99b37">SymbolHandle</a> *variables, const int num)</td></tr> |
| <tr class="memdesc:a16947c6577a274d7bd8dbc3b0aac0107"><td class="mdescLeft"> </td><td class="mdescRight">associate arrays with variables for deferred compute <a href="classmxnet_1_1Imperative.html#a16947c6577a274d7bd8dbc3b0aac0107">More...</a><br /></td></tr> |
| <tr class="separator:a16947c6577a274d7bd8dbc3b0aac0107"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a10c6d32c54563a974c9265a2397c821f"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a10c6d32c54563a974c9265a2397c821f">DeferredComputeClear</a> (<a class="el" href="classmxnet_1_1NDArrayHandle.html">NDArrayHandle</a> *arrays, const int num)</td></tr> |
| <tr class="memdesc:a10c6d32c54563a974c9265a2397c821f"><td class="mdescLeft"> </td><td class="mdescRight">clear info node associated with array <a href="classmxnet_1_1Imperative.html#a10c6d32c54563a974c9265a2397c821f">More...</a><br /></td></tr> |
| <tr class="separator:a10c6d32c54563a974c9265a2397c821f"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:ae3a1279e546061bf6062817c93dce0dc"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#ae3a1279e546061bf6062817c93dce0dc">Invoke</a> (const <a class="el" href="structmxnet_1_1Context.html">Context</a> &default_ctx, const <a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> &attrs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &inputs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &outputs)</td></tr> |
| <tr class="separator:ae3a1279e546061bf6062817c93dce0dc"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:adb41ed890acd4cbe672f088b0284fe84"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#adb41ed890acd4cbe672f088b0284fe84">InvokeOp</a> (const <a class="el" href="structmxnet_1_1Context.html">Context</a> &ctx, const <a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> &attrs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &inputs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &outputs, const std::vector< <a class="el" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">OpReqType</a> > &req, const <a class="el" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a> dispatch_mode, <a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> state=<a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a>())</td></tr> |
| <tr class="separator:adb41ed890acd4cbe672f088b0284fe84"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a06fa7ec786cf0ea3a0eb7c8a174171cb"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a06fa7ec786cf0ea3a0eb7c8a174171cb">MarkVariables</a> (const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &variables, const std::vector< uint32_t > &grad_reqs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &gradients)</td></tr> |
| <tr class="memdesc:a06fa7ec786cf0ea3a0eb7c8a174171cb"><td class="mdescLeft"> </td><td class="mdescRight">mark variables for computing gradients. <a href="classmxnet_1_1Imperative.html#a06fa7ec786cf0ea3a0eb7c8a174171cb">More...</a><br /></td></tr> |
| <tr class="separator:a06fa7ec786cf0ea3a0eb7c8a174171cb"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:aed3baaf0efb809226df9df765fefaaa0"><td class="memItemLeft" align="right" valign="top">void </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#aed3baaf0efb809226df9df765fefaaa0">DropGrads</a> (const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &variables)</td></tr> |
| <tr class="memdesc:aed3baaf0efb809226df9df765fefaaa0"><td class="mdescLeft"> </td><td class="mdescRight">unmark nonleaf variables to free the memory. <a href="classmxnet_1_1Imperative.html#aed3baaf0efb809226df9df765fefaaa0">More...</a><br /></td></tr> |
| <tr class="separator:aed3baaf0efb809226df9df765fefaaa0"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:afa4950a3cf57431f1d0d639c3f7afa7d"><td class="memItemLeft" align="right" valign="top">std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#afa4950a3cf57431f1d0d639c3f7afa7d">Backward</a> (const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &outputs, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &ograds, const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > &variables, bool is_train, bool retain_graph, bool create_graph)</td></tr> |
| <tr class="memdesc:afa4950a3cf57431f1d0d639c3f7afa7d"><td class="mdescLeft"> </td><td class="mdescRight">compute the gradient of outputs w.r.t variables. <a href="classmxnet_1_1Imperative.html#afa4950a3cf57431f1d0d639c3f7afa7d">More...</a><br /></td></tr> |
| <tr class="separator:afa4950a3cf57431f1d0d639c3f7afa7d"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a8cd9d3861381bbb2e3328b661d8e0d38"><td class="memItemLeft" align="right" valign="top">std::vector< <a class="el" href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">nnvm::ObjectPtr</a> > </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a8cd9d3861381bbb2e3328b661d8e0d38">ListNonleafVariables</a> (const <a class="el" href="classnnvm_1_1Symbol.html">nnvm::Symbol</a> &sym) const</td></tr> |
| <tr class="memdesc:a8cd9d3861381bbb2e3328b661d8e0d38"><td class="mdescLeft"> </td><td class="mdescRight">Return the marked nonleaf nodes. <a href="classmxnet_1_1Imperative.html#a8cd9d3861381bbb2e3328b661d8e0d38">More...</a><br /></td></tr> |
| <tr class="separator:a8cd9d3861381bbb2e3328b661d8e0d38"><td class="memSeparator" colspan="2"> </td></tr> |
| </table><table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-static-methods"></a> |
| Static Public Member Functions</h2></td></tr> |
| <tr class="memitem:a55cc9d57cddd0aeae5771e1c7d72b00d"><td class="memItemLeft" align="right" valign="top">static <a class="el" href="classmxnet_1_1Imperative.html">Imperative</a> * </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a55cc9d57cddd0aeae5771e1c7d72b00d">Get</a> ()</td></tr> |
| <tr class="separator:a55cc9d57cddd0aeae5771e1c7d72b00d"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a0b5c5c6c9e7288156a7c633cb5433900"><td class="memItemLeft" align="right" valign="top">static bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a0b5c5c6c9e7288156a7c633cb5433900">PreferBulkExecInference</a> ()</td></tr> |
| <tr class="memdesc:a0b5c5c6c9e7288156a7c633cb5433900"><td class="mdescLeft"> </td><td class="mdescRight">Should op execution bulking be employed during inference. <a href="classmxnet_1_1Imperative.html#a0b5c5c6c9e7288156a7c633cb5433900">More...</a><br /></td></tr> |
| <tr class="separator:a0b5c5c6c9e7288156a7c633cb5433900"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a99b8ef35060523298e8e7e969d829920"><td class="memItemLeft" align="right" valign="top">static bool </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a99b8ef35060523298e8e7e969d829920">PreferBulkExecTrain</a> ()</td></tr> |
| <tr class="memdesc:a99b8ef35060523298e8e7e969d829920"><td class="mdescLeft"> </td><td class="mdescRight">Should op execution bulking be employed during training. <a href="classmxnet_1_1Imperative.html#a99b8ef35060523298e8e7e969d829920">More...</a><br /></td></tr> |
| <tr class="separator:a99b8ef35060523298e8e7e969d829920"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:af19dbcd37d7adf9d205b9c2efdb84b6b"><td class="memItemLeft" align="right" valign="top">static int </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#af19dbcd37d7adf9d205b9c2efdb84b6b">BulkExecMaxNodeTrainFwd</a> ()</td></tr> |
| <tr class="memdesc:af19dbcd37d7adf9d205b9c2efdb84b6b"><td class="mdescLeft"> </td><td class="mdescRight">The max number of op nodes in a bulk during forward pass of training. <a href="classmxnet_1_1Imperative.html#af19dbcd37d7adf9d205b9c2efdb84b6b">More...</a><br /></td></tr> |
| <tr class="separator:af19dbcd37d7adf9d205b9c2efdb84b6b"><td class="memSeparator" colspan="2"> </td></tr> |
| <tr class="memitem:a482d9ab2ab7d969c55f7c13797a662a0"><td class="memItemLeft" align="right" valign="top">static int </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#a482d9ab2ab7d969c55f7c13797a662a0">BulkExecMaxNodeTrainBwd</a> ()</td></tr> |
| <tr class="memdesc:a482d9ab2ab7d969c55f7c13797a662a0"><td class="mdescLeft"> </td><td class="mdescRight">The max number of op nodes in a bulk during backward pass of training. <a href="classmxnet_1_1Imperative.html#a482d9ab2ab7d969c55f7c13797a662a0">More...</a><br /></td></tr> |
| <tr class="separator:a482d9ab2ab7d969c55f7c13797a662a0"><td class="memSeparator" colspan="2"> </td></tr> |
| </table><table class="memberdecls"> |
| <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="friends"></a> |
| Friends</h2></td></tr> |
| <tr class="memitem:ad41557e205cf8f4d6ef70b60587c37cb"><td class="memItemLeft" align="right" valign="top">class </td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1Imperative.html#ad41557e205cf8f4d6ef70b60587c37cb">NDArray</a></td></tr> |
| <tr class="separator:ad41557e205cf8f4d6ef70b60587c37cb"><td class="memSeparator" colspan="2"> </td></tr> |
| </table> |
| <a name="details" id="details"></a><h2 class="groupheader">Detailed Description</h2> |
| <div class="textblock"><p>runtime functions for <a class="el" href="classmxnet_1_1NDArray.html" title="ndarray interface">NDArray</a> </p> |
| </div><h2 class="groupheader">Member Function Documentation</h2> |
| <a id="afa4950a3cf57431f1d0d639c3f7afa7d"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#afa4950a3cf57431f1d0d639c3f7afa7d">◆ </a></span>Backward()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">std::vector<<a class="el" href="classmxnet_1_1NDArray.html">NDArray</a>*> mxnet::Imperative::Backward </td> |
| <td>(</td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>outputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>ograds</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>variables</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>is_train</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>retain_graph</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>create_graph</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>compute the gradient of outputs w.r.t variables. </p> |
| |
| </div> |
| </div> |
| <a id="a482d9ab2ab7d969c55f7c13797a662a0"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a482d9ab2ab7d969c55f7c13797a662a0">◆ </a></span>BulkExecMaxNodeTrainBwd()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static int mxnet::Imperative::BulkExecMaxNodeTrainBwd </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>The max number of op nodes in a bulk during backward pass of training. </p> |
| |
| </div> |
| </div> |
| <a id="af19dbcd37d7adf9d205b9c2efdb84b6b"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af19dbcd37d7adf9d205b9c2efdb84b6b">◆ </a></span>BulkExecMaxNodeTrainFwd()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static int mxnet::Imperative::BulkExecMaxNodeTrainFwd </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>The max number of op nodes in a bulk during forward pass of training. </p> |
| |
| </div> |
| </div> |
| <a id="a10c6d32c54563a974c9265a2397c821f"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a10c6d32c54563a974c9265a2397c821f">◆ </a></span>DeferredComputeClear()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void mxnet::Imperative::DeferredComputeClear </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="classmxnet_1_1NDArrayHandle.html">NDArrayHandle</a> * </td> |
| <td class="paramname"><em>arrays</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const int </td> |
| <td class="paramname"><em>num</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>clear info node associated with array </p> |
| |
| </div> |
| </div> |
| <a id="aed3baaf0efb809226df9df765fefaaa0"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#aed3baaf0efb809226df9df765fefaaa0">◆ </a></span>DropGrads()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void mxnet::Imperative::DropGrads </td> |
| <td>(</td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>variables</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>unmark nonleaf variables to free the memory. </p> |
| |
| </div> |
| </div> |
| <a id="a55cc9d57cddd0aeae5771e1c7d72b00d"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a55cc9d57cddd0aeae5771e1c7d72b00d">◆ </a></span>Get()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static <a class="el" href="classmxnet_1_1Imperative.html">Imperative</a>* mxnet::Imperative::Get </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| <dl class="section return"><dt>Returns</dt><dd>AutogradRuntime singleton </dd></dl> |
| |
| </div> |
| </div> |
| <a id="a70d6fd0885c9b17d2cacf5122a74d6f6"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a70d6fd0885c9b17d2cacf5122a74d6f6">◆ </a></span>get_opt_constraints()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname"><a class="el" href="namespacemxnet.html#abdd24c24bd41ef5d0eff7628845e7a36">OptConstraint</a> mxnet::Imperative::get_opt_constraints </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>return current optimization constraints. </p> |
| |
| </div> |
| </div> |
| <a id="a0f901368c813c144b8ec6aacac7567db"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a0f901368c813c144b8ec6aacac7567db">◆ </a></span>GetDeferredComputeSymbol()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname"><a class="el" href="classnnvm_1_1Symbol.html">nnvm::Symbol</a> mxnet::Imperative::GetDeferredComputeSymbol </td> |
| <td>(</td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>outputs</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>obtain symbol representation of deferred compute session. </p> |
| |
| </div> |
| </div> |
| <a id="ae3a1279e546061bf6062817c93dce0dc"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ae3a1279e546061bf6062817c93dce0dc">◆ </a></span>Invoke()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname"><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> mxnet::Imperative::Invoke </td> |
| <td>(</td> |
| <td class="paramtype">const <a class="el" href="structmxnet_1_1Context.html">Context</a> & </td> |
| <td class="paramname"><em>default_ctx</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const <a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> & </td> |
| <td class="paramname"><em>attrs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>inputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>outputs</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="adb41ed890acd4cbe672f088b0284fe84"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#adb41ed890acd4cbe672f088b0284fe84">◆ </a></span>InvokeOp()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname"><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> mxnet::Imperative::InvokeOp </td> |
| <td>(</td> |
| <td class="paramtype">const <a class="el" href="structmxnet_1_1Context.html">Context</a> & </td> |
| <td class="paramname"><em>ctx</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const <a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> & </td> |
| <td class="paramname"><em>attrs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>inputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>outputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="namespacemxnet.html#a7cd7094ef222e9ae1f520c3b6a214398">OpReqType</a> > & </td> |
| <td class="paramname"><em>req</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const <a class="el" href="namespacemxnet.html#aad5fec2b4bceabbf4b884460cc2e0a2b">DispatchMode</a> </td> |
| <td class="paramname"><em>dispatch_mode</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype"><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> </td> |
| <td class="paramname"><em>state</em> = <code><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a>()</code> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <a id="a350f6f69b794358586436f58992a3e38"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a350f6f69b794358586436f58992a3e38">◆ </a></span>is_deferred_compute()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::is_deferred_compute </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>whether deferred compute mode is on. </p> |
| |
| </div> |
| </div> |
| <a id="a63ff2923f4b9daa8c35d1a683d17949f"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a63ff2923f4b9daa8c35d1a683d17949f">◆ </a></span>is_np_default_dtype()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::is_np_default_dtype </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>return current numpy default dtype compatibility status. </p> |
| |
| </div> |
| </div> |
| <a id="af65cc2bc2b45b328d99ba443211b8be7"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af65cc2bc2b45b328d99ba443211b8be7">◆ </a></span>is_np_shape()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">int mxnet::Imperative::is_np_shape </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>return current numpy compatibility status, GlobalOn(2), ThreadLocalOn(1), Off(0). </p> |
| |
| </div> |
| </div> |
| <a id="acbbd0eafd83cd96a40cce3d01147b1a3"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#acbbd0eafd83cd96a40cce3d01147b1a3">◆ </a></span>is_recording()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::is_recording </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>whether operator recording is on. </p> |
| |
| </div> |
| </div> |
| <a id="af6024ec51f5fcd73935994f7b43f62ef"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#af6024ec51f5fcd73935994f7b43f62ef">◆ </a></span>is_training()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::is_training </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>whether operator recording is on. </p> |
| |
| </div> |
| </div> |
| <a id="a8cd9d3861381bbb2e3328b661d8e0d38"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a8cd9d3861381bbb2e3328b661d8e0d38">◆ </a></span>ListNonleafVariables()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">std::vector<<a class="el" href="namespacennvm.html#a573897937ff3b88f0014feabb8e44a92">nnvm::ObjectPtr</a>> mxnet::Imperative::ListNonleafVariables </td> |
| <td>(</td> |
| <td class="paramtype">const <a class="el" href="classnnvm_1_1Symbol.html">nnvm::Symbol</a> & </td> |
| <td class="paramname"><em>sym</em></td><td>)</td> |
| <td> const</td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Return the marked nonleaf nodes. </p> |
| |
| </div> |
| </div> |
| <a id="a06fa7ec786cf0ea3a0eb7c8a174171cb"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a06fa7ec786cf0ea3a0eb7c8a174171cb">◆ </a></span>MarkVariables()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void mxnet::Imperative::MarkVariables </td> |
| <td>(</td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>variables</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< uint32_t > & </td> |
| <td class="paramname"><em>grad_reqs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>gradients</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>mark variables for computing gradients. </p> |
| |
| </div> |
| </div> |
| <a id="a0b5c5c6c9e7288156a7c633cb5433900"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a0b5c5c6c9e7288156a7c633cb5433900">◆ </a></span>PreferBulkExecInference()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static bool mxnet::Imperative::PreferBulkExecInference </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Should op execution bulking be employed during inference. </p> |
| |
| </div> |
| </div> |
| <a id="a99b8ef35060523298e8e7e969d829920"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a99b8ef35060523298e8e7e969d829920">◆ </a></span>PreferBulkExecTrain()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">static bool mxnet::Imperative::PreferBulkExecTrain </td> |
| <td>(</td> |
| <td class="paramname"></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span><span class="mlabel">static</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>Should op execution bulking be employed during training. </p> |
| |
| </div> |
| </div> |
| <a id="a2efa644fae290825385c50e2f8feb74d"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a2efa644fae290825385c50e2f8feb74d">◆ </a></span>RecordDeferredCompute()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void mxnet::Imperative::RecordDeferredCompute </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> && </td> |
| <td class="paramname"><em>attrs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>inputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>outputs</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>to record operator, return corresponding node. </p> |
| |
| </div> |
| </div> |
| <a id="a2b0f55c31882412c44468d8ba1a97fd1"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a2b0f55c31882412c44468d8ba1a97fd1">◆ </a></span>RecordOp()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void mxnet::Imperative::RecordOp </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="structnnvm_1_1NodeAttrs.html">nnvm::NodeAttrs</a> && </td> |
| <td class="paramname"><em>attrs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>inputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const std::vector< <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a> * > & </td> |
| <td class="paramname"><em>outputs</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const <a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a> & </td> |
| <td class="paramname"><em>state</em> = <code><a class="el" href="classmxnet_1_1OpStatePtr.html">OpStatePtr</a>()</code>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">std::vector< bool > * </td> |
| <td class="paramname"><em>p_save_inputs</em> = <code>nullptr</code>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">std::vector< bool > * </td> |
| <td class="paramname"><em>p_save_outputs</em> = <code>nullptr</code> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>to record operator, return corresponding node. </p> |
| |
| </div> |
| </div> |
| <a id="a20d520b1969ab70928e6e84898d632c4"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a20d520b1969ab70928e6e84898d632c4">◆ </a></span>set_is_deferred_compute()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::set_is_deferred_compute </td> |
| <td>(</td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>is_deferred_compute</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>turn on or turn off operator recording for autograd. </p> |
| |
| </div> |
| </div> |
| <a id="afb64b62624152e84ca54d9371268ab40"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#afb64b62624152e84ca54d9371268ab40">◆ </a></span>set_is_np_default_dtype()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::set_is_np_default_dtype </td> |
| <td>(</td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>is_np_default_dtype</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>specify numpy default dtype off or global on. </p> |
| |
| </div> |
| </div> |
| <a id="afc403b976f76aae9dcfc3790d9190832"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#afc403b976f76aae9dcfc3790d9190832">◆ </a></span>set_is_np_shape()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::set_is_np_shape </td> |
| <td>(</td> |
| <td class="paramtype">int </td> |
| <td class="paramname"><em>is_np_shape</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>specify numpy compatibility off, thread local on or global on. </p> |
| |
| </div> |
| </div> |
| <a id="abc9227372b3d4f4b5ec8ed33ba6e68ad"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#abc9227372b3d4f4b5ec8ed33ba6e68ad">◆ </a></span>set_is_recording()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::set_is_recording </td> |
| <td>(</td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>is_recording</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>turn on or turn off operator recording for autograd. </p> |
| |
| </div> |
| </div> |
| <a id="a2c1495dbdc4439fd534307fa981b77f0"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a2c1495dbdc4439fd534307fa981b77f0">◆ </a></span>set_is_training()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">bool mxnet::Imperative::set_is_training </td> |
| <td>(</td> |
| <td class="paramtype">bool </td> |
| <td class="paramname"><em>is_train</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>turn on or turn off operator recording for autograd. </p> |
| |
| </div> |
| </div> |
| <a id="a76a9e1537fff16c402a7cc6062d4d1b0"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a76a9e1537fff16c402a7cc6062d4d1b0">◆ </a></span>set_opt_constraints()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname"><a class="el" href="namespacemxnet.html#abdd24c24bd41ef5d0eff7628845e7a36">OptConstraint</a> mxnet::Imperative::set_opt_constraints </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="namespacemxnet.html#abdd24c24bd41ef5d0eff7628845e7a36">OptConstraint</a> </td> |
| <td class="paramname"><em>constraints</em></td><td>)</td> |
| <td></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">inline</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>set optimization constraints. </p> |
| |
| </div> |
| </div> |
| <a id="a16947c6577a274d7bd8dbc3b0aac0107"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#a16947c6577a274d7bd8dbc3b0aac0107">◆ </a></span>SetDeferredComputeVariable()</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">void mxnet::Imperative::SetDeferredComputeVariable </td> |
| <td>(</td> |
| <td class="paramtype"><a class="el" href="classmxnet_1_1NDArrayHandle.html">NDArrayHandle</a> * </td> |
| <td class="paramname"><em>arrays</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype"><a class="el" href="include_2mxnet_2c__api_8h.html#a301e9a7722b326126defdd39ccd99b37">SymbolHandle</a> * </td> |
| <td class="paramname"><em>variables</em>, </td> |
| </tr> |
| <tr> |
| <td class="paramkey"></td> |
| <td></td> |
| <td class="paramtype">const int </td> |
| <td class="paramname"><em>num</em> </td> |
| </tr> |
| <tr> |
| <td></td> |
| <td>)</td> |
| <td></td><td></td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| <p>associate arrays with variables for deferred compute </p> |
| |
| </div> |
| </div> |
| <h2 class="groupheader">Friends And Related Function Documentation</h2> |
| <a id="ad41557e205cf8f4d6ef70b60587c37cb"></a> |
| <h2 class="memtitle"><span class="permalink"><a href="#ad41557e205cf8f4d6ef70b60587c37cb">◆ </a></span>NDArray</h2> |
| |
| <div class="memitem"> |
| <div class="memproto"> |
| <table class="mlabels"> |
| <tr> |
| <td class="mlabels-left"> |
| <table class="memname"> |
| <tr> |
| <td class="memname">friend class <a class="el" href="classmxnet_1_1NDArray.html">NDArray</a></td> |
| </tr> |
| </table> |
| </td> |
| <td class="mlabels-right"> |
| <span class="mlabels"><span class="mlabel">friend</span></span> </td> |
| </tr> |
| </table> |
| </div><div class="memdoc"> |
| |
| </div> |
| </div> |
| <hr/>The documentation for this class was generated from the following file:<ul> |
| <li>/work/mxnet/include/mxnet/<a class="el" href="imperative_8h_source.html">imperative.h</a></li> |
| </ul> |
| </div><!-- contents --> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Jan 5 2023 03:47:41 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.17 |
| </small></address> |
| </body> |
| </html> |