update Select op doc

This commit is contained in:
LiWanpeng 2026-06-10 09:43:24 +08:00
parent 14be21c4b9
commit 96fe10c20a
5 changed files with 21 additions and 21 deletions

Binary file not shown.

View File

@ -95,7 +95,7 @@ Select
.. code-block:: c
:linenos:
:emphasize-lines: 35-36
:emphasize-lines: 69-72
//FT78NE示例
#include <stdio.h>
@ -188,9 +188,9 @@ Select
.. code-block:: c
:linenos:
:emphasize-lines: 30-31
:emphasize-lines: 69-72
//FT78NE示例
//FT04示例
#include <stdio.h>
#include <select.h>

View File

@ -469,9 +469,9 @@
<span class="linenos">32</span>
<span class="linenos">33</span><span class="w"> </span><span class="c1">//先计算is_broadcast</span>
<span class="linenos">34</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">input0_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">input0_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">input0_dims</span><span class="p">);</span>
<span class="hll"><span class="linenos">35</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">input1_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">input1_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">input1_dims</span><span class="p">);</span>
</span><span class="hll"><span class="linenos">36</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">cond_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">cond_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">cond_dims</span><span class="p">);</span>
</span><span class="linenos">37</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">output_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">output_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims</span><span class="p">);</span>
<span class="linenos">35</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">input1_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">input1_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">input1_dims</span><span class="p">);</span>
<span class="linenos">36</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">cond_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">cond_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">cond_dims</span><span class="p">);</span>
<span class="linenos">37</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">output_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">output_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims</span><span class="p">);</span>
<span class="linenos">38</span>
<span class="linenos">39</span><span class="w"> </span><span class="k">if</span><span class="p">((</span><span class="n">input0_num</span><span class="w"> </span><span class="o">==</span><span class="w"> </span><span class="n">output_num</span><span class="w"> </span><span class="o">&amp;&amp;</span><span class="w"> </span><span class="n">input1_num</span><span class="w"> </span><span class="o">==</span><span class="w"> </span><span class="n">output_num</span><span class="p">)</span><span class="w"> </span><span class="o">&amp;&amp;</span><span class="w"> </span><span class="p">(</span><span class="n">cond_num</span><span class="w"> </span><span class="o">==</span><span class="w"> </span><span class="n">output_num</span><span class="p">))</span><span class="w"> </span><span class="p">{</span>
<span class="linenos">40</span><span class="w"> </span><span class="n">is_broadcast</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
@ -503,11 +503,11 @@
<span class="linenos">66</span><span class="w"> </span><span class="n">GetBroadCastIndex</span><span class="p">(</span><span class="n">input0_dims</span><span class="p">,</span><span class="w"> </span><span class="n">input0_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">index_list2</span><span class="p">);</span>
<span class="linenos">67</span><span class="w"> </span><span class="n">GetBroadCastIndex</span><span class="p">(</span><span class="n">input1_dims</span><span class="p">,</span><span class="w"> </span><span class="n">input1_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">index_list3</span><span class="p">);</span>
<span class="linenos">68</span>
<span class="linenos">69</span><span class="w"> </span><span class="n">fp_select_s</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">,</span><span class="w"> </span><span class="n">core_mask</span><span class="p">);</span>
<span class="linenos">70</span>
<span class="linenos">71</span><span class="w"> </span><span class="p">}</span><span class="w"> </span><span class="k">else</span><span class="w"> </span><span class="p">{</span>
<span class="linenos">72</span><span class="w"> </span><span class="n">fp_select_s</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">,</span><span class="w"> </span><span class="n">core_mask</span><span class="p">);</span>
<span class="linenos">73</span><span class="w"> </span><span class="p">}</span>
<span class="hll"><span class="linenos">69</span><span class="w"> </span><span class="n">fp_select_s</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">,</span><span class="w"> </span><span class="n">core_mask</span><span class="p">);</span>
</span><span class="hll"><span class="linenos">70</span>
</span><span class="hll"><span class="linenos">71</span><span class="w"> </span><span class="p">}</span><span class="w"> </span><span class="k">else</span><span class="w"> </span><span class="p">{</span>
</span><span class="hll"><span class="linenos">72</span><span class="w"> </span><span class="n">fp_select_s</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">,</span><span class="w"> </span><span class="n">core_mask</span><span class="p">);</span>
</span><span class="linenos">73</span><span class="w"> </span><span class="p">}</span>
<span class="linenos">74</span><span class="w"> </span><span class="k">return</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
<span class="linenos">75</span><span class="p">}</span>
</pre></div>
@ -554,7 +554,7 @@
<dd></dd></dl>
<p><strong>C调用示例私有存储版本</strong></p>
<div class="highlight-c notranslate"><div class="highlight"><pre><span></span><span class="linenos"> 1</span><span class="c1">//FT78NE示例</span>
<div class="highlight-c notranslate"><div class="highlight"><pre><span></span><span class="linenos"> 1</span><span class="c1">//FT04示例</span>
<span class="linenos"> 2</span><span class="cp">#include</span><span class="w"> </span><span class="cpf">&lt;stdio.h&gt;</span>
<span class="linenos"> 3</span><span class="cp">#include</span><span class="w"> </span><span class="cpf">&lt;select.h&gt;</span>
<span class="linenos"> 4</span>
@ -583,9 +583,9 @@
<span class="linenos">27</span>
<span class="linenos">28</span><span class="w"> </span><span class="n">params</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="p">)</span><span class="n">output_dims</span><span class="p">;</span>
<span class="linenos">29</span><span class="w"> </span><span class="n">params</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="p">)</span><span class="n">index_list1</span><span class="p">;</span>
<span class="hll"><span class="linenos">30</span><span class="w"> </span><span class="n">params</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="p">)</span><span class="n">index_list2</span><span class="p">;</span>
</span><span class="hll"><span class="linenos">31</span><span class="w"> </span><span class="n">params</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="p">)</span><span class="n">index_list3</span><span class="p">;</span>
</span><span class="linenos">32</span>
<span class="linenos">30</span><span class="w"> </span><span class="n">params</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="p">)</span><span class="n">index_list2</span><span class="p">;</span>
<span class="linenos">31</span><span class="w"> </span><span class="n">params</span><span class="p">[</span><span class="mi">4</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="p">)</span><span class="n">index_list3</span><span class="p">;</span>
<span class="linenos">32</span>
<span class="linenos">33</span><span class="w"> </span><span class="c1">//先计算is_broadcast</span>
<span class="linenos">34</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">input0_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">input0_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">input0_dims</span><span class="p">);</span>
<span class="linenos">35</span><span class="w"> </span><span class="kt">unsigned</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="kt">long</span><span class="w"> </span><span class="n">input1_num</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">get_total_elements</span><span class="p">(</span><span class="n">input1_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">input1_dims</span><span class="p">);</span>
@ -622,11 +622,11 @@
<span class="linenos">66</span><span class="w"> </span><span class="n">GetBroadCastIndex</span><span class="p">(</span><span class="n">input0_dims</span><span class="p">,</span><span class="w"> </span><span class="n">input0_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">index_list2</span><span class="p">);</span>
<span class="linenos">67</span><span class="w"> </span><span class="n">GetBroadCastIndex</span><span class="p">(</span><span class="n">input1_dims</span><span class="p">,</span><span class="w"> </span><span class="n">input1_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims</span><span class="p">,</span><span class="w"> </span><span class="n">output_dims_num</span><span class="p">,</span><span class="w"> </span><span class="n">index_list3</span><span class="p">);</span>
<span class="linenos">68</span>
<span class="linenos">69</span><span class="w"> </span><span class="n">fp_select_p</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">);</span>
<span class="linenos">70</span>
<span class="linenos">71</span><span class="w"> </span><span class="p">}</span><span class="w"> </span><span class="k">else</span><span class="w"> </span><span class="p">{</span>
<span class="linenos">72</span><span class="w"> </span><span class="n">fp_select_p</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">);</span>
<span class="linenos">73</span><span class="w"> </span><span class="p">}</span>
<span class="hll"><span class="linenos">69</span><span class="w"> </span><span class="n">fp_select_p</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">);</span>
</span><span class="hll"><span class="linenos">70</span>
</span><span class="hll"><span class="linenos">71</span><span class="w"> </span><span class="p">}</span><span class="w"> </span><span class="k">else</span><span class="w"> </span><span class="p">{</span>
</span><span class="hll"><span class="linenos">72</span><span class="w"> </span><span class="n">fp_select_p</span><span class="p">(</span><span class="n">input0</span><span class="p">,</span><span class="w"> </span><span class="n">input1</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">,</span><span class="w"> </span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">params</span><span class="p">);</span>
</span><span class="linenos">73</span><span class="w"> </span><span class="p">}</span>
<span class="linenos">74</span>
<span class="linenos">75</span><span class="w"> </span><span class="k">return</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
<span class="linenos">76</span><span class="p">}</span>

File diff suppressed because one or more lines are too long