zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-12-06 21:22:39 +00:00
parent 8ba8a39e43
commit 04b749a588
896 changed files with 12293 additions and 11212 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/build/html/transforms_8h_source.html vendored
View File

@@ -249,7 +249,7 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00223" name="l00223"></a><span class="lineno"> 223</span> </div>
<div class="line"><a id="l00224" name="l00224"></a><span class="lineno"> 224</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="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:23</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_a15dda19aa7fa1fc5fca35df5cf963297"><div class="ttname"><a href="namespacemlx_1_1core.html#a15dda19aa7fa1fc5fca35df5cf963297">mlx::core::async_eval</a></div><div class="ttdeci">void async_eval(std::vector&lt; array &gt; outputs)</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_a179a632200366c223d6ab56d3e032592"><div class="ttname"><a href="namespacemlx_1_1core.html#a179a632200366c223d6ab56d3e032592">mlx::core::jvp</a></div><div class="ttdeci">std::pair&lt; std::vector&lt; array &gt;, std::vector&lt; array &gt; &gt; jvp(const std::function&lt; std::vector&lt; array &gt;(const std::vector&lt; array &gt; &amp;)&gt; &amp;fun, const std::vector&lt; array &gt; &amp;primals, const std::vector&lt; array &gt; &amp;tangents)</div><div class="ttdoc">Computes the output and Jacobian-vector product (JVP) of a function.</div></div>
@@ -263,7 +263,7 @@ $(function(){ initResizable(false); });
<div class="ttc" id="anamespacemlx_1_1core_html_ab79d66ddf1ec38b2f2c01234892a2230"><div class="ttname"><a href="namespacemlx_1_1core.html#ab79d66ddf1ec38b2f2c01234892a2230">mlx::core::ValueAndGradFn</a></div><div class="ttdeci">std::function&lt; std::pair&lt; std::vector&lt; array &gt;, std::vector&lt; array &gt; &gt;( const std::vector&lt; array &gt; &amp;)&gt; ValueAndGradFn</div><div class="ttdef"><b>Definition</b> transforms.h:63</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_abf49b337a00997231c0f7fd389efa8f3"><div class="ttname"><a href="namespacemlx_1_1core.html#abf49b337a00997231c0f7fd389efa8f3">mlx::core::value_and_grad</a></div><div class="ttdeci">ValueAndGradFn value_and_grad(const std::function&lt; std::vector&lt; array &gt;(const std::vector&lt; array &gt; &amp;)&gt; &amp;fun, const std::vector&lt; int &gt; &amp;argnums)</div><div class="ttdoc">Returns a function which computes the value and gradient of the input function with respect to a vect...</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_ac3caec2fa65375ed4c3bf1206177b84c"><div class="ttname"><a href="namespacemlx_1_1core.html#ac3caec2fa65375ed4c3bf1206177b84c">mlx::core::vmap</a></div><div class="ttdeci">std::function&lt; array(const array &amp;)&gt; vmap(const std::function&lt; array(const array &amp;)&gt; &amp;fun, int in_axis=0, int out_axis=0)</div><div class="ttdoc">Automatically vectorize a unary function over the requested axes.</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_af89751d79339f3e4d9318ea97d64d114"><div class="ttname"><a href="namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114">mlx::core::enable_for_arrays_t</a></div><div class="ttdeci">typename std::enable_if_t&lt; is_arrays_v&lt; T... &gt; &gt; enable_for_arrays_t</div><div class="ttdef"><b>Definition</b> array.h:611</div></div>
<div class="ttc" id="anamespacemlx_1_1core_html_af89751d79339f3e4d9318ea97d64d114"><div class="ttname"><a href="namespacemlx_1_1core.html#af89751d79339f3e4d9318ea97d64d114">mlx::core::enable_for_arrays_t</a></div><div class="ttdeci">typename std::enable_if_t&lt; is_arrays_v&lt; T... &gt; &gt; enable_for_arrays_t</div><div class="ttdef"><b>Definition</b> array.h:614</div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->
<hr class="footer"/><address class="footer"><small>