Google Git
Sign in
apache / mxnet-site / d5c5a235eef72f64ccaf224b834a20ef3287469b / . / versions / master / api / cpp / docs / api / classmxnet_1_1op_1_1SimpleOpRegEntry.html
blob: 4c7bd1156d83d63c70722369ebf20bb96e11faab [file] [log] [blame]
<!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::op::SimpleOpRegEntry 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&amp;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&amp;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="namespacemxnet_1_1op.html">op</a></li><li class="navelem"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html">SimpleOpRegEntry</a></li> </ul>
</div>
</div><!-- top -->
<div class="header">
<div class="summary">
<a href="#pub-types">Public Types</a> &#124;
<a href="#pub-methods">Public Member Functions</a> &#124;
<a href="#pub-attribs">Public Attributes</a> &#124;
<a href="classmxnet_1_1op_1_1SimpleOpRegEntry-members.html">List of all members</a> </div>
<div class="headertitle">
<div class="title">mxnet::op::SimpleOpRegEntry Class Reference<span class="mlabels"><span class="mlabel">abstract</span></span></div> </div>
</div><!--header-->
<div class="contents">
<p>registry entry to register simple operators via functions.
<a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#details">More...</a></p>
<p><code>#include &lt;<a class="el" href="operator__util_8h_source.html">operator_util.h</a>&gt;</code></p>
<div class="dynheader">
Collaboration diagram for mxnet::op::SimpleOpRegEntry:</div>
<div class="dyncontent">
<div class="center"><img src="classmxnet_1_1op_1_1SimpleOpRegEntry__coll__graph.png" border="0" usemap="#mxnet_1_1op_1_1SimpleOpRegEntry_coll__map" alt="Collaboration graph"/></div>
<!-- MAP 0 -->
</div>
<table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-types"></a>
Public Types</h2></td></tr>
<tr class="memitem:ac2fd6ca9171b44a1ab131430cc5866a3"><td class="memItemLeft" align="right" valign="top">typedef <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html">SimpleOpRegEntry</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a></td></tr>
<tr class="memdesc:ac2fd6ca9171b44a1ab131430cc5866a3"><td class="mdescLeft">&#160;</td><td class="mdescRight">declare self type <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">More...</a><br /></td></tr>
<tr class="separator:ac2fd6ca9171b44a1ab131430cc5866a3"><td class="memSeparator" colspan="2">&#160;</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:ac258ea0f99d31b36e7dd682c25b3aeca"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac258ea0f99d31b36e7dd682c25b3aeca">set_symbol_op_name</a> (char const *symbol_name)=0</td></tr>
<tr class="memdesc:ac258ea0f99d31b36e7dd682c25b3aeca"><td class="mdescLeft">&#160;</td><td class="mdescRight">set a seperate name for symbol This must be called before set_function. Default: this is set to be same as the name of operator. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac258ea0f99d31b36e7dd682c25b3aeca">More...</a><br /></td></tr>
<tr class="separator:ac258ea0f99d31b36e7dd682c25b3aeca"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a0725f1b670044149a088355ffeb6977f"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a0725f1b670044149a088355ffeb6977f">set_enable_scalar</a> (bool enable_scalar, <a class="el" href="namespacemxnet_1_1op.html#ae7310844fa858f1ec6b6428dd6977a13">SimpleOpScalarOption</a> type_mask=<a class="el" href="namespacemxnet_1_1op.html#ae7310844fa858f1ec6b6428dd6977a13aa4cab4197f615e92b1fdb61c51e1b9c7">kArrayBeforeScalar</a>)=0</td></tr>
<tr class="memdesc:a0725f1b670044149a088355ffeb6977f"><td class="mdescLeft">&#160;</td><td class="mdescRight">set number of scalar arguments needed to be passed in env A function cannot have both kwargs and scalar arguments. Default: this is set to false <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a0725f1b670044149a088355ffeb6977f">More...</a><br /></td></tr>
<tr class="separator:a0725f1b670044149a088355ffeb6977f"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a4cdffb8337b41a643c05632b29e65a2b"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a4cdffb8337b41a643c05632b29e65a2b">set_enable_kwargs</a> (bool enable_kwargs)=0</td></tr>
<tr class="memdesc:a4cdffb8337b41a643c05632b29e65a2b"><td class="mdescLeft">&#160;</td><td class="mdescRight">set whether to enable kwargs A function cannot have both kwargs and scalar arguments. Default: this is set to false <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a4cdffb8337b41a643c05632b29e65a2b">More...</a><br /></td></tr>
<tr class="separator:a4cdffb8337b41a643c05632b29e65a2b"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:af67ac155e01e797e7f633789453fb5f3"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#af67ac155e01e797e7f633789453fb5f3">set_resource_request</a> (const std::vector&lt; <a class="el" href="structmxnet_1_1ResourceRequest.html">ResourceRequest</a> &gt; &amp;reqs)=0</td></tr>
<tr class="memdesc:af67ac155e01e797e7f633789453fb5f3"><td class="mdescLeft">&#160;</td><td class="mdescRight">set resource request By default there is no resource request. The resource will be presented in both forward and backward. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#af67ac155e01e797e7f633789453fb5f3">More...</a><br /></td></tr>
<tr class="separator:af67ac155e01e797e7f633789453fb5f3"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a7b615f5c6e5edc53a7ba008e631dcd52"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a7b615f5c6e5edc53a7ba008e631dcd52">set_resource_request</a> (<a class="el" href="structmxnet_1_1ResourceRequest.html">ResourceRequest</a> req)=0</td></tr>
<tr class="memdesc:a7b615f5c6e5edc53a7ba008e631dcd52"><td class="mdescLeft">&#160;</td><td class="mdescRight">set resource request By default there is no resource request. The resource will be presented in both forward and backward. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a7b615f5c6e5edc53a7ba008e631dcd52">More...</a><br /></td></tr>
<tr class="separator:a7b615f5c6e5edc53a7ba008e631dcd52"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a3be9c65fc5f747b6a05e3c2889d8f72b"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a3be9c65fc5f747b6a05e3c2889d8f72b">set_shape_function</a> (<a class="el" href="namespacemxnet_1_1op.html#aa4ae310a00e36e5bfa260fc65205395e">SourceShapeFunction</a> fshapeinfer)=0</td></tr>
<tr class="memdesc:a3be9c65fc5f747b6a05e3c2889d8f72b"><td class="mdescLeft">&#160;</td><td class="mdescRight">set source inference function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a3be9c65fc5f747b6a05e3c2889d8f72b">More...</a><br /></td></tr>
<tr class="separator:a3be9c65fc5f747b6a05e3c2889d8f72b"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a1cc9dca87282218902486f66c98acf8b"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a1cc9dca87282218902486f66c98acf8b">set_shape_function</a> (<a class="el" href="namespacemxnet_1_1op.html#aa97f14f8f0d4f6e57f792ad14f967fef">UnaryShapeFunction</a> fshapeinfer)=0</td></tr>
<tr class="memdesc:a1cc9dca87282218902486f66c98acf8b"><td class="mdescLeft">&#160;</td><td class="mdescRight">set shape inference function. Default: out_shape = in_shape <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a1cc9dca87282218902486f66c98acf8b">More...</a><br /></td></tr>
<tr class="separator:a1cc9dca87282218902486f66c98acf8b"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a8a67eaf7bab4a18d5b46f3a1639f426f"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a8a67eaf7bab4a18d5b46f3a1639f426f">set_shape_function</a> (<a class="el" href="namespacemxnet_1_1op.html#a8973cbc1527a6c57c6e19c9f1ed82075">BinaryShapeFunction</a> fshapeinfer)=0</td></tr>
<tr class="memdesc:a8a67eaf7bab4a18d5b46f3a1639f426f"><td class="mdescLeft">&#160;</td><td class="mdescRight">set shape inference function to be the binary inference function Default: out_shape = lhs_shape, and lhs_shape must equal rhs_shape. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a8a67eaf7bab4a18d5b46f3a1639f426f">More...</a><br /></td></tr>
<tr class="separator:a8a67eaf7bab4a18d5b46f3a1639f426f"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a1f9bcff5039f065f73c7ec3a2558d799"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a1f9bcff5039f065f73c7ec3a2558d799">set_function</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#a65e8bade5ae0539689b705b921ed9a9d">SourceFunction</a> fsource, <a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268">SimpleOpRegOption</a> register_symbolic=<a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268a534486ab7c3f672542bfa09b7b6982d1">kRegisterSymbolic</a>)=0</td></tr>
<tr class="memdesc:a1f9bcff5039f065f73c7ec3a2558d799"><td class="mdescLeft">&#160;</td><td class="mdescRight">set function of the function to be fsource <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a1f9bcff5039f065f73c7ec3a2558d799">More...</a><br /></td></tr>
<tr class="separator:a1f9bcff5039f065f73c7ec3a2558d799"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a2960b7ef7ff755d06336b9d87a4a0ced"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a2960b7ef7ff755d06336b9d87a4a0ced">set_function</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#aa37d5216112b72e8110fc8654860910b">UnaryFunction</a> funary, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_in_out, <a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268">SimpleOpRegOption</a> register_symbolic=<a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268a534486ab7c3f672542bfa09b7b6982d1">kRegisterSymbolic</a>)=0</td></tr>
<tr class="memdesc:a2960b7ef7ff755d06336b9d87a4a0ced"><td class="mdescLeft">&#160;</td><td class="mdescRight">set function of the function to be funary <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a2960b7ef7ff755d06336b9d87a4a0ced">More...</a><br /></td></tr>
<tr class="separator:a2960b7ef7ff755d06336b9d87a4a0ced"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a52c14c9d53298c63c9776fc2b7095149"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a52c14c9d53298c63c9776fc2b7095149">set_function</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#a880eff3ebc68810ef133a6d8b77facc9">BinaryFunction</a> fbinary, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_lhs_out, <a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268">SimpleOpRegOption</a> register_symbolic=<a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268a534486ab7c3f672542bfa09b7b6982d1">kRegisterSymbolic</a>)=0</td></tr>
<tr class="memdesc:a52c14c9d53298c63c9776fc2b7095149"><td class="mdescLeft">&#160;</td><td class="mdescRight">set function of the function to be funary <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a52c14c9d53298c63c9776fc2b7095149">More...</a><br /></td></tr>
<tr class="separator:a52c14c9d53298c63c9776fc2b7095149"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ab8fe122e15142dc85c679071acf53302"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ab8fe122e15142dc85c679071acf53302">set_gradient</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#a1e5767bffea568a1c38379060cbb68b4">UnaryGradFunctionT0</a> fgrad, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_out_in_grad)=0</td></tr>
<tr class="memdesc:ab8fe122e15142dc85c679071acf53302"><td class="mdescLeft">&#160;</td><td class="mdescRight">set gradient of the function of this function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ab8fe122e15142dc85c679071acf53302">More...</a><br /></td></tr>
<tr class="separator:ab8fe122e15142dc85c679071acf53302"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:af586bed58bb245dab4c4f1ffd6aeede4"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#af586bed58bb245dab4c4f1ffd6aeede4">set_gradient</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#a6ad11641816278f8fdb6fe87c7fd3740">UnaryGradFunctionT1</a> fgrad, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_out_in_grad)=0</td></tr>
<tr class="memdesc:af586bed58bb245dab4c4f1ffd6aeede4"><td class="mdescLeft">&#160;</td><td class="mdescRight">set gradient of the function of this function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#af586bed58bb245dab4c4f1ffd6aeede4">More...</a><br /></td></tr>
<tr class="separator:af586bed58bb245dab4c4f1ffd6aeede4"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ac6e523d1552de21667e4b5bb4d783cc1"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac6e523d1552de21667e4b5bb4d783cc1">set_gradient</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#ac29102128f7e7785493bbb5b7db80997">UnaryGradFunctionT2</a> fgrad, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_out_in_grad)=0</td></tr>
<tr class="memdesc:ac6e523d1552de21667e4b5bb4d783cc1"><td class="mdescLeft">&#160;</td><td class="mdescRight">set gradient of the function of this function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac6e523d1552de21667e4b5bb4d783cc1">More...</a><br /></td></tr>
<tr class="separator:ac6e523d1552de21667e4b5bb4d783cc1"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a15d795e1aa76b5ed12ca5a9b7b091cd2"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a15d795e1aa76b5ed12ca5a9b7b091cd2">set_gradient</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#abcef841f759a7874ef50f21ae0226c8a">BinaryGradFunctionT0</a> fgrad, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_out_lhs_grad)=0</td></tr>
<tr class="memdesc:a15d795e1aa76b5ed12ca5a9b7b091cd2"><td class="mdescLeft">&#160;</td><td class="mdescRight">set gradient of the function of this function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a15d795e1aa76b5ed12ca5a9b7b091cd2">More...</a><br /></td></tr>
<tr class="separator:a15d795e1aa76b5ed12ca5a9b7b091cd2"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a4df35e5bab55c3b2e742455d598e2e90"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a4df35e5bab55c3b2e742455d598e2e90">set_gradient</a> (int dev_mask, <a class="el" href="namespacemxnet_1_1op.html#a027d1f190d964f7494816733aff6c3f2">BinaryGradFunctionT1</a> fgrad, <a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a> inplace_out_lhs_grad)=0</td></tr>
<tr class="memdesc:a4df35e5bab55c3b2e742455d598e2e90"><td class="mdescLeft">&#160;</td><td class="mdescRight">set gradient of the function of this function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a4df35e5bab55c3b2e742455d598e2e90">More...</a><br /></td></tr>
<tr class="separator:a4df35e5bab55c3b2e742455d598e2e90"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a1cce9a6aaa175711b6be5b864b055f6a"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a1cce9a6aaa175711b6be5b864b055f6a">describe</a> (const std::string &amp;description)=0</td></tr>
<tr class="memdesc:a1cce9a6aaa175711b6be5b864b055f6a"><td class="mdescLeft">&#160;</td><td class="mdescRight">Describe the function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a1cce9a6aaa175711b6be5b864b055f6a">More...</a><br /></td></tr>
<tr class="separator:a1cce9a6aaa175711b6be5b864b055f6a"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a405bfb5929d459d45d93c2eaa645bfb5"><td class="memItemLeft" align="right" valign="top">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a> &amp;&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a405bfb5929d459d45d93c2eaa645bfb5">add_arguments</a> (const std::vector&lt; dmlc::ParamFieldInfo &gt; &amp;args)=0</td></tr>
<tr class="memdesc:a405bfb5929d459d45d93c2eaa645bfb5"><td class="mdescLeft">&#160;</td><td class="mdescRight">Describe the function. <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a405bfb5929d459d45d93c2eaa645bfb5">More...</a><br /></td></tr>
<tr class="separator:a405bfb5929d459d45d93c2eaa645bfb5"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a94815080350681ad70e7815da1bc0eae"><td class="memItemLeft" align="right" valign="top">virtual&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a94815080350681ad70e7815da1bc0eae">~SimpleOpRegEntry</a> ()</td></tr>
<tr class="memdesc:a94815080350681ad70e7815da1bc0eae"><td class="mdescLeft">&#160;</td><td class="mdescRight">virtual destructor <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a94815080350681ad70e7815da1bc0eae">More...</a><br /></td></tr>
<tr class="separator:a94815080350681ad70e7815da1bc0eae"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table><table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="pub-attribs"></a>
Public Attributes</h2></td></tr>
<tr class="memitem:a02c180f08b215e9553d3c93fa783f9a4"><td class="memItemLeft" align="right" valign="top">std::string&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a02c180f08b215e9553d3c93fa783f9a4">name</a></td></tr>
<tr class="memdesc:a02c180f08b215e9553d3c93fa783f9a4"><td class="mdescLeft">&#160;</td><td class="mdescRight">name of the operator <a href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#a02c180f08b215e9553d3c93fa783f9a4">More...</a><br /></td></tr>
<tr class="separator:a02c180f08b215e9553d3c93fa783f9a4"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table>
<a name="details" id="details"></a><h2 class="groupheader">Detailed Description</h2>
<div class="textblock"><p>registry entry to register simple operators via functions. </p>
</div><h2 class="groupheader">Member Typedef Documentation</h2>
<a id="ac2fd6ca9171b44a1ab131430cc5866a3"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ac2fd6ca9171b44a1ab131430cc5866a3">&#9670;&nbsp;</a></span>TSelf</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">typedef <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html">SimpleOpRegEntry</a> <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">mxnet::op::SimpleOpRegEntry::TSelf</a></td>
</tr>
</table>
</div><div class="memdoc">
<p>declare self type </p>
</div>
</div>
<h2 class="groupheader">Constructor &amp; Destructor Documentation</h2>
<a id="a94815080350681ad70e7815da1bc0eae"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a94815080350681ad70e7815da1bc0eae">&#9670;&nbsp;</a></span>~SimpleOpRegEntry()</h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual mxnet::op::SimpleOpRegEntry::~SimpleOpRegEntry </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">virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>virtual destructor </p>
</div>
</div>
<h2 class="groupheader">Member Function Documentation</h2>
<a id="a405bfb5929d459d45d93c2eaa645bfb5"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a405bfb5929d459d45d93c2eaa645bfb5">&#9670;&nbsp;</a></span>add_arguments()</h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::add_arguments </td>
<td>(</td>
<td class="paramtype">const std::vector&lt; dmlc::ParamFieldInfo &gt; &amp;&#160;</td>
<td class="paramname"><em>args</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>Describe the function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">args</td><td>argument information. Add additional arguments to the function. </td></tr>
</table>
</dd>
</dl>
<dl class="section return"><dt>Returns</dt><dd>reference to self. </dd></dl>
</div>
</div>
<a id="a1cce9a6aaa175711b6be5b864b055f6a"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a1cce9a6aaa175711b6be5b864b055f6a">&#9670;&nbsp;</a></span>describe()</h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::describe </td>
<td>(</td>
<td class="paramtype">const std::string &amp;&#160;</td>
<td class="paramname"><em>description</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>Describe the function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">description</td><td>The description of the function. </td></tr>
</table>
</dd>
</dl>
<dl class="section return"><dt>Returns</dt><dd>reference to self. </dd></dl>
</div>
</div>
<a id="a4cdffb8337b41a643c05632b29e65a2b"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a4cdffb8337b41a643c05632b29e65a2b">&#9670;&nbsp;</a></span>set_enable_kwargs()</h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_enable_kwargs </td>
<td>(</td>
<td class="paramtype">bool&#160;</td>
<td class="paramname"><em>enable_kwargs</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set whether to enable kwargs A function cannot have both kwargs and scalar arguments. Default: this is set to false </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">enable_kwargs</td><td>whether to enable kwargs </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a0725f1b670044149a088355ffeb6977f"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a0725f1b670044149a088355ffeb6977f">&#9670;&nbsp;</a></span>set_enable_scalar()</h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_enable_scalar </td>
<td>(</td>
<td class="paramtype">bool&#160;</td>
<td class="paramname"><em>enable_scalar</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#ae7310844fa858f1ec6b6428dd6977a13">SimpleOpScalarOption</a>&#160;</td>
<td class="paramname"><em>type_mask</em> = <code><a class="el" href="namespacemxnet_1_1op.html#ae7310844fa858f1ec6b6428dd6977a13aa4cab4197f615e92b1fdb61c51e1b9c7">kArrayBeforeScalar</a></code>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set number of scalar arguments needed to be passed in env A function cannot have both kwargs and scalar arguments. Default: this is set to false </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">enable_scalar</td><td>whether to enable scalar argument </td></tr>
<tr><td class="paramname">type_mask</td><td>the position of the scalar argument. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a52c14c9d53298c63c9776fc2b7095149"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a52c14c9d53298c63c9776fc2b7095149">&#9670;&nbsp;</a></span>set_function() <span class="overload">[1/3]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_function </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a880eff3ebc68810ef133a6d8b77facc9">BinaryFunction</a>&#160;</td>
<td class="paramname"><em>fbinary</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_lhs_out</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268">SimpleOpRegOption</a>&#160;</td>
<td class="paramname"><em>register_symbolic</em> = <code><a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268a534486ab7c3f672542bfa09b7b6982d1">kRegisterSymbolic</a></code>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set function of the function to be funary </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fbinary</td><td>The binary function that peforms the operation. </td></tr>
<tr><td class="paramname">inplace_lhs_out</td><td>Whether do inplace optimization on lhs and out. </td></tr>
<tr><td class="paramname">register_symbolic</td><td>Whether register a symbolic operator as well. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a1f9bcff5039f065f73c7ec3a2558d799"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a1f9bcff5039f065f73c7ec3a2558d799">&#9670;&nbsp;</a></span>set_function() <span class="overload">[2/3]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_function </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a65e8bade5ae0539689b705b921ed9a9d">SourceFunction</a>&#160;</td>
<td class="paramname"><em>fsource</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268">SimpleOpRegOption</a>&#160;</td>
<td class="paramname"><em>register_symbolic</em> = <code><a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268a534486ab7c3f672542bfa09b7b6982d1">kRegisterSymbolic</a></code>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set function of the function to be fsource </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fsource</td><td>The unary function that peforms the operation. </td></tr>
<tr><td class="paramname">register_symbolic</td><td>Whether register a symbolic operator as well. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a2960b7ef7ff755d06336b9d87a4a0ced"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a2960b7ef7ff755d06336b9d87a4a0ced">&#9670;&nbsp;</a></span>set_function() <span class="overload">[3/3]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_function </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#aa37d5216112b72e8110fc8654860910b">UnaryFunction</a>&#160;</td>
<td class="paramname"><em>funary</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_in_out</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268">SimpleOpRegOption</a>&#160;</td>
<td class="paramname"><em>register_symbolic</em> = <code><a class="el" href="namespacemxnet_1_1op.html#a01bb0bb658bedb4d58bf7d11e4b17268a534486ab7c3f672542bfa09b7b6982d1">kRegisterSymbolic</a></code>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set function of the function to be funary </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">funary</td><td>The unary function that peforms the operation. </td></tr>
<tr><td class="paramname">inplace_in_out</td><td>Whether do inplace optimization on in and out. </td></tr>
<tr><td class="paramname">register_symbolic</td><td>Whether register a symbolic operator as well. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a15d795e1aa76b5ed12ca5a9b7b091cd2"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a15d795e1aa76b5ed12ca5a9b7b091cd2">&#9670;&nbsp;</a></span>set_gradient() <span class="overload">[1/5]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_gradient </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#abcef841f759a7874ef50f21ae0226c8a">BinaryGradFunctionT0</a>&#160;</td>
<td class="paramname"><em>fgrad</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_out_lhs_grad</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set gradient of the function of this function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fgrad</td><td>The gradient function to be set. </td></tr>
<tr><td class="paramname">inplace_out_lhs_grad</td><td>whether out_grad and lhs_grad can share memory. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a4df35e5bab55c3b2e742455d598e2e90"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a4df35e5bab55c3b2e742455d598e2e90">&#9670;&nbsp;</a></span>set_gradient() <span class="overload">[2/5]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_gradient </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a027d1f190d964f7494816733aff6c3f2">BinaryGradFunctionT1</a>&#160;</td>
<td class="paramname"><em>fgrad</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_out_lhs_grad</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set gradient of the function of this function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fgrad</td><td>The gradient function to be set. </td></tr>
<tr><td class="paramname">inplace_out_lhs_grad</td><td>whether out_grad and lhs_grad can share memory. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="ab8fe122e15142dc85c679071acf53302"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ab8fe122e15142dc85c679071acf53302">&#9670;&nbsp;</a></span>set_gradient() <span class="overload">[3/5]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_gradient </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a1e5767bffea568a1c38379060cbb68b4">UnaryGradFunctionT0</a>&#160;</td>
<td class="paramname"><em>fgrad</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_out_in_grad</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set gradient of the function of this function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fgrad</td><td>The gradient function to be set. </td></tr>
<tr><td class="paramname">inplace_out_in_grad</td><td>whether out_grad and in_grad can share memory. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="af586bed58bb245dab4c4f1ffd6aeede4"></a>
<h2 class="memtitle"><span class="permalink"><a href="#af586bed58bb245dab4c4f1ffd6aeede4">&#9670;&nbsp;</a></span>set_gradient() <span class="overload">[4/5]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_gradient </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a6ad11641816278f8fdb6fe87c7fd3740">UnaryGradFunctionT1</a>&#160;</td>
<td class="paramname"><em>fgrad</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_out_in_grad</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set gradient of the function of this function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fgrad</td><td>The gradient function to be set. </td></tr>
<tr><td class="paramname">inplace_out_in_grad</td><td>whether out_grad and in_grad can share memory. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="ac6e523d1552de21667e4b5bb4d783cc1"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ac6e523d1552de21667e4b5bb4d783cc1">&#9670;&nbsp;</a></span>set_gradient() <span class="overload">[5/5]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_gradient </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>dev_mask</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#ac29102128f7e7785493bbb5b7db80997">UnaryGradFunctionT2</a>&#160;</td>
<td class="paramname"><em>fgrad</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a3c3b8c77865c7c715cfe05ee2b2fa8da">SimpleOpInplaceOption</a>&#160;</td>
<td class="paramname"><em>inplace_out_in_grad</em>&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set gradient of the function of this function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">dev_mask</td><td>The device mask of the function can act on. </td></tr>
<tr><td class="paramname">fgrad</td><td>The gradient function to be set. </td></tr>
<tr><td class="paramname">inplace_out_in_grad</td><td>whether out_grad and in_grad can share memory. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="af67ac155e01e797e7f633789453fb5f3"></a>
<h2 class="memtitle"><span class="permalink"><a href="#af67ac155e01e797e7f633789453fb5f3">&#9670;&nbsp;</a></span>set_resource_request() <span class="overload">[1/2]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_resource_request </td>
<td>(</td>
<td class="paramtype">const std::vector&lt; <a class="el" href="structmxnet_1_1ResourceRequest.html">ResourceRequest</a> &gt; &amp;&#160;</td>
<td class="paramname"><em>reqs</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set resource request By default there is no resource request. The resource will be presented in both forward and backward. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">reqs</td><td>the request. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a7b615f5c6e5edc53a7ba008e631dcd52"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a7b615f5c6e5edc53a7ba008e631dcd52">&#9670;&nbsp;</a></span>set_resource_request() <span class="overload">[2/2]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_resource_request </td>
<td>(</td>
<td class="paramtype"><a class="el" href="structmxnet_1_1ResourceRequest.html">ResourceRequest</a>&#160;</td>
<td class="paramname"><em>req</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set resource request By default there is no resource request. The resource will be presented in both forward and backward. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">req</td><td>the request. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a8a67eaf7bab4a18d5b46f3a1639f426f"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a8a67eaf7bab4a18d5b46f3a1639f426f">&#9670;&nbsp;</a></span>set_shape_function() <span class="overload">[1/3]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_shape_function </td>
<td>(</td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#a8973cbc1527a6c57c6e19c9f1ed82075">BinaryShapeFunction</a>&#160;</td>
<td class="paramname"><em>fshapeinfer</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set shape inference function to be the binary inference function Default: out_shape = lhs_shape, and lhs_shape must equal rhs_shape. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">fshapeinfer</td><td>The binary function that peforms the operation. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a3be9c65fc5f747b6a05e3c2889d8f72b"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a3be9c65fc5f747b6a05e3c2889d8f72b">&#9670;&nbsp;</a></span>set_shape_function() <span class="overload">[2/3]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_shape_function </td>
<td>(</td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#aa4ae310a00e36e5bfa260fc65205395e">SourceShapeFunction</a>&#160;</td>
<td class="paramname"><em>fshapeinfer</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set source inference function. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">fshapeinfer</td><td>The source function that peforms the operation. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="a1cc9dca87282218902486f66c98acf8b"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a1cc9dca87282218902486f66c98acf8b">&#9670;&nbsp;</a></span>set_shape_function() <span class="overload">[3/3]</span></h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_shape_function </td>
<td>(</td>
<td class="paramtype"><a class="el" href="namespacemxnet_1_1op.html#aa97f14f8f0d4f6e57f792ad14f967fef">UnaryShapeFunction</a>&#160;</td>
<td class="paramname"><em>fshapeinfer</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set shape inference function. Default: out_shape = in_shape </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">fshapeinfer</td><td>The unary function that peforms the operation. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<a id="ac258ea0f99d31b36e7dd682c25b3aeca"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ac258ea0f99d31b36e7dd682c25b3aeca">&#9670;&nbsp;</a></span>set_symbol_op_name()</h2>
<div class="memitem">
<div class="memproto">
<table class="mlabels">
<tr>
<td class="mlabels-left">
<table class="memname">
<tr>
<td class="memname">virtual <a class="el" href="classmxnet_1_1op_1_1SimpleOpRegEntry.html#ac2fd6ca9171b44a1ab131430cc5866a3">TSelf</a>&amp; mxnet::op::SimpleOpRegEntry::set_symbol_op_name </td>
<td>(</td>
<td class="paramtype">char const *&#160;</td>
<td class="paramname"><em>symbol_name</em></td><td>)</td>
<td></td>
</tr>
</table>
</td>
<td class="mlabels-right">
<span class="mlabels"><span class="mlabel">pure virtual</span></span> </td>
</tr>
</table>
</div><div class="memdoc">
<p>set a seperate name for symbol This must be called before set_function. Default: this is set to be same as the name of operator. </p>
<dl class="params"><dt>Parameters</dt><dd>
<table class="params">
<tr><td class="paramname">symbol_name</td><td>the name of symbolic operator. </td></tr>
</table>
</dd>
</dl>
</div>
</div>
<h2 class="groupheader">Member Data Documentation</h2>
<a id="a02c180f08b215e9553d3c93fa783f9a4"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a02c180f08b215e9553d3c93fa783f9a4">&#9670;&nbsp;</a></span>name</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">std::string mxnet::op::SimpleOpRegEntry::name</td>
</tr>
</table>
</div><div class="memdoc">
<p>name of the operator </p>
</div>
</div>
<hr/>The documentation for this class was generated from the following file:<ul>
<li>/work/mxnet/include/mxnet/<a class="el" href="operator__util_8h_source.html">operator_util.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 &#160;<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/>
</a> 1.8.17
</small></address>
</body>
</html>