zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-21 04:31:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-10-25 20:23:45 +00:00
parent 57e53f85ee
commit 73830de153
717 changed files with 33254 additions and 34561 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docs/build/html/ops_8h_source.html vendored
View File

@@ -1192,11 +1192,11 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l01285" name="l01285"></a><span class="lineno"> 1285</span> <span class="keywordtype">int</span> groups = 1,</div>
<div class="line"><a id="l01286" name="l01286"></a><span class="lineno"> 1286</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
<div class="line"><a id="l01287" name="l01287"></a><span class="lineno"> 1287</span> </div>
<div class="line"><a id="l01289" name="l01289"></a><span class="lineno"><a class="line" href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e"> 1289</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e">quantized_matmul</a>(</div>
<div class="line"><a id="l01290" name="l01290"></a><span class="lineno"> 1290</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; x,</div>
<div class="line"><a id="l01291" name="l01291"></a><span class="lineno"> 1291</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; w,</div>
<div class="line"><a id="l01292" name="l01292"></a><span class="lineno"> 1292</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; scales,</div>
<div class="line"><a id="l01293" name="l01293"></a><span class="lineno"> 1293</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; biases,</div>
<div class="line"><a id="l01289" name="l01289"></a><span class="lineno"><a class="line" href="group__ops.html#gabfa4208fb1f9b1cdd0abc563b19175af"> 1289</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gabfa4208fb1f9b1cdd0abc563b19175af">quantized_matmul</a>(</div>
<div class="line"><a id="l01290" name="l01290"></a><span class="lineno"> 1290</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x,</div>
<div class="line"><a id="l01291" name="l01291"></a><span class="lineno"> 1291</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> w,</div>
<div class="line"><a id="l01292" name="l01292"></a><span class="lineno"> 1292</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> scales,</div>
<div class="line"><a id="l01293" name="l01293"></a><span class="lineno"> 1293</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> biases,</div>
<div class="line"><a id="l01294" name="l01294"></a><span class="lineno"> 1294</span> <span class="keywordtype">bool</span> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a> = <span class="keyword">true</span>,</div>
<div class="line"><a id="l01295" name="l01295"></a><span class="lineno"> 1295</span> <span class="keywordtype">int</span> group_size = 64,</div>
<div class="line"><a id="l01296" name="l01296"></a><span class="lineno"> 1296</span> <span class="keywordtype">int</span> bits = 4,</div>
@@ -1435,7 +1435,6 @@ $(function(){ initResizable(false); });
<div class="ttc" id="agroup__ops_html_ga577c911618575314de63d1060656a26e"><div class="ttname"><a href="group__ops.html#ga577c911618575314de63d1060656a26e">mlx::core::meshgrid</a></div><div class="ttdeci">std::vector&lt; array &gt; meshgrid(const std::vector&lt; array &gt; &amp;arrays, bool sparse=false, std::string indexing=&quot;xy&quot;, StreamOrDevice s={})</div><div class="ttdoc">A vector of coordinate arrays from coordinate vectors.</div></div>
<div class="ttc" id="agroup__ops_html_ga5b596906bf8cdc8d97ed6ddc9aeb4c23"><div class="ttname"><a href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23">mlx::core::conjugate</a></div><div class="ttdeci">array conjugate(const array &amp;a, StreamOrDevice s={})</div></div>
<div class="ttc" id="agroup__ops_html_ga5efb19aa0dfa42d8a3d5e1dfd569cd6d"><div class="ttname"><a href="group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d">mlx::core::tanh</a></div><div class="ttdeci">array tanh(const array &amp;a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Tangent of the elements of an array.</div></div>
<div class="ttc" id="agroup__ops_html_ga6381c7062b232ac21f3fdc3136aaf86e"><div class="ttname"><a href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e">mlx::core::quantized_matmul</a></div><div class="ttdeci">array quantized_matmul(const array &amp;x, const array &amp;w, const array &amp;scales, const array &amp;biases, bool transpose=true, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Quantized matmul multiplies x with a quantized matrix w.</div></div>
<div class="ttc" id="agroup__ops_html_ga654fec16a9746b390916697a2ab2546e"><div class="ttname"><a href="group__ops.html#ga654fec16a9746b390916697a2ab2546e">mlx::core::inner</a></div><div class="ttdeci">array inner(const array &amp;a, const array &amp;b, StreamOrDevice s={})</div><div class="ttdoc">Compute the inner product of two vectors.</div></div>
<div class="ttc" id="agroup__ops_html_ga6b76c8ea46b19e6866af155fa5910be6"><div class="ttname"><a href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6">mlx::core::block_masked_mm</a></div><div class="ttdeci">array block_masked_mm(array a, array b, int block_size, std::optional&lt; array &gt; mask_out=std::nullopt, std::optional&lt; array &gt; mask_lhs=std::nullopt, std::optional&lt; array &gt; mask_rhs=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix product with block masking.</div></div>
<div class="ttc" id="agroup__ops_html_ga6caba9c92b5989123501f909cc7da354"><div class="ttname"><a href="group__ops.html#ga6caba9c92b5989123501f909cc7da354">mlx::core::arctan2</a></div><div class="ttdeci">array arctan2(const array &amp;a, const array &amp;b, StreamOrDevice s={})</div><div class="ttdoc">Inverse tangent of the ratio of two arrays.</div></div>
@@ -1509,6 +1508,7 @@ $(function(){ initResizable(false); });
<div class="ttc" id="agroup__ops_html_gabca78d34ce93f0de2814e62225bb2a53"><div class="ttname"><a href="group__ops.html#gabca78d34ce93f0de2814e62225bb2a53">mlx::core::logical_not</a></div><div class="ttdeci">array logical_not(const array &amp;a, StreamOrDevice s={})</div><div class="ttdoc">Logical not of an array.</div></div>
<div class="ttc" id="agroup__ops_html_gabdc36fa65697d0361c8d67495de77129"><div class="ttname"><a href="group__ops.html#gabdc36fa65697d0361c8d67495de77129">mlx::core::concatenate</a></div><div class="ttdeci">array concatenate(const std::vector&lt; array &gt; &amp;arrays, int axis, StreamOrDevice s={})</div><div class="ttdoc">Concatenate arrays along a given axis.</div></div>
<div class="ttc" id="agroup__ops_html_gabf786129c7660ed8d5acb5499bc6fefd"><div class="ttname"><a href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd">mlx::core::trace</a></div><div class="ttdeci">array trace(const array &amp;a, int offset, int axis1, int axis2, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Return the sum along a specified diagonal in the given array.</div></div>
<div class="ttc" id="agroup__ops_html_gabfa4208fb1f9b1cdd0abc563b19175af"><div class="ttname"><a href="group__ops.html#gabfa4208fb1f9b1cdd0abc563b19175af">mlx::core::quantized_matmul</a></div><div class="ttdeci">array quantized_matmul(array x, array w, array scales, array biases, bool transpose=true, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Quantized matmul multiplies x with a quantized matrix w.</div></div>
<div class="ttc" id="agroup__ops_html_gabff758a5c1ce32ad7e8b78aba0164077"><div class="ttname"><a href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077">mlx::core::dequantize</a></div><div class="ttdeci">array dequantize(const array &amp;w, const array &amp;scales, const array &amp;biases, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Dequantize a matrix produced by quantize()</div></div>
<div class="ttc" id="agroup__ops_html_gac1869f3b7094869b44fe7ac4ce58638b"><div class="ttname"><a href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">mlx::core::transpose</a></div><div class="ttdeci">array transpose(const array &amp;a, std::vector&lt; int &gt; axes, StreamOrDevice s={})</div><div class="ttdoc">Permutes the dimensions according to the given axes.</div></div>
<div class="ttc" id="agroup__ops_html_gac1b30830a972fb9a2601379ad2b32405"><div class="ttname"><a href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405">mlx::core::partition</a></div><div class="ttdeci">array partition(const array &amp;a, int kth, StreamOrDevice s={})</div><div class="ttdoc">Returns a partitioned copy of the flattened array such that the smaller kth elements are first.</div></div>