<ahref="device_2tensor__foreach_8h.html">Go to the documentation of this file.</a><divclass="fragment"><divclass="line"><aname="l00001"></a><spanclass="lineno"> 1</span> <spanclass="comment">/***************************************************************************************************</span></div><divclass="line"><aname="l00002"></a><spanclass="lineno"> 2</span> <spanclass="comment"> * Copyright (c) 2017-2019, NVIDIA CORPORATION. All rights reserved.</span></div><divclass="line"><aname="l00003"></a><spanclass="lineno"> 3</span> <spanclass="comment"> *</span></div><divclass="line"><aname="l00004"></a><spanclass="lineno"> 4</span> <spanclass="comment"> * Redistribution and use in source and binary forms, with or without modification, are permitted</span></div><divclass="line"><aname="l00005"></a><spanclass="lineno"> 5</span> <spanclass="comment"> * provided that the following conditions are met:</span></div><divclass="line"><aname="l00006"></a><spanclass="lineno"> 6</span> <spanclass="comment"> * * Redistributions of source code must retain the above copyright notice, this list of</span></div><divclass="line"><aname="l00007"></a><spanclass="lineno"> 7</span> <spanclass="comment"> * conditions and the following disclaimer.</span></div><divclass="line"><aname="l00008"></a><spanclass="lineno"> 8</span> <spanclass="comment"> * * Redistributions in binary form must reproduce the above copyright notice, this list of</span></div><divclass="line"><aname="l00009"></a><spanclass="lineno"> 9</span> <spanclass="comment"> * conditions and the following disclaimer in the documentation and/or other materials</span></div><divclass="line"><aname="l00010"></a><spanclass="lineno"> 10</span> <spanclass="comment"> * provided with the distribution.</span></div><divclass="line"><aname="l00011"></a><spanclass="lineno"> 11</span> <spanclass="comment"> * * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used</span></div><divclass="line"><aname="l00012"></a><spanclass="lineno"> 12</span> <spanclass="comment"> * to endorse or promote products derived from this software without specific prior written</span></div><divclass="line"><aname="l00013"></a><spanclass="lineno"> 13</span> <spanclass="comment"> * permission.</span></div><divclass="line"><aname="l00014"></a><spanclass="lineno"> 14</span> <spanclass="comment"> *</span></div><divclass="line"><aname="l00015"></a><spanclass="lineno"> 15</span> <spanclass="comment"> * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR</span></div><divclass="line"><aname="l00016"></a><spanclass="lineno"> 16</span> <spanclass="comment"> * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND</span></div><divclass="line"><aname="l00017"></a><spanclass="lineno"> 17</span> <spanclass="comment"> * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE</span></div><divclass="line"><aname="l00018"></a><spanclass="lineno"> 18</span> <spanclass="comment"> * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,</span></div><divclass="line"><aname="l00019"></a><spanclass="lineno"> 19</span> <spanclass="comment"> * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;</span></div><divclass="line"><aname="l00020"></a><spanclass="lineno"> 20</span> <spanclass="comment"> * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,</span></div><divclass="line"><aname="l00021"></a><spanclass="lineno"> 21</span> <spanclass="comment"> * STRICT LIABILITY, OR TOR (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE</span></div><divclass="line"><ana
<divclass="ttc"id="structcutlass_1_1reference_1_1device_1_1TensorDiagonalForEach_html_adab64f903f234d0266400bd2416134ee"><divclass="ttname"><ahref="structcutlass_1_1reference_1_1device_1_1TensorDiagonalForEach.html#adab64f903f234d0266400bd2416134ee">cutlass::reference::device::TensorDiagonalForEach::TensorDiagonalForEach</a></div><divclass="ttdeci">TensorDiagonalForEach(Coord< Rank > size, Params params=Params(), int start=0, int end=-1, int block_size=128)</div><divclass="ttdoc">Constructor performs the operation. </div><divclass="ttdef"><b>Definition:</b> device/tensor_foreach.h:75</div></div>
<divclass="ttc"id="structcutlass_1_1reference_1_1device_1_1TensorForEach_html_ad693751cf94aea83a14235a5ec7c7e92"><divclass="ttname"><ahref="structcutlass_1_1reference_1_1device_1_1TensorForEach.html#ad693751cf94aea83a14235a5ec7c7e92">cutlass::reference::device::TensorForEach::TensorForEach</a></div><divclass="ttdeci">TensorForEach(Coord< Rank > size, Params params=Params(), int grid_size=0, int block_size=0)</div><divclass="ttdoc">Constructor performs the operation. </div><divclass="ttdef"><b>Definition:</b> device/tensor_foreach.h:42</div></div>
<divclass="ttc"id="structcutlass_1_1reference_1_1device_1_1TensorDiagonalForEach_html"><divclass="ttname"><ahref="structcutlass_1_1reference_1_1device_1_1TensorDiagonalForEach.html">cutlass::reference::device::TensorDiagonalForEach</a></div><divclass="ttdoc">Launches a kernel calling a functor for each element along a tensor&#39;s diagonal. </div><divclass="ttdef"><b>Definition:</b> device/tensor_foreach.h:72</div></div>
<divclass="ttc"id="structcutlass_1_1reference_1_1device_1_1BlockForEach_html_a161e212b9b7ddbac36888de97538e106"><divclass="ttname"><ahref="structcutlass_1_1reference_1_1device_1_1BlockForEach.html#a161e212b9b7ddbac36888de97538e106">cutlass::reference::device::BlockForEach::BlockForEach</a></div><divclass="ttdeci">BlockForEach(Element *ptr, size_t capacity, typename Func::Params params=typename Func::Params(), int grid_size=0, int block_size=0)</div><divclass="ttdoc">Constructor performs the operation. </div><divclass="ttdef"><b>Definition:</b> device/tensor_foreach.h:95</div></div>
<divclass="ttc"id="structcutlass_1_1reference_1_1device_1_1TensorForEach_html"><divclass="ttname"><ahref="structcutlass_1_1reference_1_1device_1_1TensorForEach.html">cutlass::reference::device::TensorForEach</a></div><divclass="ttdoc">Launches a kernel calling a functor for each element in a tensor&#39;s index space. </div><divclass="ttdef"><b>Definition:</b> device/tensor_foreach.h:39</div></div>
<divclass="ttc"id="structcutlass_1_1Coord_html"><divclass="ttname"><ahref="structcutlass_1_1Coord.html">cutlass::Coord</a></div><divclass="ttdoc">Statically-sized array specifying Coords within a tensor. </div><divclass="ttdef"><b>Definition:</b> coord.h:43</div></div>
<divclass="ttc"id="cutlass_8h_html"><divclass="ttname"><ahref="cutlass_8h.html">cutlass.h</a></div><divclass="ttdoc">Basic include for CUTLASS. </div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->
<hrclass="footer"/><addressclass="footer"><small>
Generated by  <ahref="http://www.doxygen.org/index.html">