mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
1456 lines
339 KiB
HTML
1456 lines
339 KiB
HTML
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
|
|
<html xmlns="http://www.w3.org/1999/xhtml" lang="en-US">
|
|
<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.10.0"/>
|
|
<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"/>
|
|
<script type="text/javascript" src="jquery.js"></script>
|
|
<script type="text/javascript" src="dynsections.js"></script>
|
|
<script type="text/javascript" src="clipboard.js"></script>
|
|
<script type="text/javascript" src="cookie.js"></script>
|
|
<link href="search/search.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="search/searchdata.js"></script>
|
|
<script type="text/javascript" src="search/search.js"></script>
|
|
<link href="doxygen.css" rel="stylesheet" type="text/css" />
|
|
</head>
|
|
<body>
|
|
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
|
|
<div id="titlearea">
|
|
<table cellspacing="0" cellpadding="0">
|
|
<tbody>
|
|
<tr id="projectrow">
|
|
<td id="projectalign">
|
|
<div id="projectname">MLX
|
|
</div>
|
|
</td>
|
|
</tr>
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
<!-- end header part -->
|
|
<!-- Generated by Doxygen 1.10.0 -->
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
var searchBox = new SearchBox("searchBox", "search/",'.html');
|
|
/* @license-end */
|
|
</script>
|
|
<script type="text/javascript" src="menudata.js"></script>
|
|
<script type="text/javascript" src="menu.js"></script>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function() {
|
|
initMenu('',true,false,'search.php','Search');
|
|
$(function() { init_search(); });
|
|
});
|
|
/* @license-end */
|
|
</script>
|
|
<div id="main-nav"></div>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function() { codefold.init(0); });
|
|
/* @license-end */
|
|
</script>
|
|
<!-- window showing the filter options -->
|
|
<div id="MSearchSelectWindow"
|
|
onmouseover="return searchBox.OnSearchSelectShow()"
|
|
onmouseout="return searchBox.OnSearchSelectHide()"
|
|
onkeydown="return searchBox.OnSearchSelectKey(event)">
|
|
</div>
|
|
|
|
<!-- iframe showing the search results (closed by default) -->
|
|
<div id="MSearchResultsWindow">
|
|
<div id="MSearchResults">
|
|
<div class="SRPage">
|
|
<div id="SRIndex">
|
|
<div id="SRResults"></div>
|
|
<div class="SRStatus" id="Loading">Loading...</div>
|
|
<div class="SRStatus" id="Searching">Searching...</div>
|
|
<div class="SRStatus" id="NoMatches">No Matches</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
|
|
<div id="nav-path" class="navpath">
|
|
<ul>
|
|
<li class="navelem"><a class="el" href="dir_938ab0ecf10b8b860ff766c820f665fd.html">mlx</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="headertitle"><div class="title">ops.h</div></div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
<a href="ops_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a id="l00001" name="l00001"></a><span class="lineno"> 1</span><span class="comment">// Copyright © 2023-2024 Apple Inc.</span></div>
|
|
<div class="line"><a id="l00002" name="l00002"></a><span class="lineno"> 2</span> </div>
|
|
<div class="line"><a id="l00003" name="l00003"></a><span class="lineno"> 3</span><span class="preprocessor">#pragma once</span></div>
|
|
<div class="line"><a id="l00004" name="l00004"></a><span class="lineno"> 4</span> </div>
|
|
<div class="line"><a id="l00005" name="l00005"></a><span class="lineno"> 5</span><span class="preprocessor">#include <optional></span></div>
|
|
<div class="line"><a id="l00006" name="l00006"></a><span class="lineno"> 6</span> </div>
|
|
<div class="line"><a id="l00007" name="l00007"></a><span class="lineno"> 7</span><span class="preprocessor">#include "<a class="code" href="array_8h.html">mlx/array.h</a>"</span></div>
|
|
<div class="line"><a id="l00008" name="l00008"></a><span class="lineno"> 8</span><span class="preprocessor">#include "<a class="code" href="device_8h.html">mlx/device.h</a>"</span></div>
|
|
<div class="line"><a id="l00009" name="l00009"></a><span class="lineno"> 9</span><span class="preprocessor">#include "<a class="code" href="stream_8h.html">mlx/stream.h</a>"</span></div>
|
|
<div class="line"><a id="l00010" name="l00010"></a><span class="lineno"> 10</span><span class="preprocessor">#include "<a class="code" href="utils_8h.html">mlx/utils.h</a>"</span></div>
|
|
<div class="line"><a id="l00011" name="l00011"></a><span class="lineno"> 11</span> </div>
|
|
<div class="line"><a id="l00012" name="l00012"></a><span class="lineno"> 12</span><span class="keyword">namespace </span><a class="code hl_namespace" href="namespacemlx_1_1core.html">mlx::core</a> {</div>
|
|
<div class="line"><a id="l00013" name="l00013"></a><span class="lineno"> 13</span> </div>
|
|
<div class="line"><a id="l00022" name="l00022"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ca088b8090b9f84f2e08345cf3f835a"> 22</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(</div>
|
|
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno"> 23</span> <span class="keywordtype">double</span> start,</div>
|
|
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno"> 24</span> <span class="keywordtype">double</span> stop,</div>
|
|
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span> <span class="keywordtype">double</span> step,</div>
|
|
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"> 26</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
|
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"> 27</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"><a class="line" href="group__ops.html#ga4c36b841dc5cba391dad029be5a0ad98"> 28</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> start, <span class="keywordtype">double</span> stop, <span class="keywordtype">double</span> step, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d7cf9eb15e2daf1469058907e8abc85"> 29</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> start, <span class="keywordtype">double</span> stop, <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="l00030" name="l00030"></a><span class="lineno"><a class="line" href="group__ops.html#ga74566a14e69ba6a25f5a35e7ade5c282"> 30</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> start, <span class="keywordtype">double</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno"><a class="line" href="group__ops.html#ga345aa27af3dae3646b8b4b1068e89a3e"> 31</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> stop, <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="l00032" name="l00032"></a><span class="lineno"><a class="line" href="group__ops.html#gaae179075d0fe23f4bd53fdf8c41f4c70"> 32</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">double</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno"> 33</span> </div>
|
|
<div class="line"><a id="l00034" name="l00034"></a><span class="lineno"><a class="line" href="group__ops.html#ga6b945f513077c2978afc1a952c884860"> 34</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">int</span> start, <span class="keywordtype">int</span> stop, <span class="keywordtype">int</span> step, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"><a class="line" href="group__ops.html#ga1c39fcc6eaa1c1867735c7f849d708d6"> 35</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">int</span> start, <span class="keywordtype">int</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"><a class="line" href="group__ops.html#gafe6e4580452c873cac294f16129e633f"> 36</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#a369aa886219b83cf219e7a7862ce260b">arange</a>(<span class="keywordtype">int</span> stop, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span> </div>
|
|
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"><a class="line" href="group__ops.html#ga968bcabed902311dcfbd903b0fb886ec"> 39</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#ga968bcabed902311dcfbd903b0fb886ec">linspace</a>(</div>
|
|
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span> <span class="keywordtype">double</span> start,</div>
|
|
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span> <span class="keywordtype">double</span> stop,</div>
|
|
<div class="line"><a id="l00042" name="l00042"></a><span class="lineno"> 42</span> <span class="keywordtype">int</span> num = 50,</div>
|
|
<div class="line"><a id="l00043" name="l00043"></a><span class="lineno"> 43</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#a6894543b340321193dfb8052c438a319">float32</a>,</div>
|
|
<div class="line"><a id="l00044" name="l00044"></a><span class="lineno"> 44</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00045" name="l00045"></a><span class="lineno"> 45</span> </div>
|
|
<div class="line"><a id="l00047" name="l00047"></a><span class="lineno"><a class="line" href="group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62"> 47</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#ga0e58c24fc5668e5a521e5b45e8370a62">astype</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a, <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="l00048" name="l00048"></a><span class="lineno"> 48</span> </div>
|
|
<div class="line"><a id="l00050" name="l00050"></a><span class="lineno"><a class="line" href="group__ops.html#ga8de80ecef30fc560003d40f61a38b99d"> 50</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#ga8de80ecef30fc560003d40f61a38b99d">as_strided</a>(</div>
|
|
<div class="line"><a id="l00051" name="l00051"></a><span class="lineno"> 51</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l00052" name="l00052"></a><span class="lineno"> 52</span> std::vector<int> shape,</div>
|
|
<div class="line"><a id="l00053" name="l00053"></a><span class="lineno"> 53</span> std::vector<size_t> strides,</div>
|
|
<div class="line"><a id="l00054" name="l00054"></a><span class="lineno"> 54</span> <span class="keywordtype">size_t</span> offset,</div>
|
|
<div class="line"><a id="l00055" name="l00055"></a><span class="lineno"> 55</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00056" name="l00056"></a><span class="lineno"> 56</span> </div>
|
|
<div class="line"><a id="l00058" name="l00058"></a><span class="lineno"><a class="line" href="group__ops.html#gae306e93af12f774bd80bad6c231b09d6"> 58</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#a479648542a2bea151b947b18f0e79dd2">copy</a>(<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="l00059" name="l00059"></a><span class="lineno"> 59</span> </div>
|
|
<div class="line"><a id="l00061" name="l00061"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84"> 61</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#ga7e85a476530c96ab73b894d1bddacf84">full</a>(</div>
|
|
<div class="line"><a id="l00062" name="l00062"></a><span class="lineno"> 62</span> std::vector<int> shape,</div>
|
|
<div class="line"><a id="l00063" name="l00063"></a><span class="lineno"> 63</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> vals,</div>
|
|
<div class="line"><a id="l00064" name="l00064"></a><span class="lineno"> 64</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype,</div>
|
|
<div class="line"><a id="l00065" name="l00065"></a><span class="lineno"> 65</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00066" name="l00066"></a><span class="lineno"><a class="line" href="group__ops.html#ga9e3b15c3ec6b7922296777c5e1878d52"> 66</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#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::vector<int> shape, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> vals, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00067" name="l00067"></a><span class="lineno"> 67</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00068" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00068" name="l00068"></a><span class="lineno"><a class="line" href="group__ops.html#gac34ff7f1e217b93e878053aaa5a51e58"> 68</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#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::vector<int> shape, T val, <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="l00069" name="l00069"></a><span class="lineno"> 69</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::move(shape), array(val, dtype), <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00070" name="l00070"></a><span class="lineno"> 70</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00071" name="l00071"></a><span class="lineno"> 71</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00072" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00072" name="l00072"></a><span class="lineno"><a class="line" href="group__ops.html#ga5e33cf83db1a65c35e688142bd41d35c"> 72</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#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::vector<int> shape, T val, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00073" name="l00073"></a><span class="lineno"> 73</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">full</a>(std::move(shape), array(val), <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00074" name="l00074"></a><span class="lineno"> 74</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00075" name="l00075"></a><span class="lineno"> 75</span> </div>
|
|
<div class="line"><a id="l00077" name="l00077"></a><span class="lineno"><a class="line" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8"> 77</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#ga34573eb2ead179839579142f3aedb3b8">zeros</a>(<span class="keyword">const</span> std::vector<int>& shape, <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="foldopen" id="foldopen00078" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00078" name="l00078"></a><span class="lineno"><a class="line" href="group__ops.html#ga7c1487f8a830c9df75a961b3a24527ee"> 78</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#ga34573eb2ead179839579142f3aedb3b8">zeros</a>(<span class="keyword">const</span> std::vector<int>& shape, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00079" name="l00079"></a><span class="lineno"> 79</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8">zeros</a>(shape, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00080" name="l00080"></a><span class="lineno"> 80</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00081" name="l00081"></a><span class="lineno"><a class="line" href="group__ops.html#gafbb857094d784b38c78683a091ffdbde"> 81</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#gafbb857094d784b38c78683a091ffdbde">zeros_like</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="l00082" name="l00082"></a><span class="lineno"> 82</span> </div>
|
|
<div class="line"><a id="l00084" name="l00084"></a><span class="lineno"><a class="line" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2"> 84</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#ga54eeed455321a54c8e72e16552a978f2">ones</a>(<span class="keyword">const</span> std::vector<int>& shape, <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="foldopen" id="foldopen00085" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00085" name="l00085"></a><span class="lineno"><a class="line" href="group__ops.html#ga6cf4b5e8580e4436302c519d05897dab"> 85</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#ga54eeed455321a54c8e72e16552a978f2">ones</a>(<span class="keyword">const</span> std::vector<int>& shape, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00086" name="l00086"></a><span class="lineno"> 86</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">ones</a>(shape, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00087" name="l00087"></a><span class="lineno"> 87</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00088" name="l00088"></a><span class="lineno"><a class="line" href="group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42"> 88</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#ga94f8d3b1906fee99da9cbe39f7be7d42">ones_like</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="l00089" name="l00089"></a><span class="lineno"> 89</span> </div>
|
|
<div class="line"><a id="l00092" name="l00092"></a><span class="lineno"><a class="line" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3"> 92</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#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <span class="keywordtype">int</span> k, <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="foldopen" id="foldopen00093" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00093" name="l00093"></a><span class="lineno"><a class="line" href="group__ops.html#ga2c9011310a1fa7c82f942f54102c36dd"> 93</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#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <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="l00094" name="l00094"></a><span class="lineno"> 94</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, n, 0, dtype, s);</div>
|
|
<div class="line"><a id="l00095" name="l00095"></a><span class="lineno"> 95</span>}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00096" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00096" name="l00096"></a><span class="lineno"><a class="line" href="group__ops.html#ga61657db78ef35d41112d362c869c25d2"> 96</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#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00097" name="l00097"></a><span class="lineno"> 97</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, m, 0, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00098" name="l00098"></a><span class="lineno"> 98</span>}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00099" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00099" name="l00099"></a><span class="lineno"><a class="line" href="group__ops.html#ga908a15b42834be498a46856c99dfc779"> 99</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#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <span class="keywordtype">int</span> k, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00100" name="l00100"></a><span class="lineno"> 100</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, m, k, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00101" name="l00101"></a><span class="lineno"> 101</span>}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00102" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00102" name="l00102"></a><span class="lineno"><a class="line" href="group__ops.html#gab777fcf6d4a89172c69ec3492548dc0f"> 102</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#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(<span class="keywordtype">int</span> n, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00103" name="l00103"></a><span class="lineno"> 103</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">eye</a>(n, n, 0, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00104" name="l00104"></a><span class="lineno"> 104</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00105" name="l00105"></a><span class="lineno"> 105</span> </div>
|
|
<div class="line"><a id="l00108" name="l00108"></a><span class="lineno"><a class="line" href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600"> 108</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#ga484eaa10d5e19a4ca46d3a9cd9fab600">identity</a>(<span class="keywordtype">int</span> n, <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="foldopen" id="foldopen00109" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00109" name="l00109"></a><span class="lineno"><a class="line" href="group__ops.html#gad994d65ac6019c26b5ad6c41179d3424"> 109</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#ga484eaa10d5e19a4ca46d3a9cd9fab600">identity</a>(<span class="keywordtype">int</span> n, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00110" name="l00110"></a><span class="lineno"> 110</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600">identity</a>(n, <a class="code hl_variable" href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">float32</a>, s);</div>
|
|
<div class="line"><a id="l00111" name="l00111"></a><span class="lineno"> 111</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00112" name="l00112"></a><span class="lineno"> 112</span> </div>
|
|
<div class="line"><a id="l00113" name="l00113"></a><span class="lineno"><a class="line" href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78"> 113</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#ga4f3389e5b89e70e862e7d2b40d6c7f78">tri</a>(<span class="keywordtype">int</span> n, <span class="keywordtype">int</span> m, <span class="keywordtype">int</span> k, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> type, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00114" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00114" name="l00114"></a><span class="lineno"><a class="line" href="group__ops.html#gac19a1bd6ed6d5c7bc9d258820189dbb5"> 114</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#ga4f3389e5b89e70e862e7d2b40d6c7f78">tri</a>(<span class="keywordtype">int</span> n, <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> type, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00115" name="l00115"></a><span class="lineno"> 115</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78">tri</a>(n, n, 0, type, s);</div>
|
|
<div class="line"><a id="l00116" name="l00116"></a><span class="lineno"> 116</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00117" name="l00117"></a><span class="lineno"> 117</span> </div>
|
|
<div class="line"><a id="l00118" name="l00118"></a><span class="lineno"><a class="line" href="group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2"> 118</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#ga83e0bb45dc770cf014531d873b78c5a2">tril</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x, <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="l00119" name="l00119"></a><span class="lineno"><a class="line" href="group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412"> 119</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#gaa9df5917876eeb0cb28b7fa81f880412">triu</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> x, <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="l00120" name="l00120"></a><span class="lineno"> 120</span> </div>
|
|
<div class="line"><a id="l00122" name="l00122"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b0e3d46a39d99b1628b422049790082"> 122</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#ga7b0e3d46a39d99b1628b422049790082">reshape</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, std::vector<int> shape, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00123" name="l00123"></a><span class="lineno"> 123</span> </div>
|
|
<div class="line"><a id="l00125" name="l00125"></a><span class="lineno"><a class="line" href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f"> 125</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#ga50aa98754b412bb57c083f6e3e95061f">flatten</a>(</div>
|
|
<div class="line"><a id="l00126" name="l00126"></a><span class="lineno"> 126</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="l00127" name="l00127"></a><span class="lineno"> 127</span> <span class="keywordtype">int</span> start_axis,</div>
|
|
<div class="line"><a id="l00128" name="l00128"></a><span class="lineno"> 128</span> <span class="keywordtype">int</span> end_axis = -1,</div>
|
|
<div class="line"><a id="l00129" name="l00129"></a><span class="lineno"> 129</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00130" name="l00130"></a><span class="lineno"> 130</span> </div>
|
|
<div class="line"><a id="l00132" name="l00132"></a><span class="lineno"><a class="line" href="group__ops.html#gaa6adbc9c86f0ab27d8810a02e9e719fd"> 132</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#ga50aa98754b412bb57c083f6e3e95061f">flatten</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="l00133" name="l00133"></a><span class="lineno"> 133</span> </div>
|
|
<div class="line"><a id="l00135" name="l00135"></a><span class="lineno"><a class="line" href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576"> 135</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#ga710daa7ec721bd4d3f326082cb195576">squeeze</a>(</div>
|
|
<div class="line"><a id="l00136" name="l00136"></a><span class="lineno"> 136</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="l00137" name="l00137"></a><span class="lineno"> 137</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00138" name="l00138"></a><span class="lineno"> 138</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00139" name="l00139"></a><span class="lineno"> 139</span> </div>
|
|
<div class="foldopen" id="foldopen00141" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00141" name="l00141"></a><span class="lineno"><a class="line" href="group__ops.html#ga700dd51b77379a3d2260a55783e8ebf3"> 141</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#ga710daa7ec721bd4d3f326082cb195576">squeeze</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="l00142" name="l00142"></a><span class="lineno"> 142</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576">squeeze</a>(a, std::vector<int>{axis}, s);</div>
|
|
<div class="line"><a id="l00143" name="l00143"></a><span class="lineno"> 143</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00144" name="l00144"></a><span class="lineno"> 144</span> </div>
|
|
<div class="line"><a id="l00146" name="l00146"></a><span class="lineno"><a class="line" href="group__ops.html#ga58bad3c61fd85b95927a987ba1cf5dad"> 146</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#ga710daa7ec721bd4d3f326082cb195576">squeeze</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="l00147" name="l00147"></a><span class="lineno"> 147</span> </div>
|
|
<div class="line"><a id="l00149" name="l00149"></a><span class="lineno"><a class="line" href="group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133"> 149</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>(</div>
|
|
<div class="line"><a id="l00150" name="l00150"></a><span class="lineno"> 150</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="l00151" name="l00151"></a><span class="lineno"> 151</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00152" name="l00152"></a><span class="lineno"> 152</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00153" name="l00153"></a><span class="lineno"> 153</span> </div>
|
|
<div class="line"><a id="l00155" name="l00155"></a><span class="lineno"><a class="line" href="group__ops.html#ga7a80adb4a5a36d18b5f234d4b034950a"> 155</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="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#gad66135407dbb41b3c5d2cdfd51226c21"> 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#gad66135407dbb41b3c5d2cdfd51226c21">slice</a>(</div>
|
|
<div class="line"><a id="l00159" name="l00159"></a><span class="lineno"> 159</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="l00160" name="l00160"></a><span class="lineno"> 160</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00161" name="l00161"></a><span class="lineno"> 161</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00162" name="l00162"></a><span class="lineno"> 162</span> std::vector<int> strides,</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#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00164" name="l00164"></a><span class="lineno"> 164</span> </div>
|
|
<div class="line"><a id="l00166" name="l00166"></a><span class="lineno"><a class="line" href="group__ops.html#ga90b243e25b5fbbd7a9732f7ac48f6403"> 166</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#gad66135407dbb41b3c5d2cdfd51226c21">slice</a>(</div>
|
|
<div class="line"><a id="l00167" name="l00167"></a><span class="lineno"> 167</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="l00168" name="l00168"></a><span class="lineno"> 168</span> <span class="keyword">const</span> std::vector<int>& start,</div>
|
|
<div class="line"><a id="l00169" name="l00169"></a><span class="lineno"> 169</span> <span class="keyword">const</span> std::vector<int>& stop,</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#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00171" name="l00171"></a><span class="lineno"> 171</span> </div>
|
|
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno"><a class="line" href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708"> 173</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#ga3875660e4ce2c8add8bfcf8144078708">slice_update</a>(</div>
|
|
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno"> 174</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& src,</div>
|
|
<div class="line"><a id="l00175" name="l00175"></a><span class="lineno"> 175</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& update,</div>
|
|
<div class="line"><a id="l00176" name="l00176"></a><span class="lineno"> 176</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno"> 177</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00178" name="l00178"></a><span class="lineno"> 178</span> std::vector<int> strides,</div>
|
|
<div class="line"><a id="l00179" name="l00179"></a><span class="lineno"> 179</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00180" name="l00180"></a><span class="lineno"> 180</span> </div>
|
|
<div class="line"><a id="l00182" name="l00182"></a><span class="lineno"><a class="line" href="group__ops.html#ga03ffbbb4d989a463ef43f41ebf7eabef"> 182</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#ga3875660e4ce2c8add8bfcf8144078708">slice_update</a>(</div>
|
|
<div class="line"><a id="l00183" name="l00183"></a><span class="lineno"> 183</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& src,</div>
|
|
<div class="line"><a id="l00184" name="l00184"></a><span class="lineno"> 184</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& update,</div>
|
|
<div class="line"><a id="l00185" name="l00185"></a><span class="lineno"> 185</span> std::vector<int> start,</div>
|
|
<div class="line"><a id="l00186" name="l00186"></a><span class="lineno"> 186</span> std::vector<int> stop,</div>
|
|
<div class="line"><a id="l00187" name="l00187"></a><span class="lineno"> 187</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00188" name="l00188"></a><span class="lineno"> 188</span> </div>
|
|
<div class="line"><a id="l00190" name="l00190"></a><span class="lineno"> 190</span>std::vector<array></div>
|
|
<div class="line"><a id="l00191" name="l00191"></a><span class="lineno"><a class="line" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d"> 191</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="keywordtype">int</span> num_splits, <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="l00192" name="l00192"></a><span class="lineno"><a class="line" href="group__ops.html#ga56882d24e5fde59c266774624c892d41"> 192</a></span>std::vector<array> <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="keywordtype">int</span> num_splits, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00193" name="l00193"></a><span class="lineno"><a class="line" href="group__ops.html#ga2cfcb1a53924882e30476c9016c5de74"> 193</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga7534290bceab5fb3831a05d67bebce7d">split</a>(</div>
|
|
<div class="line"><a id="l00194" name="l00194"></a><span class="lineno"> 194</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="l00195" name="l00195"></a><span class="lineno"> 195</span> <span class="keyword">const</span> std::vector<int>& indices,</div>
|
|
<div class="line"><a id="l00196" name="l00196"></a><span class="lineno"> 196</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00197" name="l00197"></a><span class="lineno"> 197</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00198" name="l00198"></a><span class="lineno"> 198</span>std::vector<array></div>
|
|
<div class="line"><a id="l00199" name="l00199"></a><span class="lineno"><a class="line" href="group__ops.html#gac324dfa3e26d3a14a35ab7962e36f0e1"> 199</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> std::vector<int>& indices, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00200" name="l00200"></a><span class="lineno"> 200</span> </div>
|
|
<div class="line"><a id="l00202" name="l00202"></a><span class="lineno"><a class="line" href="group__ops.html#ga577c911618575314de63d1060656a26e"> 202</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga577c911618575314de63d1060656a26e">meshgrid</a>(</div>
|
|
<div class="line"><a id="l00203" name="l00203"></a><span class="lineno"> 203</span> <span class="keyword">const</span> std::vector<array>& arrays,</div>
|
|
<div class="line"><a id="l00204" name="l00204"></a><span class="lineno"> 204</span> <span class="keywordtype">bool</span> sparse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00205" name="l00205"></a><span class="lineno"> 205</span> std::string indexing = <span class="stringliteral">"xy"</span>,</div>
|
|
<div class="line"><a id="l00206" name="l00206"></a><span class="lineno"> 206</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00207" name="l00207"></a><span class="lineno"> 207</span> </div>
|
|
<div class="line"><a id="l00211" name="l00211"></a><span class="lineno"><a class="line" href="group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8"> 211</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>
|
|
<div class="line"><a id="l00212" name="l00212"></a><span class="lineno"> 212</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="l00213" name="l00213"></a><span class="lineno"> 213</span> <span class="keyword">const</span> std::optional<array>& a_min = std::nullopt,</div>
|
|
<div class="line"><a id="l00214" name="l00214"></a><span class="lineno"> 214</span> <span class="keyword">const</span> std::optional<array>& a_max = std::nullopt,</div>
|
|
<div class="line"><a id="l00215" name="l00215"></a><span class="lineno"> 215</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00216" name="l00216"></a><span class="lineno"> 216</span> </div>
|
|
<div class="line"><a id="l00218" name="l00218"></a><span class="lineno"><a class="line" href="group__ops.html#gabdc36fa65697d0361c8d67495de77129"> 218</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#gabdc36fa65697d0361c8d67495de77129">concatenate</a>(</div>
|
|
<div class="line"><a id="l00219" name="l00219"></a><span class="lineno"> 219</span> <span class="keyword">const</span> std::vector<array>& arrays,</div>
|
|
<div class="line"><a id="l00220" name="l00220"></a><span class="lineno"> 220</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00221" name="l00221"></a><span class="lineno"> 221</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00222" name="l00222"></a><span class="lineno"><a class="line" href="group__ops.html#gaa95c34ca3a8877f2c50cb60e7fa312b8"> 222</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#gabdc36fa65697d0361c8d67495de77129">concatenate</a>(<span class="keyword">const</span> std::vector<array>& arrays, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00223" name="l00223"></a><span class="lineno"> 223</span> </div>
|
|
<div class="line"><a id="l00225" name="l00225"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032"> 225</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#gaf8f2ec2b98a4b59eca73d7471df6e032">stack</a>(<span class="keyword">const</span> std::vector<array>& arrays, <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="l00226" name="l00226"></a><span class="lineno"><a class="line" href="group__ops.html#ga82216209dce901296fc737fe8efa5c94"> 226</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#gaf8f2ec2b98a4b59eca73d7471df6e032">stack</a>(<span class="keyword">const</span> std::vector<array>& arrays, <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="l00229" name="l00229"></a><span class="lineno"><a class="line" href="group__ops.html#gab49e3a687e826554ed1574186e8ae974"> 229</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#gab49e3a687e826554ed1574186e8ae974">repeat</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& arr, <span class="keywordtype">int</span> repeats, <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="l00230" name="l00230"></a><span class="lineno"><a class="line" href="group__ops.html#ga4f75f5d5db999f02f43ecbc6dccf3ba6"> 230</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#gab49e3a687e826554ed1574186e8ae974">repeat</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& arr, <span class="keywordtype">int</span> repeats, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00231" name="l00231"></a><span class="lineno"> 231</span> </div>
|
|
<div class="line"><a id="l00232" name="l00232"></a><span class="lineno"><a class="line" href="group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361"> 232</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#gab105a57b9a4d84496fe1e4d60e13d361">tile</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& arr, std::vector<int> reps, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00233" name="l00233"></a><span class="lineno"> 233</span> </div>
|
|
<div class="line"><a id="l00235" name="l00235"></a><span class="lineno"><a class="line" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b"> 235</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, std::vector<int> axes, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00236" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00236" name="l00236"></a><span class="lineno"><a class="line" href="group__ops.html#ga260ac332956f3a6bf1dfdb9095c84dc5"> 236</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#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(</div>
|
|
<div class="line"><a id="l00237" name="l00237"></a><span class="lineno"> 237</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="l00238" name="l00238"></a><span class="lineno"> 238</span> std::initializer_list<int> axes,</div>
|
|
<div class="line"><a id="l00239" name="l00239"></a><span class="lineno"> 239</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00240" name="l00240"></a><span class="lineno"> 240</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">transpose</a>(a, std::vector<int>(axes), s);</div>
|
|
<div class="line"><a id="l00241" name="l00241"></a><span class="lineno"> 241</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00242" name="l00242"></a><span class="lineno"> 242</span> </div>
|
|
<div class="line"><a id="l00244" name="l00244"></a><span class="lineno"><a class="line" href="group__ops.html#gabc46eed81ab6c6247903e4ec0c4ec1fb"> 244</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#gabc46eed81ab6c6247903e4ec0c4ec1fb">swapaxes</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> axis1, <span class="keywordtype">int</span> axis2, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00245" name="l00245"></a><span class="lineno"> 245</span> </div>
|
|
<div class="line"><a id="l00247" name="l00247"></a><span class="lineno"><a class="line" href="group__ops.html#ga24067d10a842db2c9d509ea48135a2c3"> 247</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#ga24067d10a842db2c9d509ea48135a2c3">moveaxis</a>(</div>
|
|
<div class="line"><a id="l00248" name="l00248"></a><span class="lineno"> 248</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="l00249" name="l00249"></a><span class="lineno"> 249</span> <span class="keywordtype">int</span> source,</div>
|
|
<div class="line"><a id="l00250" name="l00250"></a><span class="lineno"> 250</span> <span class="keywordtype">int</span> destination,</div>
|
|
<div class="line"><a id="l00251" name="l00251"></a><span class="lineno"> 251</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00252" name="l00252"></a><span class="lineno"> 252</span> </div>
|
|
<div class="line"><a id="l00254" name="l00254"></a><span class="lineno"><a class="line" href="group__ops.html#ga4fd04d6472ae7cdb94a174edfe338ea3"> 254</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#ga4fd04d6472ae7cdb94a174edfe338ea3">pad</a>(</div>
|
|
<div class="line"><a id="l00255" name="l00255"></a><span class="lineno"> 255</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="l00256" name="l00256"></a><span class="lineno"> 256</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00257" name="l00257"></a><span class="lineno"> 257</span> <span class="keyword">const</span> std::vector<int>& low_pad_size,</div>
|
|
<div class="line"><a id="l00258" name="l00258"></a><span class="lineno"> 258</span> <span class="keyword">const</span> std::vector<int>& high_pad_size,</div>
|
|
<div class="line"><a id="l00259" name="l00259"></a><span class="lineno"> 259</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="l00260" name="l00260"></a><span class="lineno"> 260</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00261" name="l00261"></a><span class="lineno"> 261</span> </div>
|
|
<div class="line"><a id="l00263" name="l00263"></a><span class="lineno"><a class="line" href="group__ops.html#gaca4e00d5e4ed9e9f29d56a17f19c2e31"> 263</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#ga4fd04d6472ae7cdb94a174edfe338ea3">pad</a>(</div>
|
|
<div class="line"><a id="l00264" name="l00264"></a><span class="lineno"> 264</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="l00265" name="l00265"></a><span class="lineno"> 265</span> <span class="keyword">const</span> std::vector<std::pair<int, int>>& pad_width,</div>
|
|
<div class="line"><a id="l00266" name="l00266"></a><span class="lineno"> 266</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="l00267" name="l00267"></a><span class="lineno"> 267</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00268" name="l00268"></a><span class="lineno"><a class="line" href="group__ops.html#ga608164c4f5b4ac2fba374219285d90bc"> 268</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#ga4fd04d6472ae7cdb94a174edfe338ea3">pad</a>(</div>
|
|
<div class="line"><a id="l00269" name="l00269"></a><span class="lineno"> 269</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="l00270" name="l00270"></a><span class="lineno"> 270</span> <span class="keyword">const</span> std::pair<int, int>& pad_width,</div>
|
|
<div class="line"><a id="l00271" name="l00271"></a><span class="lineno"> 271</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="l00272" name="l00272"></a><span class="lineno"> 272</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00273" name="l00273"></a><span class="lineno"><a class="line" href="group__ops.html#ga2e57e26c4d7f3d18802c164656cf915c"> 273</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#ga4fd04d6472ae7cdb94a174edfe338ea3">pad</a>(</div>
|
|
<div class="line"><a id="l00274" name="l00274"></a><span class="lineno"> 274</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="l00275" name="l00275"></a><span class="lineno"> 275</span> <span class="keywordtype">int</span> pad_width,</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> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00278" name="l00278"></a><span class="lineno"> 278</span> </div>
|
|
<div class="line"><a id="l00280" name="l00280"></a><span class="lineno"><a class="line" href="group__ops.html#ga68da0176fefbe0c0096783c6fd926c6a"> 280</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>
|
|
<div class="line"><a id="l00281" name="l00281"></a><span class="lineno"> 281</span> </div>
|
|
<div class="line"><a id="l00283" name="l00283"></a><span class="lineno"><a class="line" href="group__ops.html#gad256e86cc1a6e6b3832e392baa90318d"> 283</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#gad256e86cc1a6e6b3832e392baa90318d">broadcast_to</a>(</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>& a,</div>
|
|
<div class="line"><a id="l00285" name="l00285"></a><span class="lineno"> 285</span> <span class="keyword">const</span> std::vector<int>& shape,</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"> 287</span> </div>
|
|
<div class="line"><a id="l00289" name="l00289"></a><span class="lineno"><a class="line" href="group__ops.html#gab783890428b596f715dc7dd2057eae99"> 289</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gab783890428b596f715dc7dd2057eae99">broadcast_arrays</a>(</div>
|
|
<div class="line"><a id="l00290" name="l00290"></a><span class="lineno"> 290</span> <span class="keyword">const</span> std::vector<array>& inputs,</div>
|
|
<div class="line"><a id="l00291" name="l00291"></a><span class="lineno"> 291</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00292" name="l00292"></a><span class="lineno"> 292</span> </div>
|
|
<div class="line"><a id="l00294" name="l00294"></a><span class="lineno"><a class="line" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde"> 294</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#ga33638dc3a9972dd02be12d0eb85f9bde">equal</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="foldopen" id="foldopen00295" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00295" name="l00295"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30cf69f3d22f65615f5e1696dd5703f"> 295</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#a937503d72b66c661bf3f5fdcd98ef97c">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="l00296" name="l00296"></a><span class="lineno"> 296</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00297" name="l00297"></a><span class="lineno"> 297</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00298" name="l00298"></a><span class="lineno"> 298</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00299" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00299" name="l00299"></a><span class="lineno"><a class="line" href="group__ops.html#gaf115782d009ac2a547fcca395c9ec797"> 299</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#a937503d72b66c661bf3f5fdcd98ef97c">operator==</a>(T 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="l00300" name="l00300"></a><span class="lineno"> 300</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00301" name="l00301"></a><span class="lineno"> 301</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00302" name="l00302"></a><span class="lineno"> 302</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00303" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00303" name="l00303"></a><span class="lineno"><a class="line" href="group__ops.html#ga3ad3ed7aece2650943a35082dbe3a0a5"> 303</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#a937503d72b66c661bf3f5fdcd98ef97c">operator==</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00304" name="l00304"></a><span class="lineno"> 304</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00305" name="l00305"></a><span class="lineno"> 305</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00306" name="l00306"></a><span class="lineno"> 306</span> </div>
|
|
<div class="line"><a id="l00308" name="l00308"></a><span class="lineno"><a class="line" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103"> 308</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#ga28f22c5d2c399eee53be7b3facc11103">not_equal</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="foldopen" id="foldopen00309" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00309" name="l00309"></a><span class="lineno"><a class="line" href="group__ops.html#ga0ac483d85f23252ca8757e9926d5a3c5"> 309</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#a94d00a1b7f8a4717ab3f26f45e4da655">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="l00310" name="l00310"></a><span class="lineno"> 310</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00311" name="l00311"></a><span class="lineno"> 311</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00312" name="l00312"></a><span class="lineno"> 312</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00313" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00313" name="l00313"></a><span class="lineno"><a class="line" href="group__ops.html#ga3fecba9f3cb9a19afd8ca492cf509ce0"> 313</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#a94d00a1b7f8a4717ab3f26f45e4da655">operator!=</a>(T 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="l00314" name="l00314"></a><span class="lineno"> 314</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00315" name="l00315"></a><span class="lineno"> 315</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00316" name="l00316"></a><span class="lineno"> 316</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00317" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00317" name="l00317"></a><span class="lineno"><a class="line" href="group__ops.html#gaebbf1cfde388c7480159a03c92c9a385"> 317</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#a94d00a1b7f8a4717ab3f26f45e4da655">operator!=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00318" name="l00318"></a><span class="lineno"> 318</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga28f22c5d2c399eee53be7b3facc11103">not_equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00319" name="l00319"></a><span class="lineno"> 319</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00320" name="l00320"></a><span class="lineno"> 320</span> </div>
|
|
<div class="line"><a id="l00322" name="l00322"></a><span class="lineno"><a class="line" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d"> 322</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#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</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="foldopen" id="foldopen00323" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00323" name="l00323"></a><span class="lineno"><a class="line" href="group__ops.html#ga74fd2777adef10e6fe628a9cdadb01cb"> 323</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#ga74fd2777adef10e6fe628a9cdadb01cb">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="l00324" name="l00324"></a><span class="lineno"> 324</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(a, b);</div>
|
|
<div class="line"><a id="l00325" name="l00325"></a><span class="lineno"> 325</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00326" name="l00326"></a><span class="lineno"> 326</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00327" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00327" name="l00327"></a><span class="lineno"><a class="line" href="group__ops.html#ga32e106e794e2c32e4e7decee2df2477f"> 327</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#ga74fd2777adef10e6fe628a9cdadb01cb">operator></a>(T 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="l00328" name="l00328"></a><span class="lineno"> 328</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00329" name="l00329"></a><span class="lineno"> 329</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00330" name="l00330"></a><span class="lineno"> 330</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00331" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00331" name="l00331"></a><span class="lineno"><a class="line" href="group__ops.html#ga96552b90e89923c5d2064cc427775ec5"> 331</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#ga74fd2777adef10e6fe628a9cdadb01cb">operator></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00332" name="l00332"></a><span class="lineno"> 332</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">greater</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00333" name="l00333"></a><span class="lineno"> 333</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00334" name="l00334"></a><span class="lineno"> 334</span> </div>
|
|
<div class="line"><a id="l00336" name="l00336"></a><span class="lineno"><a class="line" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430"> 336</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#ga7153071bcfff6faad21332163fb9a430">greater_equal</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="foldopen" id="foldopen00337" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00337" name="l00337"></a><span class="lineno"><a class="line" href="group__ops.html#ga3a41895f25ed083a36994d95fa102546"> 337</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#ga3a41895f25ed083a36994d95fa102546">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="l00338" name="l00338"></a><span class="lineno"> 338</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00339" name="l00339"></a><span class="lineno"> 339</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00340" name="l00340"></a><span class="lineno"> 340</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00341" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00341" name="l00341"></a><span class="lineno"><a class="line" href="group__ops.html#gaf509f2cb3b18963232f20d6c3bd229b2"> 341</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#ga3a41895f25ed083a36994d95fa102546">operator>=</a>(T 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="l00342" name="l00342"></a><span class="lineno"> 342</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00343" name="l00343"></a><span class="lineno"> 343</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00344" name="l00344"></a><span class="lineno"> 344</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00345" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00345" name="l00345"></a><span class="lineno"><a class="line" href="group__ops.html#gafa0eb25d5978674bfc9e59d4145ec590"> 345</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#ga3a41895f25ed083a36994d95fa102546">operator>=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00346" name="l00346"></a><span class="lineno"> 346</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">greater_equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00347" name="l00347"></a><span class="lineno"> 347</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00348" name="l00348"></a><span class="lineno"> 348</span> </div>
|
|
<div class="line"><a id="l00350" name="l00350"></a><span class="lineno"><a class="line" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a"> 350</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#ga9142b8d717699a8abfa2a7398891ff8a">less</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="foldopen" id="foldopen00351" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00351" name="l00351"></a><span class="lineno"><a class="line" href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3"> 351</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#gaee41e2b8f61d563200ff03575ac1d6c3">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="l00352" name="l00352"></a><span class="lineno"> 352</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(a, b);</div>
|
|
<div class="line"><a id="l00353" name="l00353"></a><span class="lineno"> 353</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00354" name="l00354"></a><span class="lineno"> 354</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00355" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00355" name="l00355"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ef8ea11cf15ce628c54201fa42748ef"> 355</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#gaee41e2b8f61d563200ff03575ac1d6c3">operator<</a>(T 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="l00356" name="l00356"></a><span class="lineno"> 356</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00357" name="l00357"></a><span class="lineno"> 357</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00358" name="l00358"></a><span class="lineno"> 358</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00359" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00359" name="l00359"></a><span class="lineno"><a class="line" href="group__ops.html#ga95e72226dc7a79c40b3d16f990922050"> 359</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#gaee41e2b8f61d563200ff03575ac1d6c3">operator<</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00360" name="l00360"></a><span class="lineno"> 360</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">less</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00361" name="l00361"></a><span class="lineno"> 361</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00362" name="l00362"></a><span class="lineno"> 362</span> </div>
|
|
<div class="line"><a id="l00364" name="l00364"></a><span class="lineno"><a class="line" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09"> 364</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#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</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="foldopen" id="foldopen00365" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00365" name="l00365"></a><span class="lineno"><a class="line" href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6"> 365</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#ga4c8b8a1632944acaae50f0de6c23ece6">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="l00366" name="l00366"></a><span class="lineno"> 366</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(a, b);</div>
|
|
<div class="line"><a id="l00367" name="l00367"></a><span class="lineno"> 367</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00368" name="l00368"></a><span class="lineno"> 368</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00369" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00369" name="l00369"></a><span class="lineno"><a class="line" href="group__ops.html#ga150a9be467c9f91482a6d6fc13504bc4"> 369</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#ga4c8b8a1632944acaae50f0de6c23ece6">operator<=</a>(T 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="l00370" name="l00370"></a><span class="lineno"> 370</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00371" name="l00371"></a><span class="lineno"> 371</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00372" name="l00372"></a><span class="lineno"> 372</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00373" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00373" name="l00373"></a><span class="lineno"><a class="line" href="group__ops.html#ga624eeccef0cc4b130e1325abfea057cb"> 373</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#ga4c8b8a1632944acaae50f0de6c23ece6">operator<=</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00374" name="l00374"></a><span class="lineno"> 374</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">less_equal</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00375" name="l00375"></a><span class="lineno"> 375</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00376" name="l00376"></a><span class="lineno"> 376</span> </div>
|
|
<div class="line"><a id="l00378" name="l00378"></a><span class="lineno"><a class="line" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645"> 378</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#ga8f3059336ee0c87207b1f8c6ab312645">array_equal</a>(</div>
|
|
<div class="line"><a id="l00379" name="l00379"></a><span class="lineno"> 379</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="l00380" name="l00380"></a><span class="lineno"> 380</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="l00381" name="l00381"></a><span class="lineno"> 381</span> <span class="keywordtype">bool</span> equal_nan,</div>
|
|
<div class="line"><a id="l00382" name="l00382"></a><span class="lineno"> 382</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00383" name="l00383"></a><span class="lineno"> 383</span><span class="keyword">inline</span> array</div>
|
|
<div class="foldopen" id="foldopen00384" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00384" name="l00384"></a><span class="lineno"><a class="line" href="group__ops.html#gaf79cf0271ca0105d7b14295a90d0ed14"> 384</a></span><a class="code hl_function" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">array_equal</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="l00385" name="l00385"></a><span class="lineno"> 385</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">array_equal</a>(a, b, <span class="keyword">false</span>, s);</div>
|
|
<div class="line"><a id="l00386" name="l00386"></a><span class="lineno"> 386</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00387" name="l00387"></a><span class="lineno"> 387</span> </div>
|
|
<div class="line"><a id="l00388" name="l00388"></a><span class="lineno"><a class="line" href="group__ops.html#ga175592792471b0ffb45196dca4711ba6"> 388</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#ga175592792471b0ffb45196dca4711ba6">isnan</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="l00389" name="l00389"></a><span class="lineno"> 389</span> </div>
|
|
<div class="line"><a id="l00390" name="l00390"></a><span class="lineno"><a class="line" href="group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844"> 390</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#ga8fc238d5e5d1153e69da8b36015d9844">isinf</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="l00391" name="l00391"></a><span class="lineno"> 391</span> </div>
|
|
<div class="line"><a id="l00392" name="l00392"></a><span class="lineno"><a class="line" href="group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993"> 392</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#gad80f7c4a58c12b6cb30a8b9a73008993">isposinf</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="l00393" name="l00393"></a><span class="lineno"> 393</span> </div>
|
|
<div class="line"><a id="l00394" name="l00394"></a><span class="lineno"><a class="line" href="group__ops.html#ga1940523da381ed7be50656a3bc465ff3"> 394</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#ga1940523da381ed7be50656a3bc465ff3">isneginf</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="l00395" name="l00395"></a><span class="lineno"> 395</span> </div>
|
|
<div class="line"><a id="l00397" name="l00397"></a><span class="lineno"><a class="line" href="group__ops.html#ga8a2056f8c9bb30914c40bcf509386491"> 397</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#ga8a2056f8c9bb30914c40bcf509386491">where</a>(</div>
|
|
<div class="line"><a id="l00398" name="l00398"></a><span class="lineno"> 398</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& condition,</div>
|
|
<div class="line"><a id="l00399" name="l00399"></a><span class="lineno"> 399</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="l00400" name="l00400"></a><span class="lineno"> 400</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& y,</div>
|
|
<div class="line"><a id="l00401" name="l00401"></a><span class="lineno"> 401</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00402" name="l00402"></a><span class="lineno"> 402</span> </div>
|
|
<div class="line"><a id="l00404" name="l00404"></a><span class="lineno"><a class="line" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68"> 404</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#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00405" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00405" name="l00405"></a><span class="lineno"><a class="line" href="group__ops.html#ga3689e12e8f42dadb4cbe2b07dc4099f4"> 405</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#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</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="l00406" name="l00406"></a><span class="lineno"> 406</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00407" name="l00407"></a><span class="lineno"> 407</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00408" name="l00408"></a><span class="lineno"> 408</span> </div>
|
|
<div class="line"><a id="l00410" name="l00410"></a><span class="lineno"><a class="line" href="group__ops.html#gaf0cd4257de7542daf9faf5e605e31020"> 410</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#gaf0cd4257de7542daf9faf5e605e31020">allclose</a>(</div>
|
|
<div class="line"><a id="l00411" name="l00411"></a><span class="lineno"> 411</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="l00412" name="l00412"></a><span class="lineno"> 412</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="l00413" name="l00413"></a><span class="lineno"> 413</span> <span class="keywordtype">double</span> rtol = 1e-5,</div>
|
|
<div class="line"><a id="l00414" name="l00414"></a><span class="lineno"> 414</span> <span class="keywordtype">double</span> atol = 1e-8,</div>
|
|
<div class="line"><a id="l00415" name="l00415"></a><span class="lineno"> 415</span> <span class="keywordtype">bool</span> equal_nan = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00416" name="l00416"></a><span class="lineno"> 416</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00417" name="l00417"></a><span class="lineno"> 417</span> </div>
|
|
<div class="line"><a id="l00420" name="l00420"></a><span class="lineno"><a class="line" href="group__ops.html#ga51eac95c04400921c54716de14b52491"> 420</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#ga51eac95c04400921c54716de14b52491">isclose</a>(</div>
|
|
<div class="line"><a id="l00421" name="l00421"></a><span class="lineno"> 421</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="l00422" name="l00422"></a><span class="lineno"> 422</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="l00423" name="l00423"></a><span class="lineno"> 423</span> <span class="keywordtype">double</span> rtol = 1e-5,</div>
|
|
<div class="line"><a id="l00424" name="l00424"></a><span class="lineno"> 424</span> <span class="keywordtype">double</span> atol = 1e-8,</div>
|
|
<div class="line"><a id="l00425" name="l00425"></a><span class="lineno"> 425</span> <span class="keywordtype">bool</span> equal_nan = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00426" name="l00426"></a><span class="lineno"> 426</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00427" name="l00427"></a><span class="lineno"> 427</span> </div>
|
|
<div class="line"><a id="l00432" name="l00432"></a><span class="lineno"><a class="line" href="group__ops.html#gac0919c6ba53aea35a7683dea7e9a9a59"> 432</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#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(</div>
|
|
<div class="line"><a id="l00433" name="l00433"></a><span class="lineno"> 433</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="l00434" name="l00434"></a><span class="lineno"> 434</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00435" name="l00435"></a><span class="lineno"> 435</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00436" name="l00436"></a><span class="lineno"> 436</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00437" name="l00437"></a><span class="lineno"> 437</span> </div>
|
|
<div class="line"><a id="l00442" name="l00442"></a><span class="lineno"><a class="line" href="group__ops.html#gae2d5fcc5b62d673cca76c08b7b4afbbc"> 442</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#ga3b1b90ef1275ca17655b6d7f25d3ee68">all</a>(</div>
|
|
<div class="line"><a id="l00443" name="l00443"></a><span class="lineno"> 443</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="l00444" name="l00444"></a><span class="lineno"> 444</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00445" name="l00445"></a><span class="lineno"> 445</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00446" name="l00446"></a><span class="lineno"> 446</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00447" name="l00447"></a><span class="lineno"> 447</span> </div>
|
|
<div class="line"><a id="l00449" name="l00449"></a><span class="lineno"><a class="line" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f"> 449</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#ga8598dd718fb05cb28535e250372d4e6f">any</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00450" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00450" name="l00450"></a><span class="lineno"><a class="line" href="group__ops.html#gad37df97f253a963bece124198dbaf9ba"> 450</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#ga8598dd718fb05cb28535e250372d4e6f">any</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="l00451" name="l00451"></a><span class="lineno"> 451</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">any</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00452" name="l00452"></a><span class="lineno"> 452</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00453" name="l00453"></a><span class="lineno"> 453</span> </div>
|
|
<div class="line"><a id="l00458" name="l00458"></a><span class="lineno"><a class="line" href="group__ops.html#gaf240618fc8b06debf5f56e97e84f18ef"> 458</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#ga8598dd718fb05cb28535e250372d4e6f">any</a>(</div>
|
|
<div class="line"><a id="l00459" name="l00459"></a><span class="lineno"> 459</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="l00460" name="l00460"></a><span class="lineno"> 460</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00461" name="l00461"></a><span class="lineno"> 461</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00462" name="l00462"></a><span class="lineno"> 462</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00463" name="l00463"></a><span class="lineno"> 463</span> </div>
|
|
<div class="line"><a id="l00468" name="l00468"></a><span class="lineno"><a class="line" href="group__ops.html#gab1d56277d468a55227f4dad6bc2fc1ce"> 468</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#ga8598dd718fb05cb28535e250372d4e6f">any</a>(</div>
|
|
<div class="line"><a id="l00469" name="l00469"></a><span class="lineno"> 469</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="l00470" name="l00470"></a><span class="lineno"> 470</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00471" name="l00471"></a><span class="lineno"> 471</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00472" name="l00472"></a><span class="lineno"> 472</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00473" name="l00473"></a><span class="lineno"> 473</span> </div>
|
|
<div class="line"><a id="l00475" name="l00475"></a><span class="lineno"><a class="line" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6"> 475</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#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00476" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00476" name="l00476"></a><span class="lineno"><a class="line" href="group__ops.html#ga3627754d7868487bdab1bd83f05d9c81"> 476</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#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</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="l00477" name="l00477"></a><span class="lineno"> 477</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00478" name="l00478"></a><span class="lineno"> 478</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00479" name="l00479"></a><span class="lineno"> 479</span> </div>
|
|
<div class="line"><a id="l00481" name="l00481"></a><span class="lineno"><a class="line" href="group__ops.html#gaccd0a6be2c5b5128fdc2d87b5c8e67f4"> 481</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#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(</div>
|
|
<div class="line"><a id="l00482" name="l00482"></a><span class="lineno"> 482</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="l00483" name="l00483"></a><span class="lineno"> 483</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00484" name="l00484"></a><span class="lineno"> 484</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00485" name="l00485"></a><span class="lineno"> 485</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00486" name="l00486"></a><span class="lineno"> 486</span> </div>
|
|
<div class="line"><a id="l00488" name="l00488"></a><span class="lineno"><a class="line" href="group__ops.html#gafcd39b0bf39a56c26a967981c7ab8a8d"> 488</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#gade905ee92eb6ab7edfc312aeddfbaeb6">sum</a>(</div>
|
|
<div class="line"><a id="l00489" name="l00489"></a><span class="lineno"> 489</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="l00490" name="l00490"></a><span class="lineno"> 490</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00491" name="l00491"></a><span class="lineno"> 491</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00492" name="l00492"></a><span class="lineno"> 492</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00493" name="l00493"></a><span class="lineno"> 493</span> </div>
|
|
<div class="line"><a id="l00495" name="l00495"></a><span class="lineno"><a class="line" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef"> 495</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#gade46e768fd46b8b640eb16f26abeecef">mean</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00496" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00496" name="l00496"></a><span class="lineno"><a class="line" href="group__ops.html#ga52b59fdd8e8430538e564f5bbcfa31e6"> 496</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#gade46e768fd46b8b640eb16f26abeecef">mean</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="l00497" name="l00497"></a><span class="lineno"> 497</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mean</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00498" name="l00498"></a><span class="lineno"> 498</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00499" name="l00499"></a><span class="lineno"> 499</span> </div>
|
|
<div class="line"><a id="l00501" name="l00501"></a><span class="lineno"><a class="line" href="group__ops.html#ga066161f3d3e395a1d76c638cb680d444"> 501</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#gade46e768fd46b8b640eb16f26abeecef">mean</a>(</div>
|
|
<div class="line"><a id="l00502" name="l00502"></a><span class="lineno"> 502</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="l00503" name="l00503"></a><span class="lineno"> 503</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00504" name="l00504"></a><span class="lineno"> 504</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00505" name="l00505"></a><span class="lineno"> 505</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00506" name="l00506"></a><span class="lineno"> 506</span> </div>
|
|
<div class="line"><a id="l00508" name="l00508"></a><span class="lineno"><a class="line" href="group__ops.html#ga45fba73eab0e3b6e128ed3ce2f43a5da"> 508</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#gade46e768fd46b8b640eb16f26abeecef">mean</a>(</div>
|
|
<div class="line"><a id="l00509" name="l00509"></a><span class="lineno"> 509</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="l00510" name="l00510"></a><span class="lineno"> 510</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00511" name="l00511"></a><span class="lineno"> 511</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00512" name="l00512"></a><span class="lineno"> 512</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00513" name="l00513"></a><span class="lineno"> 513</span> </div>
|
|
<div class="line"><a id="l00515" name="l00515"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9"> 515</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#ga7e133df686439588a8cd1fb10ce0c6e9">var</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> keepdims, <span class="keywordtype">int</span> ddof = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00516" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00516" name="l00516"></a><span class="lineno"><a class="line" href="group__ops.html#ga7d7b38d118fa2613214078ef0f7d5a42"> 516</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#ga7e133df686439588a8cd1fb10ce0c6e9">var</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="l00517" name="l00517"></a><span class="lineno"> 517</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(a, <span class="keyword">false</span>, 0, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00518" name="l00518"></a><span class="lineno"> 518</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00519" name="l00519"></a><span class="lineno"> 519</span> </div>
|
|
<div class="line"><a id="l00522" name="l00522"></a><span class="lineno"><a class="line" href="group__ops.html#ga78ddeb966cbe7a5b0aa17e1de43025f2"> 522</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#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(</div>
|
|
<div class="line"><a id="l00523" name="l00523"></a><span class="lineno"> 523</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="l00524" name="l00524"></a><span class="lineno"> 524</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00525" name="l00525"></a><span class="lineno"> 525</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00526" name="l00526"></a><span class="lineno"> 526</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00527" name="l00527"></a><span class="lineno"> 527</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00528" name="l00528"></a><span class="lineno"> 528</span> </div>
|
|
<div class="line"><a id="l00531" name="l00531"></a><span class="lineno"><a class="line" href="group__ops.html#ga4fbf3e3f98f2e4956faf87af320aa9d0"> 531</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#ga7e133df686439588a8cd1fb10ce0c6e9">var</a>(</div>
|
|
<div class="line"><a id="l00532" name="l00532"></a><span class="lineno"> 532</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="l00533" name="l00533"></a><span class="lineno"> 533</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00534" name="l00534"></a><span class="lineno"> 534</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00535" name="l00535"></a><span class="lineno"> 535</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00536" name="l00536"></a><span class="lineno"> 536</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00537" name="l00537"></a><span class="lineno"> 537</span> </div>
|
|
<div class="line"><a id="l00539" name="l00539"></a><span class="lineno"><a class="line" href="group__ops.html#ga2a466024f8061febc0a64be557644cb0"> 539</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#ga2a466024f8061febc0a64be557644cb0">std</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> keepdims, <span class="keywordtype">int</span> ddof = 0, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00540" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00540" name="l00540"></a><span class="lineno"><a class="line" href="group__ops.html#gafdcb04d77c64405a3990078a77dd984c"> 540</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#ga2a466024f8061febc0a64be557644cb0">std</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="l00541" name="l00541"></a><span class="lineno"> 541</span> <span class="keywordflow">return</span> std(a, <span class="keyword">false</span>, 0, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00542" name="l00542"></a><span class="lineno"> 542</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00543" name="l00543"></a><span class="lineno"> 543</span> </div>
|
|
<div class="line"><a id="l00546" name="l00546"></a><span class="lineno"><a class="line" href="group__ops.html#ga7f649970bf38b987b6ef847054f3c2f8"> 546</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#ga2a466024f8061febc0a64be557644cb0">std</a>(</div>
|
|
<div class="line"><a id="l00547" name="l00547"></a><span class="lineno"> 547</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="l00548" name="l00548"></a><span class="lineno"> 548</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00549" name="l00549"></a><span class="lineno"> 549</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00550" name="l00550"></a><span class="lineno"> 550</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00551" name="l00551"></a><span class="lineno"> 551</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00552" name="l00552"></a><span class="lineno"> 552</span> </div>
|
|
<div class="line"><a id="l00555" name="l00555"></a><span class="lineno"><a class="line" href="group__ops.html#ga62721a206df671ef5797449eea97af9f"> 555</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#ga2a466024f8061febc0a64be557644cb0">std</a>(</div>
|
|
<div class="line"><a id="l00556" name="l00556"></a><span class="lineno"> 556</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="l00557" name="l00557"></a><span class="lineno"> 557</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00558" name="l00558"></a><span class="lineno"> 558</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00559" name="l00559"></a><span class="lineno"> 559</span> <span class="keywordtype">int</span> ddof = 0,</div>
|
|
<div class="line"><a id="l00560" name="l00560"></a><span class="lineno"> 560</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00561" name="l00561"></a><span class="lineno"> 561</span> </div>
|
|
<div class="line"><a id="l00563" name="l00563"></a><span class="lineno"><a class="line" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544"> 563</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#ga4a09b7241d564d92548bc2773eb1d544">prod</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00564" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00564" name="l00564"></a><span class="lineno"><a class="line" href="group__ops.html#ga61832191f3c42ea549cf04953edc3602"> 564</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#ga4a09b7241d564d92548bc2773eb1d544">prod</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="l00565" name="l00565"></a><span class="lineno"> 565</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00566" name="l00566"></a><span class="lineno"> 566</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00567" name="l00567"></a><span class="lineno"> 567</span> </div>
|
|
<div class="line"><a id="l00569" name="l00569"></a><span class="lineno"><a class="line" href="group__ops.html#ga2b3935108f641e20a70dbf63f540d970"> 569</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#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(</div>
|
|
<div class="line"><a id="l00570" name="l00570"></a><span class="lineno"> 570</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="l00571" name="l00571"></a><span class="lineno"> 571</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00572" name="l00572"></a><span class="lineno"> 572</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00573" name="l00573"></a><span class="lineno"> 573</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00574" name="l00574"></a><span class="lineno"> 574</span> </div>
|
|
<div class="line"><a id="l00576" name="l00576"></a><span class="lineno"><a class="line" href="group__ops.html#ga8a10a10b81c69996d0aca8ba401f8ff0"> 576</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#ga4a09b7241d564d92548bc2773eb1d544">prod</a>(</div>
|
|
<div class="line"><a id="l00577" name="l00577"></a><span class="lineno"> 577</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="l00578" name="l00578"></a><span class="lineno"> 578</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00579" name="l00579"></a><span class="lineno"> 579</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00580" name="l00580"></a><span class="lineno"> 580</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00581" name="l00581"></a><span class="lineno"> 581</span> </div>
|
|
<div class="line"><a id="l00583" name="l00583"></a><span class="lineno"><a class="line" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7"> 583</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#ga7fed87d96cc7741d8267f4eac83f5fe7">max</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00584" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00584" name="l00584"></a><span class="lineno"><a class="line" href="group__ops.html#ga25be91d70a5f40341db0615a0b8bfedc"> 584</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#ga7fed87d96cc7741d8267f4eac83f5fe7">max</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="l00585" name="l00585"></a><span class="lineno"> 585</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00586" name="l00586"></a><span class="lineno"> 586</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00587" name="l00587"></a><span class="lineno"> 587</span> </div>
|
|
<div class="line"><a id="l00589" name="l00589"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ca7b6b91fe2459a7d83897bf013827f"> 589</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#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(</div>
|
|
<div class="line"><a id="l00590" name="l00590"></a><span class="lineno"> 590</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="l00591" name="l00591"></a><span class="lineno"> 591</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00592" name="l00592"></a><span class="lineno"> 592</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00593" name="l00593"></a><span class="lineno"> 593</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00594" name="l00594"></a><span class="lineno"> 594</span> </div>
|
|
<div class="line"><a id="l00596" name="l00596"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b638050e03a93f2896c981bc2850a47"> 596</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#ga7fed87d96cc7741d8267f4eac83f5fe7">max</a>(</div>
|
|
<div class="line"><a id="l00597" name="l00597"></a><span class="lineno"> 597</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="l00598" name="l00598"></a><span class="lineno"> 598</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00599" name="l00599"></a><span class="lineno"> 599</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00600" name="l00600"></a><span class="lineno"> 600</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00601" name="l00601"></a><span class="lineno"> 601</span> </div>
|
|
<div class="line"><a id="l00603" name="l00603"></a><span class="lineno"><a class="line" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12"> 603</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#gab27599802617a4c8f9964ab5f4ffee12">min</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00604" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00604" name="l00604"></a><span class="lineno"><a class="line" href="group__ops.html#ga0140b91e9cdfc3fef0da8e332f65a9e8"> 604</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#gab27599802617a4c8f9964ab5f4ffee12">min</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="l00605" name="l00605"></a><span class="lineno"> 605</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gab27599802617a4c8f9964ab5f4ffee12">min</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00606" name="l00606"></a><span class="lineno"> 606</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00607" name="l00607"></a><span class="lineno"> 607</span> </div>
|
|
<div class="line"><a id="l00609" name="l00609"></a><span class="lineno"><a class="line" href="group__ops.html#ga6efb83cd46436678c8f8c4af15cc00f5"> 609</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#gab27599802617a4c8f9964ab5f4ffee12">min</a>(</div>
|
|
<div class="line"><a id="l00610" name="l00610"></a><span class="lineno"> 610</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="l00611" name="l00611"></a><span class="lineno"> 611</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00612" name="l00612"></a><span class="lineno"> 612</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00613" name="l00613"></a><span class="lineno"> 613</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00614" name="l00614"></a><span class="lineno"> 614</span> </div>
|
|
<div class="line"><a id="l00616" name="l00616"></a><span class="lineno"><a class="line" href="group__ops.html#ga36fa315eef677f4143868f552cd26d03"> 616</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#gab27599802617a4c8f9964ab5f4ffee12">min</a>(</div>
|
|
<div class="line"><a id="l00617" name="l00617"></a><span class="lineno"> 617</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="l00618" name="l00618"></a><span class="lineno"> 618</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00619" name="l00619"></a><span class="lineno"> 619</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00620" name="l00620"></a><span class="lineno"> 620</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00621" name="l00621"></a><span class="lineno"> 621</span> </div>
|
|
<div class="line"><a id="l00623" name="l00623"></a><span class="lineno"><a class="line" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71"> 623</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#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00624" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00624" name="l00624"></a><span class="lineno"><a class="line" href="group__ops.html#ga6bc577c5ab10cd9c848ba81321595070"> 624</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#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</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="l00625" name="l00625"></a><span class="lineno"> 625</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</a>(a, <span class="keyword">false</span>, s);</div>
|
|
<div class="line"><a id="l00626" name="l00626"></a><span class="lineno"> 626</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00627" name="l00627"></a><span class="lineno"> 627</span> </div>
|
|
<div class="line"><a id="l00629" name="l00629"></a><span class="lineno"><a class="line" href="group__ops.html#gaf66dc3c77b88e4009e0678eda41eca81"> 629</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#ga7c3bd5ef430a71dfd298e626741e3c71">argmin</a>(</div>
|
|
<div class="line"><a id="l00630" name="l00630"></a><span class="lineno"> 630</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="l00631" name="l00631"></a><span class="lineno"> 631</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00632" name="l00632"></a><span class="lineno"> 632</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00633" name="l00633"></a><span class="lineno"> 633</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00634" name="l00634"></a><span class="lineno"> 634</span> </div>
|
|
<div class="line"><a id="l00636" name="l00636"></a><span class="lineno"><a class="line" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83"> 636</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#gae60b0b5339b9c50b9970260faf613e83">argmax</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00637" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00637" name="l00637"></a><span class="lineno"><a class="line" href="group__ops.html#gae6f6c5a840320b336fdc9687e0ed56c8"> 637</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#gae60b0b5339b9c50b9970260faf613e83">argmax</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="l00638" name="l00638"></a><span class="lineno"> 638</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">argmax</a>(a, <span class="keyword">false</span>, s);</div>
|
|
<div class="line"><a id="l00639" name="l00639"></a><span class="lineno"> 639</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00640" name="l00640"></a><span class="lineno"> 640</span> </div>
|
|
<div class="line"><a id="l00642" name="l00642"></a><span class="lineno"><a class="line" href="group__ops.html#ga2efa67466510fc26ab9ea8dff30f2ba5"> 642</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#gae60b0b5339b9c50b9970260faf613e83">argmax</a>(</div>
|
|
<div class="line"><a id="l00643" name="l00643"></a><span class="lineno"> 643</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="l00644" name="l00644"></a><span class="lineno"> 644</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00645" name="l00645"></a><span class="lineno"> 645</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00646" name="l00646"></a><span class="lineno"> 646</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00647" name="l00647"></a><span class="lineno"> 647</span> </div>
|
|
<div class="line"><a id="l00649" name="l00649"></a><span class="lineno"><a class="line" href="group__ops.html#ga7fb616054665b3c2d61fa234f501f079"> 649</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#ga7fb616054665b3c2d61fa234f501f079">sort</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="l00650" name="l00650"></a><span class="lineno"> 650</span> </div>
|
|
<div class="line"><a id="l00652" name="l00652"></a><span class="lineno"><a class="line" href="group__ops.html#gaae1bc47aa737f705d0e5884270063fea"> 652</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#ga7fb616054665b3c2d61fa234f501f079">sort</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="l00653" name="l00653"></a><span class="lineno"> 653</span> </div>
|
|
<div class="line"><a id="l00655" name="l00655"></a><span class="lineno"><a class="line" href="group__ops.html#ga8df3b2703bf671457422894dd870cdc5"> 655</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#ga8df3b2703bf671457422894dd870cdc5">argsort</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="l00656" name="l00656"></a><span class="lineno"> 656</span> </div>
|
|
<div class="line"><a id="l00658" name="l00658"></a><span class="lineno"><a class="line" href="group__ops.html#ga7878e0daa5a75f44e57b5fe948fa3ef6"> 658</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#ga8df3b2703bf671457422894dd870cdc5">argsort</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="l00659" name="l00659"></a><span class="lineno"> 659</span> </div>
|
|
<div class="line"><a id="l00664" name="l00664"></a><span class="lineno"><a class="line" href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405"> 664</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#gac1b30830a972fb9a2601379ad2b32405">partition</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> kth, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00665" name="l00665"></a><span class="lineno"> 665</span> </div>
|
|
<div class="line"><a id="l00670" name="l00670"></a><span class="lineno"><a class="line" href="group__ops.html#ga4fbea3a5f66cf81e3c119d1661119321"> 670</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#gac1b30830a972fb9a2601379ad2b32405">partition</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> kth, <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="l00671" name="l00671"></a><span class="lineno"> 671</span> </div>
|
|
<div class="line"><a id="l00676" name="l00676"></a><span class="lineno"><a class="line" href="group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd"> 676</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#gaf301c49c10fa9b95a9e8dc52ead1a8dd">argpartition</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> kth, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00677" name="l00677"></a><span class="lineno"> 677</span> </div>
|
|
<div class="line"><a id="l00682" name="l00682"></a><span class="lineno"><a class="line" href="group__ops.html#ga7b15c654c7463def57857a0e239989a3"> 682</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#gaf301c49c10fa9b95a9e8dc52ead1a8dd">argpartition</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> kth, <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="l00683" name="l00683"></a><span class="lineno"> 683</span> </div>
|
|
<div class="line"><a id="l00685" name="l00685"></a><span class="lineno"><a class="line" href="group__ops.html#ga5487dd887c43e5341f3e68ffe47f0f5a"> 685</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#ga5487dd887c43e5341f3e68ffe47f0f5a">topk</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, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00686" name="l00686"></a><span class="lineno"> 686</span> </div>
|
|
<div class="line"><a id="l00688" name="l00688"></a><span class="lineno"><a class="line" href="group__ops.html#ga35b8436c79ff953f6c809598b646f498"> 688</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#ga5487dd887c43e5341f3e68ffe47f0f5a">topk</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, <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="l00689" name="l00689"></a><span class="lineno"> 689</span> </div>
|
|
<div class="line"><a id="l00691" name="l00691"></a><span class="lineno"><a class="line" href="group__ops.html#gacff4eb57c085d571e722083680267ac5"> 691</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#gacff4eb57c085d571e722083680267ac5">logsumexp</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> keepdims, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00692" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00692" name="l00692"></a><span class="lineno"><a class="line" href="group__ops.html#ga59be50b4e92f1dc20b53460cefa3910d"> 692</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#gacff4eb57c085d571e722083680267ac5">logsumexp</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="l00693" name="l00693"></a><span class="lineno"> 693</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(a, <span class="keyword">false</span>, <a class="code hl_function" href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">to_stream</a>(s));</div>
|
|
<div class="line"><a id="l00694" name="l00694"></a><span class="lineno"> 694</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00695" name="l00695"></a><span class="lineno"> 695</span> </div>
|
|
<div class="line"><a id="l00697" name="l00697"></a><span class="lineno"><a class="line" href="group__ops.html#gae3969c7bd24c4f3ab97831df28239689"> 697</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#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(</div>
|
|
<div class="line"><a id="l00698" name="l00698"></a><span class="lineno"> 698</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="l00699" name="l00699"></a><span class="lineno"> 699</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00700" name="l00700"></a><span class="lineno"> 700</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00701" name="l00701"></a><span class="lineno"> 701</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00702" name="l00702"></a><span class="lineno"> 702</span> </div>
|
|
<div class="line"><a id="l00704" name="l00704"></a><span class="lineno"><a class="line" href="group__ops.html#gafef5cb2159c16a60a95470cc823bdd44"> 704</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#gacff4eb57c085d571e722083680267ac5">logsumexp</a>(</div>
|
|
<div class="line"><a id="l00705" name="l00705"></a><span class="lineno"> 705</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="l00706" name="l00706"></a><span class="lineno"> 706</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00707" name="l00707"></a><span class="lineno"> 707</span> <span class="keywordtype">bool</span> keepdims = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l00708" name="l00708"></a><span class="lineno"> 708</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00709" name="l00709"></a><span class="lineno"> 709</span> </div>
|
|
<div class="line"><a id="l00711" name="l00711"></a><span class="lineno"><a class="line" href="group__ops.html#ga5528e80f5e8bad71e106a0cf9edd8920"> 711</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#ga5528e80f5e8bad71e106a0cf9edd8920">abs</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="l00712" name="l00712"></a><span class="lineno"> 712</span> </div>
|
|
<div class="line"><a id="l00714" name="l00714"></a><span class="lineno"><a class="line" href="group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6"> 714</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#ga95d9a9425533b5ed1707eb00184dffc6">negative</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="l00715" name="l00715"></a><span class="lineno"><a class="line" href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816"> 715</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#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(<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="l00716" name="l00716"></a><span class="lineno"> 716</span> </div>
|
|
<div class="line"><a id="l00718" name="l00718"></a><span class="lineno"><a class="line" href="group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa"> 718</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#ga20f1a1a8c0cd6206485f9363f3915faa">sign</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="l00719" name="l00719"></a><span class="lineno"> 719</span> </div>
|
|
<div class="line"><a id="l00721" name="l00721"></a><span class="lineno"><a class="line" href="group__ops.html#gabca78d34ce93f0de2814e62225bb2a53"> 721</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#gabca78d34ce93f0de2814e62225bb2a53">logical_not</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="l00722" name="l00722"></a><span class="lineno"> 722</span> </div>
|
|
<div class="line"><a id="l00724" name="l00724"></a><span class="lineno"><a class="line" href="group__ops.html#ga768977cda8d68cf23f464a6af9907876"> 724</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#ga768977cda8d68cf23f464a6af9907876">logical_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="l00725" name="l00725"></a><span class="lineno"><a class="line" href="group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3"> 725</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#gaee1d774bb0843601d7a0a4257d616ae3">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="l00726" name="l00726"></a><span class="lineno"> 726</span> </div>
|
|
<div class="line"><a id="l00728" name="l00728"></a><span class="lineno"><a class="line" href="group__ops.html#ga23768728e4dd070c917fbb0ed0d0c2ec"> 728</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#ga23768728e4dd070c917fbb0ed0d0c2ec">logical_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="l00729" name="l00729"></a><span class="lineno"><a class="line" href="group__ops.html#ga27af56a98270d4d76d139f0f9171b83a"> 729</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#ga27af56a98270d4d76d139f0f9171b83a">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="l00730" name="l00730"></a><span class="lineno"> 730</span> </div>
|
|
<div class="line"><a id="l00732" name="l00732"></a><span class="lineno"><a class="line" href="group__ops.html#ga4d29556bb93e2f66916116cf1f062b36"> 732</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#ga4d29556bb93e2f66916116cf1f062b36">reciprocal</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="l00733" name="l00733"></a><span class="lineno"> 733</span> </div>
|
|
<div class="line"><a id="l00735" name="l00735"></a><span class="lineno"><a class="line" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7"> 735</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#ga2d32d67cfd76785a72c43d89b94dc7d7">add</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="l00736" name="l00736"></a><span class="lineno"><a class="line" href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c"> 736</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#ga26e5a043eaaaf066d1400adac9c11d0c">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="l00737" name="l00737"></a><span class="lineno"> 737</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00738" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00738" name="l00738"></a><span class="lineno"><a class="line" href="group__ops.html#ga7d0ec8d01e7cefa6a6b25f11876761b5"> 738</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#ga26e5a043eaaaf066d1400adac9c11d0c">operator+</a>(T 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="l00739" name="l00739"></a><span class="lineno"> 739</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">add</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00740" name="l00740"></a><span class="lineno"> 740</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00741" name="l00741"></a><span class="lineno"> 741</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00742" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00742" name="l00742"></a><span class="lineno"><a class="line" href="group__ops.html#ga7cc080a4f9d4a667f2099aa0dbfefadd"> 742</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#ga26e5a043eaaaf066d1400adac9c11d0c">operator+</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00743" name="l00743"></a><span class="lineno"> 743</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2d32d67cfd76785a72c43d89b94dc7d7">add</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00744" name="l00744"></a><span class="lineno"> 744</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00745" name="l00745"></a><span class="lineno"> 745</span> </div>
|
|
<div class="line"><a id="l00747" name="l00747"></a><span class="lineno"><a class="line" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133"> 747</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#ga196c240d3d0fcbb4713802c485e15133">subtract</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="l00748" name="l00748"></a><span class="lineno"><a class="line" href="group__ops.html#ga0c7f3cb36d4ca516c7a33142f88b9181"> 748</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#gade2eea48989f4caaf36e89f7bd2a8816">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="l00749" name="l00749"></a><span class="lineno"> 749</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00750" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00750" name="l00750"></a><span class="lineno"><a class="line" href="group__ops.html#gae68d3d0691ba951501218e98439f3465"> 750</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#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(T 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="l00751" name="l00751"></a><span class="lineno"> 751</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">subtract</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00752" name="l00752"></a><span class="lineno"> 752</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00753" name="l00753"></a><span class="lineno"> 753</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00754" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00754" name="l00754"></a><span class="lineno"><a class="line" href="group__ops.html#gaf5e5d882c51ad0a0ea315c274d5439b2"> 754</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#gade2eea48989f4caaf36e89f7bd2a8816">operator-</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00755" name="l00755"></a><span class="lineno"> 755</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">subtract</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00756" name="l00756"></a><span class="lineno"> 756</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00757" name="l00757"></a><span class="lineno"> 757</span> </div>
|
|
<div class="line"><a id="l00759" name="l00759"></a><span class="lineno"><a class="line" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38"> 759</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#gaf57392e641640b5d06e4c99518391c38">multiply</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="l00760" name="l00760"></a><span class="lineno"><a class="line" href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0"> 760</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#ga26c33f5cdb6fc10d272acd6e208034e0">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="l00761" name="l00761"></a><span class="lineno"> 761</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00762" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00762" name="l00762"></a><span class="lineno"><a class="line" href="group__ops.html#gac22a67f7de797b1ae59029843cbdcab6"> 762</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#ga26c33f5cdb6fc10d272acd6e208034e0">operator*</a>(T 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="l00763" name="l00763"></a><span class="lineno"> 763</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">multiply</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00764" name="l00764"></a><span class="lineno"> 764</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00765" name="l00765"></a><span class="lineno"> 765</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00766" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00766" name="l00766"></a><span class="lineno"><a class="line" href="group__ops.html#ga6f2369ed5fae8ff9b1528670a004dde2"> 766</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#ga26c33f5cdb6fc10d272acd6e208034e0">operator*</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00767" name="l00767"></a><span class="lineno"> 767</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">multiply</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00768" name="l00768"></a><span class="lineno"> 768</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00769" name="l00769"></a><span class="lineno"> 769</span> </div>
|
|
<div class="line"><a id="l00771" name="l00771"></a><span class="lineno"><a class="line" href="group__ops.html#ga77472dd06cfa7a30a42e4fd927bd859f"> 771</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#ga77472dd06cfa7a30a42e4fd927bd859f">divide</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="l00772" name="l00772"></a><span class="lineno"><a class="line" href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf"> 772</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#gaeedf77f722b394429f1a7f6c367883bf">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="l00773" name="l00773"></a><span class="lineno"><a class="line" href="group__ops.html#ga7366ec7f453be2a4dc449f0faa1bf554"> 773</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#gaeedf77f722b394429f1a7f6c367883bf">operator/</a>(<span class="keywordtype">double</span> 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="l00774" name="l00774"></a><span class="lineno"><a class="line" href="group__ops.html#gadfb324ae9b4feb2c7ea0ac6ade639f38"> 774</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#gaeedf77f722b394429f1a7f6c367883bf">operator/</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">double</span> b);</div>
|
|
<div class="line"><a id="l00775" name="l00775"></a><span class="lineno"> 775</span> </div>
|
|
<div class="line"><a id="l00777" name="l00777"></a><span class="lineno"> 777</span>std::vector<array></div>
|
|
<div class="line"><a id="l00778" name="l00778"></a><span class="lineno"><a class="line" href="group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894"> 778</a></span><a class="code hl_function" href="group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894">divmod</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="l00779" name="l00779"></a><span class="lineno"> 779</span> </div>
|
|
<div class="line"><a id="l00781" name="l00781"></a><span class="lineno"><a class="line" href="group__ops.html#ga05b4c6054d028107869511f927da01cd"> 781</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#ga05b4c6054d028107869511f927da01cd">floor_divide</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="l00782" name="l00782"></a><span class="lineno"> 782</span> </div>
|
|
<div class="line"><a id="l00784" name="l00784"></a><span class="lineno"><a class="line" href="group__ops.html#ga99f5c904f724156a814d7817188351d2"> 784</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#ga99f5c904f724156a814d7817188351d2">remainder</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="l00785" name="l00785"></a><span class="lineno"><a class="line" href="group__ops.html#gab3bfbf82b1e4de7b00bbcf1a2255fbde"> 785</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#gab3bfbf82b1e4de7b00bbcf1a2255fbde">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="l00786" name="l00786"></a><span class="lineno"> 786</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00787" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00787" name="l00787"></a><span class="lineno"><a class="line" href="group__ops.html#ga50817666f0b82afcbf4a123486af9908"> 787</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#gab3bfbf82b1e4de7b00bbcf1a2255fbde">operator%</a>(T 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="l00788" name="l00788"></a><span class="lineno"> 788</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga99f5c904f724156a814d7817188351d2">remainder</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(a), b);</div>
|
|
<div class="line"><a id="l00789" name="l00789"></a><span class="lineno"> 789</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00790" name="l00790"></a><span class="lineno"> 790</span><span class="keyword">template</span> <<span class="keyword">typename</span> T></div>
|
|
<div class="foldopen" id="foldopen00791" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00791" name="l00791"></a><span class="lineno"><a class="line" href="group__ops.html#ga46c01daa07433542a477d216e13a8480"> 791</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#gab3bfbf82b1e4de7b00bbcf1a2255fbde">operator%</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, T b) {</div>
|
|
<div class="line"><a id="l00792" name="l00792"></a><span class="lineno"> 792</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga99f5c904f724156a814d7817188351d2">remainder</a>(a, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>(b));</div>
|
|
<div class="line"><a id="l00793" name="l00793"></a><span class="lineno"> 793</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00794" name="l00794"></a><span class="lineno"> 794</span> </div>
|
|
<div class="line"><a id="l00796" name="l00796"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d"> 796</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#ga7ade2ea305e2e4219c3609443fb5db8d">maximum</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="l00797" name="l00797"></a><span class="lineno"> 797</span> </div>
|
|
<div class="line"><a id="l00799" name="l00799"></a><span class="lineno"><a class="line" href="group__ops.html#ga49ba00c090f81f331c91b0c97040bce0"> 799</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#ga49ba00c090f81f331c91b0c97040bce0">minimum</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="l00800" name="l00800"></a><span class="lineno"> 800</span> </div>
|
|
<div class="line"><a id="l00802" name="l00802"></a><span class="lineno"><a class="line" href="group__ops.html#ga8d656904aa2690b60955ae745aecfc30"> 802</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#ga8d656904aa2690b60955ae745aecfc30">floor</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="l00803" name="l00803"></a><span class="lineno"> 803</span> </div>
|
|
<div class="line"><a id="l00805" name="l00805"></a><span class="lineno"><a class="line" href="group__ops.html#ga1404ecceff83fd9b9139b7520f55e096"> 805</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#ga1404ecceff83fd9b9139b7520f55e096">ceil</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="l00806" name="l00806"></a><span class="lineno"> 806</span> </div>
|
|
<div class="line"><a id="l00808" name="l00808"></a><span class="lineno"><a class="line" href="group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e"> 808</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#ga1234e4c39cfa79f19d4bdb5b8ea4d45e">square</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="l00809" name="l00809"></a><span class="lineno"> 809</span> </div>
|
|
<div class="line"><a id="l00811" name="l00811"></a><span class="lineno"><a class="line" href="group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125"> 811</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#ga8a3b04e23e347d99ecf411fd6f4e5125">exp</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="l00812" name="l00812"></a><span class="lineno"> 812</span> </div>
|
|
<div class="line"><a id="l00814" name="l00814"></a><span class="lineno"><a class="line" href="group__ops.html#gaebf0a73ad3732fba39df37826c235692"> 814</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#gaebf0a73ad3732fba39df37826c235692">sin</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="l00815" name="l00815"></a><span class="lineno"> 815</span> </div>
|
|
<div class="line"><a id="l00817" name="l00817"></a><span class="lineno"><a class="line" href="group__ops.html#ga39dfdf72b556012aa35ff27a94116e74"> 817</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#ga39dfdf72b556012aa35ff27a94116e74">cos</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="l00818" name="l00818"></a><span class="lineno"> 818</span> </div>
|
|
<div class="line"><a id="l00820" name="l00820"></a><span class="lineno"><a class="line" href="group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5"> 820</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#ga3f10e89a4bcb1a8fa44fb33b8d1176a5">tan</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="l00821" name="l00821"></a><span class="lineno"> 821</span> </div>
|
|
<div class="line"><a id="l00823" name="l00823"></a><span class="lineno"><a class="line" href="group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619"> 823</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#ga8770e8c8f23f13343911f4c9d6e1c619">arcsin</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="l00824" name="l00824"></a><span class="lineno"> 824</span> </div>
|
|
<div class="line"><a id="l00826" name="l00826"></a><span class="lineno"><a class="line" href="group__ops.html#ga08bec7cb10c84466487b507fc5bf9776"> 826</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#ga08bec7cb10c84466487b507fc5bf9776">arccos</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="l00827" name="l00827"></a><span class="lineno"> 827</span> </div>
|
|
<div class="line"><a id="l00829" name="l00829"></a><span class="lineno"><a class="line" href="group__ops.html#gaa041f3f070e68f4946db07516b7d092e"> 829</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#gaa041f3f070e68f4946db07516b7d092e">arctan</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="l00830" name="l00830"></a><span class="lineno"> 830</span> </div>
|
|
<div class="line"><a id="l00832" name="l00832"></a><span class="lineno"><a class="line" href="group__ops.html#ga6caba9c92b5989123501f909cc7da354"> 832</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#ga6caba9c92b5989123501f909cc7da354">arctan2</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="l00833" name="l00833"></a><span class="lineno"> 833</span> </div>
|
|
<div class="line"><a id="l00835" name="l00835"></a><span class="lineno"><a class="line" href="group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7"> 835</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#gaf532375c6563dbd6e329bdedf0224dd7">sinh</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="l00836" name="l00836"></a><span class="lineno"> 836</span> </div>
|
|
<div class="line"><a id="l00838" name="l00838"></a><span class="lineno"><a class="line" href="group__ops.html#ga2181b71cda88007a3092be4795ff0715"> 838</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#ga2181b71cda88007a3092be4795ff0715">cosh</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="l00839" name="l00839"></a><span class="lineno"> 839</span> </div>
|
|
<div class="line"><a id="l00841" name="l00841"></a><span class="lineno"><a class="line" href="group__ops.html#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d"> 841</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#ga5efb19aa0dfa42d8a3d5e1dfd569cd6d">tanh</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="l00842" name="l00842"></a><span class="lineno"> 842</span> </div>
|
|
<div class="line"><a id="l00844" name="l00844"></a><span class="lineno"><a class="line" href="group__ops.html#gac62e2cedc49ef2c90dd8584000317450"> 844</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#gac62e2cedc49ef2c90dd8584000317450">arcsinh</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="l00845" name="l00845"></a><span class="lineno"> 845</span> </div>
|
|
<div class="line"><a id="l00847" name="l00847"></a><span class="lineno"><a class="line" href="group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5"> 847</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#gaafafcfcebdf7248679c8543d0c0497e5">arccosh</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="l00848" name="l00848"></a><span class="lineno"> 848</span> </div>
|
|
<div class="line"><a id="l00850" name="l00850"></a><span class="lineno"><a class="line" href="group__ops.html#gab46a35925a04c5a9d2ec7898ee55358e"> 850</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#gab46a35925a04c5a9d2ec7898ee55358e">arctanh</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="l00851" name="l00851"></a><span class="lineno"> 851</span> </div>
|
|
<div class="line"><a id="l00853" name="l00853"></a><span class="lineno"><a class="line" href="group__ops.html#ga3a70569b50e1083c5ded199d73fb960c"> 853</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#ga3a70569b50e1083c5ded199d73fb960c">degrees</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="l00854" name="l00854"></a><span class="lineno"> 854</span> </div>
|
|
<div class="line"><a id="l00856" name="l00856"></a><span class="lineno"><a class="line" href="group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e"> 856</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#ga07aa8059adba5b9a8818027b8aafd31e">radians</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="l00857" name="l00857"></a><span class="lineno"> 857</span> </div>
|
|
<div class="line"><a id="l00859" name="l00859"></a><span class="lineno"><a class="line" href="group__ops.html#ga6fb22d4926133573e430fcc92f4eef31"> 859</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#ga6fb22d4926133573e430fcc92f4eef31">log</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="l00860" name="l00860"></a><span class="lineno"> 860</span> </div>
|
|
<div class="line"><a id="l00862" name="l00862"></a><span class="lineno"><a class="line" href="group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b"> 862</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#ga144228d7222d15af3a135b8b0f3fa21b">log2</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="l00863" name="l00863"></a><span class="lineno"> 863</span> </div>
|
|
<div class="line"><a id="l00865" name="l00865"></a><span class="lineno"><a class="line" href="group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e"> 865</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#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e">log10</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="l00866" name="l00866"></a><span class="lineno"> 866</span> </div>
|
|
<div class="line"><a id="l00868" name="l00868"></a><span class="lineno"><a class="line" href="group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604"> 868</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#ga20a1f4270c35b0fa544f5105a87a1604">log1p</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="l00869" name="l00869"></a><span class="lineno"> 869</span> </div>
|
|
<div class="line"><a id="l00871" name="l00871"></a><span class="lineno"><a class="line" href="group__ops.html#gaf985df6609c6bd75a14a844655d89eaa"> 871</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#gaf985df6609c6bd75a14a844655d89eaa">logaddexp</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="l00872" name="l00872"></a><span class="lineno"> 872</span> </div>
|
|
<div class="line"><a id="l00874" name="l00874"></a><span class="lineno"><a class="line" href="group__ops.html#ga708abf8f79609cd6831db7c38cafac0e"> 874</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#ga708abf8f79609cd6831db7c38cafac0e">sigmoid</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="l00875" name="l00875"></a><span class="lineno"> 875</span> </div>
|
|
<div class="line"><a id="l00877" name="l00877"></a><span class="lineno"><a class="line" href="group__ops.html#ga292a335240fd5d6d625fb7a340ff5eb0"> 877</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#ga292a335240fd5d6d625fb7a340ff5eb0">erf</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="l00878" name="l00878"></a><span class="lineno"> 878</span> </div>
|
|
<div class="line"><a id="l00880" name="l00880"></a><span class="lineno"><a class="line" href="group__ops.html#ga76fb9062c64264e34d2e07013390557c"> 880</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#ga76fb9062c64264e34d2e07013390557c">erfinv</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="l00881" name="l00881"></a><span class="lineno"> 881</span> </div>
|
|
<div class="line"><a id="l00883" name="l00883"></a><span class="lineno"><a class="line" href="group__ops.html#ga54ca54f06bfb2be15b163a5209e2a0f0"> 883</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#ga54ca54f06bfb2be15b163a5209e2a0f0">expm1</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="l00884" name="l00884"></a><span class="lineno"> 884</span> </div>
|
|
<div class="line"><a id="l00886" name="l00886"></a><span class="lineno"><a class="line" href="group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f"> 886</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#ga36bc28f1deb2fe668ca9ae1e447b6b1f">stop_gradient</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="l00887" name="l00887"></a><span class="lineno"> 887</span> </div>
|
|
<div class="line"><a id="l00889" name="l00889"></a><span class="lineno"><a class="line" href="group__ops.html#ga2d74d43f007a069384e89d8416525331"> 889</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#ga2d74d43f007a069384e89d8416525331">round</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> decimals, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00890" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00890" name="l00890"></a><span class="lineno"><a class="line" href="group__ops.html#gaf18fb7e98bf8cf3b7fbc5e64c988a95b"> 890</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#ga2d74d43f007a069384e89d8416525331">round</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="l00891" name="l00891"></a><span class="lineno"> 891</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#ga2d74d43f007a069384e89d8416525331">round</a>(a, 0, s);</div>
|
|
<div class="line"><a id="l00892" name="l00892"></a><span class="lineno"> 892</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00893" name="l00893"></a><span class="lineno"> 893</span> </div>
|
|
<div class="line"><a id="l00895" name="l00895"></a><span class="lineno"><a class="line" href="group__ops.html#ga753d59f5a9f5f2362865ee83b4dced2a"> 895</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="l00896" name="l00896"></a><span class="lineno"> 896</span> </div>
|
|
<div class="line"><a id="l00898" name="l00898"></a><span class="lineno"><a class="line" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444"> 898</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#gab6e7f655a9ff15350ca5379692f9d444">gather</a>(</div>
|
|
<div class="line"><a id="l00899" name="l00899"></a><span class="lineno"> 899</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="l00900" name="l00900"></a><span class="lineno"> 900</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00901" name="l00901"></a><span class="lineno"> 901</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00902" name="l00902"></a><span class="lineno"> 902</span> <span class="keyword">const</span> std::vector<int>& slice_sizes,</div>
|
|
<div class="line"><a id="l00903" name="l00903"></a><span class="lineno"> 903</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00904" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00904" name="l00904"></a><span class="lineno"><a class="line" href="group__ops.html#gadb4337ca5d4f88fe9e7c083bc478158b"> 904</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#gab6e7f655a9ff15350ca5379692f9d444">gather</a>(</div>
|
|
<div class="line"><a id="l00905" name="l00905"></a><span class="lineno"> 905</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="l00906" name="l00906"></a><span class="lineno"> 906</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="l00907" name="l00907"></a><span class="lineno"> 907</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00908" name="l00908"></a><span class="lineno"> 908</span> <span class="keyword">const</span> std::vector<int>& slice_sizes,</div>
|
|
<div class="line"><a id="l00909" name="l00909"></a><span class="lineno"> 909</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00910" name="l00910"></a><span class="lineno"> 910</span> <span class="keywordflow">return</span> <a class="code hl_function" href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">gather</a>(a, {indices}, std::vector<int>{axis}, slice_sizes, s);</div>
|
|
<div class="line"><a id="l00911" name="l00911"></a><span class="lineno"> 911</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00912" name="l00912"></a><span class="lineno"> 912</span> </div>
|
|
<div class="line"><a id="l00914" name="l00914"></a><span class="lineno"><a class="line" href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7"> 914</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#gac2fc270882fcfa81eb8bd068cc0d86d7">take</a>(</div>
|
|
<div class="line"><a id="l00915" name="l00915"></a><span class="lineno"> 915</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="l00916" name="l00916"></a><span class="lineno"> 916</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="l00917" name="l00917"></a><span class="lineno"> 917</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00918" name="l00918"></a><span class="lineno"> 918</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00919" name="l00919"></a><span class="lineno"> 919</span> </div>
|
|
<div class="line"><a id="l00921" name="l00921"></a><span class="lineno"><a class="line" href="group__ops.html#ga45d0f423a5e030440ef753f36c5aabf1"> 921</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#gac2fc270882fcfa81eb8bd068cc0d86d7">take</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>& indices, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00922" name="l00922"></a><span class="lineno"> 922</span> </div>
|
|
<div class="line"><a id="l00924" name="l00924"></a><span class="lineno"><a class="line" href="group__ops.html#gae0a81d4f983e296a87302e36d65bfc76"> 924</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#gae0a81d4f983e296a87302e36d65bfc76">take_along_axis</a>(</div>
|
|
<div class="line"><a id="l00925" name="l00925"></a><span class="lineno"> 925</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="l00926" name="l00926"></a><span class="lineno"> 926</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="l00927" name="l00927"></a><span class="lineno"> 927</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00928" name="l00928"></a><span class="lineno"> 928</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l00929" name="l00929"></a><span class="lineno"> 929</span> </div>
|
|
<div class="line"><a id="l00931" name="l00931"></a><span class="lineno"><a class="line" href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61"> 931</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="l00932" name="l00932"></a><span class="lineno"> 932</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="l00933" name="l00933"></a><span class="lineno"> 933</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00934" name="l00934"></a><span class="lineno"> 934</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="l00935" name="l00935"></a><span class="lineno"> 935</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00936" name="l00936"></a><span class="lineno"> 936</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00937" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00937" name="l00937"></a><span class="lineno"><a class="line" href="group__ops.html#gac2c2b379a3ce959dbe1c4a68f112edfe"> 937</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="l00938" name="l00938"></a><span class="lineno"> 938</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="l00939" name="l00939"></a><span class="lineno"> 939</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="l00940" name="l00940"></a><span class="lineno"> 940</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="l00941" name="l00941"></a><span class="lineno"> 941</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00942" name="l00942"></a><span class="lineno"> 942</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00943" name="l00943"></a><span class="lineno"> 943</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="l00944" name="l00944"></a><span class="lineno"> 944</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00945" name="l00945"></a><span class="lineno"> 945</span> </div>
|
|
<div class="line"><a id="l00947" name="l00947"></a><span class="lineno"><a class="line" href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174"> 947</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="l00948" name="l00948"></a><span class="lineno"> 948</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="l00949" name="l00949"></a><span class="lineno"> 949</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00950" name="l00950"></a><span class="lineno"> 950</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="l00951" name="l00951"></a><span class="lineno"> 951</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00952" name="l00952"></a><span class="lineno"> 952</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00953" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00953" name="l00953"></a><span class="lineno"><a class="line" href="group__ops.html#gac13318518e5703f1273c5366eb523a5a"> 953</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="l00954" name="l00954"></a><span class="lineno"> 954</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="l00955" name="l00955"></a><span class="lineno"> 955</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="l00956" name="l00956"></a><span class="lineno"> 956</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="l00957" name="l00957"></a><span class="lineno"> 957</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00958" name="l00958"></a><span class="lineno"> 958</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00959" name="l00959"></a><span class="lineno"> 959</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="l00960" name="l00960"></a><span class="lineno"> 960</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00961" name="l00961"></a><span class="lineno"> 961</span> </div>
|
|
<div class="line"><a id="l00963" name="l00963"></a><span class="lineno"><a class="line" href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0"> 963</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="l00964" name="l00964"></a><span class="lineno"> 964</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="l00965" name="l00965"></a><span class="lineno"> 965</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00966" name="l00966"></a><span class="lineno"> 966</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="l00967" name="l00967"></a><span class="lineno"> 967</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00968" name="l00968"></a><span class="lineno"> 968</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00969" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00969" name="l00969"></a><span class="lineno"><a class="line" href="group__ops.html#gaf83c53c453faa9083ba27e4b97539339"> 969</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="l00970" name="l00970"></a><span class="lineno"> 970</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="l00971" name="l00971"></a><span class="lineno"> 971</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="l00972" name="l00972"></a><span class="lineno"> 972</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="l00973" name="l00973"></a><span class="lineno"> 973</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00974" name="l00974"></a><span class="lineno"> 974</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00975" name="l00975"></a><span class="lineno"> 975</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="l00976" name="l00976"></a><span class="lineno"> 976</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00977" name="l00977"></a><span class="lineno"> 977</span> </div>
|
|
<div class="line"><a id="l00979" name="l00979"></a><span class="lineno"><a class="line" href="group__ops.html#ga05881a4157cd113c9392d168a79e6673"> 979</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="l00980" name="l00980"></a><span class="lineno"> 980</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="l00981" name="l00981"></a><span class="lineno"> 981</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00982" name="l00982"></a><span class="lineno"> 982</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="l00983" name="l00983"></a><span class="lineno"> 983</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00984" name="l00984"></a><span class="lineno"> 984</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen00985" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00985" name="l00985"></a><span class="lineno"><a class="line" href="group__ops.html#ga9adda5f9202bb3486e4d9e1114e3a56f"> 985</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="l00986" name="l00986"></a><span class="lineno"> 986</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="l00987" name="l00987"></a><span class="lineno"> 987</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="l00988" name="l00988"></a><span class="lineno"> 988</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="l00989" name="l00989"></a><span class="lineno"> 989</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l00990" name="l00990"></a><span class="lineno"> 990</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l00991" name="l00991"></a><span class="lineno"> 991</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="l00992" name="l00992"></a><span class="lineno"> 992</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00994" name="l00994"></a><span class="lineno"><a class="line" href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5"> 994</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="l00995" name="l00995"></a><span class="lineno"> 995</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="l00996" name="l00996"></a><span class="lineno"> 996</span> <span class="keyword">const</span> std::vector<array>& indices,</div>
|
|
<div class="line"><a id="l00997" name="l00997"></a><span class="lineno"> 997</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="l00998" name="l00998"></a><span class="lineno"> 998</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l00999" name="l00999"></a><span class="lineno"> 999</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="foldopen" id="foldopen01000" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01000" name="l01000"></a><span class="lineno"><a class="line" href="group__ops.html#ga51fa762a997c243ca7a19e1ed3e83199"> 1000</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="l01001" name="l01001"></a><span class="lineno"> 1001</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="l01002" name="l01002"></a><span class="lineno"> 1002</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="l01003" name="l01003"></a><span class="lineno"> 1003</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="l01004" name="l01004"></a><span class="lineno"> 1004</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01005" name="l01005"></a><span class="lineno"> 1005</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01006" name="l01006"></a><span class="lineno"> 1006</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="l01007" name="l01007"></a><span class="lineno"> 1007</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01008" name="l01008"></a><span class="lineno"> 1008</span> </div>
|
|
<div class="line"><a id="l01010" name="l01010"></a><span class="lineno"><a class="line" href="group__ops.html#ga297f853b3d90ec8ae81263977ba2ddb1"> 1010</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="l01011" name="l01011"></a><span class="lineno"> 1011</span> </div>
|
|
<div class="line"><a id="l01013" name="l01013"></a><span class="lineno"><a class="line" href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1"> 1013</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="l01014" name="l01014"></a><span class="lineno"> 1014</span> </div>
|
|
<div class="line"><a id="l01016" name="l01016"></a><span class="lineno"><a class="line" href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"> 1016</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="l01017" name="l01017"></a><span class="lineno"> 1017</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="l01018" name="l01018"></a><span class="lineno"> 1018</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
|
<div class="line"><a id="l01019" name="l01019"></a><span class="lineno"> 1019</span> <span class="keywordtype">bool</span> precise = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01020" name="l01020"></a><span class="lineno"> 1020</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01021" name="l01021"></a><span class="lineno"> 1021</span> </div>
|
|
<div class="line"><a id="l01023" name="l01023"></a><span class="lineno"><a class="line" href="group__ops.html#ga1ae3614d07d873892a530d14c3857d0b"> 1023</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="l01024" name="l01024"></a><span class="lineno"> 1024</span> </div>
|
|
<div class="line"><a id="l01026" name="l01026"></a><span class="lineno"> 1026</span><span class="keyword">inline</span> array</div>
|
|
<div class="foldopen" id="foldopen01027" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01027" name="l01027"></a><span class="lineno"><a class="line" href="group__ops.html#ga06f570d73716a24303e6de3aaba4457b"> 1027</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="l01028" name="l01028"></a><span class="lineno"> 1028</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="l01029" name="l01029"></a><span class="lineno"> 1029</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01030" name="l01030"></a><span class="lineno"> 1030</span> </div>
|
|
<div class="line"><a id="l01032" name="l01032"></a><span class="lineno"><a class="line" href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d"> 1032</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="l01033" name="l01033"></a><span class="lineno"> 1033</span> </div>
|
|
<div class="line"><a id="l01035" name="l01035"></a><span class="lineno"><a class="line" href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420"> 1035</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="l01036" name="l01036"></a><span class="lineno"> 1036</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="l01037" name="l01037"></a><span class="lineno"> 1037</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01038" name="l01038"></a><span class="lineno"> 1038</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01039" name="l01039"></a><span class="lineno"> 1039</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01040" name="l01040"></a><span class="lineno"> 1040</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01041" name="l01041"></a><span class="lineno"> 1041</span> </div>
|
|
<div class="line"><a id="l01043" name="l01043"></a><span class="lineno"><a class="line" href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f"> 1043</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="l01044" name="l01044"></a><span class="lineno"> 1044</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="l01045" name="l01045"></a><span class="lineno"> 1045</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01046" name="l01046"></a><span class="lineno"> 1046</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01047" name="l01047"></a><span class="lineno"> 1047</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01048" name="l01048"></a><span class="lineno"> 1048</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01049" name="l01049"></a><span class="lineno"> 1049</span> </div>
|
|
<div class="line"><a id="l01051" name="l01051"></a><span class="lineno"><a class="line" href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143"> 1051</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="l01052" name="l01052"></a><span class="lineno"> 1052</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="l01053" name="l01053"></a><span class="lineno"> 1053</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01054" name="l01054"></a><span class="lineno"> 1054</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01055" name="l01055"></a><span class="lineno"> 1055</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01056" name="l01056"></a><span class="lineno"> 1056</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01057" name="l01057"></a><span class="lineno"> 1057</span> </div>
|
|
<div class="line"><a id="l01059" name="l01059"></a><span class="lineno"><a class="line" href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8"> 1059</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="l01060" name="l01060"></a><span class="lineno"> 1060</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="l01061" name="l01061"></a><span class="lineno"> 1061</span> <span class="keywordtype">int</span> axis,</div>
|
|
<div class="line"><a id="l01062" name="l01062"></a><span class="lineno"> 1062</span> <span class="keywordtype">bool</span> reverse = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01063" name="l01063"></a><span class="lineno"> 1063</span> <span class="keywordtype">bool</span> inclusive = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l01064" name="l01064"></a><span class="lineno"> 1064</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01065" name="l01065"></a><span class="lineno"> 1065</span> </div>
|
|
<div class="line"><a id="l01067" name="l01067"></a><span class="lineno"><a class="line" href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e"> 1067</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="l01068" name="l01068"></a><span class="lineno"> 1068</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> input,</div>
|
|
<div class="line"><a id="l01069" name="l01069"></a><span class="lineno"> 1069</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> weight,</div>
|
|
<div class="line"><a id="l01070" name="l01070"></a><span class="lineno"> 1070</span> std::vector<int> stride = {},</div>
|
|
<div class="line"><a id="l01071" name="l01071"></a><span class="lineno"> 1071</span> std::vector<int> padding_lo = {},</div>
|
|
<div class="line"><a id="l01072" name="l01072"></a><span class="lineno"> 1072</span> std::vector<int> padding_hi = {},</div>
|
|
<div class="line"><a id="l01073" name="l01073"></a><span class="lineno"> 1073</span> std::vector<int> kernel_dilation = {},</div>
|
|
<div class="line"><a id="l01074" name="l01074"></a><span class="lineno"> 1074</span> std::vector<int> input_dilation = {},</div>
|
|
<div class="line"><a id="l01075" name="l01075"></a><span class="lineno"> 1075</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01076" name="l01076"></a><span class="lineno"> 1076</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01077" name="l01077"></a><span class="lineno"> 1077</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01078" name="l01078"></a><span class="lineno"> 1078</span> </div>
|
|
<div class="foldopen" id="foldopen01080" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01080" name="l01080"></a><span class="lineno"><a class="line" href="group__ops.html#gab59f89942cd1efaadffe9e8762e3c99d"> 1080</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="l01081" name="l01081"></a><span class="lineno"> 1081</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="l01082" name="l01082"></a><span class="lineno"> 1082</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="l01083" name="l01083"></a><span class="lineno"> 1083</span> std::vector<int> stride = {},</div>
|
|
<div class="line"><a id="l01084" name="l01084"></a><span class="lineno"> 1084</span> std::vector<int> padding = {},</div>
|
|
<div class="line"><a id="l01085" name="l01085"></a><span class="lineno"> 1085</span> std::vector<int> kernel_dilation = {},</div>
|
|
<div class="line"><a id="l01086" name="l01086"></a><span class="lineno"> 1086</span> std::vector<int> input_dilation = {},</div>
|
|
<div class="line"><a id="l01087" name="l01087"></a><span class="lineno"> 1087</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01088" name="l01088"></a><span class="lineno"> 1088</span> <span class="keywordtype">bool</span> flip = <span class="keyword">false</span>,</div>
|
|
<div class="line"><a id="l01089" name="l01089"></a><span class="lineno"> 1089</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {}) {</div>
|
|
<div class="line"><a id="l01090" name="l01090"></a><span class="lineno"> 1090</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="l01091" name="l01091"></a><span class="lineno"> 1091</span> <span class="comment">/* const array& input = */</span> input,</div>
|
|
<div class="line"><a id="l01092" name="l01092"></a><span class="lineno"> 1092</span> <span class="comment">/* const array& weight = */</span> weight,</div>
|
|
<div class="line"><a id="l01093" name="l01093"></a><span class="lineno"> 1093</span> <span class="comment">/* std::vector<int> stride = */</span> stride,</div>
|
|
<div class="line"><a id="l01094" name="l01094"></a><span class="lineno"> 1094</span> <span class="comment">/* std::vector<int> padding_lo = */</span> padding,</div>
|
|
<div class="line"><a id="l01095" name="l01095"></a><span class="lineno"> 1095</span> <span class="comment">/* std::vector<int> padding_hi = */</span> padding,</div>
|
|
<div class="line"><a id="l01096" name="l01096"></a><span class="lineno"> 1096</span> <span class="comment">/* std::vector<int> kernel_dilation = */</span> kernel_dilation,</div>
|
|
<div class="line"><a id="l01097" name="l01097"></a><span class="lineno"> 1097</span> <span class="comment">/* std::vector<int> input_dilation = */</span> input_dilation,</div>
|
|
<div class="line"><a id="l01098" name="l01098"></a><span class="lineno"> 1098</span> <span class="comment">/* int groups = */</span> groups,</div>
|
|
<div class="line"><a id="l01099" name="l01099"></a><span class="lineno"> 1099</span> <span class="comment">/* bool flip = */</span> flip,</div>
|
|
<div class="line"><a id="l01100" name="l01100"></a><span class="lineno"> 1100</span> <span class="comment">/* StreamOrDevice s = */</span> s);</div>
|
|
<div class="line"><a id="l01101" name="l01101"></a><span class="lineno"> 1101</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l01102" name="l01102"></a><span class="lineno"> 1102</span> </div>
|
|
<div class="line"><a id="l01104" name="l01104"></a><span class="lineno"><a class="line" href="group__ops.html#ga30d47e08093c03a3676f235f9f559411"> 1104</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="l01105" name="l01105"></a><span class="lineno"> 1105</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="l01106" name="l01106"></a><span class="lineno"> 1106</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="l01107" name="l01107"></a><span class="lineno"> 1107</span> <span class="keywordtype">int</span> stride = 1,</div>
|
|
<div class="line"><a id="l01108" name="l01108"></a><span class="lineno"> 1108</span> <span class="keywordtype">int</span> padding = 0,</div>
|
|
<div class="line"><a id="l01109" name="l01109"></a><span class="lineno"> 1109</span> <span class="keywordtype">int</span> dilation = 1,</div>
|
|
<div class="line"><a id="l01110" name="l01110"></a><span class="lineno"> 1110</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01111" name="l01111"></a><span class="lineno"> 1111</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01112" name="l01112"></a><span class="lineno"> 1112</span> </div>
|
|
<div class="line"><a id="l01114" name="l01114"></a><span class="lineno"><a class="line" href="group__ops.html#ga73b02833229678786e7f302d458d5a83"> 1114</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="l01115" name="l01115"></a><span class="lineno"> 1115</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="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>& weight,</div>
|
|
<div class="line"><a id="l01117" name="l01117"></a><span class="lineno"> 1117</span> <span class="keyword">const</span> std::pair<int, int>& stride = {1, 1},</div>
|
|
<div class="line"><a id="l01118" name="l01118"></a><span class="lineno"> 1118</span> <span class="keyword">const</span> std::pair<int, int>& padding = {0, 0},</div>
|
|
<div class="line"><a id="l01119" name="l01119"></a><span class="lineno"> 1119</span> <span class="keyword">const</span> std::pair<int, int>& dilation = {1, 1},</div>
|
|
<div class="line"><a id="l01120" name="l01120"></a><span class="lineno"> 1120</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01121" name="l01121"></a><span class="lineno"> 1121</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01122" name="l01122"></a><span class="lineno"> 1122</span> </div>
|
|
<div class="line"><a id="l01124" name="l01124"></a><span class="lineno"><a class="line" href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca"> 1124</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="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>& input,</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>& weight,</div>
|
|
<div class="line"><a id="l01127" name="l01127"></a><span class="lineno"> 1127</span> <span class="keyword">const</span> std::tuple<int, int, int>& stride = {1, 1, 1},</div>
|
|
<div class="line"><a id="l01128" name="l01128"></a><span class="lineno"> 1128</span> <span class="keyword">const</span> std::tuple<int, int, int>& padding = {0, 0, 0},</div>
|
|
<div class="line"><a id="l01129" name="l01129"></a><span class="lineno"> 1129</span> <span class="keyword">const</span> std::tuple<int, int, int>& dilation = {1, 1, 1},</div>
|
|
<div class="line"><a id="l01130" name="l01130"></a><span class="lineno"> 1130</span> <span class="keywordtype">int</span> groups = 1,</div>
|
|
<div class="line"><a id="l01131" name="l01131"></a><span class="lineno"> 1131</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01132" name="l01132"></a><span class="lineno"> 1132</span> </div>
|
|
<div class="line"><a id="l01134" name="l01134"></a><span class="lineno"><a class="line" href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e"> 1134</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e">quantized_matmul</a>(</div>
|
|
<div class="line"><a id="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>& x,</div>
|
|
<div class="line"><a id="l01136" name="l01136"></a><span class="lineno"> 1136</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="l01137" name="l01137"></a><span class="lineno"> 1137</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="l01138" name="l01138"></a><span class="lineno"> 1138</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="l01139" name="l01139"></a><span class="lineno"> 1139</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="l01140" name="l01140"></a><span class="lineno"> 1140</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01141" name="l01141"></a><span class="lineno"> 1141</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01142" name="l01142"></a><span class="lineno"> 1142</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01143" name="l01143"></a><span class="lineno"> 1143</span> </div>
|
|
<div class="line"><a id="l01145" name="l01145"></a><span class="lineno"><a class="line" href="group__ops.html#gab43cc28690da7cdd43b43065adbd31da"> 1145</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="l01146" name="l01146"></a><span class="lineno"> 1146</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="l01147" name="l01147"></a><span class="lineno"> 1147</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01148" name="l01148"></a><span class="lineno"> 1148</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01149" name="l01149"></a><span class="lineno"> 1149</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01150" name="l01150"></a><span class="lineno"> 1150</span> </div>
|
|
<div class="line"><a id="l01152" name="l01152"></a><span class="lineno"><a class="line" href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077"> 1152</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="l01153" name="l01153"></a><span class="lineno"> 1153</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="l01154" name="l01154"></a><span class="lineno"> 1154</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="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>& biases,</div>
|
|
<div class="line"><a id="l01156" name="l01156"></a><span class="lineno"> 1156</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01157" name="l01157"></a><span class="lineno"> 1157</span> <span class="keywordtype">int</span> bits = 4,</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#ga9aaa4381942a6d9a31d2ab8af8436e78"> 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#ga9aaa4381942a6d9a31d2ab8af8436e78">block_sparse_qmm</a>(</div>
|
|
<div class="line"><a id="l01162" name="l01162"></a><span class="lineno"> 1162</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="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>& w,</div>
|
|
<div class="line"><a id="l01164" name="l01164"></a><span class="lineno"> 1164</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="l01165" name="l01165"></a><span class="lineno"> 1165</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="l01166" name="l01166"></a><span class="lineno"> 1166</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01167" name="l01167"></a><span class="lineno"> 1167</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01168" name="l01168"></a><span class="lineno"> 1168</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="l01169" name="l01169"></a><span class="lineno"> 1169</span> <span class="keywordtype">int</span> group_size = 64,</div>
|
|
<div class="line"><a id="l01170" name="l01170"></a><span class="lineno"> 1170</span> <span class="keywordtype">int</span> bits = 4,</div>
|
|
<div class="line"><a id="l01171" name="l01171"></a><span class="lineno"> 1171</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01172" name="l01172"></a><span class="lineno"> 1172</span> </div>
|
|
<div class="line"><a id="l01174" name="l01174"></a><span class="lineno"><a class="line" href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70"> 1174</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="l01175" name="l01175"></a><span class="lineno"> 1175</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="l01176" name="l01176"></a><span class="lineno"> 1176</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="l01177" name="l01177"></a><span class="lineno"> 1177</span> <span class="keyword">const</span> <span class="keywordtype">int</span> axis = 2,</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="l01179" name="l01179"></a><span class="lineno"> 1179</span> </div>
|
|
<div class="line"><a id="l01180" name="l01180"></a><span class="lineno"><a class="line" href="group__ops.html#gad7fe00b566f89d607639c1a497cabbc6"> 1180</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="l01181" name="l01181"></a><span class="lineno"> 1181</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="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>& b,</div>
|
|
<div class="line"><a id="l01183" name="l01183"></a><span class="lineno"> 1183</span> <span class="keyword">const</span> std::vector<int>& axes_a,</div>
|
|
<div class="line"><a id="l01184" name="l01184"></a><span class="lineno"> 1184</span> <span class="keyword">const</span> std::vector<int>& axes_b,</div>
|
|
<div class="line"><a id="l01185" name="l01185"></a><span class="lineno"> 1185</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01186" name="l01186"></a><span class="lineno"> 1186</span> </div>
|
|
<div class="line"><a id="l01188" name="l01188"></a><span class="lineno"><a class="line" href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d"> 1188</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="l01189" name="l01189"></a><span class="lineno"> 1189</span> </div>
|
|
<div class="line"><a id="l01191" name="l01191"></a><span class="lineno"><a class="line" href="group__ops.html#ga654fec16a9746b390916697a2ab2546e"> 1191</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="l01192" name="l01192"></a><span class="lineno"> 1192</span> </div>
|
|
<div class="line"><a id="l01194" name="l01194"></a><span class="lineno"><a class="line" href="group__ops.html#ga82a53e083205a965387b3c3e2463244a"> 1194</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="l01195" name="l01195"></a><span class="lineno"> 1195</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> c,</div>
|
|
<div class="line"><a id="l01196" name="l01196"></a><span class="lineno"> 1196</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l01197" name="l01197"></a><span class="lineno"> 1197</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
|
<div class="line"><a id="l01198" name="l01198"></a><span class="lineno"> 1198</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& alpha = 1.f,</div>
|
|
<div class="line"><a id="l01199" name="l01199"></a><span class="lineno"> 1199</span> <span class="keyword">const</span> <span class="keywordtype">float</span>& beta = 1.f,</div>
|
|
<div class="line"><a id="l01200" name="l01200"></a><span class="lineno"> 1200</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01201" name="l01201"></a><span class="lineno"> 1201</span> </div>
|
|
<div class="line"><a id="l01203" name="l01203"></a><span class="lineno"><a class="line" href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6"> 1203</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="l01204" name="l01204"></a><span class="lineno"> 1204</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> a,</div>
|
|
<div class="line"><a id="l01205" name="l01205"></a><span class="lineno"> 1205</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> b,</div>
|
|
<div class="line"><a id="l01206" name="l01206"></a><span class="lineno"> 1206</span> <span class="keywordtype">int</span> block_size,</div>
|
|
<div class="line"><a id="l01207" name="l01207"></a><span class="lineno"> 1207</span> std::optional<array> mask_out = std::nullopt,</div>
|
|
<div class="line"><a id="l01208" name="l01208"></a><span class="lineno"> 1208</span> std::optional<array> mask_lhs = std::nullopt,</div>
|
|
<div class="line"><a id="l01209" name="l01209"></a><span class="lineno"> 1209</span> std::optional<array> mask_rhs = std::nullopt,</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#gaf5d82380cd204e7c33324cebcd1461ce"> 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#gaf5d82380cd204e7c33324cebcd1461ce">block_sparse_mm</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> a,</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> b,</div>
|
|
<div class="line"><a id="l01216" name="l01216"></a><span class="lineno"> 1216</span> std::optional<array> lhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01217" name="l01217"></a><span class="lineno"> 1217</span> std::optional<array> rhs_indices = std::nullopt,</div>
|
|
<div class="line"><a id="l01218" name="l01218"></a><span class="lineno"> 1218</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01219" name="l01219"></a><span class="lineno"> 1219</span> </div>
|
|
<div class="line"><a id="l01221" name="l01221"></a><span class="lineno"><a class="line" href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6"> 1221</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="l01222" name="l01222"></a><span class="lineno"> 1222</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="l01223" name="l01223"></a><span class="lineno"> 1223</span> <span class="keywordtype">int</span> offset = 0,</div>
|
|
<div class="line"><a id="l01224" name="l01224"></a><span class="lineno"> 1224</span> <span class="keywordtype">int</span> axis1 = 0,</div>
|
|
<div class="line"><a id="l01225" name="l01225"></a><span class="lineno"> 1225</span> <span class="keywordtype">int</span> axis2 = 1,</div>
|
|
<div class="line"><a id="l01226" name="l01226"></a><span class="lineno"> 1226</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01227" name="l01227"></a><span class="lineno"> 1227</span> </div>
|
|
<div class="line"><a id="l01229" name="l01229"></a><span class="lineno"><a class="line" href="group__ops.html#ga11af511875640e1fa88e0ca87e199344"> 1229</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="l01230" name="l01230"></a><span class="lineno"> 1230</span> </div>
|
|
<div class="line"><a id="l01236" name="l01236"></a><span class="lineno"><a class="line" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759"> 1236</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759">depends</a>(</div>
|
|
<div class="line"><a id="l01237" name="l01237"></a><span class="lineno"> 1237</span> <span class="keyword">const</span> std::vector<array>& inputs,</div>
|
|
<div class="line"><a id="l01238" name="l01238"></a><span class="lineno"> 1238</span> <span class="keyword">const</span> std::vector<array>& dependencies);</div>
|
|
<div class="line"><a id="l01239" name="l01239"></a><span class="lineno"> 1239</span> </div>
|
|
<div class="line"><a id="l01241" name="l01241"></a><span class="lineno"><a class="line" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396"> 1241</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="l01242" name="l01242"></a><span class="lineno"><a class="line" href="group__ops.html#ga08ca172ce80157c916c89dd0b45b95c5"> 1242</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaba4d25e7a2bf87ba4feb7837ec7fa396">atleast_1d</a>(</div>
|
|
<div class="line"><a id="l01243" name="l01243"></a><span class="lineno"> 1243</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
|
<div class="line"><a id="l01244" name="l01244"></a><span class="lineno"> 1244</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01245" name="l01245"></a><span class="lineno"><a class="line" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087"> 1245</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="l01246" name="l01246"></a><span class="lineno"><a class="line" href="group__ops.html#ga9950299a80c2562f13448758f856d1f5"> 1246</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">atleast_2d</a>(</div>
|
|
<div class="line"><a id="l01247" name="l01247"></a><span class="lineno"> 1247</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
|
<div class="line"><a id="l01248" name="l01248"></a><span class="lineno"> 1248</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01249" name="l01249"></a><span class="lineno"><a class="line" href="group__ops.html#ga4afd919601e67782ff964465919956a0"> 1249</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="l01250" name="l01250"></a><span class="lineno"><a class="line" href="group__ops.html#gaffdf742ad79440a60dda40062a8074fe"> 1250</a></span>std::vector<array> <a class="code hl_function" href="group__ops.html#ga4afd919601e67782ff964465919956a0">atleast_3d</a>(</div>
|
|
<div class="line"><a id="l01251" name="l01251"></a><span class="lineno"> 1251</span> <span class="keyword">const</span> std::vector<array>& a,</div>
|
|
<div class="line"><a id="l01252" name="l01252"></a><span class="lineno"> 1252</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01253" name="l01253"></a><span class="lineno"> 1253</span> </div>
|
|
<div class="line"><a id="l01258" name="l01258"></a><span class="lineno"><a class="line" href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636"> 1258</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="l01259" name="l01259"></a><span class="lineno"> 1259</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="l01260" name="l01260"></a><span class="lineno"> 1260</span> std::vector<int> axes,</div>
|
|
<div class="line"><a id="l01261" name="l01261"></a><span class="lineno"> 1261</span> <span class="keywordtype">bool</span> inverted,</div>
|
|
<div class="line"><a id="l01262" name="l01262"></a><span class="lineno"> 1262</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="l01263" name="l01263"></a><span class="lineno"> 1263</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
|
<div class="line"><a id="l01264" name="l01264"></a><span class="lineno"> 1264</span> </div>
|
|
<div class="line"><a id="l01265" name="l01265"></a><span class="lineno"><a class="line" href="group__ops.html#ga5b596906bf8cdc8d97ed6ddc9aeb4c23"> 1265</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="l01266" name="l01266"></a><span class="lineno"> 1266</span> </div>
|
|
<div class="line"><a id="l01268" name="l01268"></a><span class="lineno"><a class="line" href="group__ops.html#ga752fd2707dabb05d0308ba3d55346ada"> 1268</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="l01269" name="l01269"></a><span class="lineno"><a class="line" href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff"> 1269</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="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#ga8af4f22c08c11c4ffab7e3d45e0f3cd6"> 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#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="l01273" name="l01273"></a><span class="lineno"><a class="line" href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02"> 1273</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="l01274" name="l01274"></a><span class="lineno"> 1274</span> </div>
|
|
<div class="line"><a id="l01276" name="l01276"></a><span class="lineno"><a class="line" href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b"> 1276</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="l01277" name="l01277"></a><span class="lineno"><a class="line" href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181"> 1277</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="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#ga89682bf78491761e062d4ee7bef0c829"> 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#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="l01281" name="l01281"></a><span class="lineno"><a class="line" href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e"> 1281</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="l01282" name="l01282"></a><span class="lineno"> 1282</span> </div>
|
|
<div class="line"><a id="l01284" name="l01284"></a><span class="lineno"><a class="line" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23"> 1284</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="l01285" name="l01285"></a><span class="lineno"><a class="line" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a"> 1285</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="l01286" name="l01286"></a><span class="lineno"> 1286</span> </div>
|
|
<div class="line"><a id="l01289" name="l01289"></a><span class="lineno"> 1289</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:20</div></div>
|
|
<div class="ttc" id="adevice_8h_html"><div class="ttname"><a href="device_8h.html">device.h</a></div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga05881a4157cd113c9392d168a79e6673"><div class="ttname"><a href="group__ops.html#ga05881a4157cd113c9392d168a79e6673">mlx::core::scatter_max</a></div><div class="ttdeci">array scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and max updates to given linear indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga05b4c6054d028107869511f927da01cd"><div class="ttname"><a href="group__ops.html#ga05b4c6054d028107869511f927da01cd">mlx::core::floor_divide</a></div><div class="ttdeci">array floor_divide(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute integer division.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga07aa8059adba5b9a8818027b8aafd31e"><div class="ttname"><a href="group__ops.html#ga07aa8059adba5b9a8818027b8aafd31e">mlx::core::radians</a></div><div class="ttdeci">array radians(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Convert the elements of an array from Degrees to Radians.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga08bec7cb10c84466487b507fc5bf9776"><div class="ttname"><a href="group__ops.html#ga08bec7cb10c84466487b507fc5bf9776">mlx::core::arccos</a></div><div class="ttdeci">array arccos(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Arc Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0ca16b7579dfc899f3f7fd40245ba7c5"><div class="ttname"><a href="group__ops.html#ga0ca16b7579dfc899f3f7fd40245ba7c5">mlx::core::scatter_min</a></div><div class="ttdeci">array scatter_min(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and min updates to given linear indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0d49e0c7011d0573c369c13c8f045a09"><div class="ttname"><a href="group__ops.html#ga0d49e0c7011d0573c369c13c8f045a09">mlx::core::less_equal</a></div><div class="ttdeci">array less_equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a <= b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0d71dfbc14ef3ed564b0c5ee26af680f"><div class="ttname"><a href="group__ops.html#ga0d71dfbc14ef3ed564b0c5ee26af680f">mlx::core::cumprod</a></div><div class="ttdeci">array cumprod(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative product of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga0e58c24fc5668e5a521e5b45e8370a62"><div class="ttname"><a href="group__ops.html#ga0e58c24fc5668e5a521e5b45e8370a62">mlx::core::astype</a></div><div class="ttdeci">array astype(array a, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Convert an array to the given data type.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga102f23aa0b0c3d3296a321c694617aa1"><div class="ttname"><a href="group__ops.html#ga102f23aa0b0c3d3296a321c694617aa1">mlx::core::rsqrt</a></div><div class="ttdeci">array rsqrt(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Square root and reciprocal the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga11af511875640e1fa88e0ca87e199344"><div class="ttname"><a href="group__ops.html#ga11af511875640e1fa88e0ca87e199344">mlx::core::diag</a></div><div class="ttdeci">array diag(const array &a, int k=0, StreamOrDevice s={})</div><div class="ttdoc">Extract diagonal from a 2d array or create a diagonal matrix.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1234e4c39cfa79f19d4bdb5b8ea4d45e"><div class="ttname"><a href="group__ops.html#ga1234e4c39cfa79f19d4bdb5b8ea4d45e">mlx::core::square</a></div><div class="ttdeci">array square(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Square the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1404ecceff83fd9b9139b7520f55e096"><div class="ttname"><a href="group__ops.html#ga1404ecceff83fd9b9139b7520f55e096">mlx::core::ceil</a></div><div class="ttdeci">array ceil(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Ceil the element of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga144228d7222d15af3a135b8b0f3fa21b"><div class="ttname"><a href="group__ops.html#ga144228d7222d15af3a135b8b0f3fa21b">mlx::core::log2</a></div><div class="ttdeci">array log2(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Log base 2 of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga157cd7c23f9b306fee2e1eb2b9bf1dd8"><div class="ttname"><a href="group__ops.html#ga157cd7c23f9b306fee2e1eb2b9bf1dd8">mlx::core::clip</a></div><div class="ttdeci">array clip(const array &a, const std::optional< array > &a_min=std::nullopt, const std::optional< array > &a_max=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Clip (limit) the values in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga175592792471b0ffb45196dca4711ba6"><div class="ttname"><a href="group__ops.html#ga175592792471b0ffb45196dca4711ba6">mlx::core::isnan</a></div><div class="ttdeci">array isnan(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1940523da381ed7be50656a3bc465ff3"><div class="ttname"><a href="group__ops.html#ga1940523da381ed7be50656a3bc465ff3">mlx::core::isneginf</a></div><div class="ttdeci">array isneginf(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga196c240d3d0fcbb4713802c485e15133"><div class="ttname"><a href="group__ops.html#ga196c240d3d0fcbb4713802c485e15133">mlx::core::subtract</a></div><div class="ttdeci">array subtract(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Subtract two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga19c1bf6929fe8d66b9cd408946aea6a8"><div class="ttname"><a href="group__ops.html#ga19c1bf6929fe8d66b9cd408946aea6a8">mlx::core::cummin</a></div><div class="ttdeci">array cummin(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative min of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga1fdcc7fc8819caf2e6f1c327ed4e9b9e"><div class="ttname"><a href="group__ops.html#ga1fdcc7fc8819caf2e6f1c327ed4e9b9e">mlx::core::log10</a></div><div class="ttdeci">array log10(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Log base 10 of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga20a1f4270c35b0fa544f5105a87a1604"><div class="ttname"><a href="group__ops.html#ga20a1f4270c35b0fa544f5105a87a1604">mlx::core::log1p</a></div><div class="ttdeci">array log1p(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Natural logarithm of one plus elements in the array: log(1 + a).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga20f1a1a8c0cd6206485f9363f3915faa"><div class="ttname"><a href="group__ops.html#ga20f1a1a8c0cd6206485f9363f3915faa">mlx::core::sign</a></div><div class="ttdeci">array sign(const array &a, StreamOrDevice s={})</div><div class="ttdoc">The sign of the elements in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2181b71cda88007a3092be4795ff0715"><div class="ttname"><a href="group__ops.html#ga2181b71cda88007a3092be4795ff0715">mlx::core::cosh</a></div><div class="ttdeci">array cosh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga2236e5dfc7e52e28abf6c21675d0a51e"><div class="ttname"><a href="group__ops.html#ga2236e5dfc7e52e28abf6c21675d0a51e">mlx::core::conv_general</a></div><div class="ttdeci">array conv_general(array input, array weight, std::vector< int > stride={}, std::vector< int > padding_lo={}, std::vector< int > padding_hi={}, std::vector< int > kernel_dilation={}, std::vector< int > input_dilation={}, int groups=1, bool flip=false, StreamOrDevice s={})</div><div class="ttdoc">General convolution with a filter.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga23768728e4dd070c917fbb0ed0d0c2ec"><div class="ttname"><a href="group__ops.html#ga23768728e4dd070c917fbb0ed0d0c2ec">mlx::core::logical_or</a></div><div class="ttdeci">array logical_or(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Logical or of two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga24067d10a842db2c9d509ea48135a2c3"><div class="ttname"><a href="group__ops.html#ga24067d10a842db2c9d509ea48135a2c3">mlx::core::moveaxis</a></div><div class="ttdeci">array moveaxis(const array &a, int source, int destination, StreamOrDevice s={})</div><div class="ttdoc">Move an axis of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga26c33f5cdb6fc10d272acd6e208034e0"><div class="ttname"><a href="group__ops.html#ga26c33f5cdb6fc10d272acd6e208034e0">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_ga26e5a043eaaaf066d1400adac9c11d0c"><div class="ttname"><a href="group__ops.html#ga26e5a043eaaaf066d1400adac9c11d0c">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_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_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>
|
|
<div class="ttc" id="agroup__ops_html_ga2d74d43f007a069384e89d8416525331"><div class="ttname"><a href="group__ops.html#ga2d74d43f007a069384e89d8416525331">mlx::core::round</a></div><div class="ttdeci">array round(const array &a, int decimals, StreamOrDevice s={})</div><div class="ttdoc">Round a floating point number.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga30d47e08093c03a3676f235f9f559411"><div class="ttname"><a href="group__ops.html#ga30d47e08093c03a3676f235f9f559411">mlx::core::conv1d</a></div><div class="ttdeci">array conv1d(const array &input, const array &weight, int stride=1, int padding=0, int dilation=1, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">1D convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3188638fba3a60e264baf69956a1e08b"><div class="ttname"><a href="group__ops.html#ga3188638fba3a60e264baf69956a1e08b">mlx::core::bitwise_xor</a></div><div class="ttdeci">array bitwise_xor(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Bitwise exclusive or.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga33638dc3a9972dd02be12d0eb85f9bde"><div class="ttname"><a href="group__ops.html#ga33638dc3a9972dd02be12d0eb85f9bde">mlx::core::equal</a></div><div class="ttdeci">array 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_ga34573eb2ead179839579142f3aedb3b8"><div class="ttname"><a href="group__ops.html#ga34573eb2ead179839579142f3aedb3b8">mlx::core::zeros</a></div><div class="ttdeci">array zeros(const std::vector< int > &shape, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with zeros.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga36bc28f1deb2fe668ca9ae1e447b6b1f"><div class="ttname"><a href="group__ops.html#ga36bc28f1deb2fe668ca9ae1e447b6b1f">mlx::core::stop_gradient</a></div><div class="ttdeci">array stop_gradient(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Stop the flow of gradients.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3708b5bcb61e2c63d213c4ce6ad0ffc0"><div class="ttname"><a href="group__ops.html#ga3708b5bcb61e2c63d213c4ce6ad0ffc0">mlx::core::scatter_prod</a></div><div class="ttdeci">array scatter_prod(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and prod updates to given indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3875660e4ce2c8add8bfcf8144078708"><div class="ttname"><a href="group__ops.html#ga3875660e4ce2c8add8bfcf8144078708">mlx::core::slice_update</a></div><div class="ttdeci">array slice_update(const array &src, const array &update, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})</div><div class="ttdoc">Update a slice from the source array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga39dfdf72b556012aa35ff27a94116e74"><div class="ttname"><a href="group__ops.html#ga39dfdf72b556012aa35ff27a94116e74">mlx::core::cos</a></div><div class="ttdeci">array cos(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3a41895f25ed083a36994d95fa102546"><div class="ttname"><a href="group__ops.html#ga3a41895f25ed083a36994d95fa102546">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:337</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3a70569b50e1083c5ded199d73fb960c"><div class="ttname"><a href="group__ops.html#ga3a70569b50e1083c5ded199d73fb960c">mlx::core::degrees</a></div><div class="ttdeci">array degrees(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Convert the elements of an array from Radians to Degrees.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3b1b90ef1275ca17655b6d7f25d3ee68"><div class="ttname"><a href="group__ops.html#ga3b1b90ef1275ca17655b6d7f25d3ee68">mlx::core::all</a></div><div class="ttdeci">array all(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">True if all elements in the array are true (or non-zero).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga3f10e89a4bcb1a8fa44fb33b8d1176a5"><div class="ttname"><a href="group__ops.html#ga3f10e89a4bcb1a8fa44fb33b8d1176a5">mlx::core::tan</a></div><div class="ttdeci">array tan(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Tangent of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga45e9e68246b0d1cf03c3cc9c9e7e6ae3"><div class="ttname"><a href="group__ops.html#ga45e9e68246b0d1cf03c3cc9c9e7e6ae3">mlx::core::eye</a></div><div class="ttdeci">array eye(int n, int m, int k, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape (n,m) with ones in the specified diagonal k, and zeros everywhere el...</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga484eaa10d5e19a4ca46d3a9cd9fab600"><div class="ttname"><a href="group__ops.html#ga484eaa10d5e19a4ca46d3a9cd9fab600">mlx::core::identity</a></div><div class="ttdeci">array identity(int n, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Create a square matrix of shape (n,n) of zeros, and ones in the major diagonal.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga498b61f7e8f056ae00297fa0dc17303a"><div class="ttname"><a href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a">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_ga49ba00c090f81f331c91b0c97040bce0"><div class="ttname"><a href="group__ops.html#ga49ba00c090f81f331c91b0c97040bce0">mlx::core::minimum</a></div><div class="ttdeci">array minimum(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Element-wise minimum between two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4a09b7241d564d92548bc2773eb1d544"><div class="ttname"><a href="group__ops.html#ga4a09b7241d564d92548bc2773eb1d544">mlx::core::prod</a></div><div class="ttdeci">array prod(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The product of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4afd919601e67782ff964465919956a0"><div class="ttname"><a href="group__ops.html#ga4afd919601e67782ff964465919956a0">mlx::core::atleast_3d</a></div><div class="ttdeci">array atleast_3d(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4c8b8a1632944acaae50f0de6c23ece6"><div class="ttname"><a href="group__ops.html#ga4c8b8a1632944acaae50f0de6c23ece6">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:365</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4d29556bb93e2f66916116cf1f062b36"><div class="ttname"><a href="group__ops.html#ga4d29556bb93e2f66916116cf1f062b36">mlx::core::reciprocal</a></div><div class="ttdeci">array reciprocal(const array &a, StreamOrDevice s={})</div><div class="ttdoc">The reciprocal (1/x) of the elements in an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4f3389e5b89e70e862e7d2b40d6c7f78"><div class="ttname"><a href="group__ops.html#ga4f3389e5b89e70e862e7d2b40d6c7f78">mlx::core::tri</a></div><div class="ttdeci">array tri(int n, int m, int k, Dtype type, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga4fd04d6472ae7cdb94a174edfe338ea3"><div class="ttname"><a href="group__ops.html#ga4fd04d6472ae7cdb94a174edfe338ea3">mlx::core::pad</a></div><div class="ttdeci">array pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={})</div><div class="ttdoc">Pad an array with a constant value.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga50aa98754b412bb57c083f6e3e95061f"><div class="ttname"><a href="group__ops.html#ga50aa98754b412bb57c083f6e3e95061f">mlx::core::flatten</a></div><div class="ttdeci">array flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})</div><div class="ttdoc">Flatten the dimensions in the range [start_axis, end_axis] .</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga51eac95c04400921c54716de14b52491"><div class="ttname"><a href="group__ops.html#ga51eac95c04400921c54716de14b52491">mlx::core::isclose</a></div><div class="ttdeci">array isclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})</div><div class="ttdoc">Returns a boolean array where two arrays are element-wise equal within the specified tolerance.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga52392a2a98f09a80da8d338c4908bd02"><div class="ttname"><a href="group__ops.html#ga52392a2a98f09a80da8d338c4908bd02">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_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_ga54eeed455321a54c8e72e16552a978f2"><div class="ttname"><a href="group__ops.html#ga54eeed455321a54c8e72e16552a978f2">mlx::core::ones</a></div><div class="ttdeci">array ones(const std::vector< int > &shape, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with ones.</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_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_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_ga6381c7062b232ac21f3fdc3136aaf86e"><div class="ttname"><a href="group__ops.html#ga6381c7062b232ac21f3fdc3136aaf86e">mlx::core::quantized_matmul</a></div><div class="ttdeci">array quantized_matmul(const array &x, const array &w, const array &scales, const array &biases, bool transpose=true, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Quantized matmul multiplies x with a quantized matrix w.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_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>
|
|
<div class="ttc" id="agroup__ops_html_ga6b76c8ea46b19e6866af155fa5910be6"><div class="ttname"><a href="group__ops.html#ga6b76c8ea46b19e6866af155fa5910be6">mlx::core::block_masked_mm</a></div><div class="ttdeci">array block_masked_mm(array a, array b, int block_size, std::optional< array > mask_out=std::nullopt, std::optional< array > mask_lhs=std::nullopt, std::optional< array > mask_rhs=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix product with block masking.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6caba9c92b5989123501f909cc7da354"><div class="ttname"><a href="group__ops.html#ga6caba9c92b5989123501f909cc7da354">mlx::core::arctan2</a></div><div class="ttdeci">array arctan2(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Inverse tangent of the ratio of two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6d5f5f72362488b956cdc4615ef6c636"><div class="ttname"><a href="group__ops.html#ga6d5f5f72362488b956cdc4615ef6c636">mlx::core::number_of_elements</a></div><div class="ttdeci">array number_of_elements(const array &a, std::vector< int > axes, bool inverted, Dtype dtype=int32, StreamOrDevice s={})</div><div class="ttdoc">Extract the number of elements along some axes as a scalar array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6e9907d2f14dc4803e4306b3dbc4b3ca"><div class="ttname"><a href="group__ops.html#ga6e9907d2f14dc4803e4306b3dbc4b3ca">mlx::core::conv3d</a></div><div class="ttdeci">array conv3d(const array &input, const array &weight, const std::tuple< int, int, int > &stride={1, 1, 1}, const std::tuple< int, int, int > &padding={0, 0, 0}, const std::tuple< int, int, int > &dilation={1, 1, 1}, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">3D convolution with a filter</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga6fb22d4926133573e430fcc92f4eef31"><div class="ttname"><a href="group__ops.html#ga6fb22d4926133573e430fcc92f4eef31">mlx::core::log</a></div><div class="ttdeci">array log(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Natural logarithm of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga708abf8f79609cd6831db7c38cafac0e"><div class="ttname"><a href="group__ops.html#ga708abf8f79609cd6831db7c38cafac0e">mlx::core::sigmoid</a></div><div class="ttdeci">array sigmoid(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Element-wise logistic sigmoid of the array: 1 / (1 + exp(-x).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga710daa7ec721bd4d3f326082cb195576"><div class="ttname"><a href="group__ops.html#ga710daa7ec721bd4d3f326082cb195576">mlx::core::squeeze</a></div><div class="ttdeci">array squeeze(const array &a, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Remove singleton dimensions at the given axes.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7153071bcfff6faad21332163fb9a430"><div class="ttname"><a href="group__ops.html#ga7153071bcfff6faad21332163fb9a430">mlx::core::greater_equal</a></div><div class="ttdeci">array greater_equal(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a >= b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga717f11149a8c7b4cc3e33bbcc0a97133"><div class="ttname"><a href="group__ops.html#ga717f11149a8c7b4cc3e33bbcc0a97133">mlx::core::expand_dims</a></div><div class="ttdeci">array expand_dims(const array &a, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Add a singleton dimension at the given axes.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga73b02833229678786e7f302d458d5a83"><div class="ttname"><a href="group__ops.html#ga73b02833229678786e7f302d458d5a83">mlx::core::conv2d</a></div><div class="ttdeci">array conv2d(const array &input, const array &weight, const std::pair< int, int > &stride={1, 1}, const std::pair< int, int > &padding={0, 0}, const std::pair< int, int > &dilation={1, 1}, int groups=1, StreamOrDevice s={})</div><div class="ttdoc">2D convolution with a filter</div></div>
|
|
<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:323</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>
|
|
<div class="ttc" id="agroup__ops_html_ga7972058715c26559dff9c9ae2a3ef76d"><div class="ttname"><a href="group__ops.html#ga7972058715c26559dff9c9ae2a3ef76d">mlx::core::power</a></div><div class="ttdeci">array power(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Raise elements of a to the power of b element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7ade2ea305e2e4219c3609443fb5db8d"><div class="ttname"><a href="group__ops.html#ga7ade2ea305e2e4219c3609443fb5db8d">mlx::core::maximum</a></div><div class="ttdeci">array maximum(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Element-wise maximum between two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7b0e3d46a39d99b1628b422049790082"><div class="ttname"><a href="group__ops.html#ga7b0e3d46a39d99b1628b422049790082">mlx::core::reshape</a></div><div class="ttdeci">array reshape(const array &a, std::vector< int > shape, StreamOrDevice s={})</div><div class="ttdoc">Reshape an array to the given shape.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7c3bd5ef430a71dfd298e626741e3c71"><div class="ttname"><a href="group__ops.html#ga7c3bd5ef430a71dfd298e626741e3c71">mlx::core::argmin</a></div><div class="ttdeci">array argmin(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Returns the index of the minimum value in the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7e133df686439588a8cd1fb10ce0c6e9"><div class="ttname"><a href="group__ops.html#ga7e133df686439588a8cd1fb10ce0c6e9">mlx::core::var</a></div><div class="ttdeci">array var(const array &a, bool keepdims, int ddof=0, StreamOrDevice s={})</div><div class="ttdoc">Computes the variance of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7e85a476530c96ab73b894d1bddacf84"><div class="ttname"><a href="group__ops.html#ga7e85a476530c96ab73b894d1bddacf84">mlx::core::full</a></div><div class="ttdeci">array full(std::vector< int > shape, array vals, Dtype dtype, StreamOrDevice s={})</div><div class="ttdoc">Fill an array of the given shape with the given value(s).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7e9bb08b43c8fd0444b7d3c9e09dc1c6"><div class="ttname"><a href="group__ops.html#ga7e9bb08b43c8fd0444b7d3c9e09dc1c6">mlx::core::softmax</a></div><div class="ttdeci">array softmax(const array &a, const std::vector< int > &axes, bool precise=false, StreamOrDevice s={})</div><div class="ttdoc">Softmax of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7fb616054665b3c2d61fa234f501f079"><div class="ttname"><a href="group__ops.html#ga7fb616054665b3c2d61fa234f501f079">mlx::core::sort</a></div><div class="ttdeci">array sort(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Returns a sorted copy of the flattened array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga7fed87d96cc7741d8267f4eac83f5fe7"><div class="ttname"><a href="group__ops.html#ga7fed87d96cc7741d8267f4eac83f5fe7">mlx::core::max</a></div><div class="ttdeci">array max(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The maximum of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga82a53e083205a965387b3c3e2463244a"><div class="ttname"><a href="group__ops.html#ga82a53e083205a965387b3c3e2463244a">mlx::core::addmm</a></div><div class="ttdeci">array addmm(array c, array a, array b, const float &alpha=1.f, const float &beta=1.f, StreamOrDevice s={})</div><div class="ttdoc">Compute D = beta * C + alpha * (A @ B)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga83e0bb45dc770cf014531d873b78c5a2"><div class="ttname"><a href="group__ops.html#ga83e0bb45dc770cf014531d873b78c5a2">mlx::core::tril</a></div><div class="ttdeci">array tril(array x, int k=0, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8598dd718fb05cb28535e250372d4e6f"><div class="ttname"><a href="group__ops.html#ga8598dd718fb05cb28535e250372d4e6f">mlx::core::any</a></div><div class="ttdeci">array any(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">True if any elements in the array are true (or non-zero).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga866af24e10db2797e1c5a5986dbf6c0d"><div class="ttname"><a href="group__ops.html#ga866af24e10db2797e1c5a5986dbf6c0d">mlx::core::outer</a></div><div class="ttdeci">array outer(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the outer product of two vectors.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8770e8c8f23f13343911f4c9d6e1c619"><div class="ttname"><a href="group__ops.html#ga8770e8c8f23f13343911f4c9d6e1c619">mlx::core::arcsin</a></div><div class="ttdeci">array arcsin(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Arc Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga89682bf78491761e062d4ee7bef0c829"><div class="ttname"><a href="group__ops.html#ga89682bf78491761e062d4ee7bef0c829">mlx::core::left_shift</a></div><div class="ttdeci">array left_shift(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Shift bits to the left.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8a2056f8c9bb30914c40bcf509386491"><div class="ttname"><a href="group__ops.html#ga8a2056f8c9bb30914c40bcf509386491">mlx::core::where</a></div><div class="ttdeci">array where(const array &condition, const array &x, const array &y, StreamOrDevice s={})</div><div class="ttdoc">Select from x or y depending on condition.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8a3b04e23e347d99ecf411fd6f4e5125"><div class="ttname"><a href="group__ops.html#ga8a3b04e23e347d99ecf411fd6f4e5125">mlx::core::exp</a></div><div class="ttdeci">array exp(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Exponential of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8af4f22c08c11c4ffab7e3d45e0f3cd6"><div class="ttname"><a href="group__ops.html#ga8af4f22c08c11c4ffab7e3d45e0f3cd6">mlx::core::bitwise_or</a></div><div class="ttdeci">array bitwise_or(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Bitwise inclusive or.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8d656904aa2690b60955ae745aecfc30"><div class="ttname"><a href="group__ops.html#ga8d656904aa2690b60955ae745aecfc30">mlx::core::floor</a></div><div class="ttdeci">array floor(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Floor the element of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8de80ecef30fc560003d40f61a38b99d"><div class="ttname"><a href="group__ops.html#ga8de80ecef30fc560003d40f61a38b99d">mlx::core::as_strided</a></div><div class="ttdeci">array as_strided(array a, std::vector< int > shape, std::vector< size_t > 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_ga8df3b2703bf671457422894dd870cdc5"><div class="ttname"><a href="group__ops.html#ga8df3b2703bf671457422894dd870cdc5">mlx::core::argsort</a></div><div class="ttdeci">array argsort(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Returns indices that sort the flattened array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8f3059336ee0c87207b1f8c6ab312645"><div class="ttname"><a href="group__ops.html#ga8f3059336ee0c87207b1f8c6ab312645">mlx::core::array_equal</a></div><div class="ttdeci">array array_equal(const array &a, const array &b, bool equal_nan, StreamOrDevice s={})</div><div class="ttdoc">True if two arrays have the same shape and elements.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga8fc238d5e5d1153e69da8b36015d9844"><div class="ttname"><a href="group__ops.html#ga8fc238d5e5d1153e69da8b36015d9844">mlx::core::isinf</a></div><div class="ttdeci">array isinf(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga9142b8d717699a8abfa2a7398891ff8a"><div class="ttname"><a href="group__ops.html#ga9142b8d717699a8abfa2a7398891ff8a">mlx::core::less</a></div><div class="ttdeci">array less(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a < b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga9236b085a88ead3128ed8079d009cac6"><div class="ttname"><a href="group__ops.html#ga9236b085a88ead3128ed8079d009cac6">mlx::core::diagonal</a></div><div class="ttdeci">array diagonal(const array &a, int offset=0, int axis1=0, int axis2=1, StreamOrDevice s={})</div><div class="ttdoc">Extract a diagonal or construct a diagonal array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga94f8d3b1906fee99da9cbe39f7be7d42"><div class="ttname"><a href="group__ops.html#ga94f8d3b1906fee99da9cbe39f7be7d42">mlx::core::ones_like</a></div><div class="ttdeci">array ones_like(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga95d9a9425533b5ed1707eb00184dffc6"><div class="ttname"><a href="group__ops.html#ga95d9a9425533b5ed1707eb00184dffc6">mlx::core::negative</a></div><div class="ttdeci">array negative(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Negate an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga968bcabed902311dcfbd903b0fb886ec"><div class="ttname"><a href="group__ops.html#ga968bcabed902311dcfbd903b0fb886ec">mlx::core::linspace</a></div><div class="ttdeci">array linspace(double start, double stop, int num=50, Dtype dtype=float32, StreamOrDevice s={})</div><div class="ttdoc">A 1D array of num evenly spaced numbers in the range [start, stop]</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga99f5c904f724156a814d7817188351d2"><div class="ttname"><a href="group__ops.html#ga99f5c904f724156a814d7817188351d2">mlx::core::remainder</a></div><div class="ttdeci">array remainder(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the element-wise remainder of division.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_ga9aaa4381942a6d9a31d2ab8af8436e78"><div class="ttname"><a href="group__ops.html#ga9aaa4381942a6d9a31d2ab8af8436e78">mlx::core::block_sparse_qmm</a></div><div class="ttdeci">array block_sparse_qmm(const array &x, const array &w, const array &scales, const array &biases, std::optional< array > lhs_indices=std::nullopt, std::optional< array > rhs_indices=std::nullopt, bool transpose=true, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix products with matrix-level gather.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa041f3f070e68f4946db07516b7d092e"><div class="ttname"><a href="group__ops.html#gaa041f3f070e68f4946db07516b7d092e">mlx::core::arctan</a></div><div class="ttdeci">array arctan(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Arc Tangent of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa30ebc0a8376dbc3f7e46a47052b5894"><div class="ttname"><a href="group__ops.html#gaa30ebc0a8376dbc3f7e46a47052b5894">mlx::core::divmod</a></div><div class="ttdeci">std::vector< array > divmod(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Compute the element-wise quotient and remainder.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaa9df5917876eeb0cb28b7fa81f880412"><div class="ttname"><a href="group__ops.html#gaa9df5917876eeb0cb28b7fa81f880412">mlx::core::triu</a></div><div class="ttdeci">array triu(array x, int k=0, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaafafcfcebdf7248679c8543d0c0497e5"><div class="ttname"><a href="group__ops.html#gaafafcfcebdf7248679c8543d0c0497e5">mlx::core::arccosh</a></div><div class="ttdeci">array arccosh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Inverse Hyperbolic Cosine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gab105a57b9a4d84496fe1e4d60e13d361"><div class="ttname"><a href="group__ops.html#gab105a57b9a4d84496fe1e4d60e13d361">mlx::core::tile</a></div><div class="ttdeci">array tile(const array &arr, std::vector< int > reps, StreamOrDevice s={})</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_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_gab6e7f655a9ff15350ca5379692f9d444"><div class="ttname"><a href="group__ops.html#gab6e7f655a9ff15350ca5379692f9d444">mlx::core::gather</a></div><div class="ttdeci">array gather(const array &a, const std::vector< array > &indices, const std::vector< int > &axes, const std::vector< int > &slice_sizes, StreamOrDevice s={})</div><div class="ttdoc">Gather array entries given indices and slices.</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_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>
|
|
<div class="ttc" id="agroup__ops_html_gabdc36fa65697d0361c8d67495de77129"><div class="ttname"><a href="group__ops.html#gabdc36fa65697d0361c8d67495de77129">mlx::core::concatenate</a></div><div class="ttdeci">array concatenate(const std::vector< array > &arrays, int axis, StreamOrDevice s={})</div><div class="ttdoc">Concatenate arrays along a given axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gabff758a5c1ce32ad7e8b78aba0164077"><div class="ttname"><a href="group__ops.html#gabff758a5c1ce32ad7e8b78aba0164077">mlx::core::dequantize</a></div><div class="ttdeci">array dequantize(const array &w, const array &scales, const array &biases, int group_size=64, int bits=4, StreamOrDevice s={})</div><div class="ttdoc">Dequantize a matrix produced by quantize()</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac1869f3b7094869b44fe7ac4ce58638b"><div class="ttname"><a href="group__ops.html#gac1869f3b7094869b44fe7ac4ce58638b">mlx::core::transpose</a></div><div class="ttdeci">array transpose(const array &a, std::vector< int > axes, StreamOrDevice s={})</div><div class="ttdoc">Permutes the dimensions according to the given axes.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac1b30830a972fb9a2601379ad2b32405"><div class="ttname"><a href="group__ops.html#gac1b30830a972fb9a2601379ad2b32405">mlx::core::partition</a></div><div class="ttdeci">array partition(const array &a, int kth, StreamOrDevice s={})</div><div class="ttdoc">Returns a partitioned copy of the flattened array such that the smaller kth elements are first.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac2fc270882fcfa81eb8bd068cc0d86d7"><div class="ttname"><a href="group__ops.html#gac2fc270882fcfa81eb8bd068cc0d86d7">mlx::core::take</a></div><div class="ttdeci">array take(const array &a, const array &indices, int axis, StreamOrDevice s={})</div><div class="ttdoc">Take array slices at the given indices of the specified axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac3a6fe18694e84b3d63458e9553ac181"><div class="ttname"><a href="group__ops.html#gac3a6fe18694e84b3d63458e9553ac181">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_gac4a51a68fbe1725436b026d2fbb95759"><div class="ttname"><a href="group__ops.html#gac4a51a68fbe1725436b026d2fbb95759">mlx::core::depends</a></div><div class="ttdeci">std::vector< array > depends(const std::vector< array > &inputs, const std::vector< array > &dependencies)</div><div class="ttdoc">Implements the identity function but allows injecting dependencies to other arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gac62e2cedc49ef2c90dd8584000317450"><div class="ttname"><a href="group__ops.html#gac62e2cedc49ef2c90dd8584000317450">mlx::core::arcsinh</a></div><div class="ttdeci">array arcsinh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Inverse Hyperbolic Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gacd14c2b5cfebf343fc2d672722f8d174"><div class="ttname"><a href="group__ops.html#gacd14c2b5cfebf343fc2d672722f8d174">mlx::core::scatter_add</a></div><div class="ttdeci">array scatter_add(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter and add updates to given indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gacff4eb57c085d571e722083680267ac5"><div class="ttname"><a href="group__ops.html#gacff4eb57c085d571e722083680267ac5">mlx::core::logsumexp</a></div><div class="ttdeci">array logsumexp(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">The logsumexp of all elements of the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad256e86cc1a6e6b3832e392baa90318d"><div class="ttname"><a href="group__ops.html#gad256e86cc1a6e6b3832e392baa90318d">mlx::core::broadcast_to</a></div><div class="ttdeci">array broadcast_to(const array &a, const std::vector< int > &shape, StreamOrDevice s={})</div><div class="ttdoc">Broadcast an array to a given shape.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad438be8f90bae9d37c6853b8f4225d61"><div class="ttname"><a href="group__ops.html#gad438be8f90bae9d37c6853b8f4225d61">mlx::core::scatter</a></div><div class="ttdeci">array scatter(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})</div><div class="ttdoc">Scatter updates to given linear indices.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad656c30f9fd7d9467e405657b325aa7e"><div class="ttname"><a href="group__ops.html#gad656c30f9fd7d9467e405657b325aa7e">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_gad66135407dbb41b3c5d2cdfd51226c21"><div class="ttname"><a href="group__ops.html#gad66135407dbb41b3c5d2cdfd51226c21">mlx::core::slice</a></div><div class="ttdeci">array slice(const array &a, std::vector< int > start, std::vector< int > stop, std::vector< int > strides, StreamOrDevice s={})</div><div class="ttdoc">Slice an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gad80f7c4a58c12b6cb30a8b9a73008993"><div class="ttname"><a href="group__ops.html#gad80f7c4a58c12b6cb30a8b9a73008993">mlx::core::isposinf</a></div><div class="ttdeci">array isposinf(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaddc825a5c173e195ab0fda83ad630420"><div class="ttname"><a href="group__ops.html#gaddc825a5c173e195ab0fda83ad630420">mlx::core::cumsum</a></div><div class="ttdeci">array cumsum(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative sum of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gade2eea48989f4caaf36e89f7bd2a8816"><div class="ttname"><a href="group__ops.html#gade2eea48989f4caaf36e89f7bd2a8816">mlx::core::operator-</a></div><div class="ttdeci">array operator-(const array &a)</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gade46e768fd46b8b640eb16f26abeecef"><div class="ttname"><a href="group__ops.html#gade46e768fd46b8b640eb16f26abeecef">mlx::core::mean</a></div><div class="ttdeci">array mean(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Computes the mean of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gade905ee92eb6ab7edfc312aeddfbaeb6"><div class="ttname"><a href="group__ops.html#gade905ee92eb6ab7edfc312aeddfbaeb6">mlx::core::sum</a></div><div class="ttdeci">array sum(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Sums the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gae0a81d4f983e296a87302e36d65bfc76"><div class="ttname"><a href="group__ops.html#gae0a81d4f983e296a87302e36d65bfc76">mlx::core::take_along_axis</a></div><div class="ttdeci">array take_along_axis(const array &a, const array &indices, int axis, StreamOrDevice s={})</div><div class="ttdoc">Take array entries given indices along the axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gae60b0b5339b9c50b9970260faf613e83"><div class="ttname"><a href="group__ops.html#gae60b0b5339b9c50b9970260faf613e83">mlx::core::argmax</a></div><div class="ttdeci">array argmax(const array &a, bool keepdims, StreamOrDevice s={})</div><div class="ttdoc">Returns the index of the maximum value in the array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaebf0a73ad3732fba39df37826c235692"><div class="ttname"><a href="group__ops.html#gaebf0a73ad3732fba39df37826c235692">mlx::core::sin</a></div><div class="ttdeci">array sin(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaee1d774bb0843601d7a0a4257d616ae3"><div class="ttname"><a href="group__ops.html#gaee1d774bb0843601d7a0a4257d616ae3">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_gaee37cac8476e8f8d666bcded5bc59143"><div class="ttname"><a href="group__ops.html#gaee37cac8476e8f8d666bcded5bc59143">mlx::core::cummax</a></div><div class="ttdeci">array cummax(const array &a, int axis, bool reverse=false, bool inclusive=true, StreamOrDevice s={})</div><div class="ttdoc">Cumulative max of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaee41e2b8f61d563200ff03575ac1d6c3"><div class="ttname"><a href="group__ops.html#gaee41e2b8f61d563200ff03575ac1d6c3">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:351</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaeeb7f5bb88aa32a3ac2be1f39c5f8087"><div class="ttname"><a href="group__ops.html#gaeeb7f5bb88aa32a3ac2be1f39c5f8087">mlx::core::atleast_2d</a></div><div class="ttdeci">array atleast_2d(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaeedf77f722b394429f1a7f6c367883bf"><div class="ttname"><a href="group__ops.html#gaeedf77f722b394429f1a7f6c367883bf">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_gaf0cd4257de7542daf9faf5e605e31020"><div class="ttname"><a href="group__ops.html#gaf0cd4257de7542daf9faf5e605e31020">mlx::core::allclose</a></div><div class="ttdeci">array allclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})</div><div class="ttdoc">True if the two arrays are equal within the specified tolerance.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf0d232de4cbfffda1e2c838f8afdf6ff"><div class="ttname"><a href="group__ops.html#gaf0d232de4cbfffda1e2c838f8afdf6ff">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_gaf301c49c10fa9b95a9e8dc52ead1a8dd"><div class="ttname"><a href="group__ops.html#gaf301c49c10fa9b95a9e8dc52ead1a8dd">mlx::core::argpartition</a></div><div class="ttdeci">array argpartition(const array &a, int kth, StreamOrDevice s={})</div><div class="ttdoc">Returns indices that partition the flattened array such that the smaller kth elements are first.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf4ec7bfc1ad13b891f1f3ef1772ef04d"><div class="ttname"><a href="group__ops.html#gaf4ec7bfc1ad13b891f1f3ef1772ef04d">mlx::core::greater</a></div><div class="ttdeci">array greater(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Returns bool array with (a > b) element-wise.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf532375c6563dbd6e329bdedf0224dd7"><div class="ttname"><a href="group__ops.html#gaf532375c6563dbd6e329bdedf0224dd7">mlx::core::sinh</a></div><div class="ttdeci">array sinh(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Hyperbolic Sine of the elements of an array.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf57392e641640b5d06e4c99518391c38"><div class="ttname"><a href="group__ops.html#gaf57392e641640b5d06e4c99518391c38">mlx::core::multiply</a></div><div class="ttdeci">array multiply(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Multiply two arrays.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf5c9735f4690327e1500e04e728fae70"><div class="ttname"><a href="group__ops.html#gaf5c9735f4690327e1500e04e728fae70">mlx::core::tensordot</a></div><div class="ttdeci">array tensordot(const array &a, const array &b, const int axis=2, StreamOrDevice s={})</div><div class="ttdoc">Returns a contraction of a and b over multiple dimensions.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf5d82380cd204e7c33324cebcd1461ce"><div class="ttname"><a href="group__ops.html#gaf5d82380cd204e7c33324cebcd1461ce">mlx::core::block_sparse_mm</a></div><div class="ttdeci">array block_sparse_mm(array a, array b, std::optional< array > lhs_indices=std::nullopt, std::optional< array > rhs_indices=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Compute matrix product with matrix-level gather.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf8f2ec2b98a4b59eca73d7471df6e032"><div class="ttname"><a href="group__ops.html#gaf8f2ec2b98a4b59eca73d7471df6e032">mlx::core::stack</a></div><div class="ttdeci">array stack(const std::vector< array > &arrays, int axis, StreamOrDevice s={})</div><div class="ttdoc">Stack arrays along a new axis.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gaf985df6609c6bd75a14a844655d89eaa"><div class="ttname"><a href="group__ops.html#gaf985df6609c6bd75a14a844655d89eaa">mlx::core::logaddexp</a></div><div class="ttdeci">array logaddexp(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Log-add-exp of one elements in the array: log(exp(a) + exp(b)).</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gafa376ad57d38ba87378f0272dc379b23"><div class="ttname"><a href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23">mlx::core::right_shift</a></div><div class="ttdeci">array right_shift(const array &a, const array &b, StreamOrDevice s={})</div><div class="ttdoc">Shift bits to the right.</div></div>
|
|
<div class="ttc" id="agroup__ops_html_gafbb857094d784b38c78683a091ffdbde"><div class="ttname"><a href="group__ops.html#gafbb857094d784b38c78683a091ffdbde">mlx::core::zeros_like</a></div><div class="ttdeci">array zeros_like(const array &a, StreamOrDevice s={})</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_html"><div class="ttname"><a href="namespacemlx_1_1core.html">mlx::core</a></div><div class="ttdef"><b>Definition</b> allocator.h:7</div></div>
|
|
<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_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:69</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:73</div></div>
|
|
<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:14</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:15</div></div>
|
|
<div class="ttc" id="autils_8h_html"><div class="ttname"><a href="utils_8h.html">utils.h</a></div></div>
|
|
</div><!-- fragment --></div><!-- contents -->
|
|
<!-- start footer part -->
|
|
<hr class="footer"/><address class="footer"><small>
|
|
Generated by <a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.10.0
|
|
</small></address>
|
|
</body>
|
|
</html>
|