mirror of
https://github.com/ml-explore/mlx.git
synced 2025-11-05 11:28:12 +08:00
rebase
This commit is contained in:
73
docs/build/html/ops_8h_source.html
vendored
73
docs/build/html/ops_8h_source.html
vendored
@@ -1367,38 +1367,41 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="line"><a id="l01474" name="l01474"></a><span class="lineno"> 1474</span></div>
|
||||
<div class="line"><a id="l01476" name="l01476"></a><span class="lineno"><a class="line" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23"> 1476</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gafa376ad57d38ba87378f0272dc379b23">right_shift</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01477" name="l01477"></a><span class="lineno"><a class="line" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a"> 1477</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga498b61f7e8f056ae00297fa0dc17303a">operator>></a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& b);</div>
|
||||
<div class="line"><a id="l01478" name="l01478"></a><span class="lineno"> 1478</span> </div>
|
||||
<div class="line"><a id="l01479" name="l01479"></a><span class="lineno"><a class="line" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02"> 1479</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02">view</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a>& dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01480" name="l01480"></a><span class="lineno"> 1480</span></div>
|
||||
<div class="line"><a id="l01482" name="l01482"></a><span class="lineno"><a class="line" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c"> 1482</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01483" name="l01483"></a><span class="lineno"><a class="line" href="group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02"> 1483</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01484" name="l01484"></a><span class="lineno"><a class="line" href="group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436"> 1484</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01485" name="l01485"></a><span class="lineno"><a class="line" href="group__ops.html#ga665f502ecc96f1f4467556b784abf9ae"> 1485</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01486" name="l01486"></a><span class="lineno"> 1486</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01487" name="l01487"></a><span class="lineno"> 1487</span> <span class="keywordtype">int</span> shift,</div>
|
||||
<div class="line"><a id="l01488" name="l01488"></a><span class="lineno"> 1488</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01489" name="l01489"></a><span class="lineno"> 1489</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01490" name="l01490"></a><span class="lineno"><a class="line" href="group__ops.html#ga79137f90bc44ac9e35f408c012701df9"> 1490</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01491" name="l01491"></a><span class="lineno"><a class="line" href="group__ops.html#ga9d76930fb567a7d459ff96fb851abe36"> 1491</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01492" name="l01492"></a><span class="lineno"> 1492</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01493" name="l01493"></a><span class="lineno"> 1493</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift,</div>
|
||||
<div class="line"><a id="l01494" name="l01494"></a><span class="lineno"> 1494</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01495" name="l01495"></a><span class="lineno"> 1495</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01496" name="l01496"></a><span class="lineno"> 1496</span> </div>
|
||||
<div class="line"><a id="l01497" name="l01497"></a><span class="lineno"> 1497</span><span class="comment">/* The real part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01498" name="l01498"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764"> 1498</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764">real</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01499" name="l01499"></a><span class="lineno"> 1499</span> </div>
|
||||
<div class="line"><a id="l01500" name="l01500"></a><span class="lineno"> 1500</span><span class="comment">/* The imaginary part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01501" name="l01501"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029"> 1501</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029">imag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01502" name="l01502"></a><span class="lineno"> 1502</span> </div>
|
||||
<div class="line"><a id="l01503" name="l01503"></a><span class="lineno"> 1503</span><span class="comment">/* Ensure the array's underlying memory is contiguous. */</span></div>
|
||||
<div class="line"><a id="l01504" name="l01504"></a><span class="lineno"><a class="line" href="group__ops.html#ga8ab10aa6c41416d739791164a52b25d5"> 1504</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8ab10aa6c41416d739791164a52b25d5">contiguous</a>(</div>
|
||||
<div class="line"><a id="l01505" name="l01505"></a><span class="lineno"> 1505</span> <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a,</div>
|
||||
<div class="line"><a id="l01506" name="l01506"></a><span class="lineno"> 1506</span> <span class="keywordtype">bool</span> allow_col_major = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01507" name="l01507"></a><span class="lineno"> 1507</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01508" name="l01508"></a><span class="lineno"> 1508</span></div>
|
||||
<div class="line"><a id="l01510" name="l01510"></a><span class="lineno"> 1510</span> </div>
|
||||
<div class="line"><a id="l01511" name="l01511"></a><span class="lineno"> 1511</span>} <span class="comment">// namespace mlx::core</span></div>
|
||||
<div class="line"><a id="l01478" name="l01478"></a><span class="lineno"> 1478</span></div>
|
||||
<div class="line"><a id="l01480" name="l01480"></a><span class="lineno"><a class="line" href="group__ops.html#gaf1182ae7c049fbc9ee190f3e0fffbf83"> 1480</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf1182ae7c049fbc9ee190f3e0fffbf83">bitwise_invert</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="l01481" name="l01481"></a><span class="lineno"><a class="line" href="group__ops.html#ga849365a62878579a33b3d3ad09bbc7be"> 1481</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga849365a62878579a33b3d3ad09bbc7be">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="l01482" name="l01482"></a><span class="lineno"> 1482</span> </div>
|
||||
<div class="line"><a id="l01483" name="l01483"></a><span class="lineno"><a class="line" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02"> 1483</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga3602aa91b7b124a0b41ec1b2137a1b02">view</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_struct" href="structmlx_1_1core_1_1_dtype.html">Dtype</a>& dtype, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01484" name="l01484"></a><span class="lineno"> 1484</span></div>
|
||||
<div class="line"><a id="l01486" name="l01486"></a><span class="lineno"><a class="line" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c"> 1486</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01487" name="l01487"></a><span class="lineno"><a class="line" href="group__ops.html#ga5011d1a5735c64e5b91afa56c7e2cc02"> 1487</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01488" name="l01488"></a><span class="lineno"><a class="line" href="group__ops.html#ga8694ec137165752cb6d8a36a6b7c3436"> 1488</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keywordtype">int</span> shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01489" name="l01489"></a><span class="lineno"><a class="line" href="group__ops.html#ga665f502ecc96f1f4467556b784abf9ae"> 1489</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01490" name="l01490"></a><span class="lineno"> 1490</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="l01491" name="l01491"></a><span class="lineno"> 1491</span> <span class="keywordtype">int</span> shift,</div>
|
||||
<div class="line"><a id="l01492" name="l01492"></a><span class="lineno"> 1492</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01493" name="l01493"></a><span class="lineno"> 1493</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01494" name="l01494"></a><span class="lineno"><a class="line" href="group__ops.html#ga79137f90bc44ac9e35f408c012701df9"> 1494</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift, <span class="keywordtype">int</span> axis, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01495" name="l01495"></a><span class="lineno"><a class="line" href="group__ops.html#ga9d76930fb567a7d459ff96fb851abe36"> 1495</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gac40e48c69f9c715a767912c30836e75c">roll</a>(</div>
|
||||
<div class="line"><a id="l01496" name="l01496"></a><span class="lineno"> 1496</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="l01497" name="l01497"></a><span class="lineno"> 1497</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">Shape</a>& shift,</div>
|
||||
<div class="line"><a id="l01498" name="l01498"></a><span class="lineno"> 1498</span> <span class="keyword">const</span> std::vector<int>& axes,</div>
|
||||
<div class="line"><a id="l01499" name="l01499"></a><span class="lineno"> 1499</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01500" name="l01500"></a><span class="lineno"> 1500</span> </div>
|
||||
<div class="line"><a id="l01501" name="l01501"></a><span class="lineno"> 1501</span><span class="comment">/* The real part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01502" name="l01502"></a><span class="lineno"><a class="line" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764"> 1502</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#gaf8913cabeb9fb193ba687aaeb2087764">real</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01503" name="l01503"></a><span class="lineno"> 1503</span> </div>
|
||||
<div class="line"><a id="l01504" name="l01504"></a><span class="lineno"> 1504</span><span class="comment">/* The imaginary part of a complex array. */</span></div>
|
||||
<div class="line"><a id="l01505" name="l01505"></a><span class="lineno"><a class="line" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029"> 1505</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029">imag</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>& a, <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01506" name="l01506"></a><span class="lineno"> 1506</span> </div>
|
||||
<div class="line"><a id="l01507" name="l01507"></a><span class="lineno"> 1507</span><span class="comment">/* Ensure the array's underlying memory is contiguous. */</span></div>
|
||||
<div class="line"><a id="l01508" name="l01508"></a><span class="lineno"><a class="line" href="group__ops.html#ga8ab10aa6c41416d739791164a52b25d5"> 1508</a></span><a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a> <a class="code hl_function" href="group__ops.html#ga8ab10aa6c41416d739791164a52b25d5">contiguous</a>(</div>
|
||||
<div class="line"><a id="l01509" name="l01509"></a><span class="lineno"> 1509</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="l01510" name="l01510"></a><span class="lineno"> 1510</span> <span class="keywordtype">bool</span> allow_col_major = <span class="keyword">false</span>,</div>
|
||||
<div class="line"><a id="l01511" name="l01511"></a><span class="lineno"> 1511</span> <a class="code hl_typedef" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s = {});</div>
|
||||
<div class="line"><a id="l01512" name="l01512"></a><span class="lineno"> 1512</span></div>
|
||||
<div class="line"><a id="l01514" name="l01514"></a><span class="lineno"> 1514</span> </div>
|
||||
<div class="line"><a id="l01515" name="l01515"></a><span class="lineno"> 1515</span>} <span class="comment">// namespace mlx::core</span></div>
|
||||
<div class="ttc" id="aarray_8h_html"><div class="ttname"><a href="array_8h.html">array.h</a></div></div>
|
||||
<div class="ttc" id="aclassmlx_1_1core_1_1array_html"><div class="ttname"><a href="classmlx_1_1core_1_1array.html">mlx::core::array</a></div><div class="ttdef"><b>Definition</b> array.h:24</div></div>
|
||||
<div class="ttc" id="adevice_8h_html"><div class="ttname"><a href="device_8h.html">device.h</a></div></div>
|
||||
@@ -1501,6 +1504,7 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="agroup__ops_html_ga7ff592a64d528f0cf4f3d098465da029"><div class="ttname"><a href="group__ops.html#ga7ff592a64d528f0cf4f3d098465da029">mlx::core::imag</a></div><div class="ttdeci">array imag(const array &a, StreamOrDevice s={})</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_ga849365a62878579a33b3d3ad09bbc7be"><div class="ttname"><a href="group__ops.html#ga849365a62878579a33b3d3ad09bbc7be">mlx::core::operator~</a></div><div class="ttdeci">array operator~(const array &a)</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_ga872d2c1806e67ce2596b24d056681074"><div class="ttname"><a href="group__ops.html#ga872d2c1806e67ce2596b24d056681074">mlx::core::hadamard_transform</a></div><div class="ttdeci">array hadamard_transform(const array &a, std::optional< float > scale=std::nullopt, StreamOrDevice s={})</div><div class="ttdoc">Multiply the array by the Hadamard matrix of corresponding size.</div></div>
|
||||
@@ -1574,6 +1578,7 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<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_gaf1182ae7c049fbc9ee190f3e0fffbf83"><div class="ttname"><a href="group__ops.html#gaf1182ae7c049fbc9ee190f3e0fffbf83">mlx::core::bitwise_invert</a></div><div class="ttdeci">array bitwise_invert(const array &a, StreamOrDevice s={})</div><div class="ttdoc">Invert the bits.</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>
|
||||
@@ -1589,8 +1594,8 @@ $(function(){initNavTree('ops_8h_source.html',''); initResizable(true); });
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a4734a596e57434492ddfe79f2cb9dbf9"><div class="ttname"><a href="namespacemlx_1_1core.html#a4734a596e57434492ddfe79f2cb9dbf9">mlx::core::to_stream</a></div><div class="ttdeci">Stream to_stream(StreamOrDevice s)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a479648542a2bea151b947b18f0e79dd2"><div class="ttname"><a href="namespacemlx_1_1core.html#a479648542a2bea151b947b18f0e79dd2">mlx::core::copy</a></div><div class="ttdeci">void copy(const array &src, array &dst, CopyType ctype)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a5adff87687b78bfc18dafbc654925cdb"><div class="ttname"><a href="namespacemlx_1_1core.html#a5adff87687b78bfc18dafbc654925cdb">mlx::core::slice</a></div><div class="ttdeci">void slice(const array &in, array &out, const Shape &start_indices, const Shape &strides)</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a5d6373aad1444edc9de1eb07bfe5cad3"><div class="ttname"><a href="namespacemlx_1_1core.html#a5d6373aad1444edc9de1eb07bfe5cad3">mlx::core::int32</a></div><div class="ttdeci">constexpr Dtype int32</div><div class="ttdef"><b>Definition</b> dtype.h:76</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a6894543b340321193dfb8052c438a319"><div class="ttname"><a href="namespacemlx_1_1core.html#a6894543b340321193dfb8052c438a319">mlx::core::float32</a></div><div class="ttdeci">constexpr Dtype float32</div><div class="ttdef"><b>Definition</b> dtype.h:80</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_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:77</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:81</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a694e23f2d59606643728ad443d621416"><div class="ttname"><a href="namespacemlx_1_1core.html#a694e23f2d59606643728ad443d621416">mlx::core::Shape</a></div><div class="ttdeci">std::vector< ShapeElem > Shape</div><div class="ttdef"><b>Definition</b> array.h:21</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a76a2e310857f60f5ea6f1388d45b964d"><div class="ttname"><a href="namespacemlx_1_1core.html#a76a2e310857f60f5ea6f1388d45b964d">mlx::core::concatenate</a></div><div class="ttdeci">void concatenate(std::string &acc, T first)</div><div class="ttdef"><b>Definition</b> utils.h:62</div></div>
|
||||
<div class="ttc" id="anamespacemlx_1_1core_html_a79939016d0972ded7db37130da2a8b5c"><div class="ttname"><a href="namespacemlx_1_1core.html#a79939016d0972ded7db37130da2a8b5c">mlx::core::Strides</a></div><div class="ttdeci">std::vector< int64_t > Strides</div><div class="ttdef"><b>Definition</b> array.h:22</div></div>
|
||||
|
||||
Reference in New Issue
Block a user