mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-19 10:48:09 +08:00
rebase
This commit is contained in:
CircleCI Docs
774
docs/build/html/ops_8h_source.html
vendored
774
docs/build/html/ops_8h_source.html
vendored
@@ -3,7 +3,7 @@
|
||||
<head>
|
||||
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
|
||||
<meta http-equiv="X-UA-Compatible" content="IE=11"/>
|
||||
<meta name="generator" content="Doxygen 1.13.1"/>
|
||||
<meta name="generator" content="Doxygen 1.13.2"/>
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1"/>
|
||||
<title>MLX: mlx/ops.h Source File</title>
|
||||
<link href="tabs.css" rel="stylesheet" type="text/css"/>
|
||||
@@ -52,7 +52,7 @@
|
||||
</table>
|
||||
</div>
|
||||
<!-- end header part -->
|
||||
<!-- Generated by Doxygen 1.13.1 -->
|
||||
<!-- Generated by Doxygen 1.13.2 -->
|
||||
<script type="text/javascript">
|
||||
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
||||
var searchBox = new SearchBox("searchBox", "search/",'.html');
|
||||
@@ -256,26 +256,26 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="line"><a id="l00156" name="l00156"></a><span class="lineno"> 156</span></div>
|
||||
<div class="line"><a id="l00158" name="l00158"></a><span class="lineno"><a class="line" href="group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a"> 158</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#ga717f11149a8c7b4cc3e33bbcc0a97133">expand_dims</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00159" name="l00159"></a><span class="lineno"> 159</span></div>
|
||||
<div class="line"><a id="l00161" name="l00161"></a><span class="lineno"><a class="line" href="group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d"> 161</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#ga29718cd5005dbcde0396b6fd65cc041d">slice</a>(</div>
|
||||
<div class="line"><a id="l00161" name="l00161"></a><span class="lineno"><a class="line" href="group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d"> 161</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">slice</a>(</div>
|
||||
<div class="line"><a id="l00162" name="l00162"></a><span class="lineno"> 162</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00163" name="l00163"></a><span class="lineno"> 163</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> start,</div>
|
||||
<div class="line"><a id="l00164" name="l00164"></a><span class="lineno"> 164</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> stop,</div>
|
||||
<div class="line"><a id="l00165" name="l00165"></a><span class="lineno"> 165</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> strides,</div>
|
||||
<div class="line"><a id="l00166" name="l00166"></a><span class="lineno"> 166</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen00167" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l00167" name="l00167"></a><span class="lineno"><a class="line" href="group__ops.html#gaedcbdf4040f5e6a02a74b1ed1c6c2ebc"> 167</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d">slice</a>(</div>
|
||||
<div class="line"><a id="l00167" name="l00167"></a><span class="lineno"><a class="line" href="group__ops.html#gaedcbdf4040f5e6a02a74b1ed1c6c2ebc"> 167</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">slice</a>(</div>
|
||||
<div class="line"><a id="l00168" name="l00168"></a><span class="lineno"> 168</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00169" name="l00169"></a><span class="lineno"> 169</span> std::initializer_list<int> start,</div>
|
||||
<div class="line"><a id="l00170" name="l00170"></a><span class="lineno"> 170</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> stop,</div>
|
||||
<div class="line"><a id="l00171" name="l00171"></a><span class="lineno"> 171</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> strides,</div>
|
||||
<div class="line"><a id="l00172" name="l00172"></a><span class="lineno"> 172</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno"> 173</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d">slice</a>(a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>(start), std::move(stop), std::move(strides), s);</div>
|
||||
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno"> 173</span> <span class="keywordflow">return</span> <a class="code hl_function" href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">slice</a>(a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>(start), std::move(stop), std::move(strides), s);</div>
|
||||
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno"> 174</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l00175" name="l00175"></a><span class="lineno"> 175</span></div>
|
||||
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno"><a class="line" href="group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d"> 177</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#ga29718cd5005dbcde0396b6fd65cc041d">slice</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> start, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno"><a class="line" href="group__ops.html#gaec56dcb94d5e7f7b885fb60b4bf4aa9d"> 177</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">slice</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> start, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00178" name="l00178"></a><span class="lineno"> 178</span></div>
|
||||
<div class="line"><a id="l00180" name="l00180"></a><span class="lineno"><a class="line" href="group__ops.html#ga797996e53ea34317a55dc2f314736b89"> 180</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#ga29718cd5005dbcde0396b6fd65cc041d">slice</a>(</div>
|
||||
<div class="line"><a id="l00180" name="l00180"></a><span class="lineno"><a class="line" href="group__ops.html#ga797996e53ea34317a55dc2f314736b89"> 180</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">slice</a>(</div>
|
||||
<div class="line"><a id="l00181" name="l00181"></a><span class="lineno"> 181</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00182" name="l00182"></a><span class="lineno"> 182</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& start,</div>
|
||||
<div class="line"><a id="l00183" name="l00183"></a><span class="lineno"> 183</span> std::vector<int> axes,</div>
|
||||
@@ -312,10 +312,10 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="line"><a id="l00218" name="l00218"></a><span class="lineno"> 218</span>std::vector<array></div>
|
||||
<div class="line"><a id="l00219" name="l00219"></a><span class="lineno"><a class="line" href="group__ops.html#ga9ea089f42b9940510619052b7166d9ac"> 219</a></span><a class="code hl_function" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">split</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& indices, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00220" name="l00220"></a><span class="lineno"> 220</span></div>
|
||||
<div class="line"><a id="l00222" name="l00222"></a><span class="lineno"><a class="line" href="group__ops.html#ga577c911618575314de63d1060656a26e"> 222</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga577c911618575314de63d1060656a26e">meshgrid</a>(</div>
|
||||
<div class="line"><a id="l00222" name="l00222"></a><span class="lineno"><a class="line" href="group__ops.html#ga5ecddb74ba7861eb82eca8653501d5dc"> 222</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga5ecddb74ba7861eb82eca8653501d5dc">meshgrid</a>(</div>
|
||||
<div class="line"><a id="l00223" name="l00223"></a><span class="lineno"> 223</span> <span class="keyword">const</span> std::vector<array>& arrays,</div>
|
||||
<div class="line"><a id="l00224" name="l00224"></a><span class="lineno"> 224</span> <span class="keywordtype">bool</span> sparse = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l00225" name="l00225"></a><span class="lineno"> 225</span> std::string indexing = <span class="stringliteral">"xy"</span>,</div>
|
||||
<div class="line"><a id="l00225" name="l00225"></a><span class="lineno"> 225</span> <span class="keyword">const</span> std::string& indexing = <span class="stringliteral">"xy"</span>,</div>
|
||||
<div class="line"><a id="l00226" name="l00226"></a><span class="lineno"> 226</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00227" name="l00227"></a><span class="lineno"> 227</span></div>
|
||||
<div class="line"><a id="l00231" name="l00231"></a><span class="lineno"><a class="line" href="group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8"> 231</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#ga157cd7c23f9b306fee2e1eb2b9bf1dd8">clip</a>(</div>
|
||||
@@ -353,32 +353,32 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="line"><a id="l00267" name="l00267"></a><span class="lineno"> 267</span> <span class="keywordtype">int</span> destination,</div>
|
||||
<div class="line"><a id="l00268" name="l00268"></a><span class="lineno"> 268</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00269" name="l00269"></a><span class="lineno"> 269</span></div>
|
||||
<div class="line"><a id="l00271" name="l00271"></a><span class="lineno"><a class="line" href="group__ops.html#ga58e03f39aa5d63aba1646f530706a920"> 271</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#ga58e03f39aa5d63aba1646f530706a920">pad</a>(</div>
|
||||
<div class="line"><a id="l00271" name="l00271"></a><span class="lineno"><a class="line" href="group__ops.html#gab95ebd20bd7c6d1c840007cc020cbc0c"> 271</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#gab95ebd20bd7c6d1c840007cc020cbc0c">pad</a>(</div>
|
||||
<div class="line"><a id="l00272" name="l00272"></a><span class="lineno"> 272</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00273" name="l00273"></a><span class="lineno"> 273</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l00274" name="l00274"></a><span class="lineno"> 274</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& low_pad_size,</div>
|
||||
<div class="line"><a id="l00275" name="l00275"></a><span class="lineno"> 275</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& high_pad_size,</div>
|
||||
<div class="line"><a id="l00276" name="l00276"></a><span class="lineno"> 276</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
||||
<div class="line"><a id="l00277" name="l00277"></a><span class="lineno"> 277</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00277" name="l00277"></a><span class="lineno"> 277</span> <span class="keyword">const</span> std::string& mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00278" name="l00278"></a><span class="lineno"> 278</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00279" name="l00279"></a><span class="lineno"> 279</span></div>
|
||||
<div class="line"><a id="l00281" name="l00281"></a><span class="lineno"><a class="line" href="group__ops.html#ga8495c9902b0a67f288918cb528d5abeb"> 281</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#ga58e03f39aa5d63aba1646f530706a920">pad</a>(</div>
|
||||
<div class="line"><a id="l00281" name="l00281"></a><span class="lineno"><a class="line" href="group__ops.html#gad89f464e92c356faab1f1c2e763b1fb7"> 281</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#gab95ebd20bd7c6d1c840007cc020cbc0c">pad</a>(</div>
|
||||
<div class="line"><a id="l00282" name="l00282"></a><span class="lineno"> 282</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00283" name="l00283"></a><span class="lineno"> 283</span> <span class="keyword">const</span> std::vector<std::pair<int, int>>& pad_width,</div>
|
||||
<div class="line"><a id="l00284" name="l00284"></a><span class="lineno"> 284</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
||||
<div class="line"><a id="l00285" name="l00285"></a><span class="lineno"> 285</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00285" name="l00285"></a><span class="lineno"> 285</span> <span class="keyword">const</span> std::string& mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00286" name="l00286"></a><span class="lineno"> 286</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00287" name="l00287"></a><span class="lineno"><a class="line" href="group__ops.html#gaeaae2f9637b0b3e2b4c6c0517e285935"> 287</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#ga58e03f39aa5d63aba1646f530706a920">pad</a>(</div>
|
||||
<div class="line"><a id="l00287" name="l00287"></a><span class="lineno"><a class="line" href="group__ops.html#gab2d5a17d3d5225fed34905e786c31c5f"> 287</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#gab95ebd20bd7c6d1c840007cc020cbc0c">pad</a>(</div>
|
||||
<div class="line"><a id="l00288" name="l00288"></a><span class="lineno"> 288</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00289" name="l00289"></a><span class="lineno"> 289</span> <span class="keyword">const</span> std::pair<int, int>& pad_width,</div>
|
||||
<div class="line"><a id="l00290" name="l00290"></a><span class="lineno"> 290</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
||||
<div class="line"><a id="l00291" name="l00291"></a><span class="lineno"> 291</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00291" name="l00291"></a><span class="lineno"> 291</span> <span class="keyword">const</span> std::string& mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00292" name="l00292"></a><span class="lineno"> 292</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00293" name="l00293"></a><span class="lineno"><a class="line" href="group__ops.html#ga5b8684361f57cbfd5e6eedf82a62ded0"> 293</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#ga58e03f39aa5d63aba1646f530706a920">pad</a>(</div>
|
||||
<div class="line"><a id="l00293" name="l00293"></a><span class="lineno"><a class="line" href="group__ops.html#gaa73ac5674467b1d5c74de6fef7204c44"> 293</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#gab95ebd20bd7c6d1c840007cc020cbc0c">pad</a>(</div>
|
||||
<div class="line"><a id="l00294" name="l00294"></a><span class="lineno"> 294</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00295" name="l00295"></a><span class="lineno"> 295</span> <span class="keywordtype">int</span> pad_width,</div>
|
||||
<div class="line"><a id="l00296" name="l00296"></a><span class="lineno"> 296</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& pad_value = <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(0),</div>
|
||||
<div class="line"><a id="l00297" name="l00297"></a><span class="lineno"> 297</span> <span class="keyword">const</span> std::string mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00297" name="l00297"></a><span class="lineno"> 297</span> <span class="keyword">const</span> std::string& mode = <span class="stringliteral">"constant"</span>,</div>
|
||||
<div class="line"><a id="l00298" name="l00298"></a><span class="lineno"> 298</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00299" name="l00299"></a><span class="lineno"> 299</span></div>
|
||||
<div class="line"><a id="l00301" name="l00301"></a><span class="lineno"><a class="line" href="group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a"> 301</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#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
@@ -950,7 +950,7 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="line"><a id="l00920" name="l00920"></a><span class="lineno"> 920</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l00921" name="l00921"></a><span class="lineno"> 921</span></div>
|
||||
<div class="line"><a id="l00923" name="l00923"></a><span class="lineno"><a class="line" href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a"> 923</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#ga753d59f5a9f5f2362865ee83b4dced2a">matmul</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00923" name="l00923"></a><span class="lineno"><a class="line" href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a"> 923</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="namespacemlx_1_1core.html#aaacf0afe13d77a5c49ce96f1e833eb2d">matmul</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00924" name="l00924"></a><span class="lineno"> 924</span></div>
|
||||
<div class="line"><a id="l00926" name="l00926"></a><span class="lineno"><a class="line" href="group__ops.html#ga8fcc3ad0677c834c36b72d5b2ebba6d0"> 926</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#ga8fcc3ad0677c834c36b72d5b2ebba6d0">gather</a>(</div>
|
||||
<div class="line"><a id="l00927" name="l00927"></a><span class="lineno"> 927</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
@@ -994,404 +994,411 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="line"><a id="l00968" name="l00968"></a><span class="lineno"> 968</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l00969" name="l00969"></a><span class="lineno"> 969</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00970" name="l00970"></a><span class="lineno"> 970</span></div>
|
||||
<div class="line"><a id="l01069" name="l01069"></a><span class="lineno"><a class="line" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61"> 1069</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#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(</div>
|
||||
<div class="line"><a id="l01070" name="l01070"></a><span class="lineno"> 1070</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01071" name="l01071"></a><span class="lineno"> 1071</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01072" name="l01072"></a><span class="lineno"> 1072</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01073" name="l01073"></a><span class="lineno"> 1073</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01074" name="l01074"></a><span class="lineno"> 1074</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01075" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01075" name="l01075"></a><span class="lineno"><a class="line" href="group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe"> 1075</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(</div>
|
||||
<div class="line"><a id="l01076" name="l01076"></a><span class="lineno"> 1076</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01077" name="l01077"></a><span class="lineno"> 1077</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01078" name="l01078"></a><span class="lineno"> 1078</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01079" name="l01079"></a><span class="lineno"> 1079</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01080" name="l01080"></a><span class="lineno"> 1080</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01081" name="l01081"></a><span class="lineno"> 1081</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01082" name="l01082"></a><span class="lineno"> 1082</span>}</div>
|
||||
<div class="line"><a id="l00972" name="l00972"></a><span class="lineno"><a class="line" href="group__ops.html#gab3fd98c0d06b84b836f93bddbd7a2a0d"> 972</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#gab3fd98c0d06b84b836f93bddbd7a2a0d">scatter_add_axis</a>(</div>
|
||||
<div class="line"><a id="l00973" name="l00973"></a><span class="lineno"> 973</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l00974" name="l00974"></a><span class="lineno"> 974</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l00975" name="l00975"></a><span class="lineno"> 975</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& values,</div>
|
||||
<div class="line"><a id="l00976" name="l00976"></a><span class="lineno"> 976</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l00977" name="l00977"></a><span class="lineno"> 977</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l00978" name="l00978"></a><span class="lineno"> 978</span></div>
|
||||
<div class="line"><a id="l01077" name="l01077"></a><span class="lineno"><a class="line" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61"> 1077</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#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(</div>
|
||||
<div class="line"><a id="l01078" name="l01078"></a><span class="lineno"> 1078</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01079" name="l01079"></a><span class="lineno"> 1079</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01080" name="l01080"></a><span class="lineno"> 1080</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01081" name="l01081"></a><span class="lineno"> 1081</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01082" name="l01082"></a><span class="lineno"> 1082</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01083" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01083" name="l01083"></a><span class="lineno"><a class="line" href="group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe"> 1083</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(</div>
|
||||
<div class="line"><a id="l01084" name="l01084"></a><span class="lineno"> 1084</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01085" name="l01085"></a><span class="lineno"> 1085</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01086" name="l01086"></a><span class="lineno"> 1086</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01087" name="l01087"></a><span class="lineno"> 1087</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01088" name="l01088"></a><span class="lineno"> 1088</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01089" name="l01089"></a><span class="lineno"> 1089</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">scatter</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01090" name="l01090"></a><span class="lineno"> 1090</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01083" name="l01083"></a><span class="lineno"> 1083</span></div>
|
||||
<div class="line"><a id="l01085" name="l01085"></a><span class="lineno"><a class="line" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174"> 1085</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#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(</div>
|
||||
<div class="line"><a id="l01086" name="l01086"></a><span class="lineno"> 1086</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01087" name="l01087"></a><span class="lineno"> 1087</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01088" name="l01088"></a><span class="lineno"> 1088</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01089" name="l01089"></a><span class="lineno"> 1089</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01090" name="l01090"></a><span class="lineno"> 1090</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01091" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01091" name="l01091"></a><span class="lineno"><a class="line" href="group__ops.html#gac13318518e5703f1273c5366eb523a5a"> 1091</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(</div>
|
||||
<div class="line"><a id="l01092" name="l01092"></a><span class="lineno"> 1092</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01093" name="l01093"></a><span class="lineno"> 1093</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01094" name="l01094"></a><span class="lineno"> 1094</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01095" name="l01095"></a><span class="lineno"> 1095</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01096" name="l01096"></a><span class="lineno"> 1096</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01097" name="l01097"></a><span class="lineno"> 1097</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01098" name="l01098"></a><span class="lineno"> 1098</span>}</div>
|
||||
<div class="line"><a id="l01091" name="l01091"></a><span class="lineno"> 1091</span></div>
|
||||
<div class="line"><a id="l01093" name="l01093"></a><span class="lineno"><a class="line" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174"> 1093</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#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(</div>
|
||||
<div class="line"><a id="l01094" name="l01094"></a><span class="lineno"> 1094</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01095" name="l01095"></a><span class="lineno"> 1095</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01096" name="l01096"></a><span class="lineno"> 1096</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01097" name="l01097"></a><span class="lineno"> 1097</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01098" name="l01098"></a><span class="lineno"> 1098</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01099" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01099" name="l01099"></a><span class="lineno"><a class="line" href="group__ops.html#gac13318518e5703f1273c5366eb523a5a"> 1099</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(</div>
|
||||
<div class="line"><a id="l01100" name="l01100"></a><span class="lineno"> 1100</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01101" name="l01101"></a><span class="lineno"> 1101</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01102" name="l01102"></a><span class="lineno"> 1102</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01103" name="l01103"></a><span class="lineno"> 1103</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01104" name="l01104"></a><span class="lineno"> 1104</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01105" name="l01105"></a><span class="lineno"> 1105</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">scatter_add</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01106" name="l01106"></a><span class="lineno"> 1106</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01099" name="l01099"></a><span class="lineno"> 1099</span></div>
|
||||
<div class="line"><a id="l01101" name="l01101"></a><span class="lineno"><a class="line" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0"> 1101</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#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(</div>
|
||||
<div class="line"><a id="l01102" name="l01102"></a><span class="lineno"> 1102</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01103" name="l01103"></a><span class="lineno"> 1103</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01104" name="l01104"></a><span class="lineno"> 1104</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01105" name="l01105"></a><span class="lineno"> 1105</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01106" name="l01106"></a><span class="lineno"> 1106</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01107" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01107" name="l01107"></a><span class="lineno"><a class="line" href="group__ops.html#gaf83c53c453faa9083ba27e4b97539339"> 1107</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(</div>
|
||||
<div class="line"><a id="l01108" name="l01108"></a><span class="lineno"> 1108</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01109" name="l01109"></a><span class="lineno"> 1109</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01110" name="l01110"></a><span class="lineno"> 1110</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01111" name="l01111"></a><span class="lineno"> 1111</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01112" name="l01112"></a><span class="lineno"> 1112</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01113" name="l01113"></a><span class="lineno"> 1113</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01114" name="l01114"></a><span class="lineno"> 1114</span>}</div>
|
||||
<div class="line"><a id="l01107" name="l01107"></a><span class="lineno"> 1107</span></div>
|
||||
<div class="line"><a id="l01109" name="l01109"></a><span class="lineno"><a class="line" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0"> 1109</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#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(</div>
|
||||
<div class="line"><a id="l01110" name="l01110"></a><span class="lineno"> 1110</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01111" name="l01111"></a><span class="lineno"> 1111</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01112" name="l01112"></a><span class="lineno"> 1112</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01113" name="l01113"></a><span class="lineno"> 1113</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01114" name="l01114"></a><span class="lineno"> 1114</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01115" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01115" name="l01115"></a><span class="lineno"><a class="line" href="group__ops.html#gaf83c53c453faa9083ba27e4b97539339"> 1115</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(</div>
|
||||
<div class="line"><a id="l01116" name="l01116"></a><span class="lineno"> 1116</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01117" name="l01117"></a><span class="lineno"> 1117</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01118" name="l01118"></a><span class="lineno"> 1118</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01119" name="l01119"></a><span class="lineno"> 1119</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01120" name="l01120"></a><span class="lineno"> 1120</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01121" name="l01121"></a><span class="lineno"> 1121</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">scatter_prod</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01122" name="l01122"></a><span class="lineno"> 1122</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01115" name="l01115"></a><span class="lineno"> 1115</span></div>
|
||||
<div class="line"><a id="l01117" name="l01117"></a><span class="lineno"><a class="line" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673"> 1117</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#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(</div>
|
||||
<div class="line"><a id="l01118" name="l01118"></a><span class="lineno"> 1118</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01119" name="l01119"></a><span class="lineno"> 1119</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01120" name="l01120"></a><span class="lineno"> 1120</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01121" name="l01121"></a><span class="lineno"> 1121</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01122" name="l01122"></a><span class="lineno"> 1122</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01123" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01123" name="l01123"></a><span class="lineno"><a class="line" href="group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f"> 1123</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(</div>
|
||||
<div class="line"><a id="l01124" name="l01124"></a><span class="lineno"> 1124</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01125" name="l01125"></a><span class="lineno"> 1125</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01126" name="l01126"></a><span class="lineno"> 1126</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01127" name="l01127"></a><span class="lineno"> 1127</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01128" name="l01128"></a><span class="lineno"> 1128</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01129" name="l01129"></a><span class="lineno"> 1129</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01130" name="l01130"></a><span class="lineno"> 1130</span>}</div>
|
||||
<div class="line"><a id="l01123" name="l01123"></a><span class="lineno"> 1123</span></div>
|
||||
<div class="line"><a id="l01125" name="l01125"></a><span class="lineno"><a class="line" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673"> 1125</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#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(</div>
|
||||
<div class="line"><a id="l01126" name="l01126"></a><span class="lineno"> 1126</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01127" name="l01127"></a><span class="lineno"> 1127</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01128" name="l01128"></a><span class="lineno"> 1128</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01129" name="l01129"></a><span class="lineno"> 1129</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01130" name="l01130"></a><span class="lineno"> 1130</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01131" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01131" name="l01131"></a><span class="lineno"><a class="line" href="group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f"> 1131</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(</div>
|
||||
<div class="line"><a id="l01132" name="l01132"></a><span class="lineno"> 1132</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01133" name="l01133"></a><span class="lineno"> 1133</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01134" name="l01134"></a><span class="lineno"> 1134</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01135" name="l01135"></a><span class="lineno"> 1135</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01136" name="l01136"></a><span class="lineno"> 1136</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01137" name="l01137"></a><span class="lineno"> 1137</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">scatter_max</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01138" name="l01138"></a><span class="lineno"> 1138</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01131" name="l01131"></a><span class="lineno"> 1131</span></div>
|
||||
<div class="line"><a id="l01132" name="l01132"></a><span class="lineno"><a class="line" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5"> 1132</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#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(</div>
|
||||
<div class="line"><a id="l01133" name="l01133"></a><span class="lineno"> 1133</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01134" name="l01134"></a><span class="lineno"> 1134</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01135" name="l01135"></a><span class="lineno"> 1135</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01136" name="l01136"></a><span class="lineno"> 1136</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01137" name="l01137"></a><span class="lineno"> 1137</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01138" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01138" name="l01138"></a><span class="lineno"><a class="line" href="group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199"> 1138</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(</div>
|
||||
<div class="line"><a id="l01139" name="l01139"></a><span class="lineno"> 1139</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01140" name="l01140"></a><span class="lineno"> 1140</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01141" name="l01141"></a><span class="lineno"> 1141</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01142" name="l01142"></a><span class="lineno"> 1142</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01143" name="l01143"></a><span class="lineno"> 1143</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01144" name="l01144"></a><span class="lineno"> 1144</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01145" name="l01145"></a><span class="lineno"> 1145</span>}</div>
|
||||
<div class="line"><a id="l01139" name="l01139"></a><span class="lineno"> 1139</span></div>
|
||||
<div class="line"><a id="l01140" name="l01140"></a><span class="lineno"><a class="line" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5"> 1140</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#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(</div>
|
||||
<div class="line"><a id="l01141" name="l01141"></a><span class="lineno"> 1141</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01142" name="l01142"></a><span class="lineno"> 1142</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
||||
<div class="line"><a id="l01143" name="l01143"></a><span class="lineno"> 1143</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01144" name="l01144"></a><span class="lineno"> 1144</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01145" name="l01145"></a><span class="lineno"> 1145</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="foldopen" id="foldopen01146" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01146" name="l01146"></a><span class="lineno"><a class="line" href="group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199"> 1146</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(</div>
|
||||
<div class="line"><a id="l01147" name="l01147"></a><span class="lineno"> 1147</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01148" name="l01148"></a><span class="lineno"> 1148</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& indices,</div>
|
||||
<div class="line"><a id="l01149" name="l01149"></a><span class="lineno"> 1149</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& updates,</div>
|
||||
<div class="line"><a id="l01150" name="l01150"></a><span class="lineno"> 1150</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01151" name="l01151"></a><span class="lineno"> 1151</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01152" name="l01152"></a><span class="lineno"> 1152</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">scatter_min</a>(a, {indices}, updates, std::vector<int>{axis}, s);</div>
|
||||
<div class="line"><a id="l01153" name="l01153"></a><span class="lineno"> 1153</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01146" name="l01146"></a><span class="lineno"> 1146</span></div>
|
||||
<div class="line"><a id="l01148" name="l01148"></a><span class="lineno"><a class="line" href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1"> 1148</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#ga297f853b3d90ec8ae81263977ba2ddb1">sqrt</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01149" name="l01149"></a><span class="lineno"> 1149</span></div>
|
||||
<div class="line"><a id="l01151" name="l01151"></a><span class="lineno"><a class="line" href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1"> 1151</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#ga102f23aa0b0c3d3296a321c694617aa1">rsqrt</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01152" name="l01152"></a><span class="lineno"> 1152</span></div>
|
||||
<div class="line"><a id="l01154" name="l01154"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"> 1154</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#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(</div>
|
||||
<div class="line"><a id="l01155" name="l01155"></a><span class="lineno"> 1155</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01156" name="l01156"></a><span class="lineno"> 1156</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01157" name="l01157"></a><span class="lineno"> 1157</span> <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01158" name="l01158"></a><span class="lineno"> 1158</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01159" name="l01159"></a><span class="lineno"> 1159</span></div>
|
||||
<div class="line"><a id="l01161" name="l01161"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ae3614d07d873892a530d14c3857d0b"> 1161</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#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01162" name="l01162"></a><span class="lineno"> 1162</span></div>
|
||||
<div class="line"><a id="l01164" name="l01164"></a><span class="lineno"> 1164</span><span class="keyword">inline</span> array</div>
|
||||
<div class="foldopen" id="foldopen01165" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01165" name="l01165"></a><span class="lineno"><a class="line" href="group__ops.html#ga06f570d73716a24303e6de3aaba4457b"> 1165</a></span><a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01166" name="l01166"></a><span class="lineno"> 1166</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(a, std::vector<int>{axis}, precise, s);</div>
|
||||
<div class="line"><a id="l01167" name="l01167"></a><span class="lineno"> 1167</span>}</div>
|
||||
<div class="line"><a id="l01154" name="l01154"></a><span class="lineno"> 1154</span></div>
|
||||
<div class="line"><a id="l01156" name="l01156"></a><span class="lineno"><a class="line" href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1"> 1156</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#ga297f853b3d90ec8ae81263977ba2ddb1">sqrt</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01157" name="l01157"></a><span class="lineno"> 1157</span></div>
|
||||
<div class="line"><a id="l01159" name="l01159"></a><span class="lineno"><a class="line" href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1"> 1159</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#ga102f23aa0b0c3d3296a321c694617aa1">rsqrt</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01160" name="l01160"></a><span class="lineno"> 1160</span></div>
|
||||
<div class="line"><a id="l01162" name="l01162"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"> 1162</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#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(</div>
|
||||
<div class="line"><a id="l01163" name="l01163"></a><span class="lineno"> 1163</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01164" name="l01164"></a><span class="lineno"> 1164</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01165" name="l01165"></a><span class="lineno"> 1165</span> <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01166" name="l01166"></a><span class="lineno"> 1166</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01167" name="l01167"></a><span class="lineno"> 1167</span></div>
|
||||
<div class="line"><a id="l01169" name="l01169"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ae3614d07d873892a530d14c3857d0b"> 1169</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#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01170" name="l01170"></a><span class="lineno"> 1170</span></div>
|
||||
<div class="line"><a id="l01172" name="l01172"></a><span class="lineno"> 1172</span><span class="keyword">inline</span> array</div>
|
||||
<div class="foldopen" id="foldopen01173" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01173" name="l01173"></a><span class="lineno"><a class="line" href="group__ops.html#ga06f570d73716a24303e6de3aaba4457b"> 1173</a></span><a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> axis, <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01174" name="l01174"></a><span class="lineno"> 1174</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">softmax</a>(a, std::vector<int>{axis}, precise, s);</div>
|
||||
<div class="line"><a id="l01175" name="l01175"></a><span class="lineno"> 1175</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01168" name="l01168"></a><span class="lineno"> 1168</span></div>
|
||||
<div class="line"><a id="l01170" name="l01170"></a><span class="lineno"><a class="line" href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d"> 1170</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#ga7972058715c26559dff9c9ae2a3ef76d">power</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01171" name="l01171"></a><span class="lineno"> 1171</span></div>
|
||||
<div class="line"><a id="l01173" name="l01173"></a><span class="lineno"><a class="line" href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420"> 1173</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#gaddc825a5c173e195ab0fda83ad630420">cumsum</a>(</div>
|
||||
<div class="line"><a id="l01174" name="l01174"></a><span class="lineno"> 1174</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01175" name="l01175"></a><span class="lineno"> 1175</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01176" name="l01176"></a><span class="lineno"> 1176</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01177" name="l01177"></a><span class="lineno"> 1177</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
||||
<div class="line"><a id="l01178" name="l01178"></a><span class="lineno"> 1178</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01176" name="l01176"></a><span class="lineno"> 1176</span></div>
|
||||
<div class="line"><a id="l01178" name="l01178"></a><span class="lineno"><a class="line" href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d"> 1178</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#ga7972058715c26559dff9c9ae2a3ef76d">power</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01179" name="l01179"></a><span class="lineno"> 1179</span></div>
|
||||
<div class="line"><a id="l01181" name="l01181"></a><span class="lineno"><a class="line" href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f"> 1181</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#ga0d71dfbc14ef3ed564b0c5ee26af680f">cumprod</a>(</div>
|
||||
<div class="line"><a id="l01181" name="l01181"></a><span class="lineno"><a class="line" href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420"> 1181</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#gaddc825a5c173e195ab0fda83ad630420">cumsum</a>(</div>
|
||||
<div class="line"><a id="l01182" name="l01182"></a><span class="lineno"> 1182</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01183" name="l01183"></a><span class="lineno"> 1183</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01184" name="l01184"></a><span class="lineno"> 1184</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01185" name="l01185"></a><span class="lineno"> 1185</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
||||
<div class="line"><a id="l01186" name="l01186"></a><span class="lineno"> 1186</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01187" name="l01187"></a><span class="lineno"> 1187</span></div>
|
||||
<div class="line"><a id="l01189" name="l01189"></a><span class="lineno"><a class="line" href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143"> 1189</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#gaee37cac8476e8f8d666bcded5bc59143">cummax</a>(</div>
|
||||
<div class="line"><a id="l01189" name="l01189"></a><span class="lineno"><a class="line" href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f"> 1189</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#ga0d71dfbc14ef3ed564b0c5ee26af680f">cumprod</a>(</div>
|
||||
<div class="line"><a id="l01190" name="l01190"></a><span class="lineno"> 1190</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01191" name="l01191"></a><span class="lineno"> 1191</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01192" name="l01192"></a><span class="lineno"> 1192</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01193" name="l01193"></a><span class="lineno"> 1193</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
||||
<div class="line"><a id="l01194" name="l01194"></a><span class="lineno"> 1194</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01195" name="l01195"></a><span class="lineno"> 1195</span></div>
|
||||
<div class="line"><a id="l01197" name="l01197"></a><span class="lineno"><a class="line" href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8"> 1197</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#ga19c1bf6929fe8d66b9cd408946aea6a8">cummin</a>(</div>
|
||||
<div class="line"><a id="l01197" name="l01197"></a><span class="lineno"><a class="line" href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143"> 1197</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#gaee37cac8476e8f8d666bcded5bc59143">cummax</a>(</div>
|
||||
<div class="line"><a id="l01198" name="l01198"></a><span class="lineno"> 1198</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01199" name="l01199"></a><span class="lineno"> 1199</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01200" name="l01200"></a><span class="lineno"> 1200</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01201" name="l01201"></a><span class="lineno"> 1201</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
||||
<div class="line"><a id="l01202" name="l01202"></a><span class="lineno"> 1202</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01203" name="l01203"></a><span class="lineno"> 1203</span></div>
|
||||
<div class="line"><a id="l01205" name="l01205"></a><span class="lineno"><a class="line" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e"> 1205</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#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
||||
<div class="line"><a id="l01206" name="l01206"></a><span class="lineno"> 1206</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> input,</div>
|
||||
<div class="line"><a id="l01207" name="l01207"></a><span class="lineno"> 1207</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> weight,</div>
|
||||
<div class="line"><a id="l01208" name="l01208"></a><span class="lineno"> 1208</span> std::vector<int> stride = {},</div>
|
||||
<div class="line"><a id="l01209" name="l01209"></a><span class="lineno"> 1209</span> std::vector<int> padding_lo = {},</div>
|
||||
<div class="line"><a id="l01210" name="l01210"></a><span class="lineno"> 1210</span> std::vector<int> padding_hi = {},</div>
|
||||
<div class="line"><a id="l01211" name="l01211"></a><span class="lineno"> 1211</span> std::vector<int> kernel_dilation = {},</div>
|
||||
<div class="line"><a id="l01212" name="l01212"></a><span class="lineno"> 1212</span> std::vector<int> input_dilation = {},</div>
|
||||
<div class="line"><a id="l01213" name="l01213"></a><span class="lineno"> 1213</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01214" name="l01214"></a><span class="lineno"> 1214</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01215" name="l01215"></a><span class="lineno"> 1215</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01216" name="l01216"></a><span class="lineno"> 1216</span></div>
|
||||
<div class="foldopen" id="foldopen01218" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01218" name="l01218"></a><span class="lineno"><a class="line" href="group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d"> 1218</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
||||
<div class="line"><a id="l01219" name="l01219"></a><span class="lineno"> 1219</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01220" name="l01220"></a><span class="lineno"> 1220</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01221" name="l01221"></a><span class="lineno"> 1221</span> std::vector<int> stride = {},</div>
|
||||
<div class="line"><a id="l01222" name="l01222"></a><span class="lineno"> 1222</span> std::vector<int> padding = {},</div>
|
||||
<div class="line"><a id="l01223" name="l01223"></a><span class="lineno"> 1223</span> std::vector<int> kernel_dilation = {},</div>
|
||||
<div class="line"><a id="l01224" name="l01224"></a><span class="lineno"> 1224</span> std::vector<int> input_dilation = {},</div>
|
||||
<div class="line"><a id="l01225" name="l01225"></a><span class="lineno"> 1225</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01226" name="l01226"></a><span class="lineno"> 1226</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01227" name="l01227"></a><span class="lineno"> 1227</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01228" name="l01228"></a><span class="lineno"> 1228</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
||||
<div class="line"><a id="l01229" name="l01229"></a><span class="lineno"> 1229</span> <span class="comment">/* const array& input = */</span> input,</div>
|
||||
<div class="line"><a id="l01230" name="l01230"></a><span class="lineno"> 1230</span> <span class="comment">/* const array& weight = */</span> weight,</div>
|
||||
<div class="line"><a id="l01231" name="l01231"></a><span class="lineno"> 1231</span> <span class="comment">/* std::vector<int> stride = */</span> stride,</div>
|
||||
<div class="line"><a id="l01232" name="l01232"></a><span class="lineno"> 1232</span> <span class="comment">/* std::vector<int> padding_lo = */</span> padding,</div>
|
||||
<div class="line"><a id="l01233" name="l01233"></a><span class="lineno"> 1233</span> <span class="comment">/* std::vector<int> padding_hi = */</span> padding,</div>
|
||||
<div class="line"><a id="l01234" name="l01234"></a><span class="lineno"> 1234</span> <span class="comment">/* std::vector<int> kernel_dilation = */</span> kernel_dilation,</div>
|
||||
<div class="line"><a id="l01235" name="l01235"></a><span class="lineno"> 1235</span> <span class="comment">/* std::vector<int> input_dilation = */</span> input_dilation,</div>
|
||||
<div class="line"><a id="l01236" name="l01236"></a><span class="lineno"> 1236</span> <span class="comment">/* int groups = */</span> groups,</div>
|
||||
<div class="line"><a id="l01237" name="l01237"></a><span class="lineno"> 1237</span> <span class="comment">/* bool flip = */</span> flip,</div>
|
||||
<div class="line"><a id="l01238" name="l01238"></a><span class="lineno"> 1238</span> <span class="comment">/* StreamOrDevice s = */</span> s);</div>
|
||||
<div class="line"><a id="l01239" name="l01239"></a><span class="lineno"> 1239</span>}</div>
|
||||
<div class="line"><a id="l01205" name="l01205"></a><span class="lineno"><a class="line" href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8"> 1205</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#ga19c1bf6929fe8d66b9cd408946aea6a8">cummin</a>(</div>
|
||||
<div class="line"><a id="l01206" name="l01206"></a><span class="lineno"> 1206</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01207" name="l01207"></a><span class="lineno"> 1207</span> <span class="keywordtype">int</span> axis,</div>
|
||||
<div class="line"><a id="l01208" name="l01208"></a><span class="lineno"> 1208</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01209" name="l01209"></a><span class="lineno"> 1209</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
||||
<div class="line"><a id="l01210" name="l01210"></a><span class="lineno"> 1210</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01211" name="l01211"></a><span class="lineno"> 1211</span></div>
|
||||
<div class="line"><a id="l01213" name="l01213"></a><span class="lineno"><a class="line" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e"> 1213</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#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
||||
<div class="line"><a id="l01214" name="l01214"></a><span class="lineno"> 1214</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> input,</div>
|
||||
<div class="line"><a id="l01215" name="l01215"></a><span class="lineno"> 1215</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> weight,</div>
|
||||
<div class="line"><a id="l01216" name="l01216"></a><span class="lineno"> 1216</span> std::vector<int> stride = {},</div>
|
||||
<div class="line"><a id="l01217" name="l01217"></a><span class="lineno"> 1217</span> std::vector<int> padding_lo = {},</div>
|
||||
<div class="line"><a id="l01218" name="l01218"></a><span class="lineno"> 1218</span> std::vector<int> padding_hi = {},</div>
|
||||
<div class="line"><a id="l01219" name="l01219"></a><span class="lineno"> 1219</span> std::vector<int> kernel_dilation = {},</div>
|
||||
<div class="line"><a id="l01220" name="l01220"></a><span class="lineno"> 1220</span> std::vector<int> input_dilation = {},</div>
|
||||
<div class="line"><a id="l01221" name="l01221"></a><span class="lineno"> 1221</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01222" name="l01222"></a><span class="lineno"> 1222</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01223" name="l01223"></a><span class="lineno"> 1223</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01224" name="l01224"></a><span class="lineno"> 1224</span></div>
|
||||
<div class="foldopen" id="foldopen01226" data-start="{" data-end="}">
|
||||
<div class="line"><a id="l01226" name="l01226"></a><span class="lineno"><a class="line" href="group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d"> 1226</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
||||
<div class="line"><a id="l01227" name="l01227"></a><span class="lineno"> 1227</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01228" name="l01228"></a><span class="lineno"> 1228</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01229" name="l01229"></a><span class="lineno"> 1229</span> std::vector<int> stride = {},</div>
|
||||
<div class="line"><a id="l01230" name="l01230"></a><span class="lineno"> 1230</span> std::vector<int> padding = {},</div>
|
||||
<div class="line"><a id="l01231" name="l01231"></a><span class="lineno"> 1231</span> std::vector<int> kernel_dilation = {},</div>
|
||||
<div class="line"><a id="l01232" name="l01232"></a><span class="lineno"> 1232</span> std::vector<int> input_dilation = {},</div>
|
||||
<div class="line"><a id="l01233" name="l01233"></a><span class="lineno"> 1233</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01234" name="l01234"></a><span class="lineno"> 1234</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01235" name="l01235"></a><span class="lineno"> 1235</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
||||
<div class="line"><a id="l01236" name="l01236"></a><span class="lineno"> 1236</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">conv_general</a>(</div>
|
||||
<div class="line"><a id="l01237" name="l01237"></a><span class="lineno"> 1237</span> <span class="comment">/* const array& input = */</span> input,</div>
|
||||
<div class="line"><a id="l01238" name="l01238"></a><span class="lineno"> 1238</span> <span class="comment">/* const array& weight = */</span> weight,</div>
|
||||
<div class="line"><a id="l01239" name="l01239"></a><span class="lineno"> 1239</span> <span class="comment">/* std::vector<int> stride = */</span> stride,</div>
|
||||
<div class="line"><a id="l01240" name="l01240"></a><span class="lineno"> 1240</span> <span class="comment">/* std::vector<int> padding_lo = */</span> padding,</div>
|
||||
<div class="line"><a id="l01241" name="l01241"></a><span class="lineno"> 1241</span> <span class="comment">/* std::vector<int> padding_hi = */</span> padding,</div>
|
||||
<div class="line"><a id="l01242" name="l01242"></a><span class="lineno"> 1242</span> <span class="comment">/* std::vector<int> kernel_dilation = */</span> kernel_dilation,</div>
|
||||
<div class="line"><a id="l01243" name="l01243"></a><span class="lineno"> 1243</span> <span class="comment">/* std::vector<int> input_dilation = */</span> input_dilation,</div>
|
||||
<div class="line"><a id="l01244" name="l01244"></a><span class="lineno"> 1244</span> <span class="comment">/* int groups = */</span> groups,</div>
|
||||
<div class="line"><a id="l01245" name="l01245"></a><span class="lineno"> 1245</span> <span class="comment">/* bool flip = */</span> flip,</div>
|
||||
<div class="line"><a id="l01246" name="l01246"></a><span class="lineno"> 1246</span> <span class="comment">/* StreamOrDevice s = */</span> s);</div>
|
||||
<div class="line"><a id="l01247" name="l01247"></a><span class="lineno"> 1247</span>}</div>
|
||||
</div>
|
||||
<div class="line"><a id="l01240" name="l01240"></a><span class="lineno"> 1240</span></div>
|
||||
<div class="line"><a id="l01242" name="l01242"></a><span class="lineno"><a class="line" href="group__ops.html#ga30d47e08093c03a3676f235f9f559411"> 1242</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#ga30d47e08093c03a3676f235f9f559411">conv1d</a>(</div>
|
||||
<div class="line"><a id="l01243" name="l01243"></a><span class="lineno"> 1243</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01244" name="l01244"></a><span class="lineno"> 1244</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01245" name="l01245"></a><span class="lineno"> 1245</span> <span class="keywordtype">int</span> stride = 1,</div>
|
||||
<div class="line"><a id="l01246" name="l01246"></a><span class="lineno"> 1246</span> <span class="keywordtype">int</span> padding = 0,</div>
|
||||
<div class="line"><a id="l01247" name="l01247"></a><span class="lineno"> 1247</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
||||
<div class="line"><a id="l01248" name="l01248"></a><span class="lineno"> 1248</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01249" name="l01249"></a><span class="lineno"> 1249</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01250" name="l01250"></a><span class="lineno"> 1250</span></div>
|
||||
<div class="line"><a id="l01252" name="l01252"></a><span class="lineno"><a class="line" href="group__ops.html#ga73b02833229678786e7f302d458d5a83"> 1252</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#ga73b02833229678786e7f302d458d5a83">conv2d</a>(</div>
|
||||
<div class="line"><a id="l01253" name="l01253"></a><span class="lineno"> 1253</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01254" name="l01254"></a><span class="lineno"> 1254</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01255" name="l01255"></a><span class="lineno"> 1255</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
||||
<div class="line"><a id="l01256" name="l01256"></a><span class="lineno"> 1256</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
||||
<div class="line"><a id="l01257" name="l01257"></a><span class="lineno"> 1257</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
||||
<div class="line"><a id="l01258" name="l01258"></a><span class="lineno"> 1258</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01259" name="l01259"></a><span class="lineno"> 1259</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01260" name="l01260"></a><span class="lineno"> 1260</span></div>
|
||||
<div class="line"><a id="l01262" name="l01262"></a><span class="lineno"><a class="line" href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca"> 1262</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#ga6e9907d2f14dc4803e4306b3dbc4b3ca">conv3d</a>(</div>
|
||||
<div class="line"><a id="l01263" name="l01263"></a><span class="lineno"> 1263</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01264" name="l01264"></a><span class="lineno"> 1264</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01265" name="l01265"></a><span class="lineno"> 1265</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01266" name="l01266"></a><span class="lineno"> 1266</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
||||
<div class="line"><a id="l01267" name="l01267"></a><span class="lineno"> 1267</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01268" name="l01268"></a><span class="lineno"> 1268</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01269" name="l01269"></a><span class="lineno"> 1269</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01270" name="l01270"></a><span class="lineno"> 1270</span></div>
|
||||
<div class="line"><a id="l01272" name="l01272"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30bf1adcd78d1c2595d07b215731714"> 1272</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#gaa30bf1adcd78d1c2595d07b215731714">conv_transpose1d</a>(</div>
|
||||
<div class="line"><a id="l01273" name="l01273"></a><span class="lineno"> 1273</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01274" name="l01274"></a><span class="lineno"> 1274</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01275" name="l01275"></a><span class="lineno"> 1275</span> <span class="keywordtype">int</span> stride = 1,</div>
|
||||
<div class="line"><a id="l01276" name="l01276"></a><span class="lineno"> 1276</span> <span class="keywordtype">int</span> padding = 0,</div>
|
||||
<div class="line"><a id="l01277" name="l01277"></a><span class="lineno"> 1277</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
||||
<div class="line"><a id="l01278" name="l01278"></a><span class="lineno"> 1278</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01279" name="l01279"></a><span class="lineno"> 1279</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01280" name="l01280"></a><span class="lineno"> 1280</span></div>
|
||||
<div class="line"><a id="l01282" name="l01282"></a><span class="lineno"><a class="line" href="group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d"> 1282</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#gaebb59971cb9bc45005dc1d398e4f0a3d">conv_transpose2d</a>(</div>
|
||||
<div class="line"><a id="l01283" name="l01283"></a><span class="lineno"> 1283</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01284" name="l01284"></a><span class="lineno"> 1284</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01285" name="l01285"></a><span class="lineno"> 1285</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
||||
<div class="line"><a id="l01286" name="l01286"></a><span class="lineno"> 1286</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
||||
<div class="line"><a id="l01287" name="l01287"></a><span class="lineno"> 1287</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
||||
<div class="line"><a id="l01288" name="l01288"></a><span class="lineno"> 1288</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01289" name="l01289"></a><span class="lineno"> 1289</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01290" name="l01290"></a><span class="lineno"> 1290</span></div>
|
||||
<div class="line"><a id="l01292" name="l01292"></a><span class="lineno"><a class="line" href="group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530"> 1292</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#ga8db814da631d9cd32a8d6563bf4ac530">conv_transpose3d</a>(</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>& input,</div>
|
||||
<div class="line"><a id="l01294" name="l01294"></a><span class="lineno"> 1294</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01295" name="l01295"></a><span class="lineno"> 1295</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01296" name="l01296"></a><span class="lineno"> 1296</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
||||
<div class="line"><a id="l01297" name="l01297"></a><span class="lineno"> 1297</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01298" name="l01298"></a><span class="lineno"> 1298</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01299" name="l01299"></a><span class="lineno"> 1299</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01300" name="l01300"></a><span class="lineno"> 1300</span></div>
|
||||
<div class="line"><a id="l01302" name="l01302"></a><span class="lineno"><a class="line" href="group__ops.html#gabfa4208fb1f9b1cdd0abc563b19175af"> 1302</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="l01303" name="l01303"></a><span class="lineno"> 1303</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x,</div>
|
||||
<div class="line"><a id="l01304" name="l01304"></a><span class="lineno"> 1304</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> w,</div>
|
||||
<div class="line"><a id="l01305" name="l01305"></a><span class="lineno"> 1305</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> scales,</div>
|
||||
<div class="line"><a id="l01306" name="l01306"></a><span class="lineno"> 1306</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> biases,</div>
|
||||
<div class="line"><a id="l01307" name="l01307"></a><span class="lineno"> 1307</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="l01308" name="l01308"></a><span class="lineno"> 1308</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01309" name="l01309"></a><span class="lineno"> 1309</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01310" name="l01310"></a><span class="lineno"> 1310</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01311" name="l01311"></a><span class="lineno"> 1311</span></div>
|
||||
<div class="line"><a id="l01313" name="l01313"></a><span class="lineno"><a class="line" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da"> 1313</a></span>std::tuple<array, array, array> <a class="code hl_function" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da">quantize</a>(</div>
|
||||
<div class="line"><a id="l01314" name="l01314"></a><span class="lineno"> 1314</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
||||
<div class="line"><a id="l01315" name="l01315"></a><span class="lineno"> 1315</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01316" name="l01316"></a><span class="lineno"> 1316</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01317" name="l01317"></a><span class="lineno"> 1317</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01318" name="l01318"></a><span class="lineno"> 1318</span></div>
|
||||
<div class="line"><a id="l01320" name="l01320"></a><span class="lineno"><a class="line" href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077"> 1320</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#gabff758a5c1ce32ad7e8b78aba0164077">dequantize</a>(</div>
|
||||
<div class="line"><a id="l01321" name="l01321"></a><span class="lineno"> 1321</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
||||
<div class="line"><a id="l01322" name="l01322"></a><span class="lineno"> 1322</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
||||
<div class="line"><a id="l01323" name="l01323"></a><span class="lineno"> 1323</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
||||
<div class="line"><a id="l01324" name="l01324"></a><span class="lineno"> 1324</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01325" name="l01325"></a><span class="lineno"> 1325</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01326" name="l01326"></a><span class="lineno"> 1326</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01327" name="l01327"></a><span class="lineno"> 1327</span></div>
|
||||
<div class="line"><a id="l01329" name="l01329"></a><span class="lineno"><a class="line" href="group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0"> 1329</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#ga368a0dc0e5dfb76922e7aa55a95f12f0">gather_qmm</a>(</div>
|
||||
<div class="line"><a id="l01330" name="l01330"></a><span class="lineno"> 1330</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& x,</div>
|
||||
<div class="line"><a id="l01331" name="l01331"></a><span class="lineno"> 1331</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
||||
<div class="line"><a id="l01332" name="l01332"></a><span class="lineno"> 1332</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
||||
<div class="line"><a id="l01333" name="l01333"></a><span class="lineno"> 1333</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
||||
<div class="line"><a id="l01334" name="l01334"></a><span class="lineno"> 1334</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01335" name="l01335"></a><span class="lineno"> 1335</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01336" name="l01336"></a><span class="lineno"> 1336</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="l01337" name="l01337"></a><span class="lineno"> 1337</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01338" name="l01338"></a><span class="lineno"> 1338</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01339" name="l01339"></a><span class="lineno"> 1339</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01340" name="l01340"></a><span class="lineno"> 1340</span></div>
|
||||
<div class="line"><a id="l01342" name="l01342"></a><span class="lineno"><a class="line" href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70"> 1342</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#gaf5c9735f4690327e1500e04e728fae70">tensordot</a>(</div>
|
||||
<div class="line"><a id="l01343" name="l01343"></a><span class="lineno"> 1343</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01344" name="l01344"></a><span class="lineno"> 1344</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
||||
<div class="line"><a id="l01345" name="l01345"></a><span class="lineno"> 1345</span> <span class="keyword">const</span> <span class="keywordtype">int</span> axis = 2,</div>
|
||||
<div class="line"><a id="l01346" name="l01346"></a><span class="lineno"> 1346</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01347" name="l01347"></a><span class="lineno"> 1347</span> </div>
|
||||
<div class="line"><a id="l01348" name="l01348"></a><span class="lineno"><a class="line" href="group__ops.html#gad7fe00b566f89d607639c1a497cabbc6"> 1348</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#gaf5c9735f4690327e1500e04e728fae70">tensordot</a>(</div>
|
||||
<div class="line"><a id="l01349" name="l01349"></a><span class="lineno"> 1349</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01350" name="l01350"></a><span class="lineno"> 1350</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
||||
<div class="line"><a id="l01351" name="l01351"></a><span class="lineno"> 1351</span> <span class="keyword">const</span> std::vector<int>& axes_a,</div>
|
||||
<div class="line"><a id="l01352" name="l01352"></a><span class="lineno"> 1352</span> <span class="keyword">const</span> std::vector<int>& axes_b,</div>
|
||||
<div class="line"><a id="l01353" name="l01353"></a><span class="lineno"> 1353</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01354" name="l01354"></a><span class="lineno"> 1354</span></div>
|
||||
<div class="line"><a id="l01356" name="l01356"></a><span class="lineno"><a class="line" href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d"> 1356</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#ga866af24e10db2797e1c5a5986dbf6c0d">outer</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01357" name="l01357"></a><span class="lineno"> 1357</span></div>
|
||||
<div class="line"><a id="l01359" name="l01359"></a><span class="lineno"><a class="line" href="group__ops.html#ga654fec16a9746b390916697a2ab2546e"> 1359</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#ga654fec16a9746b390916697a2ab2546e">inner</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01360" name="l01360"></a><span class="lineno"> 1360</span></div>
|
||||
<div class="line"><a id="l01362" name="l01362"></a><span class="lineno"><a class="line" href="group__ops.html#ga82a53e083205a965387b3c3e2463244a"> 1362</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#ga82a53e083205a965387b3c3e2463244a">addmm</a>(</div>
|
||||
<div class="line"><a id="l01363" name="l01363"></a><span class="lineno"> 1363</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> c,</div>
|
||||
<div class="line"><a id="l01364" name="l01364"></a><span class="lineno"> 1364</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
||||
<div class="line"><a id="l01365" name="l01365"></a><span class="lineno"> 1365</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
||||
<div class="line"><a id="l01366" name="l01366"></a><span class="lineno"> 1366</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& alpha = 1.f,</div>
|
||||
<div class="line"><a id="l01367" name="l01367"></a><span class="lineno"> 1367</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& beta = 1.f,</div>
|
||||
<div class="line"><a id="l01368" name="l01368"></a><span class="lineno"> 1368</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01369" name="l01369"></a><span class="lineno"> 1369</span></div>
|
||||
<div class="line"><a id="l01371" name="l01371"></a><span class="lineno"><a class="line" href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6"> 1371</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#ga6b76c8ea46b19e6866af155fa5910be6">block_masked_mm</a>(</div>
|
||||
<div class="line"><a id="l01248" name="l01248"></a><span class="lineno"> 1248</span></div>
|
||||
<div class="line"><a id="l01250" name="l01250"></a><span class="lineno"><a class="line" href="group__ops.html#ga30d47e08093c03a3676f235f9f559411"> 1250</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#ga30d47e08093c03a3676f235f9f559411">conv1d</a>(</div>
|
||||
<div class="line"><a id="l01251" name="l01251"></a><span class="lineno"> 1251</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01252" name="l01252"></a><span class="lineno"> 1252</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01253" name="l01253"></a><span class="lineno"> 1253</span> <span class="keywordtype">int</span> stride = 1,</div>
|
||||
<div class="line"><a id="l01254" name="l01254"></a><span class="lineno"> 1254</span> <span class="keywordtype">int</span> padding = 0,</div>
|
||||
<div class="line"><a id="l01255" name="l01255"></a><span class="lineno"> 1255</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
||||
<div class="line"><a id="l01256" name="l01256"></a><span class="lineno"> 1256</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01257" name="l01257"></a><span class="lineno"> 1257</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01258" name="l01258"></a><span class="lineno"> 1258</span></div>
|
||||
<div class="line"><a id="l01260" name="l01260"></a><span class="lineno"><a class="line" href="group__ops.html#ga73b02833229678786e7f302d458d5a83"> 1260</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#ga73b02833229678786e7f302d458d5a83">conv2d</a>(</div>
|
||||
<div class="line"><a id="l01261" name="l01261"></a><span class="lineno"> 1261</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01262" name="l01262"></a><span class="lineno"> 1262</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01263" name="l01263"></a><span class="lineno"> 1263</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
||||
<div class="line"><a id="l01264" name="l01264"></a><span class="lineno"> 1264</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
||||
<div class="line"><a id="l01265" name="l01265"></a><span class="lineno"> 1265</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
||||
<div class="line"><a id="l01266" name="l01266"></a><span class="lineno"> 1266</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01267" name="l01267"></a><span class="lineno"> 1267</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01268" name="l01268"></a><span class="lineno"> 1268</span></div>
|
||||
<div class="line"><a id="l01270" name="l01270"></a><span class="lineno"><a class="line" href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca"> 1270</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#ga6e9907d2f14dc4803e4306b3dbc4b3ca">conv3d</a>(</div>
|
||||
<div class="line"><a id="l01271" name="l01271"></a><span class="lineno"> 1271</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01272" name="l01272"></a><span class="lineno"> 1272</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01273" name="l01273"></a><span class="lineno"> 1273</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01274" name="l01274"></a><span class="lineno"> 1274</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
||||
<div class="line"><a id="l01275" name="l01275"></a><span class="lineno"> 1275</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01276" name="l01276"></a><span class="lineno"> 1276</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01277" name="l01277"></a><span class="lineno"> 1277</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01278" name="l01278"></a><span class="lineno"> 1278</span></div>
|
||||
<div class="line"><a id="l01280" name="l01280"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30bf1adcd78d1c2595d07b215731714"> 1280</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#gaa30bf1adcd78d1c2595d07b215731714">conv_transpose1d</a>(</div>
|
||||
<div class="line"><a id="l01281" name="l01281"></a><span class="lineno"> 1281</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01282" name="l01282"></a><span class="lineno"> 1282</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01283" name="l01283"></a><span class="lineno"> 1283</span> <span class="keywordtype">int</span> stride = 1,</div>
|
||||
<div class="line"><a id="l01284" name="l01284"></a><span class="lineno"> 1284</span> <span class="keywordtype">int</span> padding = 0,</div>
|
||||
<div class="line"><a id="l01285" name="l01285"></a><span class="lineno"> 1285</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
||||
<div class="line"><a id="l01286" name="l01286"></a><span class="lineno"> 1286</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01287" name="l01287"></a><span class="lineno"> 1287</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01288" name="l01288"></a><span class="lineno"> 1288</span></div>
|
||||
<div class="line"><a id="l01290" name="l01290"></a><span class="lineno"><a class="line" href="group__ops.html#gaebb59971cb9bc45005dc1d398e4f0a3d"> 1290</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#gaebb59971cb9bc45005dc1d398e4f0a3d">conv_transpose2d</a>(</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>& input,</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>& weight,</div>
|
||||
<div class="line"><a id="l01293" name="l01293"></a><span class="lineno"> 1293</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
||||
<div class="line"><a id="l01294" name="l01294"></a><span class="lineno"> 1294</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
||||
<div class="line"><a id="l01295" name="l01295"></a><span class="lineno"> 1295</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
||||
<div class="line"><a id="l01296" name="l01296"></a><span class="lineno"> 1296</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01297" name="l01297"></a><span class="lineno"> 1297</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01298" name="l01298"></a><span class="lineno"> 1298</span></div>
|
||||
<div class="line"><a id="l01300" name="l01300"></a><span class="lineno"><a class="line" href="group__ops.html#ga8db814da631d9cd32a8d6563bf4ac530"> 1300</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#ga8db814da631d9cd32a8d6563bf4ac530">conv_transpose3d</a>(</div>
|
||||
<div class="line"><a id="l01301" name="l01301"></a><span class="lineno"> 1301</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& input,</div>
|
||||
<div class="line"><a id="l01302" name="l01302"></a><span class="lineno"> 1302</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& weight,</div>
|
||||
<div class="line"><a id="l01303" name="l01303"></a><span class="lineno"> 1303</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01304" name="l01304"></a><span class="lineno"> 1304</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
||||
<div class="line"><a id="l01305" name="l01305"></a><span class="lineno"> 1305</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
||||
<div class="line"><a id="l01306" name="l01306"></a><span class="lineno"> 1306</span> <span class="keywordtype">int</span> groups = 1,</div>
|
||||
<div class="line"><a id="l01307" name="l01307"></a><span class="lineno"> 1307</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01308" name="l01308"></a><span class="lineno"> 1308</span></div>
|
||||
<div class="line"><a id="l01310" name="l01310"></a><span class="lineno"><a class="line" href="group__ops.html#gabfa4208fb1f9b1cdd0abc563b19175af"> 1310</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="l01311" name="l01311"></a><span class="lineno"> 1311</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x,</div>
|
||||
<div class="line"><a id="l01312" name="l01312"></a><span class="lineno"> 1312</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> w,</div>
|
||||
<div class="line"><a id="l01313" name="l01313"></a><span class="lineno"> 1313</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> scales,</div>
|
||||
<div class="line"><a id="l01314" name="l01314"></a><span class="lineno"> 1314</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> biases,</div>
|
||||
<div class="line"><a id="l01315" name="l01315"></a><span class="lineno"> 1315</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="l01316" name="l01316"></a><span class="lineno"> 1316</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01317" name="l01317"></a><span class="lineno"> 1317</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01318" name="l01318"></a><span class="lineno"> 1318</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01319" name="l01319"></a><span class="lineno"> 1319</span></div>
|
||||
<div class="line"><a id="l01321" name="l01321"></a><span class="lineno"><a class="line" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da"> 1321</a></span>std::tuple<array, array, array> <a class="code hl_function" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da">quantize</a>(</div>
|
||||
<div class="line"><a id="l01322" name="l01322"></a><span class="lineno"> 1322</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
||||
<div class="line"><a id="l01323" name="l01323"></a><span class="lineno"> 1323</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01324" name="l01324"></a><span class="lineno"> 1324</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01325" name="l01325"></a><span class="lineno"> 1325</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01326" name="l01326"></a><span class="lineno"> 1326</span></div>
|
||||
<div class="line"><a id="l01328" name="l01328"></a><span class="lineno"><a class="line" href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077"> 1328</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#gabff758a5c1ce32ad7e8b78aba0164077">dequantize</a>(</div>
|
||||
<div class="line"><a id="l01329" name="l01329"></a><span class="lineno"> 1329</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
||||
<div class="line"><a id="l01330" name="l01330"></a><span class="lineno"> 1330</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
||||
<div class="line"><a id="l01331" name="l01331"></a><span class="lineno"> 1331</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
||||
<div class="line"><a id="l01332" name="l01332"></a><span class="lineno"> 1332</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01333" name="l01333"></a><span class="lineno"> 1333</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01334" name="l01334"></a><span class="lineno"> 1334</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01335" name="l01335"></a><span class="lineno"> 1335</span></div>
|
||||
<div class="line"><a id="l01337" name="l01337"></a><span class="lineno"><a class="line" href="group__ops.html#ga368a0dc0e5dfb76922e7aa55a95f12f0"> 1337</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#ga368a0dc0e5dfb76922e7aa55a95f12f0">gather_qmm</a>(</div>
|
||||
<div class="line"><a id="l01338" name="l01338"></a><span class="lineno"> 1338</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& x,</div>
|
||||
<div class="line"><a id="l01339" name="l01339"></a><span class="lineno"> 1339</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& w,</div>
|
||||
<div class="line"><a id="l01340" name="l01340"></a><span class="lineno"> 1340</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& scales,</div>
|
||||
<div class="line"><a id="l01341" name="l01341"></a><span class="lineno"> 1341</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& biases,</div>
|
||||
<div class="line"><a id="l01342" name="l01342"></a><span class="lineno"> 1342</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01343" name="l01343"></a><span class="lineno"> 1343</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01344" name="l01344"></a><span class="lineno"> 1344</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="l01345" name="l01345"></a><span class="lineno"> 1345</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
||||
<div class="line"><a id="l01346" name="l01346"></a><span class="lineno"> 1346</span> <span class="keywordtype">int</span> bits = 4,</div>
|
||||
<div class="line"><a id="l01347" name="l01347"></a><span class="lineno"> 1347</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01348" name="l01348"></a><span class="lineno"> 1348</span></div>
|
||||
<div class="line"><a id="l01350" name="l01350"></a><span class="lineno"><a class="line" href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70"> 1350</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#gaf5c9735f4690327e1500e04e728fae70">tensordot</a>(</div>
|
||||
<div class="line"><a id="l01351" name="l01351"></a><span class="lineno"> 1351</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01352" name="l01352"></a><span class="lineno"> 1352</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
||||
<div class="line"><a id="l01353" name="l01353"></a><span class="lineno"> 1353</span> <span class="keyword">const</span> <span class="keywordtype">int</span> axis = 2,</div>
|
||||
<div class="line"><a id="l01354" name="l01354"></a><span class="lineno"> 1354</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01355" name="l01355"></a><span class="lineno"> 1355</span> </div>
|
||||
<div class="line"><a id="l01356" name="l01356"></a><span class="lineno"><a class="line" href="group__ops.html#gad7fe00b566f89d607639c1a497cabbc6"> 1356</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#gaf5c9735f4690327e1500e04e728fae70">tensordot</a>(</div>
|
||||
<div class="line"><a id="l01357" name="l01357"></a><span class="lineno"> 1357</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01358" name="l01358"></a><span class="lineno"> 1358</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b,</div>
|
||||
<div class="line"><a id="l01359" name="l01359"></a><span class="lineno"> 1359</span> <span class="keyword">const</span> std::vector<int>& axes_a,</div>
|
||||
<div class="line"><a id="l01360" name="l01360"></a><span class="lineno"> 1360</span> <span class="keyword">const</span> std::vector<int>& axes_b,</div>
|
||||
<div class="line"><a id="l01361" name="l01361"></a><span class="lineno"> 1361</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01362" name="l01362"></a><span class="lineno"> 1362</span></div>
|
||||
<div class="line"><a id="l01364" name="l01364"></a><span class="lineno"><a class="line" href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d"> 1364</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#ga866af24e10db2797e1c5a5986dbf6c0d">outer</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01365" name="l01365"></a><span class="lineno"> 1365</span></div>
|
||||
<div class="line"><a id="l01367" name="l01367"></a><span class="lineno"><a class="line" href="group__ops.html#ga654fec16a9746b390916697a2ab2546e"> 1367</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#ga654fec16a9746b390916697a2ab2546e">inner</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01368" name="l01368"></a><span class="lineno"> 1368</span></div>
|
||||
<div class="line"><a id="l01370" name="l01370"></a><span class="lineno"><a class="line" href="group__ops.html#ga82a53e083205a965387b3c3e2463244a"> 1370</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#ga82a53e083205a965387b3c3e2463244a">addmm</a>(</div>
|
||||
<div class="line"><a id="l01371" name="l01371"></a><span class="lineno"> 1371</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> c,</div>
|
||||
<div class="line"><a id="l01372" name="l01372"></a><span class="lineno"> 1372</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
||||
<div class="line"><a id="l01373" name="l01373"></a><span class="lineno"> 1373</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
||||
<div class="line"><a id="l01374" name="l01374"></a><span class="lineno"> 1374</span> <span class="keywordtype">int</span> block_size,</div>
|
||||
<div class="line"><a id="l01375" name="l01375"></a><span class="lineno"> 1375</span> std::optional<array> mask_out = std::nullopt,</div>
|
||||
<div class="line"><a id="l01376" name="l01376"></a><span class="lineno"> 1376</span> std::optional<array> mask_lhs = std::nullopt,</div>
|
||||
<div class="line"><a id="l01377" name="l01377"></a><span class="lineno"> 1377</span> std::optional<array> mask_rhs = std::nullopt,</div>
|
||||
<div class="line"><a id="l01378" name="l01378"></a><span class="lineno"> 1378</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01379" name="l01379"></a><span class="lineno"> 1379</span></div>
|
||||
<div class="line"><a id="l01381" name="l01381"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7"> 1381</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#ga8d50480266d258cac40ff51bcb0fc6a7">gather_mm</a>(</div>
|
||||
<div class="line"><a id="l01382" name="l01382"></a><span class="lineno"> 1382</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
||||
<div class="line"><a id="l01383" name="l01383"></a><span class="lineno"> 1383</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
||||
<div class="line"><a id="l01384" name="l01384"></a><span class="lineno"> 1384</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01385" name="l01385"></a><span class="lineno"> 1385</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01374" name="l01374"></a><span class="lineno"> 1374</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& alpha = 1.f,</div>
|
||||
<div class="line"><a id="l01375" name="l01375"></a><span class="lineno"> 1375</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& beta = 1.f,</div>
|
||||
<div class="line"><a id="l01376" name="l01376"></a><span class="lineno"> 1376</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01377" name="l01377"></a><span class="lineno"> 1377</span></div>
|
||||
<div class="line"><a id="l01379" name="l01379"></a><span class="lineno"><a class="line" href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6"> 1379</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#ga6b76c8ea46b19e6866af155fa5910be6">block_masked_mm</a>(</div>
|
||||
<div class="line"><a id="l01380" name="l01380"></a><span class="lineno"> 1380</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
||||
<div class="line"><a id="l01381" name="l01381"></a><span class="lineno"> 1381</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
||||
<div class="line"><a id="l01382" name="l01382"></a><span class="lineno"> 1382</span> <span class="keywordtype">int</span> block_size,</div>
|
||||
<div class="line"><a id="l01383" name="l01383"></a><span class="lineno"> 1383</span> std::optional<array> mask_out = std::nullopt,</div>
|
||||
<div class="line"><a id="l01384" name="l01384"></a><span class="lineno"> 1384</span> std::optional<array> mask_lhs = std::nullopt,</div>
|
||||
<div class="line"><a id="l01385" name="l01385"></a><span class="lineno"> 1385</span> std::optional<array> mask_rhs = std::nullopt,</div>
|
||||
<div class="line"><a id="l01386" name="l01386"></a><span class="lineno"> 1386</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01387" name="l01387"></a><span class="lineno"> 1387</span></div>
|
||||
<div class="line"><a id="l01389" name="l01389"></a><span class="lineno"><a class="line" href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6"> 1389</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#ga9236b085a88ead3128ed8079d009cac6">diagonal</a>(</div>
|
||||
<div class="line"><a id="l01390" name="l01390"></a><span class="lineno"> 1390</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01391" name="l01391"></a><span class="lineno"> 1391</span> <span class="keywordtype">int</span> offset = 0,</div>
|
||||
<div class="line"><a id="l01392" name="l01392"></a><span class="lineno"> 1392</span> <span class="keywordtype">int</span> axis1 = 0,</div>
|
||||
<div class="line"><a id="l01393" name="l01393"></a><span class="lineno"> 1393</span> <span class="keywordtype">int</span> axis2 = 1,</div>
|
||||
<div class="line"><a id="l01389" name="l01389"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d50480266d258cac40ff51bcb0fc6a7"> 1389</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#ga8d50480266d258cac40ff51bcb0fc6a7">gather_mm</a>(</div>
|
||||
<div class="line"><a id="l01390" name="l01390"></a><span class="lineno"> 1390</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
||||
<div class="line"><a id="l01391" name="l01391"></a><span class="lineno"> 1391</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
||||
<div class="line"><a id="l01392" name="l01392"></a><span class="lineno"> 1392</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01393" name="l01393"></a><span class="lineno"> 1393</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
||||
<div class="line"><a id="l01394" name="l01394"></a><span class="lineno"> 1394</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01395" name="l01395"></a><span class="lineno"> 1395</span></div>
|
||||
<div class="line"><a id="l01397" name="l01397"></a><span class="lineno"><a class="line" href="group__ops.html#ga11af511875640e1fa88e0ca87e199344"> 1397</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#ga11af511875640e1fa88e0ca87e199344">diag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> k = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01398" name="l01398"></a><span class="lineno"> 1398</span></div>
|
||||
<div class="line"><a id="l01400" name="l01400"></a><span class="lineno"><a class="line" href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd"> 1400</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#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(</div>
|
||||
<div class="line"><a id="l01401" name="l01401"></a><span class="lineno"> 1401</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01402" name="l01402"></a><span class="lineno"> 1402</span> <span class="keywordtype">int</span> offset,</div>
|
||||
<div class="line"><a id="l01403" name="l01403"></a><span class="lineno"> 1403</span> <span class="keywordtype">int</span> axis1,</div>
|
||||
<div class="line"><a id="l01404" name="l01404"></a><span class="lineno"> 1404</span> <span class="keywordtype">int</span> axis2,</div>
|
||||
<div class="line"><a id="l01405" name="l01405"></a><span class="lineno"> 1405</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
||||
<div class="line"><a id="l01406" name="l01406"></a><span class="lineno"> 1406</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01407" name="l01407"></a><span class="lineno"><a class="line" href="group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd"> 1407</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#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(</div>
|
||||
<div class="line"><a id="l01408" name="l01408"></a><span class="lineno"> 1408</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01409" name="l01409"></a><span class="lineno"> 1409</span> <span class="keywordtype">int</span> offset,</div>
|
||||
<div class="line"><a id="l01410" name="l01410"></a><span class="lineno"> 1410</span> <span class="keywordtype">int</span> axis1,</div>
|
||||
<div class="line"><a id="l01411" name="l01411"></a><span class="lineno"> 1411</span> <span class="keywordtype">int</span> axis2,</div>
|
||||
<div class="line"><a id="l01412" name="l01412"></a><span class="lineno"> 1412</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01413" name="l01413"></a><span class="lineno"><a class="line" href="group__ops.html#gaf25c00108feaafaa6350a4434cb0062e"> 1413</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#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01414" name="l01414"></a><span class="lineno"> 1414</span></div>
|
||||
<div class="line"><a id="l01420" name="l01420"></a><span class="lineno"><a class="line" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759"> 1420</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759">depends</a>(</div>
|
||||
<div class="line"><a id="l01421" name="l01421"></a><span class="lineno"> 1421</span> <span class="keyword">const</span> std::vector<array>& inputs,</div>
|
||||
<div class="line"><a id="l01422" name="l01422"></a><span class="lineno"> 1422</span> <span class="keyword">const</span> std::vector<array>& dependencies);</div>
|
||||
<div class="line"><a id="l01423" name="l01423"></a><span class="lineno"> 1423</span></div>
|
||||
<div class="line"><a id="l01425" name="l01425"></a><span class="lineno"><a class="line" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396"> 1425</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#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01426" name="l01426"></a><span class="lineno"><a class="line" href="group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5"> 1426</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(</div>
|
||||
<div class="line"><a id="l01427" name="l01427"></a><span class="lineno"> 1427</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
||||
<div class="line"><a id="l01428" name="l01428"></a><span class="lineno"> 1428</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01429" name="l01429"></a><span class="lineno"><a class="line" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087"> 1429</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#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01430" name="l01430"></a><span class="lineno"><a class="line" href="group__ops.html#ga9950299a80c2562f13448758f856d1f5"> 1430</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(</div>
|
||||
<div class="line"><a id="l01431" name="l01431"></a><span class="lineno"> 1431</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
||||
<div class="line"><a id="l01432" name="l01432"></a><span class="lineno"> 1432</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01433" name="l01433"></a><span class="lineno"><a class="line" href="group__ops.html#ga4afd919601e67782ff964465919956a0"> 1433</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#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01434" name="l01434"></a><span class="lineno"><a class="line" href="group__ops.html#gaffdf742ad79440a60dda40062a8074fe"> 1434</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(</div>
|
||||
<div class="line"><a id="l01397" name="l01397"></a><span class="lineno"><a class="line" href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6"> 1397</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#ga9236b085a88ead3128ed8079d009cac6">diagonal</a>(</div>
|
||||
<div class="line"><a id="l01398" name="l01398"></a><span class="lineno"> 1398</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01399" name="l01399"></a><span class="lineno"> 1399</span> <span class="keywordtype">int</span> offset = 0,</div>
|
||||
<div class="line"><a id="l01400" name="l01400"></a><span class="lineno"> 1400</span> <span class="keywordtype">int</span> axis1 = 0,</div>
|
||||
<div class="line"><a id="l01401" name="l01401"></a><span class="lineno"> 1401</span> <span class="keywordtype">int</span> axis2 = 1,</div>
|
||||
<div class="line"><a id="l01402" name="l01402"></a><span class="lineno"> 1402</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01403" name="l01403"></a><span class="lineno"> 1403</span></div>
|
||||
<div class="line"><a id="l01405" name="l01405"></a><span class="lineno"><a class="line" href="group__ops.html#ga11af511875640e1fa88e0ca87e199344"> 1405</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#ga11af511875640e1fa88e0ca87e199344">diag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> k = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01406" name="l01406"></a><span class="lineno"> 1406</span></div>
|
||||
<div class="line"><a id="l01408" name="l01408"></a><span class="lineno"><a class="line" href="group__ops.html#gabf786129c7660ed8d5acb5499bc6fefd"> 1408</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#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(</div>
|
||||
<div class="line"><a id="l01409" name="l01409"></a><span class="lineno"> 1409</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01410" name="l01410"></a><span class="lineno"> 1410</span> <span class="keywordtype">int</span> offset,</div>
|
||||
<div class="line"><a id="l01411" name="l01411"></a><span class="lineno"> 1411</span> <span class="keywordtype">int</span> axis1,</div>
|
||||
<div class="line"><a id="l01412" name="l01412"></a><span class="lineno"> 1412</span> <span class="keywordtype">int</span> axis2,</div>
|
||||
<div class="line"><a id="l01413" name="l01413"></a><span class="lineno"> 1413</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
||||
<div class="line"><a id="l01414" name="l01414"></a><span class="lineno"> 1414</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01415" name="l01415"></a><span class="lineno"><a class="line" href="group__ops.html#ga5ed43c2dbf7d6cbddbaa2fd682deaafd"> 1415</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#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(</div>
|
||||
<div class="line"><a id="l01416" name="l01416"></a><span class="lineno"> 1416</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01417" name="l01417"></a><span class="lineno"> 1417</span> <span class="keywordtype">int</span> offset,</div>
|
||||
<div class="line"><a id="l01418" name="l01418"></a><span class="lineno"> 1418</span> <span class="keywordtype">int</span> axis1,</div>
|
||||
<div class="line"><a id="l01419" name="l01419"></a><span class="lineno"> 1419</span> <span class="keywordtype">int</span> axis2,</div>
|
||||
<div class="line"><a id="l01420" name="l01420"></a><span class="lineno"> 1420</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01421" name="l01421"></a><span class="lineno"><a class="line" href="group__ops.html#gaf25c00108feaafaa6350a4434cb0062e"> 1421</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#gabf786129c7660ed8d5acb5499bc6fefd">trace</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01422" name="l01422"></a><span class="lineno"> 1422</span></div>
|
||||
<div class="line"><a id="l01428" name="l01428"></a><span class="lineno"><a class="line" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759"> 1428</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759">depends</a>(</div>
|
||||
<div class="line"><a id="l01429" name="l01429"></a><span class="lineno"> 1429</span> <span class="keyword">const</span> std::vector<array>& inputs,</div>
|
||||
<div class="line"><a id="l01430" name="l01430"></a><span class="lineno"> 1430</span> <span class="keyword">const</span> std::vector<array>& dependencies);</div>
|
||||
<div class="line"><a id="l01431" name="l01431"></a><span class="lineno"> 1431</span></div>
|
||||
<div class="line"><a id="l01433" name="l01433"></a><span class="lineno"><a class="line" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396"> 1433</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#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01434" name="l01434"></a><span class="lineno"><a class="line" href="group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5"> 1434</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(</div>
|
||||
<div class="line"><a id="l01435" name="l01435"></a><span class="lineno"> 1435</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
||||
<div class="line"><a id="l01436" name="l01436"></a><span class="lineno"> 1436</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01437" name="l01437"></a><span class="lineno"> 1437</span></div>
|
||||
<div class="line"><a id="l01442" name="l01442"></a><span class="lineno"><a class="line" href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636"> 1442</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#ga6d5f5f72362488b956cdc4615ef6c636">number_of_elements</a>(</div>
|
||||
<div class="line"><a id="l01443" name="l01443"></a><span class="lineno"> 1443</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01444" name="l01444"></a><span class="lineno"> 1444</span> std::vector<int> axes,</div>
|
||||
<div class="line"><a id="l01445" name="l01445"></a><span class="lineno"> 1445</span> <span class="keywordtype">bool</span> inverted,</div>
|
||||
<div class="line"><a id="l01446" name="l01446"></a><span class="lineno"> 1446</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype = <a class="code hl_variable" href="namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3">int32</a>,</div>
|
||||
<div class="line"><a id="l01447" name="l01447"></a><span class="lineno"> 1447</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01448" name="l01448"></a><span class="lineno"> 1448</span> </div>
|
||||
<div class="line"><a id="l01449" name="l01449"></a><span class="lineno"><a class="line" href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23"> 1449</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#ga5b596906bf8cdc8d97ed6ddc9aeb4c23">conjugate</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01450" name="l01450"></a><span class="lineno"> 1450</span></div>
|
||||
<div class="line"><a id="l01452" name="l01452"></a><span class="lineno"><a class="line" href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada"> 1452</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#ga752fd2707dabb05d0308ba3d55346ada">bitwise_and</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01453" name="l01453"></a><span class="lineno"><a class="line" href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff"> 1453</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#gaf0d232de4cbfffda1e2c838f8afdf6ff">operator&</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01454" name="l01454"></a><span class="lineno"> 1454</span></div>
|
||||
<div class="line"><a id="l01456" name="l01456"></a><span class="lineno"><a class="line" href="group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6"> 1456</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#ga8af4f22c08c11c4ffab7e3d45e0f3cd6">bitwise_or</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01457" name="l01457"></a><span class="lineno"><a class="line" href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02"> 1457</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#ga52392a2a98f09a80da8d338c4908bd02">operator|</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01437" name="l01437"></a><span class="lineno"><a class="line" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087"> 1437</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#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01438" name="l01438"></a><span class="lineno"><a class="line" href="group__ops.html#ga9950299a80c2562f13448758f856d1f5"> 1438</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(</div>
|
||||
<div class="line"><a id="l01439" name="l01439"></a><span class="lineno"> 1439</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
||||
<div class="line"><a id="l01440" name="l01440"></a><span class="lineno"> 1440</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01441" name="l01441"></a><span class="lineno"><a class="line" href="group__ops.html#ga4afd919601e67782ff964465919956a0"> 1441</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#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01442" name="l01442"></a><span class="lineno"><a class="line" href="group__ops.html#gaffdf742ad79440a60dda40062a8074fe"> 1442</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(</div>
|
||||
<div class="line"><a id="l01443" name="l01443"></a><span class="lineno"> 1443</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
||||
<div class="line"><a id="l01444" name="l01444"></a><span class="lineno"> 1444</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01445" name="l01445"></a><span class="lineno"> 1445</span></div>
|
||||
<div class="line"><a id="l01450" name="l01450"></a><span class="lineno"><a class="line" href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636"> 1450</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#ga6d5f5f72362488b956cdc4615ef6c636">number_of_elements</a>(</div>
|
||||
<div class="line"><a id="l01451" name="l01451"></a><span class="lineno"> 1451</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01452" name="l01452"></a><span class="lineno"> 1452</span> std::vector<int> axes,</div>
|
||||
<div class="line"><a id="l01453" name="l01453"></a><span class="lineno"> 1453</span> <span class="keywordtype">bool</span> inverted,</div>
|
||||
<div class="line"><a id="l01454" name="l01454"></a><span class="lineno"> 1454</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype = <a class="code hl_variable" href="namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3">int32</a>,</div>
|
||||
<div class="line"><a id="l01455" name="l01455"></a><span class="lineno"> 1455</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01456" name="l01456"></a><span class="lineno"> 1456</span> </div>
|
||||
<div class="line"><a id="l01457" name="l01457"></a><span class="lineno"><a class="line" href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23"> 1457</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#ga5b596906bf8cdc8d97ed6ddc9aeb4c23">conjugate</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01458" name="l01458"></a><span class="lineno"> 1458</span></div>
|
||||
<div class="line"><a id="l01460" name="l01460"></a><span class="lineno"><a class="line" href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b"> 1460</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#ga3188638fba3a60e264baf69956a1e08b">bitwise_xor</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01461" name="l01461"></a><span class="lineno"><a class="line" href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181"> 1461</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#gac3a6fe18694e84b3d63458e9553ac181">operator^</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01460" name="l01460"></a><span class="lineno"><a class="line" href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada"> 1460</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#ga752fd2707dabb05d0308ba3d55346ada">bitwise_and</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01461" name="l01461"></a><span class="lineno"><a class="line" href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff"> 1461</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#gaf0d232de4cbfffda1e2c838f8afdf6ff">operator&</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01462" name="l01462"></a><span class="lineno"> 1462</span></div>
|
||||
<div class="line"><a id="l01464" name="l01464"></a><span class="lineno"><a class="line" href="group__ops.html#ga89682bf78491761e062d4ee7bef0c829"> 1464</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#ga89682bf78491761e062d4ee7bef0c829">left_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01465" name="l01465"></a><span class="lineno"><a class="line" href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e"> 1465</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#gad656c30f9fd7d9467e405657b325aa7e">operator<<</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01464" name="l01464"></a><span class="lineno"><a class="line" href="group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6"> 1464</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#ga8af4f22c08c11c4ffab7e3d45e0f3cd6">bitwise_or</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01465" name="l01465"></a><span class="lineno"><a class="line" href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02"> 1465</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#ga52392a2a98f09a80da8d338c4908bd02">operator|</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01466" name="l01466"></a><span class="lineno"> 1466</span></div>
|
||||
<div class="line"><a id="l01468" name="l01468"></a><span class="lineno"><a class="line" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23"> 1468</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#gafa376ad57d38ba87378f0272dc379b23">right_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01469" name="l01469"></a><span class="lineno"><a class="line" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a"> 1469</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#ga498b61f7e8f056ae00297fa0dc17303a">operator>></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01470" name="l01470"></a><span class="lineno"> 1470</span> </div>
|
||||
<div class="line"><a id="l01471" name="l01471"></a><span class="lineno"><a class="line" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02"> 1471</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#ga3602aa91b7b124a0b41ec1b2137a1b02">view</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a>& dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01472" name="l01472"></a><span class="lineno"> 1472</span></div>
|
||||
<div class="line"><a id="l01474" name="l01474"></a><span class="lineno"><a class="line" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c"> 1474</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01475" name="l01475"></a><span class="lineno"><a class="line" href="group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02"> 1475</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01476" name="l01476"></a><span class="lineno"><a class="line" href="group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436"> 1476</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01477" name="l01477"></a><span class="lineno"><a class="line" href="group__ops.html#ga665f502ecc96f1f4467556b784abf9ae"> 1477</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01478" name="l01478"></a><span class="lineno"> 1478</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01479" name="l01479"></a><span class="lineno"> 1479</span> <span class="keywordtype">int</span> shift,</div>
|
||||
<div class="line"><a id="l01480" name="l01480"></a><span class="lineno"> 1480</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01481" name="l01481"></a><span class="lineno"> 1481</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01482" name="l01482"></a><span class="lineno"><a class="line" href="group__ops.html#ga79137f90bc44ac9e35f408c012701df9"> 1482</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01483" name="l01483"></a><span class="lineno"><a class="line" href="group__ops.html#ga9d76930fb567a7d459ff96fb851abe36"> 1483</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01484" name="l01484"></a><span class="lineno"> 1484</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01485" name="l01485"></a><span class="lineno"> 1485</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift,</div>
|
||||
<div class="line"><a id="l01486" name="l01486"></a><span class="lineno"> 1486</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01487" name="l01487"></a><span class="lineno"> 1487</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01488" name="l01488"></a><span class="lineno"> 1488</span> </div>
|
||||
<div class="line"><a id="l01489" name="l01489"></a><span class="lineno"> 1489</span><span class="comment">/* The real part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01490" name="l01490"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764"> 1490</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#gaf8913cabeb9fb193ba687aaeb2087764">real</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01491" name="l01491"></a><span class="lineno"> 1491</span> </div>
|
||||
<div class="line"><a id="l01492" name="l01492"></a><span class="lineno"> 1492</span><span class="comment">/* The imaginary part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01493" name="l01493"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029"> 1493</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#ga7ff592a64d528f0cf4f3d098465da029">imag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01494" name="l01494"></a><span class="lineno"> 1494</span> </div>
|
||||
<div class="line"><a id="l01495" name="l01495"></a><span class="lineno"> 1495</span><span class="comment">/* Ensure the array's underlying memory is contiguous. */</span></div>
|
||||
<div class="line"><a id="l01496" name="l01496"></a><span class="lineno"><a class="line" href="group__ops.html#ga8ab10aa6c41416d739791164a52b25d5"> 1496</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#ga8ab10aa6c41416d739791164a52b25d5">contiguous</a>(</div>
|
||||
<div class="line"><a id="l01497" name="l01497"></a><span class="lineno"> 1497</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01498" name="l01498"></a><span class="lineno"> 1498</span> <span class="keywordtype">bool</span> allow_col_major = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01499" name="l01499"></a><span class="lineno"> 1499</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01500" name="l01500"></a><span class="lineno"> 1500</span></div>
|
||||
<div class="line"><a id="l01468" name="l01468"></a><span class="lineno"><a class="line" href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b"> 1468</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#ga3188638fba3a60e264baf69956a1e08b">bitwise_xor</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01469" name="l01469"></a><span class="lineno"><a class="line" href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181"> 1469</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#gac3a6fe18694e84b3d63458e9553ac181">operator^</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01470" name="l01470"></a><span class="lineno"> 1470</span></div>
|
||||
<div class="line"><a id="l01472" name="l01472"></a><span class="lineno"><a class="line" href="group__ops.html#ga89682bf78491761e062d4ee7bef0c829"> 1472</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#ga89682bf78491761e062d4ee7bef0c829">left_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01473" name="l01473"></a><span class="lineno"><a class="line" href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e"> 1473</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#gad656c30f9fd7d9467e405657b325aa7e">operator<<</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01474" name="l01474"></a><span class="lineno"> 1474</span></div>
|
||||
<div class="line"><a id="l01476" name="l01476"></a><span class="lineno"><a class="line" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23"> 1476</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#gafa376ad57d38ba87378f0272dc379b23">right_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01477" name="l01477"></a><span class="lineno"><a class="line" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a"> 1477</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#ga498b61f7e8f056ae00297fa0dc17303a">operator>></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01478" name="l01478"></a><span class="lineno"> 1478</span> </div>
|
||||
<div class="line"><a id="l01479" name="l01479"></a><span class="lineno"><a class="line" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02"> 1479</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#ga3602aa91b7b124a0b41ec1b2137a1b02">view</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a>& dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01480" name="l01480"></a><span class="lineno"> 1480</span></div>
|
||||
<div class="line"><a id="l01482" name="l01482"></a><span class="lineno"><a class="line" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c"> 1482</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01483" name="l01483"></a><span class="lineno"><a class="line" href="group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02"> 1483</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01484" name="l01484"></a><span class="lineno"><a class="line" href="group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436"> 1484</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01485" name="l01485"></a><span class="lineno"><a class="line" href="group__ops.html#ga665f502ecc96f1f4467556b784abf9ae"> 1485</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01486" name="l01486"></a><span class="lineno"> 1486</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01487" name="l01487"></a><span class="lineno"> 1487</span> <span class="keywordtype">int</span> shift,</div>
|
||||
<div class="line"><a id="l01488" name="l01488"></a><span class="lineno"> 1488</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01489" name="l01489"></a><span class="lineno"> 1489</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01490" name="l01490"></a><span class="lineno"><a class="line" href="group__ops.html#ga79137f90bc44ac9e35f408c012701df9"> 1490</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01491" name="l01491"></a><span class="lineno"><a class="line" href="group__ops.html#ga9d76930fb567a7d459ff96fb851abe36"> 1491</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#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01492" name="l01492"></a><span class="lineno"> 1492</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01493" name="l01493"></a><span class="lineno"> 1493</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift,</div>
|
||||
<div class="line"><a id="l01494" name="l01494"></a><span class="lineno"> 1494</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01495" name="l01495"></a><span class="lineno"> 1495</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01496" name="l01496"></a><span class="lineno"> 1496</span> </div>
|
||||
<div class="line"><a id="l01497" name="l01497"></a><span class="lineno"> 1497</span><span class="comment">/* The real part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01498" name="l01498"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764"> 1498</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#gaf8913cabeb9fb193ba687aaeb2087764">real</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01499" name="l01499"></a><span class="lineno"> 1499</span> </div>
|
||||
<div class="line"><a id="l01500" name="l01500"></a><span class="lineno"> 1500</span><span class="comment">/* The imaginary part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01501" name="l01501"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029"> 1501</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#ga7ff592a64d528f0cf4f3d098465da029">imag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01502" name="l01502"></a><span class="lineno"> 1502</span> </div>
|
||||
<div class="line"><a id="l01503" name="l01503"></a><span class="lineno"> 1503</span>} <span class="comment">// namespace mlx::core</span></div>
|
||||
<div class="line"><a id="l01503" name="l01503"></a><span class="lineno"> 1503</span><span class="comment">/* Ensure the array's underlying memory is contiguous. */</span></div>
|
||||
<div class="line"><a id="l01504" name="l01504"></a><span class="lineno"><a class="line" href="group__ops.html#ga8ab10aa6c41416d739791164a52b25d5"> 1504</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#ga8ab10aa6c41416d739791164a52b25d5">contiguous</a>(</div>
|
||||
<div class="line"><a id="l01505" name="l01505"></a><span class="lineno"> 1505</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01506" name="l01506"></a><span class="lineno"> 1506</span> <span class="keywordtype">bool</span> allow_col_major = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01507" name="l01507"></a><span class="lineno"> 1507</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01508" name="l01508"></a><span class="lineno"> 1508</span></div>
|
||||
<div class="line"><a id="l01510" name="l01510"></a><span class="lineno"> 1510</span> </div>
|
||||
<div class="line"><a id="l01511" name="l01511"></a><span class="lineno"> 1511</span>} <span class="comment">// namespace mlx::core</span></div>
|
||||
<div class="ttc" id="aarray_8h_html"><div class="ttname"><a href="array_8h.html">array.h</a></div></div>
|
||||
<div class="ttc" id="aclassmlx_1_1core_1_1array_html"><div class="ttname"><a href="classmlx_1_1core_1_1array.html">mlx::core::array</a></div><div class="ttdef"><b>Definition</b> array.h:24</div></div>
|
||||
<div class="ttc" id="adevice_8h_html"><div class="ttname"><a href="device_8h.html">device.h</a></div></div>
|
||||
@@ -1427,7 +1434,6 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="agroup__ops_html_ga27af56a98270d4d76d139f0f9171b83a"><div class="ttname"><a href="group__ops.html#ga27af56a98270d4d76d139f0f9171b83a">mlx::core::operator||</a></div><div class="ttdeci">array operator||(const array &a, const array &b)</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga28f22c5d2c399eee53be7b3facc11103"><div class="ttname"><a href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">mlx::core::not_equal</a></div><div class="ttdeci">array not_equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns the bool array with (a != b) element-wise.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga292a335240fd5d6d625fb7a340ff5eb0"><div class="ttname"><a href="group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0">mlx::core::erf</a></div><div class="ttdeci">array erf(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Computes the error function of the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga29718cd5005dbcde0396b6fd65cc041d"><div class="ttname"><a href="group__ops.html#ga29718cd5005dbcde0396b6fd65cc041d">mlx::core::slice</a></div><div class="ttdeci">array slice(const array &a, Shape start, Shape stop, Shape strides, StreamOrDevice s={})</div><div class="ttdoc">Slice an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga297f853b3d90ec8ae81263977ba2ddb1"><div class="ttname"><a href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1">mlx::core::sqrt</a></div><div class="ttdeci">array sqrt(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Square root the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga2a466024f8061febc0a64be557644cb0"><div class="ttname"><a href="group__ops.html#ga2a466024f8061febc0a64be557644cb0">mlx::core::std</a></div><div class="ttdeci">array std(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})</div><div class="ttdoc">Computes the standard deviation of the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga2d32d67cfd76785a72c43d89b94dc7d7"><div class="ttname"><a href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">mlx::core::add</a></div><div class="ttdeci">array add(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Add two arrays.</div></div>
|
||||
@@ -1460,9 +1466,8 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="agroup__ops_html_ga5487dd887c43e5341f3e68ffe47f0f5a"><div class="ttname"><a href="group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a">mlx::core::topk</a></div><div class="ttdeci">array topk(const array &a, int k, StreamOrDevice s={})</div><div class="ttdoc">Returns topk elements of the flattened array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga54ca54f06bfb2be15b163a5209e2a0f0"><div class="ttname"><a href="group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0">mlx::core::expm1</a></div><div class="ttdeci">array expm1(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Computes the expm1 function of the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga5528e80f5e8bad71e106a0cf9edd8920"><div class="ttname"><a href="group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920">mlx::core::abs</a></div><div class="ttdeci">array abs(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Absolute value of elements in an array.</div></div>
|
||||
<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< array > meshgrid(const std::vector< array > &arrays, bool sparse=false, std::string indexing="xy", StreamOrDevice s={})</div><div class="ttdoc">A vector of coordinate arrays from coordinate vectors.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga58e03f39aa5d63aba1646f530706a920"><div class="ttname"><a href="group__ops.html#ga58e03f39aa5d63aba1646f530706a920">mlx::core::pad</a></div><div class="ttdeci">array pad(const array &a, const std::vector< int > &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value=array(0), const std::string mode="constant", StreamOrDevice s={})</div><div class="ttdoc">Pad an array with a constant value.</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 &a, StreamOrDevice s={})</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga5ecddb74ba7861eb82eca8653501d5dc"><div class="ttname"><a href="group__ops.html#ga5ecddb74ba7861eb82eca8653501d5dc">mlx::core::meshgrid</a></div><div class="ttdeci">std::vector< array > meshgrid(const std::vector< array > &arrays, bool sparse=false, const std::string &indexing="xy", StreamOrDevice s={})</div><div class="ttdoc">A vector of coordinate arrays from coordinate vectors.</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 &a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Tangent of the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga6085b03f2662ef2a61de523fd609f3bf"><div class="ttname"><a href="group__ops.html#ga6085b03f2662ef2a61de523fd609f3bf">mlx::core::as_strided</a></div><div class="ttdeci">array as_strided(array a, Shape shape, Strides strides, size_t offset, StreamOrDevice s={})</div><div class="ttdoc">Create a view of an array with the given shape and strides.</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 &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the inner product of two vectors.</div></div>
|
||||
@@ -1482,7 +1487,6 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="agroup__ops_html_ga74fd2777adef10e6fe628a9cdadb01cb"><div class="ttname"><a href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb">mlx::core::operator></a></div><div class="ttdeci">array operator>(const array &a, const array &b)</div><div class="ttdef"><b>Definition</b> ops.h:341</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga752fd2707dabb05d0308ba3d55346ada"><div class="ttname"><a href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada">mlx::core::bitwise_and</a></div><div class="ttdeci">array bitwise_and(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Bitwise and.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga7534290bceab5fb3831a05d67bebce7d"><div class="ttname"><a href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">mlx::core::split</a></div><div class="ttdeci">std::vector< array > split(const array &a, int num_splits, int axis, StreamOrDevice s={})</div><div class="ttdoc">Split an array into sub-arrays along a given axis.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga753d59f5a9f5f2362865ee83b4dced2a"><div class="ttname"><a href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a">mlx::core::matmul</a></div><div class="ttdeci">array matmul(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Matrix-matrix multiplication.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga768977cda8d68cf23f464a6af9907876"><div class="ttname"><a href="group__ops.html#ga768977cda8d68cf23f464a6af9907876">mlx::core::logical_and</a></div><div class="ttdeci">array logical_and(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Logical and of two arrays.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga76fb9062c64264e34d2e07013390557c"><div class="ttname"><a href="group__ops.html#ga76fb9062c64264e34d2e07013390557c">mlx::core::erfinv</a></div><div class="ttdeci">array erfinv(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Computes the inverse error function of the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_ga77472dd06cfa7a30a42e4fd927bd859f"><div class="ttname"><a href="group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f">mlx::core::divide</a></div><div class="ttdeci">array divide(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Divide two arrays.</div></div>
|
||||
@@ -1529,10 +1533,12 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="agroup__ops_html_gab1467c6a9e675152e768afd6dcfb61de"><div class="ttname"><a href="group__ops.html#gab1467c6a9e675152e768afd6dcfb61de">mlx::core::nan_to_num</a></div><div class="ttdeci">array nan_to_num(const array &a, float nan=0.0f, const std::optional< float > posinf=std::nullopt, const std::optional< float > neginf=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Replace NaN and infinities with finite numbers.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab27599802617a4c8f9964ab5f4ffee12"><div class="ttname"><a href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">mlx::core::min</a></div><div class="ttdeci">array min(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The minimum of all elements of the array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab3bfbf82b1e4de7b00bbcf1a2255fbde"><div class="ttname"><a href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde">mlx::core::operator%</a></div><div class="ttdeci">array operator%(const array &a, const array &b)</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab3fd98c0d06b84b836f93bddbd7a2a0d"><div class="ttname"><a href="group__ops.html#gab3fd98c0d06b84b836f93bddbd7a2a0d">mlx::core::scatter_add_axis</a></div><div class="ttdeci">array scatter_add_axis(const array &a, const array &indices, const array &values, int axis, StreamOrDevice s={})</div><div class="ttdoc">Add the values into the array at the given indices along the axis.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab43cc28690da7cdd43b43065adbd31da"><div class="ttname"><a href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da">mlx::core::quantize</a></div><div class="ttdeci">std::tuple< array, array, array > quantize(const array &w, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Quantize a matrix along its last axis.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab46a35925a04c5a9d2ec7898ee55358e"><div class="ttname"><a href="group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e">mlx::core::arctanh</a></div><div class="ttdeci">array arctanh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Inverse Hyperbolic Tangent of the elements of an array.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab49e3a687e826554ed1574186e8ae974"><div class="ttname"><a href="group__ops.html#gab49e3a687e826554ed1574186e8ae974">mlx::core::repeat</a></div><div class="ttdeci">array repeat(const array &arr, int repeats, int axis, StreamOrDevice s={})</div><div class="ttdoc">Repeat an array along an axis.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab783890428b596f715dc7dd2057eae99"><div class="ttname"><a href="group__ops.html#gab783890428b596f715dc7dd2057eae99">mlx::core::broadcast_arrays</a></div><div class="ttdeci">std::vector< array > broadcast_arrays(const std::vector< array > &inputs, StreamOrDevice s={})</div><div class="ttdoc">Broadcast a vector of arrays against one another.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gab95ebd20bd7c6d1c840007cc020cbc0c"><div class="ttname"><a href="group__ops.html#gab95ebd20bd7c6d1c840007cc020cbc0c">mlx::core::pad</a></div><div class="ttdeci">array pad(const array &a, const std::vector< int > &axes, const Shape &low_pad_size, const Shape &high_pad_size, const array &pad_value=array(0), const std::string &mode="constant", StreamOrDevice s={})</div><div class="ttdoc">Pad an array with a constant value.</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gaba4d25e7a2bf87ba4feb7837ec7fa396"><div class="ttname"><a href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">mlx::core::atleast_1d</a></div><div class="ttdeci">array atleast_1d(const array &a, StreamOrDevice s={})</div><div class="ttdoc">convert an array to an atleast ndim array</div></div>
|
||||
<div class="ttc" id="agroup__ops_html_gabc46eed81ab6c6247903e4ec0c4ec1fb"><div class="ttname"><a href="group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb">mlx::core::swapaxes</a></div><div class="ttdeci">array swapaxes(const array &a, int axis1, int axis2, StreamOrDevice s={})</div><div class="ttdoc">Swap two axes of an array.</div></div>
|
||||
<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 &a, StreamOrDevice s={})</div><div class="ttdoc">Logical not of an array.</div></div>
|
||||
@@ -1582,6 +1588,7 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a369aa886219b83cf219e7a7862ce260b"><div class="ttname"><a href="namespacemlx_1_1core.html#a369aa886219b83cf219e7a7862ce260b">mlx::core::arange</a></div><div class="ttdeci">void arange(const std::vector< array > &inputs, array &out, double start, double step)</div><div class="ttdef"><b>Definition</b> arange.h:24</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a4734a596e57434492ddfe79f2cb9dbf9"><div class="ttname"><a href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">mlx::core::to_stream</a></div><div class="ttdeci">Stream to_stream(StreamOrDevice s)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a479648542a2bea151b947b18f0e79dd2"><div class="ttname"><a href="namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2">mlx::core::copy</a></div><div class="ttdeci">void copy(const array &src, array &dst, CopyType ctype)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a5adff87687b78bfc18dafbc654925cdb"><div class="ttname"><a href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">mlx::core::slice</a></div><div class="ttdeci">void slice(const array &in, array &out, const Shape &start_indices, const Shape &strides)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a5d6373aad1444edc9de1eb07bfe5cad3"><div class="ttname"><a href="namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3">mlx::core::int32</a></div><div class="ttdeci">constexpr Dtype int32</div><div class="ttdef"><b>Definition</b> dtype.h:76</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a6894543b340321193dfb8052c438a319"><div class="ttname"><a href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">mlx::core::float32</a></div><div class="ttdeci">constexpr Dtype float32</div><div class="ttdef"><b>Definition</b> dtype.h:80</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a694e23f2d59606643728ad443d621416"><div class="ttname"><a href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">mlx::core::Shape</a></div><div class="ttdeci">std::vector< ShapeElem > Shape</div><div class="ttdef"><b>Definition</b> array.h:21</div></div>
|
||||
@@ -1590,6 +1597,7 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a937503d72b66c661bf3f5fdcd98ef97c"><div class="ttname"><a href="namespacemlx_1_1core.html#a937503d72b66c661bf3f5fdcd98ef97c">mlx::core::operator==</a></div><div class="ttdeci">bool operator==(const Device &lhs, const Device &rhs)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a94d00a1b7f8a4717ab3f26f45e4da655"><div class="ttname"><a href="namespacemlx_1_1core.html#a94d00a1b7f8a4717ab3f26f45e4da655">mlx::core::operator!=</a></div><div class="ttdeci">bool operator!=(const Device &lhs, const Device &rhs)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a95fc1013cc48fbfee0c54310711a5e58"><div class="ttname"><a href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">mlx::core::StreamOrDevice</a></div><div class="ttdeci">std::variant< std::monostate, Stream, Device > StreamOrDevice</div><div class="ttdef"><b>Definition</b> utils.h:15</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_aaacf0afe13d77a5c49ce96f1e833eb2d"><div class="ttname"><a href="namespacemlx_1_1core.html#aaacf0afe13d77a5c49ce96f1e833eb2d">mlx::core::matmul</a></div><div class="ttdeci">void matmul(const array &a, const array &b, array &out, bool a_transposed, bool b_transposed, size_t lda, size_t ldb, float alpha, float beta)</div></div>
|
||||
<div class="ttc" id="astream_8h_html"><div class="ttname"><a href="stream_8h.html">stream.h</a></div></div>
|
||||
<div class="ttc" id="astructmlx_1_1core_1_1_dtype_html"><div class="ttname"><a href="structmlx_1_1core_1_1_dtype.html">mlx::core::Dtype</a></div><div class="ttdef"><b>Definition</b> dtype.h:13</div></div>
|
||||
<div class="ttc" id="autils_8h_html"><div class="ttname"><a href="utils_8h.html">utils.h</a></div></div>
|
||||
@@ -1599,7 +1607,7 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
|
||||
<ul>
|
||||
<li class="navelem"><a class="el" href="dir_938ab0ecf10b8b860ff766c820f665fd.html">mlx</a></li><li class="navelem"><a class="el" href="ops_8h.html">ops.h</a></li>
|
||||
<li class="footer">Generated by <a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.13.1 </li>
|
||||
<li class="footer">Generated by <a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.13.2 </li>
|
||||
</ul>
|
||||
</div>
|
||||
</body>
|
||||
|
||||
Reference in New Issue
Block a user