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

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-10-18 19:13:44 +00:00
parent 69fcd40979
commit 57e53f85ee
590 changed files with 10972 additions and 5085 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docs/build/html/common_2unary_8h_source.html vendored
View File

@@ -117,26 +117,26 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno"> 24</span> }</div>
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span>}</div>
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"> 26</span> </div>
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"> 27</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> Op&gt;</div>
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"> 28</span><span class="keywordtype">void</span> unary_op(<span class="keyword">const</span> T* a, T* out, Op <a class="code hl_variable" href="common_2binary_8h.html#a70228731d29946574b238d21fb4b360c">op</a>, <span class="keywordtype">size_t</span> shape, <span class="keywordtype">size_t</span> stride) {</div>
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"> 27</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> U = T, <span class="keyword">typename</span> Op&gt;</div>
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"> 28</span><span class="keywordtype">void</span> unary_op(<span class="keyword">const</span> T* a, U* out, Op <a class="code hl_variable" href="common_2binary_8h.html#a70228731d29946574b238d21fb4b360c">op</a>, <span class="keywordtype">size_t</span> shape, <span class="keywordtype">size_t</span> stride) {</div>
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"> 29</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; shape; i += 1) {</div>
<div class="line"><a id="l00030" name="l00030"></a><span class="lineno"> 30</span> out[i] = <a class="code hl_variable" href="common_2binary_8h.html#a70228731d29946574b238d21fb4b360c">op</a>(*a);</div>
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno"> 31</span> a += stride;</div>
<div class="line"><a id="l00032" name="l00032"></a><span class="lineno"> 32</span> }</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"> 34</span> </div>
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"> 35</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> Op&gt;</div>
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"> 35</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> U = T, <span class="keyword">typename</span> Op&gt;</div>
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"> 36</span><span class="keywordtype">void</span> unary_op(<span class="keyword">const</span> array&amp; a, array&amp; out, Op <a class="code hl_variable" href="common_2binary_8h.html#a70228731d29946574b238d21fb4b360c">op</a>) {</div>
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span> <span class="keyword">const</span> T* a_ptr = a.data&lt;T&gt;();</div>
<div class="line"><a id="l00038" name="l00038"></a><span class="lineno"> 38</span> <span class="keywordflow">if</span> (a.flags().contiguous) {</div>
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"> 39</span> set_unary_output_data(a, out);</div>
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span> T* dst = out.data&lt;T&gt;();</div>
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span> U* dst = out.data&lt;U&gt;();</div>
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i = 0; i &lt; a.data_size(); ++i) {</div>
<div class="line"><a id="l00042" name="l00042"></a><span class="lineno"> 42</span> dst[i] = <a class="code hl_variable" href="common_2binary_8h.html#a70228731d29946574b238d21fb4b360c">op</a>(a_ptr[i]);</div>
<div class="line"><a id="l00043" name="l00043"></a><span class="lineno"> 43</span> }</div>
<div class="line"><a id="l00044" name="l00044"></a><span class="lineno"> 44</span> } <span class="keywordflow">else</span> {</div>
<div class="line"><a id="l00045" name="l00045"></a><span class="lineno"> 45</span> out.set_data(<a class="code hl_function" href="namespacemlx_1_1core_1_1allocator.html#a86ac0a11ff78f21e717f641716c34abc">allocator::malloc_or_wait</a>(out.nbytes()));</div>
<div class="line"><a id="l00046" name="l00046"></a><span class="lineno"> 46</span> T* dst = out.data&lt;T&gt;();</div>
<div class="line"><a id="l00046" name="l00046"></a><span class="lineno"> 46</span> U* dst = out.data&lt;U&gt;();</div>
<div class="line"><a id="l00047" name="l00047"></a><span class="lineno"> 47</span> <span class="keywordtype">size_t</span> shape = a.ndim() &gt; 0 ? a.shape(-1) : 1;</div>
<div class="line"><a id="l00048" name="l00048"></a><span class="lineno"> 48</span> <span class="keywordtype">size_t</span> stride = a.ndim() &gt; 0 ? a.strides(-1) : 1;</div>
<div class="line"><a id="l00049" name="l00049"></a><span class="lineno"> 49</span> <span class="keywordflow">if</span> (a.ndim() &lt;= 1) {</div>