mirror of
https://github.com/ml-explore/mlx.git
synced 2025-11-04 02:28:13 +08:00
rebase
This commit is contained in:
CircleCI Docs
83
docs/build/html/scatter_8h.html
vendored
83
docs/build/html/scatter_8h.html
vendored
@@ -99,84 +99,18 @@ $(function(){ initResizable(false); });
|
||||
<table class="memberdecls">
|
||||
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a id="func-members" name="func-members"></a>
|
||||
Functions</h2></td></tr>
|
||||
<tr class="memitem:ac1fd901b5ef9f39be32316a17297781e" id="r_ac1fd901b5ef9f39be32316a17297781e"><td class="memTemplParams" colspan="2">template<typename T , typename IdxT , typename Op , int NIDX> </td></tr>
|
||||
<tr class="memitem:ac1fd901b5ef9f39be32316a17297781e"><td class="memTemplItemLeft" align="right" valign="top">METAL_FUNC void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#ac1fd901b5ef9f39be32316a17297781e">scatter_1d_index_impl</a> (const device T *updates, device <a class="el" href="structmlx__atomic.html">mlx_atomic</a>< T > *out, const constant int *out_shape, const constant size_t *out_strides, const constant size_t &out_ndim, const constant int *upd_shape, const constant size_t &upd_ndim, const constant size_t &upd_size, const thread array< const device IdxT *, NIDX > &idx_buffers, uint2 gid)</td></tr>
|
||||
<tr class="separator:ac1fd901b5ef9f39be32316a17297781e"><td class="memSeparator" colspan="2"> </td></tr>
|
||||
<tr class="memitem:a75f61cc43682a72dd66a1da43a1809fe" id="r_a75f61cc43682a72dd66a1da43a1809fe"><td class="memTemplParams" colspan="2">template<typename T , typename IdxT , typename Op , int NIDX> </td></tr>
|
||||
<tr class="memitem:a75f61cc43682a72dd66a1da43a1809fe"><td class="memTemplItemLeft" align="right" valign="top">METAL_FUNC void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#a75f61cc43682a72dd66a1da43a1809fe">scatter_impl</a> (const device T *updates, device <a class="el" href="structmlx__atomic.html">mlx_atomic</a>< T > *out, const constant int *upd_shape, const constant size_t *upd_strides, const constant size_t &upd_ndim, const constant size_t &upd_size, const constant int *out_shape, const constant size_t *out_strides, const constant size_t &out_ndim, const constant int *axes, const thread <a class="el" href="struct_indices.html">Indices</a>< IdxT, NIDX > &indices, uint2 gid)</td></tr>
|
||||
<tr class="separator:a75f61cc43682a72dd66a1da43a1809fe"><td class="memSeparator" colspan="2"> </td></tr>
|
||||
<tr class="memitem:ad1ce39d0b6d733a95e739121fcc61bd1" id="r_ad1ce39d0b6d733a95e739121fcc61bd1"><td class="memTemplParams" colspan="2">template<typename T , typename IdxT , typename Op , int NIDX, bool UPD_ROW_CONTIG, int NWORK> </td></tr>
|
||||
<tr class="memitem:ad1ce39d0b6d733a95e739121fcc61bd1"><td class="memTemplItemLeft" align="right" valign="top">METAL_FUNC void </td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#ad1ce39d0b6d733a95e739121fcc61bd1">scatter_impl</a> (const device T *updates, device <a class="el" href="structmlx__atomic.html">mlx_atomic</a>< T > *out, const constant int *upd_shape, const constant size_t *upd_strides, const constant size_t &upd_ndim, const constant size_t &upd_size, const constant int *out_shape, const constant size_t *out_strides, const constant size_t &out_ndim, const constant int *axes, const constant size_t &idx_size, const thread <a class="el" href="struct_indices.html">Indices</a>< IdxT, NIDX > &indices, uint2 gid)</td></tr>
|
||||
<tr class="separator:ad1ce39d0b6d733a95e739121fcc61bd1"><td class="memSeparator" colspan="2"> </td></tr>
|
||||
</table>
|
||||
<h2 class="groupheader">Function Documentation</h2>
|
||||
<a id="ac1fd901b5ef9f39be32316a17297781e" name="ac1fd901b5ef9f39be32316a17297781e"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ac1fd901b5ef9f39be32316a17297781e">◆ </a></span>scatter_1d_index_impl()</h2>
|
||||
<a id="ad1ce39d0b6d733a95e739121fcc61bd1" name="ad1ce39d0b6d733a95e739121fcc61bd1"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#ad1ce39d0b6d733a95e739121fcc61bd1">◆ </a></span>scatter_impl()</h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
<div class="memtemplate">
|
||||
template<typename T , typename IdxT , typename Op , int NIDX> </div>
|
||||
<table class="memname">
|
||||
<tr>
|
||||
<td class="memname">METAL_FUNC void scatter_1d_index_impl </td>
|
||||
<td>(</td>
|
||||
<td class="paramtype">const device T *</td> <td class="paramname"><span class="paramname"><em>updates</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">device <a class="el" href="structmlx__atomic.html">mlx_atomic</a>< T > *</td> <td class="paramname"><span class="paramname"><em>out</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant int *</td> <td class="paramname"><span class="paramname"><em>out_shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant size_t *</td> <td class="paramname"><span class="paramname"><em>out_strides</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant size_t &</td> <td class="paramname"><span class="paramname"><em>out_ndim</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant int *</td> <td class="paramname"><span class="paramname"><em>upd_shape</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant size_t &</td> <td class="paramname"><span class="paramname"><em>upd_ndim</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant size_t &</td> <td class="paramname"><span class="paramname"><em>upd_size</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const thread array< const device IdxT *, NIDX > &</td> <td class="paramname"><span class="paramname"><em>idx_buffers</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">uint2</td> <td class="paramname"><span class="paramname"><em>gid</em></span> )</td>
|
||||
</tr>
|
||||
</table>
|
||||
</div><div class="memdoc">
|
||||
|
||||
</div>
|
||||
</div>
|
||||
<a id="a75f61cc43682a72dd66a1da43a1809fe" name="a75f61cc43682a72dd66a1da43a1809fe"></a>
|
||||
<h2 class="memtitle"><span class="permalink"><a href="#a75f61cc43682a72dd66a1da43a1809fe">◆ </a></span>scatter_impl()</h2>
|
||||
|
||||
<div class="memitem">
|
||||
<div class="memproto">
|
||||
<div class="memtemplate">
|
||||
template<typename T , typename IdxT , typename Op , int NIDX> </div>
|
||||
template<typename T , typename IdxT , typename Op , int NIDX, bool UPD_ROW_CONTIG, int NWORK> </div>
|
||||
<table class="memname">
|
||||
<tr>
|
||||
<td class="memname">METAL_FUNC void scatter_impl </td>
|
||||
@@ -228,6 +162,11 @@ template<typename T , typename IdxT , typename Op , int NIDX> </div>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant int *</td> <td class="paramname"><span class="paramname"><em>axes</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
<td class="paramtype">const constant size_t &</td> <td class="paramname"><span class="paramname"><em>idx_size</em></span>, </td>
|
||||
</tr>
|
||||
<tr>
|
||||
<td class="paramkey"></td>
|
||||
<td></td>
|
||||
|
||||
Reference in New Issue
Block a user