| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| <meta name="generator" content="Doxygen 1.8.17"/> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <title>mxnet: /work/mxnet/3rdparty/mshadow/mshadow/base.h Source File</title> |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="jquery.js"></script> |
| <script type="text/javascript" src="dynsections.js"></script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="search/searchdata.js"></script> |
| <script type="text/javascript" src="search/search.js"></script> |
| <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| <div id="titlearea"> |
| <table cellspacing="0" cellpadding="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| <td id="projectalign" style="padding-left: 0.5em;"> |
| <div id="projectname">mxnet |
| </div> |
| </td> |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| <!-- end header part --> |
| <!-- Generated by Doxygen 1.8.17 --> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| /* @license-end */ |
| </script> |
| <script type="text/javascript" src="menudata.js"></script> |
| <script type="text/javascript" src="menu.js"></script> |
| <script type="text/javascript"> |
| /* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&dn=gpl-2.0.txt GPL-v2 */ |
| $(function() { |
| initMenu('',true,false,'search.php','Search'); |
| $(document).ready(function() { init_search(); }); |
| }); |
| /* @license-end */</script> |
| <div id="main-nav"></div> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| </div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="dir_8cab8f464681f7cc51cee77e79a434cd.html">3rdparty</a></li><li class="navelem"><a class="el" href="dir_3e48ced36faa4eaa1b41f6d960bf0edb.html">mshadow</a></li><li class="navelem"><a class="el" href="dir_00b035bb2ad81894e6ad291054ea5f82.html">mshadow</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">base.h</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="3rdparty_2mshadow_2mshadow_2base_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">/*</span></div> |
| <div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> * Licensed to the Apache Software Foundation (ASF) under one</span></div> |
| <div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> * or more contributor license agreements. See the NOTICE file</span></div> |
| <div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> * distributed with this work for additional information</span></div> |
| <div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> * regarding copyright ownership. The ASF licenses this file</span></div> |
| <div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * to you under the Apache License, Version 2.0 (the</span></div> |
| <div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> * "License"); you may not use this file except in compliance</span></div> |
| <div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> * with the License. You may obtain a copy of the License at</span></div> |
| <div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span></div> |
| <div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="comment"> *</span></div> |
| <div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="comment"> * Unless required by applicable law or agreed to in writing,</span></div> |
| <div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="comment"> * software distributed under the License is distributed on an</span></div> |
| <div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="comment"> * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY</span></div> |
| <div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="comment"> * KIND, either express or implied. See the License for the</span></div> |
| <div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="comment"> * specific language governing permissions and limitations</span></div> |
| <div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * under the License.</span></div> |
| <div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> */</span></div> |
| <div class="line"><a name="l00019"></a><span class="lineno"> 19</span>  </div> |
| <div class="line"><a name="l00026"></a><span class="lineno"> 26</span> <span class="preprocessor">#ifndef MSHADOW_BASE_H_</span></div> |
| <div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#define MSHADOW_BASE_H_</span></div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span> <span class="preprocessor">#ifdef _MSC_VER</span></div> |
| <div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="preprocessor">#ifndef _CRT_SECURE_NO_WARNINGS</span></div> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#define _CRT_SECURE_NO_WARNINGS</span></div> |
| <div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#ifndef _CRT_SECURE_NO_DEPRECATE</span></div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#define _CRT_SECURE_NO_DEPRECATE</span></div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#ifndef NOMINMAX</span></div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#define NOMINMAX</span></div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="preprocessor">#include <algorithm></span></div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span> <span class="preprocessor">#include <cfloat></span></div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="preprocessor">#include <climits></span></div> |
| <div class="line"><a name="l00042"></a><span class="lineno"> 42</span> <span class="preprocessor">#include <cmath></span></div> |
| <div class="line"><a name="l00043"></a><span class="lineno"> 43</span> <span class="preprocessor">#include <cstdio></span></div> |
| <div class="line"><a name="l00044"></a><span class="lineno"> 44</span> <span class="preprocessor">#include <functional></span></div> |
| <div class="line"><a name="l00045"></a><span class="lineno"> 45</span> <span class="preprocessor">#include <limits></span></div> |
| <div class="line"><a name="l00046"></a><span class="lineno"> 46</span> <span class="preprocessor">#include <sstream></span></div> |
| <div class="line"><a name="l00047"></a><span class="lineno"> 47</span> <span class="preprocessor">#include <string></span></div> |
| <div class="line"><a name="l00048"></a><span class="lineno"> 48</span>  </div> |
| <div class="line"><a name="l00049"></a><span class="lineno"> 49</span> <span class="preprocessor">#ifdef _MSC_VER</span></div> |
| <div class="line"><a name="l00050"></a><span class="lineno"> 50</span> <span class="keyword">typedef</span> <span class="keywordtype">signed</span> <span class="keywordtype">char</span> int8_t;</div> |
| <div class="line"><a name="l00052"></a><span class="lineno"> 52</span> <span class="keyword">typedef</span> __int16 int16_t;</div> |
| <div class="line"><a name="l00053"></a><span class="lineno"> 53</span> <span class="keyword">typedef</span> __int32 int32_t;</div> |
| <div class="line"><a name="l00054"></a><span class="lineno"> 54</span> <span class="keyword">typedef</span> __int64 int64_t;</div> |
| <div class="line"><a name="l00055"></a><span class="lineno"> 55</span> <span class="keyword">typedef</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> uint8_t;</div> |
| <div class="line"><a name="l00056"></a><span class="lineno"> 56</span> <span class="keyword">typedef</span> <span class="keywordtype">unsigned</span> __int16 uint16_t;</div> |
| <div class="line"><a name="l00057"></a><span class="lineno"> 57</span> <span class="keyword">typedef</span> <span class="keywordtype">unsigned</span> __int32 uint32_t;</div> |
| <div class="line"><a name="l00058"></a><span class="lineno"> 58</span> <span class="keyword">typedef</span> <span class="keywordtype">unsigned</span> __int64 uint64_t;</div> |
| <div class="line"><a name="l00060"></a><span class="lineno"> 60</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00061"></a><span class="lineno"> 61</span> <span class="preprocessor">#include <inttypes.h></span></div> |
| <div class="line"><a name="l00062"></a><span class="lineno"> 62</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00063"></a><span class="lineno"> 63</span> <span class="comment">// macro defintiions</span></div> |
| <div class="line"><a name="l00068"></a><span class="lineno"> 68</span> <span class="comment"></span><span class="preprocessor">#ifndef MSHADOW_STAND_ALONE</span></div> |
| <div class="line"><a name="l00069"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#adb97209063773b312b79a275688a2093"> 69</a></span> <span class="preprocessor">#define MSHADOW_STAND_ALONE 0</span></div> |
| <div class="line"><a name="l00070"></a><span class="lineno"> 70</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00071"></a><span class="lineno"> 71</span>  </div> |
| <div class="line"><a name="l00072"></a><span class="lineno"> 72</span> <span class="preprocessor">#ifndef MSHADOW_ALLOC_PAD</span></div> |
| <div class="line"><a name="l00073"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a72d4e933c0003ccdc01b5d5f054dcf39"> 73</a></span> <span class="preprocessor">#define MSHADOW_ALLOC_PAD true</span></div> |
| <div class="line"><a name="l00074"></a><span class="lineno"> 74</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  </div> |
| <div class="line"><a name="l00083"></a><span class="lineno"> 83</span> <span class="preprocessor">#ifndef MSHADOW_MIN_PAD_RATIO</span></div> |
| <div class="line"><a name="l00084"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#aaddfd904e59a1fc9fb07019cff73adee"> 84</a></span> <span class="preprocessor"> #define MSHADOW_MIN_PAD_RATIO 2</span></div> |
| <div class="line"><a name="l00085"></a><span class="lineno"> 85</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  </div> |
| <div class="line"><a name="l00087"></a><span class="lineno"> 87</span> <span class="preprocessor">#if MSHADOW_STAND_ALONE</span></div> |
| <div class="line"><a name="l00088"></a><span class="lineno"> 88</span> <span class="preprocessor"> #define MSHADOW_USE_CBLAS 0</span></div> |
| <div class="line"><a name="l00089"></a><span class="lineno"> 89</span> <span class="preprocessor"> #define MSHADOW_USE_MKL 0</span></div> |
| <div class="line"><a name="l00090"></a><span class="lineno"> 90</span> <span class="preprocessor"> #define MSHADOW_USE_CUDA 0</span></div> |
| <div class="line"><a name="l00091"></a><span class="lineno"> 91</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00092"></a><span class="lineno"> 92</span>  </div> |
| <div class="line"><a name="l00097"></a><span class="lineno"> 97</span> <span class="preprocessor">#ifndef MSHADOW_FORCE_STREAM</span></div> |
| <div class="line"><a name="l00098"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a19a013475c0af9faff9348c649d80e03"> 98</a></span> <span class="preprocessor">#define MSHADOW_FORCE_STREAM 1</span></div> |
| <div class="line"><a name="l00099"></a><span class="lineno"> 99</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  </div> |
| <div class="line"><a name="l00102"></a><span class="lineno"> 102</span> <span class="preprocessor">#ifndef MSHADOW_USE_CBLAS</span></div> |
| <div class="line"><a name="l00103"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a6c5fbf2591134429036ca9af316e4600"> 103</a></span> <span class="preprocessor"> #define MSHADOW_USE_CBLAS 0</span></div> |
| <div class="line"><a name="l00104"></a><span class="lineno"> 104</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  </div> |
| <div class="line"><a name="l00106"></a><span class="lineno"> 106</span> <span class="preprocessor">#ifndef MSHADOW_USE_MKL</span></div> |
| <div class="line"><a name="l00107"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5fb3349b887a553f578daa6e4b4f1e56"> 107</a></span> <span class="preprocessor"> #define MSHADOW_USE_MKL 1</span></div> |
| <div class="line"><a name="l00108"></a><span class="lineno"> 108</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  </div> |
| <div class="line"><a name="l00114"></a><span class="lineno"> 114</span> <span class="preprocessor">#ifndef MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00115"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a52e5c06e325bfed605a4f63a48fe0020"> 115</a></span> <span class="preprocessor"> #define MSHADOW_USE_CUDA 1</span></div> |
| <div class="line"><a name="l00116"></a><span class="lineno"> 116</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  </div> |
| <div class="line"><a name="l00121"></a><span class="lineno"> 121</span> <span class="preprocessor">#ifndef MSHADOW_USE_CUDNN</span></div> |
| <div class="line"><a name="l00122"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#affa4511f720838acfdbbc5f1da36a6e6"> 122</a></span> <span class="preprocessor"> #define MSHADOW_USE_CUDNN 0</span></div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  </div> |
| <div class="line"><a name="l00128"></a><span class="lineno"> 128</span> <span class="preprocessor">#ifndef MSHADOW_USE_CUTENSOR</span></div> |
| <div class="line"><a name="l00129"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#ac72d801c1e8fb17cf9a0a6b84a880c81"> 129</a></span> <span class="preprocessor"> #define MSHADOW_USE_CUTENSOR 0</span></div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  </div> |
| <div class="line"><a name="l00135"></a><span class="lineno"> 135</span> <span class="preprocessor">#ifndef MSHADOW_USE_CUSOLVER</span></div> |
| <div class="line"><a name="l00136"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a97592c78424f2386677ad622d93d27ee"> 136</a></span> <span class="preprocessor"> #define MSHADOW_USE_CUSOLVER MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00137"></a><span class="lineno"> 137</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  </div> |
| <div class="line"><a name="l00143"></a><span class="lineno"> 143</span> <span class="preprocessor">#ifndef MSHADOW_OLD_CUDA</span></div> |
| <div class="line"><a name="l00144"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a48188fc0839e136ff0f0f5ebafe24e19"> 144</a></span> <span class="preprocessor">#define MSHADOW_OLD_CUDA 0</span></div> |
| <div class="line"><a name="l00145"></a><span class="lineno"> 145</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  </div> |
| <div class="line"><a name="l00148"></a><span class="lineno"> 148</span> <span class="preprocessor">#ifndef MSHADOW_USE_SSE</span></div> |
| <div class="line"><a name="l00149"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a08b353dc670d04847f78a70a72856058"> 149</a></span> <span class="preprocessor"> #define MSHADOW_USE_SSE 1</span></div> |
| <div class="line"><a name="l00150"></a><span class="lineno"> 150</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00151"></a><span class="lineno"> 151</span>  </div> |
| <div class="line"><a name="l00153"></a><span class="lineno"> 153</span> <span class="preprocessor">#ifndef MSHADOW_USE_F16C</span></div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span> <span class="preprocessor"> #if defined(_MSC_VER) || defined(__CUDACC__)</span></div> |
| <div class="line"><a name="l00155"></a><span class="lineno"> 155</span> <span class="preprocessor"> #define MSHADOW_USE_F16C 0</span></div> |
| <div class="line"><a name="l00156"></a><span class="lineno"> 156</span> <span class="preprocessor"> #elif defined(__clang__) && \</span></div> |
| <div class="line"><a name="l00157"></a><span class="lineno"> 157</span> <span class="preprocessor"> ((__clang_major__ < 8) || ((__clang_major__ == 8) && (__clang_minor__ < 1)))</span></div> |
| <div class="line"><a name="l00158"></a><span class="lineno"> 158</span> <span class="preprocessor"> #define MSHADOW_USE_F16C 0</span></div> |
| <div class="line"><a name="l00159"></a><span class="lineno"> 159</span> <span class="preprocessor"> #else</span></div> |
| <div class="line"><a name="l00160"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af914aa3e4887150dd618b6162f88803f"> 160</a></span> <span class="preprocessor"> #define MSHADOW_USE_F16C 1</span></div> |
| <div class="line"><a name="l00161"></a><span class="lineno"> 161</span> <span class="preprocessor"> #endif</span></div> |
| <div class="line"><a name="l00162"></a><span class="lineno"> 162</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  </div> |
| <div class="line"><a name="l00165"></a><span class="lineno"> 165</span> <span class="preprocessor">#ifndef MSHADOW_USE_NVML</span></div> |
| <div class="line"><a name="l00166"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a2ff0316ca02738dcfd65dacba7b964a0"> 166</a></span> <span class="preprocessor"> #define MSHADOW_USE_NVML 0</span></div> |
| <div class="line"><a name="l00167"></a><span class="lineno"> 167</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00168"></a><span class="lineno"> 168</span> <span class="comment">// SSE is conflict with cudacc</span></div> |
| <div class="line"><a name="l00169"></a><span class="lineno"> 169</span> <span class="preprocessor">#ifdef __CUDACC__</span></div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span> <span class="preprocessor"> #undef MSHADOW_USE_SSE</span></div> |
| <div class="line"><a name="l00171"></a><span class="lineno"> 171</span> <span class="preprocessor"> #define MSHADOW_USE_SSE 0</span></div> |
| <div class="line"><a name="l00172"></a><span class="lineno"> 172</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00173"></a><span class="lineno"> 173</span>  </div> |
| <div class="line"><a name="l00174"></a><span class="lineno"> 174</span> <span class="preprocessor">#if MSHADOW_USE_CBLAS</span></div> |
| <div class="line"><a name="l00175"></a><span class="lineno"> 175</span> <span class="keyword">extern</span> <span class="stringliteral">"C"</span> {</div> |
| <div class="line"><a name="l00176"></a><span class="lineno"> 176</span> <span class="preprocessor"> #include <cblas.h></span></div> |
| <div class="line"><a name="l00177"></a><span class="lineno"> 177</span> }</div> |
| <div class="line"><a name="l00178"></a><span class="lineno"> 178</span> <span class="preprocessor">#elif MSHADOW_USE_MKL</span></div> |
| <div class="line"><a name="l00179"></a><span class="lineno"> 179</span> <span class="preprocessor"> #if MSHADOW_INT64_TENSOR_SIZE == 1</span></div> |
| <div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  <span class="comment">// Define MKL_INT here to use exactly the same 64bits integer type definitions.</span></div> |
| <div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  <span class="comment">// If MKL_INT will not be defined here, the mkl header defines it as long long int.</span></div> |
| <div class="line"><a name="l00182"></a><span class="lineno"> 182</span> <span class="preprocessor"> #define MKL_INT int64_t</span></div> |
| <div class="line"><a name="l00183"></a><span class="lineno"> 183</span> <span class="preprocessor"> #define MKL_UINT uint64_t</span></div> |
| <div class="line"><a name="l00184"></a><span class="lineno"> 184</span> <span class="preprocessor"> #endif</span></div> |
| <div class="line"><a name="l00185"></a><span class="lineno"> 185</span> <span class="preprocessor"> #include <mkl_blas.h></span></div> |
| <div class="line"><a name="l00186"></a><span class="lineno"> 186</span> <span class="preprocessor"> #include <mkl_cblas.h></span></div> |
| <div class="line"><a name="l00187"></a><span class="lineno"> 187</span> <span class="preprocessor"> #include <mkl_vsl.h></span></div> |
| <div class="line"><a name="l00188"></a><span class="lineno"> 188</span> <span class="preprocessor"> #include <mkl_vsl_functions.h></span></div> |
| <div class="line"><a name="l00189"></a><span class="lineno"> 189</span> <span class="preprocessor"> #include <mkl_version.h></span></div> |
| <div class="line"><a name="l00190"></a><span class="lineno"> 190</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00191"></a><span class="lineno"> 191</span>  </div> |
| <div class="line"><a name="l00192"></a><span class="lineno"> 192</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00193"></a><span class="lineno"> 193</span> <span class="preprocessor"> #include <cuda.h></span></div> |
| <div class="line"><a name="l00194"></a><span class="lineno"> 194</span> <span class="preprocessor"> #include <cublas_v2.h></span></div> |
| <div class="line"><a name="l00195"></a><span class="lineno"> 195</span> <span class="preprocessor"> #include <curand.h></span></div> |
| <div class="line"><a name="l00196"></a><span class="lineno"> 196</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  </div> |
| <div class="line"><a name="l00198"></a><span class="lineno"> 198</span> <span class="preprocessor">#if MSHADOW_USE_CUDNN == 1</span></div> |
| <div class="line"><a name="l00199"></a><span class="lineno"> 199</span> <span class="preprocessor"> #include <cudnn.h></span></div> |
| <div class="line"><a name="l00200"></a><span class="lineno"> 200</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  </div> |
| <div class="line"><a name="l00202"></a><span class="lineno"> 202</span> <span class="preprocessor">#if MSHADOW_USE_CUTENSOR == 1</span></div> |
| <div class="line"><a name="l00203"></a><span class="lineno"> 203</span> <span class="preprocessor"> #include <cutensor.h></span></div> |
| <div class="line"><a name="l00204"></a><span class="lineno"> 204</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  </div> |
| <div class="line"><a name="l00206"></a><span class="lineno"> 206</span> <span class="preprocessor">#if MSHADOW_USE_CUSOLVER == 1</span></div> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span> <span class="preprocessor"> #include <cusolverDn.h></span></div> |
| <div class="line"><a name="l00208"></a><span class="lineno"> 208</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  </div> |
| <div class="line"><a name="l00210"></a><span class="lineno"> 210</span> <span class="preprocessor">#if MSHADOW_USE_NVML</span></div> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span> <span class="preprocessor"> #include <nvml.h></span></div> |
| <div class="line"><a name="l00212"></a><span class="lineno"> 212</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  </div> |
| <div class="line"><a name="l00214"></a><span class="lineno"> 214</span> <span class="comment">// --------------------------------</span></div> |
| <div class="line"><a name="l00215"></a><span class="lineno"> 215</span> <span class="comment">// MSHADOW_XINLINE is used for inlining template code for both CUDA and CPU code</span></div> |
| <div class="line"><a name="l00216"></a><span class="lineno"> 216</span> <span class="preprocessor">#ifdef MSHADOW_XINLINE</span></div> |
| <div class="line"><a name="l00217"></a><span class="lineno"> 217</span> <span class="preprocessor"> #error "MSHADOW_XINLINE must not be defined"</span></div> |
| <div class="line"><a name="l00218"></a><span class="lineno"> 218</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00219"></a><span class="lineno"> 219</span> <span class="preprocessor">#ifdef _MSC_VER</span></div> |
| <div class="line"><a name="l00220"></a><span class="lineno"> 220</span> <span class="preprocessor">#define MSHADOW_FORCE_INLINE __forceinline</span></div> |
| <div class="line"><a name="l00221"></a><span class="lineno"> 221</span> <span class="preprocessor">#pragma warning(disable : 4068)</span></div> |
| <div class="line"><a name="l00222"></a><span class="lineno"> 222</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00223"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#ad9835bbd312a2aead44b7870947ef884"> 223</a></span> <span class="preprocessor">#define MSHADOW_FORCE_INLINE inline __attribute__((always_inline))</span></div> |
| <div class="line"><a name="l00224"></a><span class="lineno"> 224</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00225"></a><span class="lineno"> 225</span> <span class="preprocessor">#ifdef __CUDACC__</span></div> |
| <div class="line"><a name="l00226"></a><span class="lineno"> 226</span> <span class="preprocessor"> #define MSHADOW_XINLINE MSHADOW_FORCE_INLINE __device__ __host__</span></div> |
| <div class="line"><a name="l00227"></a><span class="lineno"> 227</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00228"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69"> 228</a></span> <span class="preprocessor"> #define MSHADOW_XINLINE MSHADOW_FORCE_INLINE</span></div> |
| <div class="line"><a name="l00229"></a><span class="lineno"> 229</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00230"></a><span class="lineno"> 230</span>  </div> |
| <div class="line"><a name="l00231"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#ae3f37fc5e892a11db794e475fae404a8"> 231</a></span> <span class="preprocessor">#define MSHADOW_CINLINE MSHADOW_FORCE_INLINE</span></div> |
| <div class="line"><a name="l00232"></a><span class="lineno"> 232</span>  </div> |
| <div class="line"><a name="l00239"></a><span class="lineno"> 239</span> <span class="preprocessor">#ifndef MSHADOW_DEFAULT_DTYPE</span></div> |
| <div class="line"><a name="l00240"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a1588e3cdf209b0cf13609d1c0e3838c7"> 240</a></span> <span class="preprocessor">#define MSHADOW_DEFAULT_DTYPE = ::mshadow::default_real_t</span></div> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  </div> |
| <div class="line"><a name="l00246"></a><span class="lineno"> 246</span> <span class="preprocessor">#ifndef MSHADOW_USE_GLOG</span></div> |
| <div class="line"><a name="l00247"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a6444c537a4443d99d62604642e6422c8"> 247</a></span> <span class="preprocessor">#define MSHADOW_USE_GLOG DMLC_USE_GLOG</span></div> |
| <div class="line"><a name="l00248"></a><span class="lineno"> 248</span> <span class="preprocessor">#endif // MSHADOW_USE_GLOG</span></div> |
| <div class="line"><a name="l00249"></a><span class="lineno"> 249</span>  </div> |
| <div class="line"><a name="l00250"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a53a342d1f8cc22d36ed49e4c51c19f0c"> 250</a></span> <span class="preprocessor">#define MSHADOW_THROW_EXCEPTION noexcept(false)</span></div> |
| <div class="line"><a name="l00251"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a26c198967f39ff94cd560ea82e773f8d"> 251</a></span> <span class="preprocessor">#define MSHADOW_NO_EXCEPTION noexcept(true)</span></div> |
| <div class="line"><a name="l00252"></a><span class="lineno"> 252</span>  </div> |
| <div class="line"><a name="l00253"></a><span class="lineno"> 253</span> <span class="preprocessor">#if defined(_MSC_VER)</span></div> |
| <div class="line"><a name="l00254"></a><span class="lineno"> 254</span> <span class="preprocessor">#define MSHADOW_ALIGNED(x) __declspec(align(x))</span></div> |
| <div class="line"><a name="l00255"></a><span class="lineno"> 255</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00256"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a9a2941102b46ebdb39e04d66da40d238"> 256</a></span> <span class="preprocessor">#define MSHADOW_ALIGNED(x) __attribute__ ((aligned(x)))</span></div> |
| <div class="line"><a name="l00257"></a><span class="lineno"> 257</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  </div> |
| <div class="line"><a name="l00264"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a8f433b4dd005a854eec58178ffd3d4bd"> 264</a></span> <span class="preprocessor">#define MSHADOW_CUDA_CALL(func) \</span></div> |
| <div class="line"><a name="l00265"></a><span class="lineno"> 265</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l00266"></a><span class="lineno"> 266</span> <span class="preprocessor"> cudaError_t e = (func); \</span></div> |
| <div class="line"><a name="l00267"></a><span class="lineno"> 267</span> <span class="preprocessor"> if (e == cudaErrorCudartUnloading) { \</span></div> |
| <div class="line"><a name="l00268"></a><span class="lineno"> 268</span> <span class="preprocessor"> throw dmlc::Error(cudaGetErrorString(e)); \</span></div> |
| <div class="line"><a name="l00269"></a><span class="lineno"> 269</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l00270"></a><span class="lineno"> 270</span> <span class="preprocessor"> CHECK_EQ(e, cudaSuccess) \</span></div> |
| <div class="line"><a name="l00271"></a><span class="lineno"> 271</span> <span class="preprocessor"> << "CUDA: " << cudaGetErrorString(e); \</span></div> |
| <div class="line"><a name="l00272"></a><span class="lineno"> 272</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l00273"></a><span class="lineno"> 273</span>  </div> |
| <div class="line"><a name="l00278"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a144171091678fc9f3cbc8a9612db899d"> 278</a></span> <span class="preprocessor">#define MSHADOW_CATCH_ERROR(func) \</span></div> |
| <div class="line"><a name="l00279"></a><span class="lineno"> 279</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l00280"></a><span class="lineno"> 280</span> <span class="preprocessor"> try { \</span></div> |
| <div class="line"><a name="l00281"></a><span class="lineno"> 281</span> <span class="preprocessor"> (func); \</span></div> |
| <div class="line"><a name="l00282"></a><span class="lineno"> 282</span> <span class="preprocessor"> } catch (const dmlc::Error &e) { \</span></div> |
| <div class="line"><a name="l00283"></a><span class="lineno"> 283</span> <span class="preprocessor"> std::string what = e.what(); \</span></div> |
| <div class="line"><a name="l00284"></a><span class="lineno"> 284</span> <span class="preprocessor"> if (what.find("driver shutting down") == std::string::npos) { \</span></div> |
| <div class="line"><a name="l00285"></a><span class="lineno"> 285</span> <span class="preprocessor"> LOG(ERROR) << "Ignore CUDA Error " << what; \</span></div> |
| <div class="line"><a name="l00286"></a><span class="lineno"> 286</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l00287"></a><span class="lineno"> 287</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l00288"></a><span class="lineno"> 288</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l00289"></a><span class="lineno"> 289</span>  </div> |
| <div class="line"><a name="l00290"></a><span class="lineno"> 290</span> <span class="preprocessor">#include "<a class="code" href="half_8h.html">./half.h</a>"</span></div> |
| <div class="line"><a name="l00291"></a><span class="lineno"> 291</span> <span class="preprocessor">#include "<a class="code" href="bfloat_8h.html">./bfloat.h</a>"</span></div> |
| <div class="line"><a name="l00292"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856"> 292</a></span> <span class="preprocessor">#define MSHADOW_HALF_BF_OPERATOR(RTYPE, OP) \</span></div> |
| <div class="line"><a name="l00293"></a><span class="lineno"> 293</span> <span class="preprocessor"> MSHADOW_XINLINE RTYPE operator OP(mshadow::half::half_t a, mshadow::bfloat::bf16_t b) { \</span></div> |
| <div class="line"><a name="l00294"></a><span class="lineno"> 294</span> <span class="preprocessor"> return float(a) OP float(b); </span><span class="comment">/* NOLINT(*) */</span><span class="preprocessor"> \</span></div> |
| <div class="line"><a name="l00295"></a><span class="lineno"> 295</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l00296"></a><span class="lineno"> 296</span> <span class="preprocessor"> MSHADOW_XINLINE RTYPE operator OP(mshadow::bfloat::bf16_t a, mshadow::half::half_t b) { \</span></div> |
| <div class="line"><a name="l00297"></a><span class="lineno"> 297</span> <span class="preprocessor"> return float(a) OP float(b); </span><span class="comment">/* NOLINT(*) */</span><span class="preprocessor"> \</span></div> |
| <div class="line"><a name="l00298"></a><span class="lineno"> 298</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l00299"></a><span class="lineno"> 299</span>  </div> |
| <div class="line"><a name="l00301"></a><span class="lineno"> 301</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">float</span>, +)</div> |
| <div class="line"><a name="l00303"></a><span class="lineno"> 303</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">float</span>, -)</div> |
| <div class="line"><a name="l00305"></a><span class="lineno"> 305</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">float</span>, *)</div> |
| <div class="line"><a name="l00307"></a><span class="lineno"> 307</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">float</span>, /)</div> |
| <div class="line"><a name="l00309"></a><span class="lineno"> 309</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">bool</span>, >)</div> |
| <div class="line"><a name="l00311"></a><span class="lineno"> 311</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">bool</span>, <)</div> |
| <div class="line"><a name="l00313"></a><span class="lineno"> 313</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">bool</span>, >=)</div> |
| <div class="line"><a name="l00315"></a><span class="lineno"> 315</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a>(<span class="keywordtype">bool</span>, <=)</div> |
| <div class="line"><a name="l00316"></a><span class="lineno"> 316</span>  </div> |
| <div class="line"><a name="l00317"></a><span class="lineno"> 317</span> <span class="preprocessor">#include "dmlc/logging.h"</span></div> |
| <div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  </div> |
| <div class="line"><a name="l00319"></a><span class="lineno"><a class="line" href="namespacemshadow.html"> 319</a></span> <span class="keyword">namespace </span><a class="code" href="namespacemshadow.html">mshadow</a> {</div> |
| <div class="line"><a name="l00321"></a><span class="lineno"><a class="line" href="namespacemshadow.html#ac4cba6b672478463350f7d84b47e99e3"> 321</a></span> <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <a class="code" href="namespacemshadow.html#ac4cba6b672478463350f7d84b47e99e3">kRandBufferSize</a> = 1000000;</div> |
| <div class="line"><a name="l00323"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a8afab1f2134d65e09679faa324ecf9ce"> 323</a></span> <span class="keyword">const</span> <span class="keywordtype">float</span> <a class="code" href="namespacemshadow.html#a8afab1f2134d65e09679faa324ecf9ce">kPi</a> = 3.1415926f;</div> |
| <div class="line"><a name="l00325"></a><span class="lineno"> 325</span> <span class="preprocessor">#if MSHADOW_INT64_TENSOR_SIZE == 1</span></div> |
| <div class="line"><a name="l00326"></a><span class="lineno"> 326</span>  <span class="keyword">typedef</span> int64_t <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a>;</div> |
| <div class="line"><a name="l00327"></a><span class="lineno"> 327</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00328"></a><span class="lineno"><a class="line" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926"> 328</a></span>  <span class="keyword">typedef</span> int32_t <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a>;</div> |
| <div class="line"><a name="l00329"></a><span class="lineno"> 329</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00330"></a><span class="lineno"> 330</span>  </div> |
| <div class="line"><a name="l00331"></a><span class="lineno"> 331</span> <span class="preprocessor">#ifdef _WIN32</span></div> |
| <div class="line"><a name="l00332"></a><span class="lineno"> 332</span>  </div> |
| <div class="line"><a name="l00333"></a><span class="lineno"> 333</span>  <span class="keyword">typedef</span> int64_t <a class="code" href="namespacemshadow.html#aa94c153a2f8bcd10be82a4f5598ae5b9">openmp_index_t</a>;</div> |
| <div class="line"><a name="l00334"></a><span class="lineno"> 334</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00335"></a><span class="lineno"> 335</span>  </div> |
| <div class="line"><a name="l00336"></a><span class="lineno"><a class="line" href="namespacemshadow.html#aa94c153a2f8bcd10be82a4f5598ae5b9"> 336</a></span>  <span class="keyword">typedef</span> <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a> <a class="code" href="namespacemshadow.html#aa94c153a2f8bcd10be82a4f5598ae5b9">openmp_index_t</a>;</div> |
| <div class="line"><a name="l00337"></a><span class="lineno"> 337</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00338"></a><span class="lineno"> 338</span>  </div> |
| <div class="line"><a name="l00339"></a><span class="lineno"> 339</span>  </div> |
| <div class="line"><a name="l00340"></a><span class="lineno"> 340</span> <span class="preprocessor">#if (MSHADOW_USE_MKL && MXNET_USE_LAPACK) || MXNET_USE_ILP64_LAPACKE</span></div> |
| <div class="line"><a name="l00341"></a><span class="lineno"> 341</span>  <span class="comment">// lapack_index_t could be replaced by index_t and removed when all blas library support large tensor</span></div> |
| <div class="line"><a name="l00342"></a><span class="lineno"> 342</span>  <span class="keyword">typedef</span> <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a> <a class="code" href="namespacemshadow.html#a8b60ab0b7ba58978695b4ab7e63268b9">lapack_index_t</a>;</div> |
| <div class="line"><a name="l00343"></a><span class="lineno"> 343</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00344"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a8b60ab0b7ba58978695b4ab7e63268b9"> 344</a></span>  <span class="keyword">typedef</span> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow.html#a8b60ab0b7ba58978695b4ab7e63268b9">lapack_index_t</a>;</div> |
| <div class="line"><a name="l00345"></a><span class="lineno"> 345</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00346"></a><span class="lineno"> 346</span>  </div> |
| <div class="line"><a name="l00348"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce"> 348</a></span> <span class="keyword">typedef</span> <span class="keywordtype">float</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a>;</div> |
| <div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  </div> |
| <div class="line"><a name="l00351"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1"> 351</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1">TypeFlag</a> {</div> |
| <div class="line"><a name="l00352"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633"> 352</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">kFloat32</a> = 0,</div> |
| <div class="line"><a name="l00353"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28"> 353</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28">kFloat64</a> = 1,</div> |
| <div class="line"><a name="l00354"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4"> 354</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4">kFloat16</a> = 2,</div> |
| <div class="line"><a name="l00355"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0"> 355</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0">kUint8</a> = 3,</div> |
| <div class="line"><a name="l00356"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368"> 356</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368">kInt32</a> = 4,</div> |
| <div class="line"><a name="l00357"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c"> 357</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c">kInt8</a> = 5,</div> |
| <div class="line"><a name="l00358"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23"> 358</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23">kInt64</a> = 6,</div> |
| <div class="line"><a name="l00359"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a"> 359</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a">kBool</a> = 7,</div> |
| <div class="line"><a name="l00360"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17"> 360</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17">kInt16</a> = 8,</div> |
| <div class="line"><a name="l00361"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908"> 361</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908">kUint16</a> = 9,</div> |
| <div class="line"><a name="l00362"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7"> 362</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7">kUint32</a> = 10,</div> |
| <div class="line"><a name="l00363"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade"> 363</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade">kUint64</a> = 11,</div> |
| <div class="line"><a name="l00364"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e"> 364</a></span>  <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e">kBfloat16</a> = 12</div> |
| <div class="line"><a name="l00365"></a><span class="lineno"> 365</span> };</div> |
| <div class="line"><a name="l00366"></a><span class="lineno"> 366</span>  </div> |
| <div class="line"><a name="l00367"></a><span class="lineno"> 367</span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00368"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType.html"> 368</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a>;</div> |
| <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  </div> |
| <div class="line"><a name="l00370"></a><span class="lineno"> 370</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00371"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01float_01_4.html"> 371</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><float> {</div> |
| <div class="line"><a name="l00372"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01float_01_4.html#a9d3121fa39e365da9a2446e48a53d931"> 372</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">kFloat32</a>;</div> |
| <div class="line"><a name="l00373"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01float_01_4.html#a504a7e54598690e29928a89a76f15a09"> 373</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00374"></a><span class="lineno"> 374</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00375"></a><span class="lineno"> 375</span> <span class="preprocessor">#if (CUDA_VERSION >= 8000)</span></div> |
| <div class="line"><a name="l00376"></a><span class="lineno"> 376</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudaDataType_t kCudaFlag = CUDA_R_32F;</div> |
| <div class="line"><a name="l00377"></a><span class="lineno"> 377</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00378"></a><span class="lineno"> 378</span> <span class="preprocessor">#if MSHADOW_USE_CUDNN</span></div> |
| <div class="line"><a name="l00379"></a><span class="lineno"> 379</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnDataType_t kCudnnFlag = CUDNN_DATA_FLOAT;</div> |
| <div class="line"><a name="l00380"></a><span class="lineno"> 380</span>  <span class="keyword">typedef</span> <span class="keywordtype">float</span> ScaleType;</div> |
| <div class="line"><a name="l00381"></a><span class="lineno"> 381</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00382"></a><span class="lineno"> 382</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00383"></a><span class="lineno"> 383</span> };</div> |
| <div class="line"><a name="l00384"></a><span class="lineno"> 384</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00385"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01double_01_4.html"> 385</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><double> {</div> |
| <div class="line"><a name="l00386"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01double_01_4.html#a658d2110dd485145654b8af031b0cf0b"> 386</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28">kFloat64</a>;</div> |
| <div class="line"><a name="l00387"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01double_01_4.html#aedb50ba7ff19dc149a3acedafb7f6d65"> 387</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00388"></a><span class="lineno"> 388</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00389"></a><span class="lineno"> 389</span> <span class="preprocessor">#if (CUDA_VERSION >= 8000)</span></div> |
| <div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudaDataType_t kCudaFlag = CUDA_R_64F;</div> |
| <div class="line"><a name="l00391"></a><span class="lineno"> 391</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00392"></a><span class="lineno"> 392</span> <span class="preprocessor">#if MSHADOW_USE_CUDNN</span></div> |
| <div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnDataType_t kCudnnFlag = CUDNN_DATA_DOUBLE;</div> |
| <div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  <span class="keyword">typedef</span> <span class="keywordtype">double</span> ScaleType;</div> |
| <div class="line"><a name="l00395"></a><span class="lineno"> 395</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00396"></a><span class="lineno"> 396</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00397"></a><span class="lineno"> 397</span> };</div> |
| <div class="line"><a name="l00398"></a><span class="lineno"> 398</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00399"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01half_1_1half__t_01_4.html"> 399</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><half::half_t> {</div> |
| <div class="line"><a name="l00400"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01half_1_1half__t_01_4.html#a0c637cd624136613106ced1caf5c3b43"> 400</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4">kFloat16</a>;</div> |
| <div class="line"><a name="l00401"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01half_1_1half__t_01_4.html#a10fa4c69e1ef38e88804c7f02ca023ef"> 401</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00402"></a><span class="lineno"> 402</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00403"></a><span class="lineno"> 403</span> <span class="preprocessor">#if (CUDA_VERSION >= 8000)</span></div> |
| <div class="line"><a name="l00404"></a><span class="lineno"> 404</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudaDataType_t kCudaFlag = CUDA_R_16F;</div> |
| <div class="line"><a name="l00405"></a><span class="lineno"> 405</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00406"></a><span class="lineno"> 406</span> <span class="preprocessor">#if MSHADOW_USE_CUDNN</span></div> |
| <div class="line"><a name="l00407"></a><span class="lineno"> 407</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnDataType_t kCudnnFlag = CUDNN_DATA_HALF;</div> |
| <div class="line"><a name="l00408"></a><span class="lineno"> 408</span>  <span class="keyword">typedef</span> <span class="keywordtype">float</span> ScaleType;</div> |
| <div class="line"><a name="l00409"></a><span class="lineno"> 409</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00410"></a><span class="lineno"> 410</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00411"></a><span class="lineno"> 411</span> };</div> |
| <div class="line"><a name="l00412"></a><span class="lineno"> 412</span> <span class="keyword">template</span> <></div> |
| <div class="line"><a name="l00413"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01bfloat_1_1bf16__t_01_4.html"> 413</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><bfloat::bf16_t> {</div> |
| <div class="line"><a name="l00414"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01bfloat_1_1bf16__t_01_4.html#abf8e679c7d024771f8b41f3e87ae044a"> 414</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e">kBfloat16</a>;</div> |
| <div class="line"><a name="l00415"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01bfloat_1_1bf16__t_01_4.html#a4b583091e70524e4590811ade2c6770c"> 415</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00416"></a><span class="lineno"> 416</span> };</div> |
| <div class="line"><a name="l00417"></a><span class="lineno"> 417</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00418"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint8__t_01_4.html"> 418</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><uint8_t> {</div> |
| <div class="line"><a name="l00419"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint8__t_01_4.html#a3a760afb8012b5db5f75bfa1c5f5f1f8"> 419</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0">kUint8</a>;</div> |
| <div class="line"><a name="l00420"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint8__t_01_4.html#a73bfeffab7e4d68994c298a80c68b536"> 420</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00421"></a><span class="lineno"> 421</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00422"></a><span class="lineno"> 422</span> <span class="preprocessor">#if (CUDA_VERSION >= 8000)</span></div> |
| <div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudaDataType_t kCudaFlag = CUDA_R_8U;</div> |
| <div class="line"><a name="l00424"></a><span class="lineno"> 424</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00425"></a><span class="lineno"> 425</span> <span class="preprocessor">#if (MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 6)</span></div> |
| <div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  <span class="comment">// no uint8 in cudnn for now</span></div> |
| <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnDataType_t kCudnnFlag = CUDNN_DATA_INT8;</div> |
| <div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="keyword">typedef</span> uint8_t ScaleType;</div> |
| <div class="line"><a name="l00429"></a><span class="lineno"> 429</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00430"></a><span class="lineno"> 430</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00431"></a><span class="lineno"> 431</span> };</div> |
| <div class="line"><a name="l00432"></a><span class="lineno"> 432</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00433"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int8__t_01_4.html"> 433</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><int8_t> {</div> |
| <div class="line"><a name="l00434"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int8__t_01_4.html#a5d224a971dcc10ced072758683ce3fbb"> 434</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c">kInt8</a>;</div> |
| <div class="line"><a name="l00435"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int8__t_01_4.html#ab40b8f46826e7af9a45c1b05398705d2"> 435</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00436"></a><span class="lineno"> 436</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00437"></a><span class="lineno"> 437</span> <span class="preprocessor">#if (CUDA_VERSION >= 8000)</span></div> |
| <div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudaDataType_t kCudaFlag = CUDA_R_8I;</div> |
| <div class="line"><a name="l00439"></a><span class="lineno"> 439</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00440"></a><span class="lineno"> 440</span> <span class="preprocessor">#if (MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 6)</span></div> |
| <div class="line"><a name="l00441"></a><span class="lineno"> 441</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnDataType_t kCudnnFlag = CUDNN_DATA_INT8;</div> |
| <div class="line"><a name="l00442"></a><span class="lineno"> 442</span>  <span class="keyword">typedef</span> int8_t ScaleType;</div> |
| <div class="line"><a name="l00443"></a><span class="lineno"> 443</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00444"></a><span class="lineno"> 444</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00445"></a><span class="lineno"> 445</span> };</div> |
| <div class="line"><a name="l00446"></a><span class="lineno"> 446</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00447"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int32__t_01_4.html"> 447</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><int32_t> {</div> |
| <div class="line"><a name="l00448"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int32__t_01_4.html#a97dcb369c90d76a380bca80c62f23be3"> 448</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368">kInt32</a>;</div> |
| <div class="line"><a name="l00449"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int32__t_01_4.html#a0f5130ae65b507a23e7bd8c3098b7a1e"> 449</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00450"></a><span class="lineno"> 450</span> <span class="preprocessor">#if MSHADOW_USE_CUDA</span></div> |
| <div class="line"><a name="l00451"></a><span class="lineno"> 451</span> <span class="preprocessor">#if (CUDA_VERSION >= 8000)</span></div> |
| <div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudaDataType_t kCudaFlag = CUDA_R_32I;</div> |
| <div class="line"><a name="l00453"></a><span class="lineno"> 453</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00454"></a><span class="lineno"> 454</span> <span class="preprocessor">#if (MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 6)</span></div> |
| <div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnDataType_t kCudnnFlag = CUDNN_DATA_INT32;</div> |
| <div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  <span class="keyword">typedef</span> int32_t ScaleType;</div> |
| <div class="line"><a name="l00457"></a><span class="lineno"> 457</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00458"></a><span class="lineno"> 458</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00459"></a><span class="lineno"> 459</span> };</div> |
| <div class="line"><a name="l00460"></a><span class="lineno"> 460</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00461"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int64__t_01_4.html"> 461</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><int64_t> {</div> |
| <div class="line"><a name="l00462"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int64__t_01_4.html#aed15ad327b09b35a4a6d7e52cb8c25aa"> 462</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23">kInt64</a>;</div> |
| <div class="line"><a name="l00463"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int64__t_01_4.html#ac21b9b3c8bc8edb01e183ebf97d05a09"> 463</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00464"></a><span class="lineno"> 464</span> };</div> |
| <div class="line"><a name="l00465"></a><span class="lineno"> 465</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00466"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01bool_01_4.html"> 466</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><bool> {</div> |
| <div class="line"><a name="l00467"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01bool_01_4.html#a01e3a14bc3d13155ad0bf31927a197b4"> 467</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a">kBool</a>;</div> |
| <div class="line"><a name="l00468"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01bool_01_4.html#a1510a26bddc961b00ad25a2870db47db"> 468</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00469"></a><span class="lineno"> 469</span> };</div> |
| <div class="line"><a name="l00470"></a><span class="lineno"> 470</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00471"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int16__t_01_4.html"> 471</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><int16_t> {</div> |
| <div class="line"><a name="l00472"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int16__t_01_4.html#a9c9f68ee9926a631376271f9bc47efb7"> 472</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17">kInt16</a>;</div> |
| <div class="line"><a name="l00473"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01int16__t_01_4.html#a2198709e1eee3a1ae3f29b55ad1c45e8"> 473</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00474"></a><span class="lineno"> 474</span> };</div> |
| <div class="line"><a name="l00475"></a><span class="lineno"> 475</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00476"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint16__t_01_4.html"> 476</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><uint16_t> {</div> |
| <div class="line"><a name="l00477"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint16__t_01_4.html#ac79270e8afccac6e1922b3fdeb43d8e1"> 477</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908">kUint16</a>;</div> |
| <div class="line"><a name="l00478"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint16__t_01_4.html#a54e7ca5dc02cd297084cbb25ac341bd8"> 478</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00479"></a><span class="lineno"> 479</span> };</div> |
| <div class="line"><a name="l00480"></a><span class="lineno"> 480</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00481"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint32__t_01_4.html"> 481</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><uint32_t> {</div> |
| <div class="line"><a name="l00482"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint32__t_01_4.html#a198a1388e5a5c7336899d28faa788525"> 482</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7">kUint32</a>;</div> |
| <div class="line"><a name="l00483"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint32__t_01_4.html#a496e40141a0e259b10c0ecd0ace6056f"> 483</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00484"></a><span class="lineno"> 484</span> };</div> |
| <div class="line"><a name="l00485"></a><span class="lineno"> 485</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00486"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint64__t_01_4.html"> 486</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1DataType.html">DataType</a><uint64_t> {</div> |
| <div class="line"><a name="l00487"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint64__t_01_4.html#a3997c7814cc6ae0ec89e7cc9a11717a6"> 487</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kFlag = <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade">kUint64</a>;</div> |
| <div class="line"><a name="l00488"></a><span class="lineno"><a class="line" href="structmshadow_1_1DataType_3_01uint64__t_01_4.html#a144beb412eb92b06b4f290d5d94e88ac"> 488</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kLanes = 1;</div> |
| <div class="line"><a name="l00489"></a><span class="lineno"> 489</span> };</div> |
| <div class="line"><a name="l00490"></a><span class="lineno"> 490</span>  </div> |
| <div class="line"><a name="l00492"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee"> 492</a></span> <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">default_type_flag</a> = <a class="code" href="structmshadow_1_1DataType.html">DataType<default_real_t>::kFlag</a>;</div> |
| <div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  </div> |
| <div class="line"><a name="l00495"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a6959787874454734212c8c30b07a8876"> 495</a></span> <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow.html#a6959787874454734212c8c30b07a8876">index_type_flag</a> = <a class="code" href="structmshadow_1_1DataType.html">DataType<lapack_index_t>::kFlag</a>;</div> |
| <div class="line"><a name="l00496"></a><span class="lineno"> 496</span>  </div> |
| <div class="line"><a name="l00498"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4"> 498</a></span> <span class="keyword">enum</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4">LayoutFlag</a> {</div> |
| <div class="line"><a name="l00499"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c"> 499</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">kUNKNOWN</a> = -1,</div> |
| <div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  </div> |
| <div class="line"><a name="l00501"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775"> 501</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775">kNCHW</a> = 0,</div> |
| <div class="line"><a name="l00502"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281"> 502</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281">kNHWC</a>,</div> |
| <div class="line"><a name="l00503"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a6e334c4df779fcc6fa6b24411fce1289"> 503</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a6e334c4df779fcc6fa6b24411fce1289">kCHWN</a>,</div> |
| <div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  </div> |
| <div class="line"><a name="l00505"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a0cfb685a803481d9ff4fe19d813b261a"> 505</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a0cfb685a803481d9ff4fe19d813b261a">kNCW</a> = 1 << 3,</div> |
| <div class="line"><a name="l00506"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a3f4908b910c47da082d60aa719c43fc3"> 506</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a3f4908b910c47da082d60aa719c43fc3">kNWC</a>,</div> |
| <div class="line"><a name="l00507"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae5189456ea58c60d8c546ab8bd1a2c08"> 507</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae5189456ea58c60d8c546ab8bd1a2c08">kCWN</a>,</div> |
| <div class="line"><a name="l00508"></a><span class="lineno"> 508</span>  </div> |
| <div class="line"><a name="l00509"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822"> 509</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822">kNCDHW</a> = 1 << 5,</div> |
| <div class="line"><a name="l00510"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd"> 510</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd">kNDHWC</a>,</div> |
| <div class="line"><a name="l00511"></a><span class="lineno"><a class="line" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a30ea75730b725ac965fb4e43c34622d8"> 511</a></span>  <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a30ea75730b725ac965fb4e43c34622d8">kCDHWN</a></div> |
| <div class="line"><a name="l00512"></a><span class="lineno"> 512</span> };</div> |
| <div class="line"><a name="l00513"></a><span class="lineno"> 513</span>  </div> |
| <div class="line"><a name="l00514"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a20e2ca3ab09e592f109bc7d7fca7e71d"> 514</a></span> <span class="keyword">inline</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4">LayoutFlag</a> <a class="code" href="namespacemshadow.html#a20e2ca3ab09e592f109bc7d7fca7e71d">layoutFlag</a>(std::string layoutstr) {</div> |
| <div class="line"><a name="l00515"></a><span class="lineno"> 515</span>  <span class="keywordflow">switch</span> (layoutstr.length()) {</div> |
| <div class="line"><a name="l00516"></a><span class="lineno"> 516</span>  <span class="keywordflow">case</span> 4:</div> |
| <div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"NHWC"</span>)</div> |
| <div class="line"><a name="l00518"></a><span class="lineno"> 518</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281">kNHWC</a>;</div> |
| <div class="line"><a name="l00519"></a><span class="lineno"> 519</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"NCHW"</span>)</div> |
| <div class="line"><a name="l00520"></a><span class="lineno"> 520</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775">kNCHW</a>;</div> |
| <div class="line"><a name="l00521"></a><span class="lineno"> 521</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"CHWN"</span>)</div> |
| <div class="line"><a name="l00522"></a><span class="lineno"> 522</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a6e334c4df779fcc6fa6b24411fce1289">kCHWN</a>;</div> |
| <div class="line"><a name="l00523"></a><span class="lineno"> 523</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">kUNKNOWN</a>;</div> |
| <div class="line"><a name="l00524"></a><span class="lineno"> 524</span>  <span class="keywordflow">case</span> 3:</div> |
| <div class="line"><a name="l00525"></a><span class="lineno"> 525</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"NWC"</span>)</div> |
| <div class="line"><a name="l00526"></a><span class="lineno"> 526</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a3f4908b910c47da082d60aa719c43fc3">kNWC</a>;</div> |
| <div class="line"><a name="l00527"></a><span class="lineno"> 527</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"NCW"</span>)</div> |
| <div class="line"><a name="l00528"></a><span class="lineno"> 528</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a0cfb685a803481d9ff4fe19d813b261a">kNCW</a>;</div> |
| <div class="line"><a name="l00529"></a><span class="lineno"> 529</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"CWN"</span>)</div> |
| <div class="line"><a name="l00530"></a><span class="lineno"> 530</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae5189456ea58c60d8c546ab8bd1a2c08">kCWN</a>;</div> |
| <div class="line"><a name="l00531"></a><span class="lineno"> 531</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">kUNKNOWN</a>;</div> |
| <div class="line"><a name="l00532"></a><span class="lineno"> 532</span>  <span class="keywordflow">case</span> 5:</div> |
| <div class="line"><a name="l00533"></a><span class="lineno"> 533</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"NDHWC"</span>)</div> |
| <div class="line"><a name="l00534"></a><span class="lineno"> 534</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd">kNDHWC</a>;</div> |
| <div class="line"><a name="l00535"></a><span class="lineno"> 535</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"NCDHW"</span>)</div> |
| <div class="line"><a name="l00536"></a><span class="lineno"> 536</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822">kNCDHW</a>;</div> |
| <div class="line"><a name="l00537"></a><span class="lineno"> 537</span>  <span class="keywordflow">if</span> (layoutstr == <span class="stringliteral">"CDHWN"</span>)</div> |
| <div class="line"><a name="l00538"></a><span class="lineno"> 538</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a30ea75730b725ac965fb4e43c34622d8">kCDHWN</a>;</div> |
| <div class="line"><a name="l00539"></a><span class="lineno"> 539</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">kUNKNOWN</a>;</div> |
| <div class="line"><a name="l00540"></a><span class="lineno"> 540</span>  <span class="keywordflow">default</span>:</div> |
| <div class="line"><a name="l00541"></a><span class="lineno"> 541</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">kUNKNOWN</a>;</div> |
| <div class="line"><a name="l00542"></a><span class="lineno"> 542</span>  }</div> |
| <div class="line"><a name="l00543"></a><span class="lineno"> 543</span> }</div> |
| <div class="line"><a name="l00544"></a><span class="lineno"> 544</span>  </div> |
| <div class="line"><a name="l00545"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a2744c70ac323798d108e5b99ed9875c6"> 545</a></span> <span class="keyword">inline</span> std::string <a class="code" href="namespacemshadow.html#a2744c70ac323798d108e5b99ed9875c6">toString</a>(<a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4">LayoutFlag</a> layout) {</div> |
| <div class="line"><a name="l00546"></a><span class="lineno"> 546</span>  <span class="keywordflow">switch</span> (layout) {</div> |
| <div class="line"><a name="l00547"></a><span class="lineno"> 547</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">kUNKNOWN</a>:</div> |
| <div class="line"><a name="l00548"></a><span class="lineno"> 548</span>  <span class="keywordflow">return</span> <span class="stringliteral">""</span>;</div> |
| <div class="line"><a name="l00549"></a><span class="lineno"> 549</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775">kNCHW</a>:</div> |
| <div class="line"><a name="l00550"></a><span class="lineno"> 550</span>  <span class="keywordflow">return</span> <span class="stringliteral">"NCHW"</span>;</div> |
| <div class="line"><a name="l00551"></a><span class="lineno"> 551</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281">kNHWC</a>:</div> |
| <div class="line"><a name="l00552"></a><span class="lineno"> 552</span>  <span class="keywordflow">return</span> <span class="stringliteral">"NHWC"</span>;</div> |
| <div class="line"><a name="l00553"></a><span class="lineno"> 553</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a6e334c4df779fcc6fa6b24411fce1289">kCHWN</a>:</div> |
| <div class="line"><a name="l00554"></a><span class="lineno"> 554</span>  <span class="keywordflow">return</span> <span class="stringliteral">"CHWN"</span>;</div> |
| <div class="line"><a name="l00555"></a><span class="lineno"> 555</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a0cfb685a803481d9ff4fe19d813b261a">kNCW</a>:</div> |
| <div class="line"><a name="l00556"></a><span class="lineno"> 556</span>  <span class="keywordflow">return</span> <span class="stringliteral">"NCW"</span>;</div> |
| <div class="line"><a name="l00557"></a><span class="lineno"> 557</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a3f4908b910c47da082d60aa719c43fc3">kNWC</a>:</div> |
| <div class="line"><a name="l00558"></a><span class="lineno"> 558</span>  <span class="keywordflow">return</span> <span class="stringliteral">"NWC"</span>;</div> |
| <div class="line"><a name="l00559"></a><span class="lineno"> 559</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae5189456ea58c60d8c546ab8bd1a2c08">kCWN</a>:</div> |
| <div class="line"><a name="l00560"></a><span class="lineno"> 560</span>  <span class="keywordflow">return</span> <span class="stringliteral">"CWN"</span>;</div> |
| <div class="line"><a name="l00561"></a><span class="lineno"> 561</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822">kNCDHW</a>:</div> |
| <div class="line"><a name="l00562"></a><span class="lineno"> 562</span>  <span class="keywordflow">return</span> <span class="stringliteral">"NCDHW"</span>;</div> |
| <div class="line"><a name="l00563"></a><span class="lineno"> 563</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd">kNDHWC</a>:</div> |
| <div class="line"><a name="l00564"></a><span class="lineno"> 564</span>  <span class="keywordflow">return</span> <span class="stringliteral">"NDHWC"</span>;</div> |
| <div class="line"><a name="l00565"></a><span class="lineno"> 565</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a30ea75730b725ac965fb4e43c34622d8">kCDHWN</a>:</div> |
| <div class="line"><a name="l00566"></a><span class="lineno"> 566</span>  <span class="keywordflow">return</span> <span class="stringliteral">"CDHWN"</span>;</div> |
| <div class="line"><a name="l00567"></a><span class="lineno"> 567</span>  <span class="keywordflow">default</span>:</div> |
| <div class="line"><a name="l00568"></a><span class="lineno"> 568</span>  <span class="keywordflow">return</span> <span class="stringliteral">""</span>;</div> |
| <div class="line"><a name="l00569"></a><span class="lineno"> 569</span>  }</div> |
| <div class="line"><a name="l00570"></a><span class="lineno"> 570</span> }</div> |
| <div class="line"><a name="l00571"></a><span class="lineno"> 571</span>  </div> |
| <div class="line"><a name="l00572"></a><span class="lineno"> 572</span> <span class="keyword">template</span><<span class="keywordtype">int</span> layout></div> |
| <div class="line"><a name="l00573"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType.html"> 573</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1LayoutType.html">LayoutType</a>;</div> |
| <div class="line"><a name="l00574"></a><span class="lineno"> 574</span>  </div> |
| <div class="line"><a name="l00575"></a><span class="lineno"> 575</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00576"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNCHW_01_4.html"> 576</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1LayoutType.html">LayoutType</a><<a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775">kNCHW</a>> {</div> |
| <div class="line"><a name="l00577"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNCHW_01_4.html#a9244a6553f6801877a321ec815f01ae8"> 577</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a> kNdim = 4;</div> |
| <div class="line"><a name="l00578"></a><span class="lineno"> 578</span> <span class="preprocessor">#if (MSHADOW_USE_CUDA && MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 4)</span></div> |
| <div class="line"><a name="l00579"></a><span class="lineno"> 579</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnTensorFormat_t kCudnnFlag = CUDNN_TENSOR_NCHW;</div> |
| <div class="line"><a name="l00580"></a><span class="lineno"> 580</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00581"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNCHW_01_4.html#a493ac509b0bc6c405466e665af218292"> 581</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kCudnnFlag = -1;</div> |
| <div class="line"><a name="l00582"></a><span class="lineno"> 582</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00583"></a><span class="lineno"> 583</span> };</div> |
| <div class="line"><a name="l00584"></a><span class="lineno"> 584</span>  </div> |
| <div class="line"><a name="l00585"></a><span class="lineno"> 585</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00586"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNHWC_01_4.html"> 586</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1LayoutType.html">LayoutType</a><<a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281">kNHWC</a>> {</div> |
| <div class="line"><a name="l00587"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNHWC_01_4.html#a809055ce2162321e9a4326651ee4c897"> 587</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a> kNdim = 4;</div> |
| <div class="line"><a name="l00588"></a><span class="lineno"> 588</span> <span class="preprocessor">#if (MSHADOW_USE_CUDA && MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 4)</span></div> |
| <div class="line"><a name="l00589"></a><span class="lineno"> 589</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnTensorFormat_t kCudnnFlag = CUDNN_TENSOR_NHWC;</div> |
| <div class="line"><a name="l00590"></a><span class="lineno"> 590</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00591"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNHWC_01_4.html#a9bc9929da63ab8dfbd4c8171624cf485"> 591</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kCudnnFlag = -1;</div> |
| <div class="line"><a name="l00592"></a><span class="lineno"> 592</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00593"></a><span class="lineno"> 593</span> };</div> |
| <div class="line"><a name="l00594"></a><span class="lineno"> 594</span>  </div> |
| <div class="line"><a name="l00596"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a36590a455cab713cf4bbb115c4dba247"> 596</a></span> <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow.html#a36590a455cab713cf4bbb115c4dba247">default_layout</a> = <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775">kNCHW</a>;</div> |
| <div class="line"><a name="l00597"></a><span class="lineno"> 597</span>  </div> |
| <div class="line"><a name="l00598"></a><span class="lineno"> 598</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00599"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNCDHW_01_4.html"> 599</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1LayoutType.html">LayoutType</a><<a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822">kNCDHW</a>> {</div> |
| <div class="line"><a name="l00600"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNCDHW_01_4.html#a5cebe847b04e020d4b89e8ccfeda39af"> 600</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a> kNdim = 5;</div> |
| <div class="line"><a name="l00601"></a><span class="lineno"> 601</span> <span class="preprocessor">#if (MSHADOW_USE_CUDA && MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 4)</span></div> |
| <div class="line"><a name="l00602"></a><span class="lineno"> 602</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnTensorFormat_t kCudnnFlag = CUDNN_TENSOR_NCHW;</div> |
| <div class="line"><a name="l00603"></a><span class="lineno"> 603</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00604"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNCDHW_01_4.html#a593764b7925e07d3becd851404f4ccd0"> 604</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kCudnnFlag = -1;</div> |
| <div class="line"><a name="l00605"></a><span class="lineno"> 605</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00606"></a><span class="lineno"> 606</span> };</div> |
| <div class="line"><a name="l00607"></a><span class="lineno"> 607</span>  </div> |
| <div class="line"><a name="l00608"></a><span class="lineno"> 608</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00609"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNDHWC_01_4.html"> 609</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1LayoutType.html">LayoutType</a><<a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd">kNDHWC</a>> {</div> |
| <div class="line"><a name="l00610"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNDHWC_01_4.html#ad8442031a3dbf59b4c8be28b58c0c3ff"> 610</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <a class="code" href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">index_t</a> kNdim = 5;</div> |
| <div class="line"><a name="l00611"></a><span class="lineno"> 611</span> <span class="preprocessor">#if (MSHADOW_USE_CUDA && MSHADOW_USE_CUDNN == 1 && CUDNN_MAJOR >= 4)</span></div> |
| <div class="line"><a name="l00612"></a><span class="lineno"> 612</span>  <span class="keyword">static</span> <span class="keyword">const</span> cudnnTensorFormat_t kCudnnFlag = CUDNN_TENSOR_NHWC;</div> |
| <div class="line"><a name="l00613"></a><span class="lineno"> 613</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l00614"></a><span class="lineno"><a class="line" href="structmshadow_1_1LayoutType_3_01kNDHWC_01_4.html#aacf7a3b6c883d5a9864bbd0a2430ec01"> 614</a></span>  <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> kCudnnFlag = -1;</div> |
| <div class="line"><a name="l00615"></a><span class="lineno"> 615</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00616"></a><span class="lineno"> 616</span> };</div> |
| <div class="line"><a name="l00617"></a><span class="lineno"> 617</span>  </div> |
| <div class="line"><a name="l00619"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a29fa60cc6436415a699971c1a3fd760a"> 619</a></span> <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow.html#a29fa60cc6436415a699971c1a3fd760a">default_layout_5d</a> = <a class="code" href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822">kNCDHW</a>;</div> |
| <div class="line"><a name="l00620"></a><span class="lineno"> 620</span>  </div> |
| <div class="line"><a name="l00622"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1op.html"> 622</a></span> <span class="keyword">namespace </span>op {</div> |
| <div class="line"><a name="l00623"></a><span class="lineno"> 623</span> <span class="comment">// binary operator</span></div> |
| <div class="line"><a name="l00625"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1mul.html"> 625</a></span> <span class="comment"></span><span class="keyword">struct </span><a class="code" href="structmshadow_1_1op_1_1mul.html">mul</a>{</div> |
| <div class="line"><a name="l00627"></a><span class="lineno"> 627</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00628"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1mul.html#a38a7497e4f710d7da3f5b9563d754b21"> 628</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1op_1_1mul.html#a38a7497e4f710d7da3f5b9563d754b21">Map</a>(DType a, DType b) {</div> |
| <div class="line"><a name="l00629"></a><span class="lineno"> 629</span>  <span class="keywordflow">return</span> a * b;</div> |
| <div class="line"><a name="l00630"></a><span class="lineno"> 630</span>  }</div> |
| <div class="line"><a name="l00631"></a><span class="lineno"> 631</span> };</div> |
| <div class="line"><a name="l00633"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1plus.html"> 633</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1op_1_1plus.html">plus</a> {</div> |
| <div class="line"><a name="l00635"></a><span class="lineno"> 635</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00636"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1plus.html#a83fb5b1656992c6fd442a819de83b292"> 636</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1op_1_1plus.html#a83fb5b1656992c6fd442a819de83b292">Map</a>(DType a, DType b) {</div> |
| <div class="line"><a name="l00637"></a><span class="lineno"> 637</span>  <span class="keywordflow">return</span> a + b;</div> |
| <div class="line"><a name="l00638"></a><span class="lineno"> 638</span>  }</div> |
| <div class="line"><a name="l00639"></a><span class="lineno"> 639</span> };</div> |
| <div class="line"><a name="l00641"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1minus.html"> 641</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1op_1_1minus.html">minus</a> {</div> |
| <div class="line"><a name="l00643"></a><span class="lineno"> 643</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00644"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1minus.html#a29e530a31da95225c916e13f1bbcbca6"> 644</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1op_1_1minus.html#a29e530a31da95225c916e13f1bbcbca6">Map</a>(DType a, DType b) {</div> |
| <div class="line"><a name="l00645"></a><span class="lineno"> 645</span>  <span class="keywordflow">return</span> a - b;</div> |
| <div class="line"><a name="l00646"></a><span class="lineno"> 646</span>  }</div> |
| <div class="line"><a name="l00647"></a><span class="lineno"> 647</span> };</div> |
| <div class="line"><a name="l00649"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1div.html"> 649</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1op_1_1div.html">div</a> {</div> |
| <div class="line"><a name="l00651"></a><span class="lineno"> 651</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00652"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1div.html#a5ad9c92273fb4c1a29fc3eb3fba3ae5a"> 652</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1op_1_1div.html#a5ad9c92273fb4c1a29fc3eb3fba3ae5a">Map</a>(DType a, DType b) {</div> |
| <div class="line"><a name="l00653"></a><span class="lineno"> 653</span>  <span class="keywordflow">return</span> a / b;</div> |
| <div class="line"><a name="l00654"></a><span class="lineno"> 654</span>  }</div> |
| <div class="line"><a name="l00655"></a><span class="lineno"> 655</span> };</div> |
| <div class="line"><a name="l00657"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1right.html"> 657</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1op_1_1right.html">right</a> {</div> |
| <div class="line"><a name="l00659"></a><span class="lineno"> 659</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00660"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1right.html#a9377077b54a983fd220a0ba0263227b3"> 660</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1op_1_1right.html#a9377077b54a983fd220a0ba0263227b3">Map</a>(DType a, DType b) {</div> |
| <div class="line"><a name="l00661"></a><span class="lineno"> 661</span>  <span class="keywordflow">return</span> b;</div> |
| <div class="line"><a name="l00662"></a><span class="lineno"> 662</span>  }</div> |
| <div class="line"><a name="l00663"></a><span class="lineno"> 663</span> };</div> |
| <div class="line"><a name="l00664"></a><span class="lineno"> 664</span> <span class="comment">// unary operator/ function: example</span></div> |
| <div class="line"><a name="l00665"></a><span class="lineno"> 665</span> <span class="comment">// these operators can be defined by user,</span></div> |
| <div class="line"><a name="l00666"></a><span class="lineno"> 666</span> <span class="comment">// in the same style as binary and unary operator</span></div> |
| <div class="line"><a name="l00667"></a><span class="lineno"> 667</span> <span class="comment">// to use, simply write F<op::identity>( src )</span></div> |
| <div class="line"><a name="l00669"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1identity.html"> 669</a></span> <span class="comment"></span><span class="keyword">struct </span><a class="code" href="structmshadow_1_1op_1_1identity.html">identity</a>{</div> |
| <div class="line"><a name="l00671"></a><span class="lineno"> 671</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00672"></a><span class="lineno"><a class="line" href="structmshadow_1_1op_1_1identity.html#a260f8047dca47cb16ff0adaf16f5e8e6"> 672</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1op_1_1identity.html#a260f8047dca47cb16ff0adaf16f5e8e6">Map</a>(DType a) {</div> |
| <div class="line"><a name="l00673"></a><span class="lineno"> 673</span>  <span class="keywordflow">return</span> a;</div> |
| <div class="line"><a name="l00674"></a><span class="lineno"> 674</span>  }</div> |
| <div class="line"><a name="l00675"></a><span class="lineno"> 675</span> };</div> |
| <div class="line"><a name="l00676"></a><span class="lineno"> 676</span> } <span class="comment">// namespace op</span></div> |
| <div class="line"><a name="l00678"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1sv.html"> 678</a></span> <span class="comment"></span><span class="keyword">namespace </span>sv {</div> |
| <div class="line"><a name="l00680"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1saveto.html"> 680</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1sv_1_1saveto.html">saveto</a> {</div> |
| <div class="line"><a name="l00682"></a><span class="lineno"> 682</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00683"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1saveto.html#af99e9c2ec6d116febb41fc73d201d80e"> 683</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1sv_1_1saveto.html#af99e9c2ec6d116febb41fc73d201d80e">Save</a>(DType &a, DType b) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00684"></a><span class="lineno"> 684</span>  a = b;</div> |
| <div class="line"><a name="l00685"></a><span class="lineno"> 685</span>  }</div> |
| <div class="line"><a name="l00687"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1saveto.html#aaf49b727bb5fe9098081c265100cb340"> 687</a></span>  <span class="keyword">inline</span> <span class="keyword">static</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a> <a class="code" href="structmshadow_1_1sv_1_1saveto.html#aaf49b727bb5fe9098081c265100cb340">AlphaBLAS</a>(<span class="keywordtype">void</span>) { <span class="keywordflow">return</span> 1.0f; }</div> |
| <div class="line"><a name="l00689"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1saveto.html#a4dab404c405fe7476eadbd02e727738b"> 689</a></span>  <span class="keyword">inline</span> <span class="keyword">static</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a> <a class="code" href="structmshadow_1_1sv_1_1saveto.html#a4dab404c405fe7476eadbd02e727738b">BetaBLAS</a>(<span class="keywordtype">void</span>) { <span class="keywordflow">return</span> 0.0f; }</div> |
| <div class="line"><a name="l00691"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1saveto.html#ac78ceb69e8291f8fa767c2eda2a732a5"> 691</a></span>  <span class="keyword">typedef</span> <a class="code" href="structmshadow_1_1op_1_1right.html">op::right</a> <a class="code" href="structmshadow_1_1sv_1_1saveto.html#ac78ceb69e8291f8fa767c2eda2a732a5">OPType</a>;</div> |
| <div class="line"><a name="l00692"></a><span class="lineno"> 692</span> };</div> |
| <div class="line"><a name="l00694"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1plusto.html"> 694</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1sv_1_1plusto.html">plusto</a> {</div> |
| <div class="line"><a name="l00696"></a><span class="lineno"> 696</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00697"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1plusto.html#ac7556832c791c8d048cbb7fd606f6e8e"> 697</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1sv_1_1plusto.html#ac7556832c791c8d048cbb7fd606f6e8e">Save</a>(DType &a, DType b) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00698"></a><span class="lineno"> 698</span>  a += b;</div> |
| <div class="line"><a name="l00699"></a><span class="lineno"> 699</span>  }</div> |
| <div class="line"><a name="l00701"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1plusto.html#a9acc887f74cce8e18f1a86818154ca1c"> 701</a></span>  <span class="keyword">inline</span> <span class="keyword">static</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a> <a class="code" href="structmshadow_1_1sv_1_1plusto.html#a9acc887f74cce8e18f1a86818154ca1c">AlphaBLAS</a>(<span class="keywordtype">void</span>) { <span class="keywordflow">return</span> 1.0f; }</div> |
| <div class="line"><a name="l00703"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1plusto.html#a916ea217bd0200ab7b23a4e2ac26887f"> 703</a></span>  <span class="keyword">inline</span> <span class="keyword">static</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a> <a class="code" href="structmshadow_1_1sv_1_1plusto.html#a916ea217bd0200ab7b23a4e2ac26887f">BetaBLAS</a>(<span class="keywordtype">void</span>) { <span class="keywordflow">return</span> 1.0f; }</div> |
| <div class="line"><a name="l00705"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1plusto.html#a0b3849b290f3207a59b8975e2b29b065"> 705</a></span>  <span class="keyword">typedef</span> <a class="code" href="structmshadow_1_1op_1_1plus.html">op::plus</a> <a class="code" href="structmshadow_1_1sv_1_1plusto.html#a0b3849b290f3207a59b8975e2b29b065">OPType</a>;</div> |
| <div class="line"><a name="l00706"></a><span class="lineno"> 706</span> };</div> |
| <div class="line"><a name="l00708"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1minusto.html"> 708</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1sv_1_1minusto.html">minusto</a> {</div> |
| <div class="line"><a name="l00710"></a><span class="lineno"> 710</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00711"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1minusto.html#a50bab98db98ee42cd377f2e2f02e37ca"> 711</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1sv_1_1minusto.html#a50bab98db98ee42cd377f2e2f02e37ca">Save</a>(DType &a, DType b) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00712"></a><span class="lineno"> 712</span>  a -= b;</div> |
| <div class="line"><a name="l00713"></a><span class="lineno"> 713</span>  }</div> |
| <div class="line"><a name="l00715"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1minusto.html#a0da791365b9910dd337e228888521127"> 715</a></span>  <span class="keyword">inline</span> <span class="keyword">static</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a> <a class="code" href="structmshadow_1_1sv_1_1minusto.html#a0da791365b9910dd337e228888521127">AlphaBLAS</a>(<span class="keywordtype">void</span>) { <span class="keywordflow">return</span> -1.0f; }</div> |
| <div class="line"><a name="l00717"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1minusto.html#a48fae34ea7d461ced6ac9c98c5e38019"> 717</a></span>  <span class="keyword">inline</span> <span class="keyword">static</span> <a class="code" href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">default_real_t</a> <a class="code" href="structmshadow_1_1sv_1_1minusto.html#a48fae34ea7d461ced6ac9c98c5e38019">BetaBLAS</a>(<span class="keywordtype">void</span>) { <span class="keywordflow">return</span> 1.0f; }</div> |
| <div class="line"><a name="l00719"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1minusto.html#ac9baef2b9d3a92b962c9ad7f7ee1122d"> 719</a></span>  <span class="keyword">typedef</span> <a class="code" href="structmshadow_1_1op_1_1minus.html">op::minus</a> <a class="code" href="structmshadow_1_1sv_1_1minusto.html#ac9baef2b9d3a92b962c9ad7f7ee1122d">OPType</a>;</div> |
| <div class="line"><a name="l00720"></a><span class="lineno"> 720</span> };</div> |
| <div class="line"><a name="l00722"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1multo.html"> 722</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1sv_1_1multo.html">multo</a> {</div> |
| <div class="line"><a name="l00724"></a><span class="lineno"> 724</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00725"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1multo.html#ab3b6328d24ef3ce0cf5c0329e254f193"> 725</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1sv_1_1multo.html#ab3b6328d24ef3ce0cf5c0329e254f193">Save</a>(DType &a, DType b) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00726"></a><span class="lineno"> 726</span>  a *= b;</div> |
| <div class="line"><a name="l00727"></a><span class="lineno"> 727</span>  }</div> |
| <div class="line"><a name="l00729"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1multo.html#acc47e9c85d6e583e5785ba0f62fa4431"> 729</a></span>  <span class="keyword">typedef</span> <a class="code" href="structmshadow_1_1op_1_1mul.html">op::mul</a> <a class="code" href="structmshadow_1_1sv_1_1multo.html#acc47e9c85d6e583e5785ba0f62fa4431">OPType</a>;</div> |
| <div class="line"><a name="l00730"></a><span class="lineno"> 730</span> };</div> |
| <div class="line"><a name="l00732"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1divto.html"> 732</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1sv_1_1divto.html">divto</a> {</div> |
| <div class="line"><a name="l00734"></a><span class="lineno"> 734</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00735"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1divto.html#aaa2299b2a710d1b5d61b6dc6f44c9339"> 735</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1sv_1_1divto.html#aaa2299b2a710d1b5d61b6dc6f44c9339">Save</a>(DType& a, DType b) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00736"></a><span class="lineno"> 736</span>  a /= b;</div> |
| <div class="line"><a name="l00737"></a><span class="lineno"> 737</span>  }</div> |
| <div class="line"><a name="l00739"></a><span class="lineno"><a class="line" href="structmshadow_1_1sv_1_1divto.html#ab8864c99eea0286fce8f5a8f791f0f33"> 739</a></span>  <span class="keyword">typedef</span> <a class="code" href="structmshadow_1_1op_1_1div.html">op::div</a> <a class="code" href="structmshadow_1_1sv_1_1divto.html#ab8864c99eea0286fce8f5a8f791f0f33">OPType</a>;</div> |
| <div class="line"><a name="l00740"></a><span class="lineno"> 740</span> };</div> |
| <div class="line"><a name="l00741"></a><span class="lineno"> 741</span> } <span class="comment">// namespace sv</span></div> |
| <div class="line"><a name="l00742"></a><span class="lineno"> 742</span>  </div> |
| <div class="line"><a name="l00743"></a><span class="lineno"> 743</span> <span class="preprocessor">#ifndef __CUDA_ARCH__</span></div> |
| <div class="line"><a name="l00744"></a><span class="lineno"> 744</span> <span class="keyword">using</span> std::isnan;</div> |
| <div class="line"><a name="l00745"></a><span class="lineno"> 745</span> <span class="keyword">using</span> std::isinf;</div> |
| <div class="line"><a name="l00746"></a><span class="lineno"> 746</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l00747"></a><span class="lineno"> 747</span>  </div> |
| <div class="line"><a name="l00751"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html"> 751</a></span> <span class="keyword">namespace </span>isnan_typed {</div> |
| <div class="line"><a name="l00752"></a><span class="lineno"> 752</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00753"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759"> 753</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">IsNan</a>(<span class="keyword">volatile</span> DType val) {</div> |
| <div class="line"><a name="l00754"></a><span class="lineno"> 754</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00755"></a><span class="lineno"> 755</span>  }</div> |
| <div class="line"><a name="l00756"></a><span class="lineno"> 756</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00757"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html#ad4b4eb847832e78bfa17be14f68c43d9"> 757</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">IsNan</a>(<span class="keyword">volatile</span> <span class="keywordtype">float</span> val) {</div> |
| <div class="line"><a name="l00758"></a><span class="lineno"> 758</span>  <span class="keywordflow">return</span> isnan(val);</div> |
| <div class="line"><a name="l00759"></a><span class="lineno"> 759</span>  }</div> |
| <div class="line"><a name="l00760"></a><span class="lineno"> 760</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00761"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html#a85d9d2f79f833f934cd5f14821347c9d"> 761</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">IsNan</a>(<span class="keyword">volatile</span> <span class="keywordtype">double</span> val) {</div> |
| <div class="line"><a name="l00762"></a><span class="lineno"> 762</span>  <span class="keywordflow">return</span> isnan(val);</div> |
| <div class="line"><a name="l00763"></a><span class="lineno"> 763</span>  }</div> |
| <div class="line"><a name="l00764"></a><span class="lineno"> 764</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00765"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html#ac1bcb29820df6644b2a72b230969a9d1"> 765</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">IsNan</a>(<span class="keyword">volatile</span> <span class="keywordtype">long</span> <span class="keywordtype">double</span> val) {</div> |
| <div class="line"><a name="l00766"></a><span class="lineno"> 766</span>  <span class="keywordflow">return</span> isnan(val);</div> |
| <div class="line"><a name="l00767"></a><span class="lineno"> 767</span>  }</div> |
| <div class="line"><a name="l00768"></a><span class="lineno"> 768</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00769"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html#abae8abe8600d9f57faddd8e59ad9ed50"> 769</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">IsNan</a>(<span class="keyword">volatile</span> mshadow::half::half_t val) {</div> |
| <div class="line"><a name="l00770"></a><span class="lineno"> 770</span>  <span class="keywordflow">return</span> (val.half_ & (~<a class="code" href="half_8h.html#a4068a67a48bb294a6bf436e28be90de9">MSHADOW_HALF_SIGN_BIT</a>)) > <a class="code" href="half_8h.html#afdc51a9766b6923a3742828222103e3d">MSHADOW_HALF_EXPONENT_BITS</a>;</div> |
| <div class="line"><a name="l00771"></a><span class="lineno"> 771</span>  }</div> |
| <div class="line"><a name="l00772"></a><span class="lineno"> 772</span>  <span class="keyword">template</span> <></div> |
| <div class="line"><a name="l00773"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isnan__typed.html#a049f83082ac746a19d6f747dd6570bca"> 773</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">IsNan</a>(<span class="keyword">volatile</span> mshadow::bfloat::bf16_t val) {</div> |
| <div class="line"><a name="l00774"></a><span class="lineno"> 774</span>  <span class="keywordflow">return</span> (val.bf16_ & (~<a class="code" href="bfloat_8h.html#a9ec28a1a9aa829211bb9963eb967c245">MSHADOW_BF16_SIGN_BIT</a>)) > <a class="code" href="bfloat_8h.html#ad3e1c33244a89e2dd8a1f5697c10ff7b">MSHADOW_BF16_EXPONENT_BITS</a>;</div> |
| <div class="line"><a name="l00775"></a><span class="lineno"> 775</span>  }</div> |
| <div class="line"><a name="l00776"></a><span class="lineno"> 776</span> } <span class="comment">// namespace isnan_typed</span></div> |
| <div class="line"><a name="l00777"></a><span class="lineno"> 777</span>  </div> |
| <div class="line"><a name="l00781"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html"> 781</a></span> <span class="keyword">namespace </span>isinf_typed {</div> |
| <div class="line"><a name="l00782"></a><span class="lineno"> 782</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00783"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d"> 783</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">IsInf</a>(<span class="keyword">volatile</span> DType val) {</div> |
| <div class="line"><a name="l00784"></a><span class="lineno"> 784</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00785"></a><span class="lineno"> 785</span>  }</div> |
| <div class="line"><a name="l00786"></a><span class="lineno"> 786</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00787"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html#afb2055fccae27375062b8561a052eca1"> 787</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">IsInf</a>(<span class="keyword">volatile</span> <span class="keywordtype">float</span> val) {</div> |
| <div class="line"><a name="l00788"></a><span class="lineno"> 788</span>  <span class="keywordflow">return</span> isinf(val);</div> |
| <div class="line"><a name="l00789"></a><span class="lineno"> 789</span>  }</div> |
| <div class="line"><a name="l00790"></a><span class="lineno"> 790</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00791"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html#a5f845acb36e946a8b8451ee674a582b8"> 791</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">IsInf</a>(<span class="keyword">volatile</span> <span class="keywordtype">double</span> val) {</div> |
| <div class="line"><a name="l00792"></a><span class="lineno"> 792</span>  <span class="keywordflow">return</span> isinf(val);</div> |
| <div class="line"><a name="l00793"></a><span class="lineno"> 793</span>  }</div> |
| <div class="line"><a name="l00794"></a><span class="lineno"> 794</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00795"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html#a05067eb40dc864fa670aad5a8f843568"> 795</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">IsInf</a>(<span class="keyword">volatile</span> <span class="keywordtype">long</span> <span class="keywordtype">double</span> val) {</div> |
| <div class="line"><a name="l00796"></a><span class="lineno"> 796</span>  <span class="keywordflow">return</span> isinf(val);</div> |
| <div class="line"><a name="l00797"></a><span class="lineno"> 797</span>  }</div> |
| <div class="line"><a name="l00798"></a><span class="lineno"> 798</span>  <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00799"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html#a9300383296ccfacbfec50a6b8d33d557"> 799</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">IsInf</a>(<span class="keyword">volatile</span> mshadow::half::half_t val) {</div> |
| <div class="line"><a name="l00800"></a><span class="lineno"> 800</span>  <span class="keywordflow">return</span> (val.half_ & (~<a class="code" href="half_8h.html#a4068a67a48bb294a6bf436e28be90de9">MSHADOW_HALF_SIGN_BIT</a>)) == <a class="code" href="half_8h.html#afdc51a9766b6923a3742828222103e3d">MSHADOW_HALF_EXPONENT_BITS</a>;</div> |
| <div class="line"><a name="l00801"></a><span class="lineno"> 801</span>  }</div> |
| <div class="line"><a name="l00802"></a><span class="lineno"> 802</span>  <span class="keyword">template</span> <></div> |
| <div class="line"><a name="l00803"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1isinf__typed.html#ab7c52a222b457ce3d5027d8641d7a102"> 803</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">IsInf</a>(<span class="keyword">volatile</span> mshadow::bfloat::bf16_t val) {</div> |
| <div class="line"><a name="l00804"></a><span class="lineno"> 804</span>  <span class="keywordflow">return</span> (val.bf16_ & (~<a class="code" href="bfloat_8h.html#a9ec28a1a9aa829211bb9963eb967c245">MSHADOW_BF16_SIGN_BIT</a>)) == <a class="code" href="bfloat_8h.html#ad3e1c33244a89e2dd8a1f5697c10ff7b">MSHADOW_BF16_EXPONENT_BITS</a>;</div> |
| <div class="line"><a name="l00805"></a><span class="lineno"> 805</span>  }</div> |
| <div class="line"><a name="l00806"></a><span class="lineno"> 806</span> } <span class="comment">// namespace isinf_typed</span></div> |
| <div class="line"><a name="l00807"></a><span class="lineno"> 807</span>  </div> |
| <div class="line"><a name="l00809"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red.html"> 809</a></span> <span class="keyword">namespace </span>red {</div> |
| <div class="line"><a name="l00810"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html"> 810</a></span> <span class="keyword">namespace </span><a class="code" href="namespacemxnet_1_1common_1_1cuda_1_1rtc.html#a09828bc99a0f05955a61b9e4feabb9b8">limits</a> {</div> |
| <div class="line"><a name="l00815"></a><span class="lineno"> 815</span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00816"></a><span class="lineno"> 816</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> DType <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a5244fd39689a5526efa712f85579b148">MinValue</a>(<span class="keywordtype">void</span>);</div> |
| <div class="line"><a name="l00818"></a><span class="lineno"> 818</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00819"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a232631c75acb76794420d172a5cc335e"> 819</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">float</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a232631c75acb76794420d172a5cc335e">MinValue<float></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00820"></a><span class="lineno"> 820</span>  <span class="keywordflow">return</span> -FLT_MAX;</div> |
| <div class="line"><a name="l00821"></a><span class="lineno"> 821</span> }</div> |
| <div class="line"><a name="l00823"></a><span class="lineno"> 823</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00824"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a34f7a97ddbac1b447e39603de7c6f965"> 824</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">double</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a34f7a97ddbac1b447e39603de7c6f965">MinValue<double></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00825"></a><span class="lineno"> 825</span>  <span class="keywordflow">return</span> -DBL_MAX;</div> |
| <div class="line"><a name="l00826"></a><span class="lineno"> 826</span> }</div> |
| <div class="line"><a name="l00828"></a><span class="lineno"> 828</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00829"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a8642a9f6501c79c3a7ff8d8d473e5ec2"> 829</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> half::half_t MinValue<half::half_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00830"></a><span class="lineno"> 830</span>  <span class="keywordflow">return</span> <a class="code" href="half_8h.html#a6961c70512f3780000b1ee5691363da3">MSHADOW_HALF_MIN</a>;</div> |
| <div class="line"><a name="l00831"></a><span class="lineno"> 831</span> }</div> |
| <div class="line"><a name="l00833"></a><span class="lineno"> 833</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00834"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a1b4ec57952b90b03665ad88d5fa0639f"> 834</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> bfloat::bf16_t MinValue<bfloat::bf16_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00835"></a><span class="lineno"> 835</span>  <span class="keywordflow">return</span> <a class="code" href="bfloat_8h.html#ab31d5d1e2e21e0fc5dd6f40e276fdd43">MSHADOW_BF16_MIN</a>;</div> |
| <div class="line"><a name="l00836"></a><span class="lineno"> 836</span> }</div> |
| <div class="line"><a name="l00838"></a><span class="lineno"> 838</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00839"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a5ada516b5bd58791f170586cfdd9ae2f"> 839</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> uint8_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a5ada516b5bd58791f170586cfdd9ae2f">MinValue<uint8_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00840"></a><span class="lineno"> 840</span>  <span class="keywordflow">return</span> 0;</div> |
| <div class="line"><a name="l00841"></a><span class="lineno"> 841</span> }</div> |
| <div class="line"><a name="l00843"></a><span class="lineno"> 843</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00844"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a55d9bb33517b6c82d04555acba79680a"> 844</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> int8_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a55d9bb33517b6c82d04555acba79680a">MinValue<int8_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00845"></a><span class="lineno"> 845</span>  <span class="keywordflow">return</span> SCHAR_MIN;</div> |
| <div class="line"><a name="l00846"></a><span class="lineno"> 846</span> }</div> |
| <div class="line"><a name="l00848"></a><span class="lineno"> 848</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00849"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a1e417de4a028b2628d6bb5b6af1244dc"> 849</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a1e417de4a028b2628d6bb5b6af1244dc">MinValue<int32_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00850"></a><span class="lineno"> 850</span>  <span class="keywordflow">return</span> INT_MIN;</div> |
| <div class="line"><a name="l00851"></a><span class="lineno"> 851</span> }</div> |
| <div class="line"><a name="l00853"></a><span class="lineno"> 853</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00854"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a56182d9842b841740da7eefb60916cf9"> 854</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> int64_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a56182d9842b841740da7eefb60916cf9">MinValue<int64_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00855"></a><span class="lineno"> 855</span>  <span class="keywordflow">return</span> LLONG_MIN;</div> |
| <div class="line"><a name="l00856"></a><span class="lineno"> 856</span> }</div> |
| <div class="line"><a name="l00858"></a><span class="lineno"> 858</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00859"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a1adf2b0a158ad07a93b21f11e6aa3150"> 859</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a1adf2b0a158ad07a93b21f11e6aa3150">MinValue<bool></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00860"></a><span class="lineno"> 860</span>  <span class="keywordflow">return</span> <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00861"></a><span class="lineno"> 861</span> }</div> |
| <div class="line"><a name="l00863"></a><span class="lineno"> 863</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00864"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#ab06124f184da5683a517e02bcf2d6b9d"> 864</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#ab06124f184da5683a517e02bcf2d6b9d">MinValue<unsigned int></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00865"></a><span class="lineno"> 865</span>  <span class="keywordflow">return</span> 0;</div> |
| <div class="line"><a name="l00866"></a><span class="lineno"> 866</span> }</div> |
| <div class="line"><a name="l00867"></a><span class="lineno"> 867</span>  </div> |
| <div class="line"><a name="l00872"></a><span class="lineno"> 872</span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00873"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a6b0a333f168fa9d54d2f13caabd595e0"> 873</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> DType <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a6b0a333f168fa9d54d2f13caabd595e0">NegInfValue</a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00874"></a><span class="lineno"> 874</span>  <span class="keywordflow">return</span> MinValue<DType>();</div> |
| <div class="line"><a name="l00875"></a><span class="lineno"> 875</span> }</div> |
| <div class="line"><a name="l00877"></a><span class="lineno"> 877</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00878"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#af43c87655a8a1531462fa86bc7cc902e"> 878</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">float</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#af43c87655a8a1531462fa86bc7cc902e">NegInfValue<float></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00879"></a><span class="lineno"> 879</span>  <span class="keywordflow">return</span> -HUGE_VALF;</div> |
| <div class="line"><a name="l00880"></a><span class="lineno"> 880</span> }</div> |
| <div class="line"><a name="l00882"></a><span class="lineno"> 882</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00883"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a3784eade98f040fe074543259fe7164e"> 883</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">double</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a3784eade98f040fe074543259fe7164e">NegInfValue<double></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00884"></a><span class="lineno"> 884</span>  <span class="keywordflow">return</span> -HUGE_VAL;</div> |
| <div class="line"><a name="l00885"></a><span class="lineno"> 885</span> }</div> |
| <div class="line"><a name="l00887"></a><span class="lineno"> 887</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00888"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#ad1bc78ffe1a46fc55d3ab44c4f8648b4"> 888</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> half::half_t NegInfValue<half::half_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00889"></a><span class="lineno"> 889</span>  <span class="keywordflow">return</span> half::half_t::Binary(</div> |
| <div class="line"><a name="l00890"></a><span class="lineno"> 890</span>  <a class="code" href="half_8h.html#a4068a67a48bb294a6bf436e28be90de9">MSHADOW_HALF_SIGN_BIT</a> | <a class="code" href="half_8h.html#afdc51a9766b6923a3742828222103e3d">MSHADOW_HALF_EXPONENT_BITS</a>);</div> |
| <div class="line"><a name="l00891"></a><span class="lineno"> 891</span> }</div> |
| <div class="line"><a name="l00893"></a><span class="lineno"> 893</span> <span class="keyword">template</span> <></div> |
| <div class="line"><a name="l00894"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#ad064d26701f7c2e291bedb1a0c13d9d1"> 894</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> bfloat::bf16_t NegInfValue<bfloat::bf16_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00895"></a><span class="lineno"> 895</span>  <span class="keywordflow">return</span> bfloat::bf16_t::Binary(<a class="code" href="bfloat_8h.html#a9ec28a1a9aa829211bb9963eb967c245">MSHADOW_BF16_SIGN_BIT</a> | <a class="code" href="bfloat_8h.html#ad3e1c33244a89e2dd8a1f5697c10ff7b">MSHADOW_BF16_EXPONENT_BITS</a>);</div> |
| <div class="line"><a name="l00896"></a><span class="lineno"> 896</span> }</div> |
| <div class="line"><a name="l00897"></a><span class="lineno"> 897</span>  </div> |
| <div class="line"><a name="l00902"></a><span class="lineno"> 902</span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00903"></a><span class="lineno"> 903</span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> DType <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a993a69000bf5e0a5d13ef149a3311a78">MaxValue</a>(<span class="keywordtype">void</span>);</div> |
| <div class="line"><a name="l00905"></a><span class="lineno"> 905</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00906"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a6262f9d4a98d7ddc205cb0b19af4a316"> 906</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">float</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a6262f9d4a98d7ddc205cb0b19af4a316">MaxValue<float></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00907"></a><span class="lineno"> 907</span>  <span class="keywordflow">return</span> FLT_MAX;</div> |
| <div class="line"><a name="l00908"></a><span class="lineno"> 908</span> }</div> |
| <div class="line"><a name="l00910"></a><span class="lineno"> 910</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00911"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a4fdc4629369e455063f9d084570f276a"> 911</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">double</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a4fdc4629369e455063f9d084570f276a">MaxValue<double></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00912"></a><span class="lineno"> 912</span>  <span class="keywordflow">return</span> DBL_MAX;</div> |
| <div class="line"><a name="l00913"></a><span class="lineno"> 913</span> }</div> |
| <div class="line"><a name="l00915"></a><span class="lineno"> 915</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00916"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a045099a4f837139379f6d164bf663770"> 916</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> half::half_t MaxValue<half::half_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00917"></a><span class="lineno"> 917</span>  <span class="keywordflow">return</span> <a class="code" href="half_8h.html#acd26d7558448d4ce6889283913627b31">MSHADOW_HALF_MAX</a>;</div> |
| <div class="line"><a name="l00918"></a><span class="lineno"> 918</span> }</div> |
| <div class="line"><a name="l00920"></a><span class="lineno"> 920</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00921"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#ae8f74d17008d30efc4a2de04d0294502"> 921</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> bfloat::bf16_t MaxValue<bfloat::bf16_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00922"></a><span class="lineno"> 922</span>  <span class="keywordflow">return</span> <a class="code" href="bfloat_8h.html#a3d5d63ecb7104078d465a0074f9d9059">MSHADOW_BF16_MAX</a>;</div> |
| <div class="line"><a name="l00923"></a><span class="lineno"> 923</span> }</div> |
| <div class="line"><a name="l00925"></a><span class="lineno"> 925</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00926"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a853288dcb013f16211eb9ce107ccba3e"> 926</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> uint8_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a853288dcb013f16211eb9ce107ccba3e">MaxValue<uint8_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00927"></a><span class="lineno"> 927</span>  <span class="keywordflow">return</span> UCHAR_MAX;</div> |
| <div class="line"><a name="l00928"></a><span class="lineno"> 928</span> }</div> |
| <div class="line"><a name="l00930"></a><span class="lineno"> 930</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00931"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#affcf917b2d5206e3ddeec7e5f0ee995e"> 931</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> int8_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#affcf917b2d5206e3ddeec7e5f0ee995e">MaxValue<int8_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00932"></a><span class="lineno"> 932</span>  <span class="keywordflow">return</span> SCHAR_MAX;</div> |
| <div class="line"><a name="l00933"></a><span class="lineno"> 933</span> }</div> |
| <div class="line"><a name="l00935"></a><span class="lineno"> 935</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00936"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#ad5c922e935e2efb47bb238c56ff8a38b"> 936</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">int</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#ad5c922e935e2efb47bb238c56ff8a38b">MaxValue<int32_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00937"></a><span class="lineno"> 937</span>  <span class="keywordflow">return</span> INT_MAX;</div> |
| <div class="line"><a name="l00938"></a><span class="lineno"> 938</span> }</div> |
| <div class="line"><a name="l00940"></a><span class="lineno"> 940</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00941"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#ab4d32d18fb83ed9fe3b2773e0bf8d508"> 941</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> int64_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#ab4d32d18fb83ed9fe3b2773e0bf8d508">MaxValue<int64_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00942"></a><span class="lineno"> 942</span>  <span class="keywordflow">return</span> LLONG_MAX;</div> |
| <div class="line"><a name="l00943"></a><span class="lineno"> 943</span> }</div> |
| <div class="line"><a name="l00945"></a><span class="lineno"> 945</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00946"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#acc5e8630b2b3d865291c02b16c7b09c1"> 946</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">bool</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#acc5e8630b2b3d865291c02b16c7b09c1">MaxValue<bool></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00947"></a><span class="lineno"> 947</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00948"></a><span class="lineno"> 948</span> }</div> |
| <div class="line"><a name="l00950"></a><span class="lineno"> 950</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00951"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a78426e39f8f6170f709965da5aa2d118"> 951</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> uint32_t <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a78426e39f8f6170f709965da5aa2d118">MaxValue<uint32_t></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00952"></a><span class="lineno"> 952</span>  <span class="keywordflow">return</span> std::numeric_limits<uint32_t>::max();</div> |
| <div class="line"><a name="l00953"></a><span class="lineno"> 953</span> }</div> |
| <div class="line"><a name="l00954"></a><span class="lineno"> 954</span>  </div> |
| <div class="line"><a name="l00959"></a><span class="lineno"> 959</span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00960"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a64ef2d1a92579d312b18fb04aaed0d4e"> 960</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> DType <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a64ef2d1a92579d312b18fb04aaed0d4e">PosInfValue</a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00961"></a><span class="lineno"> 961</span>  <span class="keywordflow">return</span> MaxValue<DType>();</div> |
| <div class="line"><a name="l00962"></a><span class="lineno"> 962</span> }</div> |
| <div class="line"><a name="l00964"></a><span class="lineno"> 964</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00965"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a77b7882d6cafca4b12908e2fff2302ed"> 965</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">float</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a77b7882d6cafca4b12908e2fff2302ed">PosInfValue<float></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00966"></a><span class="lineno"> 966</span>  <span class="keywordflow">return</span> HUGE_VALF;</div> |
| <div class="line"><a name="l00967"></a><span class="lineno"> 967</span> }</div> |
| <div class="line"><a name="l00969"></a><span class="lineno"> 969</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00970"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a194ad52f15262c4acde6f4720f039858"> 970</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keywordtype">double</span> <a class="code" href="namespacemshadow_1_1red_1_1limits.html#a194ad52f15262c4acde6f4720f039858">PosInfValue<double></a>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00971"></a><span class="lineno"> 971</span>  <span class="keywordflow">return</span> HUGE_VAL;</div> |
| <div class="line"><a name="l00972"></a><span class="lineno"> 972</span> }</div> |
| <div class="line"><a name="l00974"></a><span class="lineno"> 974</span> <span class="keyword">template</span><></div> |
| <div class="line"><a name="l00975"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#a3ff7cabb9493716594e29aff48bcc60b"> 975</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> half::half_t PosInfValue<half::half_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00976"></a><span class="lineno"> 976</span>  <span class="keywordflow">return</span> half::half_t::Binary(<a class="code" href="half_8h.html#afdc51a9766b6923a3742828222103e3d">MSHADOW_HALF_EXPONENT_BITS</a>);</div> |
| <div class="line"><a name="l00977"></a><span class="lineno"> 977</span> }</div> |
| <div class="line"><a name="l00979"></a><span class="lineno"> 979</span> <span class="keyword">template</span> <></div> |
| <div class="line"><a name="l00980"></a><span class="lineno"><a class="line" href="namespacemshadow_1_1red_1_1limits.html#aa811df7b146909bc9d57a7c03e0356ea"> 980</a></span> <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> bfloat::bf16_t PosInfValue<bfloat::bf16_t>(<span class="keywordtype">void</span>) {</div> |
| <div class="line"><a name="l00981"></a><span class="lineno"> 981</span>  <span class="keywordflow">return</span> bfloat::bf16_t::Binary(<a class="code" href="bfloat_8h.html#ad3e1c33244a89e2dd8a1f5697c10ff7b">MSHADOW_BF16_EXPONENT_BITS</a>);</div> |
| <div class="line"><a name="l00982"></a><span class="lineno"> 982</span> }</div> |
| <div class="line"><a name="l00983"></a><span class="lineno"> 983</span>  </div> |
| <div class="line"><a name="l00984"></a><span class="lineno"> 984</span> } <span class="comment">// namespace limits</span></div> |
| <div class="line"><a name="l00985"></a><span class="lineno"> 985</span>  </div> |
| <div class="line"><a name="l00987"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html"> 987</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1red_1_1sum.html">sum</a> {</div> |
| <div class="line"><a name="l00989"></a><span class="lineno"> 989</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00990"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#abd56e307635333d02fc0f22726dbed28"> 990</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#abd56e307635333d02fc0f22726dbed28">Reduce</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType src) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00991"></a><span class="lineno"> 991</span>  dst += src;</div> |
| <div class="line"><a name="l00992"></a><span class="lineno"> 992</span>  }</div> |
| <div class="line"><a name="l00994"></a><span class="lineno"> 994</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00995"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#a36583c3bb79bbfe107b7c9480e80d5e7"> 995</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#a36583c3bb79bbfe107b7c9480e80d5e7">Reduce</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType src, <span class="keyword">volatile</span> DType& residual) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00996"></a><span class="lineno"> 996</span>  DType y = src - residual;</div> |
| <div class="line"><a name="l00997"></a><span class="lineno"> 997</span>  DType t = dst + y;</div> |
| <div class="line"><a name="l00998"></a><span class="lineno"> 998</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">isinf_typed::IsInf</a>(t)) {</div> |
| <div class="line"><a name="l00999"></a><span class="lineno"> 999</span>  residual = 0;</div> |
| <div class="line"><a name="l01000"></a><span class="lineno"> 1000</span>  } <span class="keywordflow">else</span> {</div> |
| <div class="line"><a name="l01001"></a><span class="lineno"> 1001</span>  residual = (t - dst) - y;</div> |
| <div class="line"><a name="l01002"></a><span class="lineno"> 1002</span>  }</div> |
| <div class="line"><a name="l01003"></a><span class="lineno"> 1003</span>  dst = t;</div> |
| <div class="line"><a name="l01004"></a><span class="lineno"> 1004</span>  }</div> |
| <div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01007"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#aa05b8713dd5973cac465420e52f67dda"> 1007</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#aa05b8713dd5973cac465420e52f67dda">Merge</a>(<span class="keyword">volatile</span> DType& dst_val, <span class="keyword">volatile</span> DType& src_val) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01008"></a><span class="lineno"> 1008</span>  <a class="code" href="structmshadow_1_1red_1_1sum.html#abd56e307635333d02fc0f22726dbed28">Reduce</a>(dst_val, src_val);</div> |
| <div class="line"><a name="l01009"></a><span class="lineno"> 1009</span>  }</div> |
| <div class="line"><a name="l01011"></a><span class="lineno"> 1011</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01012"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#abdbdd852a7eb8460458f24fea8ca7714"> 1012</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#abdbdd852a7eb8460458f24fea8ca7714">Merge</a>(<span class="keyword">volatile</span> DType& dst_val, <span class="keyword">volatile</span> DType& dst_residual, <span class="keyword">volatile</span> DType& src_val, <span class="keyword">volatile</span> DType& src_residual) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01013"></a><span class="lineno"> 1013</span>  DType t1 = dst_val + src_val;</div> |
| <div class="line"><a name="l01014"></a><span class="lineno"> 1014</span>  <span class="keywordflow">if</span> (<a class="code" href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">isinf_typed::IsInf</a>(t1)) {</div> |
| <div class="line"><a name="l01015"></a><span class="lineno"> 1015</span>  dst_val = t1;</div> |
| <div class="line"><a name="l01016"></a><span class="lineno"> 1016</span>  dst_residual = 0;</div> |
| <div class="line"><a name="l01017"></a><span class="lineno"> 1017</span>  } <span class="keywordflow">else</span> {</div> |
| <div class="line"><a name="l01018"></a><span class="lineno"> 1018</span>  DType e = t1 - dst_val;</div> |
| <div class="line"><a name="l01019"></a><span class="lineno"> 1019</span>  DType t2 = ((src_val - e) + (dst_val - (t1 - e))) + dst_residual + src_residual;</div> |
| <div class="line"><a name="l01020"></a><span class="lineno"> 1020</span>  dst_val = t1 + t2;</div> |
| <div class="line"><a name="l01021"></a><span class="lineno"> 1021</span>  dst_residual = t2 - (dst_val - t1);</div> |
| <div class="line"><a name="l01022"></a><span class="lineno"> 1022</span>  }</div> |
| <div class="line"><a name="l01023"></a><span class="lineno"> 1023</span>  }</div> |
| <div class="line"><a name="l01025"></a><span class="lineno"> 1025</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01026"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#a511c1d4add93f26810e6a53c0ff106d0"> 1026</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#a511c1d4add93f26810e6a53c0ff106d0">Finalize</a>(<span class="keyword">volatile</span> DType& dst) {} <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01028"></a><span class="lineno"> 1028</span> <span class="comment"></span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01029"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#af024e58661dffc3043fdab0ac6424659"> 1029</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#af024e58661dffc3043fdab0ac6424659">Finalize</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType& residual) {} <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01034"></a><span class="lineno"> 1034</span> <span class="comment"></span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01035"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#a351d0e761cb9944c4d98919535754142"> 1035</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1red_1_1sum.html#a351d0e761cb9944c4d98919535754142">PartialGrad</a>(DType redres, DType redsrc) {</div> |
| <div class="line"><a name="l01036"></a><span class="lineno"> 1036</span>  <span class="keywordflow">return</span> 1;</div> |
| <div class="line"><a name="l01037"></a><span class="lineno"> 1037</span>  }</div> |
| <div class="line"><a name="l01041"></a><span class="lineno"> 1041</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01042"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#aedc5b032275b8d11fc95fae2545cc2cb"> 1042</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#aedc5b032275b8d11fc95fae2545cc2cb">SetInitValue</a>(DType &initv) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01043"></a><span class="lineno"> 1043</span>  initv = 0;</div> |
| <div class="line"><a name="l01044"></a><span class="lineno"> 1044</span>  }</div> |
| <div class="line"><a name="l01048"></a><span class="lineno"> 1048</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01049"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1sum.html#a351e35f2cbf0f8e7e08c5f23b70f3c3f"> 1049</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1sum.html#a351e35f2cbf0f8e7e08c5f23b70f3c3f">SetInitValue</a>(DType &initv, DType &residual) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01050"></a><span class="lineno"> 1050</span>  <a class="code" href="structmshadow_1_1red_1_1sum.html#aedc5b032275b8d11fc95fae2545cc2cb">SetInitValue</a>(initv);</div> |
| <div class="line"><a name="l01051"></a><span class="lineno"> 1051</span>  residual = 0;</div> |
| <div class="line"><a name="l01052"></a><span class="lineno"> 1052</span>  }</div> |
| <div class="line"><a name="l01053"></a><span class="lineno"> 1053</span> };</div> |
| <div class="line"><a name="l01055"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html"> 1055</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1red_1_1maximum.html">maximum</a> {</div> |
| <div class="line"><a name="l01057"></a><span class="lineno"> 1057</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01058"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a99b1074ecd03f849cbdcfbad8df8ed2b"> 1058</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a99b1074ecd03f849cbdcfbad8df8ed2b">Reduce</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType src) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01059"></a><span class="lineno"> 1059</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">isnan_typed::IsNan</a>(dst)) {</div> |
| <div class="line"><a name="l01060"></a><span class="lineno"> 1060</span>  <span class="keywordflow">if</span> (!(dst >= src)) dst = src;</div> |
| <div class="line"><a name="l01061"></a><span class="lineno"> 1061</span>  }</div> |
| <div class="line"><a name="l01062"></a><span class="lineno"> 1062</span>  }</div> |
| <div class="line"><a name="l01064"></a><span class="lineno"> 1064</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01065"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a96aaae5db92ec7a3a8a5740265394a35"> 1065</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a96aaae5db92ec7a3a8a5740265394a35">Reduce</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType src, <span class="keyword">volatile</span> DType &none) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01066"></a><span class="lineno"> 1066</span>  <a class="code" href="structmshadow_1_1red_1_1maximum.html#a99b1074ecd03f849cbdcfbad8df8ed2b">Reduce</a>(dst, src);</div> |
| <div class="line"><a name="l01067"></a><span class="lineno"> 1067</span>  }</div> |
| <div class="line"><a name="l01069"></a><span class="lineno"> 1069</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01070"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a5c74a9cf04436607f101428cb8fd4326"> 1070</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a5c74a9cf04436607f101428cb8fd4326">Merge</a>(<span class="keyword">volatile</span> DType& dst_val, <span class="keyword">volatile</span> DType& src_val) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01071"></a><span class="lineno"> 1071</span>  <a class="code" href="structmshadow_1_1red_1_1maximum.html#a99b1074ecd03f849cbdcfbad8df8ed2b">Reduce</a>(dst_val, src_val);</div> |
| <div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>  }</div> |
| <div class="line"><a name="l01074"></a><span class="lineno"> 1074</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01075"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a88b427b9289c2c00736887449ed67d0a"> 1075</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a88b427b9289c2c00736887449ed67d0a">Merge</a>(<span class="keyword">volatile</span> DType& dst_val, <span class="keyword">volatile</span> DType& dst_residual, <span class="keyword">volatile</span> DType& src_val, <span class="keyword">volatile</span> DType& src_residual) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01076"></a><span class="lineno"> 1076</span>  <a class="code" href="structmshadow_1_1red_1_1maximum.html#a99b1074ecd03f849cbdcfbad8df8ed2b">Reduce</a>(dst_val, src_val);</div> |
| <div class="line"><a name="l01077"></a><span class="lineno"> 1077</span>  }</div> |
| <div class="line"><a name="l01079"></a><span class="lineno"> 1079</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01080"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a97c5d96eac2849f458678742738f1ce0"> 1080</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a97c5d96eac2849f458678742738f1ce0">Finalize</a>(<span class="keyword">volatile</span> DType& dst) {} <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01082"></a><span class="lineno"> 1082</span> <span class="comment"></span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01083"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a99c8021ef1da714c052bfdf1f4e41280"> 1083</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a99c8021ef1da714c052bfdf1f4e41280">Finalize</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType& residual) {} <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01088"></a><span class="lineno"> 1088</span> <span class="comment"></span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01089"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#ac2d455ca3b4eb12554eb31ed6a1387b7"> 1089</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1red_1_1maximum.html#ac2d455ca3b4eb12554eb31ed6a1387b7">PartialGrad</a>(DType redres, DType redsrc) {</div> |
| <div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>  <span class="keywordflow">return</span> redres == redsrc ? 1: 0;</div> |
| <div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>  }</div> |
| <div class="line"><a name="l01095"></a><span class="lineno"> 1095</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01096"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a1bffb78fd1c64479b82e9a4ffccab69e"> 1096</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a1bffb78fd1c64479b82e9a4ffccab69e">SetInitValue</a>(DType &initv) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01097"></a><span class="lineno"> 1097</span>  initv = limits::NegInfValue<DType>();</div> |
| <div class="line"><a name="l01098"></a><span class="lineno"> 1098</span>  }</div> |
| <div class="line"><a name="l01102"></a><span class="lineno"> 1102</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01103"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1maximum.html#a6f5613dfb9c095420b5fe322aa2ab1ed"> 1103</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1maximum.html#a6f5613dfb9c095420b5fe322aa2ab1ed">SetInitValue</a>(DType &initv, DType &none) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01104"></a><span class="lineno"> 1104</span>  <a class="code" href="structmshadow_1_1red_1_1maximum.html#a1bffb78fd1c64479b82e9a4ffccab69e">SetInitValue</a>(initv);</div> |
| <div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>  }</div> |
| <div class="line"><a name="l01106"></a><span class="lineno"> 1106</span> };</div> |
| <div class="line"><a name="l01108"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html"> 1108</a></span> <span class="keyword">struct </span><a class="code" href="structmshadow_1_1red_1_1minimum.html">minimum</a> {</div> |
| <div class="line"><a name="l01110"></a><span class="lineno"> 1110</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01111"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#aa000adb90fad1fb429a05b1c6c8b836b"> 1111</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#aa000adb90fad1fb429a05b1c6c8b836b">Reduce</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType src) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01112"></a><span class="lineno"> 1112</span>  <span class="keywordflow">if</span> (!<a class="code" href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">isnan_typed::IsNan</a>(dst)) {</div> |
| <div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>  <span class="keywordflow">if</span> (!(dst <= src)) dst = src;</div> |
| <div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>  }</div> |
| <div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>  }</div> |
| <div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01118"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#a01e4946a88167d905c14b39d3965e03c"> 1118</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#a01e4946a88167d905c14b39d3965e03c">Reduce</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType src, <span class="keyword">volatile</span> DType &none) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01119"></a><span class="lineno"> 1119</span>  <a class="code" href="structmshadow_1_1red_1_1minimum.html#aa000adb90fad1fb429a05b1c6c8b836b">Reduce</a>(dst, src);</div> |
| <div class="line"><a name="l01120"></a><span class="lineno"> 1120</span>  }</div> |
| <div class="line"><a name="l01122"></a><span class="lineno"> 1122</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01123"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#a9485035d020c53f31eadf7e34adc37f0"> 1123</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#a9485035d020c53f31eadf7e34adc37f0">Merge</a>(<span class="keyword">volatile</span> DType& dst_val, <span class="keyword">volatile</span> DType& src_val) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01124"></a><span class="lineno"> 1124</span>  <a class="code" href="structmshadow_1_1red_1_1minimum.html#aa000adb90fad1fb429a05b1c6c8b836b">Reduce</a>(dst_val, src_val);</div> |
| <div class="line"><a name="l01125"></a><span class="lineno"> 1125</span>  }</div> |
| <div class="line"><a name="l01127"></a><span class="lineno"> 1127</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01128"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#af14bfabf30b3f4bffdcd0827679d0390"> 1128</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#af14bfabf30b3f4bffdcd0827679d0390">Merge</a>(<span class="keyword">volatile</span> DType& dst_val, <span class="keyword">volatile</span> DType& dst_residual, <span class="keyword">volatile</span> DType& src_val, <span class="keyword">volatile</span> DType& src_residual) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01129"></a><span class="lineno"> 1129</span>  <a class="code" href="structmshadow_1_1red_1_1minimum.html#aa000adb90fad1fb429a05b1c6c8b836b">Reduce</a>(dst_val, src_val);</div> |
| <div class="line"><a name="l01130"></a><span class="lineno"> 1130</span>  }</div> |
| <div class="line"><a name="l01132"></a><span class="lineno"> 1132</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01133"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#aebd011f117a63fea3cd494ee2138b634"> 1133</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#aebd011f117a63fea3cd494ee2138b634">Finalize</a>(<span class="keyword">volatile</span> DType& dst) {} <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01135"></a><span class="lineno"> 1135</span> <span class="comment"></span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01136"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#ae844649225377caf827a51ef0d080515"> 1136</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#ae844649225377caf827a51ef0d080515">Finalize</a>(<span class="keyword">volatile</span> DType& dst, <span class="keyword">volatile</span> DType& residual) {} <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01141"></a><span class="lineno"> 1141</span> <span class="comment"></span> <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01142"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#aa3f7b2a1f96f77caca258e8c594f67bd"> 1142</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> DType <a class="code" href="structmshadow_1_1red_1_1minimum.html#aa3f7b2a1f96f77caca258e8c594f67bd">PartialGrad</a>(DType redres, DType redsrc) {</div> |
| <div class="line"><a name="l01143"></a><span class="lineno"> 1143</span>  <span class="keywordflow">return</span> redres == redsrc ? 1: 0;</div> |
| <div class="line"><a name="l01144"></a><span class="lineno"> 1144</span>  }</div> |
| <div class="line"><a name="l01148"></a><span class="lineno"> 1148</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01149"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#acbda7fe80fa3754a57b967e11a8ee44f"> 1149</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#acbda7fe80fa3754a57b967e11a8ee44f">SetInitValue</a>(DType &initv) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01150"></a><span class="lineno"> 1150</span>  initv = limits::PosInfValue<DType>();</div> |
| <div class="line"><a name="l01151"></a><span class="lineno"> 1151</span>  }</div> |
| <div class="line"><a name="l01155"></a><span class="lineno"> 1155</span>  <span class="keyword">template</span><<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l01156"></a><span class="lineno"><a class="line" href="structmshadow_1_1red_1_1minimum.html#aba3be24837548e8f5a48178889ec2132"> 1156</a></span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a> <span class="keyword">static</span> <span class="keywordtype">void</span> <a class="code" href="structmshadow_1_1red_1_1minimum.html#aba3be24837548e8f5a48178889ec2132">SetInitValue</a>(DType &initv, DType &none) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l01157"></a><span class="lineno"> 1157</span>  <a class="code" href="structmshadow_1_1red_1_1minimum.html#acbda7fe80fa3754a57b967e11a8ee44f">SetInitValue</a>(initv);</div> |
| <div class="line"><a name="l01158"></a><span class="lineno"> 1158</span>  }</div> |
| <div class="line"><a name="l01159"></a><span class="lineno"> 1159</span> };</div> |
| <div class="line"><a name="l01160"></a><span class="lineno"> 1160</span> } <span class="comment">// namespace red</span></div> |
| <div class="line"><a name="l01161"></a><span class="lineno"> 1161</span>  </div> |
| <div class="line"><a name="l01162"></a><span class="lineno"> 1162</span> <span class="preprocessor">#ifndef __NVCC__</span></div> |
| <div class="line"><a name="l01163"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#af67c41c06afea92ccf285ae8501e2316"> 1163</a></span> <span class="preprocessor">#define MSHADOW_TYPE_SWITCH(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01164"></a><span class="lineno"> 1164</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01165"></a><span class="lineno"> 1165</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01166"></a><span class="lineno"> 1166</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01167"></a><span class="lineno"> 1167</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01168"></a><span class="lineno"> 1168</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01169"></a><span class="lineno"> 1169</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01170"></a><span class="lineno"> 1170</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01171"></a><span class="lineno"> 1171</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01172"></a><span class="lineno"> 1172</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01173"></a><span class="lineno"> 1173</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01174"></a><span class="lineno"> 1174</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01175"></a><span class="lineno"> 1175</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01176"></a><span class="lineno"> 1176</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01177"></a><span class="lineno"> 1177</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01178"></a><span class="lineno"> 1178</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01179"></a><span class="lineno"> 1179</span> <span class="preprocessor"> typedef mshadow::half::half_t DType; \</span></div> |
| <div class="line"><a name="l01180"></a><span class="lineno"> 1180</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01181"></a><span class="lineno"> 1181</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01182"></a><span class="lineno"> 1182</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01183"></a><span class="lineno"> 1183</span> <span class="preprocessor"> case mshadow::kBfloat16: \</span></div> |
| <div class="line"><a name="l01184"></a><span class="lineno"> 1184</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01185"></a><span class="lineno"> 1185</span> <span class="preprocessor"> typedef mshadow::bfloat::bf16_t DType; \</span></div> |
| <div class="line"><a name="l01186"></a><span class="lineno"> 1186</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01187"></a><span class="lineno"> 1187</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01188"></a><span class="lineno"> 1188</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01189"></a><span class="lineno"> 1189</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01190"></a><span class="lineno"> 1190</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01191"></a><span class="lineno"> 1191</span> <span class="preprocessor"> typedef uint8_t DType; \</span></div> |
| <div class="line"><a name="l01192"></a><span class="lineno"> 1192</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01193"></a><span class="lineno"> 1193</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01194"></a><span class="lineno"> 1194</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01195"></a><span class="lineno"> 1195</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01196"></a><span class="lineno"> 1196</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01197"></a><span class="lineno"> 1197</span> <span class="preprocessor"> typedef int8_t DType; \</span></div> |
| <div class="line"><a name="l01198"></a><span class="lineno"> 1198</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01199"></a><span class="lineno"> 1199</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01200"></a><span class="lineno"> 1200</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01201"></a><span class="lineno"> 1201</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01202"></a><span class="lineno"> 1202</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01203"></a><span class="lineno"> 1203</span> <span class="preprocessor"> typedef int32_t DType; \</span></div> |
| <div class="line"><a name="l01204"></a><span class="lineno"> 1204</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01205"></a><span class="lineno"> 1205</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01206"></a><span class="lineno"> 1206</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01207"></a><span class="lineno"> 1207</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01208"></a><span class="lineno"> 1208</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01209"></a><span class="lineno"> 1209</span> <span class="preprocessor"> typedef int64_t DType; \</span></div> |
| <div class="line"><a name="l01210"></a><span class="lineno"> 1210</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01211"></a><span class="lineno"> 1211</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01212"></a><span class="lineno"> 1212</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01213"></a><span class="lineno"> 1213</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01214"></a><span class="lineno"> 1214</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01215"></a><span class="lineno"> 1215</span> <span class="preprocessor"> "support bool type"; \</span></div> |
| <div class="line"><a name="l01216"></a><span class="lineno"> 1216</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01217"></a><span class="lineno"> 1217</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01218"></a><span class="lineno"> 1218</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01219"></a><span class="lineno"> 1219</span> <span class="preprocessor"> "support int16 type"; \</span></div> |
| <div class="line"><a name="l01220"></a><span class="lineno"> 1220</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01221"></a><span class="lineno"> 1221</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01222"></a><span class="lineno"> 1222</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01223"></a><span class="lineno"> 1223</span> <span class="preprocessor"> "support uint16 type"; \</span></div> |
| <div class="line"><a name="l01224"></a><span class="lineno"> 1224</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01225"></a><span class="lineno"> 1225</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01226"></a><span class="lineno"> 1226</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01227"></a><span class="lineno"> 1227</span> <span class="preprocessor"> "support uint32 type"; \</span></div> |
| <div class="line"><a name="l01228"></a><span class="lineno"> 1228</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01229"></a><span class="lineno"> 1229</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01230"></a><span class="lineno"> 1230</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01231"></a><span class="lineno"> 1231</span> <span class="preprocessor"> "support uint64 type"; \</span></div> |
| <div class="line"><a name="l01232"></a><span class="lineno"> 1232</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01233"></a><span class="lineno"> 1233</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01234"></a><span class="lineno"> 1234</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01235"></a><span class="lineno"> 1235</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01236"></a><span class="lineno"> 1236</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l01237"></a><span class="lineno"> 1237</span> <span class="preprocessor">#define MSHADOW_TYPE_SWITCH(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01238"></a><span class="lineno"> 1238</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01239"></a><span class="lineno"> 1239</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01240"></a><span class="lineno"> 1240</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01241"></a><span class="lineno"> 1241</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01242"></a><span class="lineno"> 1242</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01243"></a><span class="lineno"> 1243</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01244"></a><span class="lineno"> 1244</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01245"></a><span class="lineno"> 1245</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01246"></a><span class="lineno"> 1246</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01247"></a><span class="lineno"> 1247</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01248"></a><span class="lineno"> 1248</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01249"></a><span class="lineno"> 1249</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01250"></a><span class="lineno"> 1250</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01251"></a><span class="lineno"> 1251</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01252"></a><span class="lineno"> 1252</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01253"></a><span class="lineno"> 1253</span> <span class="preprocessor"> typedef mshadow::half::half_t DType; \</span></div> |
| <div class="line"><a name="l01254"></a><span class="lineno"> 1254</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01255"></a><span class="lineno"> 1255</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01256"></a><span class="lineno"> 1256</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01257"></a><span class="lineno"> 1257</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01258"></a><span class="lineno"> 1258</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01259"></a><span class="lineno"> 1259</span> <span class="preprocessor"> typedef uint8_t DType; \</span></div> |
| <div class="line"><a name="l01260"></a><span class="lineno"> 1260</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01261"></a><span class="lineno"> 1261</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01262"></a><span class="lineno"> 1262</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01263"></a><span class="lineno"> 1263</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01264"></a><span class="lineno"> 1264</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01265"></a><span class="lineno"> 1265</span> <span class="preprocessor"> typedef int8_t DType; \</span></div> |
| <div class="line"><a name="l01266"></a><span class="lineno"> 1266</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01267"></a><span class="lineno"> 1267</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01268"></a><span class="lineno"> 1268</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01269"></a><span class="lineno"> 1269</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01270"></a><span class="lineno"> 1270</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01271"></a><span class="lineno"> 1271</span> <span class="preprocessor"> typedef int32_t DType; \</span></div> |
| <div class="line"><a name="l01272"></a><span class="lineno"> 1272</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01273"></a><span class="lineno"> 1273</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01274"></a><span class="lineno"> 1274</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01275"></a><span class="lineno"> 1275</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01276"></a><span class="lineno"> 1276</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01277"></a><span class="lineno"> 1277</span> <span class="preprocessor"> typedef int64_t DType; \</span></div> |
| <div class="line"><a name="l01278"></a><span class="lineno"> 1278</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01279"></a><span class="lineno"> 1279</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01280"></a><span class="lineno"> 1280</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01281"></a><span class="lineno"> 1281</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01282"></a><span class="lineno"> 1282</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01283"></a><span class="lineno"> 1283</span> <span class="preprocessor"> "support bool type"; \</span></div> |
| <div class="line"><a name="l01284"></a><span class="lineno"> 1284</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01285"></a><span class="lineno"> 1285</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01286"></a><span class="lineno"> 1286</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01287"></a><span class="lineno"> 1287</span> <span class="preprocessor"> "support int16 type"; \</span></div> |
| <div class="line"><a name="l01288"></a><span class="lineno"> 1288</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01289"></a><span class="lineno"> 1289</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01290"></a><span class="lineno"> 1290</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01291"></a><span class="lineno"> 1291</span> <span class="preprocessor"> "support uint16 type"; \</span></div> |
| <div class="line"><a name="l01292"></a><span class="lineno"> 1292</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01293"></a><span class="lineno"> 1293</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01294"></a><span class="lineno"> 1294</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01295"></a><span class="lineno"> 1295</span> <span class="preprocessor"> "support uint32 type"; \</span></div> |
| <div class="line"><a name="l01296"></a><span class="lineno"> 1296</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01297"></a><span class="lineno"> 1297</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01298"></a><span class="lineno"> 1298</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01299"></a><span class="lineno"> 1299</span> <span class="preprocessor"> "support uint64 type"; \</span></div> |
| <div class="line"><a name="l01300"></a><span class="lineno"> 1300</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01301"></a><span class="lineno"> 1301</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01302"></a><span class="lineno"> 1302</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01303"></a><span class="lineno"> 1303</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01304"></a><span class="lineno"> 1304</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l01305"></a><span class="lineno"> 1305</span>  </div> |
| <div class="line"><a name="l01306"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a3654cd812cf6a0011b78ef28c52e2f88"> 1306</a></span> <span class="preprocessor">#define MSHADOW_SGL_DBL_TYPE_SWITCH(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01307"></a><span class="lineno"> 1307</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01308"></a><span class="lineno"> 1308</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01309"></a><span class="lineno"> 1309</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01310"></a><span class="lineno"> 1310</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01311"></a><span class="lineno"> 1311</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01312"></a><span class="lineno"> 1312</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01313"></a><span class="lineno"> 1313</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01314"></a><span class="lineno"> 1314</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01315"></a><span class="lineno"> 1315</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01316"></a><span class="lineno"> 1316</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01317"></a><span class="lineno"> 1317</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01318"></a><span class="lineno"> 1318</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01319"></a><span class="lineno"> 1319</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01320"></a><span class="lineno"> 1320</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01321"></a><span class="lineno"> 1321</span> <span class="preprocessor"> LOG(FATAL) << "This operation only supports " \</span></div> |
| <div class="line"><a name="l01322"></a><span class="lineno"> 1322</span> <span class="preprocessor"> "32-bit and 64-bit floating point"; \</span></div> |
| <div class="line"><a name="l01323"></a><span class="lineno"> 1323</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01324"></a><span class="lineno"> 1324</span>  </div> |
| <div class="line"><a name="l01325"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a07e61e376e92683d0266b7b2cfdaaf6b"> 1325</a></span> <span class="preprocessor">#define MSHADOW_REAL_TYPE_SWITCH(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01326"></a><span class="lineno"> 1326</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01327"></a><span class="lineno"> 1327</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01328"></a><span class="lineno"> 1328</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01329"></a><span class="lineno"> 1329</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01330"></a><span class="lineno"> 1330</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01331"></a><span class="lineno"> 1331</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01332"></a><span class="lineno"> 1332</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01333"></a><span class="lineno"> 1333</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01334"></a><span class="lineno"> 1334</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01335"></a><span class="lineno"> 1335</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01336"></a><span class="lineno"> 1336</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01337"></a><span class="lineno"> 1337</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01338"></a><span class="lineno"> 1338</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01339"></a><span class="lineno"> 1339</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01340"></a><span class="lineno"> 1340</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01341"></a><span class="lineno"> 1341</span> <span class="preprocessor"> typedef mshadow::half::half_t DType; \</span></div> |
| <div class="line"><a name="l01342"></a><span class="lineno"> 1342</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01343"></a><span class="lineno"> 1343</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01344"></a><span class="lineno"> 1344</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01345"></a><span class="lineno"> 1345</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01346"></a><span class="lineno"> 1346</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01347"></a><span class="lineno"> 1347</span> <span class="preprocessor"> "floating point types not uint8"; \</span></div> |
| <div class="line"><a name="l01348"></a><span class="lineno"> 1348</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01349"></a><span class="lineno"> 1349</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01350"></a><span class="lineno"> 1350</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01351"></a><span class="lineno"> 1351</span> <span class="preprocessor"> "floating point types not int8"; \</span></div> |
| <div class="line"><a name="l01352"></a><span class="lineno"> 1352</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01353"></a><span class="lineno"> 1353</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01354"></a><span class="lineno"> 1354</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01355"></a><span class="lineno"> 1355</span> <span class="preprocessor"> "floating point types, not int32";\</span></div> |
| <div class="line"><a name="l01356"></a><span class="lineno"> 1356</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01357"></a><span class="lineno"> 1357</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01358"></a><span class="lineno"> 1358</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01359"></a><span class="lineno"> 1359</span> <span class="preprocessor"> "floating point types, not int64";\</span></div> |
| <div class="line"><a name="l01360"></a><span class="lineno"> 1360</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01361"></a><span class="lineno"> 1361</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01362"></a><span class="lineno"> 1362</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01363"></a><span class="lineno"> 1363</span> <span class="preprocessor"> "floating point types, not bool"; \</span></div> |
| <div class="line"><a name="l01364"></a><span class="lineno"> 1364</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01365"></a><span class="lineno"> 1365</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01366"></a><span class="lineno"> 1366</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01367"></a><span class="lineno"> 1367</span> <span class="preprocessor"> "floating point types, not int16";\</span></div> |
| <div class="line"><a name="l01368"></a><span class="lineno"> 1368</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01369"></a><span class="lineno"> 1369</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01370"></a><span class="lineno"> 1370</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01371"></a><span class="lineno"> 1371</span> <span class="preprocessor"> "floating point types not uint16";\</span></div> |
| <div class="line"><a name="l01372"></a><span class="lineno"> 1372</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01373"></a><span class="lineno"> 1373</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01374"></a><span class="lineno"> 1374</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01375"></a><span class="lineno"> 1375</span> <span class="preprocessor"> "floating point types not uint32";\</span></div> |
| <div class="line"><a name="l01376"></a><span class="lineno"> 1376</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01377"></a><span class="lineno"> 1377</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01378"></a><span class="lineno"> 1378</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01379"></a><span class="lineno"> 1379</span> <span class="preprocessor"> "floating point types not uint64";\</span></div> |
| <div class="line"><a name="l01380"></a><span class="lineno"> 1380</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01381"></a><span class="lineno"> 1381</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01382"></a><span class="lineno"> 1382</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01383"></a><span class="lineno"> 1383</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01384"></a><span class="lineno"> 1384</span>  </div> |
| <div class="line"><a name="l01385"></a><span class="lineno"> 1385</span> <span class="preprocessor">#ifndef __NVCC__</span></div> |
| <div class="line"><a name="l01386"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5675a926d37704ceb4b997cbc19dabc2"> 1386</a></span> <span class="preprocessor">#define MSHADOW_REAL_TYPE_SWITCH_EX(type$, DType$, DLargeType$, ...) \</span></div> |
| <div class="line"><a name="l01387"></a><span class="lineno"> 1387</span> <span class="preprocessor"> switch (type$) { \</span></div> |
| <div class="line"><a name="l01388"></a><span class="lineno"> 1388</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01389"></a><span class="lineno"> 1389</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01390"></a><span class="lineno"> 1390</span> <span class="preprocessor"> typedef float DType$; \</span></div> |
| <div class="line"><a name="l01391"></a><span class="lineno"> 1391</span> <span class="preprocessor"> typedef float DLargeType$; \</span></div> |
| <div class="line"><a name="l01392"></a><span class="lineno"> 1392</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01393"></a><span class="lineno"> 1393</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01394"></a><span class="lineno"> 1394</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01395"></a><span class="lineno"> 1395</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01396"></a><span class="lineno"> 1396</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01397"></a><span class="lineno"> 1397</span> <span class="preprocessor"> typedef double DType$; \</span></div> |
| <div class="line"><a name="l01398"></a><span class="lineno"> 1398</span> <span class="preprocessor"> typedef double DLargeType$; \</span></div> |
| <div class="line"><a name="l01399"></a><span class="lineno"> 1399</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01400"></a><span class="lineno"> 1400</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01401"></a><span class="lineno"> 1401</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01402"></a><span class="lineno"> 1402</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01403"></a><span class="lineno"> 1403</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01404"></a><span class="lineno"> 1404</span> <span class="preprocessor"> typedef mshadow::half::half_t DType$; \</span></div> |
| <div class="line"><a name="l01405"></a><span class="lineno"> 1405</span> <span class="preprocessor"> typedef float DLargeType$; \</span></div> |
| <div class="line"><a name="l01406"></a><span class="lineno"> 1406</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01407"></a><span class="lineno"> 1407</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01408"></a><span class="lineno"> 1408</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01409"></a><span class="lineno"> 1409</span> <span class="preprocessor"> case mshadow::kBfloat16: \</span></div> |
| <div class="line"><a name="l01410"></a><span class="lineno"> 1410</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01411"></a><span class="lineno"> 1411</span> <span class="preprocessor"> typedef mshadow::bfloat::bf16_t DType$; \</span></div> |
| <div class="line"><a name="l01412"></a><span class="lineno"> 1412</span> <span class="preprocessor"> typedef float DLargeType$; \</span></div> |
| <div class="line"><a name="l01413"></a><span class="lineno"> 1413</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01414"></a><span class="lineno"> 1414</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01415"></a><span class="lineno"> 1415</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01416"></a><span class="lineno"> 1416</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01417"></a><span class="lineno"> 1417</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01418"></a><span class="lineno"> 1418</span> <span class="preprocessor"> "floating point types not uint8"; \</span></div> |
| <div class="line"><a name="l01419"></a><span class="lineno"> 1419</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01420"></a><span class="lineno"> 1420</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01421"></a><span class="lineno"> 1421</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01422"></a><span class="lineno"> 1422</span> <span class="preprocessor"> "floating point types not int8"; \</span></div> |
| <div class="line"><a name="l01423"></a><span class="lineno"> 1423</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01424"></a><span class="lineno"> 1424</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01425"></a><span class="lineno"> 1425</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01426"></a><span class="lineno"> 1426</span> <span class="preprocessor"> "floating point types, not int32";\</span></div> |
| <div class="line"><a name="l01427"></a><span class="lineno"> 1427</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01428"></a><span class="lineno"> 1428</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01429"></a><span class="lineno"> 1429</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01430"></a><span class="lineno"> 1430</span> <span class="preprocessor"> "floating point types, not int64";\</span></div> |
| <div class="line"><a name="l01431"></a><span class="lineno"> 1431</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01432"></a><span class="lineno"> 1432</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01433"></a><span class="lineno"> 1433</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01434"></a><span class="lineno"> 1434</span> <span class="preprocessor"> "floating point types, not bool"; \</span></div> |
| <div class="line"><a name="l01435"></a><span class="lineno"> 1435</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01436"></a><span class="lineno"> 1436</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01437"></a><span class="lineno"> 1437</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01438"></a><span class="lineno"> 1438</span> <span class="preprocessor"> "floating point types, not int16";\</span></div> |
| <div class="line"><a name="l01439"></a><span class="lineno"> 1439</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01440"></a><span class="lineno"> 1440</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01441"></a><span class="lineno"> 1441</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01442"></a><span class="lineno"> 1442</span> <span class="preprocessor"> "floating point types not uint16";\</span></div> |
| <div class="line"><a name="l01443"></a><span class="lineno"> 1443</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01444"></a><span class="lineno"> 1444</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01445"></a><span class="lineno"> 1445</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01446"></a><span class="lineno"> 1446</span> <span class="preprocessor"> "floating point types not uint32";\</span></div> |
| <div class="line"><a name="l01447"></a><span class="lineno"> 1447</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01448"></a><span class="lineno"> 1448</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01449"></a><span class="lineno"> 1449</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01450"></a><span class="lineno"> 1450</span> <span class="preprocessor"> "floating point types not uint64";\</span></div> |
| <div class="line"><a name="l01451"></a><span class="lineno"> 1451</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01452"></a><span class="lineno"> 1452</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01453"></a><span class="lineno"> 1453</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type$; \</span></div> |
| <div class="line"><a name="l01454"></a><span class="lineno"> 1454</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01455"></a><span class="lineno"> 1455</span> <span class="preprocessor">#else</span></div> |
| <div class="line"><a name="l01456"></a><span class="lineno"> 1456</span> <span class="preprocessor">#define MSHADOW_REAL_TYPE_SWITCH_EX(type$, DType$, DLargeType$, ...) \</span></div> |
| <div class="line"><a name="l01457"></a><span class="lineno"> 1457</span> <span class="preprocessor"> switch (type$) { \</span></div> |
| <div class="line"><a name="l01458"></a><span class="lineno"> 1458</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01459"></a><span class="lineno"> 1459</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01460"></a><span class="lineno"> 1460</span> <span class="preprocessor"> typedef float DType$; \</span></div> |
| <div class="line"><a name="l01461"></a><span class="lineno"> 1461</span> <span class="preprocessor"> typedef float DLargeType$; \</span></div> |
| <div class="line"><a name="l01462"></a><span class="lineno"> 1462</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01463"></a><span class="lineno"> 1463</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01464"></a><span class="lineno"> 1464</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01465"></a><span class="lineno"> 1465</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01466"></a><span class="lineno"> 1466</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01467"></a><span class="lineno"> 1467</span> <span class="preprocessor"> typedef double DType$; \</span></div> |
| <div class="line"><a name="l01468"></a><span class="lineno"> 1468</span> <span class="preprocessor"> typedef double DLargeType$; \</span></div> |
| <div class="line"><a name="l01469"></a><span class="lineno"> 1469</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01470"></a><span class="lineno"> 1470</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01471"></a><span class="lineno"> 1471</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01472"></a><span class="lineno"> 1472</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01473"></a><span class="lineno"> 1473</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01474"></a><span class="lineno"> 1474</span> <span class="preprocessor"> typedef mshadow::half::half_t DType$; \</span></div> |
| <div class="line"><a name="l01475"></a><span class="lineno"> 1475</span> <span class="preprocessor"> typedef float DLargeType$; \</span></div> |
| <div class="line"><a name="l01476"></a><span class="lineno"> 1476</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01477"></a><span class="lineno"> 1477</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01478"></a><span class="lineno"> 1478</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01479"></a><span class="lineno"> 1479</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01480"></a><span class="lineno"> 1480</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01481"></a><span class="lineno"> 1481</span> <span class="preprocessor"> "floating point types not uint8"; \</span></div> |
| <div class="line"><a name="l01482"></a><span class="lineno"> 1482</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01483"></a><span class="lineno"> 1483</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01484"></a><span class="lineno"> 1484</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01485"></a><span class="lineno"> 1485</span> <span class="preprocessor"> "floating point types not int8"; \</span></div> |
| <div class="line"><a name="l01486"></a><span class="lineno"> 1486</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01487"></a><span class="lineno"> 1487</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01488"></a><span class="lineno"> 1488</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01489"></a><span class="lineno"> 1489</span> <span class="preprocessor"> "floating point types, not int32";\</span></div> |
| <div class="line"><a name="l01490"></a><span class="lineno"> 1490</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01491"></a><span class="lineno"> 1491</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01492"></a><span class="lineno"> 1492</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01493"></a><span class="lineno"> 1493</span> <span class="preprocessor"> "floating point types, not int64";\</span></div> |
| <div class="line"><a name="l01494"></a><span class="lineno"> 1494</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01495"></a><span class="lineno"> 1495</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01496"></a><span class="lineno"> 1496</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01497"></a><span class="lineno"> 1497</span> <span class="preprocessor"> "floating point types, not bool"; \</span></div> |
| <div class="line"><a name="l01498"></a><span class="lineno"> 1498</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01499"></a><span class="lineno"> 1499</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01500"></a><span class="lineno"> 1500</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01501"></a><span class="lineno"> 1501</span> <span class="preprocessor"> "floating point types, not int16";\</span></div> |
| <div class="line"><a name="l01502"></a><span class="lineno"> 1502</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01503"></a><span class="lineno"> 1503</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01504"></a><span class="lineno"> 1504</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01505"></a><span class="lineno"> 1505</span> <span class="preprocessor"> "floating point types not uint16";\</span></div> |
| <div class="line"><a name="l01506"></a><span class="lineno"> 1506</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01507"></a><span class="lineno"> 1507</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01508"></a><span class="lineno"> 1508</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01509"></a><span class="lineno"> 1509</span> <span class="preprocessor"> "floating point types not uint32";\</span></div> |
| <div class="line"><a name="l01510"></a><span class="lineno"> 1510</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01511"></a><span class="lineno"> 1511</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01512"></a><span class="lineno"> 1512</span> <span class="preprocessor"> LOG(FATAL) << "This operation only support " \</span></div> |
| <div class="line"><a name="l01513"></a><span class="lineno"> 1513</span> <span class="preprocessor"> "floating point types not uint64";\</span></div> |
| <div class="line"><a name="l01514"></a><span class="lineno"> 1514</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01515"></a><span class="lineno"> 1515</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01516"></a><span class="lineno"> 1516</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type$; \</span></div> |
| <div class="line"><a name="l01517"></a><span class="lineno"> 1517</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01518"></a><span class="lineno"> 1518</span> <span class="preprocessor">#endif</span></div> |
| <div class="line"><a name="l01519"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#ad91ac184f854a56c21d06fafae8b93c6"> 1519</a></span> <span class="preprocessor">#define MSHADOW_LAYOUT_SWITCH(layout, Layout, ...) \</span></div> |
| <div class="line"><a name="l01520"></a><span class="lineno"> 1520</span> <span class="preprocessor"> switch (layout) { \</span></div> |
| <div class="line"><a name="l01521"></a><span class="lineno"> 1521</span> <span class="preprocessor"> case mshadow::kNCHW: \</span></div> |
| <div class="line"><a name="l01522"></a><span class="lineno"> 1522</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01523"></a><span class="lineno"> 1523</span> <span class="preprocessor"> const int Layout = kNCHW; \</span></div> |
| <div class="line"><a name="l01524"></a><span class="lineno"> 1524</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01525"></a><span class="lineno"> 1525</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01526"></a><span class="lineno"> 1526</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01527"></a><span class="lineno"> 1527</span> <span class="preprocessor"> case mshadow::kNHWC: \</span></div> |
| <div class="line"><a name="l01528"></a><span class="lineno"> 1528</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01529"></a><span class="lineno"> 1529</span> <span class="preprocessor"> const int Layout = kNHWC; \</span></div> |
| <div class="line"><a name="l01530"></a><span class="lineno"> 1530</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01531"></a><span class="lineno"> 1531</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01532"></a><span class="lineno"> 1532</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01533"></a><span class="lineno"> 1533</span> <span class="preprocessor"> case mshadow::kNCDHW: \</span></div> |
| <div class="line"><a name="l01534"></a><span class="lineno"> 1534</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01535"></a><span class="lineno"> 1535</span> <span class="preprocessor"> const int Layout = kNCDHW; \</span></div> |
| <div class="line"><a name="l01536"></a><span class="lineno"> 1536</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01537"></a><span class="lineno"> 1537</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01538"></a><span class="lineno"> 1538</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01539"></a><span class="lineno"> 1539</span> <span class="preprocessor"> case mshadow::kNDHWC: \</span></div> |
| <div class="line"><a name="l01540"></a><span class="lineno"> 1540</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01541"></a><span class="lineno"> 1541</span> <span class="preprocessor"> const int Layout = kNDHWC; \</span></div> |
| <div class="line"><a name="l01542"></a><span class="lineno"> 1542</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01543"></a><span class="lineno"> 1543</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01544"></a><span class="lineno"> 1544</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01545"></a><span class="lineno"> 1545</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01546"></a><span class="lineno"> 1546</span> <span class="preprocessor"> LOG(FATAL) << "Unknown layout enum " << layout; \</span></div> |
| <div class="line"><a name="l01547"></a><span class="lineno"> 1547</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01548"></a><span class="lineno"> 1548</span>  </div> |
| <div class="line"><a name="l01553"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#ae3f6108e48f51f98cd491fcded1d8ae0"> 1553</a></span> <span class="preprocessor">#define MSHADOW_IDX_TYPE_SWITCH(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01554"></a><span class="lineno"> 1554</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01555"></a><span class="lineno"> 1555</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01556"></a><span class="lineno"> 1556</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01557"></a><span class="lineno"> 1557</span> <span class="preprocessor"> typedef int64_t DType; \</span></div> |
| <div class="line"><a name="l01558"></a><span class="lineno"> 1558</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01559"></a><span class="lineno"> 1559</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01560"></a><span class="lineno"> 1560</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01561"></a><span class="lineno"> 1561</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01562"></a><span class="lineno"> 1562</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01563"></a><span class="lineno"> 1563</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01564"></a><span class="lineno"> 1564</span>  </div> |
| <div class="line"><a name="l01565"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#a2c0869fdb46669460e7c56cec04dfaa7"> 1565</a></span> <span class="preprocessor">#define MSHADOW_TYPE_SWITCH_WITH_BOOL(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01566"></a><span class="lineno"> 1566</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01567"></a><span class="lineno"> 1567</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01568"></a><span class="lineno"> 1568</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01569"></a><span class="lineno"> 1569</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01570"></a><span class="lineno"> 1570</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01571"></a><span class="lineno"> 1571</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01572"></a><span class="lineno"> 1572</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01573"></a><span class="lineno"> 1573</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01574"></a><span class="lineno"> 1574</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01575"></a><span class="lineno"> 1575</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01576"></a><span class="lineno"> 1576</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01577"></a><span class="lineno"> 1577</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01578"></a><span class="lineno"> 1578</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01579"></a><span class="lineno"> 1579</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01580"></a><span class="lineno"> 1580</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01581"></a><span class="lineno"> 1581</span> <span class="preprocessor"> typedef mshadow::half::half_t DType; \</span></div> |
| <div class="line"><a name="l01582"></a><span class="lineno"> 1582</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01583"></a><span class="lineno"> 1583</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01584"></a><span class="lineno"> 1584</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01585"></a><span class="lineno"> 1585</span> <span class="preprocessor"> case mshadow::kBfloat16: \</span></div> |
| <div class="line"><a name="l01586"></a><span class="lineno"> 1586</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01587"></a><span class="lineno"> 1587</span> <span class="preprocessor"> typedef mshadow::bfloat::bf16_t DType; \</span></div> |
| <div class="line"><a name="l01588"></a><span class="lineno"> 1588</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01589"></a><span class="lineno"> 1589</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01590"></a><span class="lineno"> 1590</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01591"></a><span class="lineno"> 1591</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01592"></a><span class="lineno"> 1592</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01593"></a><span class="lineno"> 1593</span> <span class="preprocessor"> typedef uint8_t DType; \</span></div> |
| <div class="line"><a name="l01594"></a><span class="lineno"> 1594</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01595"></a><span class="lineno"> 1595</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01596"></a><span class="lineno"> 1596</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01597"></a><span class="lineno"> 1597</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01598"></a><span class="lineno"> 1598</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01599"></a><span class="lineno"> 1599</span> <span class="preprocessor"> typedef int8_t DType; \</span></div> |
| <div class="line"><a name="l01600"></a><span class="lineno"> 1600</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01601"></a><span class="lineno"> 1601</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01602"></a><span class="lineno"> 1602</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01603"></a><span class="lineno"> 1603</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01604"></a><span class="lineno"> 1604</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01605"></a><span class="lineno"> 1605</span> <span class="preprocessor"> typedef int32_t DType; \</span></div> |
| <div class="line"><a name="l01606"></a><span class="lineno"> 1606</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01607"></a><span class="lineno"> 1607</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01608"></a><span class="lineno"> 1608</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01609"></a><span class="lineno"> 1609</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01610"></a><span class="lineno"> 1610</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01611"></a><span class="lineno"> 1611</span> <span class="preprocessor"> typedef int64_t DType; \</span></div> |
| <div class="line"><a name="l01612"></a><span class="lineno"> 1612</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01613"></a><span class="lineno"> 1613</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01614"></a><span class="lineno"> 1614</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01615"></a><span class="lineno"> 1615</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01616"></a><span class="lineno"> 1616</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01617"></a><span class="lineno"> 1617</span> <span class="preprocessor"> typedef bool DType; \</span></div> |
| <div class="line"><a name="l01618"></a><span class="lineno"> 1618</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01619"></a><span class="lineno"> 1619</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01620"></a><span class="lineno"> 1620</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01621"></a><span class="lineno"> 1621</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01622"></a><span class="lineno"> 1622</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01623"></a><span class="lineno"> 1623</span> <span class="preprocessor"> "support int16 type"; \</span></div> |
| <div class="line"><a name="l01624"></a><span class="lineno"> 1624</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01625"></a><span class="lineno"> 1625</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01626"></a><span class="lineno"> 1626</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01627"></a><span class="lineno"> 1627</span> <span class="preprocessor"> "support uint16 type"; \</span></div> |
| <div class="line"><a name="l01628"></a><span class="lineno"> 1628</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01629"></a><span class="lineno"> 1629</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01630"></a><span class="lineno"> 1630</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01631"></a><span class="lineno"> 1631</span> <span class="preprocessor"> "support uint32 type"; \</span></div> |
| <div class="line"><a name="l01632"></a><span class="lineno"> 1632</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01633"></a><span class="lineno"> 1633</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01634"></a><span class="lineno"> 1634</span> <span class="preprocessor"> LOG(FATAL) << "This operation does not " \</span></div> |
| <div class="line"><a name="l01635"></a><span class="lineno"> 1635</span> <span class="preprocessor"> "support uint64 type"; \</span></div> |
| <div class="line"><a name="l01636"></a><span class="lineno"> 1636</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01637"></a><span class="lineno"> 1637</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01638"></a><span class="lineno"> 1638</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01639"></a><span class="lineno"> 1639</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01640"></a><span class="lineno"> 1640</span>  </div> |
| <div class="line"><a name="l01641"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#ad1d10c829aca3f5d033a09161dfaed33"> 1641</a></span> <span class="preprocessor">#define MSHADOW_TYPE_SWITCH_EXT(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01642"></a><span class="lineno"> 1642</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01643"></a><span class="lineno"> 1643</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01644"></a><span class="lineno"> 1644</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01645"></a><span class="lineno"> 1645</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01646"></a><span class="lineno"> 1646</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01647"></a><span class="lineno"> 1647</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01648"></a><span class="lineno"> 1648</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01649"></a><span class="lineno"> 1649</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01650"></a><span class="lineno"> 1650</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01651"></a><span class="lineno"> 1651</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01652"></a><span class="lineno"> 1652</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01653"></a><span class="lineno"> 1653</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01654"></a><span class="lineno"> 1654</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01655"></a><span class="lineno"> 1655</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01656"></a><span class="lineno"> 1656</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01657"></a><span class="lineno"> 1657</span> <span class="preprocessor"> typedef mshadow::half::half_t DType; \</span></div> |
| <div class="line"><a name="l01658"></a><span class="lineno"> 1658</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01659"></a><span class="lineno"> 1659</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01660"></a><span class="lineno"> 1660</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01661"></a><span class="lineno"> 1661</span> <span class="preprocessor"> case mshadow::kBfloat16: \</span></div> |
| <div class="line"><a name="l01662"></a><span class="lineno"> 1662</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01663"></a><span class="lineno"> 1663</span> <span class="preprocessor"> typedef mshadow::bfloat::bf16_t DType; \</span></div> |
| <div class="line"><a name="l01664"></a><span class="lineno"> 1664</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01665"></a><span class="lineno"> 1665</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01666"></a><span class="lineno"> 1666</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01667"></a><span class="lineno"> 1667</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01668"></a><span class="lineno"> 1668</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01669"></a><span class="lineno"> 1669</span> <span class="preprocessor"> typedef uint8_t DType; \</span></div> |
| <div class="line"><a name="l01670"></a><span class="lineno"> 1670</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01671"></a><span class="lineno"> 1671</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01672"></a><span class="lineno"> 1672</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01673"></a><span class="lineno"> 1673</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01674"></a><span class="lineno"> 1674</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01675"></a><span class="lineno"> 1675</span> <span class="preprocessor"> typedef int8_t DType; \</span></div> |
| <div class="line"><a name="l01676"></a><span class="lineno"> 1676</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01677"></a><span class="lineno"> 1677</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01678"></a><span class="lineno"> 1678</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01679"></a><span class="lineno"> 1679</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01680"></a><span class="lineno"> 1680</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01681"></a><span class="lineno"> 1681</span> <span class="preprocessor"> typedef int32_t DType; \</span></div> |
| <div class="line"><a name="l01682"></a><span class="lineno"> 1682</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01683"></a><span class="lineno"> 1683</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01684"></a><span class="lineno"> 1684</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01685"></a><span class="lineno"> 1685</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01686"></a><span class="lineno"> 1686</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01687"></a><span class="lineno"> 1687</span> <span class="preprocessor"> typedef int64_t DType; \</span></div> |
| <div class="line"><a name="l01688"></a><span class="lineno"> 1688</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01689"></a><span class="lineno"> 1689</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01690"></a><span class="lineno"> 1690</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01691"></a><span class="lineno"> 1691</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01692"></a><span class="lineno"> 1692</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01693"></a><span class="lineno"> 1693</span> <span class="preprocessor"> typedef int16_t DType; \</span></div> |
| <div class="line"><a name="l01694"></a><span class="lineno"> 1694</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01695"></a><span class="lineno"> 1695</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01696"></a><span class="lineno"> 1696</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01697"></a><span class="lineno"> 1697</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01698"></a><span class="lineno"> 1698</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01699"></a><span class="lineno"> 1699</span> <span class="preprocessor"> typedef uint16_t DType; \</span></div> |
| <div class="line"><a name="l01700"></a><span class="lineno"> 1700</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01701"></a><span class="lineno"> 1701</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01702"></a><span class="lineno"> 1702</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01703"></a><span class="lineno"> 1703</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01704"></a><span class="lineno"> 1704</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01705"></a><span class="lineno"> 1705</span> <span class="preprocessor"> typedef uint32_t DType; \</span></div> |
| <div class="line"><a name="l01706"></a><span class="lineno"> 1706</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01707"></a><span class="lineno"> 1707</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01708"></a><span class="lineno"> 1708</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01709"></a><span class="lineno"> 1709</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01710"></a><span class="lineno"> 1710</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01711"></a><span class="lineno"> 1711</span> <span class="preprocessor"> typedef uint64_t DType; \</span></div> |
| <div class="line"><a name="l01712"></a><span class="lineno"> 1712</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01713"></a><span class="lineno"> 1713</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01714"></a><span class="lineno"> 1714</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01715"></a><span class="lineno"> 1715</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01716"></a><span class="lineno"> 1716</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01717"></a><span class="lineno"> 1717</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01718"></a><span class="lineno"> 1718</span>  </div> |
| <div class="line"><a name="l01719"></a><span class="lineno"><a class="line" href="3rdparty_2mshadow_2mshadow_2base_8h.html#adce833400b796ef18899f74d3ab301b3"> 1719</a></span> <span class="preprocessor">#define MSHADOW_TYPE_SWITCH_EXT_WITH_BOOL(type, DType, ...) \</span></div> |
| <div class="line"><a name="l01720"></a><span class="lineno"> 1720</span> <span class="preprocessor"> switch (type) { \</span></div> |
| <div class="line"><a name="l01721"></a><span class="lineno"> 1721</span> <span class="preprocessor"> case mshadow::kFloat32: \</span></div> |
| <div class="line"><a name="l01722"></a><span class="lineno"> 1722</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01723"></a><span class="lineno"> 1723</span> <span class="preprocessor"> typedef float DType; \</span></div> |
| <div class="line"><a name="l01724"></a><span class="lineno"> 1724</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01725"></a><span class="lineno"> 1725</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01726"></a><span class="lineno"> 1726</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01727"></a><span class="lineno"> 1727</span> <span class="preprocessor"> case mshadow::kFloat64: \</span></div> |
| <div class="line"><a name="l01728"></a><span class="lineno"> 1728</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01729"></a><span class="lineno"> 1729</span> <span class="preprocessor"> typedef double DType; \</span></div> |
| <div class="line"><a name="l01730"></a><span class="lineno"> 1730</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01731"></a><span class="lineno"> 1731</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01732"></a><span class="lineno"> 1732</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01733"></a><span class="lineno"> 1733</span> <span class="preprocessor"> case mshadow::kFloat16: \</span></div> |
| <div class="line"><a name="l01734"></a><span class="lineno"> 1734</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01735"></a><span class="lineno"> 1735</span> <span class="preprocessor"> typedef mshadow::half::half_t DType; \</span></div> |
| <div class="line"><a name="l01736"></a><span class="lineno"> 1736</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01737"></a><span class="lineno"> 1737</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01738"></a><span class="lineno"> 1738</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01739"></a><span class="lineno"> 1739</span> <span class="preprocessor"> case mshadow::kBfloat16: \</span></div> |
| <div class="line"><a name="l01740"></a><span class="lineno"> 1740</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01741"></a><span class="lineno"> 1741</span> <span class="preprocessor"> typedef mshadow::bfloat::bf16_t DType; \</span></div> |
| <div class="line"><a name="l01742"></a><span class="lineno"> 1742</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01743"></a><span class="lineno"> 1743</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01744"></a><span class="lineno"> 1744</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01745"></a><span class="lineno"> 1745</span> <span class="preprocessor"> case mshadow::kUint8: \</span></div> |
| <div class="line"><a name="l01746"></a><span class="lineno"> 1746</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01747"></a><span class="lineno"> 1747</span> <span class="preprocessor"> typedef uint8_t DType; \</span></div> |
| <div class="line"><a name="l01748"></a><span class="lineno"> 1748</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01749"></a><span class="lineno"> 1749</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01750"></a><span class="lineno"> 1750</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01751"></a><span class="lineno"> 1751</span> <span class="preprocessor"> case mshadow::kInt8: \</span></div> |
| <div class="line"><a name="l01752"></a><span class="lineno"> 1752</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01753"></a><span class="lineno"> 1753</span> <span class="preprocessor"> typedef int8_t DType; \</span></div> |
| <div class="line"><a name="l01754"></a><span class="lineno"> 1754</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01755"></a><span class="lineno"> 1755</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01756"></a><span class="lineno"> 1756</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01757"></a><span class="lineno"> 1757</span> <span class="preprocessor"> case mshadow::kInt32: \</span></div> |
| <div class="line"><a name="l01758"></a><span class="lineno"> 1758</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01759"></a><span class="lineno"> 1759</span> <span class="preprocessor"> typedef int32_t DType; \</span></div> |
| <div class="line"><a name="l01760"></a><span class="lineno"> 1760</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01761"></a><span class="lineno"> 1761</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01762"></a><span class="lineno"> 1762</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01763"></a><span class="lineno"> 1763</span> <span class="preprocessor"> case mshadow::kInt64: \</span></div> |
| <div class="line"><a name="l01764"></a><span class="lineno"> 1764</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01765"></a><span class="lineno"> 1765</span> <span class="preprocessor"> typedef int64_t DType; \</span></div> |
| <div class="line"><a name="l01766"></a><span class="lineno"> 1766</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01767"></a><span class="lineno"> 1767</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01768"></a><span class="lineno"> 1768</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01769"></a><span class="lineno"> 1769</span> <span class="preprocessor"> case mshadow::kBool: \</span></div> |
| <div class="line"><a name="l01770"></a><span class="lineno"> 1770</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01771"></a><span class="lineno"> 1771</span> <span class="preprocessor"> typedef bool DType; \</span></div> |
| <div class="line"><a name="l01772"></a><span class="lineno"> 1772</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01773"></a><span class="lineno"> 1773</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01774"></a><span class="lineno"> 1774</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01775"></a><span class="lineno"> 1775</span> <span class="preprocessor"> case mshadow::kInt16: \</span></div> |
| <div class="line"><a name="l01776"></a><span class="lineno"> 1776</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01777"></a><span class="lineno"> 1777</span> <span class="preprocessor"> typedef int16_t DType; \</span></div> |
| <div class="line"><a name="l01778"></a><span class="lineno"> 1778</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01779"></a><span class="lineno"> 1779</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01780"></a><span class="lineno"> 1780</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01781"></a><span class="lineno"> 1781</span> <span class="preprocessor"> case mshadow::kUint16: \</span></div> |
| <div class="line"><a name="l01782"></a><span class="lineno"> 1782</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01783"></a><span class="lineno"> 1783</span> <span class="preprocessor"> typedef uint16_t DType; \</span></div> |
| <div class="line"><a name="l01784"></a><span class="lineno"> 1784</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01785"></a><span class="lineno"> 1785</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01786"></a><span class="lineno"> 1786</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01787"></a><span class="lineno"> 1787</span> <span class="preprocessor"> case mshadow::kUint32: \</span></div> |
| <div class="line"><a name="l01788"></a><span class="lineno"> 1788</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01789"></a><span class="lineno"> 1789</span> <span class="preprocessor"> typedef uint32_t DType; \</span></div> |
| <div class="line"><a name="l01790"></a><span class="lineno"> 1790</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01791"></a><span class="lineno"> 1791</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01792"></a><span class="lineno"> 1792</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01793"></a><span class="lineno"> 1793</span> <span class="preprocessor"> case mshadow::kUint64: \</span></div> |
| <div class="line"><a name="l01794"></a><span class="lineno"> 1794</span> <span class="preprocessor"> { \</span></div> |
| <div class="line"><a name="l01795"></a><span class="lineno"> 1795</span> <span class="preprocessor"> typedef uint64_t DType; \</span></div> |
| <div class="line"><a name="l01796"></a><span class="lineno"> 1796</span> <span class="preprocessor"> {__VA_ARGS__} \</span></div> |
| <div class="line"><a name="l01797"></a><span class="lineno"> 1797</span> <span class="preprocessor"> } \</span></div> |
| <div class="line"><a name="l01798"></a><span class="lineno"> 1798</span> <span class="preprocessor"> break; \</span></div> |
| <div class="line"><a name="l01799"></a><span class="lineno"> 1799</span> <span class="preprocessor"> default: \</span></div> |
| <div class="line"><a name="l01800"></a><span class="lineno"> 1800</span> <span class="preprocessor"> LOG(FATAL) << "Unknown type enum " << type; \</span></div> |
| <div class="line"><a name="l01801"></a><span class="lineno"> 1801</span> <span class="preprocessor"> }</span></div> |
| <div class="line"><a name="l01802"></a><span class="lineno"> 1802</span>  </div> |
| <div class="line"><a name="l01804"></a><span class="lineno"><a class="line" href="namespacemshadow.html#a2e0671a4ac1729d7f8996ba091f8e07c"> 1804</a></span> <span class="keyword">inline</span> <span class="keywordtype">size_t</span> <a class="code" href="namespacemshadow.html#a2e0671a4ac1729d7f8996ba091f8e07c">mshadow_sizeof</a>(<span class="keywordtype">int</span> type) {</div> |
| <div class="line"><a name="l01805"></a><span class="lineno"> 1805</span>  <span class="keywordtype">int</span> size = 0;</div> |
| <div class="line"><a name="l01806"></a><span class="lineno"> 1806</span>  <a class="code" href="3rdparty_2mshadow_2mshadow_2base_8h.html#adce833400b796ef18899f74d3ab301b3">MSHADOW_TYPE_SWITCH_EXT_WITH_BOOL</a>(type, DType, size = <span class="keyword">sizeof</span>(DType););</div> |
| <div class="line"><a name="l01807"></a><span class="lineno"> 1807</span>  <span class="keywordflow">return</span> size;</div> |
| <div class="line"><a name="l01808"></a><span class="lineno"> 1808</span> }</div> |
| <div class="line"><a name="l01809"></a><span class="lineno"> 1809</span>  </div> |
| <div class="line"><a name="l01810"></a><span class="lineno"> 1810</span> <span class="comment">/*/ \brief get string with the type name from type enum */</span></div> |
| <div class="line"><a name="l01811"></a><span class="lineno"><a class="line" href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072"> 1811</a></span> <span class="keyword">inline</span> std::string <a class="code" href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072">dtype_string</a>(<span class="keyword">const</span> <span class="keywordtype">int</span> dtype) {</div> |
| <div class="line"><a name="l01812"></a><span class="lineno"> 1812</span>  <span class="keywordflow">switch</span> (dtype) {</div> |
| <div class="line"><a name="l01813"></a><span class="lineno"> 1813</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">mshadow::kFloat32</a>:</div> |
| <div class="line"><a name="l01814"></a><span class="lineno"> 1814</span>  <span class="keywordflow">return</span> <span class="stringliteral">"float"</span>;</div> |
| <div class="line"><a name="l01815"></a><span class="lineno"> 1815</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28">mshadow::kFloat64</a>:</div> |
| <div class="line"><a name="l01816"></a><span class="lineno"> 1816</span>  <span class="keywordflow">return</span> <span class="stringliteral">"double"</span>;</div> |
| <div class="line"><a name="l01817"></a><span class="lineno"> 1817</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4">mshadow::kFloat16</a>:</div> |
| <div class="line"><a name="l01818"></a><span class="lineno"> 1818</span>  <span class="keywordflow">return</span> <span class="stringliteral">"half"</span>;</div> |
| <div class="line"><a name="l01819"></a><span class="lineno"> 1819</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0">mshadow::kUint8</a>:</div> |
| <div class="line"><a name="l01820"></a><span class="lineno"> 1820</span>  <span class="keywordflow">return</span> <span class="stringliteral">"unsigned char"</span>;</div> |
| <div class="line"><a name="l01821"></a><span class="lineno"> 1821</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c">mshadow::kInt8</a>:</div> |
| <div class="line"><a name="l01822"></a><span class="lineno"> 1822</span>  <span class="keywordflow">return</span> <span class="stringliteral">"char"</span>;</div> |
| <div class="line"><a name="l01823"></a><span class="lineno"> 1823</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368">mshadow::kInt32</a>:</div> |
| <div class="line"><a name="l01824"></a><span class="lineno"> 1824</span>  <span class="keywordflow">return</span> <span class="stringliteral">"int"</span>;</div> |
| <div class="line"><a name="l01825"></a><span class="lineno"> 1825</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23">mshadow::kInt64</a>:</div> |
| <div class="line"><a name="l01826"></a><span class="lineno"> 1826</span>  <span class="keywordflow">return</span> <span class="stringliteral">"long long"</span>;</div> |
| <div class="line"><a name="l01827"></a><span class="lineno"> 1827</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a">mshadow::kBool</a>:</div> |
| <div class="line"><a name="l01828"></a><span class="lineno"> 1828</span>  <span class="keywordflow">return</span> <span class="stringliteral">"bool"</span>;</div> |
| <div class="line"><a name="l01829"></a><span class="lineno"> 1829</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17">mshadow::kInt16</a>:</div> |
| <div class="line"><a name="l01830"></a><span class="lineno"> 1830</span>  <span class="keywordflow">return</span> <span class="stringliteral">"short"</span>;</div> |
| <div class="line"><a name="l01831"></a><span class="lineno"> 1831</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908">mshadow::kUint16</a>:</div> |
| <div class="line"><a name="l01832"></a><span class="lineno"> 1832</span>  <span class="keywordflow">return</span> <span class="stringliteral">"unsigned short"</span>;</div> |
| <div class="line"><a name="l01833"></a><span class="lineno"> 1833</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7">mshadow::kUint32</a>:</div> |
| <div class="line"><a name="l01834"></a><span class="lineno"> 1834</span>  <span class="keywordflow">return</span> <span class="stringliteral">"unsigned int"</span>;</div> |
| <div class="line"><a name="l01835"></a><span class="lineno"> 1835</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade">mshadow::kUint64</a>:</div> |
| <div class="line"><a name="l01836"></a><span class="lineno"> 1836</span>  <span class="keywordflow">return</span> <span class="stringliteral">"unsigned long long"</span>;</div> |
| <div class="line"><a name="l01837"></a><span class="lineno"> 1837</span>  <span class="keywordflow">default</span>:</div> |
| <div class="line"><a name="l01838"></a><span class="lineno"> 1838</span>  LOG(FATAL) << <span class="stringliteral">"Unknown type enum "</span> << dtype;</div> |
| <div class="line"><a name="l01839"></a><span class="lineno"> 1839</span>  }</div> |
| <div class="line"><a name="l01840"></a><span class="lineno"> 1840</span>  <span class="keywordflow">return</span> <span class="stringliteral">"unknown"</span>;</div> |
| <div class="line"><a name="l01841"></a><span class="lineno"> 1841</span> }</div> |
| <div class="line"><a name="l01842"></a><span class="lineno"> 1842</span>  </div> |
| <div class="line"><a name="l01843"></a><span class="lineno"> 1843</span> } <span class="comment">// namespace mshadow</span></div> |
| <div class="line"><a name="l01844"></a><span class="lineno"> 1844</span> <span class="preprocessor">#endif // MSHADOW_BASE_H_</span></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1saveto_html_ac78ceb69e8291f8fa767c2eda2a732a5"><div class="ttname"><a href="structmshadow_1_1sv_1_1saveto.html#ac78ceb69e8291f8fa767c2eda2a732a5">mshadow::sv::saveto::OPType</a></div><div class="ttdeci">op::right OPType</div><div class="ttdoc">corresponding binary operator type</div><div class="ttdef"><b>Definition:</b> base.h:691</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1minusto_html_a0da791365b9910dd337e228888521127"><div class="ttname"><a href="structmshadow_1_1sv_1_1minusto.html#a0da791365b9910dd337e228888521127">mshadow::sv::minusto::AlphaBLAS</a></div><div class="ttdeci">static default_real_t AlphaBLAS(void)</div><div class="ttdoc">helper constant to use BLAS, alpha</div><div class="ttdef"><b>Definition:</b> base.h:715</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae633c123a9c9897779d5c00ea5381775">mshadow::kNCHW</a></div><div class="ttdeci">@ kNCHW</div><div class="ttdef"><b>Definition:</b> base.h:501</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html">mshadow::red::maximum</a></div><div class="ttdoc">maximum reducer</div><div class="ttdef"><b>Definition:</b> base.h:1055</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_ae844649225377caf827a51ef0d080515"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#ae844649225377caf827a51ef0d080515">mshadow::red::minimum::Finalize</a></div><div class="ttdeci">static MSHADOW_XINLINE void Finalize(volatile DType &dst, volatile DType &residual)</div><div class="ttdoc">finalize reduction</div><div class="ttdef"><b>Definition:</b> base.h:1136</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_af14bfabf30b3f4bffdcd0827679d0390"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#af14bfabf30b3f4bffdcd0827679d0390">mshadow::red::minimum::Merge</a></div><div class="ttdeci">static MSHADOW_XINLINE void Merge(volatile DType &dst_val, volatile DType &dst_residual, volatile DType &src_val, volatile DType &src_residual)</div><div class="ttdoc">combine the results of two reducers</div><div class="ttdef"><b>Definition:</b> base.h:1128</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_af024e58661dffc3043fdab0ac6424659"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#af024e58661dffc3043fdab0ac6424659">mshadow::red::sum::Finalize</a></div><div class="ttdeci">static MSHADOW_XINLINE void Finalize(volatile DType &dst, volatile DType &residual)</div><div class="ttdoc">finalize reduction</div><div class="ttdef"><b>Definition:</b> base.h:1029</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a6b0a333f168fa9d54d2f13caabd595e0"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a6b0a333f168fa9d54d2f13caabd595e0">mshadow::red::limits::NegInfValue</a></div><div class="ttdeci">MSHADOW_XINLINE DType NegInfValue(void)</div><div class="ttdoc">negative infinity of certain types</div><div class="ttdef"><b>Definition:</b> base.h:873</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a78426e39f8f6170f709965da5aa2d118"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a78426e39f8f6170f709965da5aa2d118">mshadow::red::limits::MaxValue< uint32_t ></a></div><div class="ttdeci">MSHADOW_XINLINE uint32_t MaxValue< uint32_t >(void)</div><div class="ttdoc">maximum value of uint32_t</div><div class="ttdef"><b>Definition:</b> base.h:951</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_acc5e8630b2b3d865291c02b16c7b09c1"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#acc5e8630b2b3d865291c02b16c7b09c1">mshadow::red::limits::MaxValue< bool ></a></div><div class="ttdeci">MSHADOW_XINLINE bool MaxValue< bool >(void)</div><div class="ttdoc">maximum value of bool</div><div class="ttdef"><b>Definition:</b> base.h:946</div></div> |
| <div class="ttc" id="anamespacemshadow_html_aa94c153a2f8bcd10be82a4f5598ae5b9"><div class="ttname"><a href="namespacemshadow.html#aa94c153a2f8bcd10be82a4f5598ae5b9">mshadow::openmp_index_t</a></div><div class="ttdeci">index_t openmp_index_t</div><div class="ttdoc">openmp index for linux</div><div class="ttdef"><b>Definition:</b> base.h:336</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a5ada516b5bd58791f170586cfdd9ae2f"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a5ada516b5bd58791f170586cfdd9ae2f">mshadow::red::limits::MinValue< uint8_t ></a></div><div class="ttdeci">MSHADOW_XINLINE uint8_t MinValue< uint8_t >(void)</div><div class="ttdoc">minimum value of uint8_t</div><div class="ttdef"><b>Definition:</b> base.h:839</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1mul_html_a38a7497e4f710d7da3f5b9563d754b21"><div class="ttname"><a href="structmshadow_1_1op_1_1mul.html#a38a7497e4f710d7da3f5b9563d754b21">mshadow::op::mul::Map</a></div><div class="ttdeci">static MSHADOW_XINLINE DType Map(DType a, DType b)</div><div class="ttdoc">map a, b to result using defined operation</div><div class="ttdef"><b>Definition:</b> base.h:628</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a143ad1a73ba518ef0aad29d36f2569ee"><div class="ttname"><a href="namespacemshadow.html#a143ad1a73ba518ef0aad29d36f2569ee">mshadow::default_type_flag</a></div><div class="ttdeci">const int default_type_flag</div><div class="ttdoc">type enum value for default real type</div><div class="ttdef"><b>Definition:</b> base.h:492</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a6f5613dfb9c095420b5fe322aa2ab1ed"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a6f5613dfb9c095420b5fe322aa2ab1ed">mshadow::red::maximum::SetInitValue</a></div><div class="ttdeci">static MSHADOW_XINLINE void SetInitValue(DType &initv, DType &none)</div><div class="ttdoc">set the initial value during reduction</div><div class="ttdef"><b>Definition:</b> base.h:1103</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a77b7882d6cafca4b12908e2fff2302ed"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a77b7882d6cafca4b12908e2fff2302ed">mshadow::red::limits::PosInfValue< float ></a></div><div class="ttdeci">MSHADOW_XINLINE float PosInfValue< float >(void)</div><div class="ttdoc">positive infinity value of float</div><div class="ttdef"><b>Definition:</b> base.h:965</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a853288dcb013f16211eb9ce107ccba3e"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a853288dcb013f16211eb9ce107ccba3e">mshadow::red::limits::MaxValue< uint8_t ></a></div><div class="ttdeci">MSHADOW_XINLINE uint8_t MaxValue< uint8_t >(void)</div><div class="ttdoc">maximum value of uint8_t</div><div class="ttdef"><b>Definition:</b> base.h:926</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908">mshadow::kUint16</a></div><div class="ttdeci">@ kUint16</div><div class="ttdef"><b>Definition:</b> base.h:361</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade">mshadow::kUint64</a></div><div class="ttdeci">@ kUint64</div><div class="ttdef"><b>Definition:</b> base.h:363</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4a3f4908b910c47da082d60aa719c43fc3"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a3f4908b910c47da082d60aa719c43fc3">mshadow::kNWC</a></div><div class="ttdeci">@ kNWC</div><div class="ttdef"><b>Definition:</b> base.h:506</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4aa332dc48875dac0ef29ef245e5609822">mshadow::kNCDHW</a></div><div class="ttdeci">@ kNCDHW</div><div class="ttdef"><b>Definition:</b> base.h:509</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a1bffb78fd1c64479b82e9a4ffccab69e"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a1bffb78fd1c64479b82e9a4ffccab69e">mshadow::red::maximum::SetInitValue</a></div><div class="ttdeci">static MSHADOW_XINLINE void SetInitValue(DType &initv)</div><div class="ttdoc">set the initial value during reduction</div><div class="ttdef"><b>Definition:</b> base.h:1096</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1minusto_html_a50bab98db98ee42cd377f2e2f02e37ca"><div class="ttname"><a href="structmshadow_1_1sv_1_1minusto.html#a50bab98db98ee42cd377f2e2f02e37ca">mshadow::sv::minusto::Save</a></div><div class="ttdeci">static MSHADOW_XINLINE void Save(DType &a, DType b)</div><div class="ttdoc">save b to a using save method</div><div class="ttdef"><b>Definition:</b> base.h:711</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_affcf917b2d5206e3ddeec7e5f0ee995e"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#affcf917b2d5206e3ddeec7e5f0ee995e">mshadow::red::limits::MaxValue< int8_t ></a></div><div class="ttdeci">MSHADOW_XINLINE int8_t MaxValue< int8_t >(void)</div><div class="ttdoc">maximum value of int8_t</div><div class="ttdef"><b>Definition:</b> base.h:931</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1divto_html_ab8864c99eea0286fce8f5a8f791f0f33"><div class="ttname"><a href="structmshadow_1_1sv_1_1divto.html#ab8864c99eea0286fce8f5a8f791f0f33">mshadow::sv::divto::OPType</a></div><div class="ttdeci">op::div OPType</div><div class="ttdoc">corresponding binary operator type</div><div class="ttdef"><b>Definition:</b> base.h:739</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a232631c75acb76794420d172a5cc335e"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a232631c75acb76794420d172a5cc335e">mshadow::red::limits::MinValue< float ></a></div><div class="ttdeci">MSHADOW_XINLINE float MinValue< float >(void)</div><div class="ttdoc">minimum value of float</div><div class="ttdef"><b>Definition:</b> base.h:819</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a88b427b9289c2c00736887449ed67d0a"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a88b427b9289c2c00736887449ed67d0a">mshadow::red::maximum::Merge</a></div><div class="ttdeci">static MSHADOW_XINLINE void Merge(volatile DType &dst_val, volatile DType &dst_residual, volatile DType &src_val, volatile DType &src_residual)</div><div class="ttdoc">combine the results of two reducers</div><div class="ttdef"><b>Definition:</b> base.h:1075</div></div> |
| <div class="ttc" id="astructmshadow_1_1LayoutType_html"><div class="ttname"><a href="structmshadow_1_1LayoutType.html">mshadow::LayoutType</a></div><div class="ttdef"><b>Definition:</b> base.h:573</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c">mshadow::kInt8</a></div><div class="ttdeci">@ kInt8</div><div class="ttdef"><b>Definition:</b> base.h:357</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7">mshadow::kUint32</a></div><div class="ttdeci">@ kUint32</div><div class="ttdef"><b>Definition:</b> base.h:362</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1multo_html_acc47e9c85d6e583e5785ba0f62fa4431"><div class="ttname"><a href="structmshadow_1_1sv_1_1multo.html#acc47e9c85d6e583e5785ba0f62fa4431">mshadow::sv::multo::OPType</a></div><div class="ttdeci">op::mul OPType</div><div class="ttdoc">corresponding binary operator type</div><div class="ttdef"><b>Definition:</b> base.h:729</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1minus_html"><div class="ttname"><a href="structmshadow_1_1op_1_1minus.html">mshadow::op::minus</a></div><div class="ttdoc">minus operator</div><div class="ttdef"><b>Definition:</b> base.h:641</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a99c8021ef1da714c052bfdf1f4e41280"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a99c8021ef1da714c052bfdf1f4e41280">mshadow::red::maximum::Finalize</a></div><div class="ttdeci">static MSHADOW_XINLINE void Finalize(volatile DType &dst, volatile DType &residual)</div><div class="ttdoc">finalize reduction</div><div class="ttdef"><b>Definition:</b> base.h:1083</div></div> |
| <div class="ttc" id="abfloat_8h_html_a3d5d63ecb7104078d465a0074f9d9059"><div class="ttname"><a href="bfloat_8h.html#a3d5d63ecb7104078d465a0074f9d9059">MSHADOW_BF16_MAX</a></div><div class="ttdeci">#define MSHADOW_BF16_MAX</div><div class="ttdef"><b>Definition:</b> bfloat.h:182</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a2744c70ac323798d108e5b99ed9875c6"><div class="ttname"><a href="namespacemshadow.html#a2744c70ac323798d108e5b99ed9875c6">mshadow::toString</a></div><div class="ttdeci">std::string toString(LayoutFlag layout)</div><div class="ttdef"><b>Definition:</b> base.h:545</div></div> |
| <div class="ttc" id="a3rdparty_2mshadow_2mshadow_2base_8h_html_af205b8ab4d8733133d590df7d2639d69"><div class="ttname"><a href="3rdparty_2mshadow_2mshadow_2base_8h.html#af205b8ab4d8733133d590df7d2639d69">MSHADOW_XINLINE</a></div><div class="ttdeci">#define MSHADOW_XINLINE</div><div class="ttdef"><b>Definition:</b> base.h:228</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_abdbdd852a7eb8460458f24fea8ca7714"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#abdbdd852a7eb8460458f24fea8ca7714">mshadow::red::sum::Merge</a></div><div class="ttdeci">static MSHADOW_XINLINE void Merge(volatile DType &dst_val, volatile DType &dst_residual, volatile DType &src_val, volatile DType &src_residual)</div><div class="ttdoc">combine the results of two reducers</div><div class="ttdef"><b>Definition:</b> base.h:1012</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1saveto_html_aaf49b727bb5fe9098081c265100cb340"><div class="ttname"><a href="structmshadow_1_1sv_1_1saveto.html#aaf49b727bb5fe9098081c265100cb340">mshadow::sv::saveto::AlphaBLAS</a></div><div class="ttdeci">static default_real_t AlphaBLAS(void)</div><div class="ttdoc">helper constant to use BLAS, alpha</div><div class="ttdef"><b>Definition:</b> base.h:687</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1identity_html_a260f8047dca47cb16ff0adaf16f5e8e6"><div class="ttname"><a href="structmshadow_1_1op_1_1identity.html#a260f8047dca47cb16ff0adaf16f5e8e6">mshadow::op::identity::Map</a></div><div class="ttdeci">static MSHADOW_XINLINE DType Map(DType a)</div><div class="ttdoc">map a to result using defined operation</div><div class="ttdef"><b>Definition:</b> base.h:672</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1isnan__typed_html_a31da61266fae8c9ca52df015d132b759"><div class="ttname"><a href="namespacemshadow_1_1isnan__typed.html#a31da61266fae8c9ca52df015d132b759">mshadow::isnan_typed::IsNan</a></div><div class="ttdeci">MSHADOW_XINLINE bool IsNan(volatile DType val)</div><div class="ttdef"><b>Definition:</b> base.h:753</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1plusto_html_a916ea217bd0200ab7b23a4e2ac26887f"><div class="ttname"><a href="structmshadow_1_1sv_1_1plusto.html#a916ea217bd0200ab7b23a4e2ac26887f">mshadow::sv::plusto::BetaBLAS</a></div><div class="ttdeci">static default_real_t BetaBLAS(void)</div><div class="ttdoc">helper constant to use BLAS, beta</div><div class="ttdef"><b>Definition:</b> base.h:703</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1div_html"><div class="ttname"><a href="structmshadow_1_1op_1_1div.html">mshadow::op::div</a></div><div class="ttdoc">divide operator</div><div class="ttdef"><b>Definition:</b> base.h:649</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1saveto_html_af99e9c2ec6d116febb41fc73d201d80e"><div class="ttname"><a href="structmshadow_1_1sv_1_1saveto.html#af99e9c2ec6d116febb41fc73d201d80e">mshadow::sv::saveto::Save</a></div><div class="ttdeci">static MSHADOW_XINLINE void Save(DType &a, DType b)</div><div class="ttdoc">save b to a using save method</div><div class="ttdef"><b>Definition:</b> base.h:683</div></div> |
| <div class="ttc" id="abfloat_8h_html_a9ec28a1a9aa829211bb9963eb967c245"><div class="ttname"><a href="bfloat_8h.html#a9ec28a1a9aa829211bb9963eb967c245">MSHADOW_BF16_SIGN_BIT</a></div><div class="ttdeci">#define MSHADOW_BF16_SIGN_BIT</div><div class="ttdef"><b>Definition:</b> bfloat.h:183</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html">mshadow::red::sum</a></div><div class="ttdoc">sum reducer</div><div class="ttdef"><b>Definition:</b> base.h:987</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1minusto_html_a48fae34ea7d461ced6ac9c98c5e38019"><div class="ttname"><a href="structmshadow_1_1sv_1_1minusto.html#a48fae34ea7d461ced6ac9c98c5e38019">mshadow::sv::minusto::BetaBLAS</a></div><div class="ttdeci">static default_real_t BetaBLAS(void)</div><div class="ttdoc">helper constant to use BLAS, beta</div><div class="ttdef"><b>Definition:</b> base.h:717</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a97c5d96eac2849f458678742738f1ce0"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a97c5d96eac2849f458678742738f1ce0">mshadow::red::maximum::Finalize</a></div><div class="ttdeci">static MSHADOW_XINLINE void Finalize(volatile DType &dst)</div><div class="ttdoc">finalize reduction</div><div class="ttdef"><b>Definition:</b> base.h:1080</div></div> |
| <div class="ttc" id="abfloat_8h_html_ad3e1c33244a89e2dd8a1f5697c10ff7b"><div class="ttname"><a href="bfloat_8h.html#ad3e1c33244a89e2dd8a1f5697c10ff7b">MSHADOW_BF16_EXPONENT_BITS</a></div><div class="ttdeci">#define MSHADOW_BF16_EXPONENT_BITS</div><div class="ttdef"><b>Definition:</b> bfloat.h:184</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1divto_html_aaa2299b2a710d1b5d61b6dc6f44c9339"><div class="ttname"><a href="structmshadow_1_1sv_1_1divto.html#aaa2299b2a710d1b5d61b6dc6f44c9339">mshadow::sv::divto::Save</a></div><div class="ttdeci">static MSHADOW_XINLINE void Save(DType &a, DType b)</div><div class="ttdoc">save b to a using save method</div><div class="ttdef"><b>Definition:</b> base.h:735</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a36590a455cab713cf4bbb115c4dba247"><div class="ttname"><a href="namespacemshadow.html#a36590a455cab713cf4bbb115c4dba247">mshadow::default_layout</a></div><div class="ttdeci">const int default_layout</div><div class="ttdoc">default layout for 4d tensor</div><div class="ttdef"><b>Definition:</b> base.h:596</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1multo_html_ab3b6328d24ef3ce0cf5c0329e254f193"><div class="ttname"><a href="structmshadow_1_1sv_1_1multo.html#ab3b6328d24ef3ce0cf5c0329e254f193">mshadow::sv::multo::Save</a></div><div class="ttdeci">static MSHADOW_XINLINE void Save(DType &a, DType b)</div><div class="ttdoc">save b to a using save method</div><div class="ttdef"><b>Definition:</b> base.h:725</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_a01e4946a88167d905c14b39d3965e03c"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#a01e4946a88167d905c14b39d3965e03c">mshadow::red::minimum::Reduce</a></div><div class="ttdeci">static MSHADOW_XINLINE void Reduce(volatile DType &dst, volatile DType src, volatile DType &none)</div><div class="ttdoc">do reduction into dst</div><div class="ttdef"><b>Definition:</b> base.h:1118</div></div> |
| <div class="ttc" id="anamespacemxnet_1_1common_1_1cuda_1_1rtc_html_a09828bc99a0f05955a61b9e4feabb9b8"><div class="ttname"><a href="namespacemxnet_1_1common_1_1cuda_1_1rtc.html#a09828bc99a0f05955a61b9e4feabb9b8">mxnet::common::cuda::rtc::limits</a></div><div class="ttdeci">const char limits[]</div><div class="ttdef"><b>Definition:</b> util-inl.h:594</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_ad5c922e935e2efb47bb238c56ff8a38b"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#ad5c922e935e2efb47bb238c56ff8a38b">mshadow::red::limits::MaxValue< int32_t ></a></div><div class="ttdeci">MSHADOW_XINLINE int MaxValue< int32_t >(void)</div><div class="ttdoc">maximum value of int32_t</div><div class="ttdef"><b>Definition:</b> base.h:936</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a">mshadow::kBool</a></div><div class="ttdeci">@ kBool</div><div class="ttdef"><b>Definition:</b> base.h:359</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a55d9bb33517b6c82d04555acba79680a"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a55d9bb33517b6c82d04555acba79680a">mshadow::red::limits::MinValue< int8_t ></a></div><div class="ttdeci">MSHADOW_XINLINE int8_t MinValue< int8_t >(void)</div><div class="ttdoc">minimum value of int8_t</div><div class="ttdef"><b>Definition:</b> base.h:844</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1saveto_html_a4dab404c405fe7476eadbd02e727738b"><div class="ttname"><a href="structmshadow_1_1sv_1_1saveto.html#a4dab404c405fe7476eadbd02e727738b">mshadow::sv::saveto::BetaBLAS</a></div><div class="ttdeci">static default_real_t BetaBLAS(void)</div><div class="ttdoc">helper constant to use BLAS, beta</div><div class="ttdef"><b>Definition:</b> base.h:689</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ad9ef54b427a838279a62e95204aa3281">mshadow::kNHWC</a></div><div class="ttdeci">@ kNHWC</div><div class="ttdef"><b>Definition:</b> base.h:502</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a29fa60cc6436415a699971c1a3fd760a"><div class="ttname"><a href="namespacemshadow.html#a29fa60cc6436415a699971c1a3fd760a">mshadow::default_layout_5d</a></div><div class="ttdeci">const int default_layout_5d</div><div class="ttdoc">default layout for 5d tensor</div><div class="ttdef"><b>Definition:</b> base.h:619</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a99b1074ecd03f849cbdcfbad8df8ed2b"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a99b1074ecd03f849cbdcfbad8df8ed2b">mshadow::red::maximum::Reduce</a></div><div class="ttdeci">static MSHADOW_XINLINE void Reduce(volatile DType &dst, volatile DType src)</div><div class="ttdoc">do reduction into dst</div><div class="ttdef"><b>Definition:</b> base.h:1058</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4ae5189456ea58c60d8c546ab8bd1a2c08"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4ae5189456ea58c60d8c546ab8bd1a2c08">mshadow::kCWN</a></div><div class="ttdeci">@ kCWN</div><div class="ttdef"><b>Definition:</b> base.h:507</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4a6e334c4df779fcc6fa6b24411fce1289"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a6e334c4df779fcc6fa6b24411fce1289">mshadow::kCHWN</a></div><div class="ttdeci">@ kCHWN</div><div class="ttdef"><b>Definition:</b> base.h:503</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28">mshadow::kFloat64</a></div><div class="ttdeci">@ kFloat64</div><div class="ttdef"><b>Definition:</b> base.h:353</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1plus_html"><div class="ttname"><a href="structmshadow_1_1op_1_1plus.html">mshadow::op::plus</a></div><div class="ttdoc">plus operator</div><div class="ttdef"><b>Definition:</b> base.h:633</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1plusto_html_a9acc887f74cce8e18f1a86818154ca1c"><div class="ttname"><a href="structmshadow_1_1sv_1_1plusto.html#a9acc887f74cce8e18f1a86818154ca1c">mshadow::sv::plusto::AlphaBLAS</a></div><div class="ttdeci">static default_real_t AlphaBLAS(void)</div><div class="ttdoc">helper constant to use BLAS, alpha</div><div class="ttdef"><b>Definition:</b> base.h:701</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_aa05b8713dd5973cac465420e52f67dda"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#aa05b8713dd5973cac465420e52f67dda">mshadow::red::sum::Merge</a></div><div class="ttdeci">static MSHADOW_XINLINE void Merge(volatile DType &dst_val, volatile DType &src_val)</div><div class="ttdoc">combine the results of two reducers</div><div class="ttdef"><b>Definition:</b> base.h:1007</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a34f7a97ddbac1b447e39603de7c6f965"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a34f7a97ddbac1b447e39603de7c6f965">mshadow::red::limits::MinValue< double ></a></div><div class="ttdeci">MSHADOW_XINLINE double MinValue< double >(void)</div><div class="ttdoc">minimum value of double</div><div class="ttdef"><b>Definition:</b> base.h:824</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1plusto_html"><div class="ttname"><a href="structmshadow_1_1sv_1_1plusto.html">mshadow::sv::plusto</a></div><div class="ttdoc">save to saver: +=</div><div class="ttdef"><b>Definition:</b> base.h:694</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4">mshadow::LayoutFlag</a></div><div class="ttdeci">LayoutFlag</div><div class="ttdef"><b>Definition:</b> base.h:498</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1plus_html_a83fb5b1656992c6fd442a819de83b292"><div class="ttname"><a href="structmshadow_1_1op_1_1plus.html#a83fb5b1656992c6fd442a819de83b292">mshadow::op::plus::Map</a></div><div class="ttdeci">static MSHADOW_XINLINE DType Map(DType a, DType b)</div><div class="ttdoc">map a, b to result using defined operation</div><div class="ttdef"><b>Definition:</b> base.h:636</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_abd56e307635333d02fc0f22726dbed28"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#abd56e307635333d02fc0f22726dbed28">mshadow::red::sum::Reduce</a></div><div class="ttdeci">static MSHADOW_XINLINE void Reduce(volatile DType &dst, volatile DType src)</div><div class="ttdoc">do reduction into dst</div><div class="ttdef"><b>Definition:</b> base.h:990</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17">mshadow::kInt16</a></div><div class="ttdeci">@ kInt16</div><div class="ttdef"><b>Definition:</b> base.h:360</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_aa3f7b2a1f96f77caca258e8c594f67bd"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#aa3f7b2a1f96f77caca258e8c594f67bd">mshadow::red::minimum::PartialGrad</a></div><div class="ttdeci">static MSHADOW_XINLINE DType PartialGrad(DType redres, DType redsrc)</div><div class="ttdoc">calculate gradient of redres with respect to redsrc, redres: reduced result, redsrc: one of reduction...</div><div class="ttdef"><b>Definition:</b> base.h:1142</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a64ef2d1a92579d312b18fb04aaed0d4e"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a64ef2d1a92579d312b18fb04aaed0d4e">mshadow::red::limits::PosInfValue</a></div><div class="ttdeci">MSHADOW_XINLINE DType PosInfValue(void)</div><div class="ttdoc">positive infinity of certain types</div><div class="ttdef"><b>Definition:</b> base.h:960</div></div> |
| <div class="ttc" id="ahalf_8h_html_acd26d7558448d4ce6889283913627b31"><div class="ttname"><a href="half_8h.html#acd26d7558448d4ce6889283913627b31">MSHADOW_HALF_MAX</a></div><div class="ttdeci">#define MSHADOW_HALF_MAX</div><div class="ttdef"><b>Definition:</b> half.h:369</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_a9485035d020c53f31eadf7e34adc37f0"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#a9485035d020c53f31eadf7e34adc37f0">mshadow::red::minimum::Merge</a></div><div class="ttdeci">static MSHADOW_XINLINE void Merge(volatile DType &dst_val, volatile DType &src_val)</div><div class="ttdoc">combine the results of two reducers</div><div class="ttdef"><b>Definition:</b> base.h:1123</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1saveto_html"><div class="ttname"><a href="structmshadow_1_1sv_1_1saveto.html">mshadow::sv::saveto</a></div><div class="ttdoc">save to saver: =</div><div class="ttdef"><b>Definition:</b> base.h:680</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_a511c1d4add93f26810e6a53c0ff106d0"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#a511c1d4add93f26810e6a53c0ff106d0">mshadow::red::sum::Finalize</a></div><div class="ttdeci">static MSHADOW_XINLINE void Finalize(volatile DType &dst)</div><div class="ttdoc">finalize reduction</div><div class="ttdef"><b>Definition:</b> base.h:1026</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4a30ea75730b725ac965fb4e43c34622d8"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a30ea75730b725ac965fb4e43c34622d8">mshadow::kCDHWN</a></div><div class="ttdeci">@ kCDHWN</div><div class="ttdef"><b>Definition:</b> base.h:511</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a3784eade98f040fe074543259fe7164e"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a3784eade98f040fe074543259fe7164e">mshadow::red::limits::NegInfValue< double ></a></div><div class="ttdeci">MSHADOW_XINLINE double NegInfValue< double >(void)</div><div class="ttdoc">negative infinity value of double</div><div class="ttdef"><b>Definition:</b> base.h:883</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1right_html"><div class="ttname"><a href="structmshadow_1_1op_1_1right.html">mshadow::op::right</a></div><div class="ttdoc">get rhs</div><div class="ttdef"><b>Definition:</b> base.h:657</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a5244fd39689a5526efa712f85579b148"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a5244fd39689a5526efa712f85579b148">mshadow::red::limits::MinValue</a></div><div class="ttdeci">MSHADOW_XINLINE DType MinValue(void)</div><div class="ttdoc">minimum value of certain types</div></div> |
| <div class="ttc" id="anamespacemshadow_html_ac4cba6b672478463350f7d84b47e99e3"><div class="ttname"><a href="namespacemshadow.html#ac4cba6b672478463350f7d84b47e99e3">mshadow::kRandBufferSize</a></div><div class="ttdeci">const unsigned kRandBufferSize</div><div class="ttdoc">buffer size for each random number generator</div><div class="ttdef"><b>Definition:</b> base.h:321</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1isinf__typed_html_a5208c13b479de37653fc08402f08e56d"><div class="ttname"><a href="namespacemshadow_1_1isinf__typed.html#a5208c13b479de37653fc08402f08e56d">mshadow::isinf_typed::IsInf</a></div><div class="ttdeci">MSHADOW_XINLINE bool IsInf(volatile DType val)</div><div class="ttdef"><b>Definition:</b> base.h:783</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1minus_html_a29e530a31da95225c916e13f1bbcbca6"><div class="ttname"><a href="structmshadow_1_1op_1_1minus.html#a29e530a31da95225c916e13f1bbcbca6">mshadow::op::minus::Map</a></div><div class="ttdeci">static MSHADOW_XINLINE DType Map(DType a, DType b)</div><div class="ttdoc">map a, b to result using defined operation</div><div class="ttdef"><b>Definition:</b> base.h:644</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a5c74a9cf04436607f101428cb8fd4326"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a5c74a9cf04436607f101428cb8fd4326">mshadow::red::maximum::Merge</a></div><div class="ttdeci">static MSHADOW_XINLINE void Merge(volatile DType &dst_val, volatile DType &src_val)</div><div class="ttdoc">combine the results of two reducers</div><div class="ttdef"><b>Definition:</b> base.h:1070</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1multo_html"><div class="ttname"><a href="structmshadow_1_1sv_1_1multo.html">mshadow::sv::multo</a></div><div class="ttdoc">multiply to saver: *=</div><div class="ttdef"><b>Definition:</b> base.h:722</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23">mshadow::kInt64</a></div><div class="ttdeci">@ kInt64</div><div class="ttdef"><b>Definition:</b> base.h:358</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_a351d0e761cb9944c4d98919535754142"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#a351d0e761cb9944c4d98919535754142">mshadow::red::sum::PartialGrad</a></div><div class="ttdeci">static MSHADOW_XINLINE DType PartialGrad(DType redres, DType redsrc)</div><div class="ttdoc">calculate gradient of redres with respect to redsrc, redres: reduced result, redsrc: one of reduction...</div><div class="ttdef"><b>Definition:</b> base.h:1035</div></div> |
| <div class="ttc" id="anamespacemshadow_html_adcbc2e1131386fccb1474b0bdf045926"><div class="ttname"><a href="namespacemshadow.html#adcbc2e1131386fccb1474b0bdf045926">mshadow::index_t</a></div><div class="ttdeci">int32_t index_t</div><div class="ttdoc">type that will be used for index</div><div class="ttdef"><b>Definition:</b> base.h:328</div></div> |
| <div class="ttc" id="ahalf_8h_html_a4068a67a48bb294a6bf436e28be90de9"><div class="ttname"><a href="half_8h.html#a4068a67a48bb294a6bf436e28be90de9">MSHADOW_HALF_SIGN_BIT</a></div><div class="ttdeci">#define MSHADOW_HALF_SIGN_BIT</div><div class="ttdef"><b>Definition:</b> half.h:370</div></div> |
| <div class="ttc" id="astructmshadow_1_1DataType_html"><div class="ttname"><a href="structmshadow_1_1DataType.html">mshadow::DataType</a></div><div class="ttdef"><b>Definition:</b> base.h:368</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_aa000adb90fad1fb429a05b1c6c8b836b"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#aa000adb90fad1fb429a05b1c6c8b836b">mshadow::red::minimum::Reduce</a></div><div class="ttdeci">static MSHADOW_XINLINE void Reduce(volatile DType &dst, volatile DType src)</div><div class="ttdoc">do reduction into dst</div><div class="ttdef"><b>Definition:</b> base.h:1111</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a8afab1f2134d65e09679faa324ecf9ce"><div class="ttname"><a href="namespacemshadow.html#a8afab1f2134d65e09679faa324ecf9ce">mshadow::kPi</a></div><div class="ttdeci">const float kPi</div><div class="ttdoc">pi</div><div class="ttdef"><b>Definition:</b> base.h:323</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368">mshadow::kInt32</a></div><div class="ttdeci">@ kInt32</div><div class="ttdef"><b>Definition:</b> base.h:356</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a1adf2b0a158ad07a93b21f11e6aa3150"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a1adf2b0a158ad07a93b21f11e6aa3150">mshadow::red::limits::MinValue< bool ></a></div><div class="ttdeci">MSHADOW_XINLINE bool MinValue< bool >(void)</div><div class="ttdoc">minimum value of bool</div><div class="ttdef"><b>Definition:</b> base.h:859</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1">mshadow::TypeFlag</a></div><div class="ttdeci">TypeFlag</div><div class="ttdoc">data type flag</div><div class="ttdef"><b>Definition:</b> base.h:351</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a8b60ab0b7ba58978695b4ab7e63268b9"><div class="ttname"><a href="namespacemshadow.html#a8b60ab0b7ba58978695b4ab7e63268b9">mshadow::lapack_index_t</a></div><div class="ttdeci">int lapack_index_t</div><div class="ttdef"><b>Definition:</b> base.h:344</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a56182d9842b841740da7eefb60916cf9"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a56182d9842b841740da7eefb60916cf9">mshadow::red::limits::MinValue< int64_t ></a></div><div class="ttdeci">MSHADOW_XINLINE int64_t MinValue< int64_t >(void)</div><div class="ttdoc">minimum value of int64_t</div><div class="ttdef"><b>Definition:</b> base.h:854</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a4fdc4629369e455063f9d084570f276a"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a4fdc4629369e455063f9d084570f276a">mshadow::red::limits::MaxValue< double ></a></div><div class="ttdeci">MSHADOW_XINLINE double MaxValue< double >(void)</div><div class="ttdoc">maximum value of double</div><div class="ttdef"><b>Definition:</b> base.h:911</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_aedc5b032275b8d11fc95fae2545cc2cb"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#aedc5b032275b8d11fc95fae2545cc2cb">mshadow::red::sum::SetInitValue</a></div><div class="ttdeci">static MSHADOW_XINLINE void SetInitValue(DType &initv)</div><div class="ttdoc">set the initial value during reduction</div><div class="ttdef"><b>Definition:</b> base.h:1042</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a2e0671a4ac1729d7f8996ba091f8e07c"><div class="ttname"><a href="namespacemshadow.html#a2e0671a4ac1729d7f8996ba091f8e07c">mshadow::mshadow_sizeof</a></div><div class="ttdeci">size_t mshadow_sizeof(int type)</div><div class="ttdoc">get data type size from type enum</div><div class="ttdef"><b>Definition:</b> base.h:1804</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a37c45177ee0fe2468d562a9e5e3e38dd">mshadow::kNDHWC</a></div><div class="ttdeci">@ kNDHWC</div><div class="ttdef"><b>Definition:</b> base.h:510</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1plusto_html_a0b3849b290f3207a59b8975e2b29b065"><div class="ttname"><a href="structmshadow_1_1sv_1_1plusto.html#a0b3849b290f3207a59b8975e2b29b065">mshadow::sv::plusto::OPType</a></div><div class="ttdeci">op::plus OPType</div><div class="ttdoc">corresponding binary operator type</div><div class="ttdef"><b>Definition:</b> base.h:705</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_a96aaae5db92ec7a3a8a5740265394a35"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#a96aaae5db92ec7a3a8a5740265394a35">mshadow::red::maximum::Reduce</a></div><div class="ttdeci">static MSHADOW_XINLINE void Reduce(volatile DType &dst, volatile DType src, volatile DType &none)</div><div class="ttdoc">do reduction into dst</div><div class="ttdef"><b>Definition:</b> base.h:1065</div></div> |
| <div class="ttc" id="abfloat_8h_html_ab31d5d1e2e21e0fc5dd6f40e276fdd43"><div class="ttname"><a href="bfloat_8h.html#ab31d5d1e2e21e0fc5dd6f40e276fdd43">MSHADOW_BF16_MIN</a></div><div class="ttdeci">#define MSHADOW_BF16_MIN</div><div class="ttdoc">overloaded + operator for bf16_t</div><div class="ttdef"><b>Definition:</b> bfloat.h:181</div></div> |
| <div class="ttc" id="anamespacemshadow_html_af5aa21f20dbeaa38194b4dadbd37a072"><div class="ttname"><a href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072">mshadow::dtype_string</a></div><div class="ttdeci">std::string dtype_string(const int dtype)</div><div class="ttdef"><b>Definition:</b> base.h:1811</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_a351e35f2cbf0f8e7e08c5f23b70f3c3f"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#a351e35f2cbf0f8e7e08c5f23b70f3c3f">mshadow::red::sum::SetInitValue</a></div><div class="ttdeci">static MSHADOW_XINLINE void SetInitValue(DType &initv, DType &residual)</div><div class="ttdoc">set the initial value during reduction</div><div class="ttdef"><b>Definition:</b> base.h:1049</div></div> |
| <div class="ttc" id="anamespacemshadow_html"><div class="ttname"><a href="namespacemshadow.html">mshadow</a></div><div class="ttdoc">overloaded + operator between half_t and bf16_t</div><div class="ttdef"><b>Definition:</b> base.h:319</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a6262f9d4a98d7ddc205cb0b19af4a316"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a6262f9d4a98d7ddc205cb0b19af4a316">mshadow::red::limits::MaxValue< float ></a></div><div class="ttdeci">MSHADOW_XINLINE float MaxValue< float >(void)</div><div class="ttdoc">maximum value of float</div><div class="ttdef"><b>Definition:</b> base.h:906</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a10a73dbdb4a2b833495c2315b6da69ce"><div class="ttname"><a href="namespacemshadow.html#a10a73dbdb4a2b833495c2315b6da69ce">mshadow::default_real_t</a></div><div class="ttdeci">float default_real_t</div><div class="ttdoc">float point type that will be used in default by mshadow</div><div class="ttdef"><b>Definition:</b> base.h:348</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1identity_html"><div class="ttname"><a href="structmshadow_1_1op_1_1identity.html">mshadow::op::identity</a></div><div class="ttdoc">identity function that maps a real number to it self</div><div class="ttdef"><b>Definition:</b> base.h:669</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1minusto_html_ac9baef2b9d3a92b962c9ad7f7ee1122d"><div class="ttname"><a href="structmshadow_1_1sv_1_1minusto.html#ac9baef2b9d3a92b962c9ad7f7ee1122d">mshadow::sv::minusto::OPType</a></div><div class="ttdeci">op::minus OPType</div><div class="ttdoc">corresponding binary operator type</div><div class="ttdef"><b>Definition:</b> base.h:719</div></div> |
| <div class="ttc" id="ahalf_8h_html"><div class="ttname"><a href="half_8h.html">half.h</a></div><div class="ttdoc">definition of half (float16) type.</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1minusto_html"><div class="ttname"><a href="structmshadow_1_1sv_1_1minusto.html">mshadow::sv::minusto</a></div><div class="ttdoc">minus to saver: -=</div><div class="ttdef"><b>Definition:</b> base.h:708</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a5bbd7816d59801fc6044f1b9559bb64c">mshadow::kUNKNOWN</a></div><div class="ttdeci">@ kUNKNOWN</div><div class="ttdef"><b>Definition:</b> base.h:499</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_acbda7fe80fa3754a57b967e11a8ee44f"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#acbda7fe80fa3754a57b967e11a8ee44f">mshadow::red::minimum::SetInitValue</a></div><div class="ttdeci">static MSHADOW_XINLINE void SetInitValue(DType &initv)</div><div class="ttdoc">set the initial value during reduction</div><div class="ttdef"><b>Definition:</b> base.h:1149</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_aebd011f117a63fea3cd494ee2138b634"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#aebd011f117a63fea3cd494ee2138b634">mshadow::red::minimum::Finalize</a></div><div class="ttdeci">static MSHADOW_XINLINE void Finalize(volatile DType &dst)</div><div class="ttdoc">finalize reduction</div><div class="ttdef"><b>Definition:</b> base.h:1133</div></div> |
| <div class="ttc" id="a3rdparty_2mshadow_2mshadow_2base_8h_html_adce833400b796ef18899f74d3ab301b3"><div class="ttname"><a href="3rdparty_2mshadow_2mshadow_2base_8h.html#adce833400b796ef18899f74d3ab301b3">MSHADOW_TYPE_SWITCH_EXT_WITH_BOOL</a></div><div class="ttdeci">#define MSHADOW_TYPE_SWITCH_EXT_WITH_BOOL(type, DType,...)</div><div class="ttdef"><b>Definition:</b> base.h:1719</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_ab4d32d18fb83ed9fe3b2773e0bf8d508"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#ab4d32d18fb83ed9fe3b2773e0bf8d508">mshadow::red::limits::MaxValue< int64_t ></a></div><div class="ttdeci">MSHADOW_XINLINE int64_t MaxValue< int64_t >(void)</div><div class="ttdoc">maximum value of int64_t</div><div class="ttdef"><b>Definition:</b> base.h:941</div></div> |
| <div class="ttc" id="ahalf_8h_html_a6961c70512f3780000b1ee5691363da3"><div class="ttname"><a href="half_8h.html#a6961c70512f3780000b1ee5691363da3">MSHADOW_HALF_MIN</a></div><div class="ttdeci">#define MSHADOW_HALF_MIN</div><div class="ttdoc">overloaded + operator for half_t</div><div class="ttdef"><b>Definition:</b> half.h:368</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html_aba3be24837548e8f5a48178889ec2132"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html#aba3be24837548e8f5a48178889ec2132">mshadow::red::minimum::SetInitValue</a></div><div class="ttdeci">static MSHADOW_XINLINE void SetInitValue(DType &initv, DType &none)</div><div class="ttdoc">set the initial value during reduction</div><div class="ttdef"><b>Definition:</b> base.h:1156</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1plusto_html_ac7556832c791c8d048cbb7fd606f6e8e"><div class="ttname"><a href="structmshadow_1_1sv_1_1plusto.html#ac7556832c791c8d048cbb7fd606f6e8e">mshadow::sv::plusto::Save</a></div><div class="ttdeci">static MSHADOW_XINLINE void Save(DType &a, DType b)</div><div class="ttdoc">save b to a using save method</div><div class="ttdef"><b>Definition:</b> base.h:697</div></div> |
| <div class="ttc" id="a3rdparty_2mshadow_2mshadow_2base_8h_html_a5260fbd063fdb83af6cff734306ce856"><div class="ttname"><a href="3rdparty_2mshadow_2mshadow_2base_8h.html#a5260fbd063fdb83af6cff734306ce856">MSHADOW_HALF_BF_OPERATOR</a></div><div class="ttdeci">#define MSHADOW_HALF_BF_OPERATOR(RTYPE, OP)</div><div class="ttdef"><b>Definition:</b> base.h:292</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0">mshadow::kUint8</a></div><div class="ttdeci">@ kUint8</div><div class="ttdef"><b>Definition:</b> base.h:355</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e">mshadow::kBfloat16</a></div><div class="ttdeci">@ kBfloat16</div><div class="ttdef"><b>Definition:</b> base.h:364</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1div_html_a5ad9c92273fb4c1a29fc3eb3fba3ae5a"><div class="ttname"><a href="structmshadow_1_1op_1_1div.html#a5ad9c92273fb4c1a29fc3eb3fba3ae5a">mshadow::op::div::Map</a></div><div class="ttdeci">static MSHADOW_XINLINE DType Map(DType a, DType b)</div><div class="ttdoc">map a, b to result using defined operation</div><div class="ttdef"><b>Definition:</b> base.h:652</div></div> |
| <div class="ttc" id="anamespacemshadow_html_abbe2dc827052234b64ab2d40a5b2fdb4a0cfb685a803481d9ff4fe19d813b261a"><div class="ttname"><a href="namespacemshadow.html#abbe2dc827052234b64ab2d40a5b2fdb4a0cfb685a803481d9ff4fe19d813b261a">mshadow::kNCW</a></div><div class="ttdeci">@ kNCW</div><div class="ttdef"><b>Definition:</b> base.h:505</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1sum_html_a36583c3bb79bbfe107b7c9480e80d5e7"><div class="ttname"><a href="structmshadow_1_1red_1_1sum.html#a36583c3bb79bbfe107b7c9480e80d5e7">mshadow::red::sum::Reduce</a></div><div class="ttdeci">static MSHADOW_XINLINE void Reduce(volatile DType &dst, volatile DType src, volatile DType &residual)</div><div class="ttdoc">do stable reduction into dst</div><div class="ttdef"><b>Definition:</b> base.h:995</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a194ad52f15262c4acde6f4720f039858"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a194ad52f15262c4acde6f4720f039858">mshadow::red::limits::PosInfValue< double ></a></div><div class="ttdeci">MSHADOW_XINLINE double PosInfValue< double >(void)</div><div class="ttdoc">positive infinity value of double</div><div class="ttdef"><b>Definition:</b> base.h:970</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1mul_html"><div class="ttname"><a href="structmshadow_1_1op_1_1mul.html">mshadow::op::mul</a></div><div class="ttdoc">mul operator</div><div class="ttdef"><b>Definition:</b> base.h:625</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a20e2ca3ab09e592f109bc7d7fca7e71d"><div class="ttname"><a href="namespacemshadow.html#a20e2ca3ab09e592f109bc7d7fca7e71d">mshadow::layoutFlag</a></div><div class="ttdeci">LayoutFlag layoutFlag(std::string layoutstr)</div><div class="ttdef"><b>Definition:</b> base.h:514</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a6959787874454734212c8c30b07a8876"><div class="ttname"><a href="namespacemshadow.html#a6959787874454734212c8c30b07a8876">mshadow::index_type_flag</a></div><div class="ttdeci">const int index_type_flag</div><div class="ttdoc">TypeFlag value for type of indexes.</div><div class="ttdef"><b>Definition:</b> base.h:495</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_ab06124f184da5683a517e02bcf2d6b9d"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#ab06124f184da5683a517e02bcf2d6b9d">mshadow::red::limits::MinValue< unsigned int ></a></div><div class="ttdeci">MSHADOW_XINLINE unsigned int MinValue< unsigned int >(void)</div><div class="ttdoc">minimum value of unsigned int</div><div class="ttdef"><b>Definition:</b> base.h:864</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a993a69000bf5e0a5d13ef149a3311a78"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a993a69000bf5e0a5d13ef149a3311a78">mshadow::red::limits::MaxValue</a></div><div class="ttdeci">MSHADOW_XINLINE DType MaxValue(void)</div><div class="ttdoc">maximum value of certain types</div></div> |
| <div class="ttc" id="astructmshadow_1_1sv_1_1divto_html"><div class="ttname"><a href="structmshadow_1_1sv_1_1divto.html">mshadow::sv::divto</a></div><div class="ttdoc">divide to saver: /=</div><div class="ttdef"><b>Definition:</b> base.h:732</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4">mshadow::kFloat16</a></div><div class="ttdeci">@ kFloat16</div><div class="ttdef"><b>Definition:</b> base.h:354</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_af43c87655a8a1531462fa86bc7cc902e"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#af43c87655a8a1531462fa86bc7cc902e">mshadow::red::limits::NegInfValue< float ></a></div><div class="ttdeci">MSHADOW_XINLINE float NegInfValue< float >(void)</div><div class="ttdoc">negative infinity value of float</div><div class="ttdef"><b>Definition:</b> base.h:878</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1minimum_html"><div class="ttname"><a href="structmshadow_1_1red_1_1minimum.html">mshadow::red::minimum</a></div><div class="ttdoc">minimum reducer</div><div class="ttdef"><b>Definition:</b> base.h:1108</div></div> |
| <div class="ttc" id="anamespacemshadow_1_1red_1_1limits_html_a1e417de4a028b2628d6bb5b6af1244dc"><div class="ttname"><a href="namespacemshadow_1_1red_1_1limits.html#a1e417de4a028b2628d6bb5b6af1244dc">mshadow::red::limits::MinValue< int32_t ></a></div><div class="ttdeci">MSHADOW_XINLINE int MinValue< int32_t >(void)</div><div class="ttdoc">minimum value of int32_t</div><div class="ttdef"><b>Definition:</b> base.h:849</div></div> |
| <div class="ttc" id="astructmshadow_1_1red_1_1maximum_html_ac2d455ca3b4eb12554eb31ed6a1387b7"><div class="ttname"><a href="structmshadow_1_1red_1_1maximum.html#ac2d455ca3b4eb12554eb31ed6a1387b7">mshadow::red::maximum::PartialGrad</a></div><div class="ttdeci">static MSHADOW_XINLINE DType PartialGrad(DType redres, DType redsrc)</div><div class="ttdoc">calculate gradient of redres with respect to redsrc, redres: reduced result, redsrc: one of reduction...</div><div class="ttdef"><b>Definition:</b> base.h:1089</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633"><div class="ttname"><a href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">mshadow::kFloat32</a></div><div class="ttdeci">@ kFloat32</div><div class="ttdef"><b>Definition:</b> base.h:352</div></div> |
| <div class="ttc" id="astructmshadow_1_1op_1_1right_html_a9377077b54a983fd220a0ba0263227b3"><div class="ttname"><a href="structmshadow_1_1op_1_1right.html#a9377077b54a983fd220a0ba0263227b3">mshadow::op::right::Map</a></div><div class="ttdeci">static MSHADOW_XINLINE DType Map(DType a, DType b)</div><div class="ttdoc">map a, b to result using defined operation</div><div class="ttdef"><b>Definition:</b> base.h:660</div></div> |
| <div class="ttc" id="abfloat_8h_html"><div class="ttname"><a href="bfloat_8h.html">bfloat.h</a></div><div class="ttdoc">definition of bfloat type.</div></div> |
| <div class="ttc" id="ahalf_8h_html_afdc51a9766b6923a3742828222103e3d"><div class="ttname"><a href="half_8h.html#afdc51a9766b6923a3742828222103e3d">MSHADOW_HALF_EXPONENT_BITS</a></div><div class="ttdeci">#define MSHADOW_HALF_EXPONENT_BITS</div><div class="ttdef"><b>Definition:</b> half.h:371</div></div> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Jan 5 2023 03:47:39 for mxnet by  <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/> |
| </a> 1.8.17 |
| </small></address> |
| </body> |
| </html> |
