update Select op doc
This commit is contained in:
parent
14be21c4b9
commit
96fe10c20a
Binary file not shown.
Binary file not shown.
|
|
@ -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>
|
||||
|
||||
|
|
|
|||
|
|
@ -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">&&</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">&&</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"><stdio.h></span>
|
||||
<span class="linenos"> 3</span><span class="cp">#include</span><span class="w"> </span><span class="cpf"><select.h></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
Loading…
Reference in New Issue