<tr style="height: 56px;">
<td style="padding-left: 0.5em;">
<div id="projectname">Compute Library
-  <span id="projectnumber">18.01</span>
+  <span id="projectnumber">18.05</span>
</div>
</td>
</tr>
<div class="title">NEFullyConnectedLayer.h</div> </div>
</div><!--header-->
<div class="contents">
-<a href="_n_e_fully_connected_layer_8h.xhtml">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">/*</span></div><div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> * Copyright (c) 2017 ARM Limited.</span></div><div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> *</span></div><div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> * SPDX-License-Identifier: MIT</span></div><div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> *</span></div><div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * Permission is hereby granted, free of charge, to any person obtaining a copy</span></div><div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> * of this software and associated documentation files (the "Software"), to</span></div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> * deal in the Software without restriction, including without limitation the</span></div><div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="comment"> * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or</span></div><div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="comment"> * sell copies of the Software, and to permit persons to whom the Software is</span></div><div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="comment"> * furnished to do so, subject to the following conditions:</span></div><div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="comment"> *</span></div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="comment"> * The above copyright notice and this permission notice shall be included in all</span></div><div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="comment"> * copies or substantial portions of the Software.</span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="comment"> *</span></div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="comment"> * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR</span></div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,</span></div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE</span></div><div class="line"><a name="l00019"></a><span class="lineno"> 19</span> <span class="comment"> * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER</span></div><div class="line"><a name="l00020"></a><span class="lineno"> 20</span> <span class="comment"> * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,</span></div><div class="line"><a name="l00021"></a><span class="lineno"> 21</span> <span class="comment"> * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE</span></div><div class="line"><a name="l00022"></a><span class="lineno"> 22</span> <span class="comment"> * SOFTWARE.</span></div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span> <span class="comment"> */</span></div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span> <span class="preprocessor">#ifndef __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__</span></div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span> <span class="preprocessor">#define __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__</span></div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span> </div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#include "<a class="code" href="_i_function_8h.xhtml">arm_compute/runtime/IFunction.h</a>"</span></div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span> </div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_interleave4x4_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMInterleave4x4Kernel.h</a>"</span></div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_matrix_accumulate_biases_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.h</a>"</span></div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_matrix_multiply_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h</a>"</span></div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_transpose1x_w_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMTranspose1xWKernel.h</a>"</span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#include "<a class="code" href="_n_e_im2_col_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEIm2ColKernel.h</a>"</span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include "<a class="code" href="_n_e_transpose_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NETransposeKernel.h</a>"</span></div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#include "<a class="code" href="_memory_group_8h.xhtml">arm_compute/runtime/MemoryGroup.h</a>"</span></div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#include "<a class="code" href="runtime_2_tensor_8h.xhtml">arm_compute/runtime/Tensor.h</a>"</span></div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span> </div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="keyword">namespace </span><a class="code" href="namespacearm__compute.xhtml">arm_compute</a></div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span> {</div><div class="line"><a name="l00047"></a><span class="lineno"><a class="line" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml"> 47</a></span> <span class="keyword">class </span><a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml">NEFullyConnectedLayerReshapeWeights</a> : <span class="keyword">public</span> <a class="code" href="classarm__compute_1_1_i_function.xhtml">IFunction</a></div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span> {</div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#af20812ea72913d388a9a70f9d28e7c36">NEFullyConnectedLayerReshapeWeights</a>(std::shared_ptr<IMemoryManager> memory_manager = <span class="keyword">nullptr</span>);</div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a510f09c40e20d367e4506bc9b4e8b36e">configure</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *input, <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *output, <span class="keywordtype">bool</span> transpose_weights, <span class="keywordtype">bool</span> is_batched_fc_layer);</div><div class="line"><a name="l00060"></a><span class="lineno"> 60</span> </div><div class="line"><a name="l00061"></a><span class="lineno"> 61</span>  <span class="comment">// Inherited methods overridden:</span></div><div class="line"><a name="l00062"></a><span class="lineno"> 62</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#ad1717410afd0be936c6213a63c8005fb">run</a>() <span class="keyword">override</span>;</div><div class="line"><a name="l00063"></a><span class="lineno"> 63</span> </div><div class="line"><a name="l00064"></a><span class="lineno"> 64</span> <span class="keyword">private</span>:</div><div class="line"><a name="l00065"></a><span class="lineno"> 65</span>  <a class="code" href="classarm__compute_1_1_memory_group_base.xhtml">MemoryGroup</a> _memory_group;</div><div class="line"><a name="l00066"></a><span class="lineno"> 66</span>  <a class="code" href="classarm__compute_1_1_n_e_transpose_kernel.xhtml">NETransposeKernel</a> _transpose_kernel;</div><div class="line"><a name="l00067"></a><span class="lineno"> 67</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_transpose1x_w_kernel.xhtml">NEGEMMTranspose1xWKernel</a> _transpose1xW_kernel;</div><div class="line"><a name="l00068"></a><span class="lineno"> 68</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _transpose_output;</div><div class="line"><a name="l00069"></a><span class="lineno"> 69</span>  <span class="keywordtype">bool</span> _transpose_weights;</div><div class="line"><a name="l00070"></a><span class="lineno"> 70</span>  <span class="keywordtype">bool</span> _is_batched_fc_layer;</div><div class="line"><a name="l00071"></a><span class="lineno"> 71</span> };</div><div class="line"><a name="l00072"></a><span class="lineno"> 72</span> </div><div class="line"><a name="l00082"></a><span class="lineno"><a class="line" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml"> 82</a></span> <span class="keyword">class </span><a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> : <span class="keyword">public</span> <a class="code" href="classarm__compute_1_1_i_function.xhtml">IFunction</a></div><div class="line"><a name="l00083"></a><span class="lineno"> 83</span> {</div><div class="line"><a name="l00084"></a><span class="lineno"> 84</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a>(std::shared_ptr<IMemoryManager> memory_manager = <span class="keyword">nullptr</span>);</div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a510f09c40e20d367e4506bc9b4e8b36e">configure</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *input, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *weights, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *biases, <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *output, <span class="keywordtype">bool</span> transpose_weights = <span class="keyword">true</span>, <span class="keywordtype">bool</span> are_weights_reshaped = <span class="keyword">false</span>);</div><div class="line"><a name="l00097"></a><span class="lineno"> 97</span> </div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  <span class="comment">//Inherited methods override</span></div><div class="line"><a name="l00099"></a><span class="lineno"> 99</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#ad1717410afd0be936c6213a63c8005fb">run</a>() <span class="keyword">override</span>;</div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span> </div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span> <span class="keyword">private</span>:</div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  <a class="code" href="classarm__compute_1_1_memory_group_base.xhtml">MemoryGroup</a> _memory_group;</div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span>  <a class="code" href="classarm__compute_1_1_n_e_im2_col_kernel.xhtml">NEIm2ColKernel</a> _im2col_kernel;</div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml">NEFullyConnectedLayerReshapeWeights</a> _reshape_weights_kernel;</div><div class="line"><a name="l00105"></a><span class="lineno"> 105</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_interleave4x4_kernel.xhtml">NEGEMMInterleave4x4Kernel</a> _interleave4x4_kernel;</div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_matrix_multiply_kernel.xhtml">NEGEMMMatrixMultiplyKernel</a> _mm_kernel;</div><div class="line"><a name="l00107"></a><span class="lineno"> 107</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_matrix_accumulate_biases_kernel.xhtml">NEGEMMMatrixAccumulateBiasesKernel</a> _accumulate_biases_kernel;</div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _im2col_output;</div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _interleave4x4_output;</div><div class="line"><a name="l00110"></a><span class="lineno"> 110</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _reshape_weights_output;</div><div class="line"><a name="l00111"></a><span class="lineno"> 111</span>  <span class="keywordtype">bool</span> _are_weights_reshaped;</div><div class="line"><a name="l00112"></a><span class="lineno"> 112</span>  <span class="keywordtype">bool</span> _is_batched_fc_layer;</div><div class="line"><a name="l00113"></a><span class="lineno"> 113</span>  <span class="keywordtype">bool</span> _linearize_input;</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <span class="keywordtype">bool</span> _accumulate_biases;</div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span> };</div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span> }</div><div class="line"><a name="l00117"></a><span class="lineno"> 117</span> <span class="preprocessor">#endif </span><span class="comment">/* __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__ */</span><span class="preprocessor"></span></div><div class="ttc" id="_n_e_g_e_m_m_interleave4x4_kernel_8h_xhtml"><div class="ttname"><a href="_n_e_g_e_m_m_interleave4x4_kernel_8h.xhtml">NEGEMMInterleave4x4Kernel.h</a></div></div>
+<a href="_n_e_fully_connected_layer_8h.xhtml">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span> <span class="comment">/*</span></div><div class="line"><a name="l00002"></a><span class="lineno"> 2</span> <span class="comment"> * Copyright (c) 2017-2018 ARM Limited.</span></div><div class="line"><a name="l00003"></a><span class="lineno"> 3</span> <span class="comment"> *</span></div><div class="line"><a name="l00004"></a><span class="lineno"> 4</span> <span class="comment"> * SPDX-License-Identifier: MIT</span></div><div class="line"><a name="l00005"></a><span class="lineno"> 5</span> <span class="comment"> *</span></div><div class="line"><a name="l00006"></a><span class="lineno"> 6</span> <span class="comment"> * Permission is hereby granted, free of charge, to any person obtaining a copy</span></div><div class="line"><a name="l00007"></a><span class="lineno"> 7</span> <span class="comment"> * of this software and associated documentation files (the "Software"), to</span></div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span> <span class="comment"> * deal in the Software without restriction, including without limitation the</span></div><div class="line"><a name="l00009"></a><span class="lineno"> 9</span> <span class="comment"> * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or</span></div><div class="line"><a name="l00010"></a><span class="lineno"> 10</span> <span class="comment"> * sell copies of the Software, and to permit persons to whom the Software is</span></div><div class="line"><a name="l00011"></a><span class="lineno"> 11</span> <span class="comment"> * furnished to do so, subject to the following conditions:</span></div><div class="line"><a name="l00012"></a><span class="lineno"> 12</span> <span class="comment"> *</span></div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span> <span class="comment"> * The above copyright notice and this permission notice shall be included in all</span></div><div class="line"><a name="l00014"></a><span class="lineno"> 14</span> <span class="comment"> * copies or substantial portions of the Software.</span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span> <span class="comment"> *</span></div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span> <span class="comment"> * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR</span></div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span> <span class="comment"> * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,</span></div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span> <span class="comment"> * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE</span></div><div class="line"><a name="l00019"></a><span class="lineno"> 19</span> <span class="comment"> * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER</span></div><div class="line"><a name="l00020"></a><span class="lineno"> 20</span> <span class="comment"> * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,</span></div><div class="line"><a name="l00021"></a><span class="lineno"> 21</span> <span class="comment"> * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE</span></div><div class="line"><a name="l00022"></a><span class="lineno"> 22</span> <span class="comment"> * SOFTWARE.</span></div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span> <span class="comment"> */</span></div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span> <span class="preprocessor">#ifndef __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__</span></div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span> <span class="preprocessor">#define __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__</span></div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span> </div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span> <span class="preprocessor">#include "<a class="code" href="_i_function_8h.xhtml">arm_compute/runtime/IFunction.h</a>"</span></div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span> </div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_interleave4x4_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMInterleave4x4Kernel.h</a>"</span></div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_matrix_accumulate_biases_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMMatrixAccumulateBiasesKernel.h</a>"</span></div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_matrix_multiply_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMMatrixMultiplyKernel.h</a>"</span></div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span> <span class="preprocessor">#include "<a class="code" href="_n_e_g_e_m_m_transpose1x_w_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEGEMMTranspose1xWKernel.h</a>"</span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span> <span class="preprocessor">#include "<a class="code" href="_n_e_im2_col_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NEIm2ColKernel.h</a>"</span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span> <span class="preprocessor">#include "<a class="code" href="_n_e_transpose_kernel_8h.xhtml">arm_compute/core/NEON/kernels/NETransposeKernel.h</a>"</span></div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span> <span class="preprocessor">#include "<a class="code" href="_memory_group_8h.xhtml">arm_compute/runtime/MemoryGroup.h</a>"</span></div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span> <span class="preprocessor">#include "<a class="code" href="runtime_2_tensor_8h.xhtml">arm_compute/runtime/Tensor.h</a>"</span></div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span> </div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span> <span class="keyword">namespace </span><a class="code" href="namespacearm__compute.xhtml">arm_compute</a></div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span> {</div><div class="line"><a name="l00047"></a><span class="lineno"><a class="line" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml"> 47</a></span> <span class="keyword">class </span><a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml">NEFullyConnectedLayerReshapeWeights</a> : <span class="keyword">public</span> <a class="code" href="classarm__compute_1_1_i_function.xhtml">IFunction</a></div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span> {</div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#af20812ea72913d388a9a70f9d28e7c36">NEFullyConnectedLayerReshapeWeights</a>(std::shared_ptr<IMemoryManager> memory_manager = <span class="keyword">nullptr</span>);</div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a510f09c40e20d367e4506bc9b4e8b36e">configure</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *input, <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *output, <span class="keywordtype">bool</span> transpose_weights, <span class="keywordtype">bool</span> is_batched_fc_layer);</div><div class="line"><a name="l00069"></a><span class="lineno"> 69</span>  <span class="keyword">static</span> <a class="code" href="classarm__compute_1_1_status.xhtml">Status</a> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a70b636ca822d48228a8c1f6a4d0666b2">validate</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor_info.xhtml">ITensorInfo</a> *input, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor_info.xhtml">ITensorInfo</a> *output, <span class="keywordtype">bool</span> transpose_weights, <span class="keywordtype">bool</span> is_batched_fc_layer);</div><div class="line"><a name="l00070"></a><span class="lineno"> 70</span> </div><div class="line"><a name="l00071"></a><span class="lineno"> 71</span>  <span class="comment">// Inherited methods overridden:</span></div><div class="line"><a name="l00072"></a><span class="lineno"> 72</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#ad1717410afd0be936c6213a63c8005fb">run</a>() <span class="keyword">override</span>;</div><div class="line"><a name="l00073"></a><span class="lineno"> 73</span> </div><div class="line"><a name="l00074"></a><span class="lineno"> 74</span> <span class="keyword">private</span>:</div><div class="line"><a name="l00075"></a><span class="lineno"> 75</span>  <a class="code" href="classarm__compute_1_1_memory_group_base.xhtml">MemoryGroup</a> _memory_group;</div><div class="line"><a name="l00076"></a><span class="lineno"> 76</span>  <a class="code" href="classarm__compute_1_1_n_e_transpose_kernel.xhtml">NETransposeKernel</a> _transpose_kernel;</div><div class="line"><a name="l00077"></a><span class="lineno"> 77</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_transpose1x_w_kernel.xhtml">NEGEMMTranspose1xWKernel</a> _transpose1xW_kernel;</div><div class="line"><a name="l00078"></a><span class="lineno"> 78</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _transpose_output;</div><div class="line"><a name="l00079"></a><span class="lineno"> 79</span>  <span class="keywordtype">bool</span> _transpose_weights;</div><div class="line"><a name="l00080"></a><span class="lineno"> 80</span>  <span class="keywordtype">bool</span> _is_batched_fc_layer;</div><div class="line"><a name="l00081"></a><span class="lineno"> 81</span> };</div><div class="line"><a name="l00082"></a><span class="lineno"> 82</span> </div><div class="line"><a name="l00092"></a><span class="lineno"><a class="line" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml"> 92</a></span> <span class="keyword">class </span><a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> : <span class="keyword">public</span> <a class="code" href="classarm__compute_1_1_i_function.xhtml">IFunction</a></div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span> {</div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span> <span class="keyword">public</span>:</div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a>(std::shared_ptr<IMemoryManager> memory_manager = <span class="keyword">nullptr</span>);</div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> &) = <span class="keyword">delete</span>;</div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a>(<a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> &&) = <span class="keywordflow">default</span>;</div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> &operator=(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> &) = <span class="keyword">delete</span>;</div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> &operator=(<a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">NEFullyConnectedLayer</a> &&) = <span class="keywordflow">default</span>;</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a510f09c40e20d367e4506bc9b4e8b36e">configure</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *input, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *weights, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *biases, <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *output, <span class="keywordtype">bool</span> transpose_weights = <span class="keyword">true</span>, <span class="keywordtype">bool</span> are_weights_reshaped = <span class="keyword">false</span>);</div><div class="line"><a name="l00126"></a><span class="lineno"> 126</span>  <span class="keyword">static</span> <a class="code" href="classarm__compute_1_1_status.xhtml">Status</a> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a70b636ca822d48228a8c1f6a4d0666b2">validate</a>(<span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor_info.xhtml">ITensorInfo</a> *input, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor_info.xhtml">ITensorInfo</a> *weights, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor_info.xhtml">ITensorInfo</a> *biases, <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor_info.xhtml">ITensorInfo</a> *output, <span class="keywordtype">bool</span> transpose_weights = <span class="keyword">true</span>, <span class="keywordtype">bool</span> are_weights_reshaped = <span class="keyword">false</span>);</div><div class="line"><a name="l00127"></a><span class="lineno"> 127</span> </div><div class="line"><a name="l00128"></a><span class="lineno"> 128</span>  <span class="comment">//Inherited methods override</span></div><div class="line"><a name="l00129"></a><span class="lineno"> 129</span>  <span class="keywordtype">void</span> <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#ad1717410afd0be936c6213a63c8005fb">run</a>() <span class="keyword">override</span>;</div><div class="line"><a name="l00130"></a><span class="lineno"> 130</span> </div><div class="line"><a name="l00131"></a><span class="lineno"> 131</span> <span class="keyword">private</span>:</div><div class="line"><a name="l00132"></a><span class="lineno"> 132</span>  <a class="code" href="classarm__compute_1_1_memory_group_base.xhtml">MemoryGroup</a> _memory_group;</div><div class="line"><a name="l00133"></a><span class="lineno"> 133</span>  <a class="code" href="classarm__compute_1_1_n_e_im2_col_kernel.xhtml">NEIm2ColKernel</a> _im2col_kernel;</div><div class="line"><a name="l00134"></a><span class="lineno"> 134</span>  <a class="code" href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml">NEFullyConnectedLayerReshapeWeights</a> _reshape_weights_kernel;</div><div class="line"><a name="l00135"></a><span class="lineno"> 135</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_interleave4x4_kernel.xhtml">NEGEMMInterleave4x4Kernel</a> _interleave4x4_kernel;</div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_matrix_multiply_kernel.xhtml">NEGEMMMatrixMultiplyKernel</a> _mm_kernel;</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>  <a class="code" href="classarm__compute_1_1_n_e_g_e_m_m_matrix_accumulate_biases_kernel.xhtml">NEGEMMMatrixAccumulateBiasesKernel</a> _accumulate_biases_kernel;</div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _im2col_output;</div><div class="line"><a name="l00139"></a><span class="lineno"> 139</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _interleave4x4_output;</div><div class="line"><a name="l00140"></a><span class="lineno"> 140</span>  <a class="code" href="classarm__compute_1_1_tensor.xhtml">Tensor</a> _reshape_weights_output;</div><div class="line"><a name="l00141"></a><span class="lineno"> 141</span>  <span class="keywordtype">bool</span> _are_weights_reshaped;</div><div class="line"><a name="l00142"></a><span class="lineno"> 142</span>  <span class="keywordtype">bool</span> _is_batched_fc_layer;</div><div class="line"><a name="l00143"></a><span class="lineno"> 143</span>  <span class="keywordtype">bool</span> _linearize_input;</div><div class="line"><a name="l00144"></a><span class="lineno"> 144</span>  <span class="keywordtype">bool</span> _accumulate_biases;</div><div class="line"><a name="l00145"></a><span class="lineno"> 145</span>  <span class="keyword">const</span> <a class="code" href="classarm__compute_1_1_i_tensor.xhtml">ITensor</a> *_original_weights;</div><div class="line"><a name="l00146"></a><span class="lineno"> 146</span> };</div><div class="line"><a name="l00147"></a><span class="lineno"> 147</span> } <span class="comment">// namespace arm_compute</span></div><div class="line"><a name="l00148"></a><span class="lineno"> 148</span> <span class="preprocessor">#endif </span><span class="comment">/* __ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H__ */</span><span class="preprocessor"></span></div><div class="ttc" id="_n_e_g_e_m_m_interleave4x4_kernel_8h_xhtml"><div class="ttname"><a href="_n_e_g_e_m_m_interleave4x4_kernel_8h.xhtml">NEGEMMInterleave4x4Kernel.h</a></div></div>
<div class="ttc" id="_n_e_g_e_m_m_matrix_multiply_kernel_8h_xhtml"><div class="ttname"><a href="_n_e_g_e_m_m_matrix_multiply_kernel_8h.xhtml">NEGEMMMatrixMultiplyKernel.h</a></div></div>
<div class="ttc" id="classarm__compute_1_1_i_function_xhtml"><div class="ttname"><a href="classarm__compute_1_1_i_function.xhtml">arm_compute::IFunction</a></div><div class="ttdoc">Base class for all functions. </div><div class="ttdef"><b>Definition:</b> <a href="_i_function_8h_source.xhtml#l00030">IFunction.h:30</a></div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml">arm_compute::NEFullyConnectedLayerReshapeWeights</a></div><div class="ttdoc">Basic function to reshape the weights of Fully Connected layer with NEON. </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_fully_connected_layer_8h_source.xhtml#l00047">NEFullyConnectedLayer.h:47</a></div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights_xhtml_a510f09c40e20d367e4506bc9b4e8b36e"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a510f09c40e20d367e4506bc9b4e8b36e">arm_compute::NEFullyConnectedLayerReshapeWeights::configure</a></div><div class="ttdeci">void configure(const ITensor *input, ITensor *output, bool transpose_weights, bool is_batched_fc_layer)</div><div class="ttdoc">Set the input and output tensors. </div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_im2_col_kernel_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_im2_col_kernel.xhtml">arm_compute::NEIm2ColKernel</a></div><div class="ttdoc">Interface for the im2col reshape kernel. </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_im2_col_kernel_8h_source.xhtml#l00057">NEIm2ColKernel.h:57</a></div></div>
+<div class="ttc" id="classarm__compute_1_1_i_tensor_info_xhtml"><div class="ttname"><a href="classarm__compute_1_1_i_tensor_info.xhtml">arm_compute::ITensorInfo</a></div><div class="ttdoc">Store the tensor&#39;s metadata. </div><div class="ttdef"><b>Definition:</b> <a href="_i_tensor_info_8h_source.xhtml#l00040">ITensorInfo.h:40</a></div></div>
+<div class="ttc" id="classarm__compute_1_1_status_xhtml"><div class="ttname"><a href="classarm__compute_1_1_status.xhtml">arm_compute::Status</a></div><div class="ttdoc">Status class. </div><div class="ttdef"><b>Definition:</b> <a href="_error_8h_source.xhtml#l00051">Error.h:51</a></div></div>
<div class="ttc" id="classarm__compute_1_1_i_tensor_xhtml"><div class="ttname"><a href="classarm__compute_1_1_i_tensor.xhtml">arm_compute::ITensor</a></div><div class="ttdoc">Interface for NEON tensor. </div><div class="ttdef"><b>Definition:</b> <a href="_i_tensor_8h_source.xhtml#l00036">ITensor.h:36</a></div></div>
-<div class="ttc" id="namespacearm__compute_xhtml"><div class="ttname"><a href="namespacearm__compute.xhtml">arm_compute</a></div><div class="ttdoc">This file contains all available output stages for GEMMLowp on OpenCL. </div><div class="ttdef"><b>Definition:</b> <a href="01__library_8dox_source.xhtml#l00001">01_library.dox:1</a></div></div>
+<div class="ttc" id="namespacearm__compute_xhtml"><div class="ttname"><a href="namespacearm__compute.xhtml">arm_compute</a></div><div class="ttdoc">This file contains all available output stages for GEMMLowp on OpenCL. </div><div class="ttdef"><b>Definition:</b> <a href="00__introduction_8dox_source.xhtml#l00001">00_introduction.dox:1</a></div></div>
<div class="ttc" id="_n_e_g_e_m_m_transpose1x_w_kernel_8h_xhtml"><div class="ttname"><a href="_n_e_g_e_m_m_transpose1x_w_kernel_8h.xhtml">NEGEMMTranspose1xWKernel.h</a></div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_g_e_m_m_interleave4x4_kernel_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_g_e_m_m_interleave4x4_kernel.xhtml">arm_compute::NEGEMMInterleave4x4Kernel</a></div><div class="ttdoc">NEON kernel to interleave the elements of a matrix. </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_g_e_m_m_interleave4x4_kernel_8h_source.xhtml#l00052">NEGEMMInterleave4x4Kernel.h:52</a></div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_transpose_kernel_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_transpose_kernel.xhtml">arm_compute::NETransposeKernel</a></div><div class="ttdoc">NEON kernel which transposes the elements of a matrix. </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_transpose_kernel_8h_source.xhtml#l00038">NETransposeKernel.h:38</a></div></div>
<div class="ttc" id="classarm__compute_1_1_tensor_xhtml"><div class="ttname"><a href="classarm__compute_1_1_tensor.xhtml">arm_compute::Tensor</a></div><div class="ttdoc">Basic implementation of the tensor interface. </div><div class="ttdef"><b>Definition:</b> <a href="runtime_2_tensor_8h_source.xhtml#l00037">Tensor.h:37</a></div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights_xhtml_ad1717410afd0be936c6213a63c8005fb"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#ad1717410afd0be936c6213a63c8005fb">arm_compute::NEFullyConnectedLayerReshapeWeights::run</a></div><div class="ttdeci">void run() override</div><div class="ttdoc">Run the kernels contained in the function. </div></div>
<div class="ttc" id="_memory_group_8h_xhtml"><div class="ttname"><a href="_memory_group_8h.xhtml">MemoryGroup.h</a></div></div>
-<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">arm_compute::NEFullyConnectedLayer</a></div><div class="ttdoc">Basic function to compute a Fully Connected layer on NEON. </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_fully_connected_layer_8h_source.xhtml#l00082">NEFullyConnectedLayer.h:82</a></div></div>
+<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer.xhtml">arm_compute::NEFullyConnectedLayer</a></div><div class="ttdoc">Basic function to compute a Fully Connected layer on NEON. </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_fully_connected_layer_8h_source.xhtml#l00092">NEFullyConnectedLayer.h:92</a></div></div>
<div class="ttc" id="classarm__compute_1_1_memory_group_base_xhtml"><div class="ttname"><a href="classarm__compute_1_1_memory_group_base.xhtml">arm_compute::MemoryGroupBase< Tensor ></a></div></div>
<div class="ttc" id="_n_e_g_e_m_m_matrix_accumulate_biases_kernel_8h_xhtml"><div class="ttname"><a href="_n_e_g_e_m_m_matrix_accumulate_biases_kernel_8h.xhtml">NEGEMMMatrixAccumulateBiasesKernel.h</a></div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights_xhtml_af20812ea72913d388a9a70f9d28e7c36"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#af20812ea72913d388a9a70f9d28e7c36">arm_compute::NEFullyConnectedLayerReshapeWeights::NEFullyConnectedLayerReshapeWeights</a></div><div class="ttdeci">NEFullyConnectedLayerReshapeWeights(std::shared_ptr< IMemoryManager > memory_manager=nullptr)</div><div class="ttdoc">Constructor. </div></div>
<div class="ttc" id="_i_function_8h_xhtml"><div class="ttname"><a href="_i_function_8h.xhtml">IFunction.h</a></div></div>
<div class="ttc" id="_n_e_im2_col_kernel_8h_xhtml"><div class="ttname"><a href="_n_e_im2_col_kernel_8h.xhtml">NEIm2ColKernel.h</a></div></div>
<div class="ttc" id="runtime_2_tensor_8h_xhtml"><div class="ttname"><a href="runtime_2_tensor_8h.xhtml">Tensor.h</a></div></div>
+<div class="ttc" id="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights_xhtml_a70b636ca822d48228a8c1f6a4d0666b2"><div class="ttname"><a href="classarm__compute_1_1_n_e_fully_connected_layer_reshape_weights.xhtml#a70b636ca822d48228a8c1f6a4d0666b2">arm_compute::NEFullyConnectedLayerReshapeWeights::validate</a></div><div class="ttdeci">static Status validate(const ITensorInfo *input, const ITensorInfo *output, bool transpose_weights, bool is_batched_fc_layer)</div><div class="ttdoc">Static function to check if given info will lead to a valid configuration of CLFullyConnectedLayerRes...</div></div>
<div class="ttc" id="classarm__compute_1_1_n_e_g_e_m_m_matrix_multiply_kernel_xhtml"><div class="ttname"><a href="classarm__compute_1_1_n_e_g_e_m_m_matrix_multiply_kernel.xhtml">arm_compute::NEGEMMMatrixMultiplyKernel</a></div><div class="ttdoc">NEON kernel to multiply two input matrices "A" and "B". </div><div class="ttdef"><b>Definition:</b> <a href="_n_e_g_e_m_m_matrix_multiply_kernel_8h_source.xhtml#l00039">NEGEMMMatrixMultiplyKernel.h:39</a></div></div>
</div><!-- fragment --></div><!-- contents -->
</div><!-- doc-content -->
<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
<ul>
<li class="navelem"><a class="el" href="dir_214608ef36d61442cb2b0c1c4e9a7def.xhtml">arm_compute</a></li><li class="navelem"><a class="el" href="dir_8bc26130589aa16388b5a02f17abf2c2.xhtml">runtime</a></li><li class="navelem"><a class="el" href="dir_65e289039e1347f87d412b0a1b1a312c.xhtml">NEON</a></li><li class="navelem"><a class="el" href="dir_345bd7154d1542d1e03fd2836959a19d.xhtml">functions</a></li><li class="navelem"><a class="el" href="_n_e_fully_connected_layer_8h.xhtml">NEFullyConnectedLayer.h</a></li>
- <li class="footer">Generated on Wed Jan 24 2018 14:30:40 for Compute Library by
+ <li class="footer">Generated on Wed May 23 2018 11:36:36 for Compute Library by
<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.11 </li>
</ul>