| <!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/include/mxnet/tensor_blob.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_d44c64559bbebec7f509842c48db8b23.html">include</a></li><li class="navelem"><a class="el" href="dir_1143c7affb9ebd026cb6818dd282def7.html">mxnet</a></li> </ul> |
| </div> |
| </div><!-- top --> |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">tensor_blob.h</div> </div> |
| </div><!--header--> |
| <div class="contents"> |
| <a href="tensor__blob_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="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#ifndef MXNET_TENSOR_BLOB_H_</span></div> |
| <div class="line"><a name="l00028"></a><span class="lineno"> 28</span> <span class="preprocessor">#define MXNET_TENSOR_BLOB_H_</span></div> |
| <div class="line"><a name="l00029"></a><span class="lineno"> 29</span>  </div> |
| <div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#include <dmlc/logging.h></span></div> |
| <div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#include <<a class="code" href="json_8h.html">dmlc/json.h</a>></span></div> |
| <div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#include <<a class="code" href="dlpack_8h.html">dlpack/dlpack.h</a>></span></div> |
| <div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#include <vector></span></div> |
| <div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include <iostream></span></div> |
| <div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#include <utility></span></div> |
| <div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#include <algorithm></span></div> |
| <div class="line"><a name="l00037"></a><span class="lineno"> 37</span> <span class="preprocessor">#include "<a class="code" href="include_2mxnet_2base_8h.html">./base.h</a>"</span></div> |
| <div class="line"><a name="l00038"></a><span class="lineno"> 38</span>  </div> |
| <div class="line"><a name="l00039"></a><span class="lineno"> 39</span> <span class="keyword">namespace </span><a class="code" href="namespacemxnet.html">mxnet</a> {</div> |
| <div class="line"><a name="l00040"></a><span class="lineno"> 40</span>  </div> |
| <div class="line"><a name="l00041"></a><span class="lineno"> 41</span> <span class="comment">// redefine DLPack enumeration to be backward compatible.</span></div> |
| <div class="line"><a name="l00042"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a512ee02aece77cb0bf39b6e2ded77686"> 42</a></span> constexpr <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemxnet.html#a512ee02aece77cb0bf39b6e2ded77686">kCPU</a> = <a class="code" href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e">kDLCPU</a>;</div> |
| <div class="line"><a name="l00043"></a><span class="lineno"><a class="line" href="namespacemxnet.html#a43ebce8ff01c0bee0dddb4417ea1239d"> 43</a></span> constexpr <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemxnet.html#a43ebce8ff01c0bee0dddb4417ea1239d">kGPU</a> = <a class="code" href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680">kDLGPU</a>;</div> |
| <div class="line"><a name="l00044"></a><span class="lineno"> 44</span> <span class="comment">// extension type code under TVM function.</span></div> |
| <div class="line"><a name="l00045"></a><span class="lineno"> 45</span> <span class="comment">// Currently NNVM reserved 16 to 19 type code from TVM</span></div> |
| <div class="line"><a name="l00046"></a><span class="lineno"> 46</span> <span class="comment">// 16, 17, 18 is used by NNVM compiler already.</span></div> |
| <div class="line"><a name="l00047"></a><span class="lineno"> 47</span> <span class="comment">// Pick code 19 for MXNet NDArray</span></div> |
| <div class="line"><a name="l00048"></a><span class="lineno"><a class="line" href="namespacemxnet.html#ab96d1eea6c8d006db2b33bbdf9d90bd2"> 48</a></span> constexpr <span class="keyword">const</span> <span class="keywordtype">int</span> <a class="code" href="namespacemxnet.html#ab96d1eea6c8d006db2b33bbdf9d90bd2">kTVMNDArrayTypeCode</a> = 19;</div> |
| <div class="line"><a name="l00049"></a><span class="lineno"> 49</span>  </div> |
| <div class="line"><a name="l00050"></a><span class="lineno"> 50</span> <span class="comment">/* Forward declaration for friend declaration in TBlob */</span></div> |
| <div class="line"><a name="l00051"></a><span class="lineno"> 51</span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1NDArray.html">NDArray</a>;</div> |
| <div class="line"><a name="l00052"></a><span class="lineno"> 52</span>  </div> |
| <div class="line"><a name="l00065"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html"> 65</a></span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1TBlob.html">TBlob</a> {</div> |
| <div class="line"><a name="l00066"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ad41557e205cf8f4d6ef70b60587c37cb"> 66</a></span>  <span class="keyword">friend</span> <span class="keyword">class </span><a class="code" href="classmxnet_1_1NDArray.html">NDArray</a>;</div> |
| <div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  </div> |
| <div class="line"><a name="l00068"></a><span class="lineno"> 68</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00070"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10"> 70</a></span>  <span class="keywordtype">void</span>* <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>;</div> |
| <div class="line"><a name="l00072"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029"> 72</a></span>  <a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a> <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>;</div> |
| <div class="line"><a name="l00074"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61"> 74</a></span>  <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>;</div> |
| <div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  </div> |
| <div class="line"><a name="l00077"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#adcdc6de67f81454f0a142bfc9fd38c9e"> 77</a></span>  <a class="code" href="classmxnet_1_1TBlob.html#adcdc6de67f81454f0a142bfc9fd38c9e">TBlob</a>(<span class="keywordtype">void</span>) : <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>(nullptr), <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>(<a class="code" href="namespacemshadow.html">mshadow</a>::DataType<<a class="code" href="namespacemxnet.html#a4421326e33f2a848cb2fa3e70ab24ff4">real_t</a>>::kFlag) {</div> |
| <div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  SetDLTensor(<a class="code" href="structmshadow_1_1cpu.html#a57f74b500825e21c0c246ad5b4e5016b">cpu::kDevMask</a>, 0);</div> |
| <div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  }</div> |
| <div class="line"><a name="l00087"></a><span class="lineno"> 87</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00088"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ae2bc6bbce5faf2c99d3b0ea462a3fcfa"> 88</a></span>  <a class="code" href="classmxnet_1_1TBlob.html#ae2bc6bbce5faf2c99d3b0ea462a3fcfa">TBlob</a>(DType* <a class="code" href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd">dptr</a>, <span class="keyword">const</span> <a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a>& shape, <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>, <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a> = -1)</div> |
| <div class="line"><a name="l00089"></a><span class="lineno"> 89</span>  : <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>(<a class="code" href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd">dptr</a>), <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>(shape), <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>(<a class="code" href="namespacemshadow.html">mshadow</a>::DataType<DType>::kFlag) {</div> |
| <div class="line"><a name="l00090"></a><span class="lineno"> 90</span>  SetDLTensor(<a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>, <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>);</div> |
| <div class="line"><a name="l00091"></a><span class="lineno"> 91</span>  }</div> |
| <div class="line"><a name="l00100"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a31c0750cf2642b68af00c00029c99967"> 100</a></span>  <a class="code" href="classmxnet_1_1TBlob.html#a31c0750cf2642b68af00c00029c99967">TBlob</a>(<span class="keywordtype">void</span>* <a class="code" href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd">dptr</a>, <span class="keyword">const</span> <a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a>& shape, <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>, <span class="keywordtype">int</span> type_flag, <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a> = -1)</div> |
| <div class="line"><a name="l00101"></a><span class="lineno"> 101</span>  : <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>(<a class="code" href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd">dptr</a>), <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>(shape), <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>(type_flag) {</div> |
| <div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  SetDLTensor(<a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>, <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>);</div> |
| <div class="line"><a name="l00103"></a><span class="lineno"> 103</span>  }</div> |
| <div class="line"><a name="l00108"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a44594fd9712fb19061db95a070c46000"> 108</a></span>  <span class="keyword">explicit</span> <a class="code" href="classmxnet_1_1TBlob.html#a44594fd9712fb19061db95a070c46000">TBlob</a>(<span class="keyword">const</span> <a class="code" href="structDLTensor.html">DLTensor</a>& <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>)</div> |
| <div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  : <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>(<a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.data),</div> |
| <div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>(<a class="code" href="namespacemxnet.html">mxnet</a>::<a class="code" href="classmxnet_1_1TShape.html">TShape</a>(<a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.shape, <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.shape + <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.<a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>)),</div> |
| <div class="line"><a name="l00111"></a><span class="lineno"> 111</span>  <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>(DLDataTypeTransform(<a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.dtype)),</div> |
| <div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  dltensor_(<a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>) {</div> |
| <div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  <span class="comment">// compactness check for DLTensor</span></div> |
| <div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <span class="keywordflow">if</span> (<a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.<a class="code" href="structDLTensor.html#aa1a066fbde828cf415f77e5dbf962c2e">strides</a> != <span class="keyword">nullptr</span>) {</div> |
| <div class="line"><a name="l00115"></a><span class="lineno"> 115</span>  <span class="comment">// check strides</span></div> |
| <div class="line"><a name="l00116"></a><span class="lineno"> 116</span>  <span class="keyword">const</span> <span class="keywordtype">int</span>& <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a> = <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.<a class="code" href="structDLTensor.html#a8b28a3876c4d157ccea80956f3ec18f5">ndim</a>;</div> |
| <div class="line"><a name="l00117"></a><span class="lineno"> 117</span>  <span class="keyword">const</span> int64_t* shape = <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.<a class="code" href="structDLTensor.html#ac982d47788a5251f1cd97375a1a2e49f">shape</a>;</div> |
| <div class="line"><a name="l00118"></a><span class="lineno"> 118</span>  <span class="keyword">const</span> int64_t* strides = <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>.<a class="code" href="structDLTensor.html#aa1a066fbde828cf415f77e5dbf962c2e">strides</a>;</div> |
| <div class="line"><a name="l00119"></a><span class="lineno"> 119</span>  <span class="keywordflow">if</span> (<a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a> >= 1) {</div> |
| <div class="line"><a name="l00120"></a><span class="lineno"> 120</span>  <span class="keywordtype">bool</span> err = <span class="keyword">false</span>;</div> |
| <div class="line"><a name="l00121"></a><span class="lineno"> 121</span>  <span class="keywordflow">if</span> (strides[<a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a> - 1] != 1) {</div> |
| <div class="line"><a name="l00122"></a><span class="lineno"> 122</span>  err = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00123"></a><span class="lineno"> 123</span>  } <span class="keywordflow">else</span> {</div> |
| <div class="line"><a name="l00124"></a><span class="lineno"> 124</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a> - 2; i >= 0; --i) {</div> |
| <div class="line"><a name="l00125"></a><span class="lineno"> 125</span>  <span class="keywordflow">if</span> (strides[i] != shape[i + 1] * strides[i + 1]) {</div> |
| <div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  err = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00127"></a><span class="lineno"> 127</span>  <span class="keywordflow">break</span>;</div> |
| <div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  }</div> |
| <div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  }</div> |
| <div class="line"><a name="l00130"></a><span class="lineno"> 130</span>  }</div> |
| <div class="line"><a name="l00131"></a><span class="lineno"> 131</span>  <span class="keywordflow">if</span> (err) {</div> |
| <div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  LOG(FATAL) << <span class="stringliteral">"Unsupported DLPack because MXNet only support compact tensor now"</span>;</div> |
| <div class="line"><a name="l00133"></a><span class="lineno"> 133</span>  }</div> |
| <div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  }</div> |
| <div class="line"><a name="l00135"></a><span class="lineno"> 135</span>  }</div> |
| <div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  }</div> |
| <div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keywordtype">int</span> dim, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00145"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#aae92a89345aad79c8c3b451b9e3b84a5"> 145</a></span>  <a class="code" href="classmxnet_1_1TBlob.html#aae92a89345aad79c8c3b451b9e3b84a5">TBlob</a>(<span class="keyword">const</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a>& src) { <span class="comment">// NOLINT(*)</span></div> |
| <div class="line"><a name="l00146"></a><span class="lineno"> 146</span>  *<span class="keyword">this</span> = src;</div> |
| <div class="line"><a name="l00147"></a><span class="lineno"> 147</span>  }</div> |
| <div class="line"><a name="l00152"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ac64b2d46ddd92bf972a4e5b49afbf325"> 152</a></span>  <a class="code" href="classmxnet_1_1TBlob.html#ac64b2d46ddd92bf972a4e5b49afbf325">TBlob</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1TBlob.html">TBlob</a>& src) : <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>(src.<a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>), <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>(src.<a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>), <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>(src.<a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>) {</div> |
| <div class="line"><a name="l00153"></a><span class="lineno"> 153</span>  this->SetDLTensor(src.<a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>(), src.<a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>());</div> |
| <div class="line"><a name="l00154"></a><span class="lineno"> 154</span>  }</div> |
| <div class="line"><a name="l00163"></a><span class="lineno"> 163</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keywordtype">int</span> dim, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00164"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ad3263d9d1ef40678cd592be822d1c7be"> 164</a></span>  <span class="keyword">inline</span> <a class="code" href="classmxnet_1_1TBlob.html">TBlob</a>& <a class="code" href="classmxnet_1_1TBlob.html#ad3263d9d1ef40678cd592be822d1c7be">operator=</a>(<span class="keyword">const</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a>& src) {</div> |
| <div class="line"><a name="l00165"></a><span class="lineno"> 165</span>  <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a> = src.<a class="code" href="structmshadow_1_1Tensor.html#ad86d6759c585efb5229b3a0659973838">dptr_</a>;</div> |
| <div class="line"><a name="l00166"></a><span class="lineno"> 166</span>  <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a> = src.<a class="code" href="structmshadow_1_1Tensor.html#ad10c7414c5948e789e8761df2083c4e5">shape_</a>;</div> |
| <div class="line"><a name="l00167"></a><span class="lineno"> 167</span>  <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a> = <a class="code" href="structmshadow_1_1DataType.html">mshadow::DataType<DType>::kFlag</a>;</div> |
| <div class="line"><a name="l00168"></a><span class="lineno"> 168</span>  SetDLTensor(Device::kDevMask, -1);</div> |
| <div class="line"><a name="l00169"></a><span class="lineno"> 169</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  }</div> |
| <div class="line"><a name="l00176"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a7971ef6867cd7c94ee32de023f8dbe23"> 176</a></span>  <span class="keyword">inline</span> <a class="code" href="classmxnet_1_1TBlob.html">TBlob</a>& <a class="code" href="classmxnet_1_1TBlob.html#a7971ef6867cd7c94ee32de023f8dbe23">operator=</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1TBlob.html">TBlob</a>& src) {</div> |
| <div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a> = src.<a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>;</div> |
| <div class="line"><a name="l00178"></a><span class="lineno"> 178</span>  <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a> = src.<a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>;</div> |
| <div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a> = src.<a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>;</div> |
| <div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  SetDLTensor(src.<a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>(), src.<a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>());</div> |
| <div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div> |
| <div class="line"><a name="l00182"></a><span class="lineno"> 182</span>  }</div> |
| <div class="line"><a name="l00186"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ac3abff8bace8f3afb3143d85551a18ac"> 186</a></span>  <span class="keyword">inline</span> <span class="keywordtype">bool</span> <a class="code" href="classmxnet_1_1TBlob.html#ac3abff8bace8f3afb3143d85551a18ac">CheckContiguous</a>(<span class="keywordtype">void</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00187"></a><span class="lineno"> 187</span>  <span class="keywordflow">return</span> <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  }</div> |
| <div class="line"><a name="l00194"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a10111644a1abe2035ec3e869187847dc"> 194</a></span>  <span class="keyword">inline</span> <a class="code" href="classmxnet_1_1TBlob.html">TBlob</a> <a class="code" href="classmxnet_1_1TBlob.html#a10111644a1abe2035ec3e869187847dc">reshape</a>(<span class="keyword">const</span> <a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a>& shape)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  CHECK_EQ(this->shape_.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>(), shape.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>())</div> |
| <div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  << <span class="stringliteral">"Shape size mismatch "</span> << this->shape_.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>() << <span class="stringliteral">" v.s. "</span> << shape.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>();</div> |
| <div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  <a class="code" href="classmxnet_1_1TBlob.html">TBlob</a> ret(this->dptr_, shape, this-><a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>(), this->type_flag_, this-><a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>());</div> |
| <div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  <span class="keywordflow">return</span> ret;</div> |
| <div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  }</div> |
| <div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00208"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a9bb2615c144e237afea3875e618527d4"> 208</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, 2, DType></a> <a class="code" href="classmxnet_1_1TBlob.html#a9bb2615c144e237afea3875e618527d4">FlatTo2D</a>(</div> |
| <div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  CHECK(Device::kDevMask == this-><a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>())</div> |
| <div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  << <span class="stringliteral">"TBlob.get: device type do not match specified type"</span>;</div> |
| <div class="line"><a name="l00212"></a><span class="lineno"> 212</span>  CHECK(<a class="code" href="structmshadow_1_1DataType.html">mshadow::DataType<DType>::kFlag</a> == <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>)</div> |
| <div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  << <span class="stringliteral">"TBlob.get_with_shape: data type do not match specified type. "</span></div> |
| <div class="line"><a name="l00214"></a><span class="lineno"> 214</span>  << <span class="stringliteral">"Expected: "</span> << <a class="code" href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072">mshadow::dtype_string</a>(<a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>) << <span class="stringliteral">" v.s. given "</span></div> |
| <div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  << <a class="code" href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072">mshadow::dtype_string</a>(<a class="code" href="structmshadow_1_1DataType.html">mshadow::DataType<DType>::kFlag</a>);</div> |
| <div class="line"><a name="l00216"></a><span class="lineno"> 216</span>  <span class="keywordflow">return</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, 2, DType></a>(<span class="keyword">static_cast<</span>DType*<span class="keyword">></span>(<a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>), <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.FlatTo2D(), stream);</div> |
| <div class="line"><a name="l00217"></a><span class="lineno"> 217</span>  }</div> |
| <div class="line"><a name="l00225"></a><span class="lineno"> 225</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00226"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#af220608e394f02ad3739ac048757175b"> 226</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor_3_01Device_00_011_00_01DType_01_4.html">mshadow::Tensor<Device, 1, DType></a> <a class="code" href="classmxnet_1_1TBlob.html#af220608e394f02ad3739ac048757175b">FlatTo1D</a>(</div> |
| <div class="line"><a name="l00227"></a><span class="lineno"> 227</span>  <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00228"></a><span class="lineno"> 228</span>  <span class="keywordflow">return</span> this->get_with_shape<Device, 1, DType>(<a class="code" href="namespacemshadow.html#a05e468ef4d8882fccce53ae951b7bdbd">mshadow::Shape1</a>(<a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>()), stream);</div> |
| <div class="line"><a name="l00229"></a><span class="lineno"> 229</span>  }</div> |
| <div class="line"><a name="l00231"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851"> 231</a></span>  <span class="keyword">inline</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>(<span class="keywordtype">void</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00232"></a><span class="lineno"> 232</span>  <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.<a class="code" href="classmxnet_1_1Tuple.html#a6e0bdfe3f0f34693223f657ef676bef6">ndim</a>();</div> |
| <div class="line"><a name="l00233"></a><span class="lineno"> 233</span>  }</div> |
| <div class="line"><a name="l00240"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a44c93b8a96203a1a85fa9d4b85e123c7"> 240</a></span>  <span class="keyword">inline</span> <a class="code" href="namespacemxnet.html#abadedc1b42ee3e5b8852e63783b982ef">index_t</a> <a class="code" href="classmxnet_1_1TBlob.html#a44c93b8a96203a1a85fa9d4b85e123c7">size</a>(<a class="code" href="namespacemxnet.html#abadedc1b42ee3e5b8852e63783b982ef">index_t</a> idx)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00241"></a><span class="lineno"> 241</span>  <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>[idx];</div> |
| <div class="line"><a name="l00242"></a><span class="lineno"> 242</span>  }</div> |
| <div class="line"><a name="l00244"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#aa676224250ad5299e1b914da556cdc8d"> 244</a></span>  <span class="keyword">inline</span> <span class="keywordtype">size_t</span> <a class="code" href="classmxnet_1_1TBlob.html#aa676224250ad5299e1b914da556cdc8d">Size</a>(<span class="keywordtype">void</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00245"></a><span class="lineno"> 245</span>  <span class="keywordflow">return</span> <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>();</div> |
| <div class="line"><a name="l00246"></a><span class="lineno"> 246</span>  }</div> |
| <div class="line"><a name="l00248"></a><span class="lineno"> 248</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00249"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd"> 249</a></span>  <span class="keyword">inline</span> DType* <a class="code" href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd">dptr</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00250"></a><span class="lineno"> 250</span>  CHECK(<a class="code" href="structmshadow_1_1DataType.html">mshadow::DataType<DType>::kFlag</a> == <a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>)</div> |
| <div class="line"><a name="l00251"></a><span class="lineno"> 251</span>  << <span class="stringliteral">"TBlob.get_with_shape: data type do not match specified type. "</span></div> |
| <div class="line"><a name="l00252"></a><span class="lineno"> 252</span>  << <span class="stringliteral">"Expected: "</span> << <a class="code" href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072">mshadow::dtype_string</a>(<a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>) << <span class="stringliteral">" v.s. given "</span></div> |
| <div class="line"><a name="l00253"></a><span class="lineno"> 253</span>  << <a class="code" href="namespacemshadow.html#af5aa21f20dbeaa38194b4dadbd37a072">mshadow::dtype_string</a>(<a class="code" href="structmshadow_1_1DataType.html">mshadow::DataType<DType>::kFlag</a>);</div> |
| <div class="line"><a name="l00254"></a><span class="lineno"> 254</span>  <span class="keywordflow">return</span> <span class="keyword">static_cast<</span>DType*<span class="keyword">></span>(<a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>);</div> |
| <div class="line"><a name="l00255"></a><span class="lineno"> 255</span>  }</div> |
| <div class="line"><a name="l00257"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f"> 257</a></span>  <span class="keyword">inline</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00258"></a><span class="lineno"> 258</span>  <span class="keywordflow">return</span> dltensor_.<a class="code" href="structDLTensor.html#a552ca85f7d04138c876c636e25b3c484">ctx</a>.<a class="code" href="structDLContext.html#a9e7b666f60b074a4f48943da69cee9ce">device_type</a>;</div> |
| <div class="line"><a name="l00259"></a><span class="lineno"> 259</span>  }</div> |
| <div class="line"><a name="l00261"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd"> 261</a></span>  <span class="keyword">inline</span> <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00262"></a><span class="lineno"> 262</span>  <span class="keywordflow">return</span> dltensor_.<a class="code" href="structDLTensor.html#a552ca85f7d04138c876c636e25b3c484">ctx</a>.<a class="code" href="structDLContext.html#a7cd00ca96ac03a1f8ea2b2b3a565794d">device_id</a>;</div> |
| <div class="line"><a name="l00263"></a><span class="lineno"> 263</span>  }</div> |
| <div class="line"><a name="l00268"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda"> 268</a></span>  <span class="keyword">inline</span> <span class="keyword">const</span> <a class="code" href="structDLTensor.html">DLTensor</a>& <a class="code" href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">dltensor</a>()<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00269"></a><span class="lineno"> 269</span>  <span class="keywordflow">return</span> dltensor_;</div> |
| <div class="line"><a name="l00270"></a><span class="lineno"> 270</span>  }</div> |
| <div class="line"><a name="l00271"></a><span class="lineno"> 271</span>  </div> |
| <div class="line"><a name="l00281"></a><span class="lineno"> 281</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keywordtype">int</span> dim, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00282"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#af38cdc5d0bcce77bb4f96251167cd7a0"> 282</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a> <a class="code" href="classmxnet_1_1TBlob.html#af38cdc5d0bcce77bb4f96251167cd7a0">get</a>(<a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00283"></a><span class="lineno"> 283</span>  CHECK(Device::kDevMask == this-><a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>())</div> |
| <div class="line"><a name="l00284"></a><span class="lineno"> 284</span>  << <span class="stringliteral">"TBlob.get: device type do not match specified type"</span>;</div> |
| <div class="line"><a name="l00285"></a><span class="lineno"> 285</span>  <span class="keywordflow">return</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a>(</div> |
| <div class="line"><a name="l00286"></a><span class="lineno"> 286</span>  dptr<DType>(), <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.get<dim>(), <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>[<a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.<a class="code" href="classmxnet_1_1Tuple.html#a6e0bdfe3f0f34693223f657ef676bef6">ndim</a>() - 1], stream);</div> |
| <div class="line"><a name="l00287"></a><span class="lineno"> 287</span>  }</div> |
| <div class="line"><a name="l00298"></a><span class="lineno"> 298</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keywordtype">int</span> dim, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00299"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#ad17880c664c448bb159b356e0c2b875c"> 299</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a> <a class="code" href="classmxnet_1_1TBlob.html#ad17880c664c448bb159b356e0c2b875c">get_with_shape</a>(</div> |
| <div class="line"><a name="l00300"></a><span class="lineno"> 300</span>  <span class="keyword">const</span> <a class="code" href="structmshadow_1_1Shape.html">mshadow::Shape<dim></a>& shape,</div> |
| <div class="line"><a name="l00301"></a><span class="lineno"> 301</span>  <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00302"></a><span class="lineno"> 302</span>  CHECK(Device::kDevMask == this-><a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>())</div> |
| <div class="line"><a name="l00303"></a><span class="lineno"> 303</span>  << <span class="stringliteral">"TBlob.get: device type do not match specified type"</span>;</div> |
| <div class="line"><a name="l00304"></a><span class="lineno"> 304</span>  CHECK_EQ(this-><a class="code" href="classmxnet_1_1TBlob.html#ac3abff8bace8f3afb3143d85551a18ac">CheckContiguous</a>(), <span class="keyword">true</span>) << <span class="stringliteral">"TBlob.get_reshape: must be contiguous"</span>;</div> |
| <div class="line"><a name="l00305"></a><span class="lineno"> 305</span>  CHECK_EQ(this->shape_.<a class="code" href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">Size</a>(), <span class="keyword">static_cast<</span><span class="keywordtype">size_t</span><span class="keyword">></span>(shape.<a class="code" href="structmshadow_1_1Shape.html#ac6d667fde1a8180b64f475f5d33ea58f">Size</a>()))</div> |
| <div class="line"><a name="l00306"></a><span class="lineno"> 306</span>  << <span class="stringliteral">"TBlob.get_with_shape: new and old shape do not match total elements"</span>;</div> |
| <div class="line"><a name="l00307"></a><span class="lineno"> 307</span>  <span class="keywordflow">return</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a>(dptr<DType>(), shape, shape[dim - 1], stream);</div> |
| <div class="line"><a name="l00308"></a><span class="lineno"> 308</span>  }</div> |
| <div class="line"><a name="l00318"></a><span class="lineno"> 318</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00319"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a0d051bba948b8a69a9d1d2690fad7400"> 319</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, 3, DType></a> <a class="code" href="classmxnet_1_1TBlob.html#a0d051bba948b8a69a9d1d2690fad7400">FlatTo3D</a>(</div> |
| <div class="line"><a name="l00320"></a><span class="lineno"> 320</span>  <span class="keywordtype">int</span> axis,</div> |
| <div class="line"><a name="l00321"></a><span class="lineno"> 321</span>  <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00322"></a><span class="lineno"> 322</span>  <span class="keywordflow">return</span> this->get_with_shape<Device, 3, DType>(this->shape_.FlatTo3D(axis), stream);</div> |
| <div class="line"><a name="l00323"></a><span class="lineno"> 323</span>  }</div> |
| <div class="line"><a name="l00334"></a><span class="lineno"> 334</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00335"></a><span class="lineno"> 335</span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, 3, DType></a></div> |
| <div class="line"><a name="l00336"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a684ef7a1354dc47b808ef54cfecf81b7"> 336</a></span>  <a class="code" href="classmxnet_1_1TBlob.html#a684ef7a1354dc47b808ef54cfecf81b7">FlatTo3D</a>(<span class="keywordtype">int</span> axis_begin, <span class="keywordtype">int</span> axis_end, <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00337"></a><span class="lineno"> 337</span>  <span class="keywordflow">return</span> this->get_with_shape<Device, 3, DType>(this->shape_.FlatTo3D(axis_begin, axis_end),</div> |
| <div class="line"><a name="l00338"></a><span class="lineno"> 338</span>  stream);</div> |
| <div class="line"><a name="l00339"></a><span class="lineno"> 339</span>  }</div> |
| <div class="line"><a name="l00349"></a><span class="lineno"> 349</span>  <span class="keyword">template</span> <<span class="keyword">typename</span> Device, <span class="keywordtype">int</span> dim, <span class="keyword">typename</span> DType></div> |
| <div class="line"><a name="l00350"></a><span class="lineno"><a class="line" href="classmxnet_1_1TBlob.html#a765ee89940e22fcf1cb24e8048da95e2"> 350</a></span>  <span class="keyword">inline</span> <a class="code" href="structmshadow_1_1Tensor.html">mshadow::Tensor<Device, dim, DType></a> <a class="code" href="classmxnet_1_1TBlob.html#a765ee89940e22fcf1cb24e8048da95e2">FlatToKD</a>(</div> |
| <div class="line"><a name="l00351"></a><span class="lineno"> 351</span>  <a class="code" href="structmshadow_1_1Stream.html">mshadow::Stream<Device></a>* stream = <span class="keyword">nullptr</span>)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00352"></a><span class="lineno"> 352</span>  <a class="code" href="structmshadow_1_1Shape.html">mshadow::Shape<dim></a> shape;</div> |
| <div class="line"><a name="l00353"></a><span class="lineno"> 353</span>  shape[0] = 1;</div> |
| <div class="line"><a name="l00354"></a><span class="lineno"> 354</span>  <span class="comment">// Pad higher dimensions in case dim > ndim()</span></div> |
| <div class="line"><a name="l00355"></a><span class="lineno"> 355</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < dim - <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>(); ++i) {</div> |
| <div class="line"><a name="l00356"></a><span class="lineno"> 356</span>  shape[i] = 1;</div> |
| <div class="line"><a name="l00357"></a><span class="lineno"> 357</span>  }</div> |
| <div class="line"><a name="l00358"></a><span class="lineno"> 358</span>  <span class="comment">// Collapse higher dimensions in case dim < ndim()</span></div> |
| <div class="line"><a name="l00359"></a><span class="lineno"> 359</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>() - dim + 1; ++i) {</div> |
| <div class="line"><a name="l00360"></a><span class="lineno"> 360</span>  shape[0] *= <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>[i];</div> |
| <div class="line"><a name="l00361"></a><span class="lineno"> 361</span>  }</div> |
| <div class="line"><a name="l00362"></a><span class="lineno"> 362</span>  <span class="comment">// Preserve lower dimensions.</span></div> |
| <div class="line"><a name="l00363"></a><span class="lineno"> 363</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = std::max(0, <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>() - dim + 1); i < <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>(); ++i) {</div> |
| <div class="line"><a name="l00364"></a><span class="lineno"> 364</span>  shape[i - <a class="code" href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">ndim</a>() + dim] = <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>[i];</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>  <span class="keywordflow">return</span> this->get_with_shape<Device, dim, DType>(shape, stream);</div> |
| <div class="line"><a name="l00367"></a><span class="lineno"> 367</span>  }</div> |
| <div class="line"><a name="l00368"></a><span class="lineno"> 368</span>  </div> |
| <div class="line"><a name="l00369"></a><span class="lineno"> 369</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00370"></a><span class="lineno"> 370</span>  <span class="keyword">static</span> <a class="code" href="structDLDataType.html">DLDataType</a> DTypeTransform(<span class="keywordtype">int</span> type_flag) {</div> |
| <div class="line"><a name="l00371"></a><span class="lineno"> 371</span>  <span class="keywordflow">switch</span> (type_flag) {</div> |
| <div class="line"><a name="l00372"></a><span class="lineno"> 372</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">mshadow::kFloat32</a>:</div> |
| <div class="line"><a name="l00373"></a><span class="lineno"> 373</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a>, 32, 1};</div> |
| <div class="line"><a name="l00374"></a><span class="lineno"> 374</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28">mshadow::kFloat64</a>:</div> |
| <div class="line"><a name="l00375"></a><span class="lineno"> 375</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a>, 64, 1};</div> |
| <div class="line"><a name="l00376"></a><span class="lineno"> 376</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4">mshadow::kFloat16</a>:</div> |
| <div class="line"><a name="l00377"></a><span class="lineno"> 377</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a>, 16, 1};</div> |
| <div class="line"><a name="l00378"></a><span class="lineno"> 378</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e">mshadow::kBfloat16</a>:</div> |
| <div class="line"><a name="l00379"></a><span class="lineno"> 379</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a6a739f1b3ce2f8978da99a5ac4a887a6">kDLBfloat</a>, 16, 1};</div> |
| <div class="line"><a name="l00380"></a><span class="lineno"> 380</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0">mshadow::kUint8</a>:</div> |
| <div class="line"><a name="l00381"></a><span class="lineno"> 381</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a>, 8, 1};</div> |
| <div class="line"><a name="l00382"></a><span class="lineno"> 382</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368">mshadow::kInt32</a>:</div> |
| <div class="line"><a name="l00383"></a><span class="lineno"> 383</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a>, 32, 1};</div> |
| <div class="line"><a name="l00384"></a><span class="lineno"> 384</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c">mshadow::kInt8</a>:</div> |
| <div class="line"><a name="l00385"></a><span class="lineno"> 385</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a>, 8, 1};</div> |
| <div class="line"><a name="l00386"></a><span class="lineno"> 386</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23">mshadow::kInt64</a>:</div> |
| <div class="line"><a name="l00387"></a><span class="lineno"> 387</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a>, 64, 1};</div> |
| <div class="line"><a name="l00388"></a><span class="lineno"> 388</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a">mshadow::kBool</a>:</div> |
| <div class="line"><a name="l00389"></a><span class="lineno"> 389</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a>, 1, 1};</div> |
| <div class="line"><a name="l00390"></a><span class="lineno"> 390</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17">mshadow::kInt16</a>:</div> |
| <div class="line"><a name="l00391"></a><span class="lineno"> 391</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a>, 16, 1};</div> |
| <div class="line"><a name="l00392"></a><span class="lineno"> 392</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908">mshadow::kUint16</a>:</div> |
| <div class="line"><a name="l00393"></a><span class="lineno"> 393</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a>, 16, 1};</div> |
| <div class="line"><a name="l00394"></a><span class="lineno"> 394</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7">mshadow::kUint32</a>:</div> |
| <div class="line"><a name="l00395"></a><span class="lineno"> 395</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a>, 32, 1};</div> |
| <div class="line"><a name="l00396"></a><span class="lineno"> 396</span>  <span class="keywordflow">case</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade">mshadow::kUint64</a>:</div> |
| <div class="line"><a name="l00397"></a><span class="lineno"> 397</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>{<a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a>, 64, 1};</div> |
| <div class="line"><a name="l00398"></a><span class="lineno"> 398</span>  <span class="keywordflow">default</span>: {</div> |
| <div class="line"><a name="l00399"></a><span class="lineno"> 399</span>  LOG(FATAL) << <span class="stringliteral">"Unknown type_flag="</span> << type_flag;</div> |
| <div class="line"><a name="l00400"></a><span class="lineno"> 400</span>  <span class="keywordflow">return</span> <a class="code" href="structDLDataType.html">DLDataType</a>();</div> |
| <div class="line"><a name="l00401"></a><span class="lineno"> 401</span>  }</div> |
| <div class="line"><a name="l00402"></a><span class="lineno"> 402</span>  }</div> |
| <div class="line"><a name="l00403"></a><span class="lineno"> 403</span>  }</div> |
| <div class="line"><a name="l00404"></a><span class="lineno"> 404</span>  <span class="keyword">static</span> <span class="keywordtype">int</span> DLDataTypeTransform(<a class="code" href="structDLDataType.html">DLDataType</a> dldata_type) {</div> |
| <div class="line"><a name="l00405"></a><span class="lineno"> 405</span>  <span class="keywordflow">if</span> (dldata_type.<a class="code" href="structDLDataType.html#a24de092fe56d358c1582e41499862594">lanes</a> != 1) {</div> |
| <div class="line"><a name="l00406"></a><span class="lineno"> 406</span>  LOG(FATAL) << <span class="stringliteral">"Unsupported DLDataType whose lanes != 1"</span>;</div> |
| <div class="line"><a name="l00407"></a><span class="lineno"> 407</span>  }</div> |
| <div class="line"><a name="l00408"></a><span class="lineno"> 408</span>  <span class="keywordflow">switch</span> (dldata_type.<a class="code" href="structDLDataType.html#aaa84983c485752e4401d80ce0e2b6e54">code</a>) {</div> |
| <div class="line"><a name="l00409"></a><span class="lineno"> 409</span>  <span class="keywordflow">case</span> <a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a>:</div> |
| <div class="line"><a name="l00410"></a><span class="lineno"> 410</span>  <span class="keywordflow">switch</span> (dldata_type.<a class="code" href="structDLDataType.html#a412e36fa9c58e808a2333e9044c5d9b5">bits</a>) {</div> |
| <div class="line"><a name="l00411"></a><span class="lineno"> 411</span>  <span class="keywordflow">case</span> 16:</div> |
| <div class="line"><a name="l00412"></a><span class="lineno"> 412</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a37ab9e42757689b17620f5728296d5d4">mshadow::kFloat16</a>;</div> |
| <div class="line"><a name="l00413"></a><span class="lineno"> 413</span>  <span class="keywordflow">case</span> 32:</div> |
| <div class="line"><a name="l00414"></a><span class="lineno"> 414</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">mshadow::kFloat32</a>;</div> |
| <div class="line"><a name="l00415"></a><span class="lineno"> 415</span>  <span class="keywordflow">case</span> 64:</div> |
| <div class="line"><a name="l00416"></a><span class="lineno"> 416</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1f5a1c62216cbd2200443d501924cf28">mshadow::kFloat64</a>;</div> |
| <div class="line"><a name="l00417"></a><span class="lineno"> 417</span>  }</div> |
| <div class="line"><a name="l00418"></a><span class="lineno"> 418</span>  <span class="keywordflow">break</span>;</div> |
| <div class="line"><a name="l00419"></a><span class="lineno"> 419</span>  <span class="keywordflow">case</span> <a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a6a739f1b3ce2f8978da99a5ac4a887a6">kDLBfloat</a>:</div> |
| <div class="line"><a name="l00420"></a><span class="lineno"> 420</span>  <span class="keywordflow">switch</span> (dldata_type.<a class="code" href="structDLDataType.html#a412e36fa9c58e808a2333e9044c5d9b5">bits</a>) {</div> |
| <div class="line"><a name="l00421"></a><span class="lineno"> 421</span>  <span class="keywordflow">case</span> 16:</div> |
| <div class="line"><a name="l00422"></a><span class="lineno"> 422</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2313be720271dd8ef9d3becbed309f2e">mshadow::kBfloat16</a>;</div> |
| <div class="line"><a name="l00423"></a><span class="lineno"> 423</span>  }</div> |
| <div class="line"><a name="l00424"></a><span class="lineno"> 424</span>  <span class="keywordflow">break</span>;</div> |
| <div class="line"><a name="l00425"></a><span class="lineno"> 425</span>  <span class="keywordflow">case</span> <a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a>:</div> |
| <div class="line"><a name="l00426"></a><span class="lineno"> 426</span>  <span class="keywordflow">switch</span> (dldata_type.<a class="code" href="structDLDataType.html#a412e36fa9c58e808a2333e9044c5d9b5">bits</a>) {</div> |
| <div class="line"><a name="l00427"></a><span class="lineno"> 427</span>  <span class="keywordflow">case</span> 1:</div> |
| <div class="line"><a name="l00428"></a><span class="lineno"> 428</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad2f7fcc5a34a6b27898cfd5d1626882a">mshadow::kBool</a>;</div> |
| <div class="line"><a name="l00429"></a><span class="lineno"> 429</span>  <span class="keywordflow">case</span> 8:</div> |
| <div class="line"><a name="l00430"></a><span class="lineno"> 430</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a1a39d2f8230da3cb53528904c8a5fff0">mshadow::kUint8</a>;</div> |
| <div class="line"><a name="l00431"></a><span class="lineno"> 431</span>  <span class="keywordflow">case</span> 16:</div> |
| <div class="line"><a name="l00432"></a><span class="lineno"> 432</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1aefe5e24a51079065b7707225c9289908">mshadow::kUint16</a>;</div> |
| <div class="line"><a name="l00433"></a><span class="lineno"> 433</span>  <span class="keywordflow">case</span> 32:</div> |
| <div class="line"><a name="l00434"></a><span class="lineno"> 434</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1ad48a252ca3e6d2cbe89f1ba570d206a7">mshadow::kUint32</a>;</div> |
| <div class="line"><a name="l00435"></a><span class="lineno"> 435</span>  <span class="keywordflow">case</span> 64:</div> |
| <div class="line"><a name="l00436"></a><span class="lineno"> 436</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a2c5f8f21a24725e29edad31e3ba75ade">mshadow::kUint64</a>;</div> |
| <div class="line"><a name="l00437"></a><span class="lineno"> 437</span>  }</div> |
| <div class="line"><a name="l00438"></a><span class="lineno"> 438</span>  <span class="keywordflow">break</span>;</div> |
| <div class="line"><a name="l00439"></a><span class="lineno"> 439</span>  <span class="keywordflow">case</span> <a class="code" href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a>:</div> |
| <div class="line"><a name="l00440"></a><span class="lineno"> 440</span>  <span class="keywordflow">switch</span> (dldata_type.<a class="code" href="structDLDataType.html#a412e36fa9c58e808a2333e9044c5d9b5">bits</a>) {</div> |
| <div class="line"><a name="l00441"></a><span class="lineno"> 441</span>  <span class="keywordflow">case</span> 8:</div> |
| <div class="line"><a name="l00442"></a><span class="lineno"> 442</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a020e3574a4ec06c8e0b23a1b8a93026c">mshadow::kInt8</a>;</div> |
| <div class="line"><a name="l00443"></a><span class="lineno"> 443</span>  <span class="keywordflow">case</span> 16:</div> |
| <div class="line"><a name="l00444"></a><span class="lineno"> 444</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a36cfd4fc26c0ee29affa99a36db0db17">mshadow::kInt16</a>;</div> |
| <div class="line"><a name="l00445"></a><span class="lineno"> 445</span>  <span class="keywordflow">case</span> 32:</div> |
| <div class="line"><a name="l00446"></a><span class="lineno"> 446</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a4fbb02e389c3126918b505cd01188368">mshadow::kInt32</a>;</div> |
| <div class="line"><a name="l00447"></a><span class="lineno"> 447</span>  <span class="keywordflow">case</span> 64:</div> |
| <div class="line"><a name="l00448"></a><span class="lineno"> 448</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1accdd431fb941a9158baa00e7c5f3dc23">mshadow::kInt64</a>;</div> |
| <div class="line"><a name="l00449"></a><span class="lineno"> 449</span>  }</div> |
| <div class="line"><a name="l00450"></a><span class="lineno"> 450</span>  <span class="keywordflow">break</span>;</div> |
| <div class="line"><a name="l00451"></a><span class="lineno"> 451</span>  }</div> |
| <div class="line"><a name="l00452"></a><span class="lineno"> 452</span>  LOG(FATAL) << <span class="stringliteral">"Unknown DLDataType{"</span> << dldata_type.<a class="code" href="structDLDataType.html#aaa84983c485752e4401d80ce0e2b6e54">code</a> << <span class="stringliteral">", "</span> << dldata_type.<a class="code" href="structDLDataType.html#a412e36fa9c58e808a2333e9044c5d9b5">bits</a> << <span class="stringliteral">", "</span></div> |
| <div class="line"><a name="l00453"></a><span class="lineno"> 453</span>  << dldata_type.<a class="code" href="structDLDataType.html#a24de092fe56d358c1582e41499862594">lanes</a> << <span class="stringliteral">"}"</span>;</div> |
| <div class="line"><a name="l00454"></a><span class="lineno"> 454</span>  <span class="keywordflow">return</span> <a class="code" href="namespacemshadow.html#a936bbfe6aeead8902973c098b87f18c1a404a5fd26328cf46170f6eb3424c9633">mshadow::kFloat32</a>;</div> |
| <div class="line"><a name="l00455"></a><span class="lineno"> 455</span>  }</div> |
| <div class="line"><a name="l00456"></a><span class="lineno"> 456</span>  </div> |
| <div class="line"><a name="l00457"></a><span class="lineno"> 457</span>  <span class="keyword">inline</span> <span class="keywordtype">void</span> SetDLTensor(<span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>, <span class="keywordtype">int</span> <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>) {</div> |
| <div class="line"><a name="l00458"></a><span class="lineno"> 458</span>  dltensor_.<a class="code" href="structDLTensor.html#a8a6408813fe13f1985379f022ade0947">data</a> = <a class="code" href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">dptr_</a>;</div> |
| <div class="line"><a name="l00459"></a><span class="lineno"> 459</span>  dltensor_.<a class="code" href="structDLTensor.html#a552ca85f7d04138c876c636e25b3c484">ctx</a> = <a class="code" href="structDLContext.html">DLContext</a>{<span class="keyword">static_cast<</span><a class="code" href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573">DLDeviceType</a><span class="keyword">></span>(<a class="code" href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">dev_mask</a>), <a class="code" href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">dev_id</a>};</div> |
| <div class="line"><a name="l00460"></a><span class="lineno"> 460</span>  dltensor_.<a class="code" href="structDLTensor.html#a8b28a3876c4d157ccea80956f3ec18f5">ndim</a> = <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.<a class="code" href="classmxnet_1_1Tuple.html#a6e0bdfe3f0f34693223f657ef676bef6">ndim</a>();</div> |
| <div class="line"><a name="l00461"></a><span class="lineno"> 461</span>  dltensor_.<a class="code" href="structDLTensor.html#ae0cf02d1a8b19118838670b3b3d857f5">dtype</a> = DTypeTransform(<a class="code" href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">type_flag_</a>);</div> |
| <div class="line"><a name="l00462"></a><span class="lineno"> 462</span>  dltensor_.<a class="code" href="structDLTensor.html#ac982d47788a5251f1cd97375a1a2e49f">shape</a> = <a class="code" href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">shape_</a>.<a class="code" href="classmxnet_1_1TShape.html#a9a822284aceb0b292e0f6c50acddaf5a">data</a>();</div> |
| <div class="line"><a name="l00463"></a><span class="lineno"> 463</span>  dltensor_.<a class="code" href="structDLTensor.html#aa1a066fbde828cf415f77e5dbf962c2e">strides</a> = <span class="keyword">nullptr</span>;</div> |
| <div class="line"><a name="l00464"></a><span class="lineno"> 464</span>  dltensor_.<a class="code" href="structDLTensor.html#a647dfbb2a3b976c4e9d1e17740dca568">byte_offset</a> = 0;</div> |
| <div class="line"><a name="l00465"></a><span class="lineno"> 465</span>  }</div> |
| <div class="line"><a name="l00466"></a><span class="lineno"> 466</span>  </div> |
| <div class="line"><a name="l00467"></a><span class="lineno"> 467</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00469"></a><span class="lineno"> 469</span>  <a class="code" href="structDLTensor.html">DLTensor</a> dltensor_;</div> |
| <div class="line"><a name="l00470"></a><span class="lineno"> 470</span> };</div> |
| <div class="line"><a name="l00471"></a><span class="lineno"> 471</span> } <span class="comment">// namespace mxnet</span></div> |
| <div class="line"><a name="l00472"></a><span class="lineno"> 472</span>  </div> |
| <div class="line"><a name="l00473"></a><span class="lineno"> 473</span> <span class="keyword">namespace </span><a class="code" href="namespacedmlc.html">dmlc</a> {</div> |
| <div class="line"><a name="l00474"></a><span class="lineno"> 474</span> <span class="comment">// Add a few patches to support mxnet::TShape in dmlc/parameter.</span></div> |
| <div class="line"><a name="l00475"></a><span class="lineno"> 475</span> <a class="code" href="namespacedmlc.html#ae0e0e4f86115e4c4e7f761de5c8a6797">DMLC_DECLARE_TYPE_NAME</a>(<a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a>, <span class="stringliteral">"Shape(tuple)"</span>);</div> |
| <div class="line"><a name="l00476"></a><span class="lineno"> 476</span> <a class="code" href="namespacedmlc.html#ae0e0e4f86115e4c4e7f761de5c8a6797">DMLC_DECLARE_TYPE_NAME</a>(<a class="code" href="classmxnet_1_1Tuple.html">mxnet::Tuple<int></a>, <span class="stringliteral">"Shape(tuple)"</span>);</div> |
| <div class="line"><a name="l00477"></a><span class="lineno"> 477</span> <a class="code" href="namespacedmlc.html#ae0e0e4f86115e4c4e7f761de5c8a6797">DMLC_DECLARE_TYPE_NAME</a>(<a class="code" href="classmxnet_1_1Tuple.html">mxnet::Tuple</a><<a class="code" href="classdmlc_1_1optional.html">dmlc::optional<int></a>>, <span class="stringliteral">"Shape(tuple)"</span>);</div> |
| <div class="line"><a name="l00478"></a><span class="lineno"> 478</span> <a class="code" href="namespacedmlc.html#ae0e0e4f86115e4c4e7f761de5c8a6797">DMLC_DECLARE_TYPE_NAME</a>(<a class="code" href="classnnvm_1_1Tuple.html">nnvm::Tuple<int></a>, <span class="stringliteral">"Shape(tuple)"</span>);</div> |
| <div class="line"><a name="l00479"></a><span class="lineno"> 479</span> <a class="code" href="namespacedmlc.html#ae0e0e4f86115e4c4e7f761de5c8a6797">DMLC_DECLARE_TYPE_NAME</a>(<a class="code" href="classnnvm_1_1Tuple.html">nnvm::Tuple</a><<a class="code" href="classdmlc_1_1optional.html">dmlc::optional<int></a>>, <span class="stringliteral">"Shape(tuple)"</span>);</div> |
| <div class="line"><a name="l00480"></a><span class="lineno"> 480</span>  </div> |
| <div class="line"><a name="l00481"></a><span class="lineno"><a class="line" href="namespacedmlc_1_1parameter.html"> 481</a></span> <span class="keyword">namespace </span>parameter {</div> |
| <div class="line"><a name="l00482"></a><span class="lineno"> 482</span>  </div> |
| <div class="line"><a name="l00483"></a><span class="lineno"> 483</span> <span class="keyword">template</span> <></div> |
| <div class="line"><a name="l00484"></a><span class="lineno"><a class="line" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html"> 484</a></span> <span class="keyword">class </span>FieldEntry<<a class="code" href="namespacemxnet.html">mxnet</a>::TShape> : <span class="keyword">public</span> FieldEntryBase<FieldEntry<mxnet::TShape>, mxnet::TShape> {</div> |
| <div class="line"><a name="l00485"></a><span class="lineno"> 485</span>  <span class="keyword">public</span>:</div> |
| <div class="line"><a name="l00486"></a><span class="lineno"><a class="line" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a0755f0abe443e4b57da234ec7b49fde0"> 486</a></span>  <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a0755f0abe443e4b57da234ec7b49fde0">FieldEntry</a>() : enforce_nonzero_(false), expect_ndim_(0) {}</div> |
| <div class="line"><a name="l00487"></a><span class="lineno"> 487</span>  <span class="comment">// parent class</span></div> |
| <div class="line"><a name="l00488"></a><span class="lineno"><a class="line" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#ac6a42c993f66fceb5022b330e5672b6c"> 488</a></span>  <span class="keyword">typedef</span> FieldEntryBase<FieldEntry<mxnet::TShape>, <a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a>> <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#ac6a42c993f66fceb5022b330e5672b6c">Parent</a>;</div> |
| <div class="line"><a name="l00489"></a><span class="lineno"> 489</span>  </div> |
| <div class="line"><a name="l00490"></a><span class="lineno"><a class="line" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a6795bc6e9e883adf946f69e16c644788"> 490</a></span>  <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a6795bc6e9e883adf946f69e16c644788">Check</a>(<span class="keywordtype">void</span>* head)<span class="keyword"> const </span>{</div> |
| <div class="line"><a name="l00491"></a><span class="lineno"> 491</span>  Parent::Check(head);</div> |
| <div class="line"><a name="l00492"></a><span class="lineno"> 492</span>  <a class="code" href="classmxnet_1_1TShape.html">mxnet::TShape</a>& v = this->Get(head);</div> |
| <div class="line"><a name="l00493"></a><span class="lineno"> 493</span>  <span class="keywordflow">if</span> (expect_ndim_ != 0 && v.<a class="code" href="classmxnet_1_1Tuple.html#a6e0bdfe3f0f34693223f657ef676bef6">ndim</a>() != expect_ndim_) {</div> |
| <div class="line"><a name="l00494"></a><span class="lineno"> 494</span>  std::ostringstream os;</div> |
| <div class="line"><a name="l00495"></a><span class="lineno"> 495</span>  os << <span class="stringliteral">"value "</span> << v << <span class="stringliteral">"for Parameter "</span> << this->key_</div> |
| <div class="line"><a name="l00496"></a><span class="lineno"> 496</span>  << <span class="stringliteral">" has wrong dimensions, expected dimension="</span> << expect_ndim_;</div> |
| <div class="line"><a name="l00497"></a><span class="lineno"> 497</span>  <span class="keywordflow">throw</span> dmlc::ParamError(os.str());</div> |
| <div class="line"><a name="l00498"></a><span class="lineno"> 498</span>  }</div> |
| <div class="line"><a name="l00499"></a><span class="lineno"> 499</span>  <span class="keywordflow">if</span> (enforce_nonzero_) {</div> |
| <div class="line"><a name="l00500"></a><span class="lineno"> 500</span>  <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i = 0; i < v.<a class="code" href="classmxnet_1_1Tuple.html#a6e0bdfe3f0f34693223f657ef676bef6">ndim</a>(); ++i) {</div> |
| <div class="line"><a name="l00501"></a><span class="lineno"> 501</span>  <span class="keywordflow">if</span> (v[i] == 0U) {</div> |
| <div class="line"><a name="l00502"></a><span class="lineno"> 502</span>  std::ostringstream os;</div> |
| <div class="line"><a name="l00503"></a><span class="lineno"> 503</span>  os << <span class="stringliteral">"value "</span> << v << <span class="stringliteral">"for Parameter "</span> << this->key_</div> |
| <div class="line"><a name="l00504"></a><span class="lineno"> 504</span>  << <span class="stringliteral">" is invalid, the input shape must be nonzero in all dimensions"</span>;</div> |
| <div class="line"><a name="l00505"></a><span class="lineno"> 505</span>  <span class="keywordflow">throw</span> dmlc::ParamError(os.str());</div> |
| <div class="line"><a name="l00506"></a><span class="lineno"> 506</span>  }</div> |
| <div class="line"><a name="l00507"></a><span class="lineno"> 507</span>  }</div> |
| <div class="line"><a name="l00508"></a><span class="lineno"> 508</span>  }</div> |
| <div class="line"><a name="l00509"></a><span class="lineno"> 509</span>  }</div> |
| <div class="line"><a name="l00510"></a><span class="lineno"><a class="line" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#afa1da539bf9bcb64d3b5a55aec23e8c2"> 510</a></span>  <span class="keyword">inline</span> <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html">FieldEntry<mxnet::TShape></a>& <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#afa1da539bf9bcb64d3b5a55aec23e8c2">enforce_nonzero</a>() {</div> |
| <div class="line"><a name="l00511"></a><span class="lineno"> 511</span>  this->enforce_nonzero_ = <span class="keyword">true</span>;</div> |
| <div class="line"><a name="l00512"></a><span class="lineno"> 512</span>  <span class="keywordflow">return</span> this-><span class="keyword">self</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="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a66f7111b4492f16511239f527b014af6"> 514</a></span>  <span class="keyword">inline</span> <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html">FieldEntry<mxnet::TShape></a>& <a class="code" href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a66f7111b4492f16511239f527b014af6">set_expect_ndim</a>(<span class="keywordtype">int</span> ndim) {</div> |
| <div class="line"><a name="l00515"></a><span class="lineno"> 515</span>  expect_ndim_ = ndim;</div> |
| <div class="line"><a name="l00516"></a><span class="lineno"> 516</span>  <span class="keywordflow">return</span> this-><span class="keyword">self</span>();</div> |
| <div class="line"><a name="l00517"></a><span class="lineno"> 517</span>  }</div> |
| <div class="line"><a name="l00518"></a><span class="lineno"> 518</span>  </div> |
| <div class="line"><a name="l00519"></a><span class="lineno"> 519</span>  <span class="keyword">private</span>:</div> |
| <div class="line"><a name="l00520"></a><span class="lineno"> 520</span>  <span class="comment">// whether all the entries need to be nonzero</span></div> |
| <div class="line"><a name="l00521"></a><span class="lineno"> 521</span>  <span class="keywordtype">bool</span> enforce_nonzero_;</div> |
| <div class="line"><a name="l00522"></a><span class="lineno"> 522</span>  <span class="comment">// expected number of dimension, default = 0 means no restriction.</span></div> |
| <div class="line"><a name="l00523"></a><span class="lineno"> 523</span>  <span class="keywordtype">int</span> expect_ndim_;</div> |
| <div class="line"><a name="l00524"></a><span class="lineno"> 524</span> };</div> |
| <div class="line"><a name="l00525"></a><span class="lineno"> 525</span>  </div> |
| <div class="line"><a name="l00526"></a><span class="lineno"> 526</span> } <span class="comment">// namespace parameter</span></div> |
| <div class="line"><a name="l00527"></a><span class="lineno"> 527</span> } <span class="comment">// namespace dmlc</span></div> |
| <div class="line"><a name="l00528"></a><span class="lineno"> 528</span>  </div> |
| <div class="line"><a name="l00529"></a><span class="lineno"> 529</span> <span class="preprocessor">#endif // MXNET_TENSOR_BLOB_H_</span></div> |
| </div><!-- fragment --></div><!-- contents --> |
| <div class="ttc" id="anamespacemxnet_html"><div class="ttname"><a href="namespacemxnet.html">mxnet</a></div><div class="ttdoc">namespace of mxnet</div><div class="ttdef"><b>Definition:</b> api_registry.h:33</div></div> |
| <div class="ttc" id="adlpack_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0"><div class="ttname"><a href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a8bd73646a2aa8465201b59ab7d1a83e0">kDLFloat</a></div><div class="ttdeci">@ kDLFloat</div><div class="ttdef"><b>Definition:</b> dlpack.h:82</div></div> |
| <div class="ttc" id="anamespacemxnet_html_a512ee02aece77cb0bf39b6e2ded77686"><div class="ttname"><a href="namespacemxnet.html#a512ee02aece77cb0bf39b6e2ded77686">mxnet::kCPU</a></div><div class="ttdeci">constexpr const int kCPU</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:42</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_aae92a89345aad79c8c3b451b9e3b84a5"><div class="ttname"><a href="classmxnet_1_1TBlob.html#aae92a89345aad79c8c3b451b9e3b84a5">mxnet::TBlob::TBlob</a></div><div class="ttdeci">TBlob(const mshadow::Tensor< Device, dim, DType > &src)</div><div class="ttdoc">constructor from tensor</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:145</div></div> |
| <div class="ttc" id="astructDLDataType_html"><div class="ttname"><a href="structDLDataType.html">DLDataType</a></div><div class="ttdoc">The data type the tensor can hold.</div><div class="ttdef"><b>Definition:</b> dlpack.h:94</div></div> |
| <div class="ttc" id="astructmshadow_1_1Shape_html_ac6d667fde1a8180b64f475f5d33ea58f"><div class="ttname"><a href="structmshadow_1_1Shape.html#ac6d667fde1a8180b64f475f5d33ea58f">mshadow::Shape::Size</a></div><div class="ttdeci">MSHADOW_XINLINE index_t Size(void) const</div><div class="ttdef"><b>Definition:</b> tensor.h:158</div></div> |
| <div class="ttc" id="aclassdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4_html_a0755f0abe443e4b57da234ec7b49fde0"><div class="ttname"><a href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a0755f0abe443e4b57da234ec7b49fde0">dmlc::parameter::FieldEntry< mxnet::TShape >::FieldEntry</a></div><div class="ttdeci">FieldEntry()</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:486</div></div> |
| <div class="ttc" id="astructmshadow_1_1Stream_html"><div class="ttname"><a href="structmshadow_1_1Stream.html">mshadow::Stream</a></div><div class="ttdoc">computaion stream structure, used for asynchronous computations</div><div class="ttdef"><b>Definition:</b> tensor.h:488</div></div> |
| <div class="ttc" id="aclassmxnet_1_1Tuple_html"><div class="ttname"><a href="classmxnet_1_1Tuple.html">mxnet::Tuple</a></div><div class="ttdoc">A dynamic sized array data structure that is optimized for storing small number of elements with same...</div><div class="ttdef"><b>Definition:</b> tuple.h:57</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="aclassmxnet_1_1TBlob_html_adcdc6de67f81454f0a142bfc9fd38c9e"><div class="ttname"><a href="classmxnet_1_1TBlob.html#adcdc6de67f81454f0a142bfc9fd38c9e">mxnet::TBlob::TBlob</a></div><div class="ttdeci">TBlob(void)</div><div class="ttdoc">default constructor, default copy assign will work</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:77</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="aclassmxnet_1_1TBlob_html_ad931186405f2b40a327278a8f5ff57cd"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ad931186405f2b40a327278a8f5ff57cd">mxnet::TBlob::dptr</a></div><div class="ttdeci">DType * dptr() const</div><div class="ttdoc">get pointer in dtype</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:249</div></div> |
| <div class="ttc" id="astructDLTensor_html_a8a6408813fe13f1985379f022ade0947"><div class="ttname"><a href="structDLTensor.html#a8a6408813fe13f1985379f022ade0947">DLTensor::data</a></div><div class="ttdeci">void * data</div><div class="ttdoc">The opaque data pointer points to the allocated data. This will be CUDA device pointer or cl_mem hand...</div><div class="ttdef"><b>Definition:</b> dlpack.h:132</div></div> |
| <div class="ttc" id="aclassnnvm_1_1Tuple_html"><div class="ttname"><a href="classnnvm_1_1Tuple.html">nnvm::Tuple</a></div><div class="ttdoc">A dynamic sized array data structure that is optimized for storing small number of elements with same...</div><div class="ttdef"><b>Definition:</b> tuple.h:52</div></div> |
| <div class="ttc" id="anamespacedmlc_html"><div class="ttname"><a href="namespacedmlc.html">dmlc</a></div><div class="ttdoc">namespace for dmlc</div><div class="ttdef"><b>Definition:</b> array_view.h:12</div></div> |
| <div class="ttc" id="astructDLContext_html_a9e7b666f60b074a4f48943da69cee9ce"><div class="ttname"><a href="structDLContext.html#a9e7b666f60b074a4f48943da69cee9ce">DLContext::device_type</a></div><div class="ttdeci">DLDeviceType device_type</div><div class="ttdoc">The device type used in the device.</div><div class="ttdef"><b>Definition:</b> dlpack.h:71</div></div> |
| <div class="ttc" id="aclassdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4_html_a6795bc6e9e883adf946f69e16c644788"><div class="ttname"><a href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a6795bc6e9e883adf946f69e16c644788">dmlc::parameter::FieldEntry< mxnet::TShape >::Check</a></div><div class="ttdeci">virtual void Check(void *head) const</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:490</div></div> |
| <div class="ttc" id="aclassdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4_html"><div class="ttname"><a href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html">dmlc::parameter::FieldEntry< mxnet::TShape ></a></div><div class="ttdef"><b>Definition:</b> tensor_blob.h:484</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="aclassmxnet_1_1TBlob_html_a31c0750cf2642b68af00c00029c99967"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a31c0750cf2642b68af00c00029c99967">mxnet::TBlob::TBlob</a></div><div class="ttdeci">TBlob(void *dptr, const mxnet::TShape &shape, int dev_mask, int type_flag, int dev_id=-1)</div><div class="ttdoc">constructor that construct TBlob from contiguous memory</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:100</div></div> |
| <div class="ttc" id="astructDLTensor_html_a8b28a3876c4d157ccea80956f3ec18f5"><div class="ttname"><a href="structDLTensor.html#a8b28a3876c4d157ccea80956f3ec18f5">DLTensor::ndim</a></div><div class="ttdeci">int ndim</div><div class="ttdoc">Number of dimensions.</div><div class="ttdef"><b>Definition:</b> dlpack.h:136</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_ac3abff8bace8f3afb3143d85551a18ac"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ac3abff8bace8f3afb3143d85551a18ac">mxnet::TBlob::CheckContiguous</a></div><div class="ttdeci">bool CheckContiguous(void) const</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:186</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a9bb2615c144e237afea3875e618527d4"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a9bb2615c144e237afea3875e618527d4">mxnet::TBlob::FlatTo2D</a></div><div class="ttdeci">mshadow::Tensor< Device, 2, DType > FlatTo2D(mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">flatten the tensor to 2 dimension, collapse the higher dimensions together</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:208</div></div> |
| <div class="ttc" id="astructDLContext_html"><div class="ttname"><a href="structDLContext.html">DLContext</a></div><div class="ttdoc">A Device context for Tensor and operator.</div><div class="ttdef"><b>Definition:</b> dlpack.h:69</div></div> |
| <div class="ttc" id="adlpack_8h_html_a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e"><div class="ttname"><a href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573ab736fabc9e37c2acf3b09b88fdc65c9e">kDLCPU</a></div><div class="ttdeci">@ kDLCPU</div><div class="ttdoc">CPU device.</div><div class="ttdef"><b>Definition:</b> dlpack.h:40</div></div> |
| <div class="ttc" id="astructDLTensor_html_a647dfbb2a3b976c4e9d1e17740dca568"><div class="ttname"><a href="structDLTensor.html#a647dfbb2a3b976c4e9d1e17740dca568">DLTensor::byte_offset</a></div><div class="ttdeci">uint64_t byte_offset</div><div class="ttdoc">The offset in bytes to the beginning pointer to data.</div><div class="ttdef"><b>Definition:</b> dlpack.h:147</div></div> |
| <div class="ttc" id="astructmshadow_1_1Tensor_html"><div class="ttname"><a href="structmshadow_1_1Tensor.html">mshadow::Tensor</a></div><div class="ttdoc">general tensor</div><div class="ttdef"><b>Definition:</b> tensor.h:525</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a4e5f4953be2c16724ecff0afc5ea2e10"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a4e5f4953be2c16724ecff0afc5ea2e10">mxnet::TBlob::dptr_</a></div><div class="ttdeci">void * dptr_</div><div class="ttdoc">pointer to the data</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:70</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_aa676224250ad5299e1b914da556cdc8d"><div class="ttname"><a href="classmxnet_1_1TBlob.html#aa676224250ad5299e1b914da556cdc8d">mxnet::TBlob::Size</a></div><div class="ttdeci">size_t Size(void) const</div><div class="ttdoc">total number of elements in the tensor</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:244</div></div> |
| <div class="ttc" id="astructmshadow_1_1cpu_html_a57f74b500825e21c0c246ad5b4e5016b"><div class="ttname"><a href="structmshadow_1_1cpu.html#a57f74b500825e21c0c246ad5b4e5016b">mshadow::cpu::kDevMask</a></div><div class="ttdeci">static const int kDevMask</div><div class="ttdoc">device flag number, identifies this device</div><div class="ttdef"><b>Definition:</b> tensor.h:43</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a2caa6723ffda5908eb5c2e062ac77ebd"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a2caa6723ffda5908eb5c2e062ac77ebd">mxnet::TBlob::dev_id</a></div><div class="ttdeci">int dev_id() const</div><div class="ttdoc">device index of the corresponding device</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:261</div></div> |
| <div class="ttc" id="adlpack_8h_html_a49db572e18d84c47738fc746568fb573"><div class="ttname"><a href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573">DLDeviceType</a></div><div class="ttdeci">DLDeviceType</div><div class="ttdoc">The device type in DLContext.</div><div class="ttdef"><b>Definition:</b> dlpack.h:38</div></div> |
| <div class="ttc" id="aclassmxnet_1_1Tuple_html_a6e0bdfe3f0f34693223f657ef676bef6"><div class="ttname"><a href="classmxnet_1_1Tuple.html#a6e0bdfe3f0f34693223f657ef676bef6">mxnet::Tuple::ndim</a></div><div class="ttdeci">int ndim() const</div><div class="ttdef"><b>Definition:</b> tuple.h:217</div></div> |
| <div class="ttc" id="anamespacedmlc_html_ae0e0e4f86115e4c4e7f761de5c8a6797"><div class="ttname"><a href="namespacedmlc.html#ae0e0e4f86115e4c4e7f761de5c8a6797">dmlc::DMLC_DECLARE_TYPE_NAME</a></div><div class="ttdeci">DMLC_DECLARE_TYPE_NAME(optional< int >, "int or None")</div><div class="ttdoc">description for optional int</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="aclassmxnet_1_1TBlob_html_ad17880c664c448bb159b356e0c2b875c"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ad17880c664c448bb159b356e0c2b875c">mxnet::TBlob::get_with_shape</a></div><div class="ttdeci">mshadow::Tensor< Device, dim, DType > get_with_shape(const mshadow::Shape< dim > &shape, mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">fetch a tensor in given shape If size do not match the stored size, an error will be issued</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:299</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="aclassmxnet_1_1TBlob_html_a7971ef6867cd7c94ee32de023f8dbe23"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a7971ef6867cd7c94ee32de023f8dbe23">mxnet::TBlob::operator=</a></div><div class="ttdeci">TBlob & operator=(const TBlob &src)</div><div class="ttdoc">assignment from TBlob (copy assignment)</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:176</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_ad8bcf56e13082fc0607e21f08b14758f"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ad8bcf56e13082fc0607e21f08b14758f">mxnet::TBlob::dev_mask</a></div><div class="ttdeci">int dev_mask() const</div><div class="ttdoc">device mask of the corresponding device</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:257</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a77a8929e46d06e5d6a0026bcc9e48851"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a77a8929e46d06e5d6a0026bcc9e48851">mxnet::TBlob::ndim</a></div><div class="ttdeci">int ndim(void) const</div><div class="ttdoc">return number of dimension of the tensor inside</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:231</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="astructDLTensor_html_aa1a066fbde828cf415f77e5dbf962c2e"><div class="ttname"><a href="structDLTensor.html#aa1a066fbde828cf415f77e5dbf962c2e">DLTensor::strides</a></div><div class="ttdeci">int64_t * strides</div><div class="ttdoc">strides of the tensor (in number of elements, not bytes) can be NULL, indicating tensor is compact an...</div><div class="ttdef"><b>Definition:</b> dlpack.h:145</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a4649b740b80a6045754cb2389cfcca61"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a4649b740b80a6045754cb2389cfcca61">mxnet::TBlob::type_flag_</a></div><div class="ttdeci">int type_flag_</div><div class="ttdoc">type flag of the tensor blob</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:74</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_ae2bc6bbce5faf2c99d3b0ea462a3fcfa"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ae2bc6bbce5faf2c99d3b0ea462a3fcfa">mxnet::TBlob::TBlob</a></div><div class="ttdeci">TBlob(DType *dptr, const mxnet::TShape &shape, int dev_mask, int dev_id=-1)</div><div class="ttdoc">constructor that construct TBlob from contiguous memory</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:88</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_af38cdc5d0bcce77bb4f96251167cd7a0"><div class="ttname"><a href="classmxnet_1_1TBlob.html#af38cdc5d0bcce77bb4f96251167cd7a0">mxnet::TBlob::get</a></div><div class="ttdeci">mshadow::Tensor< Device, dim, DType > get(mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">fetch the tensor, with respect to specific dimension if dim do not match the stored dimension,...</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:282</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_ad3263d9d1ef40678cd592be822d1c7be"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ad3263d9d1ef40678cd592be822d1c7be">mxnet::TBlob::operator=</a></div><div class="ttdeci">TBlob & operator=(const mshadow::Tensor< Device, dim, DType > &src)</div><div class="ttdoc">assignment from tensor</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:164</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a765ee89940e22fcf1cb24e8048da95e2"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a765ee89940e22fcf1cb24e8048da95e2">mxnet::TBlob::FlatToKD</a></div><div class="ttdeci">mshadow::Tensor< Device, dim, DType > FlatToKD(mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">flatten the tensor to specified number of dimensions, collapse the highest dimensions or pad with hig...</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:350</div></div> |
| <div class="ttc" id="adlpack_8h_html_a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680"><div class="ttname"><a href="dlpack_8h.html#a49db572e18d84c47738fc746568fb573ae04bfdcc919948d170b2ba26da09e680">kDLGPU</a></div><div class="ttdeci">@ kDLGPU</div><div class="ttdoc">CUDA GPU device.</div><div class="ttdef"><b>Definition:</b> dlpack.h:42</div></div> |
| <div class="ttc" id="astructDLDataType_html_aaa84983c485752e4401d80ce0e2b6e54"><div class="ttname"><a href="structDLDataType.html#aaa84983c485752e4401d80ce0e2b6e54">DLDataType::code</a></div><div class="ttdeci">uint8_t code</div><div class="ttdoc">Type code of base types. We keep it uint8_t instead of DLDataTypeCode for minimal memory footprint,...</div><div class="ttdef"><b>Definition:</b> dlpack.h:100</div></div> |
| <div class="ttc" id="astructmshadow_1_1Tensor_html_ad10c7414c5948e789e8761df2083c4e5"><div class="ttname"><a href="structmshadow_1_1Tensor.html#ad10c7414c5948e789e8761df2083c4e5">mshadow::Tensor::shape_</a></div><div class="ttdeci">Shape< dimension > shape_</div><div class="ttdoc">shape of the tensor</div><div class="ttdef"><b>Definition:</b> tensor.h:541</div></div> |
| <div class="ttc" id="aclassmxnet_1_1NDArray_html"><div class="ttname"><a href="classmxnet_1_1NDArray.html">mxnet::NDArray</a></div><div class="ttdoc">ndarray interface</div><div class="ttdef"><b>Definition:</b> ndarray.h:82</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="astructDLDataType_html_a412e36fa9c58e808a2333e9044c5d9b5"><div class="ttname"><a href="structDLDataType.html#a412e36fa9c58e808a2333e9044c5d9b5">DLDataType::bits</a></div><div class="ttdeci">uint8_t bits</div><div class="ttdoc">Number of bits, common choices are 8, 16, 32.</div><div class="ttdef"><b>Definition:</b> dlpack.h:104</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="adlpack_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e"><div class="ttname"><a href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a291cf38955be5c45c3aa50cea13c5e6e">kDLInt</a></div><div class="ttdeci">@ kDLInt</div><div class="ttdef"><b>Definition:</b> dlpack.h:80</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html"><div class="ttname"><a href="classmxnet_1_1TBlob.html">mxnet::TBlob</a></div><div class="ttdoc">tensor blob class that can be used to hold tensor of any dimension, any device and any data type,...</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:65</div></div> |
| <div class="ttc" id="astructDLTensor_html_a552ca85f7d04138c876c636e25b3c484"><div class="ttname"><a href="structDLTensor.html#a552ca85f7d04138c876c636e25b3c484">DLTensor::ctx</a></div><div class="ttdeci">DLContext ctx</div><div class="ttdoc">The device context of the tensor.</div><div class="ttdef"><b>Definition:</b> dlpack.h:134</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="aclassdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4_html_a66f7111b4492f16511239f527b014af6"><div class="ttname"><a href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#a66f7111b4492f16511239f527b014af6">dmlc::parameter::FieldEntry< mxnet::TShape >::set_expect_ndim</a></div><div class="ttdeci">FieldEntry< mxnet::TShape > & set_expect_ndim(int ndim)</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:514</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a684ef7a1354dc47b808ef54cfecf81b7"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a684ef7a1354dc47b808ef54cfecf81b7">mxnet::TBlob::FlatTo3D</a></div><div class="ttdeci">mshadow::Tensor< Device, 3, DType > FlatTo3D(int axis_begin, int axis_end, mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">flatten the tensor to 3 dimension, collapse the dimension: [0, axis_begin), [axis_begin,...</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:336</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a44c93b8a96203a1a85fa9d4b85e123c7"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a44c93b8a96203a1a85fa9d4b85e123c7">mxnet::TBlob::size</a></div><div class="ttdeci">index_t size(index_t idx) const</div><div class="ttdoc">return size of i-th dimension, start counting from highest dimension. return type needs to be a signe...</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:240</div></div> |
| <div class="ttc" id="astructDLTensor_html_ae0cf02d1a8b19118838670b3b3d857f5"><div class="ttname"><a href="structDLTensor.html#ae0cf02d1a8b19118838670b3b3d857f5">DLTensor::dtype</a></div><div class="ttdeci">DLDataType dtype</div><div class="ttdoc">The data type of the pointer.</div><div class="ttdef"><b>Definition:</b> dlpack.h:138</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a4b0a53821dec20c558a030ed3df14dda"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a4b0a53821dec20c558a030ed3df14dda">mxnet::TBlob::dltensor</a></div><div class="ttdeci">const DLTensor & dltensor() const</div><div class="ttdoc">return the corresponding DLTensor</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:268</div></div> |
| <div class="ttc" id="adlpack_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a"><div class="ttname"><a href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50aa19211076c5ae99e281ea430a25b102a">kDLUInt</a></div><div class="ttdeci">@ kDLUInt</div><div class="ttdef"><b>Definition:</b> dlpack.h:81</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="aclassmxnet_1_1TShape_html_a505a8c1666a7aa7f6e4108b808e6a897"><div class="ttname"><a href="classmxnet_1_1TShape.html#a505a8c1666a7aa7f6e4108b808e6a897">mxnet::TShape::Size</a></div><div class="ttdeci">size_t Size() const</div><div class="ttdef"><b>Definition:</b> tuple.h:523</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TShape_html_a9a822284aceb0b292e0f6c50acddaf5a"><div class="ttname"><a href="classmxnet_1_1TShape.html#a9a822284aceb0b292e0f6c50acddaf5a">mxnet::TShape::data</a></div><div class="ttdeci">const dim_t * data() const</div><div class="ttdef"><b>Definition:</b> tuple.h:552</div></div> |
| <div class="ttc" id="astructmshadow_1_1Tensor_3_01Device_00_011_00_01DType_01_4_html"><div class="ttname"><a href="structmshadow_1_1Tensor_3_01Device_00_011_00_01DType_01_4.html">mshadow::Tensor< Device, 1, DType ></a></div><div class="ttdef"><b>Definition:</b> tensor.h:673</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="aclassmxnet_1_1TBlob_html_a44594fd9712fb19061db95a070c46000"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a44594fd9712fb19061db95a070c46000">mxnet::TBlob::TBlob</a></div><div class="ttdeci">TBlob(const DLTensor &dltensor)</div><div class="ttdoc">constructor that construct TBlob from DLTensor</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:108</div></div> |
| <div class="ttc" id="astructDLTensor_html_ac982d47788a5251f1cd97375a1a2e49f"><div class="ttname"><a href="structDLTensor.html#ac982d47788a5251f1cd97375a1a2e49f">DLTensor::shape</a></div><div class="ttdeci">int64_t * shape</div><div class="ttdoc">The shape of the tensor.</div><div class="ttdef"><b>Definition:</b> dlpack.h:140</div></div> |
| <div class="ttc" id="aclassdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4_html_afa1da539bf9bcb64d3b5a55aec23e8c2"><div class="ttname"><a href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#afa1da539bf9bcb64d3b5a55aec23e8c2">dmlc::parameter::FieldEntry< mxnet::TShape >::enforce_nonzero</a></div><div class="ttdeci">FieldEntry< mxnet::TShape > & enforce_nonzero()</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:510</div></div> |
| <div class="ttc" id="anamespacemxnet_html_ab96d1eea6c8d006db2b33bbdf9d90bd2"><div class="ttname"><a href="namespacemxnet.html#ab96d1eea6c8d006db2b33bbdf9d90bd2">mxnet::kTVMNDArrayTypeCode</a></div><div class="ttdeci">constexpr const int kTVMNDArrayTypeCode</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:48</div></div> |
| <div class="ttc" id="anamespacemxnet_html_abadedc1b42ee3e5b8852e63783b982ef"><div class="ttname"><a href="namespacemxnet.html#abadedc1b42ee3e5b8852e63783b982ef">mxnet::index_t</a></div><div class="ttdeci">mshadow::index_t index_t</div><div class="ttdoc">index type usually use unsigned</div><div class="ttdef"><b>Definition:</b> base.h:81</div></div> |
| <div class="ttc" id="astructmshadow_1_1Shape_html"><div class="ttname"><a href="structmshadow_1_1Shape.html">mshadow::Shape< dim ></a></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="astructmshadow_1_1Tensor_html_ad86d6759c585efb5229b3a0659973838"><div class="ttname"><a href="structmshadow_1_1Tensor.html#ad86d6759c585efb5229b3a0659973838">mshadow::Tensor::dptr_</a></div><div class="ttdeci">DType * dptr_</div><div class="ttdoc">pointer to the data</div><div class="ttdef"><b>Definition:</b> tensor.h:539</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="aclassmxnet_1_1TBlob_html_ac64b2d46ddd92bf972a4e5b49afbf325"><div class="ttname"><a href="classmxnet_1_1TBlob.html#ac64b2d46ddd92bf972a4e5b49afbf325">mxnet::TBlob::TBlob</a></div><div class="ttdeci">TBlob(const TBlob &src)</div><div class="ttdoc">constructor from TBlob (copy constructor)</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:152</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_af220608e394f02ad3739ac048757175b"><div class="ttname"><a href="classmxnet_1_1TBlob.html#af220608e394f02ad3739ac048757175b">mxnet::TBlob::FlatTo1D</a></div><div class="ttdeci">mshadow::Tensor< Device, 1, DType > FlatTo1D(mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">flatten the tensor to 1 dimension, collapse all the dimensions together.</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:226</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a0d051bba948b8a69a9d1d2690fad7400"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a0d051bba948b8a69a9d1d2690fad7400">mxnet::TBlob::FlatTo3D</a></div><div class="ttdeci">mshadow::Tensor< Device, 3, DType > FlatTo3D(int axis, mshadow::Stream< Device > *stream=nullptr) const</div><div class="ttdoc">flatten the tensor to 3 dimension, collapse the dimension before and after specified axis.</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:319</div></div> |
| <div class="ttc" id="ajson_8h_html"><div class="ttname"><a href="json_8h.html">json.h</a></div><div class="ttdoc">Lightweight JSON Reader/Writer that read save into C++ data structs. This includes STL composites and...</div></div> |
| <div class="ttc" id="anamespacemshadow_html_a05e468ef4d8882fccce53ae951b7bdbd"><div class="ttname"><a href="namespacemshadow.html#a05e468ef4d8882fccce53ae951b7bdbd">mshadow::Shape1</a></div><div class="ttdeci">MSHADOW_XINLINE Shape< 1 > Shape1(index_t s0)</div><div class="ttdoc">construct a one dimension shape, stride will equal s0</div><div class="ttdef"><b>Definition:</b> tensor.h:220</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_af0e813ff066c775051d1f1f78a4a2029"><div class="ttname"><a href="classmxnet_1_1TBlob.html#af0e813ff066c775051d1f1f78a4a2029">mxnet::TBlob::shape_</a></div><div class="ttdeci">mxnet::TShape shape_</div><div class="ttdoc">shape of the tensor</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:72</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TShape_html"><div class="ttname"><a href="classmxnet_1_1TShape.html">mxnet::TShape</a></div><div class="ttdoc">A Shape class that is used to represent shape of each tensor.</div><div class="ttdef"><b>Definition:</b> tuple.h:440</div></div> |
| <div class="ttc" id="aclassdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4_html_ac6a42c993f66fceb5022b330e5672b6c"><div class="ttname"><a href="classdmlc_1_1parameter_1_1FieldEntry_3_01mxnet_1_1TShape_01_4.html#ac6a42c993f66fceb5022b330e5672b6c">dmlc::parameter::FieldEntry< mxnet::TShape >::Parent</a></div><div class="ttdeci">FieldEntryBase< FieldEntry< mxnet::TShape >, mxnet::TShape > Parent</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:488</div></div> |
| <div class="ttc" id="astructDLContext_html_a7cd00ca96ac03a1f8ea2b2b3a565794d"><div class="ttname"><a href="structDLContext.html#a7cd00ca96ac03a1f8ea2b2b3a565794d">DLContext::device_id</a></div><div class="ttdeci">int device_id</div><div class="ttdoc">The device index.</div><div class="ttdef"><b>Definition:</b> dlpack.h:73</div></div> |
| <div class="ttc" id="astructDLDataType_html_a24de092fe56d358c1582e41499862594"><div class="ttname"><a href="structDLDataType.html#a24de092fe56d358c1582e41499862594">DLDataType::lanes</a></div><div class="ttdeci">uint16_t lanes</div><div class="ttdoc">Number of lanes in the type, used for vector types.</div><div class="ttdef"><b>Definition:</b> dlpack.h:106</div></div> |
| <div class="ttc" id="astructDLTensor_html"><div class="ttname"><a href="structDLTensor.html">DLTensor</a></div><div class="ttdoc">Plain C Tensor object, does not manage memory.</div><div class="ttdef"><b>Definition:</b> dlpack.h:112</div></div> |
| <div class="ttc" id="aclassmxnet_1_1TBlob_html_a10111644a1abe2035ec3e869187847dc"><div class="ttname"><a href="classmxnet_1_1TBlob.html#a10111644a1abe2035ec3e869187847dc">mxnet::TBlob::reshape</a></div><div class="ttdeci">TBlob reshape(const mxnet::TShape &shape) const</div><div class="ttdoc">reshape to shape</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:194</div></div> |
| <div class="ttc" id="adlpack_8h_html"><div class="ttname"><a href="dlpack_8h.html">dlpack.h</a></div><div class="ttdoc">The common header of DLPack.</div></div> |
| <div class="ttc" id="anamespacemxnet_html_a43ebce8ff01c0bee0dddb4417ea1239d"><div class="ttname"><a href="namespacemxnet.html#a43ebce8ff01c0bee0dddb4417ea1239d">mxnet::kGPU</a></div><div class="ttdeci">constexpr const int kGPU</div><div class="ttdef"><b>Definition:</b> tensor_blob.h:43</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="aclassdmlc_1_1optional_html"><div class="ttname"><a href="classdmlc_1_1optional.html">dmlc::optional</a></div><div class="ttdoc">c++17 compatible optional class.</div><div class="ttdef"><b>Definition:</b> optional.h:43</div></div> |
| <div class="ttc" id="ainclude_2mxnet_2base_8h_html"><div class="ttname"><a href="include_2mxnet_2base_8h.html">base.h</a></div><div class="ttdoc">configuration of MXNet as well as basic data structure.</div></div> |
| <div class="ttc" id="anamespacemxnet_html_a4421326e33f2a848cb2fa3e70ab24ff4"><div class="ttname"><a href="namespacemxnet.html#a4421326e33f2a848cb2fa3e70ab24ff4">mxnet::real_t</a></div><div class="ttdeci">mshadow::default_real_t real_t</div><div class="ttdoc">data type that will be used to store ndarray</div><div class="ttdef"><b>Definition:</b> base.h:85</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="adlpack_8h_html_a2a5dbaa5fb67f4845a973c09a7f2ec50a6a739f1b3ce2f8978da99a5ac4a887a6"><div class="ttname"><a href="dlpack_8h.html#a2a5dbaa5fb67f4845a973c09a7f2ec50a6a739f1b3ce2f8978da99a5ac4a887a6">kDLBfloat</a></div><div class="ttdeci">@ kDLBfloat</div><div class="ttdef"><b>Definition:</b> dlpack.h:83</div></div> |
| <!-- start footer part --> |
| <hr class="footer"/><address class="footer"><small> |
| Generated on Thu Jan 5 2023 03:47:40 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> |
