blob: bb6691bf9b7919cf2bf9c39f46dfb539a1cb2442 [file] [log] [blame]
<!-- HTML header for doxygen 1.8.15-->
<!-- Remember to use version doxygen 1.8.15 +-->
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<meta name="generator" content="Doxygen 1.8.15"/>
<meta name="robots" content="NOINDEX, NOFOLLOW" /> <!-- Prevent indexing by search engines -->
<title>Compute Library: src/core/CL/cl_kernels/gemm.cl Source File</title>
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="jquery.js"></script>
<script type="text/javascript" src="dynsections.js"></script>
<link href="navtree.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="resize.js"></script>
<script type="text/javascript" src="navtreedata.js"></script>
<script type="text/javascript" src="navtree.js"></script>
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
$(document).ready(initResizable);
/* @license-end */</script>
<link href="search/search.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="search/searchdata.js"></script>
<script type="text/javascript" src="search/search.js"></script>
<script type="text/x-mathjax-config">
MathJax.Hub.Config({
extensions: ["tex2jax.js"],
jax: ["input/TeX","output/HTML-CSS"],
});
</script><script type="text/javascript" async="async" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
<link href="stylesheet.css" rel="stylesheet" type="text/css"/>
</head>
<body>
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
<div id="titlearea">
<table cellspacing="0" cellpadding="0">
<tbody>
<tr style="height: 56px;">
<img alt="Compute Library" src="https://raw.githubusercontent.com/ARM-software/ComputeLibrary/gh-pages/ACL_logo.png" style="max-width: 100%;margin-top: 15px;margin-left: 10px"/>
<td style="padding-left: 0.5em;">
<div id="projectname">
&#160;<span id="projectnumber">19.08</span>
</div>
</td>
</tr>
</tbody>
</table>
</div>
<!-- end header part -->
<!-- Generated by Doxygen 1.8.15 -->
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
var searchBox = new SearchBox("searchBox", "search",false,'Search');
/* @license-end */
</script>
<script type="text/javascript" src="menudata.js"></script>
<script type="text/javascript" src="menu.js"></script>
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
$(function() {
initMenu('',true,false,'search.php','Search');
$(document).ready(function() { init_search(); });
});
/* @license-end */</script>
<div id="main-nav"></div>
</div><!-- top -->
<div id="side-nav" class="ui-resizable side-nav-resizable">
<div id="nav-tree">
<div id="nav-tree-contents">
<div id="nav-sync" class="sync"></div>
</div>
</div>
<div id="splitbar" style="-moz-user-select:none;"
class="ui-resizable-handle">
</div>
</div>
<script type="text/javascript">
/* @license magnet:?xt=urn:btih:cf05388f2679ee054f2beb29a391d25f4e673ac3&amp;dn=gpl-2.0.txt GPL-v2 */
$(document).ready(function(){initNavTree('gemm_8cl_source.xhtml','');});
/* @license-end */
</script>
<div id="doc-content">
<!-- window showing the filter options -->
<div id="MSearchSelectWindow"
onmouseover="return searchBox.OnSearchSelectShow()"
onmouseout="return searchBox.OnSearchSelectHide()"
onkeydown="return searchBox.OnSearchSelectKey(event)">
</div>
<!-- iframe showing the search results (closed by default) -->
<div id="MSearchResultsWindow">
<iframe src="javascript:void(0)" frameborder="0"
name="MSearchResults" id="MSearchResults">
</iframe>
</div>
<div class="header">
<div class="headertitle">
<div class="title">gemm.cl</div> </div>
</div><!--header-->
<div class="contents">
<a href="gemm_8cl.xhtml">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a name="l00001"></a><span class="lineno"> 1</span>&#160;<span class="comment">/*</span></div><div class="line"><a name="l00002"></a><span class="lineno"> 2</span>&#160;<span class="comment"> * Copyright (c) 2017-2019 ARM Limited.</span></div><div class="line"><a name="l00003"></a><span class="lineno"> 3</span>&#160;<span class="comment"> *</span></div><div class="line"><a name="l00004"></a><span class="lineno"> 4</span>&#160;<span class="comment"> * SPDX-License-Identifier: MIT</span></div><div class="line"><a name="l00005"></a><span class="lineno"> 5</span>&#160;<span class="comment"> *</span></div><div class="line"><a name="l00006"></a><span class="lineno"> 6</span>&#160;<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>&#160;<span class="comment"> * of this software and associated documentation files (the &quot;Software&quot;), to</span></div><div class="line"><a name="l00008"></a><span class="lineno"> 8</span>&#160;<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>&#160;<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>&#160;<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>&#160;<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>&#160;<span class="comment"> *</span></div><div class="line"><a name="l00013"></a><span class="lineno"> 13</span>&#160;<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>&#160;<span class="comment"> * copies or substantial portions of the Software.</span></div><div class="line"><a name="l00015"></a><span class="lineno"> 15</span>&#160;<span class="comment"> *</span></div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160;<span class="comment"> * THE SOFTWARE IS PROVIDED &quot;AS IS&quot;, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR</span></div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160;<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>&#160;<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>&#160;<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>&#160;<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>&#160;<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>&#160;<span class="comment"> * SOFTWARE.</span></div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160;<span class="comment"> */</span></div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="gemm__helpers_8h.xhtml">gemm_helpers.h</a>&quot;</span></div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160;<span class="preprocessor">#include &quot;<a class="code" href="repeat_8h.xhtml">repeat.h</a>&quot;</span></div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160;</div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160;<span class="preprocessor">#if defined(M0) &amp;&amp; defined(K0) &amp;&amp; defined(V0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(SRC_WIDTH)</span></div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160;<span class="preprocessor">#define INC2 (VEC_DATA_TYPE(uint, 2))(0, 1)</span></div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160;<span class="preprocessor">#define INC3 (VEC_DATA_TYPE(uint, 3))(0, 1, 2)</span></div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160;<span class="preprocessor">#define INC4 (VEC_DATA_TYPE(uint, 4))(0, 1, 2, 3)</span></div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160;<span class="preprocessor">#define INC8 (VEC_DATA_TYPE(uint, 8))(0, 1, 2, 3, 4, 5, 6, 7)</span></div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160;<span class="preprocessor">#define INC16 (VEC_DATA_TYPE(uint, 16))(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15)</span></div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160;<span class="preprocessor">#define CONCAT_INC(K0) INC##K0</span></div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160;<span class="preprocessor">#define INC(K0) CONCAT_INC(K0)</span></div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160;</div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160;<span class="preprocessor">#if(SRC_WIDTH % K0)</span></div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160;<span class="preprocessor">#define BOUNDARY_CONDITION_X(x, a) \</span></div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160;<span class="preprocessor"> a = select(0, a, CONVERT(((x * (VEC_DATA_TYPE(uint, K0))K0 + INC(K0)) &lt; (VEC_DATA_TYPE(uint, K0))SRC_WIDTH), VEC_DATA_TYPE(DATA_TYPE, K0))); \</span></div><div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160;<span class="preprocessor">#else // (SRC_WIDTH % K0)</span></div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160;<span class="preprocessor">#define BOUNDARY_CONDITION_X(x, a) \</span></div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160;<span class="preprocessor"> ({})</span></div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160;<span class="preprocessor">#endif // (SRC_WIDTH % K0)</span></div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160;</div><div class="line"><a name="l00082"></a><span class="lineno"> 82</span>&#160;__kernel <span class="keywordtype">void</span> gemm_reshape_lhs_matrix_nt(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>),</div><div class="line"><a name="l00083"></a><span class="lineno"> 83</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>)</div><div class="line"><a name="l00084"></a><span class="lineno"> 84</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l00085"></a><span class="lineno"> 85</span>&#160; ,</div><div class="line"><a name="l00086"></a><span class="lineno"> 86</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l00087"></a><span class="lineno"> 87</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l00088"></a><span class="lineno"> 88</span>&#160; )</div><div class="line"><a name="l00089"></a><span class="lineno"> 89</span>&#160;{</div><div class="line"><a name="l00090"></a><span class="lineno"> 90</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l00091"></a><span class="lineno"> 91</span>&#160;<span class="preprocessor">#define BLOCK_SIZE ((M0) * (K0))</span></div><div class="line"><a name="l00092"></a><span class="lineno"> 92</span>&#160;</div><div class="line"><a name="l00093"></a><span class="lineno"> 93</span>&#160; <span class="comment">// Output offset X</span></div><div class="line"><a name="l00094"></a><span class="lineno"> 94</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00095"></a><span class="lineno"> 95</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (K0)</span></div><div class="line"><a name="l00096"></a><span class="lineno"> 96</span>&#160;<span class="preprocessor">#else // defined(INTERLEAVE)</span></div><div class="line"><a name="l00097"></a><span class="lineno"> 97</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (BLOCK_SIZE)</span></div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00099"></a><span class="lineno"> 99</span>&#160;</div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span>&#160; <span class="comment">// Output step X</span></div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (K0) * (V0)</span></div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160;<span class="preprocessor">#else // Do not interleave</span></div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (K0)</span></div><div class="line"><a name="l00105"></a><span class="lineno"> 105</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160;</div><div class="line"><a name="l00107"></a><span class="lineno"> 107</span>&#160; <span class="comment">// Compute source and destination addresses</span></div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l00110"></a><span class="lineno"> 110</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l00111"></a><span class="lineno"> 111</span>&#160;</div><div class="line"><a name="l00112"></a><span class="lineno"> 112</span>&#160; <span class="comment">// ------------------ Compute input/output addresses ---------------------------</span></div><div class="line"><a name="l00113"></a><span class="lineno"> 113</span>&#160;</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>&#160; <span class="comment">// Compute the input address</span></div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span>&#160; __global uchar *input_ptr = src_ptr + src_offset_first_element_in_bytes + x * (uint)K0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + y * (uint)M0 * src_stride_y;</div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span>&#160;</div><div class="line"><a name="l00117"></a><span class="lineno"> 117</span>&#160; <span class="comment">// Compute the output address</span></div><div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160; __global uchar *output_ptr = dst_ptr + dst_offset_first_element_in_bytes + (x * (uint)BLOCK_SIZE * (uint)V0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + ((y / (uint)V0) * (uint)dst_stride_y) + ((y % V0) *</div><div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160; (uint)OUTPUT_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span>&#160;</div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span>&#160; <span class="comment">// Create variables: uint zin0=0, zin1=0, zin2=0...zin(M0-1)=0;</span></div><div class="line"><a name="l00122"></a><span class="lineno"> 122</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, uint, zin, 0);</div><div class="line"><a name="l00123"></a><span class="lineno"> 123</span>&#160;</div><div class="line"><a name="l00124"></a><span class="lineno"> 124</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l00125"></a><span class="lineno"> 125</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l00126"></a><span class="lineno"> 126</span>&#160; <span class="comment">// multiply src_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l00127"></a><span class="lineno"> 127</span>&#160;</div><div class="line"><a name="l00128"></a><span class="lineno"> 128</span>&#160; input_ptr += z * (uint)src_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l00129"></a><span class="lineno"> 129</span>&#160;</div><div class="line"><a name="l00130"></a><span class="lineno"> 130</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l00131"></a><span class="lineno"> 131</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zin, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, cross_plane_pad, src_stride_y);</div><div class="line"><a name="l00132"></a><span class="lineno"> 132</span>&#160;</div><div class="line"><a name="l00133"></a><span class="lineno"> 133</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l00134"></a><span class="lineno"> 134</span>&#160;</div><div class="line"><a name="l00135"></a><span class="lineno"> 135</span>&#160; input_ptr += z * (uint)src_stride_z;</div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span>&#160;</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>&#160;</div><div class="line"><a name="l00139"></a><span class="lineno"> 139</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l00140"></a><span class="lineno"> 140</span>&#160; output_ptr += z * (uint)dst_stride_z;</div><div class="line"><a name="l00141"></a><span class="lineno"> 141</span>&#160;</div><div class="line"><a name="l00142"></a><span class="lineno"> 142</span>&#160; <span class="comment">// ---------------------------Load input values --------------------------------</span></div><div class="line"><a name="l00143"></a><span class="lineno"> 143</span>&#160; <span class="comment">// Load values from the LHS matrix</span></div><div class="line"><a name="l00144"></a><span class="lineno"> 144</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, input_ptr, 0, src_stride_y, zin);</div><div class="line"><a name="l00145"></a><span class="lineno"> 145</span>&#160; BOUNDARY_CONDITION_X(x, a0);</div><div class="line"><a name="l00146"></a><span class="lineno"> 146</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l00147"></a><span class="lineno"> 147</span>&#160; BOUNDARY_CONDITION_X(x, a1);</div><div class="line"><a name="l00148"></a><span class="lineno"> 148</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l00149"></a><span class="lineno"> 149</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l00150"></a><span class="lineno"> 150</span>&#160; BOUNDARY_CONDITION_X(x, a2);</div><div class="line"><a name="l00151"></a><span class="lineno"> 151</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l00152"></a><span class="lineno"> 152</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l00153"></a><span class="lineno"> 153</span>&#160; BOUNDARY_CONDITION_X(x, a3);</div><div class="line"><a name="l00154"></a><span class="lineno"> 154</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l00155"></a><span class="lineno"> 155</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l00156"></a><span class="lineno"> 156</span>&#160; BOUNDARY_CONDITION_X(x, a4);</div><div class="line"><a name="l00157"></a><span class="lineno"> 157</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l00158"></a><span class="lineno"> 158</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l00159"></a><span class="lineno"> 159</span>&#160; BOUNDARY_CONDITION_X(x, a5);</div><div class="line"><a name="l00160"></a><span class="lineno"> 160</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l00161"></a><span class="lineno"> 161</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l00162"></a><span class="lineno"> 162</span>&#160; BOUNDARY_CONDITION_X(x, a6);</div><div class="line"><a name="l00163"></a><span class="lineno"> 163</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l00164"></a><span class="lineno"> 164</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l00165"></a><span class="lineno"> 165</span>&#160; BOUNDARY_CONDITION_X(x, a7);</div><div class="line"><a name="l00166"></a><span class="lineno"> 166</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l00167"></a><span class="lineno"> 167</span>&#160; <span class="comment">// ---------------------------Store output values ------------------------------</span></div><div class="line"><a name="l00168"></a><span class="lineno"> 168</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zout, 0);</div><div class="line"><a name="l00169"></a><span class="lineno"> 169</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, output_ptr, OUTPUT_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zout);</div><div class="line"><a name="l00170"></a><span class="lineno"> 170</span>&#160;</div><div class="line"><a name="l00171"></a><span class="lineno"> 171</span>&#160;<span class="preprocessor">#undef BLOCK_SIZE</span></div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span>&#160;<span class="preprocessor">#undef OUTPUT_OFFSET_X</span></div><div class="line"><a name="l00173"></a><span class="lineno"> 173</span>&#160;<span class="preprocessor">#undef OUTPUT_STEP_X</span></div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span>&#160;}</div><div class="line"><a name="l00175"></a><span class="lineno"> 175</span>&#160;</div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span>&#160;<span class="preprocessor">#if M0 == 2</span></div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00179"></a><span class="lineno"> 179</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, M0) \</span></div><div class="line"><a name="l00180"></a><span class="lineno"> 180</span>&#160;<span class="preprocessor"> res = (VEC_DATA_TYPE(DATA_TYPE, M0))(a0.s##i, a1.s##i); \</span></div><div class="line"><a name="l00181"></a><span class="lineno"> 181</span>&#160;<span class="preprocessor"> VSTORE(M0) \</span></div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160;<span class="preprocessor"> (res, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00184"></a><span class="lineno"> 184</span>&#160;<span class="preprocessor">#elif M0 == 3 // M0 == 3</span></div><div class="line"><a name="l00185"></a><span class="lineno"> 185</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00186"></a><span class="lineno"> 186</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00187"></a><span class="lineno"> 187</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, M0) \</span></div><div class="line"><a name="l00188"></a><span class="lineno"> 188</span>&#160;<span class="preprocessor"> res = (VEC_DATA_TYPE(DATA_TYPE, M0))(a0.s##i, a1.s##i, a2.s##i); \</span></div><div class="line"><a name="l00189"></a><span class="lineno"> 189</span>&#160;<span class="preprocessor"> VSTORE(M0) \</span></div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>&#160;<span class="preprocessor"> (res, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00192"></a><span class="lineno"> 192</span>&#160;<span class="preprocessor">#elif M0 == 4 // M0 == 4</span></div><div class="line"><a name="l00193"></a><span class="lineno"> 193</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00195"></a><span class="lineno"> 195</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, M0) \</span></div><div class="line"><a name="l00196"></a><span class="lineno"> 196</span>&#160;<span class="preprocessor"> res = (VEC_DATA_TYPE(DATA_TYPE, M0))(a0.s##i, a1.s##i, a2.s##i, a3.s##i); \</span></div><div class="line"><a name="l00197"></a><span class="lineno"> 197</span>&#160;<span class="preprocessor"> VSTORE(M0) \</span></div><div class="line"><a name="l00198"></a><span class="lineno"> 198</span>&#160;<span class="preprocessor"> (res, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00199"></a><span class="lineno"> 199</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00200"></a><span class="lineno"> 200</span>&#160;<span class="preprocessor">#elif M0 == 5 // M0 == 5</span></div><div class="line"><a name="l00201"></a><span class="lineno"> 201</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00202"></a><span class="lineno"> 202</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00203"></a><span class="lineno"> 203</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, 4) \</span></div><div class="line"><a name="l00204"></a><span class="lineno"> 204</span>&#160;<span class="preprocessor"> res0 = (VEC_DATA_TYPE(DATA_TYPE, 4))(a0.s##i, a1.s##i, a2.s##i, a3.s##i); \</span></div><div class="line"><a name="l00205"></a><span class="lineno"> 205</span>&#160;<span class="preprocessor"> DATA_TYPE res1 = a4.s##i; \</span></div><div class="line"><a name="l00206"></a><span class="lineno"> 206</span>&#160;<span class="preprocessor"> VSTORE(4) \</span></div><div class="line"><a name="l00207"></a><span class="lineno"> 207</span>&#160;<span class="preprocessor"> (res0, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00208"></a><span class="lineno"> 208</span>&#160;<span class="preprocessor"> *((__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE)) + 4) = res1; \</span></div><div class="line"><a name="l00209"></a><span class="lineno"> 209</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00210"></a><span class="lineno"> 210</span>&#160;<span class="preprocessor">#elif M0 == 6 // M0 == 6</span></div><div class="line"><a name="l00211"></a><span class="lineno"> 211</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00212"></a><span class="lineno"> 212</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00213"></a><span class="lineno"> 213</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, 4) \</span></div><div class="line"><a name="l00214"></a><span class="lineno"> 214</span>&#160;<span class="preprocessor"> res0 = (VEC_DATA_TYPE(DATA_TYPE, 4))(a0.s##i, a1.s##i, a2.s##i, a3.s##i); \</span></div><div class="line"><a name="l00215"></a><span class="lineno"> 215</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, 2) \</span></div><div class="line"><a name="l00216"></a><span class="lineno"> 216</span>&#160;<span class="preprocessor"> res1 = (VEC_DATA_TYPE(DATA_TYPE, 2))(a4.s##i, a5.s##i); \</span></div><div class="line"><a name="l00217"></a><span class="lineno"> 217</span>&#160;<span class="preprocessor"> VSTORE(4) \</span></div><div class="line"><a name="l00218"></a><span class="lineno"> 218</span>&#160;<span class="preprocessor"> (res0, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00219"></a><span class="lineno"> 219</span>&#160;<span class="preprocessor"> VSTORE(2) \</span></div><div class="line"><a name="l00220"></a><span class="lineno"> 220</span>&#160;<span class="preprocessor"> (res1, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE)) + 4); \</span></div><div class="line"><a name="l00221"></a><span class="lineno"> 221</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00222"></a><span class="lineno"> 222</span>&#160;<span class="preprocessor">#elif M0 == 7 // M0 == 7</span></div><div class="line"><a name="l00223"></a><span class="lineno"> 223</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00224"></a><span class="lineno"> 224</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00225"></a><span class="lineno"> 225</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, 4) \</span></div><div class="line"><a name="l00226"></a><span class="lineno"> 226</span>&#160;<span class="preprocessor"> res0 = (VEC_DATA_TYPE(DATA_TYPE, 4))(a0.s##i, a1.s##i, a2.s##i, a3.s##i); \</span></div><div class="line"><a name="l00227"></a><span class="lineno"> 227</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, 3) \</span></div><div class="line"><a name="l00228"></a><span class="lineno"> 228</span>&#160;<span class="preprocessor"> res1 = (VEC_DATA_TYPE(DATA_TYPE, 3))(a4.s##i, a5.s##i, a6.s##i); \</span></div><div class="line"><a name="l00229"></a><span class="lineno"> 229</span>&#160;<span class="preprocessor"> VSTORE(4) \</span></div><div class="line"><a name="l00230"></a><span class="lineno"> 230</span>&#160;<span class="preprocessor"> (res0, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00231"></a><span class="lineno"> 231</span>&#160;<span class="preprocessor"> VSTORE(3) \</span></div><div class="line"><a name="l00232"></a><span class="lineno"> 232</span>&#160;<span class="preprocessor"> (res1, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE)) + 4); \</span></div><div class="line"><a name="l00233"></a><span class="lineno"> 233</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00234"></a><span class="lineno"> 234</span>&#160;<span class="preprocessor">#elif M0 == 8 // M0 == 8</span></div><div class="line"><a name="l00235"></a><span class="lineno"> 235</span>&#160;<span class="preprocessor">#define TRANSPOSE_COLUMN_AND_STORE(output_ptr, output_step_x, i) \</span></div><div class="line"><a name="l00236"></a><span class="lineno"> 236</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00237"></a><span class="lineno"> 237</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, M0) \</span></div><div class="line"><a name="l00238"></a><span class="lineno"> 238</span>&#160;<span class="preprocessor"> res = (VEC_DATA_TYPE(DATA_TYPE, M0))(a0.s##i, a1.s##i, a2.s##i, a3.s##i, a4.s##i, a5.s##i, a6.s##i, a7.s##i); \</span></div><div class="line"><a name="l00239"></a><span class="lineno"> 239</span>&#160;<span class="preprocessor"> VSTORE(M0) \</span></div><div class="line"><a name="l00240"></a><span class="lineno"> 240</span>&#160;<span class="preprocessor"> (res, 0, (__global DATA_TYPE *)(output_ptr + 0x##i * output_step_x * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l00241"></a><span class="lineno"> 241</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00242"></a><span class="lineno"> 242</span>&#160;<span class="preprocessor">#else // M0 not supported</span></div><div class="line"><a name="l00243"></a><span class="lineno"> 243</span>&#160;<span class="preprocessor">#error &quot;M0 value not supported&quot;</span></div><div class="line"><a name="l00244"></a><span class="lineno"> 244</span>&#160;<span class="preprocessor">#endif // N0 conditions</span></div><div class="line"><a name="l00245"></a><span class="lineno"> 245</span>&#160;</div><div class="line"><a name="l00282"></a><span class="lineno"> 282</span>&#160;__kernel <span class="keywordtype">void</span> gemm_reshape_lhs_matrix_t(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>),</div><div class="line"><a name="l00283"></a><span class="lineno"> 283</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>)</div><div class="line"><a name="l00284"></a><span class="lineno"> 284</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l00285"></a><span class="lineno"> 285</span>&#160; ,</div><div class="line"><a name="l00286"></a><span class="lineno"> 286</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l00287"></a><span class="lineno"> 287</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l00288"></a><span class="lineno"> 288</span>&#160; )</div><div class="line"><a name="l00289"></a><span class="lineno"> 289</span>&#160;{</div><div class="line"><a name="l00290"></a><span class="lineno"> 290</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l00291"></a><span class="lineno"> 291</span>&#160;<span class="preprocessor">#define BLOCK_SIZE ((M0) * (K0))</span></div><div class="line"><a name="l00292"></a><span class="lineno"> 292</span>&#160;</div><div class="line"><a name="l00293"></a><span class="lineno"> 293</span>&#160; <span class="comment">// Output offset X</span></div><div class="line"><a name="l00294"></a><span class="lineno"> 294</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00295"></a><span class="lineno"> 295</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (M0)</span></div><div class="line"><a name="l00296"></a><span class="lineno"> 296</span>&#160;<span class="preprocessor">#else // defined(INTERLEAVE)</span></div><div class="line"><a name="l00297"></a><span class="lineno"> 297</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (BLOCK_SIZE)</span></div><div class="line"><a name="l00298"></a><span class="lineno"> 298</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00299"></a><span class="lineno"> 299</span>&#160;</div><div class="line"><a name="l00300"></a><span class="lineno"> 300</span>&#160; <span class="comment">// Output step X</span></div><div class="line"><a name="l00301"></a><span class="lineno"> 301</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00302"></a><span class="lineno"> 302</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (M0) * (V0)</span></div><div class="line"><a name="l00303"></a><span class="lineno"> 303</span>&#160;<span class="preprocessor">#else // Do not interleave</span></div><div class="line"><a name="l00304"></a><span class="lineno"> 304</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (M0)</span></div><div class="line"><a name="l00305"></a><span class="lineno"> 305</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00306"></a><span class="lineno"> 306</span>&#160;</div><div class="line"><a name="l00307"></a><span class="lineno"> 307</span>&#160; <span class="comment">// Compute source and destination addresses</span></div><div class="line"><a name="l00308"></a><span class="lineno"> 308</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l00309"></a><span class="lineno"> 309</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l00310"></a><span class="lineno"> 310</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l00311"></a><span class="lineno"> 311</span>&#160;</div><div class="line"><a name="l00312"></a><span class="lineno"> 312</span>&#160; <span class="comment">// ------------------ Compute input/output addresses ---------------------------</span></div><div class="line"><a name="l00313"></a><span class="lineno"> 313</span>&#160;</div><div class="line"><a name="l00314"></a><span class="lineno"> 314</span>&#160; <span class="comment">// Compute the input address</span></div><div class="line"><a name="l00315"></a><span class="lineno"> 315</span>&#160; __global uchar *input_ptr = src_ptr + src_offset_first_element_in_bytes + x * (uint)K0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + y * (uint)M0 * src_stride_y;</div><div class="line"><a name="l00316"></a><span class="lineno"> 316</span>&#160;</div><div class="line"><a name="l00317"></a><span class="lineno"> 317</span>&#160; <span class="comment">// Compute the output address</span></div><div class="line"><a name="l00318"></a><span class="lineno"> 318</span>&#160; __global uchar *output_ptr = dst_ptr + dst_offset_first_element_in_bytes + (x * (uint)BLOCK_SIZE * (uint)V0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + ((y / (uint)V0) * (uint)dst_stride_y) + ((y % V0) *</div><div class="line"><a name="l00319"></a><span class="lineno"> 319</span>&#160; (uint)OUTPUT_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l00320"></a><span class="lineno"> 320</span>&#160;</div><div class="line"><a name="l00321"></a><span class="lineno"> 321</span>&#160; <span class="comment">// Create variables: uint zin0=0, zin1=0, zin2=0...zin(M0-1)=0;</span></div><div class="line"><a name="l00322"></a><span class="lineno"> 322</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, uint, zin, 0);</div><div class="line"><a name="l00323"></a><span class="lineno"> 323</span>&#160;</div><div class="line"><a name="l00324"></a><span class="lineno"> 324</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l00325"></a><span class="lineno"> 325</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l00326"></a><span class="lineno"> 326</span>&#160; <span class="comment">// multiply src_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l00327"></a><span class="lineno"> 327</span>&#160;</div><div class="line"><a name="l00328"></a><span class="lineno"> 328</span>&#160; input_ptr += z * (uint)src_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l00329"></a><span class="lineno"> 329</span>&#160;</div><div class="line"><a name="l00330"></a><span class="lineno"> 330</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l00331"></a><span class="lineno"> 331</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zin, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, cross_plane_pad, src_stride_y);</div><div class="line"><a name="l00332"></a><span class="lineno"> 332</span>&#160;</div><div class="line"><a name="l00333"></a><span class="lineno"> 333</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l00334"></a><span class="lineno"> 334</span>&#160;</div><div class="line"><a name="l00335"></a><span class="lineno"> 335</span>&#160; input_ptr += z * (uint)src_stride_z;</div><div class="line"><a name="l00336"></a><span class="lineno"> 336</span>&#160;</div><div class="line"><a name="l00337"></a><span class="lineno"> 337</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l00338"></a><span class="lineno"> 338</span>&#160;</div><div class="line"><a name="l00339"></a><span class="lineno"> 339</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l00340"></a><span class="lineno"> 340</span>&#160; output_ptr += z * (uint)dst_stride_z;</div><div class="line"><a name="l00341"></a><span class="lineno"> 341</span>&#160;</div><div class="line"><a name="l00342"></a><span class="lineno"> 342</span>&#160; <span class="comment">// ---------------------------Load input values --------------------------------</span></div><div class="line"><a name="l00343"></a><span class="lineno"> 343</span>&#160;</div><div class="line"><a name="l00344"></a><span class="lineno"> 344</span>&#160; <span class="comment">// Load values from the LHS matrix</span></div><div class="line"><a name="l00345"></a><span class="lineno"> 345</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, input_ptr, 0, src_stride_y, zin);</div><div class="line"><a name="l00346"></a><span class="lineno"> 346</span>&#160; BOUNDARY_CONDITION_X(x, a0);</div><div class="line"><a name="l00347"></a><span class="lineno"> 347</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l00348"></a><span class="lineno"> 348</span>&#160; BOUNDARY_CONDITION_X(x, a1);</div><div class="line"><a name="l00349"></a><span class="lineno"> 349</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l00350"></a><span class="lineno"> 350</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l00351"></a><span class="lineno"> 351</span>&#160; BOUNDARY_CONDITION_X(x, a2);</div><div class="line"><a name="l00352"></a><span class="lineno"> 352</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l00353"></a><span class="lineno"> 353</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l00354"></a><span class="lineno"> 354</span>&#160; BOUNDARY_CONDITION_X(x, a3);</div><div class="line"><a name="l00355"></a><span class="lineno"> 355</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l00356"></a><span class="lineno"> 356</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l00357"></a><span class="lineno"> 357</span>&#160; BOUNDARY_CONDITION_X(x, a4);</div><div class="line"><a name="l00358"></a><span class="lineno"> 358</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l00359"></a><span class="lineno"> 359</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l00360"></a><span class="lineno"> 360</span>&#160; BOUNDARY_CONDITION_X(x, a5);</div><div class="line"><a name="l00361"></a><span class="lineno"> 361</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l00362"></a><span class="lineno"> 362</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l00363"></a><span class="lineno"> 363</span>&#160; BOUNDARY_CONDITION_X(x, a6);</div><div class="line"><a name="l00364"></a><span class="lineno"> 364</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l00365"></a><span class="lineno"> 365</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l00366"></a><span class="lineno"> 366</span>&#160; BOUNDARY_CONDITION_X(x, a7);</div><div class="line"><a name="l00367"></a><span class="lineno"> 367</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l00368"></a><span class="lineno"> 368</span>&#160; <span class="comment">// ---------------------------Transpose and store block -----------------------</span></div><div class="line"><a name="l00369"></a><span class="lineno"> 369</span>&#160;</div><div class="line"><a name="l00370"></a><span class="lineno"> 370</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 0);</div><div class="line"><a name="l00371"></a><span class="lineno"> 371</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 1);</div><div class="line"><a name="l00372"></a><span class="lineno"> 372</span>&#160;<span class="preprocessor">#if K0 &gt; 2</span></div><div class="line"><a name="l00373"></a><span class="lineno"> 373</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 2);</div><div class="line"><a name="l00374"></a><span class="lineno"> 374</span>&#160;<span class="preprocessor">#endif // K0 &gt; 2</span></div><div class="line"><a name="l00375"></a><span class="lineno"> 375</span>&#160;<span class="preprocessor">#if K0 &gt; 3</span></div><div class="line"><a name="l00376"></a><span class="lineno"> 376</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 3);</div><div class="line"><a name="l00377"></a><span class="lineno"> 377</span>&#160;<span class="preprocessor">#endif // K0 &gt; 3</span></div><div class="line"><a name="l00378"></a><span class="lineno"> 378</span>&#160;<span class="preprocessor">#if K0 &gt; 4</span></div><div class="line"><a name="l00379"></a><span class="lineno"> 379</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 4);</div><div class="line"><a name="l00380"></a><span class="lineno"> 380</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 5);</div><div class="line"><a name="l00381"></a><span class="lineno"> 381</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 6);</div><div class="line"><a name="l00382"></a><span class="lineno"> 382</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 7);</div><div class="line"><a name="l00383"></a><span class="lineno"> 383</span>&#160;<span class="preprocessor">#endif // K0 &gt; 4</span></div><div class="line"><a name="l00384"></a><span class="lineno"> 384</span>&#160;<span class="preprocessor">#if K0 &gt; 8</span></div><div class="line"><a name="l00385"></a><span class="lineno"> 385</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 8);</div><div class="line"><a name="l00386"></a><span class="lineno"> 386</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, 9);</div><div class="line"><a name="l00387"></a><span class="lineno"> 387</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, A);</div><div class="line"><a name="l00388"></a><span class="lineno"> 388</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, B);</div><div class="line"><a name="l00389"></a><span class="lineno"> 389</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, C);</div><div class="line"><a name="l00390"></a><span class="lineno"> 390</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, D);</div><div class="line"><a name="l00391"></a><span class="lineno"> 391</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, E);</div><div class="line"><a name="l00392"></a><span class="lineno"> 392</span>&#160; TRANSPOSE_COLUMN_AND_STORE(output_ptr, OUTPUT_STEP_X, F);</div><div class="line"><a name="l00393"></a><span class="lineno"> 393</span>&#160;<span class="preprocessor">#endif // K0 &gt; 8</span></div><div class="line"><a name="l00394"></a><span class="lineno"> 394</span>&#160;</div><div class="line"><a name="l00395"></a><span class="lineno"> 395</span>&#160;<span class="preprocessor">#undef BLOCK_SIZE</span></div><div class="line"><a name="l00396"></a><span class="lineno"> 396</span>&#160;<span class="preprocessor">#undef OUTPUT_OFFSET_X</span></div><div class="line"><a name="l00397"></a><span class="lineno"> 397</span>&#160;<span class="preprocessor">#undef OUTPUT_STEP_X</span></div><div class="line"><a name="l00398"></a><span class="lineno"> 398</span>&#160;}</div><div class="line"><a name="l00399"></a><span class="lineno"> 399</span>&#160;<span class="preprocessor">#endif // defined(M0) &amp;&amp; defined(K0) &amp;&amp; defined(V0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(SRC_WIDTH)</span></div><div class="line"><a name="l00400"></a><span class="lineno"> 400</span>&#160;</div><div class="line"><a name="l00401"></a><span class="lineno"> 401</span>&#160;<span class="preprocessor">#if defined(K0) &amp;&amp; defined(N0) &amp;&amp; defined(H0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(SRC_HEIGHT)</span></div><div class="line"><a name="l00402"></a><span class="lineno"> 402</span>&#160;</div><div class="line"><a name="l00432"></a><span class="lineno"> 432</span>&#160;__kernel <span class="keywordtype">void</span> gemm_reshape_rhs_matrix_nt(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>),</div><div class="line"><a name="l00433"></a><span class="lineno"> 433</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>))</div><div class="line"><a name="l00434"></a><span class="lineno"> 434</span>&#160;{</div><div class="line"><a name="l00435"></a><span class="lineno"> 435</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l00436"></a><span class="lineno"> 436</span>&#160;<span class="preprocessor">#define BLOCK_SIZE ((K0) * (N0))</span></div><div class="line"><a name="l00437"></a><span class="lineno"> 437</span>&#160;</div><div class="line"><a name="l00438"></a><span class="lineno"> 438</span>&#160; <span class="comment">// Output offset X</span></div><div class="line"><a name="l00439"></a><span class="lineno"> 439</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00440"></a><span class="lineno"> 440</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (N0)</span></div><div class="line"><a name="l00441"></a><span class="lineno"> 441</span>&#160;<span class="preprocessor">#else // defined(INTERLEAVE)</span></div><div class="line"><a name="l00442"></a><span class="lineno"> 442</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (BLOCK_SIZE)</span></div><div class="line"><a name="l00443"></a><span class="lineno"> 443</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00444"></a><span class="lineno"> 444</span>&#160;</div><div class="line"><a name="l00445"></a><span class="lineno"> 445</span>&#160; <span class="comment">// Output step X</span></div><div class="line"><a name="l00446"></a><span class="lineno"> 446</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00447"></a><span class="lineno"> 447</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (N0) * (H0)</span></div><div class="line"><a name="l00448"></a><span class="lineno"> 448</span>&#160;<span class="preprocessor">#else // Do not interleave</span></div><div class="line"><a name="l00449"></a><span class="lineno"> 449</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (N0)</span></div><div class="line"><a name="l00450"></a><span class="lineno"> 450</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00451"></a><span class="lineno"> 451</span>&#160;</div><div class="line"><a name="l00452"></a><span class="lineno"> 452</span>&#160; <span class="comment">// Compute source and destination addresses</span></div><div class="line"><a name="l00453"></a><span class="lineno"> 453</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l00454"></a><span class="lineno"> 454</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l00455"></a><span class="lineno"> 455</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l00456"></a><span class="lineno"> 456</span>&#160;</div><div class="line"><a name="l00457"></a><span class="lineno"> 457</span>&#160; <span class="comment">// ------------------ Compute input/output addresses ---------------------------</span></div><div class="line"><a name="l00458"></a><span class="lineno"> 458</span>&#160;</div><div class="line"><a name="l00459"></a><span class="lineno"> 459</span>&#160; <span class="comment">// Compute the input address</span></div><div class="line"><a name="l00460"></a><span class="lineno"> 460</span>&#160; __global uchar *input_ptr = src_ptr + src_offset_first_element_in_bytes + x * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + y * (uint)K0 * src_stride_y + z * (uint)src_stride_z;</div><div class="line"><a name="l00461"></a><span class="lineno"> 461</span>&#160;</div><div class="line"><a name="l00462"></a><span class="lineno"> 462</span>&#160; <span class="comment">// Compute the output address</span></div><div class="line"><a name="l00463"></a><span class="lineno"> 463</span>&#160; __global uchar *output_ptr = dst_ptr + dst_offset_first_element_in_bytes + (y * (uint)BLOCK_SIZE * (uint)H0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + ((x % (uint)H0) * (uint)OUTPUT_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + ((</div><div class="line"><a name="l00464"></a><span class="lineno"> 464</span>&#160; x / (uint)H0)</div><div class="line"><a name="l00465"></a><span class="lineno"> 465</span>&#160; * (uint)dst_stride_y)</div><div class="line"><a name="l00466"></a><span class="lineno"> 466</span>&#160; + z * (uint)dst_stride_z;</div><div class="line"><a name="l00467"></a><span class="lineno"> 467</span>&#160;</div><div class="line"><a name="l00468"></a><span class="lineno"> 468</span>&#160; <span class="comment">// ---------------------------Load input values --------------------------------</span></div><div class="line"><a name="l00469"></a><span class="lineno"> 469</span>&#160;</div><div class="line"><a name="l00470"></a><span class="lineno"> 470</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(K0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0), a, 0); </div><div class="line"><a name="l00471"></a><span class="lineno"> 471</span>&#160;</div><div class="line"><a name="l00472"></a><span class="lineno"> 472</span>&#160; <span class="comment">// Load values from the RHS matrix</span></div><div class="line"><a name="l00473"></a><span class="lineno"> 473</span>&#160; a0 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 0 * src_stride_y));</div><div class="line"><a name="l00474"></a><span class="lineno"> 474</span>&#160;<span class="preprocessor">#if K0 &gt; 1</span></div><div class="line"><a name="l00475"></a><span class="lineno"> 475</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 1 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00476"></a><span class="lineno"> 476</span>&#160; {</div><div class="line"><a name="l00477"></a><span class="lineno"> 477</span>&#160; a1 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 1 * src_stride_y));</div><div class="line"><a name="l00478"></a><span class="lineno"> 478</span>&#160; }</div><div class="line"><a name="l00479"></a><span class="lineno"> 479</span>&#160;<span class="preprocessor">#endif // K0 &gt; 1</span></div><div class="line"><a name="l00480"></a><span class="lineno"> 480</span>&#160;<span class="preprocessor">#if K0 &gt; 2</span></div><div class="line"><a name="l00481"></a><span class="lineno"> 481</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 2 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00482"></a><span class="lineno"> 482</span>&#160; {</div><div class="line"><a name="l00483"></a><span class="lineno"> 483</span>&#160; a2 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 2 * src_stride_y));</div><div class="line"><a name="l00484"></a><span class="lineno"> 484</span>&#160; }</div><div class="line"><a name="l00485"></a><span class="lineno"> 485</span>&#160;<span class="preprocessor">#endif // K0 &gt; 2</span></div><div class="line"><a name="l00486"></a><span class="lineno"> 486</span>&#160;<span class="preprocessor">#if K0 &gt; 3</span></div><div class="line"><a name="l00487"></a><span class="lineno"> 487</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 3 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00488"></a><span class="lineno"> 488</span>&#160; {</div><div class="line"><a name="l00489"></a><span class="lineno"> 489</span>&#160; a3 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 3 * src_stride_y));</div><div class="line"><a name="l00490"></a><span class="lineno"> 490</span>&#160; }</div><div class="line"><a name="l00491"></a><span class="lineno"> 491</span>&#160;<span class="preprocessor">#endif // K0 &gt; 3</span></div><div class="line"><a name="l00492"></a><span class="lineno"> 492</span>&#160;<span class="preprocessor">#if K0 &gt; 4</span></div><div class="line"><a name="l00493"></a><span class="lineno"> 493</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 4 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00494"></a><span class="lineno"> 494</span>&#160; {</div><div class="line"><a name="l00495"></a><span class="lineno"> 495</span>&#160; a4 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 4 * src_stride_y));</div><div class="line"><a name="l00496"></a><span class="lineno"> 496</span>&#160; }</div><div class="line"><a name="l00497"></a><span class="lineno"> 497</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 5 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00498"></a><span class="lineno"> 498</span>&#160; {</div><div class="line"><a name="l00499"></a><span class="lineno"> 499</span>&#160; a5 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 5 * src_stride_y));</div><div class="line"><a name="l00500"></a><span class="lineno"> 500</span>&#160; }</div><div class="line"><a name="l00501"></a><span class="lineno"> 501</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 6 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00502"></a><span class="lineno"> 502</span>&#160; {</div><div class="line"><a name="l00503"></a><span class="lineno"> 503</span>&#160; a6 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 6 * src_stride_y));</div><div class="line"><a name="l00504"></a><span class="lineno"> 504</span>&#160; }</div><div class="line"><a name="l00505"></a><span class="lineno"> 505</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 7 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00506"></a><span class="lineno"> 506</span>&#160; {</div><div class="line"><a name="l00507"></a><span class="lineno"> 507</span>&#160; a7 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 7 * src_stride_y));</div><div class="line"><a name="l00508"></a><span class="lineno"> 508</span>&#160; }</div><div class="line"><a name="l00509"></a><span class="lineno"> 509</span>&#160;<span class="preprocessor">#endif // K0 &gt; 4</span></div><div class="line"><a name="l00510"></a><span class="lineno"> 510</span>&#160;<span class="preprocessor">#if K0 &gt; 8</span></div><div class="line"><a name="l00511"></a><span class="lineno"> 511</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 8 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00512"></a><span class="lineno"> 512</span>&#160; {</div><div class="line"><a name="l00513"></a><span class="lineno"> 513</span>&#160; a8 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 8 * src_stride_y));</div><div class="line"><a name="l00514"></a><span class="lineno"> 514</span>&#160; }</div><div class="line"><a name="l00515"></a><span class="lineno"> 515</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 9 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00516"></a><span class="lineno"> 516</span>&#160; {</div><div class="line"><a name="l00517"></a><span class="lineno"> 517</span>&#160; a9 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 9 * src_stride_y));</div><div class="line"><a name="l00518"></a><span class="lineno"> 518</span>&#160; }</div><div class="line"><a name="l00519"></a><span class="lineno"> 519</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 10 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00520"></a><span class="lineno"> 520</span>&#160; {</div><div class="line"><a name="l00521"></a><span class="lineno"> 521</span>&#160; aA = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 10 * src_stride_y));</div><div class="line"><a name="l00522"></a><span class="lineno"> 522</span>&#160; }</div><div class="line"><a name="l00523"></a><span class="lineno"> 523</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 11 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00524"></a><span class="lineno"> 524</span>&#160; {</div><div class="line"><a name="l00525"></a><span class="lineno"> 525</span>&#160; aB = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 11 * src_stride_y));</div><div class="line"><a name="l00526"></a><span class="lineno"> 526</span>&#160; }</div><div class="line"><a name="l00527"></a><span class="lineno"> 527</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 12 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00528"></a><span class="lineno"> 528</span>&#160; {</div><div class="line"><a name="l00529"></a><span class="lineno"> 529</span>&#160; aC = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 12 * src_stride_y));</div><div class="line"><a name="l00530"></a><span class="lineno"> 530</span>&#160; }</div><div class="line"><a name="l00531"></a><span class="lineno"> 531</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 13 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00532"></a><span class="lineno"> 532</span>&#160; {</div><div class="line"><a name="l00533"></a><span class="lineno"> 533</span>&#160; aD = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 13 * src_stride_y));</div><div class="line"><a name="l00534"></a><span class="lineno"> 534</span>&#160; }</div><div class="line"><a name="l00535"></a><span class="lineno"> 535</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 14 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00536"></a><span class="lineno"> 536</span>&#160; {</div><div class="line"><a name="l00537"></a><span class="lineno"> 537</span>&#160; aE = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 14 * src_stride_y));</div><div class="line"><a name="l00538"></a><span class="lineno"> 538</span>&#160; }</div><div class="line"><a name="l00539"></a><span class="lineno"> 539</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 15 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00540"></a><span class="lineno"> 540</span>&#160; {</div><div class="line"><a name="l00541"></a><span class="lineno"> 541</span>&#160; aF = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 15 * src_stride_y));</div><div class="line"><a name="l00542"></a><span class="lineno"> 542</span>&#160; }</div><div class="line"><a name="l00543"></a><span class="lineno"> 543</span>&#160;<span class="preprocessor">#endif // K0 &gt; 8</span></div><div class="line"><a name="l00544"></a><span class="lineno"> 544</span>&#160;</div><div class="line"><a name="l00545"></a><span class="lineno"> 545</span>&#160; <span class="comment">// ---------------------------Store output values ------------------------------</span></div><div class="line"><a name="l00546"></a><span class="lineno"> 546</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zout, 0);</div><div class="line"><a name="l00547"></a><span class="lineno"> 547</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(K0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, output_ptr, OUTPUT_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zout);</div><div class="line"><a name="l00548"></a><span class="lineno"> 548</span>&#160;</div><div class="line"><a name="l00549"></a><span class="lineno"> 549</span>&#160;<span class="preprocessor">#undef BLOCK_SIZE</span></div><div class="line"><a name="l00550"></a><span class="lineno"> 550</span>&#160;<span class="preprocessor">#undef OUTPUT_OFFSET_X</span></div><div class="line"><a name="l00551"></a><span class="lineno"> 551</span>&#160;<span class="preprocessor">#undef OUTPUT_STEP_X</span></div><div class="line"><a name="l00552"></a><span class="lineno"> 552</span>&#160;}</div><div class="line"><a name="l00553"></a><span class="lineno"> 553</span>&#160;</div><div class="line"><a name="l00554"></a><span class="lineno"> 554</span>&#160;<span class="preprocessor">#if defined(TRANSPOSE)</span></div><div class="line"><a name="l00555"></a><span class="lineno"> 555</span>&#160;</div><div class="line"><a name="l00586"></a><span class="lineno"> 586</span>&#160;__kernel <span class="keywordtype">void</span> gemm_reshape_rhs_matrix_t(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>),</div><div class="line"><a name="l00587"></a><span class="lineno"> 587</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>))</div><div class="line"><a name="l00588"></a><span class="lineno"> 588</span>&#160;{</div><div class="line"><a name="l00589"></a><span class="lineno"> 589</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l00590"></a><span class="lineno"> 590</span>&#160;<span class="preprocessor">#define BLOCK_SIZE ((K0) * (N0))</span></div><div class="line"><a name="l00591"></a><span class="lineno"> 591</span>&#160;</div><div class="line"><a name="l00592"></a><span class="lineno"> 592</span>&#160; <span class="comment">// Output offset X</span></div><div class="line"><a name="l00593"></a><span class="lineno"> 593</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00594"></a><span class="lineno"> 594</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (K0)</span></div><div class="line"><a name="l00595"></a><span class="lineno"> 595</span>&#160;<span class="preprocessor">#else // defined(INTERLEAVE)</span></div><div class="line"><a name="l00596"></a><span class="lineno"> 596</span>&#160;<span class="preprocessor">#define OUTPUT_OFFSET_X (BLOCK_SIZE)</span></div><div class="line"><a name="l00597"></a><span class="lineno"> 597</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00598"></a><span class="lineno"> 598</span>&#160;</div><div class="line"><a name="l00599"></a><span class="lineno"> 599</span>&#160; <span class="comment">// Output step X</span></div><div class="line"><a name="l00600"></a><span class="lineno"> 600</span>&#160;<span class="preprocessor">#if defined(INTERLEAVE)</span></div><div class="line"><a name="l00601"></a><span class="lineno"> 601</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (K0) * (H0)</span></div><div class="line"><a name="l00602"></a><span class="lineno"> 602</span>&#160;<span class="preprocessor">#else // Do not interleave</span></div><div class="line"><a name="l00603"></a><span class="lineno"> 603</span>&#160;<span class="preprocessor">#define OUTPUT_STEP_X (K0)</span></div><div class="line"><a name="l00604"></a><span class="lineno"> 604</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l00605"></a><span class="lineno"> 605</span>&#160;</div><div class="line"><a name="l00606"></a><span class="lineno"> 606</span>&#160; <span class="comment">// Compute source and destination addresses</span></div><div class="line"><a name="l00607"></a><span class="lineno"> 607</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l00608"></a><span class="lineno"> 608</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l00609"></a><span class="lineno"> 609</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l00610"></a><span class="lineno"> 610</span>&#160;</div><div class="line"><a name="l00611"></a><span class="lineno"> 611</span>&#160; <span class="comment">// ------------------ Compute input/output addresses ---------------------------</span></div><div class="line"><a name="l00612"></a><span class="lineno"> 612</span>&#160;</div><div class="line"><a name="l00613"></a><span class="lineno"> 613</span>&#160; <span class="comment">// Compute the input address</span></div><div class="line"><a name="l00614"></a><span class="lineno"> 614</span>&#160; __global uchar *input_ptr = src_ptr + src_offset_first_element_in_bytes + x * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + y * (uint)K0 * src_stride_y + z * (uint)src_stride_z;</div><div class="line"><a name="l00615"></a><span class="lineno"> 615</span>&#160;</div><div class="line"><a name="l00616"></a><span class="lineno"> 616</span>&#160; <span class="comment">// Compute the output address</span></div><div class="line"><a name="l00617"></a><span class="lineno"> 617</span>&#160; __global uchar *output_ptr = dst_ptr + dst_offset_first_element_in_bytes + (y * (uint)BLOCK_SIZE * (uint)H0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + ((x % H0) * (uint)OUTPUT_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + ((x /</div><div class="line"><a name="l00618"></a><span class="lineno"> 618</span>&#160; (uint)H0) * (uint)dst_stride_y) + z * (uint)dst_stride_z;</div><div class="line"><a name="l00619"></a><span class="lineno"> 619</span>&#160;</div><div class="line"><a name="l00620"></a><span class="lineno"> 620</span>&#160; <span class="comment">// ---------------------------Load input values --------------------------------</span></div><div class="line"><a name="l00621"></a><span class="lineno"> 621</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(K0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0), a, 0); <span class="comment">//VEC_DATA_TYPE(DATA_TYPE, N0) a0=0, a1=0, ... a(K0-1)=0;</span></div><div class="line"><a name="l00622"></a><span class="lineno"> 622</span>&#160;</div><div class="line"><a name="l00623"></a><span class="lineno"> 623</span>&#160; <span class="comment">// Load values from the RHS matrix</span></div><div class="line"><a name="l00624"></a><span class="lineno"> 624</span>&#160; a0 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 0 * src_stride_y));</div><div class="line"><a name="l00625"></a><span class="lineno"> 625</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 1 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00626"></a><span class="lineno"> 626</span>&#160; {</div><div class="line"><a name="l00627"></a><span class="lineno"> 627</span>&#160; a1 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 1 * src_stride_y));</div><div class="line"><a name="l00628"></a><span class="lineno"> 628</span>&#160; }</div><div class="line"><a name="l00629"></a><span class="lineno"> 629</span>&#160;<span class="preprocessor">#if K0 &gt; 2</span></div><div class="line"><a name="l00630"></a><span class="lineno"> 630</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 2 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00631"></a><span class="lineno"> 631</span>&#160; {</div><div class="line"><a name="l00632"></a><span class="lineno"> 632</span>&#160; a2 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 2 * src_stride_y));</div><div class="line"><a name="l00633"></a><span class="lineno"> 633</span>&#160; }</div><div class="line"><a name="l00634"></a><span class="lineno"> 634</span>&#160;<span class="preprocessor">#endif // K0 &gt; 2</span></div><div class="line"><a name="l00635"></a><span class="lineno"> 635</span>&#160;<span class="preprocessor">#if K0 &gt; 3</span></div><div class="line"><a name="l00636"></a><span class="lineno"> 636</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 3 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00637"></a><span class="lineno"> 637</span>&#160; {</div><div class="line"><a name="l00638"></a><span class="lineno"> 638</span>&#160; a3 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 3 * src_stride_y));</div><div class="line"><a name="l00639"></a><span class="lineno"> 639</span>&#160; }</div><div class="line"><a name="l00640"></a><span class="lineno"> 640</span>&#160;<span class="preprocessor">#endif // K0 &gt; 3</span></div><div class="line"><a name="l00641"></a><span class="lineno"> 641</span>&#160;<span class="preprocessor">#if K0 &gt; 4</span></div><div class="line"><a name="l00642"></a><span class="lineno"> 642</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 4 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00643"></a><span class="lineno"> 643</span>&#160; {</div><div class="line"><a name="l00644"></a><span class="lineno"> 644</span>&#160; a4 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 4 * src_stride_y));</div><div class="line"><a name="l00645"></a><span class="lineno"> 645</span>&#160; }</div><div class="line"><a name="l00646"></a><span class="lineno"> 646</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 5 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00647"></a><span class="lineno"> 647</span>&#160; {</div><div class="line"><a name="l00648"></a><span class="lineno"> 648</span>&#160; a5 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 5 * src_stride_y));</div><div class="line"><a name="l00649"></a><span class="lineno"> 649</span>&#160; }</div><div class="line"><a name="l00650"></a><span class="lineno"> 650</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 6 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00651"></a><span class="lineno"> 651</span>&#160; {</div><div class="line"><a name="l00652"></a><span class="lineno"> 652</span>&#160; a6 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 6 * src_stride_y));</div><div class="line"><a name="l00653"></a><span class="lineno"> 653</span>&#160; }</div><div class="line"><a name="l00654"></a><span class="lineno"> 654</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 7 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00655"></a><span class="lineno"> 655</span>&#160; {</div><div class="line"><a name="l00656"></a><span class="lineno"> 656</span>&#160; a7 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 7 * src_stride_y));</div><div class="line"><a name="l00657"></a><span class="lineno"> 657</span>&#160; }</div><div class="line"><a name="l00658"></a><span class="lineno"> 658</span>&#160;<span class="preprocessor">#endif // K0 &gt; 4</span></div><div class="line"><a name="l00659"></a><span class="lineno"> 659</span>&#160;<span class="preprocessor">#if K0 &gt; 8</span></div><div class="line"><a name="l00660"></a><span class="lineno"> 660</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 8 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00661"></a><span class="lineno"> 661</span>&#160; {</div><div class="line"><a name="l00662"></a><span class="lineno"> 662</span>&#160; a8 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 8 * src_stride_y));</div><div class="line"><a name="l00663"></a><span class="lineno"> 663</span>&#160; }</div><div class="line"><a name="l00664"></a><span class="lineno"> 664</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 9 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00665"></a><span class="lineno"> 665</span>&#160; {</div><div class="line"><a name="l00666"></a><span class="lineno"> 666</span>&#160; a9 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 9 * src_stride_y));</div><div class="line"><a name="l00667"></a><span class="lineno"> 667</span>&#160; }</div><div class="line"><a name="l00668"></a><span class="lineno"> 668</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 10 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00669"></a><span class="lineno"> 669</span>&#160; {</div><div class="line"><a name="l00670"></a><span class="lineno"> 670</span>&#160; aA = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 10 * src_stride_y));</div><div class="line"><a name="l00671"></a><span class="lineno"> 671</span>&#160; }</div><div class="line"><a name="l00672"></a><span class="lineno"> 672</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 11 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00673"></a><span class="lineno"> 673</span>&#160; {</div><div class="line"><a name="l00674"></a><span class="lineno"> 674</span>&#160; aB = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 11 * src_stride_y));</div><div class="line"><a name="l00675"></a><span class="lineno"> 675</span>&#160; }</div><div class="line"><a name="l00676"></a><span class="lineno"> 676</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 12 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00677"></a><span class="lineno"> 677</span>&#160; {</div><div class="line"><a name="l00678"></a><span class="lineno"> 678</span>&#160; aC = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 12 * src_stride_y));</div><div class="line"><a name="l00679"></a><span class="lineno"> 679</span>&#160; }</div><div class="line"><a name="l00680"></a><span class="lineno"> 680</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 13 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00681"></a><span class="lineno"> 681</span>&#160; {</div><div class="line"><a name="l00682"></a><span class="lineno"> 682</span>&#160; aD = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 13 * src_stride_y));</div><div class="line"><a name="l00683"></a><span class="lineno"> 683</span>&#160; }</div><div class="line"><a name="l00684"></a><span class="lineno"> 684</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 14 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00685"></a><span class="lineno"> 685</span>&#160; {</div><div class="line"><a name="l00686"></a><span class="lineno"> 686</span>&#160; aE = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 14 * src_stride_y));</div><div class="line"><a name="l00687"></a><span class="lineno"> 687</span>&#160; }</div><div class="line"><a name="l00688"></a><span class="lineno"> 688</span>&#160; <span class="keywordflow">if</span>(y * (uint)K0 + 15 &lt; SRC_HEIGHT)</div><div class="line"><a name="l00689"></a><span class="lineno"> 689</span>&#160; {</div><div class="line"><a name="l00690"></a><span class="lineno"> 690</span>&#160; aF = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(input_ptr + 15 * src_stride_y));</div><div class="line"><a name="l00691"></a><span class="lineno"> 691</span>&#160; }</div><div class="line"><a name="l00692"></a><span class="lineno"> 692</span>&#160;<span class="preprocessor">#endif // K0 &gt; 8</span></div><div class="line"><a name="l00693"></a><span class="lineno"> 693</span>&#160;</div><div class="line"><a name="l00694"></a><span class="lineno"> 694</span>&#160; <span class="comment">// ---------------------------Transpose the block ------------------------------</span></div><div class="line"><a name="l00695"></a><span class="lineno"> 695</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(N0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0), res, 0); <span class="comment">//VEC_DATA_TYPE(DATA_TYPE, K0) res0=0, res1=0, res2=0,... res(N0-1)=0;</span></div><div class="line"><a name="l00696"></a><span class="lineno"> 696</span>&#160;</div><div class="line"><a name="l00697"></a><span class="lineno"> 697</span>&#160;<span class="preprocessor">#if K0 == 2</span></div><div class="line"><a name="l00698"></a><span class="lineno"> 698</span>&#160; <span class="comment">// This part computes the following transpositions:</span></div><div class="line"><a name="l00699"></a><span class="lineno"> 699</span>&#160; <span class="comment">// 2x2 -&gt; 2x2</span></div><div class="line"><a name="l00700"></a><span class="lineno"> 700</span>&#160; <span class="comment">// 2x4 -&gt; 4x2</span></div><div class="line"><a name="l00701"></a><span class="lineno"> 701</span>&#160; <span class="comment">// 2x8 -&gt; 8x2</span></div><div class="line"><a name="l00702"></a><span class="lineno"> 702</span>&#160; <span class="comment">// 2x16 -&gt; 16x2</span></div><div class="line"><a name="l00703"></a><span class="lineno"> 703</span>&#160; res0 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s0, a1.s0);</div><div class="line"><a name="l00704"></a><span class="lineno"> 704</span>&#160; res1 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s1, a1.s1);</div><div class="line"><a name="l00705"></a><span class="lineno"> 705</span>&#160;<span class="preprocessor">#if N0 &gt; 2</span></div><div class="line"><a name="l00706"></a><span class="lineno"> 706</span>&#160; res2 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s2, a1.s2);</div><div class="line"><a name="l00707"></a><span class="lineno"> 707</span>&#160;<span class="preprocessor">#endif // N0 &gt; 2</span></div><div class="line"><a name="l00708"></a><span class="lineno"> 708</span>&#160;<span class="preprocessor">#if N0 &gt; 3</span></div><div class="line"><a name="l00709"></a><span class="lineno"> 709</span>&#160; res3 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s3, a1.s3);</div><div class="line"><a name="l00710"></a><span class="lineno"> 710</span>&#160;<span class="preprocessor">#endif // N0 &gt; 3</span></div><div class="line"><a name="l00711"></a><span class="lineno"> 711</span>&#160;<span class="preprocessor">#if N0 &gt; 4</span></div><div class="line"><a name="l00712"></a><span class="lineno"> 712</span>&#160; res4 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s4, a1.s4);</div><div class="line"><a name="l00713"></a><span class="lineno"> 713</span>&#160; res5 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s5, a1.s5);</div><div class="line"><a name="l00714"></a><span class="lineno"> 714</span>&#160; res6 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s6, a1.s6);</div><div class="line"><a name="l00715"></a><span class="lineno"> 715</span>&#160; res7 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s7, a1.s7);</div><div class="line"><a name="l00716"></a><span class="lineno"> 716</span>&#160;<span class="preprocessor">#endif // N0 &gt; 4</span></div><div class="line"><a name="l00717"></a><span class="lineno"> 717</span>&#160;<span class="preprocessor">#if N0 &gt; 8</span></div><div class="line"><a name="l00718"></a><span class="lineno"> 718</span>&#160; res8 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s8, a1.s8);</div><div class="line"><a name="l00719"></a><span class="lineno"> 719</span>&#160; res9 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s9, a1.s9);</div><div class="line"><a name="l00720"></a><span class="lineno"> 720</span>&#160; resA = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sA, a1.sA);</div><div class="line"><a name="l00721"></a><span class="lineno"> 721</span>&#160; resB = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sB, a1.sB);</div><div class="line"><a name="l00722"></a><span class="lineno"> 722</span>&#160; resC = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sC, a1.sC);</div><div class="line"><a name="l00723"></a><span class="lineno"> 723</span>&#160; resD = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sD, a1.sD);</div><div class="line"><a name="l00724"></a><span class="lineno"> 724</span>&#160; resE = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sE, a1.sE);</div><div class="line"><a name="l00725"></a><span class="lineno"> 725</span>&#160; resF = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sF, a1.sF);</div><div class="line"><a name="l00726"></a><span class="lineno"> 726</span>&#160;<span class="preprocessor">#endif // N0 &gt; 8</span></div><div class="line"><a name="l00727"></a><span class="lineno"> 727</span>&#160;</div><div class="line"><a name="l00728"></a><span class="lineno"> 728</span>&#160;<span class="preprocessor">#elif K0 == 3 // K0 == 2</span></div><div class="line"><a name="l00729"></a><span class="lineno"> 729</span>&#160; <span class="comment">// This part computes the following transpositions:</span></div><div class="line"><a name="l00730"></a><span class="lineno"> 730</span>&#160; <span class="comment">// 3x2 -&gt; 2x3</span></div><div class="line"><a name="l00731"></a><span class="lineno"> 731</span>&#160; <span class="comment">// 3x4 -&gt; 4x3</span></div><div class="line"><a name="l00732"></a><span class="lineno"> 732</span>&#160; <span class="comment">// 3x8 -&gt; 8x3</span></div><div class="line"><a name="l00733"></a><span class="lineno"> 733</span>&#160; <span class="comment">// 3x16 -&gt; 16x3</span></div><div class="line"><a name="l00734"></a><span class="lineno"> 734</span>&#160; res0 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s0, a1.s0, a2.s0);</div><div class="line"><a name="l00735"></a><span class="lineno"> 735</span>&#160; res1 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s1, a1.s1, a2.s1);</div><div class="line"><a name="l00736"></a><span class="lineno"> 736</span>&#160;<span class="preprocessor">#if N0 &gt; 2</span></div><div class="line"><a name="l00737"></a><span class="lineno"> 737</span>&#160; res2 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s2, a1.s2, a2.s2);</div><div class="line"><a name="l00738"></a><span class="lineno"> 738</span>&#160;<span class="preprocessor">#endif // N0 &gt; 2</span></div><div class="line"><a name="l00739"></a><span class="lineno"> 739</span>&#160;<span class="preprocessor">#if N0 &gt; 3</span></div><div class="line"><a name="l00740"></a><span class="lineno"> 740</span>&#160; res3 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s3, a1.s3, a2.s3);</div><div class="line"><a name="l00741"></a><span class="lineno"> 741</span>&#160;<span class="preprocessor">#endif // N0 &gt; 3</span></div><div class="line"><a name="l00742"></a><span class="lineno"> 742</span>&#160;<span class="preprocessor">#if N0 &gt; 4</span></div><div class="line"><a name="l00743"></a><span class="lineno"> 743</span>&#160; res4 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s4, a1.s4, a2.s4);</div><div class="line"><a name="l00744"></a><span class="lineno"> 744</span>&#160; res5 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s5, a1.s5, a2.s5);</div><div class="line"><a name="l00745"></a><span class="lineno"> 745</span>&#160; res6 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s6, a1.s6, a2.s6);</div><div class="line"><a name="l00746"></a><span class="lineno"> 746</span>&#160; res7 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s7, a1.s7, a2.s7);</div><div class="line"><a name="l00747"></a><span class="lineno"> 747</span>&#160;<span class="preprocessor">#endif // N0 &gt; 4</span></div><div class="line"><a name="l00748"></a><span class="lineno"> 748</span>&#160;<span class="preprocessor">#if N0 &gt; 8</span></div><div class="line"><a name="l00749"></a><span class="lineno"> 749</span>&#160; res8 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s8, a1.s8, a2.s8);</div><div class="line"><a name="l00750"></a><span class="lineno"> 750</span>&#160; res9 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s9, a1.s9, a2.s9);</div><div class="line"><a name="l00751"></a><span class="lineno"> 751</span>&#160; resA = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sA, a1.sA, a2.sA);</div><div class="line"><a name="l00752"></a><span class="lineno"> 752</span>&#160; resB = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sB, a1.sB, a2.sB);</div><div class="line"><a name="l00753"></a><span class="lineno"> 753</span>&#160; resC = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sC, a1.sC, a2.sC);</div><div class="line"><a name="l00754"></a><span class="lineno"> 754</span>&#160; resD = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sD, a1.sD, a2.sD);</div><div class="line"><a name="l00755"></a><span class="lineno"> 755</span>&#160; resE = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sE, a1.sE, a2.sE);</div><div class="line"><a name="l00756"></a><span class="lineno"> 756</span>&#160; resF = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sF, a1.sF, a2.sF);</div><div class="line"><a name="l00757"></a><span class="lineno"> 757</span>&#160;<span class="preprocessor">#endif // N0 &gt; 8</span></div><div class="line"><a name="l00758"></a><span class="lineno"> 758</span>&#160;</div><div class="line"><a name="l00759"></a><span class="lineno"> 759</span>&#160;<span class="preprocessor">#elif K0 == 4 // K0 == 4</span></div><div class="line"><a name="l00760"></a><span class="lineno"> 760</span>&#160; <span class="comment">// This part computes the following transpositions:</span></div><div class="line"><a name="l00761"></a><span class="lineno"> 761</span>&#160; <span class="comment">// 4x2 -&gt; 2x4</span></div><div class="line"><a name="l00762"></a><span class="lineno"> 762</span>&#160; <span class="comment">// 4x4 -&gt; 4x4</span></div><div class="line"><a name="l00763"></a><span class="lineno"> 763</span>&#160; <span class="comment">// 4x8 -&gt; 8x4</span></div><div class="line"><a name="l00764"></a><span class="lineno"> 764</span>&#160; <span class="comment">// 4x16 -&gt; 16x4</span></div><div class="line"><a name="l00765"></a><span class="lineno"> 765</span>&#160; res0 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s0, a1.s0, a2.s0, a3.s0);</div><div class="line"><a name="l00766"></a><span class="lineno"> 766</span>&#160; res1 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s1, a1.s1, a2.s1, a3.s1);</div><div class="line"><a name="l00767"></a><span class="lineno"> 767</span>&#160;<span class="preprocessor">#if N0 &gt; 2</span></div><div class="line"><a name="l00768"></a><span class="lineno"> 768</span>&#160; res2 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s2, a1.s2, a2.s2, a3.s2);</div><div class="line"><a name="l00769"></a><span class="lineno"> 769</span>&#160;<span class="preprocessor">#endif // N0 &gt; 2</span></div><div class="line"><a name="l00770"></a><span class="lineno"> 770</span>&#160;<span class="preprocessor">#if N0 &gt; 3</span></div><div class="line"><a name="l00771"></a><span class="lineno"> 771</span>&#160; res3 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s3, a1.s3, a2.s3, a3.s3);</div><div class="line"><a name="l00772"></a><span class="lineno"> 772</span>&#160;<span class="preprocessor">#endif // N0 &gt; 3</span></div><div class="line"><a name="l00773"></a><span class="lineno"> 773</span>&#160;<span class="preprocessor">#if N0 &gt; 4</span></div><div class="line"><a name="l00774"></a><span class="lineno"> 774</span>&#160; res4 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s4, a1.s4, a2.s4, a3.s4);</div><div class="line"><a name="l00775"></a><span class="lineno"> 775</span>&#160; res5 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s5, a1.s5, a2.s5, a3.s5);</div><div class="line"><a name="l00776"></a><span class="lineno"> 776</span>&#160; res6 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s6, a1.s6, a2.s6, a3.s6);</div><div class="line"><a name="l00777"></a><span class="lineno"> 777</span>&#160; res7 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s7, a1.s7, a2.s7, a3.s7);</div><div class="line"><a name="l00778"></a><span class="lineno"> 778</span>&#160;<span class="preprocessor">#endif // N0 &gt; 4</span></div><div class="line"><a name="l00779"></a><span class="lineno"> 779</span>&#160;<span class="preprocessor">#if N0 &gt; 8</span></div><div class="line"><a name="l00780"></a><span class="lineno"> 780</span>&#160; res8 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s8, a1.s8, a2.s8, a3.s8);</div><div class="line"><a name="l00781"></a><span class="lineno"> 781</span>&#160; res9 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s9, a1.s9, a2.s9, a3.s9);</div><div class="line"><a name="l00782"></a><span class="lineno"> 782</span>&#160; resA = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sA, a1.sA, a2.sA, a3.sA);</div><div class="line"><a name="l00783"></a><span class="lineno"> 783</span>&#160; resB = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sB, a1.sB, a2.sB, a3.sB);</div><div class="line"><a name="l00784"></a><span class="lineno"> 784</span>&#160; resC = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sC, a1.sC, a2.sC, a3.sC);</div><div class="line"><a name="l00785"></a><span class="lineno"> 785</span>&#160; resD = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sD, a1.sD, a2.sD, a3.sD);</div><div class="line"><a name="l00786"></a><span class="lineno"> 786</span>&#160; resE = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sE, a1.sE, a2.sE, a3.sE);</div><div class="line"><a name="l00787"></a><span class="lineno"> 787</span>&#160; resF = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sF, a1.sF, a2.sF, a3.sF);</div><div class="line"><a name="l00788"></a><span class="lineno"> 788</span>&#160;<span class="preprocessor">#endif // N0 &gt; 8</span></div><div class="line"><a name="l00789"></a><span class="lineno"> 789</span>&#160;</div><div class="line"><a name="l00790"></a><span class="lineno"> 790</span>&#160;<span class="preprocessor">#elif K0 == 8 // K0 == 8</span></div><div class="line"><a name="l00791"></a><span class="lineno"> 791</span>&#160; <span class="comment">// This part computes the following transpositions:</span></div><div class="line"><a name="l00792"></a><span class="lineno"> 792</span>&#160; <span class="comment">// 8x2 -&gt; 2x8</span></div><div class="line"><a name="l00793"></a><span class="lineno"> 793</span>&#160; <span class="comment">// 8x4 -&gt; 4x8</span></div><div class="line"><a name="l00794"></a><span class="lineno"> 794</span>&#160; <span class="comment">// 8x8 -&gt; 8x8</span></div><div class="line"><a name="l00795"></a><span class="lineno"> 795</span>&#160; <span class="comment">// 8x16 -&gt; 16x8</span></div><div class="line"><a name="l00796"></a><span class="lineno"> 796</span>&#160; res0 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s0, a1.s0, a2.s0, a3.s0, a4.s0, a5.s0, a6.s0, a7.s0);</div><div class="line"><a name="l00797"></a><span class="lineno"> 797</span>&#160; res1 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s1, a1.s1, a2.s1, a3.s1, a4.s1, a5.s1, a6.s1, a7.s1);</div><div class="line"><a name="l00798"></a><span class="lineno"> 798</span>&#160;<span class="preprocessor">#if N0 &gt; 2</span></div><div class="line"><a name="l00799"></a><span class="lineno"> 799</span>&#160; res2 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s2, a1.s2, a2.s2, a3.s2, a4.s2, a5.s2, a6.s2, a7.s2);</div><div class="line"><a name="l00800"></a><span class="lineno"> 800</span>&#160;<span class="preprocessor">#endif // N0 &gt; 2</span></div><div class="line"><a name="l00801"></a><span class="lineno"> 801</span>&#160;<span class="preprocessor">#if N0 &gt; 3</span></div><div class="line"><a name="l00802"></a><span class="lineno"> 802</span>&#160; res3 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s3, a1.s3, a2.s3, a3.s3, a4.s3, a5.s3, a6.s3, a7.s3);</div><div class="line"><a name="l00803"></a><span class="lineno"> 803</span>&#160;<span class="preprocessor">#endif // N0 &gt; 3</span></div><div class="line"><a name="l00804"></a><span class="lineno"> 804</span>&#160;<span class="preprocessor">#if N0 &gt; 4</span></div><div class="line"><a name="l00805"></a><span class="lineno"> 805</span>&#160; res4 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s4, a1.s4, a2.s4, a3.s4, a4.s4, a5.s4, a6.s4, a7.s4);</div><div class="line"><a name="l00806"></a><span class="lineno"> 806</span>&#160; res5 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s5, a1.s5, a2.s5, a3.s5, a4.s5, a5.s5, a6.s5, a7.s5);</div><div class="line"><a name="l00807"></a><span class="lineno"> 807</span>&#160; res6 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s6, a1.s6, a2.s6, a3.s6, a4.s6, a5.s6, a6.s6, a7.s6);</div><div class="line"><a name="l00808"></a><span class="lineno"> 808</span>&#160; res7 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s7, a1.s7, a2.s7, a3.s7, a4.s7, a5.s7, a6.s7, a7.s7);</div><div class="line"><a name="l00809"></a><span class="lineno"> 809</span>&#160;<span class="preprocessor">#endif // N0 &gt; 4</span></div><div class="line"><a name="l00810"></a><span class="lineno"> 810</span>&#160;<span class="preprocessor">#if N0 &gt; 8</span></div><div class="line"><a name="l00811"></a><span class="lineno"> 811</span>&#160; res8 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s8, a1.s8, a2.s8, a3.s8, a4.s8, a5.s8, a6.s8, a7.s8);</div><div class="line"><a name="l00812"></a><span class="lineno"> 812</span>&#160; res9 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s9, a1.s9, a2.s9, a3.s9, a4.s9, a5.s9, a6.s9, a7.s9);</div><div class="line"><a name="l00813"></a><span class="lineno"> 813</span>&#160; resA = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sA, a1.sA, a2.sA, a3.sA, a4.sA, a5.sA, a6.sA, a7.sA);</div><div class="line"><a name="l00814"></a><span class="lineno"> 814</span>&#160; resB = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sB, a1.sB, a2.sB, a3.sB, a4.sB, a5.sB, a6.sB, a7.sB);</div><div class="line"><a name="l00815"></a><span class="lineno"> 815</span>&#160; resC = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sC, a1.sC, a2.sC, a3.sC, a4.sC, a5.sC, a6.sC, a7.sC);</div><div class="line"><a name="l00816"></a><span class="lineno"> 816</span>&#160; resD = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sD, a1.sD, a2.sD, a3.sD, a4.sD, a5.sD, a6.sD, a7.sD);</div><div class="line"><a name="l00817"></a><span class="lineno"> 817</span>&#160; resE = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sE, a1.sE, a2.sE, a3.sE, a4.sE, a5.sE, a6.sE, a7.sE);</div><div class="line"><a name="l00818"></a><span class="lineno"> 818</span>&#160; resF = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sF, a1.sF, a2.sF, a3.sF, a4.sF, a5.sF, a6.sF, a7.sF);</div><div class="line"><a name="l00819"></a><span class="lineno"> 819</span>&#160;<span class="preprocessor">#endif // N0 &gt; 8</span></div><div class="line"><a name="l00820"></a><span class="lineno"> 820</span>&#160;</div><div class="line"><a name="l00821"></a><span class="lineno"> 821</span>&#160;<span class="preprocessor">#elif K0 == 16 // K0 == 16</span></div><div class="line"><a name="l00822"></a><span class="lineno"> 822</span>&#160;</div><div class="line"><a name="l00823"></a><span class="lineno"> 823</span>&#160; <span class="comment">// This part computes the following transpositions:</span></div><div class="line"><a name="l00824"></a><span class="lineno"> 824</span>&#160; <span class="comment">// 16x2 -&gt; 2x16</span></div><div class="line"><a name="l00825"></a><span class="lineno"> 825</span>&#160; <span class="comment">// 16x4 -&gt; 4x16</span></div><div class="line"><a name="l00826"></a><span class="lineno"> 826</span>&#160; <span class="comment">// 16x8 -&gt; 8x16</span></div><div class="line"><a name="l00827"></a><span class="lineno"> 827</span>&#160; <span class="comment">// 16x16 -&gt; 16x16</span></div><div class="line"><a name="l00828"></a><span class="lineno"> 828</span>&#160; res0 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s0, a1.s0, a2.s0, a3.s0, a4.s0, a5.s0, a6.s0, a7.s0,</div><div class="line"><a name="l00829"></a><span class="lineno"> 829</span>&#160; a8.s0, a9.s0, aA.s0, aB.s0, aC.s0, aD.s0, aE.s0, aF.s0);</div><div class="line"><a name="l00830"></a><span class="lineno"> 830</span>&#160; res1 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s1, a1.s1, a2.s1, a3.s1, a4.s1, a5.s1, a6.s1, a7.s1,</div><div class="line"><a name="l00831"></a><span class="lineno"> 831</span>&#160; a8.s1, a9.s1, aA.s1, aB.s1, aC.s1, aD.s1, aE.s1, aF.s1);</div><div class="line"><a name="l00832"></a><span class="lineno"> 832</span>&#160;<span class="preprocessor">#if N0 &gt; 2</span></div><div class="line"><a name="l00833"></a><span class="lineno"> 833</span>&#160; res2 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s2, a1.s2, a2.s2, a3.s2, a4.s2, a5.s2, a6.s2, a7.s2,</div><div class="line"><a name="l00834"></a><span class="lineno"> 834</span>&#160; a8.s2, a9.s2, aA.s2, aB.s2, aC.s2, aD.s2, aE.s2, aF.s2);</div><div class="line"><a name="l00835"></a><span class="lineno"> 835</span>&#160;<span class="preprocessor">#endif // N0 &gt; 2</span></div><div class="line"><a name="l00836"></a><span class="lineno"> 836</span>&#160;<span class="preprocessor">#if N0 &gt; 3</span></div><div class="line"><a name="l00837"></a><span class="lineno"> 837</span>&#160; res3 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s3, a1.s3, a2.s3, a3.s3, a4.s3, a5.s3, a6.s3, a7.s3,</div><div class="line"><a name="l00838"></a><span class="lineno"> 838</span>&#160; a8.s3, a9.s3, aA.s3, aB.s3, aC.s3, aD.s3, aE.s3, aF.s3);</div><div class="line"><a name="l00839"></a><span class="lineno"> 839</span>&#160;<span class="preprocessor">#endif // N0 &gt; 3</span></div><div class="line"><a name="l00840"></a><span class="lineno"> 840</span>&#160;<span class="preprocessor">#if N0 &gt; 4</span></div><div class="line"><a name="l00841"></a><span class="lineno"> 841</span>&#160; res4 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s4, a1.s4, a2.s4, a3.s4, a4.s4, a5.s4, a6.s4, a7.s4,</div><div class="line"><a name="l00842"></a><span class="lineno"> 842</span>&#160; a8.s4, a9.s4, aA.s4, aB.s4, aC.s4, aD.s4, aE.s4, aF.s4);</div><div class="line"><a name="l00843"></a><span class="lineno"> 843</span>&#160; res5 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s5, a1.s5, a2.s5, a3.s5, a4.s5, a5.s5, a6.s5, a7.s5,</div><div class="line"><a name="l00844"></a><span class="lineno"> 844</span>&#160; a8.s5, a9.s5, aA.s5, aB.s5, aC.s5, aD.s5, aE.s5, aF.s5);</div><div class="line"><a name="l00845"></a><span class="lineno"> 845</span>&#160; res6 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s6, a1.s6, a2.s6, a3.s6, a4.s6, a5.s6, a6.s6, a7.s6,</div><div class="line"><a name="l00846"></a><span class="lineno"> 846</span>&#160; a8.s6, a9.s6, aA.s6, aB.s6, aC.s6, aD.s6, aE.s6, aF.s6);</div><div class="line"><a name="l00847"></a><span class="lineno"> 847</span>&#160; res7 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s7, a1.s7, a2.s7, a3.s7, a4.s7, a5.s7, a6.s7, a7.s7,</div><div class="line"><a name="l00848"></a><span class="lineno"> 848</span>&#160; a8.s7, a9.s7, aA.s7, aB.s7, aC.s7, aD.s7, aE.s7, aF.s7);</div><div class="line"><a name="l00849"></a><span class="lineno"> 849</span>&#160;<span class="preprocessor">#endif // N0 &gt; 4</span></div><div class="line"><a name="l00850"></a><span class="lineno"> 850</span>&#160;<span class="preprocessor">#if N0 &gt; 8</span></div><div class="line"><a name="l00851"></a><span class="lineno"> 851</span>&#160; res8 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s8, a1.s8, a2.s8, a3.s8, a4.s8, a5.s8, a6.s8, a7.s8,</div><div class="line"><a name="l00852"></a><span class="lineno"> 852</span>&#160; a8.s8, a9.s8, aA.s8, aB.s8, aC.s8, aD.s8, aE.s8, aF.s8);</div><div class="line"><a name="l00853"></a><span class="lineno"> 853</span>&#160; res9 = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.s9, a1.s9, a2.s9, a3.s9, a4.s9, a5.s9, a6.s9, a7.s9,</div><div class="line"><a name="l00854"></a><span class="lineno"> 854</span>&#160; a8.s9, a9.s9, aA.s9, aB.s9, aC.s9, aD.s9, aE.s9, aF.s9);</div><div class="line"><a name="l00855"></a><span class="lineno"> 855</span>&#160; resA = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sA, a1.sA, a2.sA, a3.sA, a4.sA, a5.sA, a6.sA, a7.sA,</div><div class="line"><a name="l00856"></a><span class="lineno"> 856</span>&#160; a8.sA, a9.sA, aA.sA, aB.sA, aC.sA, aD.sA, aE.sA, aF.sA);</div><div class="line"><a name="l00857"></a><span class="lineno"> 857</span>&#160; resB = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sB, a1.sB, a2.sB, a3.sB, a4.sB, a5.sB, a6.sB, a7.sB,</div><div class="line"><a name="l00858"></a><span class="lineno"> 858</span>&#160; a8.sB, a9.sB, aA.sB, aB.sB, aC.sB, aD.sB, aE.sB, aF.sB);</div><div class="line"><a name="l00859"></a><span class="lineno"> 859</span>&#160; resC = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sC, a1.sC, a2.sC, a3.sC, a4.sC, a5.sC, a6.sC, a7.sC,</div><div class="line"><a name="l00860"></a><span class="lineno"> 860</span>&#160; a8.sC, a9.sC, aA.sC, aB.sC, aC.sC, aD.sC, aE.sC, aF.sC);</div><div class="line"><a name="l00861"></a><span class="lineno"> 861</span>&#160; resD = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sD, a1.sD, a2.sD, a3.sD, a4.sD, a5.sD, a6.sD, a7.sD,</div><div class="line"><a name="l00862"></a><span class="lineno"> 862</span>&#160; a8.sD, a9.sD, aA.sD, aB.sD, aC.sD, aD.sD, aE.sD, aF.sD);</div><div class="line"><a name="l00863"></a><span class="lineno"> 863</span>&#160; resE = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sE, a1.sE, a2.sE, a3.sE, a4.sE, a5.sE, a6.sE, a7.sE,</div><div class="line"><a name="l00864"></a><span class="lineno"> 864</span>&#160; a8.sE, a9.sE, aA.sE, aB.sE, aC.sE, aD.sE, aE.sE, aF.sE);</div><div class="line"><a name="l00865"></a><span class="lineno"> 865</span>&#160; resF = (<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, K0))(a0.sF, a1.sF, a2.sF, a3.sF, a4.sF, a5.sF, a6.sF, a7.sF,</div><div class="line"><a name="l00866"></a><span class="lineno"> 866</span>&#160; a8.sF, a9.sF, aA.sF, aB.sF, aC.sF, aD.sF, aE.sF, aF.sF);</div><div class="line"><a name="l00867"></a><span class="lineno"> 867</span>&#160;<span class="preprocessor">#endif // N0 &gt; 8</span></div><div class="line"><a name="l00868"></a><span class="lineno"> 868</span>&#160;</div><div class="line"><a name="l00869"></a><span class="lineno"> 869</span>&#160;<span class="preprocessor">#else // N0 == 16</span></div><div class="line"><a name="l00870"></a><span class="lineno"> 870</span>&#160;<span class="preprocessor">#error &quot;Not supported N0 value&quot;</span></div><div class="line"><a name="l00871"></a><span class="lineno"> 871</span>&#160;<span class="preprocessor">#endif // N0 &gt; 2</span></div><div class="line"><a name="l00872"></a><span class="lineno"> 872</span>&#160;</div><div class="line"><a name="l00873"></a><span class="lineno"> 873</span>&#160; <span class="comment">// ---------------------------Store the output values ------------------------------</span></div><div class="line"><a name="l00874"></a><span class="lineno"> 874</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zout, 0);</div><div class="line"><a name="l00875"></a><span class="lineno"> 875</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(N0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, res, output_ptr, OUTPUT_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zout);</div><div class="line"><a name="l00876"></a><span class="lineno"> 876</span>&#160;</div><div class="line"><a name="l00877"></a><span class="lineno"> 877</span>&#160;<span class="preprocessor">#undef BLOCK_SIZE</span></div><div class="line"><a name="l00878"></a><span class="lineno"> 878</span>&#160;<span class="preprocessor">#undef OUTPUT_OFFSET_X</span></div><div class="line"><a name="l00879"></a><span class="lineno"> 879</span>&#160;<span class="preprocessor">#undef OUTPUT_STEP_X</span></div><div class="line"><a name="l00880"></a><span class="lineno"> 880</span>&#160;}</div><div class="line"><a name="l00881"></a><span class="lineno"> 881</span>&#160;<span class="preprocessor">#endif // defined(TRANSPOSE)</span></div><div class="line"><a name="l00882"></a><span class="lineno"> 882</span>&#160;<span class="preprocessor">#endif // defined(K0) &amp;&amp; defined(N0) &amp;&amp; defined(H0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(SRC_HEIGHT)</span></div><div class="line"><a name="l00883"></a><span class="lineno"> 883</span>&#160;</div><div class="line"><a name="l00884"></a><span class="lineno"> 884</span>&#160;<span class="preprocessor">#if defined(M0) &amp;&amp; defined(N0) &amp;&amp; defined(K0) &amp;&amp; defined(H0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(M) &amp;&amp; defined(N) &amp;&amp; defined(K)</span></div><div class="line"><a name="l00885"></a><span class="lineno"> 885</span>&#160;</div><div class="line"><a name="l00886"></a><span class="lineno"> 886</span>&#160;<span class="preprocessor">#define CONCAT(a, b) a##b</span></div><div class="line"><a name="l00887"></a><span class="lineno"> 887</span>&#160;</div><div class="line"><a name="l00888"></a><span class="lineno"> 888</span>&#160;<span class="preprocessor">#define ARM_DOT1(a, b, c) \</span></div><div class="line"><a name="l00889"></a><span class="lineno"> 889</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00890"></a><span class="lineno"> 890</span>&#160;<span class="preprocessor"> c = fma(a, b, c); \</span></div><div class="line"><a name="l00891"></a><span class="lineno"> 891</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00892"></a><span class="lineno"> 892</span>&#160;<span class="preprocessor">#define ARM_DOT2(a, b, c) \</span></div><div class="line"><a name="l00893"></a><span class="lineno"> 893</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00894"></a><span class="lineno"> 894</span>&#160;<span class="preprocessor"> c = fma(a.s0, b.s0, c); \</span></div><div class="line"><a name="l00895"></a><span class="lineno"> 895</span>&#160;<span class="preprocessor"> c = fma(a.s1, b.s1, c); \</span></div><div class="line"><a name="l00896"></a><span class="lineno"> 896</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00897"></a><span class="lineno"> 897</span>&#160;<span class="preprocessor">#define ARM_DOT3(a, b, c) \</span></div><div class="line"><a name="l00898"></a><span class="lineno"> 898</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00899"></a><span class="lineno"> 899</span>&#160;<span class="preprocessor"> ARM_DOT2(a, b, c); \</span></div><div class="line"><a name="l00900"></a><span class="lineno"> 900</span>&#160;<span class="preprocessor"> c = fma((a.s2), (b.s2), c); \</span></div><div class="line"><a name="l00901"></a><span class="lineno"> 901</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00902"></a><span class="lineno"> 902</span>&#160;<span class="preprocessor">#define ARM_DOT4(a, b, c) \</span></div><div class="line"><a name="l00903"></a><span class="lineno"> 903</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00904"></a><span class="lineno"> 904</span>&#160;<span class="preprocessor"> ARM_DOT3(a, b, c); \</span></div><div class="line"><a name="l00905"></a><span class="lineno"> 905</span>&#160;<span class="preprocessor"> c = fma((a.s3), (b.s3), c); \</span></div><div class="line"><a name="l00906"></a><span class="lineno"> 906</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00907"></a><span class="lineno"> 907</span>&#160;<span class="preprocessor">#define ARM_DOT8(a, b, c) \</span></div><div class="line"><a name="l00908"></a><span class="lineno"> 908</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00909"></a><span class="lineno"> 909</span>&#160;<span class="preprocessor"> ARM_DOT4((a.lo), (b.lo), c); \</span></div><div class="line"><a name="l00910"></a><span class="lineno"> 910</span>&#160;<span class="preprocessor"> ARM_DOT4((a.hi), (b.hi), c); \</span></div><div class="line"><a name="l00911"></a><span class="lineno"> 911</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00912"></a><span class="lineno"> 912</span>&#160;<span class="preprocessor">#define ARM_DOT16(a, b, c) \</span></div><div class="line"><a name="l00913"></a><span class="lineno"> 913</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00914"></a><span class="lineno"> 914</span>&#160;<span class="preprocessor"> ARM_DOT8((a.lo), (b.lo), c); \</span></div><div class="line"><a name="l00915"></a><span class="lineno"> 915</span>&#160;<span class="preprocessor"> ARM_DOT8((a.hi), (b.hi), c); \</span></div><div class="line"><a name="l00916"></a><span class="lineno"> 916</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00917"></a><span class="lineno"> 917</span>&#160;</div><div class="line"><a name="l00918"></a><span class="lineno"> 918</span>&#160;<span class="preprocessor">#if N0 == 2</span></div><div class="line"><a name="l00919"></a><span class="lineno"> 919</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(k0, a, b, c) \</span></div><div class="line"><a name="l00920"></a><span class="lineno"> 920</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00921"></a><span class="lineno"> 921</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00922"></a><span class="lineno"> 922</span>&#160;<span class="preprocessor"> ((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l00923"></a><span class="lineno"> 923</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00924"></a><span class="lineno"> 924</span>&#160;<span class="preprocessor"> ((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l00925"></a><span class="lineno"> 925</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00926"></a><span class="lineno"> 926</span>&#160;<span class="preprocessor">#elif N0 == 3 // N0 == 3</span></div><div class="line"><a name="l00927"></a><span class="lineno"> 927</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(k0, a, b, c) \</span></div><div class="line"><a name="l00928"></a><span class="lineno"> 928</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00929"></a><span class="lineno"> 929</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00930"></a><span class="lineno"> 930</span>&#160;<span class="preprocessor"> ((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l00931"></a><span class="lineno"> 931</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00932"></a><span class="lineno"> 932</span>&#160;<span class="preprocessor"> ((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l00933"></a><span class="lineno"> 933</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00934"></a><span class="lineno"> 934</span>&#160;<span class="preprocessor"> ((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l00935"></a><span class="lineno"> 935</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00936"></a><span class="lineno"> 936</span>&#160;<span class="preprocessor">#elif N0 == 4 // N0 == 4</span></div><div class="line"><a name="l00937"></a><span class="lineno"> 937</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(k0, a, b, c) \</span></div><div class="line"><a name="l00938"></a><span class="lineno"> 938</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00939"></a><span class="lineno"> 939</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00940"></a><span class="lineno"> 940</span>&#160;<span class="preprocessor"> ((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l00941"></a><span class="lineno"> 941</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00942"></a><span class="lineno"> 942</span>&#160;<span class="preprocessor"> ((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l00943"></a><span class="lineno"> 943</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00944"></a><span class="lineno"> 944</span>&#160;<span class="preprocessor"> ((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l00945"></a><span class="lineno"> 945</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00946"></a><span class="lineno"> 946</span>&#160;<span class="preprocessor"> ((a), (b##3), (c.s3)); \</span></div><div class="line"><a name="l00947"></a><span class="lineno"> 947</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00948"></a><span class="lineno"> 948</span>&#160;<span class="preprocessor">#elif N0 == 8 // N0 == 8</span></div><div class="line"><a name="l00949"></a><span class="lineno"> 949</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(k0, a, b, c) \</span></div><div class="line"><a name="l00950"></a><span class="lineno"> 950</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00951"></a><span class="lineno"> 951</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00952"></a><span class="lineno"> 952</span>&#160;<span class="preprocessor"> ((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l00953"></a><span class="lineno"> 953</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00954"></a><span class="lineno"> 954</span>&#160;<span class="preprocessor"> ((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l00955"></a><span class="lineno"> 955</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00956"></a><span class="lineno"> 956</span>&#160;<span class="preprocessor"> ((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l00957"></a><span class="lineno"> 957</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00958"></a><span class="lineno"> 958</span>&#160;<span class="preprocessor"> ((a), (b##3), (c.s3)); \</span></div><div class="line"><a name="l00959"></a><span class="lineno"> 959</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00960"></a><span class="lineno"> 960</span>&#160;<span class="preprocessor"> ((a), (b##4), (c.s4)); \</span></div><div class="line"><a name="l00961"></a><span class="lineno"> 961</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00962"></a><span class="lineno"> 962</span>&#160;<span class="preprocessor"> ((a), (b##5), (c.s5)); \</span></div><div class="line"><a name="l00963"></a><span class="lineno"> 963</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00964"></a><span class="lineno"> 964</span>&#160;<span class="preprocessor"> ((a), (b##6), (c.s6)); \</span></div><div class="line"><a name="l00965"></a><span class="lineno"> 965</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00966"></a><span class="lineno"> 966</span>&#160;<span class="preprocessor"> ((a), (b##7), (c.s7)); \</span></div><div class="line"><a name="l00967"></a><span class="lineno"> 967</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l00968"></a><span class="lineno"> 968</span>&#160;<span class="preprocessor">#elif N0 == 16 // N0 == 16</span></div><div class="line"><a name="l00969"></a><span class="lineno"> 969</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(k0, a, b, c) \</span></div><div class="line"><a name="l00970"></a><span class="lineno"> 970</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l00971"></a><span class="lineno"> 971</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00972"></a><span class="lineno"> 972</span>&#160;<span class="preprocessor"> ((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l00973"></a><span class="lineno"> 973</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00974"></a><span class="lineno"> 974</span>&#160;<span class="preprocessor"> ((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l00975"></a><span class="lineno"> 975</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00976"></a><span class="lineno"> 976</span>&#160;<span class="preprocessor"> ((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l00977"></a><span class="lineno"> 977</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00978"></a><span class="lineno"> 978</span>&#160;<span class="preprocessor"> ((a), (b##3), (c.s3)); \</span></div><div class="line"><a name="l00979"></a><span class="lineno"> 979</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00980"></a><span class="lineno"> 980</span>&#160;<span class="preprocessor"> ((a), (b##4), (c.s4)); \</span></div><div class="line"><a name="l00981"></a><span class="lineno"> 981</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00982"></a><span class="lineno"> 982</span>&#160;<span class="preprocessor"> ((a), (b##5), (c.s5)); \</span></div><div class="line"><a name="l00983"></a><span class="lineno"> 983</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00984"></a><span class="lineno"> 984</span>&#160;<span class="preprocessor"> ((a), (b##6), (c.s6)); \</span></div><div class="line"><a name="l00985"></a><span class="lineno"> 985</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00986"></a><span class="lineno"> 986</span>&#160;<span class="preprocessor"> ((a), (b##7), (c.s7)); \</span></div><div class="line"><a name="l00987"></a><span class="lineno"> 987</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00988"></a><span class="lineno"> 988</span>&#160;<span class="preprocessor"> ((a), (b##8), (c.s8)); \</span></div><div class="line"><a name="l00989"></a><span class="lineno"> 989</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00990"></a><span class="lineno"> 990</span>&#160;<span class="preprocessor"> ((a), (b##9), (c.s9)); \</span></div><div class="line"><a name="l00991"></a><span class="lineno"> 991</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00992"></a><span class="lineno"> 992</span>&#160;<span class="preprocessor"> ((a), (b##A), (c.sA)); \</span></div><div class="line"><a name="l00993"></a><span class="lineno"> 993</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00994"></a><span class="lineno"> 994</span>&#160;<span class="preprocessor"> ((a), (b##B), (c.sB)); \</span></div><div class="line"><a name="l00995"></a><span class="lineno"> 995</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00996"></a><span class="lineno"> 996</span>&#160;<span class="preprocessor"> ((a), (b##C), (c.sC)); \</span></div><div class="line"><a name="l00997"></a><span class="lineno"> 997</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l00998"></a><span class="lineno"> 998</span>&#160;<span class="preprocessor"> ((a), (b##D), (c.sD)); \</span></div><div class="line"><a name="l00999"></a><span class="lineno"> 999</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l01000"></a><span class="lineno"> 1000</span>&#160;<span class="preprocessor"> ((a), (b##E), (c.sE)); \</span></div><div class="line"><a name="l01001"></a><span class="lineno"> 1001</span>&#160;<span class="preprocessor"> CONCAT(ARM_DOT, k0) \</span></div><div class="line"><a name="l01002"></a><span class="lineno"> 1002</span>&#160;<span class="preprocessor"> ((a), (b##F), (c.sF)); \</span></div><div class="line"><a name="l01003"></a><span class="lineno"> 1003</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01004"></a><span class="lineno"> 1004</span>&#160;<span class="preprocessor">#else // N0 not supported</span></div><div class="line"><a name="l01005"></a><span class="lineno"> 1005</span>&#160;<span class="preprocessor">#error &quot;N0 value not supported&quot;</span></div><div class="line"><a name="l01006"></a><span class="lineno"> 1006</span>&#160;<span class="preprocessor">#endif // N0 conditions</span></div><div class="line"><a name="l01007"></a><span class="lineno"> 1007</span>&#160;</div><div class="line"><a name="l01065"></a><span class="lineno"> 1065</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_reshaped_only_rhs_t(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(lhs),</div><div class="line"><a name="l01066"></a><span class="lineno"> 1066</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(rhs),</div><div class="line"><a name="l01067"></a><span class="lineno"> 1067</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l01068"></a><span class="lineno"> 1068</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>),</div><div class="line"><a name="l01069"></a><span class="lineno"> 1069</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l01070"></a><span class="lineno"> 1070</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l01071"></a><span class="lineno"> 1071</span>&#160; uint lhs_stride_z,</div><div class="line"><a name="l01072"></a><span class="lineno"> 1072</span>&#160; uint rhs_stride_z,</div><div class="line"><a name="l01073"></a><span class="lineno"> 1073</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l01074"></a><span class="lineno"> 1074</span>&#160; uint bias_stride_z,</div><div class="line"><a name="l01075"></a><span class="lineno"> 1075</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l01076"></a><span class="lineno"> 1076</span>&#160; uint dst_stride_z</div><div class="line"><a name="l01077"></a><span class="lineno"> 1077</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l01078"></a><span class="lineno"> 1078</span>&#160; ,</div><div class="line"><a name="l01079"></a><span class="lineno"> 1079</span>&#160; uint lhs_cross_plane_pad</div><div class="line"><a name="l01080"></a><span class="lineno"> 1080</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l01081"></a><span class="lineno"> 1081</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l01082"></a><span class="lineno"> 1082</span>&#160; ,</div><div class="line"><a name="l01083"></a><span class="lineno"> 1083</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l01084"></a><span class="lineno"> 1084</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l01085"></a><span class="lineno"> 1085</span>&#160; )</div><div class="line"><a name="l01086"></a><span class="lineno"> 1086</span>&#160;{</div><div class="line"><a name="l01087"></a><span class="lineno"> 1087</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l01088"></a><span class="lineno"> 1088</span>&#160;<span class="preprocessor">#define RHS_BLOCK_SIZE ((K0) * (N0))</span></div><div class="line"><a name="l01089"></a><span class="lineno"> 1089</span>&#160;</div><div class="line"><a name="l01090"></a><span class="lineno"> 1090</span>&#160; <span class="comment">// RHS offset and step X</span></div><div class="line"><a name="l01091"></a><span class="lineno"> 1091</span>&#160;<span class="preprocessor">#if defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01092"></a><span class="lineno"> 1092</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (K0)</span></div><div class="line"><a name="l01093"></a><span class="lineno"> 1093</span>&#160;<span class="preprocessor">#define RHS_STEP_X ((K0) * (H0))</span></div><div class="line"><a name="l01094"></a><span class="lineno"> 1094</span>&#160;<span class="preprocessor">#define RHS_STEP_LOOP (1)</span></div><div class="line"><a name="l01095"></a><span class="lineno"> 1095</span>&#160;<span class="preprocessor">#else // defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01096"></a><span class="lineno"> 1096</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (RHS_BLOCK_SIZE)</span></div><div class="line"><a name="l01097"></a><span class="lineno"> 1097</span>&#160;<span class="preprocessor">#define RHS_STEP_X (K0)</span></div><div class="line"><a name="l01098"></a><span class="lineno"> 1098</span>&#160;<span class="preprocessor">#define RHS_STEP_LOOP (H0)</span></div><div class="line"><a name="l01099"></a><span class="lineno"> 1099</span>&#160;<span class="preprocessor">#endif // defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01100"></a><span class="lineno"> 1100</span>&#160;</div><div class="line"><a name="l01101"></a><span class="lineno"> 1101</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l01102"></a><span class="lineno"> 1102</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l01103"></a><span class="lineno"> 1103</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l01104"></a><span class="lineno"> 1104</span>&#160;</div><div class="line"><a name="l01105"></a><span class="lineno"> 1105</span>&#160;<span class="preprocessor">#if defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l01106"></a><span class="lineno"> 1106</span>&#160; <span class="keywordflow">if</span>((x * N0 &gt;= N) || (y * M0 &gt;= M))</div><div class="line"><a name="l01107"></a><span class="lineno"> 1107</span>&#160; {</div><div class="line"><a name="l01108"></a><span class="lineno"> 1108</span>&#160; <span class="keywordflow">return</span>;</div><div class="line"><a name="l01109"></a><span class="lineno"> 1109</span>&#160; }</div><div class="line"><a name="l01110"></a><span class="lineno"> 1110</span>&#160;<span class="preprocessor">#endif // defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l01111"></a><span class="lineno"> 1111</span>&#160;</div><div class="line"><a name="l01112"></a><span class="lineno"> 1112</span>&#160; <span class="comment">// Compute LHS matrix address</span></div><div class="line"><a name="l01113"></a><span class="lineno"> 1113</span>&#160; uint lhs_offset = lhs_offset_first_element_in_bytes + y * M0 * (uint)lhs_stride_y;</div><div class="line"><a name="l01114"></a><span class="lineno"> 1114</span>&#160;</div><div class="line"><a name="l01115"></a><span class="lineno"> 1115</span>&#160; <span class="comment">// Compute RHS matrix address</span></div><div class="line"><a name="l01116"></a><span class="lineno"> 1116</span>&#160; uint rhs_offset = rhs_offset_first_element_in_bytes + (x % H0) * (uint)RHS_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + (x / (uint)H0) * rhs_stride_y;</div><div class="line"><a name="l01117"></a><span class="lineno"> 1117</span>&#160;</div><div class="line"><a name="l01118"></a><span class="lineno"> 1118</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01119"></a><span class="lineno"> 1119</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l01120"></a><span class="lineno"> 1120</span>&#160; rhs_offset += (z % MATRIX_B_DEPTH) * rhs_stride_z;</div><div class="line"><a name="l01121"></a><span class="lineno"> 1121</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01122"></a><span class="lineno"> 1122</span>&#160; rhs_offset += z * rhs_stride_z;</div><div class="line"><a name="l01123"></a><span class="lineno"> 1123</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01124"></a><span class="lineno"> 1124</span>&#160;</div><div class="line"><a name="l01125"></a><span class="lineno"> 1125</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(8, uint, zlhs, 0); <span class="comment">//uint zlhs0=0,zlhs1=0,zlhs2=0,... zlhs7=0;</span></div><div class="line"><a name="l01126"></a><span class="lineno"> 1126</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zero, 0);</div><div class="line"><a name="l01127"></a><span class="lineno"> 1127</span>&#160;</div><div class="line"><a name="l01128"></a><span class="lineno"> 1128</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l01129"></a><span class="lineno"> 1129</span>&#160; <span class="comment">// The plane (zlhs) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l01130"></a><span class="lineno"> 1130</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zlhs, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, lhs_cross_plane_pad, lhs_stride_y);</div><div class="line"><a name="l01131"></a><span class="lineno"> 1131</span>&#160;</div><div class="line"><a name="l01132"></a><span class="lineno"> 1132</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l01133"></a><span class="lineno"> 1133</span>&#160; <span class="comment">// multiply lhs_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l01134"></a><span class="lineno"> 1134</span>&#160; lhs_offset += z * lhs_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l01135"></a><span class="lineno"> 1135</span>&#160;</div><div class="line"><a name="l01136"></a><span class="lineno"> 1136</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l01137"></a><span class="lineno"> 1137</span>&#160;</div><div class="line"><a name="l01138"></a><span class="lineno"> 1138</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l01139"></a><span class="lineno"> 1139</span>&#160; lhs_offset += z * lhs_stride_z;</div><div class="line"><a name="l01140"></a><span class="lineno"> 1140</span>&#160;</div><div class="line"><a name="l01141"></a><span class="lineno"> 1141</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l01142"></a><span class="lineno"> 1142</span>&#160;</div><div class="line"><a name="l01143"></a><span class="lineno"> 1143</span>&#160; <span class="comment">// Initialize the accumulators</span></div><div class="line"><a name="l01144"></a><span class="lineno"> 1144</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0), c, 0); <span class="comment">//VEC_DATA_TYPE(DATA_TYPE, N0) c0=0,c1=0,c2=0,... c(M0-1)=0;</span></div><div class="line"><a name="l01145"></a><span class="lineno"> 1145</span>&#160;</div><div class="line"><a name="l01146"></a><span class="lineno"> 1146</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l01147"></a><span class="lineno"> 1147</span>&#160; <span class="keywordflow">for</span>(; i &lt;= (K - K0); i += K0)</div><div class="line"><a name="l01148"></a><span class="lineno"> 1148</span>&#160; {</div><div class="line"><a name="l01149"></a><span class="lineno"> 1149</span>&#160; <span class="comment">// Supported cases (M0, K0):</span></div><div class="line"><a name="l01150"></a><span class="lineno"> 1150</span>&#160; <span class="comment">// 1,2 - 1,3 - 1,4 - 1,8 - 1,16</span></div><div class="line"><a name="l01151"></a><span class="lineno"> 1151</span>&#160; <span class="comment">// 2,2 - 2,3 - 2,4 - 2,8 - 2,16</span></div><div class="line"><a name="l01152"></a><span class="lineno"> 1152</span>&#160; <span class="comment">// 3,2 - 3,3 - 3,4 - 3,8 - 3,16</span></div><div class="line"><a name="l01153"></a><span class="lineno"> 1153</span>&#160; <span class="comment">// 4,2 - 4,3 - 4,4 - 4,8 - 4,16</span></div><div class="line"><a name="l01154"></a><span class="lineno"> 1154</span>&#160; <span class="comment">// 5,2 - 5,3 - 5,4 - 5,8 - 5,16</span></div><div class="line"><a name="l01155"></a><span class="lineno"> 1155</span>&#160; <span class="comment">// 6,2 - 6,3 - 6,4 - 6,8 - 6,16</span></div><div class="line"><a name="l01156"></a><span class="lineno"> 1156</span>&#160; <span class="comment">// 7,2 - 7,3 - 7,4 - 7,8 - 7,16</span></div><div class="line"><a name="l01157"></a><span class="lineno"> 1157</span>&#160; <span class="comment">// 8,2 - 8,3 - 8,4 - 8,8 - 8,16</span></div><div class="line"><a name="l01158"></a><span class="lineno"> 1158</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l01159"></a><span class="lineno"> 1159</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, lhs_ptr, lhs_offset, lhs_stride_y, zlhs);</div><div class="line"><a name="l01160"></a><span class="lineno"> 1160</span>&#160;</div><div class="line"><a name="l01161"></a><span class="lineno"> 1161</span>&#160; <span class="comment">// Load values from RHS matrix</span></div><div class="line"><a name="l01162"></a><span class="lineno"> 1162</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(N0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, rhs_ptr, rhs_offset, RHS_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zero);</div><div class="line"><a name="l01163"></a><span class="lineno"> 1163</span>&#160;</div><div class="line"><a name="l01164"></a><span class="lineno"> 1164</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l01165"></a><span class="lineno"> 1165</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a0, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c0);</div><div class="line"><a name="l01166"></a><span class="lineno"> 1166</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l01167"></a><span class="lineno"> 1167</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a1, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c1);</div><div class="line"><a name="l01168"></a><span class="lineno"> 1168</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l01169"></a><span class="lineno"> 1169</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l01170"></a><span class="lineno"> 1170</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a2, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c2);</div><div class="line"><a name="l01171"></a><span class="lineno"> 1171</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l01172"></a><span class="lineno"> 1172</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l01173"></a><span class="lineno"> 1173</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a3, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c3);</div><div class="line"><a name="l01174"></a><span class="lineno"> 1174</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l01175"></a><span class="lineno"> 1175</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l01176"></a><span class="lineno"> 1176</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a4, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c4);</div><div class="line"><a name="l01177"></a><span class="lineno"> 1177</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l01178"></a><span class="lineno"> 1178</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l01179"></a><span class="lineno"> 1179</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a5, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c5);</div><div class="line"><a name="l01180"></a><span class="lineno"> 1180</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l01181"></a><span class="lineno"> 1181</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l01182"></a><span class="lineno"> 1182</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a6, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c6);</div><div class="line"><a name="l01183"></a><span class="lineno"> 1183</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l01184"></a><span class="lineno"> 1184</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l01185"></a><span class="lineno"> 1185</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(K0, a7, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c7);</div><div class="line"><a name="l01186"></a><span class="lineno"> 1186</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l01187"></a><span class="lineno"> 1187</span>&#160;</div><div class="line"><a name="l01188"></a><span class="lineno"> 1188</span>&#160; lhs_offset += K0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01189"></a><span class="lineno"> 1189</span>&#160; rhs_offset += (N0 * RHS_STEP_X * RHS_STEP_LOOP) * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01190"></a><span class="lineno"> 1190</span>&#160; }</div><div class="line"><a name="l01191"></a><span class="lineno"> 1191</span>&#160;</div><div class="line"><a name="l01192"></a><span class="lineno"> 1192</span>&#160; <span class="comment">// Left-over accumulations</span></div><div class="line"><a name="l01193"></a><span class="lineno"> 1193</span>&#160; <span class="keywordflow">for</span>(; i &lt; K; ++i)</div><div class="line"><a name="l01194"></a><span class="lineno"> 1194</span>&#160; {</div><div class="line"><a name="l01195"></a><span class="lineno"> 1195</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l01196"></a><span class="lineno"> 1196</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, 1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, lhs_ptr, lhs_offset, lhs_stride_y, zlhs);</div><div class="line"><a name="l01197"></a><span class="lineno"> 1197</span>&#160;</div><div class="line"><a name="l01198"></a><span class="lineno"> 1198</span>&#160; <span class="comment">// Load values from RHS matrix</span></div><div class="line"><a name="l01199"></a><span class="lineno"> 1199</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(N0, 1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, rhs_ptr, rhs_offset, RHS_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zero);</div><div class="line"><a name="l01200"></a><span class="lineno"> 1200</span>&#160;</div><div class="line"><a name="l01201"></a><span class="lineno"> 1201</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l01202"></a><span class="lineno"> 1202</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a0, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c0);</div><div class="line"><a name="l01203"></a><span class="lineno"> 1203</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l01204"></a><span class="lineno"> 1204</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a1, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c1);</div><div class="line"><a name="l01205"></a><span class="lineno"> 1205</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l01206"></a><span class="lineno"> 1206</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l01207"></a><span class="lineno"> 1207</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a2, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c2);</div><div class="line"><a name="l01208"></a><span class="lineno"> 1208</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l01209"></a><span class="lineno"> 1209</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l01210"></a><span class="lineno"> 1210</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a3, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c3);</div><div class="line"><a name="l01211"></a><span class="lineno"> 1211</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l01212"></a><span class="lineno"> 1212</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l01213"></a><span class="lineno"> 1213</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a4, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c4);</div><div class="line"><a name="l01214"></a><span class="lineno"> 1214</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l01215"></a><span class="lineno"> 1215</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l01216"></a><span class="lineno"> 1216</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a5, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c5);</div><div class="line"><a name="l01217"></a><span class="lineno"> 1217</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l01218"></a><span class="lineno"> 1218</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l01219"></a><span class="lineno"> 1219</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a6, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c6);</div><div class="line"><a name="l01220"></a><span class="lineno"> 1220</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l01221"></a><span class="lineno"> 1221</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l01222"></a><span class="lineno"> 1222</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(1, a7, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c7);</div><div class="line"><a name="l01223"></a><span class="lineno"> 1223</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l01224"></a><span class="lineno"> 1224</span>&#160;</div><div class="line"><a name="l01225"></a><span class="lineno"> 1225</span>&#160; lhs_offset += <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01226"></a><span class="lineno"> 1226</span>&#160; rhs_offset += <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01227"></a><span class="lineno"> 1227</span>&#160; }</div><div class="line"><a name="l01228"></a><span class="lineno"> 1228</span>&#160;</div><div class="line"><a name="l01229"></a><span class="lineno"> 1229</span>&#160; __global uchar *dst_addr = dst_ptr + dst_offset_first_element_in_bytes + (x * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (y * (uint)M0 * dst_stride_y);</div><div class="line"><a name="l01230"></a><span class="lineno"> 1230</span>&#160;</div><div class="line"><a name="l01231"></a><span class="lineno"> 1231</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(8, uint, zout, 0); <span class="comment">//uint zout0=0,zout1=0,zout2=0,... zout7=0;</span></div><div class="line"><a name="l01232"></a><span class="lineno"> 1232</span>&#160;</div><div class="line"><a name="l01233"></a><span class="lineno"> 1233</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01234"></a><span class="lineno"> 1234</span>&#160;</div><div class="line"><a name="l01235"></a><span class="lineno"> 1235</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l01236"></a><span class="lineno"> 1236</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zout, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, dst_cross_plane_pad, dst_stride_y);</div><div class="line"><a name="l01237"></a><span class="lineno"> 1237</span>&#160;</div><div class="line"><a name="l01238"></a><span class="lineno"> 1238</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l01239"></a><span class="lineno"> 1239</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l01240"></a><span class="lineno"> 1240</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l01241"></a><span class="lineno"> 1241</span>&#160;</div><div class="line"><a name="l01242"></a><span class="lineno"> 1242</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01243"></a><span class="lineno"> 1243</span>&#160;</div><div class="line"><a name="l01244"></a><span class="lineno"> 1244</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l01245"></a><span class="lineno"> 1245</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l01246"></a><span class="lineno"> 1246</span>&#160;</div><div class="line"><a name="l01247"></a><span class="lineno"> 1247</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01248"></a><span class="lineno"> 1248</span>&#160;</div><div class="line"><a name="l01249"></a><span class="lineno"> 1249</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l01250"></a><span class="lineno"> 1250</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l01251"></a><span class="lineno"> 1251</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, ALPHA);</div><div class="line"><a name="l01252"></a><span class="lineno"> 1252</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l01253"></a><span class="lineno"> 1253</span>&#160;</div><div class="line"><a name="l01254"></a><span class="lineno"> 1254</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l01255"></a><span class="lineno"> 1255</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l01256"></a><span class="lineno"> 1256</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01257"></a><span class="lineno"> 1257</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l01258"></a><span class="lineno"> 1258</span>&#160;</div><div class="line"><a name="l01259"></a><span class="lineno"> 1259</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l01260"></a><span class="lineno"> 1260</span>&#160;</div><div class="line"><a name="l01261"></a><span class="lineno"> 1261</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l01262"></a><span class="lineno"> 1262</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l01263"></a><span class="lineno"> 1263</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l01264"></a><span class="lineno"> 1264</span>&#160;</div><div class="line"><a name="l01265"></a><span class="lineno"> 1265</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l01266"></a><span class="lineno"> 1266</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(M0, c, bias0);</div><div class="line"><a name="l01267"></a><span class="lineno"> 1267</span>&#160;</div><div class="line"><a name="l01268"></a><span class="lineno"> 1268</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01269"></a><span class="lineno"> 1269</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (get_global_id(1) * (uint)M0 * bias_stride_y) + get_global_id(</div><div class="line"><a name="l01270"></a><span class="lineno"> 1270</span>&#160; 2) * bias_stride_z;</div><div class="line"><a name="l01271"></a><span class="lineno"> 1271</span>&#160;</div><div class="line"><a name="l01272"></a><span class="lineno"> 1272</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l01273"></a><span class="lineno"> 1273</span>&#160;</div><div class="line"><a name="l01274"></a><span class="lineno"> 1274</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l01275"></a><span class="lineno"> 1275</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l01276"></a><span class="lineno"> 1276</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l01277"></a><span class="lineno"> 1277</span>&#160;</div><div class="line"><a name="l01278"></a><span class="lineno"> 1278</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l01279"></a><span class="lineno"> 1279</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(M0, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l01280"></a><span class="lineno"> 1280</span>&#160;</div><div class="line"><a name="l01281"></a><span class="lineno"> 1281</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01282"></a><span class="lineno"> 1282</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l01283"></a><span class="lineno"> 1283</span>&#160;</div><div class="line"><a name="l01284"></a><span class="lineno"> 1284</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l01285"></a><span class="lineno"> 1285</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(M0, ACTIVATION_TYPE, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, A_VAL, B_VAL);</div><div class="line"><a name="l01286"></a><span class="lineno"> 1286</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l01287"></a><span class="lineno"> 1287</span>&#160;</div><div class="line"><a name="l01288"></a><span class="lineno"> 1288</span>&#160; <span class="comment">// Store output block</span></div><div class="line"><a name="l01289"></a><span class="lineno"> 1289</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, dst_addr, dst_stride_y, zout);</div><div class="line"><a name="l01290"></a><span class="lineno"> 1290</span>&#160;</div><div class="line"><a name="l01291"></a><span class="lineno"> 1291</span>&#160;<span class="preprocessor">#undef RHS_BLOCK_SIZE</span></div><div class="line"><a name="l01292"></a><span class="lineno"> 1292</span>&#160;<span class="preprocessor">#undef RHS_OFFSET_X</span></div><div class="line"><a name="l01293"></a><span class="lineno"> 1293</span>&#160;<span class="preprocessor">#undef RHS_STEP_X</span></div><div class="line"><a name="l01294"></a><span class="lineno"> 1294</span>&#160;}</div><div class="line"><a name="l01295"></a><span class="lineno"> 1295</span>&#160;</div><div class="line"><a name="l01296"></a><span class="lineno"> 1296</span>&#160;<span class="preprocessor">#define VFMA(a, b, c) \</span></div><div class="line"><a name="l01297"></a><span class="lineno"> 1297</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01298"></a><span class="lineno"> 1298</span>&#160;<span class="preprocessor"> c = fma(a, b, c); \</span></div><div class="line"><a name="l01299"></a><span class="lineno"> 1299</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01300"></a><span class="lineno"> 1300</span>&#160;</div><div class="line"><a name="l01301"></a><span class="lineno"> 1301</span>&#160;<span class="preprocessor">#if M0 == 1</span></div><div class="line"><a name="l01302"></a><span class="lineno"> 1302</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01303"></a><span class="lineno"> 1303</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01304"></a><span class="lineno"> 1304</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01305"></a><span class="lineno"> 1305</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01306"></a><span class="lineno"> 1306</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01307"></a><span class="lineno"> 1307</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01308"></a><span class="lineno"> 1308</span>&#160;<span class="preprocessor">#elif M0 == 2 // M0 == 2</span></div><div class="line"><a name="l01309"></a><span class="lineno"> 1309</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01310"></a><span class="lineno"> 1310</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01311"></a><span class="lineno"> 1311</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01312"></a><span class="lineno"> 1312</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01313"></a><span class="lineno"> 1313</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01314"></a><span class="lineno"> 1314</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01315"></a><span class="lineno"> 1315</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01316"></a><span class="lineno"> 1316</span>&#160;<span class="preprocessor">#elif M0 == 3 // M0 == 3</span></div><div class="line"><a name="l01317"></a><span class="lineno"> 1317</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01318"></a><span class="lineno"> 1318</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01319"></a><span class="lineno"> 1319</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01320"></a><span class="lineno"> 1320</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01321"></a><span class="lineno"> 1321</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01322"></a><span class="lineno"> 1322</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01323"></a><span class="lineno"> 1323</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l01324"></a><span class="lineno"> 1324</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01325"></a><span class="lineno"> 1325</span>&#160;<span class="preprocessor">#elif M0 == 4 // M0 == 4</span></div><div class="line"><a name="l01326"></a><span class="lineno"> 1326</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01327"></a><span class="lineno"> 1327</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01328"></a><span class="lineno"> 1328</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01329"></a><span class="lineno"> 1329</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01330"></a><span class="lineno"> 1330</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01331"></a><span class="lineno"> 1331</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01332"></a><span class="lineno"> 1332</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l01333"></a><span class="lineno"> 1333</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l01334"></a><span class="lineno"> 1334</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01335"></a><span class="lineno"> 1335</span>&#160;<span class="preprocessor">#elif M0 == 5 // M0 == 5</span></div><div class="line"><a name="l01336"></a><span class="lineno"> 1336</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01337"></a><span class="lineno"> 1337</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01338"></a><span class="lineno"> 1338</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01339"></a><span class="lineno"> 1339</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01340"></a><span class="lineno"> 1340</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01341"></a><span class="lineno"> 1341</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01342"></a><span class="lineno"> 1342</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l01343"></a><span class="lineno"> 1343</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l01344"></a><span class="lineno"> 1344</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l01345"></a><span class="lineno"> 1345</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01346"></a><span class="lineno"> 1346</span>&#160;<span class="preprocessor">#elif M0 == 6 // M0 == 6</span></div><div class="line"><a name="l01347"></a><span class="lineno"> 1347</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01348"></a><span class="lineno"> 1348</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01349"></a><span class="lineno"> 1349</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01350"></a><span class="lineno"> 1350</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01351"></a><span class="lineno"> 1351</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01352"></a><span class="lineno"> 1352</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01353"></a><span class="lineno"> 1353</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l01354"></a><span class="lineno"> 1354</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l01355"></a><span class="lineno"> 1355</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l01356"></a><span class="lineno"> 1356</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##5).s##i), b, (c##5)); \</span></div><div class="line"><a name="l01357"></a><span class="lineno"> 1357</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01358"></a><span class="lineno"> 1358</span>&#160;<span class="preprocessor">#elif M0 == 7 // M0 == 7</span></div><div class="line"><a name="l01359"></a><span class="lineno"> 1359</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01360"></a><span class="lineno"> 1360</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01361"></a><span class="lineno"> 1361</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01362"></a><span class="lineno"> 1362</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01363"></a><span class="lineno"> 1363</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01364"></a><span class="lineno"> 1364</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01365"></a><span class="lineno"> 1365</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l01366"></a><span class="lineno"> 1366</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l01367"></a><span class="lineno"> 1367</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l01368"></a><span class="lineno"> 1368</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##5).s##i), b, (c##5)); \</span></div><div class="line"><a name="l01369"></a><span class="lineno"> 1369</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##6).s##i), b, (c##6)); \</span></div><div class="line"><a name="l01370"></a><span class="lineno"> 1370</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01371"></a><span class="lineno"> 1371</span>&#160;<span class="preprocessor">#elif M0 == 8 // M0 == 8</span></div><div class="line"><a name="l01372"></a><span class="lineno"> 1372</span>&#160;<span class="preprocessor">#define LD_RHS_VFMA_M0xN0(i, a, c) \</span></div><div class="line"><a name="l01373"></a><span class="lineno"> 1373</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01374"></a><span class="lineno"> 1374</span>&#160;<span class="preprocessor"> VEC_DATA_TYPE(DATA_TYPE, N0) \</span></div><div class="line"><a name="l01375"></a><span class="lineno"> 1375</span>&#160;<span class="preprocessor"> b = VLOAD(N0)(0, (__global DATA_TYPE *)(rhs_ptr + rhs_offset + 0x##i * RHS_STEP_X * sizeof(DATA_TYPE))); \</span></div><div class="line"><a name="l01376"></a><span class="lineno"> 1376</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l01377"></a><span class="lineno"> 1377</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l01378"></a><span class="lineno"> 1378</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l01379"></a><span class="lineno"> 1379</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l01380"></a><span class="lineno"> 1380</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l01381"></a><span class="lineno"> 1381</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##5).s##i), b, (c##5)); \</span></div><div class="line"><a name="l01382"></a><span class="lineno"> 1382</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##6).s##i), b, (c##6)); \</span></div><div class="line"><a name="l01383"></a><span class="lineno"> 1383</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##7).s##i), b, (c##7)); \</span></div><div class="line"><a name="l01384"></a><span class="lineno"> 1384</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01385"></a><span class="lineno"> 1385</span>&#160;<span class="preprocessor">#else // M0 not supported</span></div><div class="line"><a name="l01386"></a><span class="lineno"> 1386</span>&#160;<span class="preprocessor">#error &quot;M0 not supported&quot;</span></div><div class="line"><a name="l01387"></a><span class="lineno"> 1387</span>&#160;<span class="preprocessor">#endif // M0 not supported</span></div><div class="line"><a name="l01388"></a><span class="lineno"> 1388</span>&#160;</div><div class="line"><a name="l01445"></a><span class="lineno"> 1445</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_reshaped_only_rhs_nt(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(lhs),</div><div class="line"><a name="l01446"></a><span class="lineno"> 1446</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(rhs),</div><div class="line"><a name="l01447"></a><span class="lineno"> 1447</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l01448"></a><span class="lineno"> 1448</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>),</div><div class="line"><a name="l01449"></a><span class="lineno"> 1449</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l01450"></a><span class="lineno"> 1450</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l01451"></a><span class="lineno"> 1451</span>&#160; uint lhs_stride_z,</div><div class="line"><a name="l01452"></a><span class="lineno"> 1452</span>&#160; uint rhs_stride_z,</div><div class="line"><a name="l01453"></a><span class="lineno"> 1453</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l01454"></a><span class="lineno"> 1454</span>&#160; uint bias_stride_z,</div><div class="line"><a name="l01455"></a><span class="lineno"> 1455</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l01456"></a><span class="lineno"> 1456</span>&#160; uint dst_stride_z</div><div class="line"><a name="l01457"></a><span class="lineno"> 1457</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l01458"></a><span class="lineno"> 1458</span>&#160; ,</div><div class="line"><a name="l01459"></a><span class="lineno"> 1459</span>&#160; uint lhs_cross_plane_pad</div><div class="line"><a name="l01460"></a><span class="lineno"> 1460</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l01461"></a><span class="lineno"> 1461</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l01462"></a><span class="lineno"> 1462</span>&#160; ,</div><div class="line"><a name="l01463"></a><span class="lineno"> 1463</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l01464"></a><span class="lineno"> 1464</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l01465"></a><span class="lineno"> 1465</span>&#160; )</div><div class="line"><a name="l01466"></a><span class="lineno"> 1466</span>&#160;{</div><div class="line"><a name="l01467"></a><span class="lineno"> 1467</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l01468"></a><span class="lineno"> 1468</span>&#160;<span class="preprocessor">#define RHS_BLOCK_SIZE ((K0) * (N0))</span></div><div class="line"><a name="l01469"></a><span class="lineno"> 1469</span>&#160;</div><div class="line"><a name="l01470"></a><span class="lineno"> 1470</span>&#160; <span class="comment">// RHS offset and step X</span></div><div class="line"><a name="l01471"></a><span class="lineno"> 1471</span>&#160;<span class="preprocessor">#if defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01472"></a><span class="lineno"> 1472</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (N0)</span></div><div class="line"><a name="l01473"></a><span class="lineno"> 1473</span>&#160;<span class="preprocessor">#define RHS_STEP_X ((N0) * (H0))</span></div><div class="line"><a name="l01474"></a><span class="lineno"> 1474</span>&#160;<span class="preprocessor">#define RHS_STEP_LOOP (1)</span></div><div class="line"><a name="l01475"></a><span class="lineno"> 1475</span>&#160;<span class="preprocessor">#else // defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01476"></a><span class="lineno"> 1476</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (RHS_BLOCK_SIZE)</span></div><div class="line"><a name="l01477"></a><span class="lineno"> 1477</span>&#160;<span class="preprocessor">#define RHS_STEP_X (N0)</span></div><div class="line"><a name="l01478"></a><span class="lineno"> 1478</span>&#160;<span class="preprocessor">#define RHS_STEP_LOOP (H0)</span></div><div class="line"><a name="l01479"></a><span class="lineno"> 1479</span>&#160;<span class="preprocessor">#endif // defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01480"></a><span class="lineno"> 1480</span>&#160;</div><div class="line"><a name="l01481"></a><span class="lineno"> 1481</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l01482"></a><span class="lineno"> 1482</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l01483"></a><span class="lineno"> 1483</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l01484"></a><span class="lineno"> 1484</span>&#160;</div><div class="line"><a name="l01485"></a><span class="lineno"> 1485</span>&#160;<span class="preprocessor">#if defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l01486"></a><span class="lineno"> 1486</span>&#160; <span class="keywordflow">if</span>((x * N0 &gt;= N) || (y * M0 &gt;= M))</div><div class="line"><a name="l01487"></a><span class="lineno"> 1487</span>&#160; {</div><div class="line"><a name="l01488"></a><span class="lineno"> 1488</span>&#160; <span class="keywordflow">return</span>;</div><div class="line"><a name="l01489"></a><span class="lineno"> 1489</span>&#160; }</div><div class="line"><a name="l01490"></a><span class="lineno"> 1490</span>&#160;<span class="preprocessor">#endif // defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l01491"></a><span class="lineno"> 1491</span>&#160;</div><div class="line"><a name="l01492"></a><span class="lineno"> 1492</span>&#160; <span class="comment">// Compute LHS matrix address</span></div><div class="line"><a name="l01493"></a><span class="lineno"> 1493</span>&#160; uint lhs_offset = lhs_offset_first_element_in_bytes + y * M0 * (uint)lhs_stride_y;</div><div class="line"><a name="l01494"></a><span class="lineno"> 1494</span>&#160;</div><div class="line"><a name="l01495"></a><span class="lineno"> 1495</span>&#160; <span class="comment">// Compute RHS matrix address</span></div><div class="line"><a name="l01496"></a><span class="lineno"> 1496</span>&#160; uint rhs_offset = rhs_offset_first_element_in_bytes + (x % H0) * (uint)RHS_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + (x / (uint)H0) * rhs_stride_y;</div><div class="line"><a name="l01497"></a><span class="lineno"> 1497</span>&#160;</div><div class="line"><a name="l01498"></a><span class="lineno"> 1498</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01499"></a><span class="lineno"> 1499</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l01500"></a><span class="lineno"> 1500</span>&#160; rhs_offset += (z % MATRIX_B_DEPTH) * rhs_stride_z;</div><div class="line"><a name="l01501"></a><span class="lineno"> 1501</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01502"></a><span class="lineno"> 1502</span>&#160; rhs_offset += z * rhs_stride_z;</div><div class="line"><a name="l01503"></a><span class="lineno"> 1503</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01504"></a><span class="lineno"> 1504</span>&#160;</div><div class="line"><a name="l01505"></a><span class="lineno"> 1505</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(8, uint, zin, 0); <span class="comment">//uint zin0=0,zin1=0,zin2=0,... zin7=0;</span></div><div class="line"><a name="l01506"></a><span class="lineno"> 1506</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zero, 0); <span class="comment">//uint zero0=0,zero1=0,zero2=0,... zero7=0;</span></div><div class="line"><a name="l01507"></a><span class="lineno"> 1507</span>&#160;</div><div class="line"><a name="l01508"></a><span class="lineno"> 1508</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l01509"></a><span class="lineno"> 1509</span>&#160;</div><div class="line"><a name="l01510"></a><span class="lineno"> 1510</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l01511"></a><span class="lineno"> 1511</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zin, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, lhs_cross_plane_pad, lhs_stride_y);</div><div class="line"><a name="l01512"></a><span class="lineno"> 1512</span>&#160;</div><div class="line"><a name="l01513"></a><span class="lineno"> 1513</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l01514"></a><span class="lineno"> 1514</span>&#160; <span class="comment">// multiply lhs_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l01515"></a><span class="lineno"> 1515</span>&#160; lhs_offset += z * lhs_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l01516"></a><span class="lineno"> 1516</span>&#160;</div><div class="line"><a name="l01517"></a><span class="lineno"> 1517</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l01518"></a><span class="lineno"> 1518</span>&#160;</div><div class="line"><a name="l01519"></a><span class="lineno"> 1519</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l01520"></a><span class="lineno"> 1520</span>&#160; lhs_offset += z * lhs_stride_z;</div><div class="line"><a name="l01521"></a><span class="lineno"> 1521</span>&#160;</div><div class="line"><a name="l01522"></a><span class="lineno"> 1522</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l01523"></a><span class="lineno"> 1523</span>&#160;</div><div class="line"><a name="l01524"></a><span class="lineno"> 1524</span>&#160; <span class="comment">// Initialize the accumulators</span></div><div class="line"><a name="l01525"></a><span class="lineno"> 1525</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0), c, 0); <span class="comment">//VEC_DATA_TYPE(DATA_TYPE, N0) c0=0,c1=0,c2=0,... c(N0-1)=0;</span></div><div class="line"><a name="l01526"></a><span class="lineno"> 1526</span>&#160;</div><div class="line"><a name="l01527"></a><span class="lineno"> 1527</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l01528"></a><span class="lineno"> 1528</span>&#160; <span class="keywordflow">for</span>(; i &lt;= (K - K0); i += K0)</div><div class="line"><a name="l01529"></a><span class="lineno"> 1529</span>&#160; {</div><div class="line"><a name="l01530"></a><span class="lineno"> 1530</span>&#160; <span class="comment">// Supported cases (M0, K0):</span></div><div class="line"><a name="l01531"></a><span class="lineno"> 1531</span>&#160; <span class="comment">// 1,2 - 1,3 - 1,4 - 1,8 - 1,16</span></div><div class="line"><a name="l01532"></a><span class="lineno"> 1532</span>&#160; <span class="comment">// 2,2 - 2,3 - 2,4 - 2,8 - 2,16</span></div><div class="line"><a name="l01533"></a><span class="lineno"> 1533</span>&#160; <span class="comment">// 3,2 - 3,3 - 3,4 - 3,8 - 3,16</span></div><div class="line"><a name="l01534"></a><span class="lineno"> 1534</span>&#160; <span class="comment">// 4,2 - 4,3 - 4,4 - 4,8 - 4,16</span></div><div class="line"><a name="l01535"></a><span class="lineno"> 1535</span>&#160; <span class="comment">// 5,2 - 5,3 - 5,4 - 5,8 - 5,16</span></div><div class="line"><a name="l01536"></a><span class="lineno"> 1536</span>&#160; <span class="comment">// 6,2 - 6,3 - 6,4 - 6,8 - 6,16</span></div><div class="line"><a name="l01537"></a><span class="lineno"> 1537</span>&#160; <span class="comment">// 7,2 - 7,3 - 7,4 - 7,8 - 7,16</span></div><div class="line"><a name="l01538"></a><span class="lineno"> 1538</span>&#160; <span class="comment">// 8,2 - 8,3 - 8,4 - 8,8 - 8,16</span></div><div class="line"><a name="l01539"></a><span class="lineno"> 1539</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l01540"></a><span class="lineno"> 1540</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, lhs_ptr, lhs_offset, lhs_stride_y, zin);</div><div class="line"><a name="l01541"></a><span class="lineno"> 1541</span>&#160;</div><div class="line"><a name="l01542"></a><span class="lineno"> 1542</span>&#160; LD_RHS_VFMA_M0xN0(0, a, c);</div><div class="line"><a name="l01543"></a><span class="lineno"> 1543</span>&#160; LD_RHS_VFMA_M0xN0(1, a, c);</div><div class="line"><a name="l01544"></a><span class="lineno"> 1544</span>&#160;<span class="preprocessor">#if K0 &gt; 2</span></div><div class="line"><a name="l01545"></a><span class="lineno"> 1545</span>&#160; LD_RHS_VFMA_M0xN0(2, a, c);</div><div class="line"><a name="l01546"></a><span class="lineno"> 1546</span>&#160;<span class="preprocessor">#endif // K0 &gt; 2</span></div><div class="line"><a name="l01547"></a><span class="lineno"> 1547</span>&#160;<span class="preprocessor">#if K0 &gt; 3</span></div><div class="line"><a name="l01548"></a><span class="lineno"> 1548</span>&#160; LD_RHS_VFMA_M0xN0(3, a, c);</div><div class="line"><a name="l01549"></a><span class="lineno"> 1549</span>&#160;<span class="preprocessor">#endif // K0 &gt; 3</span></div><div class="line"><a name="l01550"></a><span class="lineno"> 1550</span>&#160;<span class="preprocessor">#if K0 &gt; 4</span></div><div class="line"><a name="l01551"></a><span class="lineno"> 1551</span>&#160; LD_RHS_VFMA_M0xN0(4, a, c);</div><div class="line"><a name="l01552"></a><span class="lineno"> 1552</span>&#160; LD_RHS_VFMA_M0xN0(5, a, c);</div><div class="line"><a name="l01553"></a><span class="lineno"> 1553</span>&#160; LD_RHS_VFMA_M0xN0(6, a, c);</div><div class="line"><a name="l01554"></a><span class="lineno"> 1554</span>&#160; LD_RHS_VFMA_M0xN0(7, a, c);</div><div class="line"><a name="l01555"></a><span class="lineno"> 1555</span>&#160;<span class="preprocessor">#endif // K0 &gt; 4</span></div><div class="line"><a name="l01556"></a><span class="lineno"> 1556</span>&#160;<span class="preprocessor">#if K0 &gt; 8</span></div><div class="line"><a name="l01557"></a><span class="lineno"> 1557</span>&#160; LD_RHS_VFMA_M0xN0(8, a, c);</div><div class="line"><a name="l01558"></a><span class="lineno"> 1558</span>&#160; LD_RHS_VFMA_M0xN0(9, a, c);</div><div class="line"><a name="l01559"></a><span class="lineno"> 1559</span>&#160; LD_RHS_VFMA_M0xN0(A, a, c);</div><div class="line"><a name="l01560"></a><span class="lineno"> 1560</span>&#160; LD_RHS_VFMA_M0xN0(B, a, c);</div><div class="line"><a name="l01561"></a><span class="lineno"> 1561</span>&#160; LD_RHS_VFMA_M0xN0(C, a, c);</div><div class="line"><a name="l01562"></a><span class="lineno"> 1562</span>&#160; LD_RHS_VFMA_M0xN0(D, a, c);</div><div class="line"><a name="l01563"></a><span class="lineno"> 1563</span>&#160; LD_RHS_VFMA_M0xN0(E, a, c);</div><div class="line"><a name="l01564"></a><span class="lineno"> 1564</span>&#160; LD_RHS_VFMA_M0xN0(F, a, c);</div><div class="line"><a name="l01565"></a><span class="lineno"> 1565</span>&#160;<span class="preprocessor">#endif // K0 &gt; 8</span></div><div class="line"><a name="l01566"></a><span class="lineno"> 1566</span>&#160;</div><div class="line"><a name="l01567"></a><span class="lineno"> 1567</span>&#160; lhs_offset += K0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01568"></a><span class="lineno"> 1568</span>&#160; rhs_offset += K0 * RHS_STEP_X * RHS_STEP_LOOP * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01569"></a><span class="lineno"> 1569</span>&#160; }</div><div class="line"><a name="l01570"></a><span class="lineno"> 1570</span>&#160;</div><div class="line"><a name="l01571"></a><span class="lineno"> 1571</span>&#160; <span class="comment">// Left-over accumulations</span></div><div class="line"><a name="l01572"></a><span class="lineno"> 1572</span>&#160; <span class="keywordflow">for</span>(; i &lt; K; ++i)</div><div class="line"><a name="l01573"></a><span class="lineno"> 1573</span>&#160; {</div><div class="line"><a name="l01574"></a><span class="lineno"> 1574</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l01575"></a><span class="lineno"> 1575</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01576"></a><span class="lineno"> 1576</span>&#160; a0 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 0 * lhs_stride_y + zin0));</div><div class="line"><a name="l01577"></a><span class="lineno"> 1577</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l01578"></a><span class="lineno"> 1578</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01579"></a><span class="lineno"> 1579</span>&#160; a1 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 1 * lhs_stride_y + zin1));</div><div class="line"><a name="l01580"></a><span class="lineno"> 1580</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l01581"></a><span class="lineno"> 1581</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l01582"></a><span class="lineno"> 1582</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01583"></a><span class="lineno"> 1583</span>&#160; a2 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 2 * lhs_stride_y + zin2));</div><div class="line"><a name="l01584"></a><span class="lineno"> 1584</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l01585"></a><span class="lineno"> 1585</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l01586"></a><span class="lineno"> 1586</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01587"></a><span class="lineno"> 1587</span>&#160; a3 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 3 * lhs_stride_y + zin3));</div><div class="line"><a name="l01588"></a><span class="lineno"> 1588</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l01589"></a><span class="lineno"> 1589</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l01590"></a><span class="lineno"> 1590</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01591"></a><span class="lineno"> 1591</span>&#160; a4 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 4 * lhs_stride_y + zin4));</div><div class="line"><a name="l01592"></a><span class="lineno"> 1592</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l01593"></a><span class="lineno"> 1593</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l01594"></a><span class="lineno"> 1594</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01595"></a><span class="lineno"> 1595</span>&#160; a5 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 5 * lhs_stride_y + zin5));</div><div class="line"><a name="l01596"></a><span class="lineno"> 1596</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l01597"></a><span class="lineno"> 1597</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l01598"></a><span class="lineno"> 1598</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01599"></a><span class="lineno"> 1599</span>&#160; a6 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 6 * lhs_stride_y + zin6));</div><div class="line"><a name="l01600"></a><span class="lineno"> 1600</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l01601"></a><span class="lineno"> 1601</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l01602"></a><span class="lineno"> 1602</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l01603"></a><span class="lineno"> 1603</span>&#160; a7 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 7 * lhs_stride_y + zin7));</div><div class="line"><a name="l01604"></a><span class="lineno"> 1604</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l01605"></a><span class="lineno"> 1605</span>&#160;</div><div class="line"><a name="l01606"></a><span class="lineno"> 1606</span>&#160; LD_RHS_VFMA_M0xN0(0, a, c);</div><div class="line"><a name="l01607"></a><span class="lineno"> 1607</span>&#160;</div><div class="line"><a name="l01608"></a><span class="lineno"> 1608</span>&#160; lhs_offset += <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01609"></a><span class="lineno"> 1609</span>&#160; rhs_offset += RHS_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01610"></a><span class="lineno"> 1610</span>&#160; }</div><div class="line"><a name="l01611"></a><span class="lineno"> 1611</span>&#160;</div><div class="line"><a name="l01612"></a><span class="lineno"> 1612</span>&#160; __global uchar *dst_addr = dst_ptr + dst_offset_first_element_in_bytes + (x * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (y * (uint)M0 * dst_stride_y);</div><div class="line"><a name="l01613"></a><span class="lineno"> 1613</span>&#160;</div><div class="line"><a name="l01614"></a><span class="lineno"> 1614</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(8, uint, zout, 0); <span class="comment">//uint zout0=0,zout1=0,zout2=0,... zout7=0;</span></div><div class="line"><a name="l01615"></a><span class="lineno"> 1615</span>&#160;</div><div class="line"><a name="l01616"></a><span class="lineno"> 1616</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01617"></a><span class="lineno"> 1617</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l01618"></a><span class="lineno"> 1618</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zout, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, dst_cross_plane_pad, dst_stride_y);</div><div class="line"><a name="l01619"></a><span class="lineno"> 1619</span>&#160;</div><div class="line"><a name="l01620"></a><span class="lineno"> 1620</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l01621"></a><span class="lineno"> 1621</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l01622"></a><span class="lineno"> 1622</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l01623"></a><span class="lineno"> 1623</span>&#160;</div><div class="line"><a name="l01624"></a><span class="lineno"> 1624</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01625"></a><span class="lineno"> 1625</span>&#160;</div><div class="line"><a name="l01626"></a><span class="lineno"> 1626</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l01627"></a><span class="lineno"> 1627</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l01628"></a><span class="lineno"> 1628</span>&#160;</div><div class="line"><a name="l01629"></a><span class="lineno"> 1629</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01630"></a><span class="lineno"> 1630</span>&#160;</div><div class="line"><a name="l01631"></a><span class="lineno"> 1631</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l01632"></a><span class="lineno"> 1632</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l01633"></a><span class="lineno"> 1633</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, ALPHA);</div><div class="line"><a name="l01634"></a><span class="lineno"> 1634</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l01635"></a><span class="lineno"> 1635</span>&#160;</div><div class="line"><a name="l01636"></a><span class="lineno"> 1636</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l01637"></a><span class="lineno"> 1637</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l01638"></a><span class="lineno"> 1638</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01639"></a><span class="lineno"> 1639</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l01640"></a><span class="lineno"> 1640</span>&#160;</div><div class="line"><a name="l01641"></a><span class="lineno"> 1641</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l01642"></a><span class="lineno"> 1642</span>&#160;</div><div class="line"><a name="l01643"></a><span class="lineno"> 1643</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l01644"></a><span class="lineno"> 1644</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l01645"></a><span class="lineno"> 1645</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l01646"></a><span class="lineno"> 1646</span>&#160;</div><div class="line"><a name="l01647"></a><span class="lineno"> 1647</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l01648"></a><span class="lineno"> 1648</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(M0, c, bias0);</div><div class="line"><a name="l01649"></a><span class="lineno"> 1649</span>&#160;</div><div class="line"><a name="l01650"></a><span class="lineno"> 1650</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01651"></a><span class="lineno"> 1651</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (get_global_id(1) * (uint)M0 * bias_stride_y) + get_global_id(</div><div class="line"><a name="l01652"></a><span class="lineno"> 1652</span>&#160; 2) * bias_stride_z;</div><div class="line"><a name="l01653"></a><span class="lineno"> 1653</span>&#160;</div><div class="line"><a name="l01654"></a><span class="lineno"> 1654</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l01655"></a><span class="lineno"> 1655</span>&#160;</div><div class="line"><a name="l01656"></a><span class="lineno"> 1656</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l01657"></a><span class="lineno"> 1657</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l01658"></a><span class="lineno"> 1658</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l01659"></a><span class="lineno"> 1659</span>&#160;</div><div class="line"><a name="l01660"></a><span class="lineno"> 1660</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l01661"></a><span class="lineno"> 1661</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(M0, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l01662"></a><span class="lineno"> 1662</span>&#160;</div><div class="line"><a name="l01663"></a><span class="lineno"> 1663</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01664"></a><span class="lineno"> 1664</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l01665"></a><span class="lineno"> 1665</span>&#160;</div><div class="line"><a name="l01666"></a><span class="lineno"> 1666</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l01667"></a><span class="lineno"> 1667</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(M0, ACTIVATION_TYPE, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, A_VAL, B_VAL);</div><div class="line"><a name="l01668"></a><span class="lineno"> 1668</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l01669"></a><span class="lineno"> 1669</span>&#160;</div><div class="line"><a name="l01670"></a><span class="lineno"> 1670</span>&#160; <span class="comment">// Store output block</span></div><div class="line"><a name="l01671"></a><span class="lineno"> 1671</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, dst_addr, dst_stride_y, zout);</div><div class="line"><a name="l01672"></a><span class="lineno"> 1672</span>&#160;</div><div class="line"><a name="l01673"></a><span class="lineno"> 1673</span>&#160;<span class="preprocessor">#undef RHS_BLOCK_SIZE</span></div><div class="line"><a name="l01674"></a><span class="lineno"> 1674</span>&#160;<span class="preprocessor">#undef RHS_OFFSET_X</span></div><div class="line"><a name="l01675"></a><span class="lineno"> 1675</span>&#160;<span class="preprocessor">#undef RHS_STEP_X</span></div><div class="line"><a name="l01676"></a><span class="lineno"> 1676</span>&#160;}</div><div class="line"><a name="l01677"></a><span class="lineno"> 1677</span>&#160;<span class="preprocessor">#endif // defined(M0) &amp;&amp; defined(N0) &amp;&amp; defined(K0) &amp;&amp; defined(H0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(M) &amp;&amp; defined(N) &amp;&amp; defined(K)</span></div><div class="line"><a name="l01678"></a><span class="lineno"> 1678</span>&#160;</div><div class="line"><a name="l01679"></a><span class="lineno"> 1679</span>&#160;<span class="preprocessor">#if defined(M0) &amp;&amp; defined(N0) &amp;&amp; defined(K0) &amp;&amp; defined(V0) &amp;&amp; defined(H0) &amp;&amp; defined(DATA_TYPE) &amp;&amp; defined(M) &amp;&amp; defined(N)</span></div><div class="line"><a name="l01680"></a><span class="lineno"> 1680</span>&#160;</div><div class="line"><a name="l01681"></a><span class="lineno"> 1681</span>&#160;<span class="preprocessor">#if K0 == 2</span></div><div class="line"><a name="l01682"></a><span class="lineno"> 1682</span>&#160;<span class="preprocessor">#define ARM_DOT_K0(a, b, c) \</span></div><div class="line"><a name="l01683"></a><span class="lineno"> 1683</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01684"></a><span class="lineno"> 1684</span>&#160;<span class="preprocessor"> c = fma(a.s0, b.s0, c); \</span></div><div class="line"><a name="l01685"></a><span class="lineno"> 1685</span>&#160;<span class="preprocessor"> c = fma(a.s1, b.s1, c); \</span></div><div class="line"><a name="l01686"></a><span class="lineno"> 1686</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01687"></a><span class="lineno"> 1687</span>&#160;<span class="preprocessor">#elif K0 == 3 // K0 == 3</span></div><div class="line"><a name="l01688"></a><span class="lineno"> 1688</span>&#160;<span class="preprocessor">#define ARM_DOT_K0(a, b, c) \</span></div><div class="line"><a name="l01689"></a><span class="lineno"> 1689</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01690"></a><span class="lineno"> 1690</span>&#160;<span class="preprocessor"> c = fma(a.s0, b.s0, c); \</span></div><div class="line"><a name="l01691"></a><span class="lineno"> 1691</span>&#160;<span class="preprocessor"> c = fma(a.s1, b.s1, c); \</span></div><div class="line"><a name="l01692"></a><span class="lineno"> 1692</span>&#160;<span class="preprocessor"> c = fma(a.s2, b.s2, c); \</span></div><div class="line"><a name="l01693"></a><span class="lineno"> 1693</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01694"></a><span class="lineno"> 1694</span>&#160;<span class="preprocessor">#elif K0 == 4 // K0 == 4</span></div><div class="line"><a name="l01695"></a><span class="lineno"> 1695</span>&#160;<span class="preprocessor">#define ARM_DOT_K0(a, b, c) \</span></div><div class="line"><a name="l01696"></a><span class="lineno"> 1696</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01697"></a><span class="lineno"> 1697</span>&#160;<span class="preprocessor"> c = fma(a.s0, b.s0, c); \</span></div><div class="line"><a name="l01698"></a><span class="lineno"> 1698</span>&#160;<span class="preprocessor"> c = fma(a.s1, b.s1, c); \</span></div><div class="line"><a name="l01699"></a><span class="lineno"> 1699</span>&#160;<span class="preprocessor"> c = fma(a.s2, b.s2, c); \</span></div><div class="line"><a name="l01700"></a><span class="lineno"> 1700</span>&#160;<span class="preprocessor"> c = fma(a.s3, b.s3, c); \</span></div><div class="line"><a name="l01701"></a><span class="lineno"> 1701</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01702"></a><span class="lineno"> 1702</span>&#160;<span class="preprocessor">#elif K0 == 8 // K0 == 8</span></div><div class="line"><a name="l01703"></a><span class="lineno"> 1703</span>&#160;<span class="preprocessor">#define ARM_DOT_K0(a, b, c) \</span></div><div class="line"><a name="l01704"></a><span class="lineno"> 1704</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01705"></a><span class="lineno"> 1705</span>&#160;<span class="preprocessor"> c = fma(a.s0, b.s0, c); \</span></div><div class="line"><a name="l01706"></a><span class="lineno"> 1706</span>&#160;<span class="preprocessor"> c = fma(a.s1, b.s1, c); \</span></div><div class="line"><a name="l01707"></a><span class="lineno"> 1707</span>&#160;<span class="preprocessor"> c = fma(a.s2, b.s2, c); \</span></div><div class="line"><a name="l01708"></a><span class="lineno"> 1708</span>&#160;<span class="preprocessor"> c = fma(a.s3, b.s3, c); \</span></div><div class="line"><a name="l01709"></a><span class="lineno"> 1709</span>&#160;<span class="preprocessor"> c = fma(a.s4, b.s4, c); \</span></div><div class="line"><a name="l01710"></a><span class="lineno"> 1710</span>&#160;<span class="preprocessor"> c = fma(a.s5, b.s5, c); \</span></div><div class="line"><a name="l01711"></a><span class="lineno"> 1711</span>&#160;<span class="preprocessor"> c = fma(a.s6, b.s6, c); \</span></div><div class="line"><a name="l01712"></a><span class="lineno"> 1712</span>&#160;<span class="preprocessor"> c = fma(a.s7, b.s7, c); \</span></div><div class="line"><a name="l01713"></a><span class="lineno"> 1713</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01714"></a><span class="lineno"> 1714</span>&#160;<span class="preprocessor">#elif K0 == 16 // K0 == 16</span></div><div class="line"><a name="l01715"></a><span class="lineno"> 1715</span>&#160;<span class="preprocessor">#define ARM_DOT_K0(a, b, c) \</span></div><div class="line"><a name="l01716"></a><span class="lineno"> 1716</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01717"></a><span class="lineno"> 1717</span>&#160;<span class="preprocessor"> c = fma(a.s0, b.s0, c); \</span></div><div class="line"><a name="l01718"></a><span class="lineno"> 1718</span>&#160;<span class="preprocessor"> c = fma(a.s1, b.s1, c); \</span></div><div class="line"><a name="l01719"></a><span class="lineno"> 1719</span>&#160;<span class="preprocessor"> c = fma(a.s2, b.s2, c); \</span></div><div class="line"><a name="l01720"></a><span class="lineno"> 1720</span>&#160;<span class="preprocessor"> c = fma(a.s3, b.s3, c); \</span></div><div class="line"><a name="l01721"></a><span class="lineno"> 1721</span>&#160;<span class="preprocessor"> c = fma(a.s4, b.s4, c); \</span></div><div class="line"><a name="l01722"></a><span class="lineno"> 1722</span>&#160;<span class="preprocessor"> c = fma(a.s5, b.s5, c); \</span></div><div class="line"><a name="l01723"></a><span class="lineno"> 1723</span>&#160;<span class="preprocessor"> c = fma(a.s6, b.s6, c); \</span></div><div class="line"><a name="l01724"></a><span class="lineno"> 1724</span>&#160;<span class="preprocessor"> c = fma(a.s7, b.s7, c); \</span></div><div class="line"><a name="l01725"></a><span class="lineno"> 1725</span>&#160;<span class="preprocessor"> c = fma(a.s8, b.s8, c); \</span></div><div class="line"><a name="l01726"></a><span class="lineno"> 1726</span>&#160;<span class="preprocessor"> c = fma(a.s9, b.s9, c); \</span></div><div class="line"><a name="l01727"></a><span class="lineno"> 1727</span>&#160;<span class="preprocessor"> c = fma(a.sA, b.sA, c); \</span></div><div class="line"><a name="l01728"></a><span class="lineno"> 1728</span>&#160;<span class="preprocessor"> c = fma(a.sB, b.sB, c); \</span></div><div class="line"><a name="l01729"></a><span class="lineno"> 1729</span>&#160;<span class="preprocessor"> c = fma(a.sC, b.sC, c); \</span></div><div class="line"><a name="l01730"></a><span class="lineno"> 1730</span>&#160;<span class="preprocessor"> c = fma(a.sD, b.sD, c); \</span></div><div class="line"><a name="l01731"></a><span class="lineno"> 1731</span>&#160;<span class="preprocessor"> c = fma(a.sE, b.sE, c); \</span></div><div class="line"><a name="l01732"></a><span class="lineno"> 1732</span>&#160;<span class="preprocessor"> c = fma(a.sF, b.sF, c); \</span></div><div class="line"><a name="l01733"></a><span class="lineno"> 1733</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01734"></a><span class="lineno"> 1734</span>&#160;<span class="preprocessor">#else // K0 not supported</span></div><div class="line"><a name="l01735"></a><span class="lineno"> 1735</span>&#160;<span class="preprocessor">#error &quot;K0 value not supported&quot;</span></div><div class="line"><a name="l01736"></a><span class="lineno"> 1736</span>&#160;<span class="preprocessor">#endif // K0 conditions</span></div><div class="line"><a name="l01737"></a><span class="lineno"> 1737</span>&#160;</div><div class="line"><a name="l01738"></a><span class="lineno"> 1738</span>&#160;<span class="preprocessor">#if N0 == 2</span></div><div class="line"><a name="l01739"></a><span class="lineno"> 1739</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(a, b, c) \</span></div><div class="line"><a name="l01740"></a><span class="lineno"> 1740</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01741"></a><span class="lineno"> 1741</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l01742"></a><span class="lineno"> 1742</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l01743"></a><span class="lineno"> 1743</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01744"></a><span class="lineno"> 1744</span>&#160;<span class="preprocessor">#elif N0 == 3 // N0 == 3</span></div><div class="line"><a name="l01745"></a><span class="lineno"> 1745</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(a, b, c) \</span></div><div class="line"><a name="l01746"></a><span class="lineno"> 1746</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01747"></a><span class="lineno"> 1747</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l01748"></a><span class="lineno"> 1748</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l01749"></a><span class="lineno"> 1749</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l01750"></a><span class="lineno"> 1750</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01751"></a><span class="lineno"> 1751</span>&#160;<span class="preprocessor">#elif N0 == 4 // N0 == 4</span></div><div class="line"><a name="l01752"></a><span class="lineno"> 1752</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(a, b, c) \</span></div><div class="line"><a name="l01753"></a><span class="lineno"> 1753</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01754"></a><span class="lineno"> 1754</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l01755"></a><span class="lineno"> 1755</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l01756"></a><span class="lineno"> 1756</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l01757"></a><span class="lineno"> 1757</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##3), (c.s3)); \</span></div><div class="line"><a name="l01758"></a><span class="lineno"> 1758</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01759"></a><span class="lineno"> 1759</span>&#160;<span class="preprocessor">#elif N0 == 8 // N0 == 8</span></div><div class="line"><a name="l01760"></a><span class="lineno"> 1760</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(a, b, c) \</span></div><div class="line"><a name="l01761"></a><span class="lineno"> 1761</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01762"></a><span class="lineno"> 1762</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l01763"></a><span class="lineno"> 1763</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l01764"></a><span class="lineno"> 1764</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l01765"></a><span class="lineno"> 1765</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##3), (c.s3)); \</span></div><div class="line"><a name="l01766"></a><span class="lineno"> 1766</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##4), (c.s4)); \</span></div><div class="line"><a name="l01767"></a><span class="lineno"> 1767</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##5), (c.s5)); \</span></div><div class="line"><a name="l01768"></a><span class="lineno"> 1768</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##6), (c.s6)); \</span></div><div class="line"><a name="l01769"></a><span class="lineno"> 1769</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##7), (c.s7)); \</span></div><div class="line"><a name="l01770"></a><span class="lineno"> 1770</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01771"></a><span class="lineno"> 1771</span>&#160;<span class="preprocessor">#elif N0 == 16 // N0 == 16</span></div><div class="line"><a name="l01772"></a><span class="lineno"> 1772</span>&#160;<span class="preprocessor">#define ARM_DOT_K0XN0(a, b, c) \</span></div><div class="line"><a name="l01773"></a><span class="lineno"> 1773</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l01774"></a><span class="lineno"> 1774</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##0), (c.s0)); \</span></div><div class="line"><a name="l01775"></a><span class="lineno"> 1775</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##1), (c.s1)); \</span></div><div class="line"><a name="l01776"></a><span class="lineno"> 1776</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##2), (c.s2)); \</span></div><div class="line"><a name="l01777"></a><span class="lineno"> 1777</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##3), (c.s3)); \</span></div><div class="line"><a name="l01778"></a><span class="lineno"> 1778</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##4), (c.s4)); \</span></div><div class="line"><a name="l01779"></a><span class="lineno"> 1779</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##5), (c.s5)); \</span></div><div class="line"><a name="l01780"></a><span class="lineno"> 1780</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##6), (c.s6)); \</span></div><div class="line"><a name="l01781"></a><span class="lineno"> 1781</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##7), (c.s7)); \</span></div><div class="line"><a name="l01782"></a><span class="lineno"> 1782</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##8), (c.s8)); \</span></div><div class="line"><a name="l01783"></a><span class="lineno"> 1783</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##9), (c.s9)); \</span></div><div class="line"><a name="l01784"></a><span class="lineno"> 1784</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##A), (c.sA)); \</span></div><div class="line"><a name="l01785"></a><span class="lineno"> 1785</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##B), (c.sB)); \</span></div><div class="line"><a name="l01786"></a><span class="lineno"> 1786</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##C), (c.sC)); \</span></div><div class="line"><a name="l01787"></a><span class="lineno"> 1787</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##D), (c.sD)); \</span></div><div class="line"><a name="l01788"></a><span class="lineno"> 1788</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##E), (c.sE)); \</span></div><div class="line"><a name="l01789"></a><span class="lineno"> 1789</span>&#160;<span class="preprocessor"> ARM_DOT_K0((a), (b##F), (c.sF)); \</span></div><div class="line"><a name="l01790"></a><span class="lineno"> 1790</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l01791"></a><span class="lineno"> 1791</span>&#160;<span class="preprocessor">#else // N0 not supported</span></div><div class="line"><a name="l01792"></a><span class="lineno"> 1792</span>&#160;<span class="preprocessor">#error &quot;N0 value not supported&quot;</span></div><div class="line"><a name="l01793"></a><span class="lineno"> 1793</span>&#160;<span class="preprocessor">#endif // N0 conditions</span></div><div class="line"><a name="l01794"></a><span class="lineno"> 1794</span>&#160;</div><div class="line"><a name="l01852"></a><span class="lineno"> 1852</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_reshaped_lhs_nt_rhs_t(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(lhs),</div><div class="line"><a name="l01853"></a><span class="lineno"> 1853</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(rhs),</div><div class="line"><a name="l01854"></a><span class="lineno"> 1854</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l01855"></a><span class="lineno"> 1855</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>),</div><div class="line"><a name="l01856"></a><span class="lineno"> 1856</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l01857"></a><span class="lineno"> 1857</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l01858"></a><span class="lineno"> 1858</span>&#160; uint k,</div><div class="line"><a name="l01859"></a><span class="lineno"> 1859</span>&#160; uint lhs_stride_z,</div><div class="line"><a name="l01860"></a><span class="lineno"> 1860</span>&#160; uint rhs_stride_z,</div><div class="line"><a name="l01861"></a><span class="lineno"> 1861</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l01862"></a><span class="lineno"> 1862</span>&#160; uint bias_stride_z,</div><div class="line"><a name="l01863"></a><span class="lineno"> 1863</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l01864"></a><span class="lineno"> 1864</span>&#160; uint dst_stride_z</div><div class="line"><a name="l01865"></a><span class="lineno"> 1865</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l01866"></a><span class="lineno"> 1866</span>&#160; ,</div><div class="line"><a name="l01867"></a><span class="lineno"> 1867</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l01868"></a><span class="lineno"> 1868</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l01869"></a><span class="lineno"> 1869</span>&#160; )</div><div class="line"><a name="l01870"></a><span class="lineno"> 1870</span>&#160;{</div><div class="line"><a name="l01871"></a><span class="lineno"> 1871</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l01872"></a><span class="lineno"> 1872</span>&#160;<span class="preprocessor">#define LHS_BLOCK_SIZE ((K0) * (M0))</span></div><div class="line"><a name="l01873"></a><span class="lineno"> 1873</span>&#160;</div><div class="line"><a name="l01874"></a><span class="lineno"> 1874</span>&#160;<span class="preprocessor">#if defined(LHS_INTERLEAVE)</span></div><div class="line"><a name="l01875"></a><span class="lineno"> 1875</span>&#160;<span class="preprocessor">#define LHS_OFFSET_X (K0)</span></div><div class="line"><a name="l01876"></a><span class="lineno"> 1876</span>&#160;<span class="preprocessor">#define LHS_STEP_X ((K0) * (V0))</span></div><div class="line"><a name="l01877"></a><span class="lineno"> 1877</span>&#160;<span class="preprocessor">#define LHS_STEP_LOOP (1)</span></div><div class="line"><a name="l01878"></a><span class="lineno"> 1878</span>&#160;<span class="preprocessor">#else // defined(INTERLEAVE)</span></div><div class="line"><a name="l01879"></a><span class="lineno"> 1879</span>&#160;<span class="preprocessor">#define LHS_OFFSET_X (LHS_BLOCK_SIZE)</span></div><div class="line"><a name="l01880"></a><span class="lineno"> 1880</span>&#160;<span class="preprocessor">#define LHS_STEP_X (K0)</span></div><div class="line"><a name="l01881"></a><span class="lineno"> 1881</span>&#160;<span class="preprocessor">#define LHS_STEP_LOOP (V0)</span></div><div class="line"><a name="l01882"></a><span class="lineno"> 1882</span>&#160;<span class="preprocessor">#endif // defined(INTERLEAVE)</span></div><div class="line"><a name="l01883"></a><span class="lineno"> 1883</span>&#160;</div><div class="line"><a name="l01884"></a><span class="lineno"> 1884</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l01885"></a><span class="lineno"> 1885</span>&#160;<span class="preprocessor">#define RHS_BLOCK_SIZE ((K0) * (N0))</span></div><div class="line"><a name="l01886"></a><span class="lineno"> 1886</span>&#160;</div><div class="line"><a name="l01887"></a><span class="lineno"> 1887</span>&#160; <span class="comment">// RHS offset and step X</span></div><div class="line"><a name="l01888"></a><span class="lineno"> 1888</span>&#160;<span class="preprocessor">#if defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01889"></a><span class="lineno"> 1889</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (K0)</span></div><div class="line"><a name="l01890"></a><span class="lineno"> 1890</span>&#160;<span class="preprocessor">#define RHS_STEP_X ((K0) * (H0))</span></div><div class="line"><a name="l01891"></a><span class="lineno"> 1891</span>&#160;<span class="preprocessor">#define RHS_STEP_LOOP (1)</span></div><div class="line"><a name="l01892"></a><span class="lineno"> 1892</span>&#160;<span class="preprocessor">#else // defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01893"></a><span class="lineno"> 1893</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (RHS_BLOCK_SIZE)</span></div><div class="line"><a name="l01894"></a><span class="lineno"> 1894</span>&#160;<span class="preprocessor">#define RHS_STEP_X (K0)</span></div><div class="line"><a name="l01895"></a><span class="lineno"> 1895</span>&#160;<span class="preprocessor">#define RHS_STEP_LOOP (H0)</span></div><div class="line"><a name="l01896"></a><span class="lineno"> 1896</span>&#160;<span class="preprocessor">#endif // defined(RHS_INTERLEAVE)</span></div><div class="line"><a name="l01897"></a><span class="lineno"> 1897</span>&#160;</div><div class="line"><a name="l01898"></a><span class="lineno"> 1898</span>&#160;<span class="preprocessor">#if defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l01899"></a><span class="lineno"> 1899</span>&#160; <span class="keywordflow">if</span>((get_global_id(0) * N0 &gt;= N) || (get_global_id(1) * M0 &gt;= M))</div><div class="line"><a name="l01900"></a><span class="lineno"> 1900</span>&#160; {</div><div class="line"><a name="l01901"></a><span class="lineno"> 1901</span>&#160; <span class="keywordflow">return</span>;</div><div class="line"><a name="l01902"></a><span class="lineno"> 1902</span>&#160; }</div><div class="line"><a name="l01903"></a><span class="lineno"> 1903</span>&#160;<span class="preprocessor">#endif // defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l01904"></a><span class="lineno"> 1904</span>&#160;</div><div class="line"><a name="l01905"></a><span class="lineno"> 1905</span>&#160; <span class="comment">// Compute LHS matrix address</span></div><div class="line"><a name="l01906"></a><span class="lineno"> 1906</span>&#160; __global uchar *lhs_addr = lhs_ptr + lhs_offset_first_element_in_bytes + (get_global_id(1) % V0) * (uint)LHS_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + (get_global_id(1) / V0) * (uint)lhs_stride_y +</div><div class="line"><a name="l01907"></a><span class="lineno"> 1907</span>&#160; (get_global_id(2) * lhs_stride_z);</div><div class="line"><a name="l01908"></a><span class="lineno"> 1908</span>&#160;</div><div class="line"><a name="l01909"></a><span class="lineno"> 1909</span>&#160; <span class="comment">// Compute RHS matrix address</span></div><div class="line"><a name="l01910"></a><span class="lineno"> 1910</span>&#160; __global uchar *rhs_addr = rhs_ptr + rhs_offset_first_element_in_bytes + (get_global_id(0) % H0) * (uint)RHS_OFFSET_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>) + (get_global_id(0) / (uint)H0) * rhs_stride_y;</div><div class="line"><a name="l01911"></a><span class="lineno"> 1911</span>&#160;</div><div class="line"><a name="l01912"></a><span class="lineno"> 1912</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01913"></a><span class="lineno"> 1913</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l01914"></a><span class="lineno"> 1914</span>&#160; rhs_addr += (get_global_id(2) % MATRIX_B_DEPTH) * rhs_stride_z;</div><div class="line"><a name="l01915"></a><span class="lineno"> 1915</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01916"></a><span class="lineno"> 1916</span>&#160; rhs_addr += get_global_id(2) * rhs_stride_z;</div><div class="line"><a name="l01917"></a><span class="lineno"> 1917</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l01918"></a><span class="lineno"> 1918</span>&#160;</div><div class="line"><a name="l01919"></a><span class="lineno"> 1919</span>&#160; <span class="comment">// Initialize the accumulators</span></div><div class="line"><a name="l01920"></a><span class="lineno"> 1920</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0), c, 0); <span class="comment">//VEC_DATA_TYPE(DATA_TYPE, N0) c0=0,c1=0,c2=0,... c(M0-1)=0;</span></div><div class="line"><a name="l01921"></a><span class="lineno"> 1921</span>&#160;</div><div class="line"><a name="l01922"></a><span class="lineno"> 1922</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, uint, zlhs, 0); <span class="comment">//uint zlhs0=0,zlhs1=0,zlhs2=0,... zlhs7=0;</span></div><div class="line"><a name="l01923"></a><span class="lineno"> 1923</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zero, 0);</div><div class="line"><a name="l01924"></a><span class="lineno"> 1924</span>&#160;</div><div class="line"><a name="l01925"></a><span class="lineno"> 1925</span>&#160; <span class="keywordflow">for</span>(<span class="keywordtype">int</span> i = 0; i &lt; k; i += K0)</div><div class="line"><a name="l01926"></a><span class="lineno"> 1926</span>&#160; {</div><div class="line"><a name="l01927"></a><span class="lineno"> 1927</span>&#160; <span class="comment">// Supported cases (M0, K0):</span></div><div class="line"><a name="l01928"></a><span class="lineno"> 1928</span>&#160; <span class="comment">// 1,2 - 1,3 - 1,4 - 1,8 - 1,16</span></div><div class="line"><a name="l01929"></a><span class="lineno"> 1929</span>&#160; <span class="comment">// 2,2 - 2,3 - 2,4 - 2,8 - 2,16</span></div><div class="line"><a name="l01930"></a><span class="lineno"> 1930</span>&#160; <span class="comment">// 3,2 - 3,3 - 3,4 - 3,8 - 3,16</span></div><div class="line"><a name="l01931"></a><span class="lineno"> 1931</span>&#160; <span class="comment">// 4,2 - 4,3 - 4,4 - 4,8 - 4,16</span></div><div class="line"><a name="l01932"></a><span class="lineno"> 1932</span>&#160; <span class="comment">// 5,2 - 5,3 - 5,4 - 5,8 - 5,16</span></div><div class="line"><a name="l01933"></a><span class="lineno"> 1933</span>&#160; <span class="comment">// 6,2 - 6,3 - 6,4 - 6,8 - 6,16</span></div><div class="line"><a name="l01934"></a><span class="lineno"> 1934</span>&#160; <span class="comment">// 7,2 - 7,3 - 7,4 - 7,8 - 7,16</span></div><div class="line"><a name="l01935"></a><span class="lineno"> 1935</span>&#160; <span class="comment">// 8,2 - 8,3 - 8,4 - 8,8 - 8,16</span></div><div class="line"><a name="l01936"></a><span class="lineno"> 1936</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l01937"></a><span class="lineno"> 1937</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, lhs_addr, 0, LHS_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zlhs);</div><div class="line"><a name="l01938"></a><span class="lineno"> 1938</span>&#160;</div><div class="line"><a name="l01939"></a><span class="lineno"> 1939</span>&#160; <span class="comment">// Load values from RHS matrix</span></div><div class="line"><a name="l01940"></a><span class="lineno"> 1940</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(N0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, rhs_addr, 0, RHS_STEP_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), zero);</div><div class="line"><a name="l01941"></a><span class="lineno"> 1941</span>&#160;</div><div class="line"><a name="l01942"></a><span class="lineno"> 1942</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l01943"></a><span class="lineno"> 1943</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a0, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c0);</div><div class="line"><a name="l01944"></a><span class="lineno"> 1944</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l01945"></a><span class="lineno"> 1945</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a1, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c1);</div><div class="line"><a name="l01946"></a><span class="lineno"> 1946</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l01947"></a><span class="lineno"> 1947</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l01948"></a><span class="lineno"> 1948</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a2, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c2);</div><div class="line"><a name="l01949"></a><span class="lineno"> 1949</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l01950"></a><span class="lineno"> 1950</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l01951"></a><span class="lineno"> 1951</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a3, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c3);</div><div class="line"><a name="l01952"></a><span class="lineno"> 1952</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l01953"></a><span class="lineno"> 1953</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l01954"></a><span class="lineno"> 1954</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a4, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c4);</div><div class="line"><a name="l01955"></a><span class="lineno"> 1955</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l01956"></a><span class="lineno"> 1956</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l01957"></a><span class="lineno"> 1957</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a5, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c5);</div><div class="line"><a name="l01958"></a><span class="lineno"> 1958</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l01959"></a><span class="lineno"> 1959</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l01960"></a><span class="lineno"> 1960</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a6, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c6);</div><div class="line"><a name="l01961"></a><span class="lineno"> 1961</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l01962"></a><span class="lineno"> 1962</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l01963"></a><span class="lineno"> 1963</span>&#160; <a class="code" href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a>(a7, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c7);</div><div class="line"><a name="l01964"></a><span class="lineno"> 1964</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l01965"></a><span class="lineno"> 1965</span>&#160;</div><div class="line"><a name="l01966"></a><span class="lineno"> 1966</span>&#160; lhs_addr += (M0 * LHS_STEP_X * LHS_STEP_LOOP) * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01967"></a><span class="lineno"> 1967</span>&#160; rhs_addr += (N0 * RHS_STEP_X * RHS_STEP_LOOP) * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l01968"></a><span class="lineno"> 1968</span>&#160; }</div><div class="line"><a name="l01969"></a><span class="lineno"> 1969</span>&#160;</div><div class="line"><a name="l01970"></a><span class="lineno"> 1970</span>&#160; __global uchar *dst_addr = dst_ptr + dst_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (get_global_id(1) * (uint)M0 * dst_stride_y);</div><div class="line"><a name="l01971"></a><span class="lineno"> 1971</span>&#160;</div><div class="line"><a name="l01972"></a><span class="lineno"> 1972</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, uint, zout, 0);</div><div class="line"><a name="l01973"></a><span class="lineno"> 1973</span>&#160;</div><div class="line"><a name="l01974"></a><span class="lineno"> 1974</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01975"></a><span class="lineno"> 1975</span>&#160;</div><div class="line"><a name="l01976"></a><span class="lineno"> 1976</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l01977"></a><span class="lineno"> 1977</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zout, get_global_id(1), HEIGHT_GEMM3D, DEPTH_GEMM3D, dst_cross_plane_pad, dst_stride_y);</div><div class="line"><a name="l01978"></a><span class="lineno"> 1978</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l01979"></a><span class="lineno"> 1979</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l01980"></a><span class="lineno"> 1980</span>&#160; dst_addr += get_global_id(2) * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l01981"></a><span class="lineno"> 1981</span>&#160;</div><div class="line"><a name="l01982"></a><span class="lineno"> 1982</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01983"></a><span class="lineno"> 1983</span>&#160;</div><div class="line"><a name="l01984"></a><span class="lineno"> 1984</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l01985"></a><span class="lineno"> 1985</span>&#160; dst_addr += get_global_id(2) * dst_stride_z;</div><div class="line"><a name="l01986"></a><span class="lineno"> 1986</span>&#160;</div><div class="line"><a name="l01987"></a><span class="lineno"> 1987</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l01988"></a><span class="lineno"> 1988</span>&#160;</div><div class="line"><a name="l01989"></a><span class="lineno"> 1989</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l01990"></a><span class="lineno"> 1990</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l01991"></a><span class="lineno"> 1991</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, ALPHA);</div><div class="line"><a name="l01992"></a><span class="lineno"> 1992</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l01993"></a><span class="lineno"> 1993</span>&#160;</div><div class="line"><a name="l01994"></a><span class="lineno"> 1994</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l01995"></a><span class="lineno"> 1995</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l01996"></a><span class="lineno"> 1996</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l01997"></a><span class="lineno"> 1997</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l01998"></a><span class="lineno"> 1998</span>&#160;</div><div class="line"><a name="l01999"></a><span class="lineno"> 1999</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l02000"></a><span class="lineno"> 2000</span>&#160;</div><div class="line"><a name="l02001"></a><span class="lineno"> 2001</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02002"></a><span class="lineno"> 2002</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02003"></a><span class="lineno"> 2003</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02004"></a><span class="lineno"> 2004</span>&#160;</div><div class="line"><a name="l02005"></a><span class="lineno"> 2005</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l02006"></a><span class="lineno"> 2006</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(M0, c, bias0);</div><div class="line"><a name="l02007"></a><span class="lineno"> 2007</span>&#160;</div><div class="line"><a name="l02008"></a><span class="lineno"> 2008</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02009"></a><span class="lineno"> 2009</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (get_global_id(1) * (uint)M0 * bias_stride_y) + get_global_id(</div><div class="line"><a name="l02010"></a><span class="lineno"> 2010</span>&#160; 2) * bias_stride_z;</div><div class="line"><a name="l02011"></a><span class="lineno"> 2011</span>&#160;</div><div class="line"><a name="l02012"></a><span class="lineno"> 2012</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l02013"></a><span class="lineno"> 2013</span>&#160;</div><div class="line"><a name="l02014"></a><span class="lineno"> 2014</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02015"></a><span class="lineno"> 2015</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02016"></a><span class="lineno"> 2016</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02017"></a><span class="lineno"> 2017</span>&#160;</div><div class="line"><a name="l02018"></a><span class="lineno"> 2018</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l02019"></a><span class="lineno"> 2019</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(M0, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l02020"></a><span class="lineno"> 2020</span>&#160;</div><div class="line"><a name="l02021"></a><span class="lineno"> 2021</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02022"></a><span class="lineno"> 2022</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l02023"></a><span class="lineno"> 2023</span>&#160;</div><div class="line"><a name="l02024"></a><span class="lineno"> 2024</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02025"></a><span class="lineno"> 2025</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(M0, ACTIVATION_TYPE, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, A_VAL, B_VAL);</div><div class="line"><a name="l02026"></a><span class="lineno"> 2026</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02027"></a><span class="lineno"> 2027</span>&#160;</div><div class="line"><a name="l02028"></a><span class="lineno"> 2028</span>&#160; <span class="comment">// Store output block</span></div><div class="line"><a name="l02029"></a><span class="lineno"> 2029</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, dst_addr, dst_stride_y, zout);</div><div class="line"><a name="l02030"></a><span class="lineno"> 2030</span>&#160;</div><div class="line"><a name="l02031"></a><span class="lineno"> 2031</span>&#160;<span class="preprocessor">#undef LHS_BLOCK_SIZE</span></div><div class="line"><a name="l02032"></a><span class="lineno"> 2032</span>&#160;<span class="preprocessor">#undef LHS_OFFSET_X</span></div><div class="line"><a name="l02033"></a><span class="lineno"> 2033</span>&#160;<span class="preprocessor">#undef LHS_STEP_X</span></div><div class="line"><a name="l02034"></a><span class="lineno"> 2034</span>&#160;<span class="preprocessor">#undef RHS_BLOCK_SIZE</span></div><div class="line"><a name="l02035"></a><span class="lineno"> 2035</span>&#160;<span class="preprocessor">#undef RHS_OFFSET_X</span></div><div class="line"><a name="l02036"></a><span class="lineno"> 2036</span>&#160;<span class="preprocessor">#undef RHS_STEP_X</span></div><div class="line"><a name="l02037"></a><span class="lineno"> 2037</span>&#160;}</div><div class="line"><a name="l02038"></a><span class="lineno"> 2038</span>&#160;</div><div class="line"><a name="l02039"></a><span class="lineno"> 2039</span>&#160;<span class="preprocessor">#endif // defined(M0) &amp;&amp; defined(N0) &amp;&amp; defined(K0) &amp;&amp; defined(V0) &amp;&amp; defined(H0) &amp;&amp; defined(K) &amp;&amp; defined(DATA_TYPE)</span></div><div class="line"><a name="l02040"></a><span class="lineno"> 2040</span>&#160;</div><div class="line"><a name="l02041"></a><span class="lineno"> 2041</span>&#160;<span class="preprocessor">#if defined(M0) &amp;&amp; defined(N0) &amp;&amp; defined(K0) &amp;&amp; defined(K) &amp;&amp; defined(DATA_TYPE)</span></div><div class="line"><a name="l02042"></a><span class="lineno"> 2042</span>&#160;</div><div class="line"><a name="l02043"></a><span class="lineno"> 2043</span>&#160;<span class="preprocessor">#define VFMA(a, b, c) \</span></div><div class="line"><a name="l02044"></a><span class="lineno"> 2044</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02045"></a><span class="lineno"> 2045</span>&#160;<span class="preprocessor"> c = fma(a, b, c); \</span></div><div class="line"><a name="l02046"></a><span class="lineno"> 2046</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02047"></a><span class="lineno"> 2047</span>&#160;</div><div class="line"><a name="l02048"></a><span class="lineno"> 2048</span>&#160;<span class="preprocessor">#if M0 == 1</span></div><div class="line"><a name="l02049"></a><span class="lineno"> 2049</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02050"></a><span class="lineno"> 2050</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02051"></a><span class="lineno"> 2051</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02052"></a><span class="lineno"> 2052</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02053"></a><span class="lineno"> 2053</span>&#160;<span class="preprocessor">#elif M0 == 2 // M0 == 2</span></div><div class="line"><a name="l02054"></a><span class="lineno"> 2054</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02055"></a><span class="lineno"> 2055</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02056"></a><span class="lineno"> 2056</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02057"></a><span class="lineno"> 2057</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02058"></a><span class="lineno"> 2058</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02059"></a><span class="lineno"> 2059</span>&#160;<span class="preprocessor">#elif M0 == 3 // M0 == 3</span></div><div class="line"><a name="l02060"></a><span class="lineno"> 2060</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02061"></a><span class="lineno"> 2061</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02062"></a><span class="lineno"> 2062</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02063"></a><span class="lineno"> 2063</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02064"></a><span class="lineno"> 2064</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l02065"></a><span class="lineno"> 2065</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02066"></a><span class="lineno"> 2066</span>&#160;<span class="preprocessor">#elif M0 == 4 // M0 == 4</span></div><div class="line"><a name="l02067"></a><span class="lineno"> 2067</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02068"></a><span class="lineno"> 2068</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02069"></a><span class="lineno"> 2069</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02070"></a><span class="lineno"> 2070</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02071"></a><span class="lineno"> 2071</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l02072"></a><span class="lineno"> 2072</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l02073"></a><span class="lineno"> 2073</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02074"></a><span class="lineno"> 2074</span>&#160;<span class="preprocessor">#elif M0 == 5 // M0 == 5</span></div><div class="line"><a name="l02075"></a><span class="lineno"> 2075</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02076"></a><span class="lineno"> 2076</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02077"></a><span class="lineno"> 2077</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02078"></a><span class="lineno"> 2078</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02079"></a><span class="lineno"> 2079</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l02080"></a><span class="lineno"> 2080</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l02081"></a><span class="lineno"> 2081</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l02082"></a><span class="lineno"> 2082</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02083"></a><span class="lineno"> 2083</span>&#160;<span class="preprocessor">#elif M0 == 6 // M0 == 6</span></div><div class="line"><a name="l02084"></a><span class="lineno"> 2084</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02085"></a><span class="lineno"> 2085</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02086"></a><span class="lineno"> 2086</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02087"></a><span class="lineno"> 2087</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02088"></a><span class="lineno"> 2088</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l02089"></a><span class="lineno"> 2089</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l02090"></a><span class="lineno"> 2090</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l02091"></a><span class="lineno"> 2091</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##5).s##i), b, (c##5)); \</span></div><div class="line"><a name="l02092"></a><span class="lineno"> 2092</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02093"></a><span class="lineno"> 2093</span>&#160;<span class="preprocessor">#elif M0 == 7 // M0 == 7</span></div><div class="line"><a name="l02094"></a><span class="lineno"> 2094</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02095"></a><span class="lineno"> 2095</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02096"></a><span class="lineno"> 2096</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02097"></a><span class="lineno"> 2097</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02098"></a><span class="lineno"> 2098</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l02099"></a><span class="lineno"> 2099</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l02100"></a><span class="lineno"> 2100</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l02101"></a><span class="lineno"> 2101</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##5).s##i), b, (c##5)); \</span></div><div class="line"><a name="l02102"></a><span class="lineno"> 2102</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##6).s##i), b, (c##6)); \</span></div><div class="line"><a name="l02103"></a><span class="lineno"> 2103</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02104"></a><span class="lineno"> 2104</span>&#160;<span class="preprocessor">#elif M0 == 8 // M0 == 8</span></div><div class="line"><a name="l02105"></a><span class="lineno"> 2105</span>&#160;<span class="preprocessor">#define RHS_VFMA_M0xN0(i, a, b, c) \</span></div><div class="line"><a name="l02106"></a><span class="lineno"> 2106</span>&#160;<span class="preprocessor"> ({ \</span></div><div class="line"><a name="l02107"></a><span class="lineno"> 2107</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##0).s##i), b, (c##0)); \</span></div><div class="line"><a name="l02108"></a><span class="lineno"> 2108</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##1).s##i), b, (c##1)); \</span></div><div class="line"><a name="l02109"></a><span class="lineno"> 2109</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##2).s##i), b, (c##2)); \</span></div><div class="line"><a name="l02110"></a><span class="lineno"> 2110</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##3).s##i), b, (c##3)); \</span></div><div class="line"><a name="l02111"></a><span class="lineno"> 2111</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##4).s##i), b, (c##4)); \</span></div><div class="line"><a name="l02112"></a><span class="lineno"> 2112</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##5).s##i), b, (c##5)); \</span></div><div class="line"><a name="l02113"></a><span class="lineno"> 2113</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##6).s##i), b, (c##6)); \</span></div><div class="line"><a name="l02114"></a><span class="lineno"> 2114</span>&#160;<span class="preprocessor"> VFMA((VEC_DATA_TYPE(DATA_TYPE, N0))((a##7).s##i), b, (c##7)); \</span></div><div class="line"><a name="l02115"></a><span class="lineno"> 2115</span>&#160;<span class="preprocessor"> })</span></div><div class="line"><a name="l02116"></a><span class="lineno"> 2116</span>&#160;<span class="preprocessor">#else // M0 not supported</span></div><div class="line"><a name="l02117"></a><span class="lineno"> 2117</span>&#160;<span class="preprocessor">#error &quot;M0 not supported&quot;</span></div><div class="line"><a name="l02118"></a><span class="lineno"> 2118</span>&#160;<span class="preprocessor">#endif // M0 not supported</span></div><div class="line"><a name="l02119"></a><span class="lineno"> 2119</span>&#160;</div><div class="line"><a name="l02175"></a><span class="lineno"> 2175</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_native(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(lhs),</div><div class="line"><a name="l02176"></a><span class="lineno"> 2176</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(rhs),</div><div class="line"><a name="l02177"></a><span class="lineno"> 2177</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l02178"></a><span class="lineno"> 2178</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>),</div><div class="line"><a name="l02179"></a><span class="lineno"> 2179</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l02180"></a><span class="lineno"> 2180</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l02181"></a><span class="lineno"> 2181</span>&#160; uint lhs_stride_z,</div><div class="line"><a name="l02182"></a><span class="lineno"> 2182</span>&#160; uint rhs_stride_z,</div><div class="line"><a name="l02183"></a><span class="lineno"> 2183</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l02184"></a><span class="lineno"> 2184</span>&#160; uint bias_stride_z,</div><div class="line"><a name="l02185"></a><span class="lineno"> 2185</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l02186"></a><span class="lineno"> 2186</span>&#160; uint dst_stride_z</div><div class="line"><a name="l02187"></a><span class="lineno"> 2187</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l02188"></a><span class="lineno"> 2188</span>&#160; ,</div><div class="line"><a name="l02189"></a><span class="lineno"> 2189</span>&#160; uint lhs_cross_plane_pad</div><div class="line"><a name="l02190"></a><span class="lineno"> 2190</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l02191"></a><span class="lineno"> 2191</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l02192"></a><span class="lineno"> 2192</span>&#160; ,</div><div class="line"><a name="l02193"></a><span class="lineno"> 2193</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l02194"></a><span class="lineno"> 2194</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l02195"></a><span class="lineno"> 2195</span>&#160; )</div><div class="line"><a name="l02196"></a><span class="lineno"> 2196</span>&#160;{</div><div class="line"><a name="l02197"></a><span class="lineno"> 2197</span>&#160; <span class="comment">// Block size</span></div><div class="line"><a name="l02198"></a><span class="lineno"> 2198</span>&#160;<span class="preprocessor">#define RHS_BLOCK_SIZE ((K0) * (N0))</span></div><div class="line"><a name="l02199"></a><span class="lineno"> 2199</span>&#160;</div><div class="line"><a name="l02200"></a><span class="lineno"> 2200</span>&#160; <span class="comment">// RHS offset and step X</span></div><div class="line"><a name="l02201"></a><span class="lineno"> 2201</span>&#160;<span class="preprocessor">#define RHS_OFFSET_X (RHS_BLOCK_SIZE)</span></div><div class="line"><a name="l02202"></a><span class="lineno"> 2202</span>&#160;</div><div class="line"><a name="l02203"></a><span class="lineno"> 2203</span>&#160; uint x = get_global_id(0);</div><div class="line"><a name="l02204"></a><span class="lineno"> 2204</span>&#160; uint y = get_global_id(1);</div><div class="line"><a name="l02205"></a><span class="lineno"> 2205</span>&#160; uint z = get_global_id(2);</div><div class="line"><a name="l02206"></a><span class="lineno"> 2206</span>&#160;</div><div class="line"><a name="l02207"></a><span class="lineno"> 2207</span>&#160;<span class="preprocessor">#if defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l02208"></a><span class="lineno"> 2208</span>&#160; <span class="keywordflow">if</span>((x * N0 &gt;= N) || (y * M0 &gt;= M))</div><div class="line"><a name="l02209"></a><span class="lineno"> 2209</span>&#160; {</div><div class="line"><a name="l02210"></a><span class="lineno"> 2210</span>&#160; <span class="keywordflow">return</span>;</div><div class="line"><a name="l02211"></a><span class="lineno"> 2211</span>&#160; }</div><div class="line"><a name="l02212"></a><span class="lineno"> 2212</span>&#160;<span class="preprocessor">#endif // defined(DUMMY_WORK_ITEMS)</span></div><div class="line"><a name="l02213"></a><span class="lineno"> 2213</span>&#160;</div><div class="line"><a name="l02214"></a><span class="lineno"> 2214</span>&#160; <span class="comment">// Compute LHS matrix address</span></div><div class="line"><a name="l02215"></a><span class="lineno"> 2215</span>&#160; uint lhs_offset = lhs_offset_first_element_in_bytes + y * M0 * (uint)lhs_stride_y;</div><div class="line"><a name="l02216"></a><span class="lineno"> 2216</span>&#160;</div><div class="line"><a name="l02217"></a><span class="lineno"> 2217</span>&#160; <span class="comment">// Compute RHS matrix address</span></div><div class="line"><a name="l02218"></a><span class="lineno"> 2218</span>&#160; uint rhs_offset = rhs_offset_first_element_in_bytes + x * N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l02219"></a><span class="lineno"> 2219</span>&#160;</div><div class="line"><a name="l02220"></a><span class="lineno"> 2220</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02221"></a><span class="lineno"> 2221</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l02222"></a><span class="lineno"> 2222</span>&#160; rhs_offset += (z % MATRIX_B_DEPTH) * rhs_stride_z;</div><div class="line"><a name="l02223"></a><span class="lineno"> 2223</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02224"></a><span class="lineno"> 2224</span>&#160; rhs_offset += z * rhs_stride_z;</div><div class="line"><a name="l02225"></a><span class="lineno"> 2225</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02226"></a><span class="lineno"> 2226</span>&#160;</div><div class="line"><a name="l02227"></a><span class="lineno"> 2227</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, uint, zlhs, 0);</div><div class="line"><a name="l02228"></a><span class="lineno"> 2228</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(16, uint, zero, 0);</div><div class="line"><a name="l02229"></a><span class="lineno"> 2229</span>&#160;</div><div class="line"><a name="l02230"></a><span class="lineno"> 2230</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l02231"></a><span class="lineno"> 2231</span>&#160; <span class="comment">// The plane (zlhs) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l02232"></a><span class="lineno"> 2232</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zlhs, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, lhs_cross_plane_pad, lhs_stride_y);</div><div class="line"><a name="l02233"></a><span class="lineno"> 2233</span>&#160;</div><div class="line"><a name="l02234"></a><span class="lineno"> 2234</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l02235"></a><span class="lineno"> 2235</span>&#160; <span class="comment">// multiply lhs_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l02236"></a><span class="lineno"> 2236</span>&#160; lhs_offset += z * lhs_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l02237"></a><span class="lineno"> 2237</span>&#160;</div><div class="line"><a name="l02238"></a><span class="lineno"> 2238</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l02239"></a><span class="lineno"> 2239</span>&#160;</div><div class="line"><a name="l02240"></a><span class="lineno"> 2240</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l02241"></a><span class="lineno"> 2241</span>&#160; lhs_offset += z * lhs_stride_z;</div><div class="line"><a name="l02242"></a><span class="lineno"> 2242</span>&#160;</div><div class="line"><a name="l02243"></a><span class="lineno"> 2243</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l02244"></a><span class="lineno"> 2244</span>&#160;</div><div class="line"><a name="l02245"></a><span class="lineno"> 2245</span>&#160; <span class="comment">// Initialize the accumulators</span></div><div class="line"><a name="l02246"></a><span class="lineno"> 2246</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0), c, 0); <span class="comment">//VEC_DATA_TYPE(DATA_TYPE, N0) c0=0,c1=0,c2=0,... c(M0-1)=0;</span></div><div class="line"><a name="l02247"></a><span class="lineno"> 2247</span>&#160;</div><div class="line"><a name="l02248"></a><span class="lineno"> 2248</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l02249"></a><span class="lineno"> 2249</span>&#160; <span class="keywordflow">for</span>(; i &lt;= (K - K0); i += K0)</div><div class="line"><a name="l02250"></a><span class="lineno"> 2250</span>&#160; {</div><div class="line"><a name="l02251"></a><span class="lineno"> 2251</span>&#160; <span class="comment">// Supported cases (M0, K0):</span></div><div class="line"><a name="l02252"></a><span class="lineno"> 2252</span>&#160; <span class="comment">// 1,2 - 1,3 - 1,4 - 1,8 - 1,16</span></div><div class="line"><a name="l02253"></a><span class="lineno"> 2253</span>&#160; <span class="comment">// 2,2 - 2,3 - 2,4 - 2,8 - 2,16</span></div><div class="line"><a name="l02254"></a><span class="lineno"> 2254</span>&#160; <span class="comment">// 3,2 - 3,3 - 3,4 - 3,8 - 3,16</span></div><div class="line"><a name="l02255"></a><span class="lineno"> 2255</span>&#160; <span class="comment">// 4,2 - 4,3 - 4,4 - 4,8 - 4,16</span></div><div class="line"><a name="l02256"></a><span class="lineno"> 2256</span>&#160; <span class="comment">// 5,2 - 5,3 - 5,4 - 5,8 - 5,16</span></div><div class="line"><a name="l02257"></a><span class="lineno"> 2257</span>&#160; <span class="comment">// 6,2 - 6,3 - 6,4 - 6,8 - 6,16</span></div><div class="line"><a name="l02258"></a><span class="lineno"> 2258</span>&#160; <span class="comment">// 7,2 - 7,3 - 7,4 - 7,8 - 7,16</span></div><div class="line"><a name="l02259"></a><span class="lineno"> 2259</span>&#160; <span class="comment">// 8,2 - 8,3 - 8,4 - 8,8 - 8,16</span></div><div class="line"><a name="l02260"></a><span class="lineno"> 2260</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l02261"></a><span class="lineno"> 2261</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, K0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, lhs_ptr, lhs_offset, lhs_stride_y, zlhs);</div><div class="line"><a name="l02262"></a><span class="lineno"> 2262</span>&#160;</div><div class="line"><a name="l02263"></a><span class="lineno"> 2263</span>&#160; <span class="comment">// Load values from RHS matrix</span></div><div class="line"><a name="l02264"></a><span class="lineno"> 2264</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(K0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, rhs_ptr, rhs_offset, rhs_stride_y, zero);</div><div class="line"><a name="l02265"></a><span class="lineno"> 2265</span>&#160;</div><div class="line"><a name="l02266"></a><span class="lineno"> 2266</span>&#160; RHS_VFMA_M0xN0(0, a, b0, c);</div><div class="line"><a name="l02267"></a><span class="lineno"> 2267</span>&#160; RHS_VFMA_M0xN0(1, a, b1, c);</div><div class="line"><a name="l02268"></a><span class="lineno"> 2268</span>&#160;<span class="preprocessor">#if K0 &gt; 2</span></div><div class="line"><a name="l02269"></a><span class="lineno"> 2269</span>&#160; RHS_VFMA_M0xN0(2, a, b2, c);</div><div class="line"><a name="l02270"></a><span class="lineno"> 2270</span>&#160;<span class="preprocessor">#endif // K0 &gt; 2</span></div><div class="line"><a name="l02271"></a><span class="lineno"> 2271</span>&#160;<span class="preprocessor">#if K0 &gt; 3</span></div><div class="line"><a name="l02272"></a><span class="lineno"> 2272</span>&#160; RHS_VFMA_M0xN0(3, a, b3, c);</div><div class="line"><a name="l02273"></a><span class="lineno"> 2273</span>&#160;<span class="preprocessor">#endif // K0 &gt; 3</span></div><div class="line"><a name="l02274"></a><span class="lineno"> 2274</span>&#160;<span class="preprocessor">#if K0 &gt; 4</span></div><div class="line"><a name="l02275"></a><span class="lineno"> 2275</span>&#160; RHS_VFMA_M0xN0(4, a, b4, c);</div><div class="line"><a name="l02276"></a><span class="lineno"> 2276</span>&#160; RHS_VFMA_M0xN0(5, a, b5, c);</div><div class="line"><a name="l02277"></a><span class="lineno"> 2277</span>&#160; RHS_VFMA_M0xN0(6, a, b6, c);</div><div class="line"><a name="l02278"></a><span class="lineno"> 2278</span>&#160; RHS_VFMA_M0xN0(7, a, b7, c);</div><div class="line"><a name="l02279"></a><span class="lineno"> 2279</span>&#160;<span class="preprocessor">#endif // K0 &gt; 4</span></div><div class="line"><a name="l02280"></a><span class="lineno"> 2280</span>&#160;<span class="preprocessor">#if K0 &gt; 8</span></div><div class="line"><a name="l02281"></a><span class="lineno"> 2281</span>&#160; RHS_VFMA_M0xN0(8, a, b8, c);</div><div class="line"><a name="l02282"></a><span class="lineno"> 2282</span>&#160; RHS_VFMA_M0xN0(9, a, b9, c);</div><div class="line"><a name="l02283"></a><span class="lineno"> 2283</span>&#160; RHS_VFMA_M0xN0(A, a, b10, c);</div><div class="line"><a name="l02284"></a><span class="lineno"> 2284</span>&#160; RHS_VFMA_M0xN0(B, a, b11, c);</div><div class="line"><a name="l02285"></a><span class="lineno"> 2285</span>&#160; RHS_VFMA_M0xN0(C, a, b12, c);</div><div class="line"><a name="l02286"></a><span class="lineno"> 2286</span>&#160; RHS_VFMA_M0xN0(D, a, b13, c);</div><div class="line"><a name="l02287"></a><span class="lineno"> 2287</span>&#160; RHS_VFMA_M0xN0(E, a, b14, c);</div><div class="line"><a name="l02288"></a><span class="lineno"> 2288</span>&#160; RHS_VFMA_M0xN0(F, a, b15, c);</div><div class="line"><a name="l02289"></a><span class="lineno"> 2289</span>&#160;<span class="preprocessor">#endif // K0 &gt; 8</span></div><div class="line"><a name="l02290"></a><span class="lineno"> 2290</span>&#160;</div><div class="line"><a name="l02291"></a><span class="lineno"> 2291</span>&#160; lhs_offset += K0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l02292"></a><span class="lineno"> 2292</span>&#160; rhs_offset += K0 * rhs_stride_y;</div><div class="line"><a name="l02293"></a><span class="lineno"> 2293</span>&#160; }</div><div class="line"><a name="l02294"></a><span class="lineno"> 2294</span>&#160;</div><div class="line"><a name="l02295"></a><span class="lineno"> 2295</span>&#160; <span class="comment">// Left-over accumulations</span></div><div class="line"><a name="l02296"></a><span class="lineno"> 2296</span>&#160; <span class="keywordflow">for</span>(; i &lt; K; ++i)</div><div class="line"><a name="l02297"></a><span class="lineno"> 2297</span>&#160; {</div><div class="line"><a name="l02298"></a><span class="lineno"> 2298</span>&#160; <span class="comment">// Load values from LHS matrix</span></div><div class="line"><a name="l02299"></a><span class="lineno"> 2299</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02300"></a><span class="lineno"> 2300</span>&#160; a0 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 0 * lhs_stride_y + zlhs0));</div><div class="line"><a name="l02301"></a><span class="lineno"> 2301</span>&#160;<span class="preprocessor">#if M0 &gt; 1</span></div><div class="line"><a name="l02302"></a><span class="lineno"> 2302</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02303"></a><span class="lineno"> 2303</span>&#160; a1 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 1 * lhs_stride_y + zlhs1));</div><div class="line"><a name="l02304"></a><span class="lineno"> 2304</span>&#160;<span class="preprocessor">#endif // M0 &gt; 1</span></div><div class="line"><a name="l02305"></a><span class="lineno"> 2305</span>&#160;<span class="preprocessor">#if M0 &gt; 2</span></div><div class="line"><a name="l02306"></a><span class="lineno"> 2306</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02307"></a><span class="lineno"> 2307</span>&#160; a2 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 2 * lhs_stride_y + zlhs2));</div><div class="line"><a name="l02308"></a><span class="lineno"> 2308</span>&#160;<span class="preprocessor">#endif // M0 &gt; 2</span></div><div class="line"><a name="l02309"></a><span class="lineno"> 2309</span>&#160;<span class="preprocessor">#if M0 &gt; 3</span></div><div class="line"><a name="l02310"></a><span class="lineno"> 2310</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02311"></a><span class="lineno"> 2311</span>&#160; a3 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 3 * lhs_stride_y + zlhs3));</div><div class="line"><a name="l02312"></a><span class="lineno"> 2312</span>&#160;<span class="preprocessor">#endif // M0 &gt; 3</span></div><div class="line"><a name="l02313"></a><span class="lineno"> 2313</span>&#160;<span class="preprocessor">#if M0 &gt; 4</span></div><div class="line"><a name="l02314"></a><span class="lineno"> 2314</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02315"></a><span class="lineno"> 2315</span>&#160; a4 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 4 * lhs_stride_y + zlhs4));</div><div class="line"><a name="l02316"></a><span class="lineno"> 2316</span>&#160;<span class="preprocessor">#endif // M0 &gt; 4</span></div><div class="line"><a name="l02317"></a><span class="lineno"> 2317</span>&#160;<span class="preprocessor">#if M0 &gt; 5</span></div><div class="line"><a name="l02318"></a><span class="lineno"> 2318</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02319"></a><span class="lineno"> 2319</span>&#160; a5 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 5 * lhs_stride_y + zlhs5));</div><div class="line"><a name="l02320"></a><span class="lineno"> 2320</span>&#160;<span class="preprocessor">#endif // M0 &gt; 5</span></div><div class="line"><a name="l02321"></a><span class="lineno"> 2321</span>&#160;<span class="preprocessor">#if M0 &gt; 6</span></div><div class="line"><a name="l02322"></a><span class="lineno"> 2322</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02323"></a><span class="lineno"> 2323</span>&#160; a6 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 6 * lhs_stride_y + zlhs6));</div><div class="line"><a name="l02324"></a><span class="lineno"> 2324</span>&#160;<span class="preprocessor">#endif // M0 &gt; 6</span></div><div class="line"><a name="l02325"></a><span class="lineno"> 2325</span>&#160;<span class="preprocessor">#if M0 &gt; 7</span></div><div class="line"><a name="l02326"></a><span class="lineno"> 2326</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l02327"></a><span class="lineno"> 2327</span>&#160; a7 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(lhs_ptr + lhs_offset + 7 * lhs_stride_y + zlhs7));</div><div class="line"><a name="l02328"></a><span class="lineno"> 2328</span>&#160;<span class="preprocessor">#endif // M0 &gt; 7</span></div><div class="line"><a name="l02329"></a><span class="lineno"> 2329</span>&#160;</div><div class="line"><a name="l02330"></a><span class="lineno"> 2330</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, N0)</div><div class="line"><a name="l02331"></a><span class="lineno"> 2331</span>&#160; <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(N0)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(rhs_ptr + rhs_offset + 0 * rhs_stride_y));</div><div class="line"><a name="l02332"></a><span class="lineno"> 2332</span>&#160; RHS_VFMA_M0xN0(0, a, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">b</a>, c);</div><div class="line"><a name="l02333"></a><span class="lineno"> 2333</span>&#160;</div><div class="line"><a name="l02334"></a><span class="lineno"> 2334</span>&#160; lhs_offset += <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l02335"></a><span class="lineno"> 2335</span>&#160; rhs_offset += rhs_stride_y;</div><div class="line"><a name="l02336"></a><span class="lineno"> 2336</span>&#160; }</div><div class="line"><a name="l02337"></a><span class="lineno"> 2337</span>&#160;</div><div class="line"><a name="l02338"></a><span class="lineno"> 2338</span>&#160; __global uchar *dst_addr = dst_ptr + dst_offset_first_element_in_bytes + (x * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (y * (uint)M0 * dst_stride_y);</div><div class="line"><a name="l02339"></a><span class="lineno"> 2339</span>&#160;</div><div class="line"><a name="l02340"></a><span class="lineno"> 2340</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(M0, uint, zout, 0);</div><div class="line"><a name="l02341"></a><span class="lineno"> 2341</span>&#160;</div><div class="line"><a name="l02342"></a><span class="lineno"> 2342</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02343"></a><span class="lineno"> 2343</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (y * M0) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l02344"></a><span class="lineno"> 2344</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a>(M0, uint, zout, y, HEIGHT_GEMM3D, DEPTH_GEMM3D, dst_cross_plane_pad, dst_stride_y);</div><div class="line"><a name="l02345"></a><span class="lineno"> 2345</span>&#160;</div><div class="line"><a name="l02346"></a><span class="lineno"> 2346</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l02347"></a><span class="lineno"> 2347</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l02348"></a><span class="lineno"> 2348</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l02349"></a><span class="lineno"> 2349</span>&#160;</div><div class="line"><a name="l02350"></a><span class="lineno"> 2350</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02351"></a><span class="lineno"> 2351</span>&#160;</div><div class="line"><a name="l02352"></a><span class="lineno"> 2352</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l02353"></a><span class="lineno"> 2353</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l02354"></a><span class="lineno"> 2354</span>&#160;</div><div class="line"><a name="l02355"></a><span class="lineno"> 2355</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02356"></a><span class="lineno"> 2356</span>&#160;</div><div class="line"><a name="l02357"></a><span class="lineno"> 2357</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l02358"></a><span class="lineno"> 2358</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l02359"></a><span class="lineno"> 2359</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, ALPHA);</div><div class="line"><a name="l02360"></a><span class="lineno"> 2360</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l02361"></a><span class="lineno"> 2361</span>&#160;</div><div class="line"><a name="l02362"></a><span class="lineno"> 2362</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l02363"></a><span class="lineno"> 2363</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l02364"></a><span class="lineno"> 2364</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02365"></a><span class="lineno"> 2365</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l02366"></a><span class="lineno"> 2366</span>&#160;</div><div class="line"><a name="l02367"></a><span class="lineno"> 2367</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l02368"></a><span class="lineno"> 2368</span>&#160;</div><div class="line"><a name="l02369"></a><span class="lineno"> 2369</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02370"></a><span class="lineno"> 2370</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02371"></a><span class="lineno"> 2371</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02372"></a><span class="lineno"> 2372</span>&#160;</div><div class="line"><a name="l02373"></a><span class="lineno"> 2373</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l02374"></a><span class="lineno"> 2374</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(M0, c, bias0);</div><div class="line"><a name="l02375"></a><span class="lineno"> 2375</span>&#160;</div><div class="line"><a name="l02376"></a><span class="lineno"> 2376</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02377"></a><span class="lineno"> 2377</span>&#160; __global uchar *bias_addr = bias_ptr + bias_offset_first_element_in_bytes + (get_global_id(0) * (uint)N0 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (get_global_id(1) * (uint)M0 * bias_stride_y) + get_global_id(</div><div class="line"><a name="l02378"></a><span class="lineno"> 2378</span>&#160; 2) * bias_stride_z;</div><div class="line"><a name="l02379"></a><span class="lineno"> 2379</span>&#160;</div><div class="line"><a name="l02380"></a><span class="lineno"> 2380</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, bias_addr, 0, bias_stride_y, zero);</div><div class="line"><a name="l02381"></a><span class="lineno"> 2381</span>&#160;</div><div class="line"><a name="l02382"></a><span class="lineno"> 2382</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02383"></a><span class="lineno"> 2383</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(M0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02384"></a><span class="lineno"> 2384</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02385"></a><span class="lineno"> 2385</span>&#160;</div><div class="line"><a name="l02386"></a><span class="lineno"> 2386</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l02387"></a><span class="lineno"> 2387</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(M0, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l02388"></a><span class="lineno"> 2388</span>&#160;</div><div class="line"><a name="l02389"></a><span class="lineno"> 2389</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02390"></a><span class="lineno"> 2390</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l02391"></a><span class="lineno"> 2391</span>&#160;</div><div class="line"><a name="l02392"></a><span class="lineno"> 2392</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02393"></a><span class="lineno"> 2393</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(M0, ACTIVATION_TYPE, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, A_VAL, B_VAL);</div><div class="line"><a name="l02394"></a><span class="lineno"> 2394</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02395"></a><span class="lineno"> 2395</span>&#160;</div><div class="line"><a name="l02396"></a><span class="lineno"> 2396</span>&#160; <span class="comment">// Store output block</span></div><div class="line"><a name="l02397"></a><span class="lineno"> 2397</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(M0, N0, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, c, dst_addr, dst_stride_y, zout);</div><div class="line"><a name="l02398"></a><span class="lineno"> 2398</span>&#160;</div><div class="line"><a name="l02399"></a><span class="lineno"> 2399</span>&#160;<span class="preprocessor">#undef RHS_BLOCK_SIZE</span></div><div class="line"><a name="l02400"></a><span class="lineno"> 2400</span>&#160;<span class="preprocessor">#undef RHS_OFFSET_X</span></div><div class="line"><a name="l02401"></a><span class="lineno"> 2401</span>&#160;<span class="preprocessor">#undef RHS_STEP_X</span></div><div class="line"><a name="l02402"></a><span class="lineno"> 2402</span>&#160;}</div><div class="line"><a name="l02403"></a><span class="lineno"> 2403</span>&#160;<span class="preprocessor">#endif // defined(M0) &amp;&amp; defined(N0) &amp;&amp; defined(K0) &amp;&amp; defined(K) &amp;&amp; defined(DATA_TYPE)</span></div><div class="line"><a name="l02404"></a><span class="lineno"> 2404</span>&#160;</div><div class="line"><a name="l02405"></a><span class="lineno"> 2405</span>&#160;<span class="preprocessor">#if defined(COLS_B) &amp;&amp; defined(MULT_TRANSPOSE1XW_WIDTH) &amp;&amp; defined(MULT_INTERLEAVE4X4_HEIGHT)</span></div><div class="line"><a name="l02406"></a><span class="lineno"> 2406</span>&#160;</div><div class="line"><a name="l02452"></a><span class="lineno"> 2452</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_interleaved_transposed_f32(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l02453"></a><span class="lineno"> 2453</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l02454"></a><span class="lineno"> 2454</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l02455"></a><span class="lineno"> 2455</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l02456"></a><span class="lineno"> 2456</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l02457"></a><span class="lineno"> 2457</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l02458"></a><span class="lineno"> 2458</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l02459"></a><span class="lineno"> 2459</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l02460"></a><span class="lineno"> 2460</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l02461"></a><span class="lineno"> 2461</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l02462"></a><span class="lineno"> 2462</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l02463"></a><span class="lineno"> 2463</span>&#160; uint dst_stride_z</div><div class="line"><a name="l02464"></a><span class="lineno"> 2464</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l02465"></a><span class="lineno"> 2465</span>&#160; ,</div><div class="line"><a name="l02466"></a><span class="lineno"> 2466</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l02467"></a><span class="lineno"> 2467</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l02468"></a><span class="lineno"> 2468</span>&#160; )</div><div class="line"><a name="l02469"></a><span class="lineno"> 2469</span>&#160;{</div><div class="line"><a name="l02470"></a><span class="lineno"> 2470</span>&#160; <span class="keywordtype">int</span> x = get_global_id(0) / MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02471"></a><span class="lineno"> 2471</span>&#160; <span class="keywordtype">int</span> y = get_global_id(1) / MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02472"></a><span class="lineno"> 2472</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l02473"></a><span class="lineno"> 2473</span>&#160;</div><div class="line"><a name="l02474"></a><span class="lineno"> 2474</span>&#160; <span class="comment">// Offset</span></div><div class="line"><a name="l02475"></a><span class="lineno"> 2475</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_a = (get_global_id(1) % MULT_INTERLEAVE4X4_HEIGHT) * 4;</div><div class="line"><a name="l02476"></a><span class="lineno"> 2476</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_b = (get_global_id(0) % MULT_TRANSPOSE1XW_WIDTH) * 4;</div><div class="line"><a name="l02477"></a><span class="lineno"> 2477</span>&#160;</div><div class="line"><a name="l02478"></a><span class="lineno"> 2478</span>&#160; <span class="comment">// src_addr_a = address of matrix A</span></div><div class="line"><a name="l02479"></a><span class="lineno"> 2479</span>&#160; <span class="comment">// src_addr_b = address of matrix B</span></div><div class="line"><a name="l02480"></a><span class="lineno"> 2480</span>&#160; <span class="keywordtype">int</span> src0_addr_in_bytes = z * src0_stride_z + y * src0_stride_y + src0_offset_first_element_in_bytes;</div><div class="line"><a name="l02481"></a><span class="lineno"> 2481</span>&#160; <span class="keywordtype">int</span> src1_addr_in_bytes = x * src1_stride_y + src1_offset_first_element_in_bytes;</div><div class="line"><a name="l02482"></a><span class="lineno"> 2482</span>&#160;</div><div class="line"><a name="l02483"></a><span class="lineno"> 2483</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02484"></a><span class="lineno"> 2484</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l02485"></a><span class="lineno"> 2485</span>&#160; src1_addr_in_bytes += (z % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l02486"></a><span class="lineno"> 2486</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02487"></a><span class="lineno"> 2487</span>&#160; src1_addr_in_bytes += z * src1_stride_z;</div><div class="line"><a name="l02488"></a><span class="lineno"> 2488</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02489"></a><span class="lineno"> 2489</span>&#160;</div><div class="line"><a name="l02490"></a><span class="lineno"> 2490</span>&#160; __global <span class="keywordtype">float</span> *src_addr_a = (__global <span class="keywordtype">float</span> *)(src0_ptr + src0_addr_in_bytes);</div><div class="line"><a name="l02491"></a><span class="lineno"> 2491</span>&#160; __global <span class="keywordtype">float</span> *src_addr_b = (__global <span class="keywordtype">float</span> *)(src1_ptr + src1_addr_in_bytes);</div><div class="line"><a name="l02492"></a><span class="lineno"> 2492</span>&#160;</div><div class="line"><a name="l02493"></a><span class="lineno"> 2493</span>&#160; <span class="comment">// Compute end row address for matrix B</span></div><div class="line"><a name="l02494"></a><span class="lineno"> 2494</span>&#160; __global <span class="keywordtype">float</span> *src_end_addr_b = src_addr_b + COLS_B;</div><div class="line"><a name="l02495"></a><span class="lineno"> 2495</span>&#160;</div><div class="line"><a name="l02496"></a><span class="lineno"> 2496</span>&#160; src_addr_a += offset_row_a;</div><div class="line"><a name="l02497"></a><span class="lineno"> 2497</span>&#160; src_addr_b += offset_row_b;</div><div class="line"><a name="l02498"></a><span class="lineno"> 2498</span>&#160;</div><div class="line"><a name="l02499"></a><span class="lineno"> 2499</span>&#160; <span class="comment">// Reset accumulators</span></div><div class="line"><a name="l02500"></a><span class="lineno"> 2500</span>&#160; float4 c0 = 0.0f;</div><div class="line"><a name="l02501"></a><span class="lineno"> 2501</span>&#160; float4 c1 = 0.0f;</div><div class="line"><a name="l02502"></a><span class="lineno"> 2502</span>&#160; float4 c2 = 0.0f;</div><div class="line"><a name="l02503"></a><span class="lineno"> 2503</span>&#160; float4 c3 = 0.0f;</div><div class="line"><a name="l02504"></a><span class="lineno"> 2504</span>&#160;</div><div class="line"><a name="l02505"></a><span class="lineno"> 2505</span>&#160; <span class="keywordflow">for</span>(; src_addr_b &lt;= (src_end_addr_b - (int)(8 * MULT_TRANSPOSE1XW_WIDTH)); src_addr_a += 8 * MULT_INTERLEAVE4X4_HEIGHT, src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH)</div><div class="line"><a name="l02506"></a><span class="lineno"> 2506</span>&#160; {</div><div class="line"><a name="l02507"></a><span class="lineno"> 2507</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02508"></a><span class="lineno"> 2508</span>&#160; float4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02509"></a><span class="lineno"> 2509</span>&#160; float4 b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02510"></a><span class="lineno"> 2510</span>&#160;</div><div class="line"><a name="l02511"></a><span class="lineno"> 2511</span>&#160; c0 += (float4)a0.s0 * b0;</div><div class="line"><a name="l02512"></a><span class="lineno"> 2512</span>&#160; c1 += (float4)a0.s1 * b0;</div><div class="line"><a name="l02513"></a><span class="lineno"> 2513</span>&#160; c2 += (float4)a0.s2 * b0;</div><div class="line"><a name="l02514"></a><span class="lineno"> 2514</span>&#160; c3 += (float4)a0.s3 * b0;</div><div class="line"><a name="l02515"></a><span class="lineno"> 2515</span>&#160;</div><div class="line"><a name="l02516"></a><span class="lineno"> 2516</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02517"></a><span class="lineno"> 2517</span>&#160; a0 = vload4(0, src_addr_a + 4 * MULT_INTERLEAVE4X4_HEIGHT);</div><div class="line"><a name="l02518"></a><span class="lineno"> 2518</span>&#160; b0 = vload4(0, src_addr_b + 4 * MULT_TRANSPOSE1XW_WIDTH);</div><div class="line"><a name="l02519"></a><span class="lineno"> 2519</span>&#160;</div><div class="line"><a name="l02520"></a><span class="lineno"> 2520</span>&#160; c0 += (float4)a0.s0 * b0;</div><div class="line"><a name="l02521"></a><span class="lineno"> 2521</span>&#160; c1 += (float4)a0.s1 * b0;</div><div class="line"><a name="l02522"></a><span class="lineno"> 2522</span>&#160; c2 += (float4)a0.s2 * b0;</div><div class="line"><a name="l02523"></a><span class="lineno"> 2523</span>&#160; c3 += (float4)a0.s3 * b0;</div><div class="line"><a name="l02524"></a><span class="lineno"> 2524</span>&#160; }</div><div class="line"><a name="l02525"></a><span class="lineno"> 2525</span>&#160;</div><div class="line"><a name="l02526"></a><span class="lineno"> 2526</span>&#160; <span class="keywordflow">for</span>(; src_addr_b &lt; src_end_addr_b; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT, src_addr_b += 4 * MULT_TRANSPOSE1XW_WIDTH)</div><div class="line"><a name="l02527"></a><span class="lineno"> 2527</span>&#160; {</div><div class="line"><a name="l02528"></a><span class="lineno"> 2528</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02529"></a><span class="lineno"> 2529</span>&#160; float4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02530"></a><span class="lineno"> 2530</span>&#160; float4 b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02531"></a><span class="lineno"> 2531</span>&#160;</div><div class="line"><a name="l02532"></a><span class="lineno"> 2532</span>&#160; c0 += (float4)a0.s0 * b0;</div><div class="line"><a name="l02533"></a><span class="lineno"> 2533</span>&#160; c1 += (float4)a0.s1 * b0;</div><div class="line"><a name="l02534"></a><span class="lineno"> 2534</span>&#160; c2 += (float4)a0.s2 * b0;</div><div class="line"><a name="l02535"></a><span class="lineno"> 2535</span>&#160; c3 += (float4)a0.s3 * b0;</div><div class="line"><a name="l02536"></a><span class="lineno"> 2536</span>&#160; }</div><div class="line"><a name="l02537"></a><span class="lineno"> 2537</span>&#160;</div><div class="line"><a name="l02538"></a><span class="lineno"> 2538</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l02539"></a><span class="lineno"> 2539</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l02540"></a><span class="lineno"> 2540</span>&#160;</div><div class="line"><a name="l02541"></a><span class="lineno"> 2541</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l02542"></a><span class="lineno"> 2542</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l02543"></a><span class="lineno"> 2543</span>&#160;</div><div class="line"><a name="l02544"></a><span class="lineno"> 2544</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l02545"></a><span class="lineno"> 2545</span>&#160;</div><div class="line"><a name="l02546"></a><span class="lineno"> 2546</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02547"></a><span class="lineno"> 2547</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l02548"></a><span class="lineno"> 2548</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l02549"></a><span class="lineno"> 2549</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l02550"></a><span class="lineno"> 2550</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02551"></a><span class="lineno"> 2551</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l02552"></a><span class="lineno"> 2552</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02553"></a><span class="lineno"> 2553</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l02554"></a><span class="lineno"> 2554</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l02555"></a><span class="lineno"> 2555</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l02556"></a><span class="lineno"> 2556</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l02557"></a><span class="lineno"> 2557</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02558"></a><span class="lineno"> 2558</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l02559"></a><span class="lineno"> 2559</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02560"></a><span class="lineno"> 2560</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l02561"></a><span class="lineno"> 2561</span>&#160;</div><div class="line"><a name="l02562"></a><span class="lineno"> 2562</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * 4) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l02563"></a><span class="lineno"> 2563</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * 4)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l02564"></a><span class="lineno"> 2564</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l02565"></a><span class="lineno"> 2565</span>&#160;</div><div class="line"><a name="l02566"></a><span class="lineno"> 2566</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l02567"></a><span class="lineno"> 2567</span>&#160; zout *= (cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l02568"></a><span class="lineno"> 2568</span>&#160;</div><div class="line"><a name="l02569"></a><span class="lineno"> 2569</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l02570"></a><span class="lineno"> 2570</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l02571"></a><span class="lineno"> 2571</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l02572"></a><span class="lineno"> 2572</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02573"></a><span class="lineno"> 2573</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l02574"></a><span class="lineno"> 2574</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l02575"></a><span class="lineno"> 2575</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02576"></a><span class="lineno"> 2576</span>&#160;</div><div class="line"><a name="l02577"></a><span class="lineno"> 2577</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l02578"></a><span class="lineno"> 2578</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l02579"></a><span class="lineno"> 2579</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <span class="keywordtype">float</span>, c, ALPHA);</div><div class="line"><a name="l02580"></a><span class="lineno"> 2580</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l02581"></a><span class="lineno"> 2581</span>&#160;</div><div class="line"><a name="l02582"></a><span class="lineno"> 2582</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l02583"></a><span class="lineno"> 2583</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l02584"></a><span class="lineno"> 2584</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(4, uint, zero, 0);</div><div class="line"><a name="l02585"></a><span class="lineno"> 2585</span>&#160;</div><div class="line"><a name="l02586"></a><span class="lineno"> 2586</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02587"></a><span class="lineno"> 2587</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)4 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>));</div><div class="line"><a name="l02588"></a><span class="lineno"> 2588</span>&#160;</div><div class="line"><a name="l02589"></a><span class="lineno"> 2589</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l02590"></a><span class="lineno"> 2590</span>&#160;</div><div class="line"><a name="l02591"></a><span class="lineno"> 2591</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02592"></a><span class="lineno"> 2592</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02593"></a><span class="lineno"> 2593</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02594"></a><span class="lineno"> 2594</span>&#160;</div><div class="line"><a name="l02595"></a><span class="lineno"> 2595</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l02596"></a><span class="lineno"> 2596</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(4, c, bias0);</div><div class="line"><a name="l02597"></a><span class="lineno"> 2597</span>&#160;</div><div class="line"><a name="l02598"></a><span class="lineno"> 2598</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02599"></a><span class="lineno"> 2599</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)4 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>)) + (get_global_id(1) * (uint)4 * src2_stride_y) + get_global_id(</div><div class="line"><a name="l02600"></a><span class="lineno"> 2600</span>&#160; 2) * src2_stride_z;</div><div class="line"><a name="l02601"></a><span class="lineno"> 2601</span>&#160;</div><div class="line"><a name="l02602"></a><span class="lineno"> 2602</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(4, 4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l02603"></a><span class="lineno"> 2603</span>&#160;</div><div class="line"><a name="l02604"></a><span class="lineno"> 2604</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02605"></a><span class="lineno"> 2605</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02606"></a><span class="lineno"> 2606</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02607"></a><span class="lineno"> 2607</span>&#160;</div><div class="line"><a name="l02608"></a><span class="lineno"> 2608</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l02609"></a><span class="lineno"> 2609</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(4, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l02610"></a><span class="lineno"> 2610</span>&#160;</div><div class="line"><a name="l02611"></a><span class="lineno"> 2611</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02612"></a><span class="lineno"> 2612</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l02613"></a><span class="lineno"> 2613</span>&#160;</div><div class="line"><a name="l02614"></a><span class="lineno"> 2614</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02615"></a><span class="lineno"> 2615</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(4, ACTIVATION_TYPE, <span class="keywordtype">float</span>, c, A_VAL, B_VAL);</div><div class="line"><a name="l02616"></a><span class="lineno"> 2616</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02617"></a><span class="lineno"> 2617</span>&#160;</div><div class="line"><a name="l02618"></a><span class="lineno"> 2618</span>&#160; <span class="comment">// Store 4x4 block</span></div><div class="line"><a name="l02619"></a><span class="lineno"> 2619</span>&#160; vstore4(c0, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l02620"></a><span class="lineno"> 2620</span>&#160; vstore4(c1, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l02621"></a><span class="lineno"> 2621</span>&#160; vstore4(c2, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l02622"></a><span class="lineno"> 2622</span>&#160; vstore4(c3, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l02623"></a><span class="lineno"> 2623</span>&#160;}</div><div class="line"><a name="l02624"></a><span class="lineno"> 2624</span>&#160;</div><div class="line"><a name="l02672"></a><span class="lineno"> 2672</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_interleaved_transposed_f32_bifrost(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l02673"></a><span class="lineno"> 2673</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l02674"></a><span class="lineno"> 2674</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l02675"></a><span class="lineno"> 2675</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l02676"></a><span class="lineno"> 2676</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l02677"></a><span class="lineno"> 2677</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l02678"></a><span class="lineno"> 2678</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l02679"></a><span class="lineno"> 2679</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l02680"></a><span class="lineno"> 2680</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l02681"></a><span class="lineno"> 2681</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l02682"></a><span class="lineno"> 2682</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l02683"></a><span class="lineno"> 2683</span>&#160; uint dst_stride_z</div><div class="line"><a name="l02684"></a><span class="lineno"> 2684</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l02685"></a><span class="lineno"> 2685</span>&#160; ,</div><div class="line"><a name="l02686"></a><span class="lineno"> 2686</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l02687"></a><span class="lineno"> 2687</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l02688"></a><span class="lineno"> 2688</span>&#160; )</div><div class="line"><a name="l02689"></a><span class="lineno"> 2689</span>&#160;{</div><div class="line"><a name="l02690"></a><span class="lineno"> 2690</span>&#160; <span class="keywordtype">int</span> x = get_global_id(0) / MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02691"></a><span class="lineno"> 2691</span>&#160; <span class="keywordtype">int</span> y = get_global_id(1) / MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02692"></a><span class="lineno"> 2692</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l02693"></a><span class="lineno"> 2693</span>&#160;</div><div class="line"><a name="l02694"></a><span class="lineno"> 2694</span>&#160; <span class="comment">// Offset</span></div><div class="line"><a name="l02695"></a><span class="lineno"> 2695</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_a = (get_global_id(1) % MULT_INTERLEAVE4X4_HEIGHT) * 4;</div><div class="line"><a name="l02696"></a><span class="lineno"> 2696</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_b = (get_global_id(0) % MULT_TRANSPOSE1XW_WIDTH) * 4;</div><div class="line"><a name="l02697"></a><span class="lineno"> 2697</span>&#160;</div><div class="line"><a name="l02698"></a><span class="lineno"> 2698</span>&#160; <span class="comment">// src_addr_a = address of matrix A</span></div><div class="line"><a name="l02699"></a><span class="lineno"> 2699</span>&#160; <span class="comment">// src_addr_b = address of matrix B</span></div><div class="line"><a name="l02700"></a><span class="lineno"> 2700</span>&#160; <span class="keywordtype">int</span> src0_addr_in_bytes = z * src0_stride_z + y * src0_stride_y + src0_offset_first_element_in_bytes;</div><div class="line"><a name="l02701"></a><span class="lineno"> 2701</span>&#160; <span class="keywordtype">int</span> src1_addr_in_bytes = x * src1_stride_y + src1_offset_first_element_in_bytes;</div><div class="line"><a name="l02702"></a><span class="lineno"> 2702</span>&#160;</div><div class="line"><a name="l02703"></a><span class="lineno"> 2703</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02704"></a><span class="lineno"> 2704</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l02705"></a><span class="lineno"> 2705</span>&#160; src1_addr_in_bytes += (z % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l02706"></a><span class="lineno"> 2706</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02707"></a><span class="lineno"> 2707</span>&#160; src1_addr_in_bytes += z * src1_stride_z;</div><div class="line"><a name="l02708"></a><span class="lineno"> 2708</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l02709"></a><span class="lineno"> 2709</span>&#160;</div><div class="line"><a name="l02710"></a><span class="lineno"> 2710</span>&#160; __global <span class="keywordtype">float</span> *src_addr_a = (__global <span class="keywordtype">float</span> *)(src0_ptr + src0_addr_in_bytes);</div><div class="line"><a name="l02711"></a><span class="lineno"> 2711</span>&#160; __global <span class="keywordtype">float</span> *src_addr_b = (__global <span class="keywordtype">float</span> *)(src1_ptr + src1_addr_in_bytes);</div><div class="line"><a name="l02712"></a><span class="lineno"> 2712</span>&#160;</div><div class="line"><a name="l02713"></a><span class="lineno"> 2713</span>&#160; src_addr_a += offset_row_a;</div><div class="line"><a name="l02714"></a><span class="lineno"> 2714</span>&#160; src_addr_b += offset_row_b;</div><div class="line"><a name="l02715"></a><span class="lineno"> 2715</span>&#160;</div><div class="line"><a name="l02716"></a><span class="lineno"> 2716</span>&#160; <span class="comment">// Reset accumulators</span></div><div class="line"><a name="l02717"></a><span class="lineno"> 2717</span>&#160; float4 c0 = 0.0f;</div><div class="line"><a name="l02718"></a><span class="lineno"> 2718</span>&#160; float4 c1 = 0.0f;</div><div class="line"><a name="l02719"></a><span class="lineno"> 2719</span>&#160; float4 c2 = 0.0f;</div><div class="line"><a name="l02720"></a><span class="lineno"> 2720</span>&#160; float4 c3 = 0.0f;</div><div class="line"><a name="l02721"></a><span class="lineno"> 2721</span>&#160;</div><div class="line"><a name="l02722"></a><span class="lineno"> 2722</span>&#160;<span class="preprocessor">#define COLS_MTX_B (COLS_B / (4 * MULT_TRANSPOSE1XW_WIDTH))</span></div><div class="line"><a name="l02723"></a><span class="lineno"> 2723</span>&#160;</div><div class="line"><a name="l02724"></a><span class="lineno"> 2724</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l02725"></a><span class="lineno"> 2725</span>&#160; <span class="keywordflow">for</span>(; i &lt;= (int)(COLS_MTX_B - 4); i += 4)</div><div class="line"><a name="l02726"></a><span class="lineno"> 2726</span>&#160; {</div><div class="line"><a name="l02727"></a><span class="lineno"> 2727</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02728"></a><span class="lineno"> 2728</span>&#160; float4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02729"></a><span class="lineno"> 2729</span>&#160; float4 b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02730"></a><span class="lineno"> 2730</span>&#160;</div><div class="line"><a name="l02731"></a><span class="lineno"> 2731</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02732"></a><span class="lineno"> 2732</span>&#160; src_addr_b += 4 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02733"></a><span class="lineno"> 2733</span>&#160;</div><div class="line"><a name="l02734"></a><span class="lineno"> 2734</span>&#160; c0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, c0.s0);</div><div class="line"><a name="l02735"></a><span class="lineno"> 2735</span>&#160; c0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, c0.s1);</div><div class="line"><a name="l02736"></a><span class="lineno"> 2736</span>&#160; c0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s2, c0.s2);</div><div class="line"><a name="l02737"></a><span class="lineno"> 2737</span>&#160; c0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s3, c0.s3);</div><div class="line"><a name="l02738"></a><span class="lineno"> 2738</span>&#160;</div><div class="line"><a name="l02739"></a><span class="lineno"> 2739</span>&#160; c1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s0, c1.s0);</div><div class="line"><a name="l02740"></a><span class="lineno"> 2740</span>&#160; c1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s1, c1.s1);</div><div class="line"><a name="l02741"></a><span class="lineno"> 2741</span>&#160; c1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s2, c1.s2);</div><div class="line"><a name="l02742"></a><span class="lineno"> 2742</span>&#160; c1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s3, c1.s3);</div><div class="line"><a name="l02743"></a><span class="lineno"> 2743</span>&#160;</div><div class="line"><a name="l02744"></a><span class="lineno"> 2744</span>&#160; c2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s0, c2.s0);</div><div class="line"><a name="l02745"></a><span class="lineno"> 2745</span>&#160; c2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s1, c2.s1);</div><div class="line"><a name="l02746"></a><span class="lineno"> 2746</span>&#160; c2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s2, c2.s2);</div><div class="line"><a name="l02747"></a><span class="lineno"> 2747</span>&#160; c2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s3, c2.s3);</div><div class="line"><a name="l02748"></a><span class="lineno"> 2748</span>&#160;</div><div class="line"><a name="l02749"></a><span class="lineno"> 2749</span>&#160; c3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s0, c3.s0);</div><div class="line"><a name="l02750"></a><span class="lineno"> 2750</span>&#160; c3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s1, c3.s1);</div><div class="line"><a name="l02751"></a><span class="lineno"> 2751</span>&#160; c3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s2, c3.s2);</div><div class="line"><a name="l02752"></a><span class="lineno"> 2752</span>&#160; c3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s3, c3.s3);</div><div class="line"><a name="l02753"></a><span class="lineno"> 2753</span>&#160;</div><div class="line"><a name="l02754"></a><span class="lineno"> 2754</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02755"></a><span class="lineno"> 2755</span>&#160; a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02756"></a><span class="lineno"> 2756</span>&#160; b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02757"></a><span class="lineno"> 2757</span>&#160;</div><div class="line"><a name="l02758"></a><span class="lineno"> 2758</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02759"></a><span class="lineno"> 2759</span>&#160; src_addr_b += 4 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02760"></a><span class="lineno"> 2760</span>&#160;</div><div class="line"><a name="l02761"></a><span class="lineno"> 2761</span>&#160; c0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, c0.s0);</div><div class="line"><a name="l02762"></a><span class="lineno"> 2762</span>&#160; c0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, c0.s1);</div><div class="line"><a name="l02763"></a><span class="lineno"> 2763</span>&#160; c0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s2, c0.s2);</div><div class="line"><a name="l02764"></a><span class="lineno"> 2764</span>&#160; c0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s3, c0.s3);</div><div class="line"><a name="l02765"></a><span class="lineno"> 2765</span>&#160;</div><div class="line"><a name="l02766"></a><span class="lineno"> 2766</span>&#160; c1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s0, c1.s0);</div><div class="line"><a name="l02767"></a><span class="lineno"> 2767</span>&#160; c1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s1, c1.s1);</div><div class="line"><a name="l02768"></a><span class="lineno"> 2768</span>&#160; c1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s2, c1.s2);</div><div class="line"><a name="l02769"></a><span class="lineno"> 2769</span>&#160; c1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s3, c1.s3);</div><div class="line"><a name="l02770"></a><span class="lineno"> 2770</span>&#160;</div><div class="line"><a name="l02771"></a><span class="lineno"> 2771</span>&#160; c2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s0, c2.s0);</div><div class="line"><a name="l02772"></a><span class="lineno"> 2772</span>&#160; c2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s1, c2.s1);</div><div class="line"><a name="l02773"></a><span class="lineno"> 2773</span>&#160; c2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s2, c2.s2);</div><div class="line"><a name="l02774"></a><span class="lineno"> 2774</span>&#160; c2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s3, c2.s3);</div><div class="line"><a name="l02775"></a><span class="lineno"> 2775</span>&#160;</div><div class="line"><a name="l02776"></a><span class="lineno"> 2776</span>&#160; c3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s0, c3.s0);</div><div class="line"><a name="l02777"></a><span class="lineno"> 2777</span>&#160; c3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s1, c3.s1);</div><div class="line"><a name="l02778"></a><span class="lineno"> 2778</span>&#160; c3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s2, c3.s2);</div><div class="line"><a name="l02779"></a><span class="lineno"> 2779</span>&#160; c3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s3, c3.s3);</div><div class="line"><a name="l02780"></a><span class="lineno"> 2780</span>&#160;</div><div class="line"><a name="l02781"></a><span class="lineno"> 2781</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02782"></a><span class="lineno"> 2782</span>&#160; a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02783"></a><span class="lineno"> 2783</span>&#160; b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02784"></a><span class="lineno"> 2784</span>&#160;</div><div class="line"><a name="l02785"></a><span class="lineno"> 2785</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02786"></a><span class="lineno"> 2786</span>&#160; src_addr_b += 4 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02787"></a><span class="lineno"> 2787</span>&#160;</div><div class="line"><a name="l02788"></a><span class="lineno"> 2788</span>&#160; c0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, c0.s0);</div><div class="line"><a name="l02789"></a><span class="lineno"> 2789</span>&#160; c0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, c0.s1);</div><div class="line"><a name="l02790"></a><span class="lineno"> 2790</span>&#160; c0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s2, c0.s2);</div><div class="line"><a name="l02791"></a><span class="lineno"> 2791</span>&#160; c0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s3, c0.s3);</div><div class="line"><a name="l02792"></a><span class="lineno"> 2792</span>&#160;</div><div class="line"><a name="l02793"></a><span class="lineno"> 2793</span>&#160; c1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s0, c1.s0);</div><div class="line"><a name="l02794"></a><span class="lineno"> 2794</span>&#160; c1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s1, c1.s1);</div><div class="line"><a name="l02795"></a><span class="lineno"> 2795</span>&#160; c1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s2, c1.s2);</div><div class="line"><a name="l02796"></a><span class="lineno"> 2796</span>&#160; c1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s3, c1.s3);</div><div class="line"><a name="l02797"></a><span class="lineno"> 2797</span>&#160;</div><div class="line"><a name="l02798"></a><span class="lineno"> 2798</span>&#160; c2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s0, c2.s0);</div><div class="line"><a name="l02799"></a><span class="lineno"> 2799</span>&#160; c2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s1, c2.s1);</div><div class="line"><a name="l02800"></a><span class="lineno"> 2800</span>&#160; c2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s2, c2.s2);</div><div class="line"><a name="l02801"></a><span class="lineno"> 2801</span>&#160; c2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s3, c2.s3);</div><div class="line"><a name="l02802"></a><span class="lineno"> 2802</span>&#160;</div><div class="line"><a name="l02803"></a><span class="lineno"> 2803</span>&#160; c3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s0, c3.s0);</div><div class="line"><a name="l02804"></a><span class="lineno"> 2804</span>&#160; c3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s1, c3.s1);</div><div class="line"><a name="l02805"></a><span class="lineno"> 2805</span>&#160; c3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s2, c3.s2);</div><div class="line"><a name="l02806"></a><span class="lineno"> 2806</span>&#160; c3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s3, c3.s3);</div><div class="line"><a name="l02807"></a><span class="lineno"> 2807</span>&#160;</div><div class="line"><a name="l02808"></a><span class="lineno"> 2808</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02809"></a><span class="lineno"> 2809</span>&#160; a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02810"></a><span class="lineno"> 2810</span>&#160; b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02811"></a><span class="lineno"> 2811</span>&#160;</div><div class="line"><a name="l02812"></a><span class="lineno"> 2812</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02813"></a><span class="lineno"> 2813</span>&#160; src_addr_b += 4 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02814"></a><span class="lineno"> 2814</span>&#160;</div><div class="line"><a name="l02815"></a><span class="lineno"> 2815</span>&#160; c0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, c0.s0);</div><div class="line"><a name="l02816"></a><span class="lineno"> 2816</span>&#160; c0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, c0.s1);</div><div class="line"><a name="l02817"></a><span class="lineno"> 2817</span>&#160; c0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s2, c0.s2);</div><div class="line"><a name="l02818"></a><span class="lineno"> 2818</span>&#160; c0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s3, c0.s3);</div><div class="line"><a name="l02819"></a><span class="lineno"> 2819</span>&#160;</div><div class="line"><a name="l02820"></a><span class="lineno"> 2820</span>&#160; c1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s0, c1.s0);</div><div class="line"><a name="l02821"></a><span class="lineno"> 2821</span>&#160; c1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s1, c1.s1);</div><div class="line"><a name="l02822"></a><span class="lineno"> 2822</span>&#160; c1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s2, c1.s2);</div><div class="line"><a name="l02823"></a><span class="lineno"> 2823</span>&#160; c1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s3, c1.s3);</div><div class="line"><a name="l02824"></a><span class="lineno"> 2824</span>&#160;</div><div class="line"><a name="l02825"></a><span class="lineno"> 2825</span>&#160; c2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s0, c2.s0);</div><div class="line"><a name="l02826"></a><span class="lineno"> 2826</span>&#160; c2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s1, c2.s1);</div><div class="line"><a name="l02827"></a><span class="lineno"> 2827</span>&#160; c2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s2, c2.s2);</div><div class="line"><a name="l02828"></a><span class="lineno"> 2828</span>&#160; c2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s3, c2.s3);</div><div class="line"><a name="l02829"></a><span class="lineno"> 2829</span>&#160;</div><div class="line"><a name="l02830"></a><span class="lineno"> 2830</span>&#160; c3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s0, c3.s0);</div><div class="line"><a name="l02831"></a><span class="lineno"> 2831</span>&#160; c3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s1, c3.s1);</div><div class="line"><a name="l02832"></a><span class="lineno"> 2832</span>&#160; c3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s2, c3.s2);</div><div class="line"><a name="l02833"></a><span class="lineno"> 2833</span>&#160; c3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s3, c3.s3);</div><div class="line"><a name="l02834"></a><span class="lineno"> 2834</span>&#160; }</div><div class="line"><a name="l02835"></a><span class="lineno"> 2835</span>&#160;</div><div class="line"><a name="l02836"></a><span class="lineno"> 2836</span>&#160; <span class="keywordflow">for</span>(; i &lt; (int)(COLS_MTX_B); ++i)</div><div class="line"><a name="l02837"></a><span class="lineno"> 2837</span>&#160; {</div><div class="line"><a name="l02838"></a><span class="lineno"> 2838</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l02839"></a><span class="lineno"> 2839</span>&#160; float4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l02840"></a><span class="lineno"> 2840</span>&#160; float4 b0 = vload4(0, src_addr_b);</div><div class="line"><a name="l02841"></a><span class="lineno"> 2841</span>&#160;</div><div class="line"><a name="l02842"></a><span class="lineno"> 2842</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l02843"></a><span class="lineno"> 2843</span>&#160; src_addr_b += 4 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l02844"></a><span class="lineno"> 2844</span>&#160;</div><div class="line"><a name="l02845"></a><span class="lineno"> 2845</span>&#160; c0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, c0.s0);</div><div class="line"><a name="l02846"></a><span class="lineno"> 2846</span>&#160; c0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, c0.s1);</div><div class="line"><a name="l02847"></a><span class="lineno"> 2847</span>&#160; c0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s2, c0.s2);</div><div class="line"><a name="l02848"></a><span class="lineno"> 2848</span>&#160; c0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s3, c0.s3);</div><div class="line"><a name="l02849"></a><span class="lineno"> 2849</span>&#160;</div><div class="line"><a name="l02850"></a><span class="lineno"> 2850</span>&#160; c1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s0, c1.s0);</div><div class="line"><a name="l02851"></a><span class="lineno"> 2851</span>&#160; c1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s1, c1.s1);</div><div class="line"><a name="l02852"></a><span class="lineno"> 2852</span>&#160; c1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s2, c1.s2);</div><div class="line"><a name="l02853"></a><span class="lineno"> 2853</span>&#160; c1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s3, c1.s3);</div><div class="line"><a name="l02854"></a><span class="lineno"> 2854</span>&#160;</div><div class="line"><a name="l02855"></a><span class="lineno"> 2855</span>&#160; c2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s0, c2.s0);</div><div class="line"><a name="l02856"></a><span class="lineno"> 2856</span>&#160; c2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s1, c2.s1);</div><div class="line"><a name="l02857"></a><span class="lineno"> 2857</span>&#160; c2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s2, c2.s2);</div><div class="line"><a name="l02858"></a><span class="lineno"> 2858</span>&#160; c2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s3, c2.s3);</div><div class="line"><a name="l02859"></a><span class="lineno"> 2859</span>&#160;</div><div class="line"><a name="l02860"></a><span class="lineno"> 2860</span>&#160; c3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s0, c3.s0);</div><div class="line"><a name="l02861"></a><span class="lineno"> 2861</span>&#160; c3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s1, c3.s1);</div><div class="line"><a name="l02862"></a><span class="lineno"> 2862</span>&#160; c3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s2, c3.s2);</div><div class="line"><a name="l02863"></a><span class="lineno"> 2863</span>&#160; c3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s3, c3.s3);</div><div class="line"><a name="l02864"></a><span class="lineno"> 2864</span>&#160; }</div><div class="line"><a name="l02865"></a><span class="lineno"> 2865</span>&#160;</div><div class="line"><a name="l02866"></a><span class="lineno"> 2866</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l02867"></a><span class="lineno"> 2867</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l02868"></a><span class="lineno"> 2868</span>&#160;</div><div class="line"><a name="l02869"></a><span class="lineno"> 2869</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l02870"></a><span class="lineno"> 2870</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l02871"></a><span class="lineno"> 2871</span>&#160;</div><div class="line"><a name="l02872"></a><span class="lineno"> 2872</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l02873"></a><span class="lineno"> 2873</span>&#160;</div><div class="line"><a name="l02874"></a><span class="lineno"> 2874</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02875"></a><span class="lineno"> 2875</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l02876"></a><span class="lineno"> 2876</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l02877"></a><span class="lineno"> 2877</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l02878"></a><span class="lineno"> 2878</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02879"></a><span class="lineno"> 2879</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l02880"></a><span class="lineno"> 2880</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02881"></a><span class="lineno"> 2881</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l02882"></a><span class="lineno"> 2882</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l02883"></a><span class="lineno"> 2883</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l02884"></a><span class="lineno"> 2884</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l02885"></a><span class="lineno"> 2885</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02886"></a><span class="lineno"> 2886</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l02887"></a><span class="lineno"> 2887</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l02888"></a><span class="lineno"> 2888</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l02889"></a><span class="lineno"> 2889</span>&#160;</div><div class="line"><a name="l02890"></a><span class="lineno"> 2890</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * 4) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l02891"></a><span class="lineno"> 2891</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * 4)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l02892"></a><span class="lineno"> 2892</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l02893"></a><span class="lineno"> 2893</span>&#160;</div><div class="line"><a name="l02894"></a><span class="lineno"> 2894</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l02895"></a><span class="lineno"> 2895</span>&#160; zout *= (cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l02896"></a><span class="lineno"> 2896</span>&#160;</div><div class="line"><a name="l02897"></a><span class="lineno"> 2897</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l02898"></a><span class="lineno"> 2898</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l02899"></a><span class="lineno"> 2899</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l02900"></a><span class="lineno"> 2900</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02901"></a><span class="lineno"> 2901</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l02902"></a><span class="lineno"> 2902</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l02903"></a><span class="lineno"> 2903</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l02904"></a><span class="lineno"> 2904</span>&#160;</div><div class="line"><a name="l02905"></a><span class="lineno"> 2905</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l02906"></a><span class="lineno"> 2906</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l02907"></a><span class="lineno"> 2907</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <span class="keywordtype">float</span>, c, ALPHA);</div><div class="line"><a name="l02908"></a><span class="lineno"> 2908</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l02909"></a><span class="lineno"> 2909</span>&#160;</div><div class="line"><a name="l02910"></a><span class="lineno"> 2910</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l02911"></a><span class="lineno"> 2911</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l02912"></a><span class="lineno"> 2912</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(4, uint, zero, 0);</div><div class="line"><a name="l02913"></a><span class="lineno"> 2913</span>&#160;</div><div class="line"><a name="l02914"></a><span class="lineno"> 2914</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02915"></a><span class="lineno"> 2915</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)4 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>));</div><div class="line"><a name="l02916"></a><span class="lineno"> 2916</span>&#160;</div><div class="line"><a name="l02917"></a><span class="lineno"> 2917</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l02918"></a><span class="lineno"> 2918</span>&#160;</div><div class="line"><a name="l02919"></a><span class="lineno"> 2919</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02920"></a><span class="lineno"> 2920</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02921"></a><span class="lineno"> 2921</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02922"></a><span class="lineno"> 2922</span>&#160;</div><div class="line"><a name="l02923"></a><span class="lineno"> 2923</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l02924"></a><span class="lineno"> 2924</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(4, c, bias0);</div><div class="line"><a name="l02925"></a><span class="lineno"> 2925</span>&#160;</div><div class="line"><a name="l02926"></a><span class="lineno"> 2926</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02927"></a><span class="lineno"> 2927</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)4 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>)) + (get_global_id(1) * (uint)4 * src2_stride_y) + get_global_id(</div><div class="line"><a name="l02928"></a><span class="lineno"> 2928</span>&#160; 2) * src2_stride_z;</div><div class="line"><a name="l02929"></a><span class="lineno"> 2929</span>&#160;</div><div class="line"><a name="l02930"></a><span class="lineno"> 2930</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(4, 4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l02931"></a><span class="lineno"> 2931</span>&#160;</div><div class="line"><a name="l02932"></a><span class="lineno"> 2932</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l02933"></a><span class="lineno"> 2933</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l02934"></a><span class="lineno"> 2934</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l02935"></a><span class="lineno"> 2935</span>&#160;</div><div class="line"><a name="l02936"></a><span class="lineno"> 2936</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l02937"></a><span class="lineno"> 2937</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(4, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l02938"></a><span class="lineno"> 2938</span>&#160;</div><div class="line"><a name="l02939"></a><span class="lineno"> 2939</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l02940"></a><span class="lineno"> 2940</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l02941"></a><span class="lineno"> 2941</span>&#160;</div><div class="line"><a name="l02942"></a><span class="lineno"> 2942</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02943"></a><span class="lineno"> 2943</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(4, ACTIVATION_TYPE, <span class="keywordtype">float</span>, c, A_VAL, B_VAL);</div><div class="line"><a name="l02944"></a><span class="lineno"> 2944</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l02945"></a><span class="lineno"> 2945</span>&#160;</div><div class="line"><a name="l02946"></a><span class="lineno"> 2946</span>&#160; <span class="comment">// Store 4x4 block</span></div><div class="line"><a name="l02947"></a><span class="lineno"> 2947</span>&#160; vstore4(c0, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l02948"></a><span class="lineno"> 2948</span>&#160; vstore4(c1, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l02949"></a><span class="lineno"> 2949</span>&#160; vstore4(c2, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l02950"></a><span class="lineno"> 2950</span>&#160; vstore4(c3, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l02951"></a><span class="lineno"> 2951</span>&#160;}</div><div class="line"><a name="l02952"></a><span class="lineno"> 2952</span>&#160;</div><div class="line"><a name="l02953"></a><span class="lineno"> 2953</span>&#160;<span class="comment">// Undefine local defines</span></div><div class="line"><a name="l02954"></a><span class="lineno"> 2954</span>&#160;<span class="preprocessor">#undef COLS_MTX_B</span></div><div class="line"><a name="l02955"></a><span class="lineno"> 2955</span>&#160;</div><div class="line"><a name="l02956"></a><span class="lineno"> 2956</span>&#160;<span class="preprocessor">#if defined(ARM_COMPUTE_OPENCL_FP16_ENABLED)</span></div><div class="line"><a name="l02957"></a><span class="lineno"> 2957</span>&#160;</div><div class="line"><a name="l03003"></a><span class="lineno"> 3003</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_interleaved_transposed_f16(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l03004"></a><span class="lineno"> 3004</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l03005"></a><span class="lineno"> 3005</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03006"></a><span class="lineno"> 3006</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l03007"></a><span class="lineno"> 3007</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l03008"></a><span class="lineno"> 3008</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l03009"></a><span class="lineno"> 3009</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l03010"></a><span class="lineno"> 3010</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l03011"></a><span class="lineno"> 3011</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03012"></a><span class="lineno"> 3012</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l03013"></a><span class="lineno"> 3013</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l03014"></a><span class="lineno"> 3014</span>&#160; uint dst_stride_z</div><div class="line"><a name="l03015"></a><span class="lineno"> 3015</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l03016"></a><span class="lineno"> 3016</span>&#160; ,</div><div class="line"><a name="l03017"></a><span class="lineno"> 3017</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l03018"></a><span class="lineno"> 3018</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l03019"></a><span class="lineno"> 3019</span>&#160; )</div><div class="line"><a name="l03020"></a><span class="lineno"> 3020</span>&#160;{</div><div class="line"><a name="l03021"></a><span class="lineno"> 3021</span>&#160; <span class="keywordtype">int</span> x = get_global_id(0) / MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03022"></a><span class="lineno"> 3022</span>&#160; <span class="keywordtype">int</span> y = get_global_id(1) / MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03023"></a><span class="lineno"> 3023</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l03024"></a><span class="lineno"> 3024</span>&#160;</div><div class="line"><a name="l03025"></a><span class="lineno"> 3025</span>&#160; <span class="comment">// Offset</span></div><div class="line"><a name="l03026"></a><span class="lineno"> 3026</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_a = (get_global_id(1) % MULT_INTERLEAVE4X4_HEIGHT) * 4;</div><div class="line"><a name="l03027"></a><span class="lineno"> 3027</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_b = (get_global_id(0) % MULT_TRANSPOSE1XW_WIDTH) * 8;</div><div class="line"><a name="l03028"></a><span class="lineno"> 3028</span>&#160;</div><div class="line"><a name="l03029"></a><span class="lineno"> 3029</span>&#160; <span class="comment">// src_addr_a = address of matrix A</span></div><div class="line"><a name="l03030"></a><span class="lineno"> 3030</span>&#160; <span class="comment">// src_addr_b = address of matrix B</span></div><div class="line"><a name="l03031"></a><span class="lineno"> 3031</span>&#160; <span class="keywordtype">int</span> src0_addr_in_bytes = z * src0_stride_z + y * src0_stride_y + src0_offset_first_element_in_bytes;</div><div class="line"><a name="l03032"></a><span class="lineno"> 3032</span>&#160; <span class="keywordtype">int</span> src1_addr_in_bytes = x * src1_stride_y + src1_offset_first_element_in_bytes;</div><div class="line"><a name="l03033"></a><span class="lineno"> 3033</span>&#160;</div><div class="line"><a name="l03034"></a><span class="lineno"> 3034</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03035"></a><span class="lineno"> 3035</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l03036"></a><span class="lineno"> 3036</span>&#160; src1_addr_in_bytes += (z % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l03037"></a><span class="lineno"> 3037</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03038"></a><span class="lineno"> 3038</span>&#160; src1_addr_in_bytes += z * src1_stride_z;</div><div class="line"><a name="l03039"></a><span class="lineno"> 3039</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03040"></a><span class="lineno"> 3040</span>&#160;</div><div class="line"><a name="l03041"></a><span class="lineno"> 3041</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_addr_a = (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src0_addr_in_bytes);</div><div class="line"><a name="l03042"></a><span class="lineno"> 3042</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_addr_b = (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src1_addr_in_bytes);</div><div class="line"><a name="l03043"></a><span class="lineno"> 3043</span>&#160;</div><div class="line"><a name="l03044"></a><span class="lineno"> 3044</span>&#160; <span class="comment">// Compute end row address for matrix B</span></div><div class="line"><a name="l03045"></a><span class="lineno"> 3045</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_end_addr_b = src_addr_b + COLS_B;</div><div class="line"><a name="l03046"></a><span class="lineno"> 3046</span>&#160;</div><div class="line"><a name="l03047"></a><span class="lineno"> 3047</span>&#160; src_addr_a += offset_row_a;</div><div class="line"><a name="l03048"></a><span class="lineno"> 3048</span>&#160; src_addr_b += offset_row_b;</div><div class="line"><a name="l03049"></a><span class="lineno"> 3049</span>&#160;</div><div class="line"><a name="l03050"></a><span class="lineno"> 3050</span>&#160; <span class="comment">// Reset accumulators</span></div><div class="line"><a name="l03051"></a><span class="lineno"> 3051</span>&#160; half8 c0 = 0.0f;</div><div class="line"><a name="l03052"></a><span class="lineno"> 3052</span>&#160; half8 c1 = 0.0f;</div><div class="line"><a name="l03053"></a><span class="lineno"> 3053</span>&#160; half8 c2 = 0.0f;</div><div class="line"><a name="l03054"></a><span class="lineno"> 3054</span>&#160; half8 c3 = 0.0f;</div><div class="line"><a name="l03055"></a><span class="lineno"> 3055</span>&#160;</div><div class="line"><a name="l03056"></a><span class="lineno"> 3056</span>&#160; <span class="keywordflow">for</span>(; src_addr_b &lt;= (src_end_addr_b - (int)(16 * MULT_TRANSPOSE1XW_WIDTH)); src_addr_a += 8 * MULT_INTERLEAVE4X4_HEIGHT, src_addr_b += 16 * MULT_TRANSPOSE1XW_WIDTH)</div><div class="line"><a name="l03057"></a><span class="lineno"> 3057</span>&#160; {</div><div class="line"><a name="l03058"></a><span class="lineno"> 3058</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03059"></a><span class="lineno"> 3059</span>&#160; half4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03060"></a><span class="lineno"> 3060</span>&#160; half8 b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03061"></a><span class="lineno"> 3061</span>&#160;</div><div class="line"><a name="l03062"></a><span class="lineno"> 3062</span>&#160; c0 += (half8)a0.s0 * b0;</div><div class="line"><a name="l03063"></a><span class="lineno"> 3063</span>&#160; c1 += (half8)a0.s1 * b0;</div><div class="line"><a name="l03064"></a><span class="lineno"> 3064</span>&#160; c2 += (half8)a0.s2 * b0;</div><div class="line"><a name="l03065"></a><span class="lineno"> 3065</span>&#160; c3 += (half8)a0.s3 * b0;</div><div class="line"><a name="l03066"></a><span class="lineno"> 3066</span>&#160;</div><div class="line"><a name="l03067"></a><span class="lineno"> 3067</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03068"></a><span class="lineno"> 3068</span>&#160; a0 = vload4(0, src_addr_a + 4 * MULT_INTERLEAVE4X4_HEIGHT);</div><div class="line"><a name="l03069"></a><span class="lineno"> 3069</span>&#160; b0 = vload8(0, src_addr_b + 8 * MULT_TRANSPOSE1XW_WIDTH);</div><div class="line"><a name="l03070"></a><span class="lineno"> 3070</span>&#160;</div><div class="line"><a name="l03071"></a><span class="lineno"> 3071</span>&#160; c0 += (half8)a0.s0 * b0;</div><div class="line"><a name="l03072"></a><span class="lineno"> 3072</span>&#160; c1 += (half8)a0.s1 * b0;</div><div class="line"><a name="l03073"></a><span class="lineno"> 3073</span>&#160; c2 += (half8)a0.s2 * b0;</div><div class="line"><a name="l03074"></a><span class="lineno"> 3074</span>&#160; c3 += (half8)a0.s3 * b0;</div><div class="line"><a name="l03075"></a><span class="lineno"> 3075</span>&#160; }</div><div class="line"><a name="l03076"></a><span class="lineno"> 3076</span>&#160;</div><div class="line"><a name="l03077"></a><span class="lineno"> 3077</span>&#160; <span class="keywordflow">for</span>(; src_addr_b &lt; src_end_addr_b; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT, src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH)</div><div class="line"><a name="l03078"></a><span class="lineno"> 3078</span>&#160; {</div><div class="line"><a name="l03079"></a><span class="lineno"> 3079</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03080"></a><span class="lineno"> 3080</span>&#160; half4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03081"></a><span class="lineno"> 3081</span>&#160; half8 b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03082"></a><span class="lineno"> 3082</span>&#160;</div><div class="line"><a name="l03083"></a><span class="lineno"> 3083</span>&#160; c0 += (half8)a0.s0 * b0;</div><div class="line"><a name="l03084"></a><span class="lineno"> 3084</span>&#160; c1 += (half8)a0.s1 * b0;</div><div class="line"><a name="l03085"></a><span class="lineno"> 3085</span>&#160; c2 += (half8)a0.s2 * b0;</div><div class="line"><a name="l03086"></a><span class="lineno"> 3086</span>&#160; c3 += (half8)a0.s3 * b0;</div><div class="line"><a name="l03087"></a><span class="lineno"> 3087</span>&#160; }</div><div class="line"><a name="l03088"></a><span class="lineno"> 3088</span>&#160;</div><div class="line"><a name="l03089"></a><span class="lineno"> 3089</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l03090"></a><span class="lineno"> 3090</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l03091"></a><span class="lineno"> 3091</span>&#160;</div><div class="line"><a name="l03092"></a><span class="lineno"> 3092</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l03093"></a><span class="lineno"> 3093</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l03094"></a><span class="lineno"> 3094</span>&#160;</div><div class="line"><a name="l03095"></a><span class="lineno"> 3095</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l03096"></a><span class="lineno"> 3096</span>&#160;</div><div class="line"><a name="l03097"></a><span class="lineno"> 3097</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03098"></a><span class="lineno"> 3098</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l03099"></a><span class="lineno"> 3099</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l03100"></a><span class="lineno"> 3100</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l03101"></a><span class="lineno"> 3101</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03102"></a><span class="lineno"> 3102</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l03103"></a><span class="lineno"> 3103</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03104"></a><span class="lineno"> 3104</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03105"></a><span class="lineno"> 3105</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03106"></a><span class="lineno"> 3106</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l03107"></a><span class="lineno"> 3107</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03108"></a><span class="lineno"> 3108</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03109"></a><span class="lineno"> 3109</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l03110"></a><span class="lineno"> 3110</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03111"></a><span class="lineno"> 3111</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03112"></a><span class="lineno"> 3112</span>&#160;</div><div class="line"><a name="l03113"></a><span class="lineno"> 3113</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * 4) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l03114"></a><span class="lineno"> 3114</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * 4)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l03115"></a><span class="lineno"> 3115</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l03116"></a><span class="lineno"> 3116</span>&#160;</div><div class="line"><a name="l03117"></a><span class="lineno"> 3117</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l03118"></a><span class="lineno"> 3118</span>&#160; zout *= (cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l03119"></a><span class="lineno"> 3119</span>&#160;</div><div class="line"><a name="l03120"></a><span class="lineno"> 3120</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l03121"></a><span class="lineno"> 3121</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l03122"></a><span class="lineno"> 3122</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l03123"></a><span class="lineno"> 3123</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03124"></a><span class="lineno"> 3124</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l03125"></a><span class="lineno"> 3125</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l03126"></a><span class="lineno"> 3126</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03127"></a><span class="lineno"> 3127</span>&#160;</div><div class="line"><a name="l03128"></a><span class="lineno"> 3128</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l03129"></a><span class="lineno"> 3129</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l03130"></a><span class="lineno"> 3130</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, c, ALPHA);</div><div class="line"><a name="l03131"></a><span class="lineno"> 3131</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l03132"></a><span class="lineno"> 3132</span>&#160;</div><div class="line"><a name="l03133"></a><span class="lineno"> 3133</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l03134"></a><span class="lineno"> 3134</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l03135"></a><span class="lineno"> 3135</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(4, uint, zero, 0);</div><div class="line"><a name="l03136"></a><span class="lineno"> 3136</span>&#160;</div><div class="line"><a name="l03137"></a><span class="lineno"> 3137</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03138"></a><span class="lineno"> 3138</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>));</div><div class="line"><a name="l03139"></a><span class="lineno"> 3139</span>&#160;</div><div class="line"><a name="l03140"></a><span class="lineno"> 3140</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03141"></a><span class="lineno"> 3141</span>&#160;</div><div class="line"><a name="l03142"></a><span class="lineno"> 3142</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03143"></a><span class="lineno"> 3143</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l03144"></a><span class="lineno"> 3144</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l03145"></a><span class="lineno"> 3145</span>&#160;</div><div class="line"><a name="l03146"></a><span class="lineno"> 3146</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l03147"></a><span class="lineno"> 3147</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(4, c, bias0);</div><div class="line"><a name="l03148"></a><span class="lineno"> 3148</span>&#160;</div><div class="line"><a name="l03149"></a><span class="lineno"> 3149</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03150"></a><span class="lineno"> 3150</span>&#160;</div><div class="line"><a name="l03151"></a><span class="lineno"> 3151</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>)) + (get_global_id(1) * (uint)4 * src2_stride_y) + get_global_id(</div><div class="line"><a name="l03152"></a><span class="lineno"> 3152</span>&#160; 2) * src2_stride_z;</div><div class="line"><a name="l03153"></a><span class="lineno"> 3153</span>&#160;</div><div class="line"><a name="l03154"></a><span class="lineno"> 3154</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(4, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03155"></a><span class="lineno"> 3155</span>&#160;</div><div class="line"><a name="l03156"></a><span class="lineno"> 3156</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03157"></a><span class="lineno"> 3157</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l03158"></a><span class="lineno"> 3158</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l03159"></a><span class="lineno"> 3159</span>&#160;</div><div class="line"><a name="l03160"></a><span class="lineno"> 3160</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l03161"></a><span class="lineno"> 3161</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(4, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l03162"></a><span class="lineno"> 3162</span>&#160;</div><div class="line"><a name="l03163"></a><span class="lineno"> 3163</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03164"></a><span class="lineno"> 3164</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l03165"></a><span class="lineno"> 3165</span>&#160;</div><div class="line"><a name="l03166"></a><span class="lineno"> 3166</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l03167"></a><span class="lineno"> 3167</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(4, ACTIVATION_TYPE, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, c, A_VAL, B_VAL);</div><div class="line"><a name="l03168"></a><span class="lineno"> 3168</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l03169"></a><span class="lineno"> 3169</span>&#160;</div><div class="line"><a name="l03170"></a><span class="lineno"> 3170</span>&#160; <span class="comment">// Store 4x8 block</span></div><div class="line"><a name="l03171"></a><span class="lineno"> 3171</span>&#160; vstore8(c0, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l03172"></a><span class="lineno"> 3172</span>&#160; vstore8(c1, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l03173"></a><span class="lineno"> 3173</span>&#160; vstore8(c2, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l03174"></a><span class="lineno"> 3174</span>&#160; vstore8(c3, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l03175"></a><span class="lineno"> 3175</span>&#160;}</div><div class="line"><a name="l03176"></a><span class="lineno"> 3176</span>&#160;</div><div class="line"><a name="l03223"></a><span class="lineno"> 3223</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_interleaved_transposed_f16_acc32(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l03224"></a><span class="lineno"> 3224</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l03225"></a><span class="lineno"> 3225</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03226"></a><span class="lineno"> 3226</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l03227"></a><span class="lineno"> 3227</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l03228"></a><span class="lineno"> 3228</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l03229"></a><span class="lineno"> 3229</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l03230"></a><span class="lineno"> 3230</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l03231"></a><span class="lineno"> 3231</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03232"></a><span class="lineno"> 3232</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l03233"></a><span class="lineno"> 3233</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l03234"></a><span class="lineno"> 3234</span>&#160; uint dst_stride_z</div><div class="line"><a name="l03235"></a><span class="lineno"> 3235</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l03236"></a><span class="lineno"> 3236</span>&#160; ,</div><div class="line"><a name="l03237"></a><span class="lineno"> 3237</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l03238"></a><span class="lineno"> 3238</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l03239"></a><span class="lineno"> 3239</span>&#160; )</div><div class="line"><a name="l03240"></a><span class="lineno"> 3240</span>&#160;{</div><div class="line"><a name="l03241"></a><span class="lineno"> 3241</span>&#160; <span class="keywordtype">int</span> x = get_global_id(0) / MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03242"></a><span class="lineno"> 3242</span>&#160; <span class="keywordtype">int</span> y = get_global_id(1) / MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03243"></a><span class="lineno"> 3243</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l03244"></a><span class="lineno"> 3244</span>&#160;</div><div class="line"><a name="l03245"></a><span class="lineno"> 3245</span>&#160; <span class="comment">// Offset</span></div><div class="line"><a name="l03246"></a><span class="lineno"> 3246</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_a = (get_global_id(1) % MULT_INTERLEAVE4X4_HEIGHT) * 4;</div><div class="line"><a name="l03247"></a><span class="lineno"> 3247</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_b = (get_global_id(0) % MULT_TRANSPOSE1XW_WIDTH) * 8;</div><div class="line"><a name="l03248"></a><span class="lineno"> 3248</span>&#160;</div><div class="line"><a name="l03249"></a><span class="lineno"> 3249</span>&#160; <span class="comment">// src_addr_a = address of matrix A</span></div><div class="line"><a name="l03250"></a><span class="lineno"> 3250</span>&#160; <span class="comment">// src_addr_b = address of matrix B</span></div><div class="line"><a name="l03251"></a><span class="lineno"> 3251</span>&#160; <span class="keywordtype">int</span> src0_addr_in_bytes = z * src0_stride_z + y * src0_stride_y + src0_offset_first_element_in_bytes;</div><div class="line"><a name="l03252"></a><span class="lineno"> 3252</span>&#160; <span class="keywordtype">int</span> src1_addr_in_bytes = x * src1_stride_y + src1_offset_first_element_in_bytes;</div><div class="line"><a name="l03253"></a><span class="lineno"> 3253</span>&#160;</div><div class="line"><a name="l03254"></a><span class="lineno"> 3254</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03255"></a><span class="lineno"> 3255</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l03256"></a><span class="lineno"> 3256</span>&#160; src1_addr_in_bytes += (z % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l03257"></a><span class="lineno"> 3257</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03258"></a><span class="lineno"> 3258</span>&#160; src1_addr_in_bytes += z * src1_stride_z;</div><div class="line"><a name="l03259"></a><span class="lineno"> 3259</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03260"></a><span class="lineno"> 3260</span>&#160;</div><div class="line"><a name="l03261"></a><span class="lineno"> 3261</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_addr_a = (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src0_addr_in_bytes);</div><div class="line"><a name="l03262"></a><span class="lineno"> 3262</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_addr_b = (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src1_addr_in_bytes);</div><div class="line"><a name="l03263"></a><span class="lineno"> 3263</span>&#160;</div><div class="line"><a name="l03264"></a><span class="lineno"> 3264</span>&#160; <span class="comment">// Compute end row address for matrix B</span></div><div class="line"><a name="l03265"></a><span class="lineno"> 3265</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_end_addr_b = src_addr_b + COLS_B;</div><div class="line"><a name="l03266"></a><span class="lineno"> 3266</span>&#160;</div><div class="line"><a name="l03267"></a><span class="lineno"> 3267</span>&#160; src_addr_a += offset_row_a;</div><div class="line"><a name="l03268"></a><span class="lineno"> 3268</span>&#160; src_addr_b += offset_row_b;</div><div class="line"><a name="l03269"></a><span class="lineno"> 3269</span>&#160;</div><div class="line"><a name="l03270"></a><span class="lineno"> 3270</span>&#160; <span class="comment">// Reset accumulators</span></div><div class="line"><a name="l03271"></a><span class="lineno"> 3271</span>&#160; float8 c0 = 0.0f;</div><div class="line"><a name="l03272"></a><span class="lineno"> 3272</span>&#160; float8 c1 = 0.0f;</div><div class="line"><a name="l03273"></a><span class="lineno"> 3273</span>&#160; float8 c2 = 0.0f;</div><div class="line"><a name="l03274"></a><span class="lineno"> 3274</span>&#160; float8 c3 = 0.0f;</div><div class="line"><a name="l03275"></a><span class="lineno"> 3275</span>&#160;</div><div class="line"><a name="l03276"></a><span class="lineno"> 3276</span>&#160; <span class="keywordflow">for</span>(; src_addr_b &lt;= (src_end_addr_b - (int)(16 * MULT_TRANSPOSE1XW_WIDTH)); src_addr_a += 8 * MULT_INTERLEAVE4X4_HEIGHT, src_addr_b += 16 * MULT_TRANSPOSE1XW_WIDTH)</div><div class="line"><a name="l03277"></a><span class="lineno"> 3277</span>&#160; {</div><div class="line"><a name="l03278"></a><span class="lineno"> 3278</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03279"></a><span class="lineno"> 3279</span>&#160; float4 a0 = convert_float4(vload4(0, src_addr_a));</div><div class="line"><a name="l03280"></a><span class="lineno"> 3280</span>&#160; float8 b0 = convert_float8(vload8(0, src_addr_b));</div><div class="line"><a name="l03281"></a><span class="lineno"> 3281</span>&#160;</div><div class="line"><a name="l03282"></a><span class="lineno"> 3282</span>&#160; c0 += (float8)a0.s0 * b0;</div><div class="line"><a name="l03283"></a><span class="lineno"> 3283</span>&#160; c1 += (float8)a0.s1 * b0;</div><div class="line"><a name="l03284"></a><span class="lineno"> 3284</span>&#160; c2 += (float8)a0.s2 * b0;</div><div class="line"><a name="l03285"></a><span class="lineno"> 3285</span>&#160; c3 += (float8)a0.s3 * b0;</div><div class="line"><a name="l03286"></a><span class="lineno"> 3286</span>&#160;</div><div class="line"><a name="l03287"></a><span class="lineno"> 3287</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03288"></a><span class="lineno"> 3288</span>&#160; a0 = convert_float4(vload4(0, src_addr_a + 4 * MULT_INTERLEAVE4X4_HEIGHT));</div><div class="line"><a name="l03289"></a><span class="lineno"> 3289</span>&#160; b0 = convert_float8(vload8(0, src_addr_b + 8 * MULT_TRANSPOSE1XW_WIDTH));</div><div class="line"><a name="l03290"></a><span class="lineno"> 3290</span>&#160;</div><div class="line"><a name="l03291"></a><span class="lineno"> 3291</span>&#160; c0 += (float8)a0.s0 * b0;</div><div class="line"><a name="l03292"></a><span class="lineno"> 3292</span>&#160; c1 += (float8)a0.s1 * b0;</div><div class="line"><a name="l03293"></a><span class="lineno"> 3293</span>&#160; c2 += (float8)a0.s2 * b0;</div><div class="line"><a name="l03294"></a><span class="lineno"> 3294</span>&#160; c3 += (float8)a0.s3 * b0;</div><div class="line"><a name="l03295"></a><span class="lineno"> 3295</span>&#160; }</div><div class="line"><a name="l03296"></a><span class="lineno"> 3296</span>&#160;</div><div class="line"><a name="l03297"></a><span class="lineno"> 3297</span>&#160; <span class="keywordflow">for</span>(; src_addr_b &lt; src_end_addr_b; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT, src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH)</div><div class="line"><a name="l03298"></a><span class="lineno"> 3298</span>&#160; {</div><div class="line"><a name="l03299"></a><span class="lineno"> 3299</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03300"></a><span class="lineno"> 3300</span>&#160; float4 a0 = convert_float4(vload4(0, src_addr_a));</div><div class="line"><a name="l03301"></a><span class="lineno"> 3301</span>&#160; float8 b0 = convert_float8(vload8(0, src_addr_b));</div><div class="line"><a name="l03302"></a><span class="lineno"> 3302</span>&#160;</div><div class="line"><a name="l03303"></a><span class="lineno"> 3303</span>&#160; c0 += (float8)a0.s0 * b0;</div><div class="line"><a name="l03304"></a><span class="lineno"> 3304</span>&#160; c1 += (float8)a0.s1 * b0;</div><div class="line"><a name="l03305"></a><span class="lineno"> 3305</span>&#160; c2 += (float8)a0.s2 * b0;</div><div class="line"><a name="l03306"></a><span class="lineno"> 3306</span>&#160; c3 += (float8)a0.s3 * b0;</div><div class="line"><a name="l03307"></a><span class="lineno"> 3307</span>&#160; }</div><div class="line"><a name="l03308"></a><span class="lineno"> 3308</span>&#160;</div><div class="line"><a name="l03309"></a><span class="lineno"> 3309</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l03310"></a><span class="lineno"> 3310</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l03311"></a><span class="lineno"> 3311</span>&#160;</div><div class="line"><a name="l03312"></a><span class="lineno"> 3312</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l03313"></a><span class="lineno"> 3313</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l03314"></a><span class="lineno"> 3314</span>&#160;</div><div class="line"><a name="l03315"></a><span class="lineno"> 3315</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l03316"></a><span class="lineno"> 3316</span>&#160;</div><div class="line"><a name="l03317"></a><span class="lineno"> 3317</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03318"></a><span class="lineno"> 3318</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l03319"></a><span class="lineno"> 3319</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l03320"></a><span class="lineno"> 3320</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l03321"></a><span class="lineno"> 3321</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03322"></a><span class="lineno"> 3322</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l03323"></a><span class="lineno"> 3323</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03324"></a><span class="lineno"> 3324</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03325"></a><span class="lineno"> 3325</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03326"></a><span class="lineno"> 3326</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l03327"></a><span class="lineno"> 3327</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03328"></a><span class="lineno"> 3328</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03329"></a><span class="lineno"> 3329</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l03330"></a><span class="lineno"> 3330</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03331"></a><span class="lineno"> 3331</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03332"></a><span class="lineno"> 3332</span>&#160;</div><div class="line"><a name="l03333"></a><span class="lineno"> 3333</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * 4) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l03334"></a><span class="lineno"> 3334</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * 4)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l03335"></a><span class="lineno"> 3335</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l03336"></a><span class="lineno"> 3336</span>&#160;</div><div class="line"><a name="l03337"></a><span class="lineno"> 3337</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l03338"></a><span class="lineno"> 3338</span>&#160; zout *= (cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l03339"></a><span class="lineno"> 3339</span>&#160;</div><div class="line"><a name="l03340"></a><span class="lineno"> 3340</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l03341"></a><span class="lineno"> 3341</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l03342"></a><span class="lineno"> 3342</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l03343"></a><span class="lineno"> 3343</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03344"></a><span class="lineno"> 3344</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l03345"></a><span class="lineno"> 3345</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l03346"></a><span class="lineno"> 3346</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03347"></a><span class="lineno"> 3347</span>&#160;</div><div class="line"><a name="l03348"></a><span class="lineno"> 3348</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l03349"></a><span class="lineno"> 3349</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l03350"></a><span class="lineno"> 3350</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <span class="keywordtype">float</span>, c, ALPHA);</div><div class="line"><a name="l03351"></a><span class="lineno"> 3351</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l03352"></a><span class="lineno"> 3352</span>&#160;</div><div class="line"><a name="l03353"></a><span class="lineno"> 3353</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l03354"></a><span class="lineno"> 3354</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(4, uint, zero, 0);</div><div class="line"><a name="l03355"></a><span class="lineno"> 3355</span>&#160;</div><div class="line"><a name="l03356"></a><span class="lineno"> 3356</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03357"></a><span class="lineno"> 3357</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>));</div><div class="line"><a name="l03358"></a><span class="lineno"> 3358</span>&#160;</div><div class="line"><a name="l03359"></a><span class="lineno"> 3359</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03360"></a><span class="lineno"> 3360</span>&#160;</div><div class="line"><a name="l03361"></a><span class="lineno"> 3361</span>&#160; float8 bias_f0 = convert_float8(bias0);</div><div class="line"><a name="l03362"></a><span class="lineno"> 3362</span>&#160;</div><div class="line"><a name="l03363"></a><span class="lineno"> 3363</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03364"></a><span class="lineno"> 3364</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <span class="keywordtype">float</span>, bias_f, BETA);</div><div class="line"><a name="l03365"></a><span class="lineno"> 3365</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l03366"></a><span class="lineno"> 3366</span>&#160;</div><div class="line"><a name="l03367"></a><span class="lineno"> 3367</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l03368"></a><span class="lineno"> 3368</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(4, c, bias_f0);</div><div class="line"><a name="l03369"></a><span class="lineno"> 3369</span>&#160;</div><div class="line"><a name="l03370"></a><span class="lineno"> 3370</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03371"></a><span class="lineno"> 3371</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>)) + (get_global_id(1) * (uint)4 * src2_stride_y) + get_global_id(</div><div class="line"><a name="l03372"></a><span class="lineno"> 3372</span>&#160; 2) * src2_stride_z;</div><div class="line"><a name="l03373"></a><span class="lineno"> 3373</span>&#160;</div><div class="line"><a name="l03374"></a><span class="lineno"> 3374</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(4, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03375"></a><span class="lineno"> 3375</span>&#160;</div><div class="line"><a name="l03376"></a><span class="lineno"> 3376</span>&#160; float8 bias_f0 = convert_float8(bias0);</div><div class="line"><a name="l03377"></a><span class="lineno"> 3377</span>&#160; float8 bias_f1 = convert_float8(bias1);</div><div class="line"><a name="l03378"></a><span class="lineno"> 3378</span>&#160; float8 bias_f2 = convert_float8(bias2);</div><div class="line"><a name="l03379"></a><span class="lineno"> 3379</span>&#160; float8 bias_f3 = convert_float8(bias3);</div><div class="line"><a name="l03380"></a><span class="lineno"> 3380</span>&#160;</div><div class="line"><a name="l03381"></a><span class="lineno"> 3381</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03382"></a><span class="lineno"> 3382</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <span class="keywordtype">float</span>, bias_f, BETA);</div><div class="line"><a name="l03383"></a><span class="lineno"> 3383</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l03384"></a><span class="lineno"> 3384</span>&#160;</div><div class="line"><a name="l03385"></a><span class="lineno"> 3385</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l03386"></a><span class="lineno"> 3386</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(4, c, bias_f);</div><div class="line"><a name="l03387"></a><span class="lineno"> 3387</span>&#160;</div><div class="line"><a name="l03388"></a><span class="lineno"> 3388</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03389"></a><span class="lineno"> 3389</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l03390"></a><span class="lineno"> 3390</span>&#160;</div><div class="line"><a name="l03391"></a><span class="lineno"> 3391</span>&#160; half8 c_h0 = convert_half8(c0);</div><div class="line"><a name="l03392"></a><span class="lineno"> 3392</span>&#160; half8 c_h1 = convert_half8(c1);</div><div class="line"><a name="l03393"></a><span class="lineno"> 3393</span>&#160; half8 c_h2 = convert_half8(c2);</div><div class="line"><a name="l03394"></a><span class="lineno"> 3394</span>&#160; half8 c_h3 = convert_half8(c3);</div><div class="line"><a name="l03395"></a><span class="lineno"> 3395</span>&#160;</div><div class="line"><a name="l03396"></a><span class="lineno"> 3396</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l03397"></a><span class="lineno"> 3397</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(4, ACTIVATION_TYPE, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, c_h, A_VAL, B_VAL);</div><div class="line"><a name="l03398"></a><span class="lineno"> 3398</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l03399"></a><span class="lineno"> 3399</span>&#160;</div><div class="line"><a name="l03400"></a><span class="lineno"> 3400</span>&#160; <span class="comment">// Store 4x8 block</span></div><div class="line"><a name="l03401"></a><span class="lineno"> 3401</span>&#160; vstore8(c_h0, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l03402"></a><span class="lineno"> 3402</span>&#160; vstore8(c_h1, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l03403"></a><span class="lineno"> 3403</span>&#160; vstore8(c_h2, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l03404"></a><span class="lineno"> 3404</span>&#160; vstore8(c_h3, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l03405"></a><span class="lineno"> 3405</span>&#160;}</div><div class="line"><a name="l03406"></a><span class="lineno"> 3406</span>&#160;</div><div class="line"><a name="l03452"></a><span class="lineno"> 3452</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_interleaved_transposed_f16_bifrost(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l03453"></a><span class="lineno"> 3453</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l03454"></a><span class="lineno"> 3454</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03455"></a><span class="lineno"> 3455</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l03456"></a><span class="lineno"> 3456</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l03457"></a><span class="lineno"> 3457</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l03458"></a><span class="lineno"> 3458</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l03459"></a><span class="lineno"> 3459</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l03460"></a><span class="lineno"> 3460</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03461"></a><span class="lineno"> 3461</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l03462"></a><span class="lineno"> 3462</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l03463"></a><span class="lineno"> 3463</span>&#160; uint dst_stride_z</div><div class="line"><a name="l03464"></a><span class="lineno"> 3464</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l03465"></a><span class="lineno"> 3465</span>&#160; ,</div><div class="line"><a name="l03466"></a><span class="lineno"> 3466</span>&#160; uint cross_plane_pad</div><div class="line"><a name="l03467"></a><span class="lineno"> 3467</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l03468"></a><span class="lineno"> 3468</span>&#160; )</div><div class="line"><a name="l03469"></a><span class="lineno"> 3469</span>&#160;{</div><div class="line"><a name="l03470"></a><span class="lineno"> 3470</span>&#160; <span class="keywordtype">int</span> x = get_global_id(0) / MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03471"></a><span class="lineno"> 3471</span>&#160; <span class="keywordtype">int</span> y = get_global_id(1) / MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03472"></a><span class="lineno"> 3472</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l03473"></a><span class="lineno"> 3473</span>&#160;</div><div class="line"><a name="l03474"></a><span class="lineno"> 3474</span>&#160; <span class="comment">// Offset</span></div><div class="line"><a name="l03475"></a><span class="lineno"> 3475</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_a = (get_global_id(1) % MULT_INTERLEAVE4X4_HEIGHT) * 4;</div><div class="line"><a name="l03476"></a><span class="lineno"> 3476</span>&#160; <span class="keyword">const</span> <span class="keywordtype">int</span> offset_row_b = (get_global_id(0) % MULT_TRANSPOSE1XW_WIDTH) * 8;</div><div class="line"><a name="l03477"></a><span class="lineno"> 3477</span>&#160;</div><div class="line"><a name="l03478"></a><span class="lineno"> 3478</span>&#160; <span class="comment">// src_addr_a = address of matrix A</span></div><div class="line"><a name="l03479"></a><span class="lineno"> 3479</span>&#160; <span class="comment">// src_addr_b = address of matrix B</span></div><div class="line"><a name="l03480"></a><span class="lineno"> 3480</span>&#160; <span class="keywordtype">int</span> src0_addr_in_bytes = z * src0_stride_z + y * src0_stride_y + src0_offset_first_element_in_bytes;</div><div class="line"><a name="l03481"></a><span class="lineno"> 3481</span>&#160; <span class="keywordtype">int</span> src1_addr_in_bytes = x * src1_stride_y + src1_offset_first_element_in_bytes;</div><div class="line"><a name="l03482"></a><span class="lineno"> 3482</span>&#160;</div><div class="line"><a name="l03483"></a><span class="lineno"> 3483</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03484"></a><span class="lineno"> 3484</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l03485"></a><span class="lineno"> 3485</span>&#160; src1_addr_in_bytes += (z % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l03486"></a><span class="lineno"> 3486</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03487"></a><span class="lineno"> 3487</span>&#160; src1_addr_in_bytes += z * src1_stride_z;</div><div class="line"><a name="l03488"></a><span class="lineno"> 3488</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03489"></a><span class="lineno"> 3489</span>&#160;</div><div class="line"><a name="l03490"></a><span class="lineno"> 3490</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_addr_a = (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src0_addr_in_bytes);</div><div class="line"><a name="l03491"></a><span class="lineno"> 3491</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_addr_b = (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src1_addr_in_bytes);</div><div class="line"><a name="l03492"></a><span class="lineno"> 3492</span>&#160;</div><div class="line"><a name="l03493"></a><span class="lineno"> 3493</span>&#160; <span class="comment">// Compute end row address for matrix B</span></div><div class="line"><a name="l03494"></a><span class="lineno"> 3494</span>&#160; __global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *src_end_addr_b = src_addr_b + COLS_B;</div><div class="line"><a name="l03495"></a><span class="lineno"> 3495</span>&#160;</div><div class="line"><a name="l03496"></a><span class="lineno"> 3496</span>&#160; src_addr_a += offset_row_a;</div><div class="line"><a name="l03497"></a><span class="lineno"> 3497</span>&#160; src_addr_b += offset_row_b;</div><div class="line"><a name="l03498"></a><span class="lineno"> 3498</span>&#160;</div><div class="line"><a name="l03499"></a><span class="lineno"> 3499</span>&#160; <span class="comment">// Reset accumulators</span></div><div class="line"><a name="l03500"></a><span class="lineno"> 3500</span>&#160; half8 c0 = 0.0f;</div><div class="line"><a name="l03501"></a><span class="lineno"> 3501</span>&#160; half8 c1 = 0.0f;</div><div class="line"><a name="l03502"></a><span class="lineno"> 3502</span>&#160; half8 c2 = 0.0f;</div><div class="line"><a name="l03503"></a><span class="lineno"> 3503</span>&#160; half8 c3 = 0.0f;</div><div class="line"><a name="l03504"></a><span class="lineno"> 3504</span>&#160;</div><div class="line"><a name="l03505"></a><span class="lineno"> 3505</span>&#160;<span class="preprocessor">#define COLS_MTX_B (COLS_B / (8 * MULT_TRANSPOSE1XW_WIDTH))</span></div><div class="line"><a name="l03506"></a><span class="lineno"> 3506</span>&#160;</div><div class="line"><a name="l03507"></a><span class="lineno"> 3507</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l03508"></a><span class="lineno"> 3508</span>&#160; <span class="keywordflow">for</span>(; i &lt;= (int)(COLS_MTX_B - 4); i += 4)</div><div class="line"><a name="l03509"></a><span class="lineno"> 3509</span>&#160; {</div><div class="line"><a name="l03510"></a><span class="lineno"> 3510</span>&#160;<span class="preprocessor">#if MULT_INTERLEAVE4X4_HEIGHT == 1</span></div><div class="line"><a name="l03511"></a><span class="lineno"> 3511</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03512"></a><span class="lineno"> 3512</span>&#160; half8 a0 = vload8(0, src_addr_a);</div><div class="line"><a name="l03513"></a><span class="lineno"> 3513</span>&#160; half8 b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03514"></a><span class="lineno"> 3514</span>&#160;</div><div class="line"><a name="l03515"></a><span class="lineno"> 3515</span>&#160; src_addr_a += 8 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03516"></a><span class="lineno"> 3516</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03517"></a><span class="lineno"> 3517</span>&#160;</div><div class="line"><a name="l03518"></a><span class="lineno"> 3518</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03519"></a><span class="lineno"> 3519</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03520"></a><span class="lineno"> 3520</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03521"></a><span class="lineno"> 3521</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03522"></a><span class="lineno"> 3522</span>&#160;</div><div class="line"><a name="l03523"></a><span class="lineno"> 3523</span>&#160; <span class="comment">// Load values from matrix B (transposed)</span></div><div class="line"><a name="l03524"></a><span class="lineno"> 3524</span>&#160; b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03525"></a><span class="lineno"> 3525</span>&#160;</div><div class="line"><a name="l03526"></a><span class="lineno"> 3526</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03527"></a><span class="lineno"> 3527</span>&#160;</div><div class="line"><a name="l03528"></a><span class="lineno"> 3528</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s4, b0, c0);</div><div class="line"><a name="l03529"></a><span class="lineno"> 3529</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s5, b0, c1);</div><div class="line"><a name="l03530"></a><span class="lineno"> 3530</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s6, b0, c2);</div><div class="line"><a name="l03531"></a><span class="lineno"> 3531</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s7, b0, c3);</div><div class="line"><a name="l03532"></a><span class="lineno"> 3532</span>&#160;</div><div class="line"><a name="l03533"></a><span class="lineno"> 3533</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03534"></a><span class="lineno"> 3534</span>&#160; a0 = vload8(0, src_addr_a);</div><div class="line"><a name="l03535"></a><span class="lineno"> 3535</span>&#160; b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03536"></a><span class="lineno"> 3536</span>&#160;</div><div class="line"><a name="l03537"></a><span class="lineno"> 3537</span>&#160; src_addr_a += 8 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03538"></a><span class="lineno"> 3538</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03539"></a><span class="lineno"> 3539</span>&#160;</div><div class="line"><a name="l03540"></a><span class="lineno"> 3540</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03541"></a><span class="lineno"> 3541</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03542"></a><span class="lineno"> 3542</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03543"></a><span class="lineno"> 3543</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03544"></a><span class="lineno"> 3544</span>&#160;</div><div class="line"><a name="l03545"></a><span class="lineno"> 3545</span>&#160; <span class="comment">// Load values from matrix B (transposed)</span></div><div class="line"><a name="l03546"></a><span class="lineno"> 3546</span>&#160; b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03547"></a><span class="lineno"> 3547</span>&#160;</div><div class="line"><a name="l03548"></a><span class="lineno"> 3548</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03549"></a><span class="lineno"> 3549</span>&#160;</div><div class="line"><a name="l03550"></a><span class="lineno"> 3550</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s4, b0, c0);</div><div class="line"><a name="l03551"></a><span class="lineno"> 3551</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s5, b0, c1);</div><div class="line"><a name="l03552"></a><span class="lineno"> 3552</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s6, b0, c2);</div><div class="line"><a name="l03553"></a><span class="lineno"> 3553</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s7, b0, c3);</div><div class="line"><a name="l03554"></a><span class="lineno"> 3554</span>&#160;<span class="preprocessor">#else // MULT_INTERLEAVE4X4_HEIGHT == 1</span></div><div class="line"><a name="l03555"></a><span class="lineno"> 3555</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03556"></a><span class="lineno"> 3556</span>&#160; half4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03557"></a><span class="lineno"> 3557</span>&#160; half8 b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03558"></a><span class="lineno"> 3558</span>&#160;</div><div class="line"><a name="l03559"></a><span class="lineno"> 3559</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03560"></a><span class="lineno"> 3560</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03561"></a><span class="lineno"> 3561</span>&#160;</div><div class="line"><a name="l03562"></a><span class="lineno"> 3562</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03563"></a><span class="lineno"> 3563</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03564"></a><span class="lineno"> 3564</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03565"></a><span class="lineno"> 3565</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03566"></a><span class="lineno"> 3566</span>&#160;</div><div class="line"><a name="l03567"></a><span class="lineno"> 3567</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03568"></a><span class="lineno"> 3568</span>&#160; a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03569"></a><span class="lineno"> 3569</span>&#160; b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03570"></a><span class="lineno"> 3570</span>&#160;</div><div class="line"><a name="l03571"></a><span class="lineno"> 3571</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03572"></a><span class="lineno"> 3572</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03573"></a><span class="lineno"> 3573</span>&#160;</div><div class="line"><a name="l03574"></a><span class="lineno"> 3574</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03575"></a><span class="lineno"> 3575</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03576"></a><span class="lineno"> 3576</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03577"></a><span class="lineno"> 3577</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03578"></a><span class="lineno"> 3578</span>&#160;</div><div class="line"><a name="l03579"></a><span class="lineno"> 3579</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03580"></a><span class="lineno"> 3580</span>&#160; a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03581"></a><span class="lineno"> 3581</span>&#160; b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03582"></a><span class="lineno"> 3582</span>&#160;</div><div class="line"><a name="l03583"></a><span class="lineno"> 3583</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03584"></a><span class="lineno"> 3584</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03585"></a><span class="lineno"> 3585</span>&#160;</div><div class="line"><a name="l03586"></a><span class="lineno"> 3586</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03587"></a><span class="lineno"> 3587</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03588"></a><span class="lineno"> 3588</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03589"></a><span class="lineno"> 3589</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03590"></a><span class="lineno"> 3590</span>&#160;</div><div class="line"><a name="l03591"></a><span class="lineno"> 3591</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03592"></a><span class="lineno"> 3592</span>&#160; a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03593"></a><span class="lineno"> 3593</span>&#160; b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03594"></a><span class="lineno"> 3594</span>&#160;</div><div class="line"><a name="l03595"></a><span class="lineno"> 3595</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03596"></a><span class="lineno"> 3596</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03597"></a><span class="lineno"> 3597</span>&#160;</div><div class="line"><a name="l03598"></a><span class="lineno"> 3598</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03599"></a><span class="lineno"> 3599</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03600"></a><span class="lineno"> 3600</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03601"></a><span class="lineno"> 3601</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03602"></a><span class="lineno"> 3602</span>&#160;<span class="preprocessor">#endif // MULT_INTERLEAVE4X4_HEIGHT == 1</span></div><div class="line"><a name="l03603"></a><span class="lineno"> 3603</span>&#160; }</div><div class="line"><a name="l03604"></a><span class="lineno"> 3604</span>&#160;</div><div class="line"><a name="l03605"></a><span class="lineno"> 3605</span>&#160; <span class="keywordflow">for</span>(; i &lt; (int)(COLS_MTX_B); ++i)</div><div class="line"><a name="l03606"></a><span class="lineno"> 3606</span>&#160; {</div><div class="line"><a name="l03607"></a><span class="lineno"> 3607</span>&#160; <span class="comment">// Load values from matrix A (interleaved) and matrix B (transposed)</span></div><div class="line"><a name="l03608"></a><span class="lineno"> 3608</span>&#160; half4 a0 = vload4(0, src_addr_a);</div><div class="line"><a name="l03609"></a><span class="lineno"> 3609</span>&#160; half8 b0 = vload8(0, src_addr_b);</div><div class="line"><a name="l03610"></a><span class="lineno"> 3610</span>&#160;</div><div class="line"><a name="l03611"></a><span class="lineno"> 3611</span>&#160; src_addr_a += 4 * MULT_INTERLEAVE4X4_HEIGHT;</div><div class="line"><a name="l03612"></a><span class="lineno"> 3612</span>&#160; src_addr_b += 8 * MULT_TRANSPOSE1XW_WIDTH;</div><div class="line"><a name="l03613"></a><span class="lineno"> 3613</span>&#160;</div><div class="line"><a name="l03614"></a><span class="lineno"> 3614</span>&#160; c0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s0, b0, c0);</div><div class="line"><a name="l03615"></a><span class="lineno"> 3615</span>&#160; c1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s1, b0, c1);</div><div class="line"><a name="l03616"></a><span class="lineno"> 3616</span>&#160; c2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s2, b0, c2);</div><div class="line"><a name="l03617"></a><span class="lineno"> 3617</span>&#160; c3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>((half8)a0.s3, b0, c3);</div><div class="line"><a name="l03618"></a><span class="lineno"> 3618</span>&#160; }</div><div class="line"><a name="l03619"></a><span class="lineno"> 3619</span>&#160;</div><div class="line"><a name="l03620"></a><span class="lineno"> 3620</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l03621"></a><span class="lineno"> 3621</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l03622"></a><span class="lineno"> 3622</span>&#160;</div><div class="line"><a name="l03623"></a><span class="lineno"> 3623</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l03624"></a><span class="lineno"> 3624</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l03625"></a><span class="lineno"> 3625</span>&#160;</div><div class="line"><a name="l03626"></a><span class="lineno"> 3626</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l03627"></a><span class="lineno"> 3627</span>&#160;</div><div class="line"><a name="l03628"></a><span class="lineno"> 3628</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03629"></a><span class="lineno"> 3629</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l03630"></a><span class="lineno"> 3630</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l03631"></a><span class="lineno"> 3631</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l03632"></a><span class="lineno"> 3632</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03633"></a><span class="lineno"> 3633</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l03634"></a><span class="lineno"> 3634</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03635"></a><span class="lineno"> 3635</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03636"></a><span class="lineno"> 3636</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03637"></a><span class="lineno"> 3637</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l03638"></a><span class="lineno"> 3638</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03639"></a><span class="lineno"> 3639</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03640"></a><span class="lineno"> 3640</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l03641"></a><span class="lineno"> 3641</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03642"></a><span class="lineno"> 3642</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03643"></a><span class="lineno"> 3643</span>&#160;</div><div class="line"><a name="l03644"></a><span class="lineno"> 3644</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * 4) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l03645"></a><span class="lineno"> 3645</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * 4)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l03646"></a><span class="lineno"> 3646</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l03647"></a><span class="lineno"> 3647</span>&#160;</div><div class="line"><a name="l03648"></a><span class="lineno"> 3648</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l03649"></a><span class="lineno"> 3649</span>&#160; zout *= (cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l03650"></a><span class="lineno"> 3650</span>&#160;</div><div class="line"><a name="l03651"></a><span class="lineno"> 3651</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l03652"></a><span class="lineno"> 3652</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l03653"></a><span class="lineno"> 3653</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l03654"></a><span class="lineno"> 3654</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03655"></a><span class="lineno"> 3655</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l03656"></a><span class="lineno"> 3656</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l03657"></a><span class="lineno"> 3657</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03658"></a><span class="lineno"> 3658</span>&#160;</div><div class="line"><a name="l03659"></a><span class="lineno"> 3659</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l03660"></a><span class="lineno"> 3660</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l03661"></a><span class="lineno"> 3661</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, c, ALPHA);</div><div class="line"><a name="l03662"></a><span class="lineno"> 3662</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l03663"></a><span class="lineno"> 3663</span>&#160;</div><div class="line"><a name="l03664"></a><span class="lineno"> 3664</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l03665"></a><span class="lineno"> 3665</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l03666"></a><span class="lineno"> 3666</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(4, uint, zero, 0);</div><div class="line"><a name="l03667"></a><span class="lineno"> 3667</span>&#160;</div><div class="line"><a name="l03668"></a><span class="lineno"> 3668</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03669"></a><span class="lineno"> 3669</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>));</div><div class="line"><a name="l03670"></a><span class="lineno"> 3670</span>&#160;</div><div class="line"><a name="l03671"></a><span class="lineno"> 3671</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03672"></a><span class="lineno"> 3672</span>&#160;</div><div class="line"><a name="l03673"></a><span class="lineno"> 3673</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03674"></a><span class="lineno"> 3674</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l03675"></a><span class="lineno"> 3675</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l03676"></a><span class="lineno"> 3676</span>&#160;</div><div class="line"><a name="l03677"></a><span class="lineno"> 3677</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l03678"></a><span class="lineno"> 3678</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(4, c, bias0);</div><div class="line"><a name="l03679"></a><span class="lineno"> 3679</span>&#160;</div><div class="line"><a name="l03680"></a><span class="lineno"> 3680</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03681"></a><span class="lineno"> 3681</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>)) + (get_global_id(1) * (uint)4 * src2_stride_y) + get_global_id(</div><div class="line"><a name="l03682"></a><span class="lineno"> 3682</span>&#160; 2) * src2_stride_z;</div><div class="line"><a name="l03683"></a><span class="lineno"> 3683</span>&#160;</div><div class="line"><a name="l03684"></a><span class="lineno"> 3684</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(4, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03685"></a><span class="lineno"> 3685</span>&#160;</div><div class="line"><a name="l03686"></a><span class="lineno"> 3686</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03687"></a><span class="lineno"> 3687</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(4, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l03688"></a><span class="lineno"> 3688</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l03689"></a><span class="lineno"> 3689</span>&#160;</div><div class="line"><a name="l03690"></a><span class="lineno"> 3690</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l03691"></a><span class="lineno"> 3691</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(4, c, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l03692"></a><span class="lineno"> 3692</span>&#160;</div><div class="line"><a name="l03693"></a><span class="lineno"> 3693</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03694"></a><span class="lineno"> 3694</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l03695"></a><span class="lineno"> 3695</span>&#160;</div><div class="line"><a name="l03696"></a><span class="lineno"> 3696</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l03697"></a><span class="lineno"> 3697</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(4, ACTIVATION_TYPE, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, c, A_VAL, B_VAL);</div><div class="line"><a name="l03698"></a><span class="lineno"> 3698</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l03699"></a><span class="lineno"> 3699</span>&#160;</div><div class="line"><a name="l03700"></a><span class="lineno"> 3700</span>&#160; <span class="comment">// Store 4x8 block</span></div><div class="line"><a name="l03701"></a><span class="lineno"> 3701</span>&#160; vstore8(c0, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l03702"></a><span class="lineno"> 3702</span>&#160; vstore8(c1, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l03703"></a><span class="lineno"> 3703</span>&#160; vstore8(c2, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l03704"></a><span class="lineno"> 3704</span>&#160; vstore8(c3, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l03705"></a><span class="lineno"> 3705</span>&#160;}</div><div class="line"><a name="l03706"></a><span class="lineno"> 3706</span>&#160;</div><div class="line"><a name="l03707"></a><span class="lineno"> 3707</span>&#160;<span class="comment">// Undefine local defines</span></div><div class="line"><a name="l03708"></a><span class="lineno"> 3708</span>&#160;<span class="preprocessor">#undef COLS_MTX_B</span></div><div class="line"><a name="l03709"></a><span class="lineno"> 3709</span>&#160;</div><div class="line"><a name="l03710"></a><span class="lineno"> 3710</span>&#160;<span class="preprocessor">#endif // defined(ARM_COMPUTE_OPENCL_FP16_ENABLED)</span></div><div class="line"><a name="l03711"></a><span class="lineno"> 3711</span>&#160;</div><div class="line"><a name="l03712"></a><span class="lineno"> 3712</span>&#160;<span class="preprocessor">#endif // defined(COLS_B) &amp;&amp; defined(MULT_TRANSPOSE1XW_WIDTH) &amp;&amp; defined(MULT_INTERLEAVE4X4_HEIGHT)</span></div><div class="line"><a name="l03713"></a><span class="lineno"> 3713</span>&#160;</div><div class="line"><a name="l03714"></a><span class="lineno"> 3714</span>&#160;<span class="preprocessor">#if defined(COLS_A) &amp;&amp; defined(NUM_ELEMS_PROCESSED_PER_THREAD_X) &amp;&amp; (NUM_ELEMS_PROCESSED_PER_THREAD_Y)</span></div><div class="line"><a name="l03715"></a><span class="lineno"> 3715</span>&#160;<span class="preprocessor">#if defined(DATA_TYPE)</span></div><div class="line"><a name="l03716"></a><span class="lineno"> 3716</span>&#160;<span class="preprocessor">#define VECTOR_TYPE VEC_DATA_TYPE(DATA_TYPE, NUM_ELEMS_PROCESSED_PER_THREAD_X)</span></div><div class="line"><a name="l03717"></a><span class="lineno"> 3717</span>&#160;</div><div class="line"><a name="l03766"></a><span class="lineno"> 3766</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_floating_point(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l03767"></a><span class="lineno"> 3767</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l03768"></a><span class="lineno"> 3768</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03769"></a><span class="lineno"> 3769</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l03770"></a><span class="lineno"> 3770</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l03771"></a><span class="lineno"> 3771</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l03772"></a><span class="lineno"> 3772</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l03773"></a><span class="lineno"> 3773</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l03774"></a><span class="lineno"> 3774</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l03775"></a><span class="lineno"> 3775</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l03776"></a><span class="lineno"> 3776</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l03777"></a><span class="lineno"> 3777</span>&#160; uint dst_stride_z</div><div class="line"><a name="l03778"></a><span class="lineno"> 3778</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l03779"></a><span class="lineno"> 3779</span>&#160; ,</div><div class="line"><a name="l03780"></a><span class="lineno"> 3780</span>&#160; uint src_cross_plane_pad</div><div class="line"><a name="l03781"></a><span class="lineno"> 3781</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l03782"></a><span class="lineno"> 3782</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l03783"></a><span class="lineno"> 3783</span>&#160; ,</div><div class="line"><a name="l03784"></a><span class="lineno"> 3784</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l03785"></a><span class="lineno"> 3785</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l03786"></a><span class="lineno"> 3786</span>&#160; )</div><div class="line"><a name="l03787"></a><span class="lineno"> 3787</span>&#160;{</div><div class="line"><a name="l03788"></a><span class="lineno"> 3788</span>&#160; <span class="keywordtype">int</span> idx = get_global_id(0) * NUM_ELEMS_PROCESSED_PER_THREAD_X;</div><div class="line"><a name="l03789"></a><span class="lineno"> 3789</span>&#160;</div><div class="line"><a name="l03790"></a><span class="lineno"> 3790</span>&#160; <span class="comment">// Compute starting address for matrix A and Matrix B</span></div><div class="line"><a name="l03791"></a><span class="lineno"> 3791</span>&#160; int2 src_addr = ((int2)(src0_offset_first_element_in_bytes, src1_offset_first_element_in_bytes));</div><div class="line"><a name="l03792"></a><span class="lineno"> 3792</span>&#160;</div><div class="line"><a name="l03793"></a><span class="lineno"> 3793</span>&#160; <span class="comment">// Update address for the matrix A</span></div><div class="line"><a name="l03794"></a><span class="lineno"> 3794</span>&#160; src_addr.s0 += get_global_id(1) * src0_stride_y * NUM_ELEMS_PROCESSED_PER_THREAD_Y;</div><div class="line"><a name="l03795"></a><span class="lineno"> 3795</span>&#160;</div><div class="line"><a name="l03796"></a><span class="lineno"> 3796</span>&#160; <span class="comment">// Update address for the matrix B</span></div><div class="line"><a name="l03797"></a><span class="lineno"> 3797</span>&#160; src_addr.s1 += idx * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>);</div><div class="line"><a name="l03798"></a><span class="lineno"> 3798</span>&#160;</div><div class="line"><a name="l03799"></a><span class="lineno"> 3799</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03800"></a><span class="lineno"> 3800</span>&#160; <span class="comment">// Since we load a 2D input tile from a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l03801"></a><span class="lineno"> 3801</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l03802"></a><span class="lineno"> 3802</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l03803"></a><span class="lineno"> 3803</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03804"></a><span class="lineno"> 3804</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l03805"></a><span class="lineno"> 3805</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03806"></a><span class="lineno"> 3806</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03807"></a><span class="lineno"> 3807</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03808"></a><span class="lineno"> 3808</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l03809"></a><span class="lineno"> 3809</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03810"></a><span class="lineno"> 3810</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03811"></a><span class="lineno"> 3811</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l03812"></a><span class="lineno"> 3812</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03813"></a><span class="lineno"> 3813</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03814"></a><span class="lineno"> 3814</span>&#160;</div><div class="line"><a name="l03815"></a><span class="lineno"> 3815</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l03816"></a><span class="lineno"> 3816</span>&#160; uint4 zin = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l03817"></a><span class="lineno"> 3817</span>&#160; zin = min(DEPTH_GEMM3D - 1, zin);</div><div class="line"><a name="l03818"></a><span class="lineno"> 3818</span>&#160;</div><div class="line"><a name="l03819"></a><span class="lineno"> 3819</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l03820"></a><span class="lineno"> 3820</span>&#160; zin *= (src_cross_plane_pad * src0_stride_y);</div><div class="line"><a name="l03821"></a><span class="lineno"> 3821</span>&#160;</div><div class="line"><a name="l03822"></a><span class="lineno"> 3822</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l03823"></a><span class="lineno"> 3823</span>&#160; <span class="comment">// multiply src0_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l03824"></a><span class="lineno"> 3824</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l03825"></a><span class="lineno"> 3825</span>&#160;</div><div class="line"><a name="l03826"></a><span class="lineno"> 3826</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03827"></a><span class="lineno"> 3827</span>&#160;</div><div class="line"><a name="l03828"></a><span class="lineno"> 3828</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l03829"></a><span class="lineno"> 3829</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z;</div><div class="line"><a name="l03830"></a><span class="lineno"> 3830</span>&#160;</div><div class="line"><a name="l03831"></a><span class="lineno"> 3831</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03832"></a><span class="lineno"> 3832</span>&#160;</div><div class="line"><a name="l03833"></a><span class="lineno"> 3833</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03834"></a><span class="lineno"> 3834</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l03835"></a><span class="lineno"> 3835</span>&#160; src_addr.s1 += (get_global_id(2) % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l03836"></a><span class="lineno"> 3836</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03837"></a><span class="lineno"> 3837</span>&#160; src_addr.s1 += get_global_id(2) * src1_stride_z;</div><div class="line"><a name="l03838"></a><span class="lineno"> 3838</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l03839"></a><span class="lineno"> 3839</span>&#160;</div><div class="line"><a name="l03840"></a><span class="lineno"> 3840</span>&#160; <span class="keywordtype">int</span> end_row_vec_a = src_addr.s0 + (COLS_A * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l03841"></a><span class="lineno"> 3841</span>&#160;</div><div class="line"><a name="l03842"></a><span class="lineno"> 3842</span>&#160; VECTOR_TYPE acc0 = 0.0f;</div><div class="line"><a name="l03843"></a><span class="lineno"> 3843</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03844"></a><span class="lineno"> 3844</span>&#160; VECTOR_TYPE acc1 = 0.0f;</div><div class="line"><a name="l03845"></a><span class="lineno"> 3845</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03846"></a><span class="lineno"> 3846</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03847"></a><span class="lineno"> 3847</span>&#160; VECTOR_TYPE acc2 = 0.0f;</div><div class="line"><a name="l03848"></a><span class="lineno"> 3848</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03849"></a><span class="lineno"> 3849</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03850"></a><span class="lineno"> 3850</span>&#160; VECTOR_TYPE acc3 = 0.0f;</div><div class="line"><a name="l03851"></a><span class="lineno"> 3851</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03852"></a><span class="lineno"> 3852</span>&#160;</div><div class="line"><a name="l03853"></a><span class="lineno"> 3853</span>&#160; <span class="keywordflow">for</span>(; src_addr.s0 &lt;= (end_row_vec_a - 2 * (int)<span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)); src_addr += (int2)(2 * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), 2 * src1_stride_y))</div><div class="line"><a name="l03854"></a><span class="lineno"> 3854</span>&#160; {</div><div class="line"><a name="l03855"></a><span class="lineno"> 3855</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03856"></a><span class="lineno"> 3856</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l03857"></a><span class="lineno"> 3857</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 2, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, a, src0_ptr, src_addr.s0, src0_stride_y, zin.s);</div><div class="line"><a name="l03858"></a><span class="lineno"> 3858</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03859"></a><span class="lineno"> 3859</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l03860"></a><span class="lineno"> 3860</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l03861"></a><span class="lineno"> 3861</span>&#160; a0 = vload2(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l03862"></a><span class="lineno"> 3862</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03863"></a><span class="lineno"> 3863</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l03864"></a><span class="lineno"> 3864</span>&#160; a1 = vload2(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l03865"></a><span class="lineno"> 3865</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03866"></a><span class="lineno"> 3866</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03867"></a><span class="lineno"> 3867</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l03868"></a><span class="lineno"> 3868</span>&#160; a2 = vload2(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l03869"></a><span class="lineno"> 3869</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03870"></a><span class="lineno"> 3870</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03871"></a><span class="lineno"> 3871</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, 2)</div><div class="line"><a name="l03872"></a><span class="lineno"> 3872</span>&#160; a3 = vload2(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l03873"></a><span class="lineno"> 3873</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03874"></a><span class="lineno"> 3874</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03875"></a><span class="lineno"> 3875</span>&#160;</div><div class="line"><a name="l03876"></a><span class="lineno"> 3876</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l03877"></a><span class="lineno"> 3877</span>&#160; VECTOR_TYPE b0 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(NUM_ELEMS_PROCESSED_PER_THREAD_X)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l03878"></a><span class="lineno"> 3878</span>&#160; VECTOR_TYPE b1 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(NUM_ELEMS_PROCESSED_PER_THREAD_X)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src1_ptr + src_addr.s1 + src1_stride_y));</div><div class="line"><a name="l03879"></a><span class="lineno"> 3879</span>&#160;</div><div class="line"><a name="l03880"></a><span class="lineno"> 3880</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l03881"></a><span class="lineno"> 3881</span>&#160; acc0 += b0 * (VECTOR_TYPE)a0.s0;</div><div class="line"><a name="l03882"></a><span class="lineno"> 3882</span>&#160; acc0 += b1 * (VECTOR_TYPE)a0.s1;</div><div class="line"><a name="l03883"></a><span class="lineno"> 3883</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03884"></a><span class="lineno"> 3884</span>&#160; acc1 += b0 * (VECTOR_TYPE)a1.s0;</div><div class="line"><a name="l03885"></a><span class="lineno"> 3885</span>&#160; acc1 += b1 * (VECTOR_TYPE)a1.s1;</div><div class="line"><a name="l03886"></a><span class="lineno"> 3886</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03887"></a><span class="lineno"> 3887</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03888"></a><span class="lineno"> 3888</span>&#160; acc2 += b0 * (VECTOR_TYPE)a2.s0;</div><div class="line"><a name="l03889"></a><span class="lineno"> 3889</span>&#160; acc2 += b1 * (VECTOR_TYPE)a2.s1;</div><div class="line"><a name="l03890"></a><span class="lineno"> 3890</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03891"></a><span class="lineno"> 3891</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03892"></a><span class="lineno"> 3892</span>&#160; acc3 += b0 * (VECTOR_TYPE)a3.s0;</div><div class="line"><a name="l03893"></a><span class="lineno"> 3893</span>&#160; acc3 += b1 * (VECTOR_TYPE)a3.s1;</div><div class="line"><a name="l03894"></a><span class="lineno"> 3894</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03895"></a><span class="lineno"> 3895</span>&#160; }</div><div class="line"><a name="l03896"></a><span class="lineno"> 3896</span>&#160;</div><div class="line"><a name="l03897"></a><span class="lineno"> 3897</span>&#160; <span class="keywordflow">for</span>(; src_addr.s0 &lt; end_row_vec_a; src_addr += (int2)(<span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>), src1_stride_y))</div><div class="line"><a name="l03898"></a><span class="lineno"> 3898</span>&#160; {</div><div class="line"><a name="l03899"></a><span class="lineno"> 3899</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03900"></a><span class="lineno"> 3900</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l03901"></a><span class="lineno"> 3901</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a0 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y + zin.s0));</div><div class="line"><a name="l03902"></a><span class="lineno"> 3902</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03903"></a><span class="lineno"> 3903</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a1 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y + zin.s1));</div><div class="line"><a name="l03904"></a><span class="lineno"> 3904</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03905"></a><span class="lineno"> 3905</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03906"></a><span class="lineno"> 3906</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a2 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y + zin.s2));</div><div class="line"><a name="l03907"></a><span class="lineno"> 3907</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03908"></a><span class="lineno"> 3908</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03909"></a><span class="lineno"> 3909</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a3 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y + zin.s3));</div><div class="line"><a name="l03910"></a><span class="lineno"> 3910</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03911"></a><span class="lineno"> 3911</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03912"></a><span class="lineno"> 3912</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l03913"></a><span class="lineno"> 3913</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a0 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l03914"></a><span class="lineno"> 3914</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03915"></a><span class="lineno"> 3915</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a1 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l03916"></a><span class="lineno"> 3916</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03917"></a><span class="lineno"> 3917</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03918"></a><span class="lineno"> 3918</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a2 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l03919"></a><span class="lineno"> 3919</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03920"></a><span class="lineno"> 3920</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03921"></a><span class="lineno"> 3921</span>&#160; <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> a3 = *((__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l03922"></a><span class="lineno"> 3922</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03923"></a><span class="lineno"> 3923</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l03924"></a><span class="lineno"> 3924</span>&#160;</div><div class="line"><a name="l03925"></a><span class="lineno"> 3925</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l03926"></a><span class="lineno"> 3926</span>&#160; VECTOR_TYPE b0 = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(NUM_ELEMS_PROCESSED_PER_THREAD_X)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l03927"></a><span class="lineno"> 3927</span>&#160;</div><div class="line"><a name="l03928"></a><span class="lineno"> 3928</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l03929"></a><span class="lineno"> 3929</span>&#160; acc0 += b0 * (VECTOR_TYPE)a0;</div><div class="line"><a name="l03930"></a><span class="lineno"> 3930</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03931"></a><span class="lineno"> 3931</span>&#160; acc1 += b0 * (VECTOR_TYPE)a1;</div><div class="line"><a name="l03932"></a><span class="lineno"> 3932</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l03933"></a><span class="lineno"> 3933</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03934"></a><span class="lineno"> 3934</span>&#160; acc2 += b0 * (VECTOR_TYPE)a2;</div><div class="line"><a name="l03935"></a><span class="lineno"> 3935</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l03936"></a><span class="lineno"> 3936</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03937"></a><span class="lineno"> 3937</span>&#160; acc3 += b0 * (VECTOR_TYPE)a3;</div><div class="line"><a name="l03938"></a><span class="lineno"> 3938</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l03939"></a><span class="lineno"> 3939</span>&#160; }</div><div class="line"><a name="l03940"></a><span class="lineno"> 3940</span>&#160;</div><div class="line"><a name="l03941"></a><span class="lineno"> 3941</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l03942"></a><span class="lineno"> 3942</span>&#160;</div><div class="line"><a name="l03943"></a><span class="lineno"> 3943</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l03944"></a><span class="lineno"> 3944</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l03945"></a><span class="lineno"> 3945</span>&#160;</div><div class="line"><a name="l03946"></a><span class="lineno"> 3946</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l03947"></a><span class="lineno"> 3947</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l03948"></a><span class="lineno"> 3948</span>&#160;</div><div class="line"><a name="l03949"></a><span class="lineno"> 3949</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l03950"></a><span class="lineno"> 3950</span>&#160;</div><div class="line"><a name="l03951"></a><span class="lineno"> 3951</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03952"></a><span class="lineno"> 3952</span>&#160;</div><div class="line"><a name="l03953"></a><span class="lineno"> 3953</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l03954"></a><span class="lineno"> 3954</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l03955"></a><span class="lineno"> 3955</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l03956"></a><span class="lineno"> 3956</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03957"></a><span class="lineno"> 3957</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l03958"></a><span class="lineno"> 3958</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03959"></a><span class="lineno"> 3959</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03960"></a><span class="lineno"> 3960</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03961"></a><span class="lineno"> 3961</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l03962"></a><span class="lineno"> 3962</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l03963"></a><span class="lineno"> 3963</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03964"></a><span class="lineno"> 3964</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l03965"></a><span class="lineno"> 3965</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l03966"></a><span class="lineno"> 3966</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l03967"></a><span class="lineno"> 3967</span>&#160;</div><div class="line"><a name="l03968"></a><span class="lineno"> 3968</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l03969"></a><span class="lineno"> 3969</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l03970"></a><span class="lineno"> 3970</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l03971"></a><span class="lineno"> 3971</span>&#160;</div><div class="line"><a name="l03972"></a><span class="lineno"> 3972</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l03973"></a><span class="lineno"> 3973</span>&#160; zout *= (dst_cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l03974"></a><span class="lineno"> 3974</span>&#160;</div><div class="line"><a name="l03975"></a><span class="lineno"> 3975</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l03976"></a><span class="lineno"> 3976</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l03977"></a><span class="lineno"> 3977</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l03978"></a><span class="lineno"> 3978</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03979"></a><span class="lineno"> 3979</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l03980"></a><span class="lineno"> 3980</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l03981"></a><span class="lineno"> 3981</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l03982"></a><span class="lineno"> 3982</span>&#160;</div><div class="line"><a name="l03983"></a><span class="lineno"> 3983</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l03984"></a><span class="lineno"> 3984</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l03985"></a><span class="lineno"> 3985</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, acc, ALPHA);</div><div class="line"><a name="l03986"></a><span class="lineno"> 3986</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l03987"></a><span class="lineno"> 3987</span>&#160;</div><div class="line"><a name="l03988"></a><span class="lineno"> 3988</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l03989"></a><span class="lineno"> 3989</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l03990"></a><span class="lineno"> 3990</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, uint, zero, 0);</div><div class="line"><a name="l03991"></a><span class="lineno"> 3991</span>&#160;</div><div class="line"><a name="l03992"></a><span class="lineno"> 3992</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l03993"></a><span class="lineno"> 3993</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)NUM_ELEMS_PROCESSED_PER_THREAD_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>));</div><div class="line"><a name="l03994"></a><span class="lineno"> 3994</span>&#160;</div><div class="line"><a name="l03995"></a><span class="lineno"> 3995</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, NUM_ELEMS_PROCESSED_PER_THREAD_X, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l03996"></a><span class="lineno"> 3996</span>&#160;</div><div class="line"><a name="l03997"></a><span class="lineno"> 3997</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l03998"></a><span class="lineno"> 3998</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l03999"></a><span class="lineno"> 3999</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l04000"></a><span class="lineno"> 4000</span>&#160;</div><div class="line"><a name="l04001"></a><span class="lineno"> 4001</span>&#160; <span class="comment">// c = c + bias[broadcasted]</span></div><div class="line"><a name="l04002"></a><span class="lineno"> 4002</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, bias0);</div><div class="line"><a name="l04003"></a><span class="lineno"> 4003</span>&#160;</div><div class="line"><a name="l04004"></a><span class="lineno"> 4004</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04005"></a><span class="lineno"> 4005</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)NUM_ELEMS_PROCESSED_PER_THREAD_X * <span class="keyword">sizeof</span>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>)) + (get_global_id(1) *</div><div class="line"><a name="l04006"></a><span class="lineno"> 4006</span>&#160; (uint)NUM_ELEMS_PROCESSED_PER_THREAD_Y * src2_stride_y) + get_global_id(2) * src2_stride_z;</div><div class="line"><a name="l04007"></a><span class="lineno"> 4007</span>&#160;</div><div class="line"><a name="l04008"></a><span class="lineno"> 4008</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, NUM_ELEMS_PROCESSED_PER_THREAD_X, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l04009"></a><span class="lineno"> 4009</span>&#160;</div><div class="line"><a name="l04010"></a><span class="lineno"> 4010</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l04011"></a><span class="lineno"> 4011</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l04012"></a><span class="lineno"> 4012</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l04013"></a><span class="lineno"> 4013</span>&#160;</div><div class="line"><a name="l04014"></a><span class="lineno"> 4014</span>&#160; <span class="comment">// c = c + bias</span></div><div class="line"><a name="l04015"></a><span class="lineno"> 4015</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l04016"></a><span class="lineno"> 4016</span>&#160;</div><div class="line"><a name="l04017"></a><span class="lineno"> 4017</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04018"></a><span class="lineno"> 4018</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l04019"></a><span class="lineno"> 4019</span>&#160;</div><div class="line"><a name="l04020"></a><span class="lineno"> 4020</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l04021"></a><span class="lineno"> 4021</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, ACTIVATION_TYPE, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, acc, A_VAL, B_VAL);</div><div class="line"><a name="l04022"></a><span class="lineno"> 4022</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l04023"></a><span class="lineno"> 4023</span>&#160;</div><div class="line"><a name="l04024"></a><span class="lineno"> 4024</span>&#160; <span class="comment">// Store output block</span></div><div class="line"><a name="l04025"></a><span class="lineno"> 4025</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, NUM_ELEMS_PROCESSED_PER_THREAD_X, <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, acc, dst_addr, dst_stride_y, zout.s);</div><div class="line"><a name="l04026"></a><span class="lineno"> 4026</span>&#160;}</div><div class="line"><a name="l04027"></a><span class="lineno"> 4027</span>&#160;<span class="preprocessor">#endif // defined(DATA_TYPE)</span></div><div class="line"><a name="l04028"></a><span class="lineno"> 4028</span>&#160;</div><div class="line"><a name="l04079"></a><span class="lineno"> 4079</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_floating_point_f32_bifrost(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l04080"></a><span class="lineno"> 4080</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l04081"></a><span class="lineno"> 4081</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l04082"></a><span class="lineno"> 4082</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l04083"></a><span class="lineno"> 4083</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l04084"></a><span class="lineno"> 4084</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l04085"></a><span class="lineno"> 4085</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l04086"></a><span class="lineno"> 4086</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l04087"></a><span class="lineno"> 4087</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l04088"></a><span class="lineno"> 4088</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l04089"></a><span class="lineno"> 4089</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l04090"></a><span class="lineno"> 4090</span>&#160; uint dst_stride_z</div><div class="line"><a name="l04091"></a><span class="lineno"> 4091</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l04092"></a><span class="lineno"> 4092</span>&#160; ,</div><div class="line"><a name="l04093"></a><span class="lineno"> 4093</span>&#160; uint src_cross_plane_pad</div><div class="line"><a name="l04094"></a><span class="lineno"> 4094</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l04095"></a><span class="lineno"> 4095</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l04096"></a><span class="lineno"> 4096</span>&#160; ,</div><div class="line"><a name="l04097"></a><span class="lineno"> 4097</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l04098"></a><span class="lineno"> 4098</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l04099"></a><span class="lineno"> 4099</span>&#160; )</div><div class="line"><a name="l04100"></a><span class="lineno"> 4100</span>&#160;{</div><div class="line"><a name="l04101"></a><span class="lineno"> 4101</span>&#160; <span class="keywordtype">int</span> idx = get_global_id(0) * NUM_ELEMS_PROCESSED_PER_THREAD_X;</div><div class="line"><a name="l04102"></a><span class="lineno"> 4102</span>&#160;</div><div class="line"><a name="l04103"></a><span class="lineno"> 4103</span>&#160; <span class="comment">// Compute starting address for matrix A and matrix B</span></div><div class="line"><a name="l04104"></a><span class="lineno"> 4104</span>&#160; int2 src_addr = ((int2)(src0_offset_first_element_in_bytes, src1_offset_first_element_in_bytes));</div><div class="line"><a name="l04105"></a><span class="lineno"> 4105</span>&#160;</div><div class="line"><a name="l04106"></a><span class="lineno"> 4106</span>&#160; <span class="comment">// Update address for matrix A</span></div><div class="line"><a name="l04107"></a><span class="lineno"> 4107</span>&#160; src_addr.s0 += get_global_id(1) * src0_stride_y * NUM_ELEMS_PROCESSED_PER_THREAD_Y;</div><div class="line"><a name="l04108"></a><span class="lineno"> 4108</span>&#160;</div><div class="line"><a name="l04109"></a><span class="lineno"> 4109</span>&#160; <span class="comment">// Update address for matrix B</span></div><div class="line"><a name="l04110"></a><span class="lineno"> 4110</span>&#160; src_addr.s1 += idx * <span class="keyword">sizeof</span>(float);</div><div class="line"><a name="l04111"></a><span class="lineno"> 4111</span>&#160;</div><div class="line"><a name="l04112"></a><span class="lineno"> 4112</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04113"></a><span class="lineno"> 4113</span>&#160; <span class="comment">// Since we load a 2D input tile from a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l04114"></a><span class="lineno"> 4114</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l04115"></a><span class="lineno"> 4115</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l04116"></a><span class="lineno"> 4116</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04117"></a><span class="lineno"> 4117</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l04118"></a><span class="lineno"> 4118</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04119"></a><span class="lineno"> 4119</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04120"></a><span class="lineno"> 4120</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04121"></a><span class="lineno"> 4121</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l04122"></a><span class="lineno"> 4122</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04123"></a><span class="lineno"> 4123</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04124"></a><span class="lineno"> 4124</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l04125"></a><span class="lineno"> 4125</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04126"></a><span class="lineno"> 4126</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04127"></a><span class="lineno"> 4127</span>&#160;</div><div class="line"><a name="l04128"></a><span class="lineno"> 4128</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l04129"></a><span class="lineno"> 4129</span>&#160; uint4 zin = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l04130"></a><span class="lineno"> 4130</span>&#160; zin = min(DEPTH_GEMM3D - 1, zin);</div><div class="line"><a name="l04131"></a><span class="lineno"> 4131</span>&#160;</div><div class="line"><a name="l04132"></a><span class="lineno"> 4132</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l04133"></a><span class="lineno"> 4133</span>&#160; zin *= (src_cross_plane_pad * src0_stride_y);</div><div class="line"><a name="l04134"></a><span class="lineno"> 4134</span>&#160;</div><div class="line"><a name="l04135"></a><span class="lineno"> 4135</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l04136"></a><span class="lineno"> 4136</span>&#160; <span class="comment">// multiply src0_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l04137"></a><span class="lineno"> 4137</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l04138"></a><span class="lineno"> 4138</span>&#160;</div><div class="line"><a name="l04139"></a><span class="lineno"> 4139</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04140"></a><span class="lineno"> 4140</span>&#160;</div><div class="line"><a name="l04141"></a><span class="lineno"> 4141</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l04142"></a><span class="lineno"> 4142</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z;</div><div class="line"><a name="l04143"></a><span class="lineno"> 4143</span>&#160;</div><div class="line"><a name="l04144"></a><span class="lineno"> 4144</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04145"></a><span class="lineno"> 4145</span>&#160;</div><div class="line"><a name="l04146"></a><span class="lineno"> 4146</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l04147"></a><span class="lineno"> 4147</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l04148"></a><span class="lineno"> 4148</span>&#160; src_addr.s1 += (get_global_id(2) % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l04149"></a><span class="lineno"> 4149</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l04150"></a><span class="lineno"> 4150</span>&#160; src_addr.s1 += get_global_id(2) * src1_stride_z;</div><div class="line"><a name="l04151"></a><span class="lineno"> 4151</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l04152"></a><span class="lineno"> 4152</span>&#160;</div><div class="line"><a name="l04153"></a><span class="lineno"> 4153</span>&#160; <span class="comment">// Initialize accumulators</span></div><div class="line"><a name="l04154"></a><span class="lineno"> 4154</span>&#160; float4 acc0 = 0.0f;</div><div class="line"><a name="l04155"></a><span class="lineno"> 4155</span>&#160;</div><div class="line"><a name="l04156"></a><span class="lineno"> 4156</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04157"></a><span class="lineno"> 4157</span>&#160; float4 acc1 = 0.0f;</div><div class="line"><a name="l04158"></a><span class="lineno"> 4158</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04159"></a><span class="lineno"> 4159</span>&#160;</div><div class="line"><a name="l04160"></a><span class="lineno"> 4160</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04161"></a><span class="lineno"> 4161</span>&#160; float4 acc2 = 0.0f;</div><div class="line"><a name="l04162"></a><span class="lineno"> 4162</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04163"></a><span class="lineno"> 4163</span>&#160;</div><div class="line"><a name="l04164"></a><span class="lineno"> 4164</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04165"></a><span class="lineno"> 4165</span>&#160; float4 acc3 = 0.0f;</div><div class="line"><a name="l04166"></a><span class="lineno"> 4166</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04167"></a><span class="lineno"> 4167</span>&#160;</div><div class="line"><a name="l04168"></a><span class="lineno"> 4168</span>&#160; <span class="comment">// A and B src indices get incremented at the same time.</span></div><div class="line"><a name="l04169"></a><span class="lineno"> 4169</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l04170"></a><span class="lineno"> 4170</span>&#160; <span class="keywordflow">for</span>(; i &lt;= ((int)COLS_A - 4); i += 4)</div><div class="line"><a name="l04171"></a><span class="lineno"> 4171</span>&#160; {</div><div class="line"><a name="l04172"></a><span class="lineno"> 4172</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04173"></a><span class="lineno"> 4173</span>&#160; <span class="comment">// Load values from matrix A and matrix B</span></div><div class="line"><a name="l04174"></a><span class="lineno"> 4174</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 4, <span class="keywordtype">float</span>, a, src0_ptr, src_addr.s0, src0_stride_y, zin.s);</div><div class="line"><a name="l04175"></a><span class="lineno"> 4175</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04176"></a><span class="lineno"> 4176</span>&#160; <span class="comment">// Load values from matrix A and matrix B</span></div><div class="line"><a name="l04177"></a><span class="lineno"> 4177</span>&#160; float4 a0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l04178"></a><span class="lineno"> 4178</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04179"></a><span class="lineno"> 4179</span>&#160; float4 a1 = vload4(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l04180"></a><span class="lineno"> 4180</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04181"></a><span class="lineno"> 4181</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04182"></a><span class="lineno"> 4182</span>&#160; float4 a2 = vload4(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l04183"></a><span class="lineno"> 4183</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04184"></a><span class="lineno"> 4184</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04185"></a><span class="lineno"> 4185</span>&#160; float4 a3 = vload4(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l04186"></a><span class="lineno"> 4186</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04187"></a><span class="lineno"> 4187</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04188"></a><span class="lineno"> 4188</span>&#160;</div><div class="line"><a name="l04189"></a><span class="lineno"> 4189</span>&#160; float4 b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04190"></a><span class="lineno"> 4190</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04191"></a><span class="lineno"> 4191</span>&#160;</div><div class="line"><a name="l04192"></a><span class="lineno"> 4192</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04193"></a><span class="lineno"> 4193</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, acc0.s0);</div><div class="line"><a name="l04194"></a><span class="lineno"> 4194</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, acc0.s1);</div><div class="line"><a name="l04195"></a><span class="lineno"> 4195</span>&#160; acc0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s2, acc0.s2);</div><div class="line"><a name="l04196"></a><span class="lineno"> 4196</span>&#160; acc0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s3, acc0.s3);</div><div class="line"><a name="l04197"></a><span class="lineno"> 4197</span>&#160;</div><div class="line"><a name="l04198"></a><span class="lineno"> 4198</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04199"></a><span class="lineno"> 4199</span>&#160;</div><div class="line"><a name="l04200"></a><span class="lineno"> 4200</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s0, b0.s0, acc1.s0);</div><div class="line"><a name="l04201"></a><span class="lineno"> 4201</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s0, b0.s1, acc1.s1);</div><div class="line"><a name="l04202"></a><span class="lineno"> 4202</span>&#160; acc1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s0, b0.s2, acc1.s2);</div><div class="line"><a name="l04203"></a><span class="lineno"> 4203</span>&#160; acc1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s0, b0.s3, acc1.s3);</div><div class="line"><a name="l04204"></a><span class="lineno"> 4204</span>&#160;</div><div class="line"><a name="l04205"></a><span class="lineno"> 4205</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04206"></a><span class="lineno"> 4206</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04207"></a><span class="lineno"> 4207</span>&#160;</div><div class="line"><a name="l04208"></a><span class="lineno"> 4208</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s0, b0.s0, acc2.s0);</div><div class="line"><a name="l04209"></a><span class="lineno"> 4209</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s0, b0.s1, acc2.s1);</div><div class="line"><a name="l04210"></a><span class="lineno"> 4210</span>&#160; acc2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s0, b0.s2, acc2.s2);</div><div class="line"><a name="l04211"></a><span class="lineno"> 4211</span>&#160; acc2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s0, b0.s3, acc2.s3);</div><div class="line"><a name="l04212"></a><span class="lineno"> 4212</span>&#160;</div><div class="line"><a name="l04213"></a><span class="lineno"> 4213</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04214"></a><span class="lineno"> 4214</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04215"></a><span class="lineno"> 4215</span>&#160;</div><div class="line"><a name="l04216"></a><span class="lineno"> 4216</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s0, b0.s0, acc3.s0);</div><div class="line"><a name="l04217"></a><span class="lineno"> 4217</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s0, b0.s1, acc3.s1);</div><div class="line"><a name="l04218"></a><span class="lineno"> 4218</span>&#160; acc3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s0, b0.s2, acc3.s2);</div><div class="line"><a name="l04219"></a><span class="lineno"> 4219</span>&#160; acc3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s0, b0.s3, acc3.s3);</div><div class="line"><a name="l04220"></a><span class="lineno"> 4220</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04221"></a><span class="lineno"> 4221</span>&#160;</div><div class="line"><a name="l04222"></a><span class="lineno"> 4222</span>&#160; <span class="comment">// Load values from matrix A and matrix B</span></div><div class="line"><a name="l04223"></a><span class="lineno"> 4223</span>&#160; b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04224"></a><span class="lineno"> 4224</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04225"></a><span class="lineno"> 4225</span>&#160;</div><div class="line"><a name="l04226"></a><span class="lineno"> 4226</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04227"></a><span class="lineno"> 4227</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s0, acc0.s0);</div><div class="line"><a name="l04228"></a><span class="lineno"> 4228</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s1, acc0.s1);</div><div class="line"><a name="l04229"></a><span class="lineno"> 4229</span>&#160; acc0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s2, acc0.s2);</div><div class="line"><a name="l04230"></a><span class="lineno"> 4230</span>&#160; acc0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b0.s3, acc0.s3);</div><div class="line"><a name="l04231"></a><span class="lineno"> 4231</span>&#160;</div><div class="line"><a name="l04232"></a><span class="lineno"> 4232</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04233"></a><span class="lineno"> 4233</span>&#160;</div><div class="line"><a name="l04234"></a><span class="lineno"> 4234</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s1, b0.s0, acc1.s0);</div><div class="line"><a name="l04235"></a><span class="lineno"> 4235</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s1, b0.s1, acc1.s1);</div><div class="line"><a name="l04236"></a><span class="lineno"> 4236</span>&#160; acc1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s1, b0.s2, acc1.s2);</div><div class="line"><a name="l04237"></a><span class="lineno"> 4237</span>&#160; acc1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s1, b0.s3, acc1.s3);</div><div class="line"><a name="l04238"></a><span class="lineno"> 4238</span>&#160;</div><div class="line"><a name="l04239"></a><span class="lineno"> 4239</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04240"></a><span class="lineno"> 4240</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04241"></a><span class="lineno"> 4241</span>&#160;</div><div class="line"><a name="l04242"></a><span class="lineno"> 4242</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s1, b0.s0, acc2.s0);</div><div class="line"><a name="l04243"></a><span class="lineno"> 4243</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s1, b0.s1, acc2.s1);</div><div class="line"><a name="l04244"></a><span class="lineno"> 4244</span>&#160; acc2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s1, b0.s2, acc2.s2);</div><div class="line"><a name="l04245"></a><span class="lineno"> 4245</span>&#160; acc2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s1, b0.s3, acc2.s3);</div><div class="line"><a name="l04246"></a><span class="lineno"> 4246</span>&#160;</div><div class="line"><a name="l04247"></a><span class="lineno"> 4247</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04248"></a><span class="lineno"> 4248</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04249"></a><span class="lineno"> 4249</span>&#160;</div><div class="line"><a name="l04250"></a><span class="lineno"> 4250</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s1, b0.s0, acc3.s0);</div><div class="line"><a name="l04251"></a><span class="lineno"> 4251</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s1, b0.s1, acc3.s1);</div><div class="line"><a name="l04252"></a><span class="lineno"> 4252</span>&#160; acc3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s1, b0.s2, acc3.s2);</div><div class="line"><a name="l04253"></a><span class="lineno"> 4253</span>&#160; acc3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s1, b0.s3, acc3.s3);</div><div class="line"><a name="l04254"></a><span class="lineno"> 4254</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04255"></a><span class="lineno"> 4255</span>&#160;</div><div class="line"><a name="l04256"></a><span class="lineno"> 4256</span>&#160; <span class="comment">// Load values from matrix A and matrix B</span></div><div class="line"><a name="l04257"></a><span class="lineno"> 4257</span>&#160; b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04258"></a><span class="lineno"> 4258</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04259"></a><span class="lineno"> 4259</span>&#160;</div><div class="line"><a name="l04260"></a><span class="lineno"> 4260</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04261"></a><span class="lineno"> 4261</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s0, acc0.s0);</div><div class="line"><a name="l04262"></a><span class="lineno"> 4262</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s1, acc0.s1);</div><div class="line"><a name="l04263"></a><span class="lineno"> 4263</span>&#160; acc0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s2, acc0.s2);</div><div class="line"><a name="l04264"></a><span class="lineno"> 4264</span>&#160; acc0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b0.s3, acc0.s3);</div><div class="line"><a name="l04265"></a><span class="lineno"> 4265</span>&#160;</div><div class="line"><a name="l04266"></a><span class="lineno"> 4266</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04267"></a><span class="lineno"> 4267</span>&#160;</div><div class="line"><a name="l04268"></a><span class="lineno"> 4268</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s2, b0.s0, acc1.s0);</div><div class="line"><a name="l04269"></a><span class="lineno"> 4269</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s2, b0.s1, acc1.s1);</div><div class="line"><a name="l04270"></a><span class="lineno"> 4270</span>&#160; acc1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s2, b0.s2, acc1.s2);</div><div class="line"><a name="l04271"></a><span class="lineno"> 4271</span>&#160; acc1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s2, b0.s3, acc1.s3);</div><div class="line"><a name="l04272"></a><span class="lineno"> 4272</span>&#160;</div><div class="line"><a name="l04273"></a><span class="lineno"> 4273</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04274"></a><span class="lineno"> 4274</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04275"></a><span class="lineno"> 4275</span>&#160;</div><div class="line"><a name="l04276"></a><span class="lineno"> 4276</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s2, b0.s0, acc2.s0);</div><div class="line"><a name="l04277"></a><span class="lineno"> 4277</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s2, b0.s1, acc2.s1);</div><div class="line"><a name="l04278"></a><span class="lineno"> 4278</span>&#160; acc2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s2, b0.s2, acc2.s2);</div><div class="line"><a name="l04279"></a><span class="lineno"> 4279</span>&#160; acc2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s2, b0.s3, acc2.s3);</div><div class="line"><a name="l04280"></a><span class="lineno"> 4280</span>&#160;</div><div class="line"><a name="l04281"></a><span class="lineno"> 4281</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04282"></a><span class="lineno"> 4282</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04283"></a><span class="lineno"> 4283</span>&#160;</div><div class="line"><a name="l04284"></a><span class="lineno"> 4284</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s2, b0.s0, acc3.s0);</div><div class="line"><a name="l04285"></a><span class="lineno"> 4285</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s2, b0.s1, acc3.s1);</div><div class="line"><a name="l04286"></a><span class="lineno"> 4286</span>&#160; acc3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s2, b0.s2, acc3.s2);</div><div class="line"><a name="l04287"></a><span class="lineno"> 4287</span>&#160; acc3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s2, b0.s3, acc3.s3);</div><div class="line"><a name="l04288"></a><span class="lineno"> 4288</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04289"></a><span class="lineno"> 4289</span>&#160;</div><div class="line"><a name="l04290"></a><span class="lineno"> 4290</span>&#160; <span class="comment">// Load values from matrix A and matrix B</span></div><div class="line"><a name="l04291"></a><span class="lineno"> 4291</span>&#160; b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04292"></a><span class="lineno"> 4292</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04293"></a><span class="lineno"> 4293</span>&#160;</div><div class="line"><a name="l04294"></a><span class="lineno"> 4294</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04295"></a><span class="lineno"> 4295</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s0, acc0.s0);</div><div class="line"><a name="l04296"></a><span class="lineno"> 4296</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s1, acc0.s1);</div><div class="line"><a name="l04297"></a><span class="lineno"> 4297</span>&#160; acc0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s2, acc0.s2);</div><div class="line"><a name="l04298"></a><span class="lineno"> 4298</span>&#160; acc0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b0.s3, acc0.s3);</div><div class="line"><a name="l04299"></a><span class="lineno"> 4299</span>&#160;</div><div class="line"><a name="l04300"></a><span class="lineno"> 4300</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04301"></a><span class="lineno"> 4301</span>&#160;</div><div class="line"><a name="l04302"></a><span class="lineno"> 4302</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s3, b0.s0, acc1.s0);</div><div class="line"><a name="l04303"></a><span class="lineno"> 4303</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s3, b0.s1, acc1.s1);</div><div class="line"><a name="l04304"></a><span class="lineno"> 4304</span>&#160; acc1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s3, b0.s2, acc1.s2);</div><div class="line"><a name="l04305"></a><span class="lineno"> 4305</span>&#160; acc1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1.s3, b0.s3, acc1.s3);</div><div class="line"><a name="l04306"></a><span class="lineno"> 4306</span>&#160;</div><div class="line"><a name="l04307"></a><span class="lineno"> 4307</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04308"></a><span class="lineno"> 4308</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04309"></a><span class="lineno"> 4309</span>&#160;</div><div class="line"><a name="l04310"></a><span class="lineno"> 4310</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s3, b0.s0, acc2.s0);</div><div class="line"><a name="l04311"></a><span class="lineno"> 4311</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s3, b0.s1, acc2.s1);</div><div class="line"><a name="l04312"></a><span class="lineno"> 4312</span>&#160; acc2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s3, b0.s2, acc2.s2);</div><div class="line"><a name="l04313"></a><span class="lineno"> 4313</span>&#160; acc2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2.s3, b0.s3, acc2.s3);</div><div class="line"><a name="l04314"></a><span class="lineno"> 4314</span>&#160;</div><div class="line"><a name="l04315"></a><span class="lineno"> 4315</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04316"></a><span class="lineno"> 4316</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04317"></a><span class="lineno"> 4317</span>&#160;</div><div class="line"><a name="l04318"></a><span class="lineno"> 4318</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s3, b0.s0, acc3.s0);</div><div class="line"><a name="l04319"></a><span class="lineno"> 4319</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s3, b0.s1, acc3.s1);</div><div class="line"><a name="l04320"></a><span class="lineno"> 4320</span>&#160; acc3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s3, b0.s2, acc3.s2);</div><div class="line"><a name="l04321"></a><span class="lineno"> 4321</span>&#160; acc3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3.s3, b0.s3, acc3.s3);</div><div class="line"><a name="l04322"></a><span class="lineno"> 4322</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04323"></a><span class="lineno"> 4323</span>&#160;</div><div class="line"><a name="l04324"></a><span class="lineno"> 4324</span>&#160; src_addr.s0 += 4 * <span class="keyword">sizeof</span>(float);</div><div class="line"><a name="l04325"></a><span class="lineno"> 4325</span>&#160; }</div><div class="line"><a name="l04326"></a><span class="lineno"> 4326</span>&#160;</div><div class="line"><a name="l04327"></a><span class="lineno"> 4327</span>&#160; <span class="keywordflow">for</span>(; i &lt; (int)COLS_A; ++i)</div><div class="line"><a name="l04328"></a><span class="lineno"> 4328</span>&#160; {</div><div class="line"><a name="l04329"></a><span class="lineno"> 4329</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04330"></a><span class="lineno"> 4330</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l04331"></a><span class="lineno"> 4331</span>&#160; <span class="keywordtype">float</span> a0 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y + zin.s0));</div><div class="line"><a name="l04332"></a><span class="lineno"> 4332</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04333"></a><span class="lineno"> 4333</span>&#160; <span class="keywordtype">float</span> a1 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y + zin.s1));</div><div class="line"><a name="l04334"></a><span class="lineno"> 4334</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04335"></a><span class="lineno"> 4335</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04336"></a><span class="lineno"> 4336</span>&#160; <span class="keywordtype">float</span> a2 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y + zin.s2));</div><div class="line"><a name="l04337"></a><span class="lineno"> 4337</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04338"></a><span class="lineno"> 4338</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04339"></a><span class="lineno"> 4339</span>&#160; <span class="keywordtype">float</span> a3 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y + zin.s3));</div><div class="line"><a name="l04340"></a><span class="lineno"> 4340</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04341"></a><span class="lineno"> 4341</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04342"></a><span class="lineno"> 4342</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l04343"></a><span class="lineno"> 4343</span>&#160; <span class="keywordtype">float</span> a0 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l04344"></a><span class="lineno"> 4344</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04345"></a><span class="lineno"> 4345</span>&#160; <span class="keywordtype">float</span> a1 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l04346"></a><span class="lineno"> 4346</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04347"></a><span class="lineno"> 4347</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04348"></a><span class="lineno"> 4348</span>&#160; <span class="keywordtype">float</span> a2 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l04349"></a><span class="lineno"> 4349</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04350"></a><span class="lineno"> 4350</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04351"></a><span class="lineno"> 4351</span>&#160; <span class="keywordtype">float</span> a3 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l04352"></a><span class="lineno"> 4352</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04353"></a><span class="lineno"> 4353</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04354"></a><span class="lineno"> 4354</span>&#160;</div><div class="line"><a name="l04355"></a><span class="lineno"> 4355</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l04356"></a><span class="lineno"> 4356</span>&#160; float4 b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04357"></a><span class="lineno"> 4357</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04358"></a><span class="lineno"> 4358</span>&#160;</div><div class="line"><a name="l04359"></a><span class="lineno"> 4359</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04360"></a><span class="lineno"> 4360</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0, b0.s0, acc0.s0);</div><div class="line"><a name="l04361"></a><span class="lineno"> 4361</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0, b0.s1, acc0.s1);</div><div class="line"><a name="l04362"></a><span class="lineno"> 4362</span>&#160; acc0.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0, b0.s2, acc0.s2);</div><div class="line"><a name="l04363"></a><span class="lineno"> 4363</span>&#160; acc0.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0, b0.s3, acc0.s3);</div><div class="line"><a name="l04364"></a><span class="lineno"> 4364</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04365"></a><span class="lineno"> 4365</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1, b0.s0, acc1.s0);</div><div class="line"><a name="l04366"></a><span class="lineno"> 4366</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1, b0.s1, acc1.s1);</div><div class="line"><a name="l04367"></a><span class="lineno"> 4367</span>&#160; acc1.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1, b0.s2, acc1.s2);</div><div class="line"><a name="l04368"></a><span class="lineno"> 4368</span>&#160; acc1.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1, b0.s3, acc1.s3);</div><div class="line"><a name="l04369"></a><span class="lineno"> 4369</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04370"></a><span class="lineno"> 4370</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04371"></a><span class="lineno"> 4371</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2, b0.s0, acc2.s0);</div><div class="line"><a name="l04372"></a><span class="lineno"> 4372</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2, b0.s1, acc2.s1);</div><div class="line"><a name="l04373"></a><span class="lineno"> 4373</span>&#160; acc2.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2, b0.s2, acc2.s2);</div><div class="line"><a name="l04374"></a><span class="lineno"> 4374</span>&#160; acc2.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2, b0.s3, acc2.s3);</div><div class="line"><a name="l04375"></a><span class="lineno"> 4375</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04376"></a><span class="lineno"> 4376</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04377"></a><span class="lineno"> 4377</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3, b0.s0, acc3.s0);</div><div class="line"><a name="l04378"></a><span class="lineno"> 4378</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3, b0.s1, acc3.s1);</div><div class="line"><a name="l04379"></a><span class="lineno"> 4379</span>&#160; acc3.s2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3, b0.s2, acc3.s2);</div><div class="line"><a name="l04380"></a><span class="lineno"> 4380</span>&#160; acc3.s3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3, b0.s3, acc3.s3);</div><div class="line"><a name="l04381"></a><span class="lineno"> 4381</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04382"></a><span class="lineno"> 4382</span>&#160;</div><div class="line"><a name="l04383"></a><span class="lineno"> 4383</span>&#160; src_addr.s0 += <span class="keyword">sizeof</span>(float);</div><div class="line"><a name="l04384"></a><span class="lineno"> 4384</span>&#160; }</div><div class="line"><a name="l04385"></a><span class="lineno"> 4385</span>&#160;</div><div class="line"><a name="l04386"></a><span class="lineno"> 4386</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l04387"></a><span class="lineno"> 4387</span>&#160;</div><div class="line"><a name="l04388"></a><span class="lineno"> 4388</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l04389"></a><span class="lineno"> 4389</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l04390"></a><span class="lineno"> 4390</span>&#160;</div><div class="line"><a name="l04391"></a><span class="lineno"> 4391</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l04392"></a><span class="lineno"> 4392</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l04393"></a><span class="lineno"> 4393</span>&#160;</div><div class="line"><a name="l04394"></a><span class="lineno"> 4394</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l04395"></a><span class="lineno"> 4395</span>&#160;</div><div class="line"><a name="l04396"></a><span class="lineno"> 4396</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l04397"></a><span class="lineno"> 4397</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l04398"></a><span class="lineno"> 4398</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l04399"></a><span class="lineno"> 4399</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l04400"></a><span class="lineno"> 4400</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04401"></a><span class="lineno"> 4401</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l04402"></a><span class="lineno"> 4402</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04403"></a><span class="lineno"> 4403</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04404"></a><span class="lineno"> 4404</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04405"></a><span class="lineno"> 4405</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l04406"></a><span class="lineno"> 4406</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04407"></a><span class="lineno"> 4407</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04408"></a><span class="lineno"> 4408</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l04409"></a><span class="lineno"> 4409</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04410"></a><span class="lineno"> 4410</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04411"></a><span class="lineno"> 4411</span>&#160;</div><div class="line"><a name="l04412"></a><span class="lineno"> 4412</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l04413"></a><span class="lineno"> 4413</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l04414"></a><span class="lineno"> 4414</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l04415"></a><span class="lineno"> 4415</span>&#160;</div><div class="line"><a name="l04416"></a><span class="lineno"> 4416</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l04417"></a><span class="lineno"> 4417</span>&#160; zout *= (dst_cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l04418"></a><span class="lineno"> 4418</span>&#160;</div><div class="line"><a name="l04419"></a><span class="lineno"> 4419</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l04420"></a><span class="lineno"> 4420</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l04421"></a><span class="lineno"> 4421</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l04422"></a><span class="lineno"> 4422</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l04423"></a><span class="lineno"> 4423</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l04424"></a><span class="lineno"> 4424</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l04425"></a><span class="lineno"> 4425</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l04426"></a><span class="lineno"> 4426</span>&#160;</div><div class="line"><a name="l04427"></a><span class="lineno"> 4427</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l04428"></a><span class="lineno"> 4428</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l04429"></a><span class="lineno"> 4429</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <span class="keywordtype">float</span>, acc, ALPHA);</div><div class="line"><a name="l04430"></a><span class="lineno"> 4430</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l04431"></a><span class="lineno"> 4431</span>&#160;</div><div class="line"><a name="l04432"></a><span class="lineno"> 4432</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l04433"></a><span class="lineno"> 4433</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l04434"></a><span class="lineno"> 4434</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, uint, zero, 0);</div><div class="line"><a name="l04435"></a><span class="lineno"> 4435</span>&#160;</div><div class="line"><a name="l04436"></a><span class="lineno"> 4436</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04437"></a><span class="lineno"> 4437</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)4 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>));</div><div class="line"><a name="l04438"></a><span class="lineno"> 4438</span>&#160;</div><div class="line"><a name="l04439"></a><span class="lineno"> 4439</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l04440"></a><span class="lineno"> 4440</span>&#160;</div><div class="line"><a name="l04441"></a><span class="lineno"> 4441</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l04442"></a><span class="lineno"> 4442</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l04443"></a><span class="lineno"> 4443</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l04444"></a><span class="lineno"> 4444</span>&#160;</div><div class="line"><a name="l04445"></a><span class="lineno"> 4445</span>&#160; <span class="comment">// acc = acc + bias[broadcasted]</span></div><div class="line"><a name="l04446"></a><span class="lineno"> 4446</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, bias0);</div><div class="line"><a name="l04447"></a><span class="lineno"> 4447</span>&#160;</div><div class="line"><a name="l04448"></a><span class="lineno"> 4448</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04449"></a><span class="lineno"> 4449</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)4 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>)) + (get_global_id(1) *</div><div class="line"><a name="l04450"></a><span class="lineno"> 4450</span>&#160; (uint)NUM_ELEMS_PROCESSED_PER_THREAD_Y * src2_stride_y) + get_global_id(2) * src2_stride_z;</div><div class="line"><a name="l04451"></a><span class="lineno"> 4451</span>&#160;</div><div class="line"><a name="l04452"></a><span class="lineno"> 4452</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 4, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l04453"></a><span class="lineno"> 4453</span>&#160;</div><div class="line"><a name="l04454"></a><span class="lineno"> 4454</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l04455"></a><span class="lineno"> 4455</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l04456"></a><span class="lineno"> 4456</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l04457"></a><span class="lineno"> 4457</span>&#160;</div><div class="line"><a name="l04458"></a><span class="lineno"> 4458</span>&#160; <span class="comment">// acc = acc + bias</span></div><div class="line"><a name="l04459"></a><span class="lineno"> 4459</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l04460"></a><span class="lineno"> 4460</span>&#160;</div><div class="line"><a name="l04461"></a><span class="lineno"> 4461</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04462"></a><span class="lineno"> 4462</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l04463"></a><span class="lineno"> 4463</span>&#160;</div><div class="line"><a name="l04464"></a><span class="lineno"> 4464</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l04465"></a><span class="lineno"> 4465</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, ACTIVATION_TYPE, <span class="keywordtype">float</span>, acc, A_VAL, B_VAL);</div><div class="line"><a name="l04466"></a><span class="lineno"> 4466</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l04467"></a><span class="lineno"> 4467</span>&#160;</div><div class="line"><a name="l04468"></a><span class="lineno"> 4468</span>&#160; <span class="comment">// Store the output block</span></div><div class="line"><a name="l04469"></a><span class="lineno"> 4469</span>&#160; vstore4(acc0, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l04470"></a><span class="lineno"> 4470</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04471"></a><span class="lineno"> 4471</span>&#160; vstore4(acc1, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l04472"></a><span class="lineno"> 4472</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04473"></a><span class="lineno"> 4473</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04474"></a><span class="lineno"> 4474</span>&#160; vstore4(acc2, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l04475"></a><span class="lineno"> 4475</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04476"></a><span class="lineno"> 4476</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04477"></a><span class="lineno"> 4477</span>&#160; vstore4(acc3, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l04478"></a><span class="lineno"> 4478</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04479"></a><span class="lineno"> 4479</span>&#160;}</div><div class="line"><a name="l04480"></a><span class="lineno"> 4480</span>&#160;</div><div class="line"><a name="l04532"></a><span class="lineno"> 4532</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_floating_point_f32_bifrost_1000(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l04533"></a><span class="lineno"> 4533</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l04534"></a><span class="lineno"> 4534</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l04535"></a><span class="lineno"> 4535</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l04536"></a><span class="lineno"> 4536</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l04537"></a><span class="lineno"> 4537</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l04538"></a><span class="lineno"> 4538</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l04539"></a><span class="lineno"> 4539</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l04540"></a><span class="lineno"> 4540</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l04541"></a><span class="lineno"> 4541</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l04542"></a><span class="lineno"> 4542</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l04543"></a><span class="lineno"> 4543</span>&#160; uint dst_stride_z</div><div class="line"><a name="l04544"></a><span class="lineno"> 4544</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l04545"></a><span class="lineno"> 4545</span>&#160; ,</div><div class="line"><a name="l04546"></a><span class="lineno"> 4546</span>&#160; uint src_cross_plane_pad</div><div class="line"><a name="l04547"></a><span class="lineno"> 4547</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l04548"></a><span class="lineno"> 4548</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l04549"></a><span class="lineno"> 4549</span>&#160; ,</div><div class="line"><a name="l04550"></a><span class="lineno"> 4550</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l04551"></a><span class="lineno"> 4551</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l04552"></a><span class="lineno"> 4552</span>&#160; )</div><div class="line"><a name="l04553"></a><span class="lineno"> 4553</span>&#160;{</div><div class="line"><a name="l04554"></a><span class="lineno"> 4554</span>&#160; <span class="comment">// Requires 2 NUM_ELEMS_PROCESSED_PER_THREAD_X, C vect2, A vect4, B (2 vload2) // to fix for NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04555"></a><span class="lineno"> 4555</span>&#160; <span class="keywordtype">int</span> idx = get_global_id(0) * NUM_ELEMS_PROCESSED_PER_THREAD_X;</div><div class="line"><a name="l04556"></a><span class="lineno"> 4556</span>&#160;</div><div class="line"><a name="l04557"></a><span class="lineno"> 4557</span>&#160; <span class="comment">// Compute starting address for matrix A and Matrix B</span></div><div class="line"><a name="l04558"></a><span class="lineno"> 4558</span>&#160; int2 src_addr = ((int2)(src0_offset_first_element_in_bytes, src1_offset_first_element_in_bytes));</div><div class="line"><a name="l04559"></a><span class="lineno"> 4559</span>&#160;</div><div class="line"><a name="l04560"></a><span class="lineno"> 4560</span>&#160; <span class="comment">// Update address for the matrix A</span></div><div class="line"><a name="l04561"></a><span class="lineno"> 4561</span>&#160; src_addr.s0 += get_global_id(1) * src0_stride_y * NUM_ELEMS_PROCESSED_PER_THREAD_Y;</div><div class="line"><a name="l04562"></a><span class="lineno"> 4562</span>&#160;</div><div class="line"><a name="l04563"></a><span class="lineno"> 4563</span>&#160; <span class="comment">// Update address for the matrix B</span></div><div class="line"><a name="l04564"></a><span class="lineno"> 4564</span>&#160; src_addr.s1 += idx * <span class="keyword">sizeof</span>(float);</div><div class="line"><a name="l04565"></a><span class="lineno"> 4565</span>&#160;</div><div class="line"><a name="l04566"></a><span class="lineno"> 4566</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04567"></a><span class="lineno"> 4567</span>&#160; <span class="comment">// Since we load a 2D input tile from a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l04568"></a><span class="lineno"> 4568</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l04569"></a><span class="lineno"> 4569</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l04570"></a><span class="lineno"> 4570</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04571"></a><span class="lineno"> 4571</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l04572"></a><span class="lineno"> 4572</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04573"></a><span class="lineno"> 4573</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04574"></a><span class="lineno"> 4574</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04575"></a><span class="lineno"> 4575</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l04576"></a><span class="lineno"> 4576</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04577"></a><span class="lineno"> 4577</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04578"></a><span class="lineno"> 4578</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l04579"></a><span class="lineno"> 4579</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04580"></a><span class="lineno"> 4580</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04581"></a><span class="lineno"> 4581</span>&#160;</div><div class="line"><a name="l04582"></a><span class="lineno"> 4582</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l04583"></a><span class="lineno"> 4583</span>&#160; uint4 zin = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l04584"></a><span class="lineno"> 4584</span>&#160; zin = min(DEPTH_GEMM3D - 1, zin);</div><div class="line"><a name="l04585"></a><span class="lineno"> 4585</span>&#160;</div><div class="line"><a name="l04586"></a><span class="lineno"> 4586</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l04587"></a><span class="lineno"> 4587</span>&#160; zin *= (src_cross_plane_pad * src0_stride_y);</div><div class="line"><a name="l04588"></a><span class="lineno"> 4588</span>&#160;</div><div class="line"><a name="l04589"></a><span class="lineno"> 4589</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l04590"></a><span class="lineno"> 4590</span>&#160; <span class="comment">// multiply src0_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l04591"></a><span class="lineno"> 4591</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l04592"></a><span class="lineno"> 4592</span>&#160;</div><div class="line"><a name="l04593"></a><span class="lineno"> 4593</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04594"></a><span class="lineno"> 4594</span>&#160;</div><div class="line"><a name="l04595"></a><span class="lineno"> 4595</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l04596"></a><span class="lineno"> 4596</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z;</div><div class="line"><a name="l04597"></a><span class="lineno"> 4597</span>&#160;</div><div class="line"><a name="l04598"></a><span class="lineno"> 4598</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04599"></a><span class="lineno"> 4599</span>&#160;</div><div class="line"><a name="l04600"></a><span class="lineno"> 4600</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l04601"></a><span class="lineno"> 4601</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l04602"></a><span class="lineno"> 4602</span>&#160; src_addr.s1 += (get_global_id(2) % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l04603"></a><span class="lineno"> 4603</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l04604"></a><span class="lineno"> 4604</span>&#160; src_addr.s1 += get_global_id(2) * src1_stride_z;</div><div class="line"><a name="l04605"></a><span class="lineno"> 4605</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l04606"></a><span class="lineno"> 4606</span>&#160;</div><div class="line"><a name="l04607"></a><span class="lineno"> 4607</span>&#160; <span class="comment">// Initialize accumulators</span></div><div class="line"><a name="l04608"></a><span class="lineno"> 4608</span>&#160; float2 acc0 = 0.0f;</div><div class="line"><a name="l04609"></a><span class="lineno"> 4609</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04610"></a><span class="lineno"> 4610</span>&#160; float2 acc1 = 0.0f;</div><div class="line"><a name="l04611"></a><span class="lineno"> 4611</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04612"></a><span class="lineno"> 4612</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04613"></a><span class="lineno"> 4613</span>&#160; float2 acc2 = 0.0f;</div><div class="line"><a name="l04614"></a><span class="lineno"> 4614</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04615"></a><span class="lineno"> 4615</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04616"></a><span class="lineno"> 4616</span>&#160; float2 acc3 = 0.0f;</div><div class="line"><a name="l04617"></a><span class="lineno"> 4617</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04618"></a><span class="lineno"> 4618</span>&#160;</div><div class="line"><a name="l04619"></a><span class="lineno"> 4619</span>&#160; <span class="comment">// A and B src indices get incremented at the same time.</span></div><div class="line"><a name="l04620"></a><span class="lineno"> 4620</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l04621"></a><span class="lineno"> 4621</span>&#160; <span class="keywordflow">for</span>(; i &lt;= ((int)COLS_A - 8); i += 8)</div><div class="line"><a name="l04622"></a><span class="lineno"> 4622</span>&#160; {</div><div class="line"><a name="l04623"></a><span class="lineno"> 4623</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04624"></a><span class="lineno"> 4624</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l04625"></a><span class="lineno"> 4625</span>&#160; float8 a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + zin.s0));</div><div class="line"><a name="l04626"></a><span class="lineno"> 4626</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04627"></a><span class="lineno"> 4627</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l04628"></a><span class="lineno"> 4628</span>&#160; float8 a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0));</div><div class="line"><a name="l04629"></a><span class="lineno"> 4629</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04630"></a><span class="lineno"> 4630</span>&#160;</div><div class="line"><a name="l04631"></a><span class="lineno"> 4631</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l04632"></a><span class="lineno"> 4632</span>&#160; float2 b0 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04633"></a><span class="lineno"> 4633</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04634"></a><span class="lineno"> 4634</span>&#160; float2 b1 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04635"></a><span class="lineno"> 4635</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04636"></a><span class="lineno"> 4636</span>&#160; float2 b2 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04637"></a><span class="lineno"> 4637</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04638"></a><span class="lineno"> 4638</span>&#160; float2 b3 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04639"></a><span class="lineno"> 4639</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04640"></a><span class="lineno"> 4640</span>&#160; float2 b4 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04641"></a><span class="lineno"> 4641</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04642"></a><span class="lineno"> 4642</span>&#160; float2 b5 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04643"></a><span class="lineno"> 4643</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04644"></a><span class="lineno"> 4644</span>&#160; float2 b6 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04645"></a><span class="lineno"> 4645</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04646"></a><span class="lineno"> 4646</span>&#160; float2 b7 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04647"></a><span class="lineno"> 4647</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04648"></a><span class="lineno"> 4648</span>&#160;</div><div class="line"><a name="l04649"></a><span class="lineno"> 4649</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04650"></a><span class="lineno"> 4650</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, acc0.s0);</div><div class="line"><a name="l04651"></a><span class="lineno"> 4651</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s0, acc0.s0);</div><div class="line"><a name="l04652"></a><span class="lineno"> 4652</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s0, acc0.s0);</div><div class="line"><a name="l04653"></a><span class="lineno"> 4653</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s0, acc0.s0);</div><div class="line"><a name="l04654"></a><span class="lineno"> 4654</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s0, acc0.s0);</div><div class="line"><a name="l04655"></a><span class="lineno"> 4655</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s0, acc0.s0);</div><div class="line"><a name="l04656"></a><span class="lineno"> 4656</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s0, acc0.s0);</div><div class="line"><a name="l04657"></a><span class="lineno"> 4657</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s0, acc0.s0);</div><div class="line"><a name="l04658"></a><span class="lineno"> 4658</span>&#160;</div><div class="line"><a name="l04659"></a><span class="lineno"> 4659</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, acc0.s1);</div><div class="line"><a name="l04660"></a><span class="lineno"> 4660</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s1, acc0.s1);</div><div class="line"><a name="l04661"></a><span class="lineno"> 4661</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s1, acc0.s1);</div><div class="line"><a name="l04662"></a><span class="lineno"> 4662</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s1, acc0.s1);</div><div class="line"><a name="l04663"></a><span class="lineno"> 4663</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s1, acc0.s1);</div><div class="line"><a name="l04664"></a><span class="lineno"> 4664</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s1, acc0.s1);</div><div class="line"><a name="l04665"></a><span class="lineno"> 4665</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s1, acc0.s1);</div><div class="line"><a name="l04666"></a><span class="lineno"> 4666</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s1, acc0.s1);</div><div class="line"><a name="l04667"></a><span class="lineno"> 4667</span>&#160;</div><div class="line"><a name="l04668"></a><span class="lineno"> 4668</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04669"></a><span class="lineno"> 4669</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04670"></a><span class="lineno"> 4670</span>&#160; a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y + zin.s1));</div><div class="line"><a name="l04671"></a><span class="lineno"> 4671</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04672"></a><span class="lineno"> 4672</span>&#160; a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l04673"></a><span class="lineno"> 4673</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04674"></a><span class="lineno"> 4674</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, acc1.s0);</div><div class="line"><a name="l04675"></a><span class="lineno"> 4675</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s0, acc1.s0);</div><div class="line"><a name="l04676"></a><span class="lineno"> 4676</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s0, acc1.s0);</div><div class="line"><a name="l04677"></a><span class="lineno"> 4677</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s0, acc1.s0);</div><div class="line"><a name="l04678"></a><span class="lineno"> 4678</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s0, acc1.s0);</div><div class="line"><a name="l04679"></a><span class="lineno"> 4679</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s0, acc1.s0);</div><div class="line"><a name="l04680"></a><span class="lineno"> 4680</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s0, acc1.s0);</div><div class="line"><a name="l04681"></a><span class="lineno"> 4681</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s0, acc1.s0);</div><div class="line"><a name="l04682"></a><span class="lineno"> 4682</span>&#160;</div><div class="line"><a name="l04683"></a><span class="lineno"> 4683</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, acc1.s1);</div><div class="line"><a name="l04684"></a><span class="lineno"> 4684</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s1, acc1.s1);</div><div class="line"><a name="l04685"></a><span class="lineno"> 4685</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s1, acc1.s1);</div><div class="line"><a name="l04686"></a><span class="lineno"> 4686</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s1, acc1.s1);</div><div class="line"><a name="l04687"></a><span class="lineno"> 4687</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s1, acc1.s1);</div><div class="line"><a name="l04688"></a><span class="lineno"> 4688</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s1, acc1.s1);</div><div class="line"><a name="l04689"></a><span class="lineno"> 4689</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s1, acc1.s1);</div><div class="line"><a name="l04690"></a><span class="lineno"> 4690</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s1, acc1.s1);</div><div class="line"><a name="l04691"></a><span class="lineno"> 4691</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04692"></a><span class="lineno"> 4692</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04693"></a><span class="lineno"> 4693</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04694"></a><span class="lineno"> 4694</span>&#160; a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y + zin.s2));</div><div class="line"><a name="l04695"></a><span class="lineno"> 4695</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04696"></a><span class="lineno"> 4696</span>&#160; a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l04697"></a><span class="lineno"> 4697</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04698"></a><span class="lineno"> 4698</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, acc2.s0);</div><div class="line"><a name="l04699"></a><span class="lineno"> 4699</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s0, acc2.s0);</div><div class="line"><a name="l04700"></a><span class="lineno"> 4700</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s0, acc2.s0);</div><div class="line"><a name="l04701"></a><span class="lineno"> 4701</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s0, acc2.s0);</div><div class="line"><a name="l04702"></a><span class="lineno"> 4702</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s0, acc2.s0);</div><div class="line"><a name="l04703"></a><span class="lineno"> 4703</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s0, acc2.s0);</div><div class="line"><a name="l04704"></a><span class="lineno"> 4704</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s0, acc2.s0);</div><div class="line"><a name="l04705"></a><span class="lineno"> 4705</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s0, acc2.s0);</div><div class="line"><a name="l04706"></a><span class="lineno"> 4706</span>&#160;</div><div class="line"><a name="l04707"></a><span class="lineno"> 4707</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, acc2.s1);</div><div class="line"><a name="l04708"></a><span class="lineno"> 4708</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s1, acc2.s1);</div><div class="line"><a name="l04709"></a><span class="lineno"> 4709</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s1, acc2.s1);</div><div class="line"><a name="l04710"></a><span class="lineno"> 4710</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s1, acc2.s1);</div><div class="line"><a name="l04711"></a><span class="lineno"> 4711</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s1, acc2.s1);</div><div class="line"><a name="l04712"></a><span class="lineno"> 4712</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s1, acc2.s1);</div><div class="line"><a name="l04713"></a><span class="lineno"> 4713</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s1, acc2.s1);</div><div class="line"><a name="l04714"></a><span class="lineno"> 4714</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s1, acc2.s1);</div><div class="line"><a name="l04715"></a><span class="lineno"> 4715</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04716"></a><span class="lineno"> 4716</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04717"></a><span class="lineno"> 4717</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04718"></a><span class="lineno"> 4718</span>&#160; a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y + zin.s3));</div><div class="line"><a name="l04719"></a><span class="lineno"> 4719</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04720"></a><span class="lineno"> 4720</span>&#160; a0 = vload8(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l04721"></a><span class="lineno"> 4721</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04722"></a><span class="lineno"> 4722</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s0, acc3.s0);</div><div class="line"><a name="l04723"></a><span class="lineno"> 4723</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s0, acc3.s0);</div><div class="line"><a name="l04724"></a><span class="lineno"> 4724</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s0, acc3.s0);</div><div class="line"><a name="l04725"></a><span class="lineno"> 4725</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s0, acc3.s0);</div><div class="line"><a name="l04726"></a><span class="lineno"> 4726</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s0, acc3.s0);</div><div class="line"><a name="l04727"></a><span class="lineno"> 4727</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s0, acc3.s0);</div><div class="line"><a name="l04728"></a><span class="lineno"> 4728</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s0, acc3.s0);</div><div class="line"><a name="l04729"></a><span class="lineno"> 4729</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s0, acc3.s0);</div><div class="line"><a name="l04730"></a><span class="lineno"> 4730</span>&#160;</div><div class="line"><a name="l04731"></a><span class="lineno"> 4731</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s0, b0.s1, acc3.s1);</div><div class="line"><a name="l04732"></a><span class="lineno"> 4732</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s1, b1.s1, acc3.s1);</div><div class="line"><a name="l04733"></a><span class="lineno"> 4733</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s2, b2.s1, acc3.s1);</div><div class="line"><a name="l04734"></a><span class="lineno"> 4734</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s3, b3.s1, acc3.s1);</div><div class="line"><a name="l04735"></a><span class="lineno"> 4735</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s4, b4.s1, acc3.s1);</div><div class="line"><a name="l04736"></a><span class="lineno"> 4736</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s5, b5.s1, acc3.s1);</div><div class="line"><a name="l04737"></a><span class="lineno"> 4737</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s6, b6.s1, acc3.s1);</div><div class="line"><a name="l04738"></a><span class="lineno"> 4738</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0.s7, b7.s1, acc3.s1);</div><div class="line"><a name="l04739"></a><span class="lineno"> 4739</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04740"></a><span class="lineno"> 4740</span>&#160;</div><div class="line"><a name="l04741"></a><span class="lineno"> 4741</span>&#160; src_addr.s0 += <span class="keyword">sizeof</span>(float) * 8;</div><div class="line"><a name="l04742"></a><span class="lineno"> 4742</span>&#160; }</div><div class="line"><a name="l04743"></a><span class="lineno"> 4743</span>&#160; <span class="comment">// float size increment</span></div><div class="line"><a name="l04744"></a><span class="lineno"> 4744</span>&#160; <span class="keywordflow">for</span>(; i &lt; (int)COLS_A; ++i)</div><div class="line"><a name="l04745"></a><span class="lineno"> 4745</span>&#160; {</div><div class="line"><a name="l04746"></a><span class="lineno"> 4746</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04747"></a><span class="lineno"> 4747</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l04748"></a><span class="lineno"> 4748</span>&#160; <span class="keywordtype">float</span> a0 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y + zin.s0));</div><div class="line"><a name="l04749"></a><span class="lineno"> 4749</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04750"></a><span class="lineno"> 4750</span>&#160; <span class="keywordtype">float</span> a1 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y + zin.s1));</div><div class="line"><a name="l04751"></a><span class="lineno"> 4751</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04752"></a><span class="lineno"> 4752</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04753"></a><span class="lineno"> 4753</span>&#160; <span class="keywordtype">float</span> a2 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y + zin.s2));</div><div class="line"><a name="l04754"></a><span class="lineno"> 4754</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04755"></a><span class="lineno"> 4755</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04756"></a><span class="lineno"> 4756</span>&#160; <span class="keywordtype">float</span> a3 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y + zin.s3));</div><div class="line"><a name="l04757"></a><span class="lineno"> 4757</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04758"></a><span class="lineno"> 4758</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04759"></a><span class="lineno"> 4759</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l04760"></a><span class="lineno"> 4760</span>&#160; <span class="keywordtype">float</span> a0 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l04761"></a><span class="lineno"> 4761</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04762"></a><span class="lineno"> 4762</span>&#160; <span class="keywordtype">float</span> a1 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l04763"></a><span class="lineno"> 4763</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04764"></a><span class="lineno"> 4764</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04765"></a><span class="lineno"> 4765</span>&#160; <span class="keywordtype">float</span> a2 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l04766"></a><span class="lineno"> 4766</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04767"></a><span class="lineno"> 4767</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04768"></a><span class="lineno"> 4768</span>&#160; <span class="keywordtype">float</span> a3 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l04769"></a><span class="lineno"> 4769</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04770"></a><span class="lineno"> 4770</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04771"></a><span class="lineno"> 4771</span>&#160;</div><div class="line"><a name="l04772"></a><span class="lineno"> 4772</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l04773"></a><span class="lineno"> 4773</span>&#160; float2 b0 = vload2(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l04774"></a><span class="lineno"> 4774</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l04775"></a><span class="lineno"> 4775</span>&#160;</div><div class="line"><a name="l04776"></a><span class="lineno"> 4776</span>&#160; <span class="comment">// Multiply and accumulate</span></div><div class="line"><a name="l04777"></a><span class="lineno"> 4777</span>&#160; acc0.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0, b0.s0, acc0.s0);</div><div class="line"><a name="l04778"></a><span class="lineno"> 4778</span>&#160; acc0.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a0, b0.s1, acc0.s1);</div><div class="line"><a name="l04779"></a><span class="lineno"> 4779</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04780"></a><span class="lineno"> 4780</span>&#160; acc1.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1, b0.s0, acc1.s0);</div><div class="line"><a name="l04781"></a><span class="lineno"> 4781</span>&#160; acc1.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a1, b0.s1, acc1.s1);</div><div class="line"><a name="l04782"></a><span class="lineno"> 4782</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04783"></a><span class="lineno"> 4783</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04784"></a><span class="lineno"> 4784</span>&#160; acc2.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2, b0.s0, acc2.s0);</div><div class="line"><a name="l04785"></a><span class="lineno"> 4785</span>&#160; acc2.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a2, b0.s1, acc2.s1);</div><div class="line"><a name="l04786"></a><span class="lineno"> 4786</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04787"></a><span class="lineno"> 4787</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04788"></a><span class="lineno"> 4788</span>&#160; acc3.s0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3, b0.s0, acc3.s0);</div><div class="line"><a name="l04789"></a><span class="lineno"> 4789</span>&#160; acc3.s1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(a3, b0.s1, acc3.s1);</div><div class="line"><a name="l04790"></a><span class="lineno"> 4790</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04791"></a><span class="lineno"> 4791</span>&#160;</div><div class="line"><a name="l04792"></a><span class="lineno"> 4792</span>&#160; src_addr.s0 += <span class="keyword">sizeof</span>(float);</div><div class="line"><a name="l04793"></a><span class="lineno"> 4793</span>&#160; }</div><div class="line"><a name="l04794"></a><span class="lineno"> 4794</span>&#160;</div><div class="line"><a name="l04795"></a><span class="lineno"> 4795</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l04796"></a><span class="lineno"> 4796</span>&#160;</div><div class="line"><a name="l04797"></a><span class="lineno"> 4797</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l04798"></a><span class="lineno"> 4798</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l04799"></a><span class="lineno"> 4799</span>&#160;</div><div class="line"><a name="l04800"></a><span class="lineno"> 4800</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l04801"></a><span class="lineno"> 4801</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l04802"></a><span class="lineno"> 4802</span>&#160;</div><div class="line"><a name="l04803"></a><span class="lineno"> 4803</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l04804"></a><span class="lineno"> 4804</span>&#160;</div><div class="line"><a name="l04805"></a><span class="lineno"> 4805</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l04806"></a><span class="lineno"> 4806</span>&#160;</div><div class="line"><a name="l04807"></a><span class="lineno"> 4807</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l04808"></a><span class="lineno"> 4808</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l04809"></a><span class="lineno"> 4809</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l04810"></a><span class="lineno"> 4810</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04811"></a><span class="lineno"> 4811</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l04812"></a><span class="lineno"> 4812</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04813"></a><span class="lineno"> 4813</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04814"></a><span class="lineno"> 4814</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04815"></a><span class="lineno"> 4815</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l04816"></a><span class="lineno"> 4816</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04817"></a><span class="lineno"> 4817</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04818"></a><span class="lineno"> 4818</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l04819"></a><span class="lineno"> 4819</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04820"></a><span class="lineno"> 4820</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04821"></a><span class="lineno"> 4821</span>&#160;</div><div class="line"><a name="l04822"></a><span class="lineno"> 4822</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l04823"></a><span class="lineno"> 4823</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l04824"></a><span class="lineno"> 4824</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l04825"></a><span class="lineno"> 4825</span>&#160;</div><div class="line"><a name="l04826"></a><span class="lineno"> 4826</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l04827"></a><span class="lineno"> 4827</span>&#160; zout *= (dst_cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l04828"></a><span class="lineno"> 4828</span>&#160;</div><div class="line"><a name="l04829"></a><span class="lineno"> 4829</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l04830"></a><span class="lineno"> 4830</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l04831"></a><span class="lineno"> 4831</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l04832"></a><span class="lineno"> 4832</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l04833"></a><span class="lineno"> 4833</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l04834"></a><span class="lineno"> 4834</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l04835"></a><span class="lineno"> 4835</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l04836"></a><span class="lineno"> 4836</span>&#160;</div><div class="line"><a name="l04837"></a><span class="lineno"> 4837</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l04838"></a><span class="lineno"> 4838</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l04839"></a><span class="lineno"> 4839</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <span class="keywordtype">float</span>, acc, ALPHA);</div><div class="line"><a name="l04840"></a><span class="lineno"> 4840</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l04841"></a><span class="lineno"> 4841</span>&#160;</div><div class="line"><a name="l04842"></a><span class="lineno"> 4842</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l04843"></a><span class="lineno"> 4843</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l04844"></a><span class="lineno"> 4844</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, uint, zero, 0);</div><div class="line"><a name="l04845"></a><span class="lineno"> 4845</span>&#160;</div><div class="line"><a name="l04846"></a><span class="lineno"> 4846</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04847"></a><span class="lineno"> 4847</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)2 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>));</div><div class="line"><a name="l04848"></a><span class="lineno"> 4848</span>&#160;</div><div class="line"><a name="l04849"></a><span class="lineno"> 4849</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 2, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l04850"></a><span class="lineno"> 4850</span>&#160;</div><div class="line"><a name="l04851"></a><span class="lineno"> 4851</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l04852"></a><span class="lineno"> 4852</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l04853"></a><span class="lineno"> 4853</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l04854"></a><span class="lineno"> 4854</span>&#160;</div><div class="line"><a name="l04855"></a><span class="lineno"> 4855</span>&#160; <span class="comment">// acc = acc + bias[broadcasted]</span></div><div class="line"><a name="l04856"></a><span class="lineno"> 4856</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, bias0);</div><div class="line"><a name="l04857"></a><span class="lineno"> 4857</span>&#160;</div><div class="line"><a name="l04858"></a><span class="lineno"> 4858</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04859"></a><span class="lineno"> 4859</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)2 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>)) + (get_global_id(1) *</div><div class="line"><a name="l04860"></a><span class="lineno"> 4860</span>&#160; (uint)NUM_ELEMS_PROCESSED_PER_THREAD_Y * src2_stride_y) + get_global_id(2) * src2_stride_z;</div><div class="line"><a name="l04861"></a><span class="lineno"> 4861</span>&#160;</div><div class="line"><a name="l04862"></a><span class="lineno"> 4862</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 2, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l04863"></a><span class="lineno"> 4863</span>&#160;</div><div class="line"><a name="l04864"></a><span class="lineno"> 4864</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l04865"></a><span class="lineno"> 4865</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <span class="keywordtype">float</span>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l04866"></a><span class="lineno"> 4866</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l04867"></a><span class="lineno"> 4867</span>&#160;</div><div class="line"><a name="l04868"></a><span class="lineno"> 4868</span>&#160; <span class="comment">// acc = acc + bias</span></div><div class="line"><a name="l04869"></a><span class="lineno"> 4869</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l04870"></a><span class="lineno"> 4870</span>&#160;</div><div class="line"><a name="l04871"></a><span class="lineno"> 4871</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l04872"></a><span class="lineno"> 4872</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l04873"></a><span class="lineno"> 4873</span>&#160;</div><div class="line"><a name="l04874"></a><span class="lineno"> 4874</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l04875"></a><span class="lineno"> 4875</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, ACTIVATION_TYPE, <span class="keywordtype">float</span>, acc, A_VAL, B_VAL);</div><div class="line"><a name="l04876"></a><span class="lineno"> 4876</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l04877"></a><span class="lineno"> 4877</span>&#160;</div><div class="line"><a name="l04878"></a><span class="lineno"> 4878</span>&#160; <span class="comment">// Store the output block</span></div><div class="line"><a name="l04879"></a><span class="lineno"> 4879</span>&#160; vstore2(acc0, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 0 * dst_stride_y + zout.s0));</div><div class="line"><a name="l04880"></a><span class="lineno"> 4880</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04881"></a><span class="lineno"> 4881</span>&#160; vstore2(acc1, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 1 * dst_stride_y + zout.s1));</div><div class="line"><a name="l04882"></a><span class="lineno"> 4882</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l04883"></a><span class="lineno"> 4883</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04884"></a><span class="lineno"> 4884</span>&#160; vstore2(acc2, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 2 * dst_stride_y + zout.s2));</div><div class="line"><a name="l04885"></a><span class="lineno"> 4885</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l04886"></a><span class="lineno"> 4886</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04887"></a><span class="lineno"> 4887</span>&#160; vstore2(acc3, 0, (__global <span class="keywordtype">float</span> *)(dst_addr + 3 * dst_stride_y + zout.s3));</div><div class="line"><a name="l04888"></a><span class="lineno"> 4888</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l04889"></a><span class="lineno"> 4889</span>&#160;}</div><div class="line"><a name="l04890"></a><span class="lineno"> 4890</span>&#160;</div><div class="line"><a name="l04891"></a><span class="lineno"> 4891</span>&#160;<span class="preprocessor">#if defined(ARM_COMPUTE_OPENCL_FP16_ENABLED)</span></div><div class="line"><a name="l04892"></a><span class="lineno"> 4892</span>&#160;</div><div class="line"><a name="l04942"></a><span class="lineno"> 4942</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_floating_point_f16_bifrost_acc32(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l04943"></a><span class="lineno"> 4943</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l04944"></a><span class="lineno"> 4944</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l04945"></a><span class="lineno"> 4945</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l04946"></a><span class="lineno"> 4946</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l04947"></a><span class="lineno"> 4947</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l04948"></a><span class="lineno"> 4948</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l04949"></a><span class="lineno"> 4949</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l04950"></a><span class="lineno"> 4950</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l04951"></a><span class="lineno"> 4951</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l04952"></a><span class="lineno"> 4952</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l04953"></a><span class="lineno"> 4953</span>&#160; uint dst_stride_z</div><div class="line"><a name="l04954"></a><span class="lineno"> 4954</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l04955"></a><span class="lineno"> 4955</span>&#160; ,</div><div class="line"><a name="l04956"></a><span class="lineno"> 4956</span>&#160; uint src_cross_plane_pad</div><div class="line"><a name="l04957"></a><span class="lineno"> 4957</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l04958"></a><span class="lineno"> 4958</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l04959"></a><span class="lineno"> 4959</span>&#160; ,</div><div class="line"><a name="l04960"></a><span class="lineno"> 4960</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l04961"></a><span class="lineno"> 4961</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l04962"></a><span class="lineno"> 4962</span>&#160; )</div><div class="line"><a name="l04963"></a><span class="lineno"> 4963</span>&#160;{</div><div class="line"><a name="l04964"></a><span class="lineno"> 4964</span>&#160; <span class="keywordtype">int</span> idx = get_global_id(0) * NUM_ELEMS_PROCESSED_PER_THREAD_X;</div><div class="line"><a name="l04965"></a><span class="lineno"> 4965</span>&#160;</div><div class="line"><a name="l04966"></a><span class="lineno"> 4966</span>&#160; <span class="comment">// Compute starting address for matrix A and Matrix B</span></div><div class="line"><a name="l04967"></a><span class="lineno"> 4967</span>&#160; int2 src_addr = ((int2)(src0_offset_first_element_in_bytes, src1_offset_first_element_in_bytes));</div><div class="line"><a name="l04968"></a><span class="lineno"> 4968</span>&#160;</div><div class="line"><a name="l04969"></a><span class="lineno"> 4969</span>&#160; <span class="comment">// Update address for the matrix A</span></div><div class="line"><a name="l04970"></a><span class="lineno"> 4970</span>&#160; src_addr.s0 += get_global_id(1) * src0_stride_y * NUM_ELEMS_PROCESSED_PER_THREAD_Y;</div><div class="line"><a name="l04971"></a><span class="lineno"> 4971</span>&#160;</div><div class="line"><a name="l04972"></a><span class="lineno"> 4972</span>&#160; <span class="comment">// Update address for the matrix B</span></div><div class="line"><a name="l04973"></a><span class="lineno"> 4973</span>&#160; src_addr.s1 += idx * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>);</div><div class="line"><a name="l04974"></a><span class="lineno"> 4974</span>&#160;</div><div class="line"><a name="l04975"></a><span class="lineno"> 4975</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l04976"></a><span class="lineno"> 4976</span>&#160; <span class="comment">// Since we load a 2D input tile from a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l04977"></a><span class="lineno"> 4977</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l04978"></a><span class="lineno"> 4978</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l04979"></a><span class="lineno"> 4979</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04980"></a><span class="lineno"> 4980</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l04981"></a><span class="lineno"> 4981</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04982"></a><span class="lineno"> 4982</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04983"></a><span class="lineno"> 4983</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04984"></a><span class="lineno"> 4984</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l04985"></a><span class="lineno"> 4985</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l04986"></a><span class="lineno"> 4986</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04987"></a><span class="lineno"> 4987</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l04988"></a><span class="lineno"> 4988</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l04989"></a><span class="lineno"> 4989</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l04990"></a><span class="lineno"> 4990</span>&#160;</div><div class="line"><a name="l04991"></a><span class="lineno"> 4991</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l04992"></a><span class="lineno"> 4992</span>&#160; uint4 zin = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l04993"></a><span class="lineno"> 4993</span>&#160; zin = min(DEPTH_GEMM3D - 1, zin);</div><div class="line"><a name="l04994"></a><span class="lineno"> 4994</span>&#160;</div><div class="line"><a name="l04995"></a><span class="lineno"> 4995</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l04996"></a><span class="lineno"> 4996</span>&#160; zin *= (src_cross_plane_pad * src0_stride_y);</div><div class="line"><a name="l04997"></a><span class="lineno"> 4997</span>&#160;</div><div class="line"><a name="l04998"></a><span class="lineno"> 4998</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l04999"></a><span class="lineno"> 4999</span>&#160; <span class="comment">// multiply src0_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l05000"></a><span class="lineno"> 5000</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l05001"></a><span class="lineno"> 5001</span>&#160;</div><div class="line"><a name="l05002"></a><span class="lineno"> 5002</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05003"></a><span class="lineno"> 5003</span>&#160;</div><div class="line"><a name="l05004"></a><span class="lineno"> 5004</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l05005"></a><span class="lineno"> 5005</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z;</div><div class="line"><a name="l05006"></a><span class="lineno"> 5006</span>&#160;</div><div class="line"><a name="l05007"></a><span class="lineno"> 5007</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05008"></a><span class="lineno"> 5008</span>&#160;</div><div class="line"><a name="l05009"></a><span class="lineno"> 5009</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l05010"></a><span class="lineno"> 5010</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l05011"></a><span class="lineno"> 5011</span>&#160; src_addr.s1 += (get_global_id(2) % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l05012"></a><span class="lineno"> 5012</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l05013"></a><span class="lineno"> 5013</span>&#160; src_addr.s1 += get_global_id(2) * src1_stride_z;</div><div class="line"><a name="l05014"></a><span class="lineno"> 5014</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l05015"></a><span class="lineno"> 5015</span>&#160;</div><div class="line"><a name="l05016"></a><span class="lineno"> 5016</span>&#160; float8 acc0 = 0.0h;</div><div class="line"><a name="l05017"></a><span class="lineno"> 5017</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05018"></a><span class="lineno"> 5018</span>&#160; float8 acc1 = 0.0h;</div><div class="line"><a name="l05019"></a><span class="lineno"> 5019</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05020"></a><span class="lineno"> 5020</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05021"></a><span class="lineno"> 5021</span>&#160; float8 acc2 = 0.0h;</div><div class="line"><a name="l05022"></a><span class="lineno"> 5022</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05023"></a><span class="lineno"> 5023</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05024"></a><span class="lineno"> 5024</span>&#160; float8 acc3 = 0.0h;</div><div class="line"><a name="l05025"></a><span class="lineno"> 5025</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05026"></a><span class="lineno"> 5026</span>&#160;</div><div class="line"><a name="l05027"></a><span class="lineno"> 5027</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l05028"></a><span class="lineno"> 5028</span>&#160; <span class="keywordflow">for</span>(; i &lt;= ((int)COLS_A - 4); i += 4)</div><div class="line"><a name="l05029"></a><span class="lineno"> 5029</span>&#160; {</div><div class="line"><a name="l05030"></a><span class="lineno"> 5030</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05031"></a><span class="lineno"> 5031</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05032"></a><span class="lineno"> 5032</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 4, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, a, src0_ptr, src_addr.s0, src0_stride_y, zin.s);</div><div class="line"><a name="l05033"></a><span class="lineno"> 5033</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05034"></a><span class="lineno"> 5034</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05035"></a><span class="lineno"> 5035</span>&#160; half4 a0 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l05036"></a><span class="lineno"> 5036</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05037"></a><span class="lineno"> 5037</span>&#160; half4 a1 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l05038"></a><span class="lineno"> 5038</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05039"></a><span class="lineno"> 5039</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05040"></a><span class="lineno"> 5040</span>&#160; half4 a2 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l05041"></a><span class="lineno"> 5041</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05042"></a><span class="lineno"> 5042</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05043"></a><span class="lineno"> 5043</span>&#160; half4 a3 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l05044"></a><span class="lineno"> 5044</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05045"></a><span class="lineno"> 5045</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05046"></a><span class="lineno"> 5046</span>&#160;</div><div class="line"><a name="l05047"></a><span class="lineno"> 5047</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l05048"></a><span class="lineno"> 5048</span>&#160; float8 b0 = convert_float8(vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1)));</div><div class="line"><a name="l05049"></a><span class="lineno"> 5049</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05050"></a><span class="lineno"> 5050</span>&#160;</div><div class="line"><a name="l05051"></a><span class="lineno"> 5051</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l05052"></a><span class="lineno"> 5052</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a0.s0, acc0);</div><div class="line"><a name="l05053"></a><span class="lineno"> 5053</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05054"></a><span class="lineno"> 5054</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a1.s0, acc1);</div><div class="line"><a name="l05055"></a><span class="lineno"> 5055</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05056"></a><span class="lineno"> 5056</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05057"></a><span class="lineno"> 5057</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a2.s0, acc2);</div><div class="line"><a name="l05058"></a><span class="lineno"> 5058</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05059"></a><span class="lineno"> 5059</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05060"></a><span class="lineno"> 5060</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a3.s0, acc3);</div><div class="line"><a name="l05061"></a><span class="lineno"> 5061</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05062"></a><span class="lineno"> 5062</span>&#160;</div><div class="line"><a name="l05063"></a><span class="lineno"> 5063</span>&#160; b0 = convert_float8(vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1)));</div><div class="line"><a name="l05064"></a><span class="lineno"> 5064</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05065"></a><span class="lineno"> 5065</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a0.s1, acc0);</div><div class="line"><a name="l05066"></a><span class="lineno"> 5066</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05067"></a><span class="lineno"> 5067</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a1.s1, acc1);</div><div class="line"><a name="l05068"></a><span class="lineno"> 5068</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05069"></a><span class="lineno"> 5069</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05070"></a><span class="lineno"> 5070</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a2.s1, acc2);</div><div class="line"><a name="l05071"></a><span class="lineno"> 5071</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05072"></a><span class="lineno"> 5072</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05073"></a><span class="lineno"> 5073</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a3.s1, acc3);</div><div class="line"><a name="l05074"></a><span class="lineno"> 5074</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05075"></a><span class="lineno"> 5075</span>&#160;</div><div class="line"><a name="l05076"></a><span class="lineno"> 5076</span>&#160; b0 = convert_float8(vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1)));</div><div class="line"><a name="l05077"></a><span class="lineno"> 5077</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05078"></a><span class="lineno"> 5078</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a0.s2, acc0);</div><div class="line"><a name="l05079"></a><span class="lineno"> 5079</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05080"></a><span class="lineno"> 5080</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a1.s2, acc1);</div><div class="line"><a name="l05081"></a><span class="lineno"> 5081</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05082"></a><span class="lineno"> 5082</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05083"></a><span class="lineno"> 5083</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a2.s2, acc2);</div><div class="line"><a name="l05084"></a><span class="lineno"> 5084</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05085"></a><span class="lineno"> 5085</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05086"></a><span class="lineno"> 5086</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a3.s2, acc3);</div><div class="line"><a name="l05087"></a><span class="lineno"> 5087</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05088"></a><span class="lineno"> 5088</span>&#160;</div><div class="line"><a name="l05089"></a><span class="lineno"> 5089</span>&#160; b0 = convert_float8(vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1)));</div><div class="line"><a name="l05090"></a><span class="lineno"> 5090</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05091"></a><span class="lineno"> 5091</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a0.s3, acc0);</div><div class="line"><a name="l05092"></a><span class="lineno"> 5092</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05093"></a><span class="lineno"> 5093</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a1.s3, acc1);</div><div class="line"><a name="l05094"></a><span class="lineno"> 5094</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05095"></a><span class="lineno"> 5095</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05096"></a><span class="lineno"> 5096</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a2.s3, acc2);</div><div class="line"><a name="l05097"></a><span class="lineno"> 5097</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05098"></a><span class="lineno"> 5098</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05099"></a><span class="lineno"> 5099</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a3.s3, acc3);</div><div class="line"><a name="l05100"></a><span class="lineno"> 5100</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05101"></a><span class="lineno"> 5101</span>&#160;</div><div class="line"><a name="l05102"></a><span class="lineno"> 5102</span>&#160; src_addr.s0 += 4 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>);</div><div class="line"><a name="l05103"></a><span class="lineno"> 5103</span>&#160; }</div><div class="line"><a name="l05104"></a><span class="lineno"> 5104</span>&#160;</div><div class="line"><a name="l05105"></a><span class="lineno"> 5105</span>&#160; <span class="keywordflow">for</span>(; i &lt; (int)COLS_A; ++i)</div><div class="line"><a name="l05106"></a><span class="lineno"> 5106</span>&#160; {</div><div class="line"><a name="l05107"></a><span class="lineno"> 5107</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05108"></a><span class="lineno"> 5108</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05109"></a><span class="lineno"> 5109</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a0 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y + zin.s0));</div><div class="line"><a name="l05110"></a><span class="lineno"> 5110</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05111"></a><span class="lineno"> 5111</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a1 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y + zin.s1));</div><div class="line"><a name="l05112"></a><span class="lineno"> 5112</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05113"></a><span class="lineno"> 5113</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05114"></a><span class="lineno"> 5114</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a2 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y + zin.s2));</div><div class="line"><a name="l05115"></a><span class="lineno"> 5115</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05116"></a><span class="lineno"> 5116</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05117"></a><span class="lineno"> 5117</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a3 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y + zin.s3));</div><div class="line"><a name="l05118"></a><span class="lineno"> 5118</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05119"></a><span class="lineno"> 5119</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05120"></a><span class="lineno"> 5120</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05121"></a><span class="lineno"> 5121</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a0 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l05122"></a><span class="lineno"> 5122</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05123"></a><span class="lineno"> 5123</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a1 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l05124"></a><span class="lineno"> 5124</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05125"></a><span class="lineno"> 5125</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05126"></a><span class="lineno"> 5126</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a2 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l05127"></a><span class="lineno"> 5127</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05128"></a><span class="lineno"> 5128</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05129"></a><span class="lineno"> 5129</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a3 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l05130"></a><span class="lineno"> 5130</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05131"></a><span class="lineno"> 5131</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05132"></a><span class="lineno"> 5132</span>&#160;</div><div class="line"><a name="l05133"></a><span class="lineno"> 5133</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l05134"></a><span class="lineno"> 5134</span>&#160; float8 b0 = convert_float8(vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1)));</div><div class="line"><a name="l05135"></a><span class="lineno"> 5135</span>&#160;</div><div class="line"><a name="l05136"></a><span class="lineno"> 5136</span>&#160; src_addr += (int2)(<span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>), src1_stride_y);</div><div class="line"><a name="l05137"></a><span class="lineno"> 5137</span>&#160;</div><div class="line"><a name="l05138"></a><span class="lineno"> 5138</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l05139"></a><span class="lineno"> 5139</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a0, acc0); <span class="comment">// b0 * (half8)a0;</span></div><div class="line"><a name="l05140"></a><span class="lineno"> 5140</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05141"></a><span class="lineno"> 5141</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a1, acc1); <span class="comment">// b0 * (half8)a1;</span></div><div class="line"><a name="l05142"></a><span class="lineno"> 5142</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05143"></a><span class="lineno"> 5143</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05144"></a><span class="lineno"> 5144</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a2, acc2); <span class="comment">// b0 * (half8)a2;</span></div><div class="line"><a name="l05145"></a><span class="lineno"> 5145</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05146"></a><span class="lineno"> 5146</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05147"></a><span class="lineno"> 5147</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (float8)a3, acc3); <span class="comment">// b0 * (half8)a3;</span></div><div class="line"><a name="l05148"></a><span class="lineno"> 5148</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05149"></a><span class="lineno"> 5149</span>&#160; }</div><div class="line"><a name="l05150"></a><span class="lineno"> 5150</span>&#160;</div><div class="line"><a name="l05151"></a><span class="lineno"> 5151</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l05152"></a><span class="lineno"> 5152</span>&#160;</div><div class="line"><a name="l05153"></a><span class="lineno"> 5153</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l05154"></a><span class="lineno"> 5154</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l05155"></a><span class="lineno"> 5155</span>&#160;</div><div class="line"><a name="l05156"></a><span class="lineno"> 5156</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l05157"></a><span class="lineno"> 5157</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l05158"></a><span class="lineno"> 5158</span>&#160;</div><div class="line"><a name="l05159"></a><span class="lineno"> 5159</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l05160"></a><span class="lineno"> 5160</span>&#160;</div><div class="line"><a name="l05161"></a><span class="lineno"> 5161</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l05162"></a><span class="lineno"> 5162</span>&#160;</div><div class="line"><a name="l05163"></a><span class="lineno"> 5163</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l05164"></a><span class="lineno"> 5164</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l05165"></a><span class="lineno"> 5165</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l05166"></a><span class="lineno"> 5166</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05167"></a><span class="lineno"> 5167</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l05168"></a><span class="lineno"> 5168</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05169"></a><span class="lineno"> 5169</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l05170"></a><span class="lineno"> 5170</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l05171"></a><span class="lineno"> 5171</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l05172"></a><span class="lineno"> 5172</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l05173"></a><span class="lineno"> 5173</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05174"></a><span class="lineno"> 5174</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l05175"></a><span class="lineno"> 5175</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05176"></a><span class="lineno"> 5176</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l05177"></a><span class="lineno"> 5177</span>&#160;</div><div class="line"><a name="l05178"></a><span class="lineno"> 5178</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l05179"></a><span class="lineno"> 5179</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l05180"></a><span class="lineno"> 5180</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l05181"></a><span class="lineno"> 5181</span>&#160;</div><div class="line"><a name="l05182"></a><span class="lineno"> 5182</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l05183"></a><span class="lineno"> 5183</span>&#160; zout *= (dst_cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l05184"></a><span class="lineno"> 5184</span>&#160;</div><div class="line"><a name="l05185"></a><span class="lineno"> 5185</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l05186"></a><span class="lineno"> 5186</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l05187"></a><span class="lineno"> 5187</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l05188"></a><span class="lineno"> 5188</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l05189"></a><span class="lineno"> 5189</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l05190"></a><span class="lineno"> 5190</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l05191"></a><span class="lineno"> 5191</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l05192"></a><span class="lineno"> 5192</span>&#160;</div><div class="line"><a name="l05193"></a><span class="lineno"> 5193</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l05194"></a><span class="lineno"> 5194</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l05195"></a><span class="lineno"> 5195</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <span class="keywordtype">float</span>, acc, ALPHA);</div><div class="line"><a name="l05196"></a><span class="lineno"> 5196</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l05197"></a><span class="lineno"> 5197</span>&#160;</div><div class="line"><a name="l05198"></a><span class="lineno"> 5198</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l05199"></a><span class="lineno"> 5199</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, uint, zero, 0);</div><div class="line"><a name="l05200"></a><span class="lineno"> 5200</span>&#160;</div><div class="line"><a name="l05201"></a><span class="lineno"> 5201</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l05202"></a><span class="lineno"> 5202</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>));</div><div class="line"><a name="l05203"></a><span class="lineno"> 5203</span>&#160;</div><div class="line"><a name="l05204"></a><span class="lineno"> 5204</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l05205"></a><span class="lineno"> 5205</span>&#160;</div><div class="line"><a name="l05206"></a><span class="lineno"> 5206</span>&#160; float8 bias_f0 = convert_float8(bias0);</div><div class="line"><a name="l05207"></a><span class="lineno"> 5207</span>&#160;</div><div class="line"><a name="l05208"></a><span class="lineno"> 5208</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l05209"></a><span class="lineno"> 5209</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <span class="keywordtype">float</span>, bias_f, BETA);</div><div class="line"><a name="l05210"></a><span class="lineno"> 5210</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l05211"></a><span class="lineno"> 5211</span>&#160;</div><div class="line"><a name="l05212"></a><span class="lineno"> 5212</span>&#160; <span class="comment">// acc = acc + bias[broadcasted]</span></div><div class="line"><a name="l05213"></a><span class="lineno"> 5213</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, bias_f0);</div><div class="line"><a name="l05214"></a><span class="lineno"> 5214</span>&#160;</div><div class="line"><a name="l05215"></a><span class="lineno"> 5215</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l05216"></a><span class="lineno"> 5216</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>)) + (get_global_id(1) *</div><div class="line"><a name="l05217"></a><span class="lineno"> 5217</span>&#160; (uint)NUM_ELEMS_PROCESSED_PER_THREAD_Y * src2_stride_y) + get_global_id(2) * src2_stride_z;</div><div class="line"><a name="l05218"></a><span class="lineno"> 5218</span>&#160;</div><div class="line"><a name="l05219"></a><span class="lineno"> 5219</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l05220"></a><span class="lineno"> 5220</span>&#160;</div><div class="line"><a name="l05221"></a><span class="lineno"> 5221</span>&#160; float8 bias_f0 = convert_float8(bias0);</div><div class="line"><a name="l05222"></a><span class="lineno"> 5222</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05223"></a><span class="lineno"> 5223</span>&#160; float8 bias_f1 = convert_float8(bias1);</div><div class="line"><a name="l05224"></a><span class="lineno"> 5224</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05225"></a><span class="lineno"> 5225</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05226"></a><span class="lineno"> 5226</span>&#160; float8 bias_f2 = convert_float8(bias2);</div><div class="line"><a name="l05227"></a><span class="lineno"> 5227</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05228"></a><span class="lineno"> 5228</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05229"></a><span class="lineno"> 5229</span>&#160; float8 bias_f3 = convert_float8(bias3);</div><div class="line"><a name="l05230"></a><span class="lineno"> 5230</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05231"></a><span class="lineno"> 5231</span>&#160;</div><div class="line"><a name="l05232"></a><span class="lineno"> 5232</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l05233"></a><span class="lineno"> 5233</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <span class="keywordtype">float</span>, bias_f, BETA);</div><div class="line"><a name="l05234"></a><span class="lineno"> 5234</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l05235"></a><span class="lineno"> 5235</span>&#160;</div><div class="line"><a name="l05236"></a><span class="lineno"> 5236</span>&#160; <span class="comment">// acc = acc + bias</span></div><div class="line"><a name="l05237"></a><span class="lineno"> 5237</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, bias_f);</div><div class="line"><a name="l05238"></a><span class="lineno"> 5238</span>&#160;</div><div class="line"><a name="l05239"></a><span class="lineno"> 5239</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l05240"></a><span class="lineno"> 5240</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l05241"></a><span class="lineno"> 5241</span>&#160;</div><div class="line"><a name="l05242"></a><span class="lineno"> 5242</span>&#160; half8 acc_h0 = convert_half8(acc0);</div><div class="line"><a name="l05243"></a><span class="lineno"> 5243</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05244"></a><span class="lineno"> 5244</span>&#160; half8 acc_h1 = convert_half8(acc1);</div><div class="line"><a name="l05245"></a><span class="lineno"> 5245</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05246"></a><span class="lineno"> 5246</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05247"></a><span class="lineno"> 5247</span>&#160; half8 acc_h2 = convert_half8(acc2);</div><div class="line"><a name="l05248"></a><span class="lineno"> 5248</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05249"></a><span class="lineno"> 5249</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05250"></a><span class="lineno"> 5250</span>&#160; half8 acc_h3 = convert_half8(acc3);</div><div class="line"><a name="l05251"></a><span class="lineno"> 5251</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05252"></a><span class="lineno"> 5252</span>&#160;</div><div class="line"><a name="l05253"></a><span class="lineno"> 5253</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l05254"></a><span class="lineno"> 5254</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, ACTIVATION_TYPE, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, acc_h, A_VAL, B_VAL);</div><div class="line"><a name="l05255"></a><span class="lineno"> 5255</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l05256"></a><span class="lineno"> 5256</span>&#160;</div><div class="line"><a name="l05257"></a><span class="lineno"> 5257</span>&#160; <span class="comment">// Store the output block</span></div><div class="line"><a name="l05258"></a><span class="lineno"> 5258</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, acc_h, dst_addr, dst_stride_y, zout.s);</div><div class="line"><a name="l05259"></a><span class="lineno"> 5259</span>&#160;}</div><div class="line"><a name="l05260"></a><span class="lineno"> 5260</span>&#160;</div><div class="line"><a name="l05311"></a><span class="lineno"> 5311</span>&#160;__kernel <span class="keywordtype">void</span> gemm_mm_floating_point_f16_bifrost(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l05312"></a><span class="lineno"> 5312</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src1),</div><div class="line"><a name="l05313"></a><span class="lineno"> 5313</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l05314"></a><span class="lineno"> 5314</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src2),</div><div class="line"><a name="l05315"></a><span class="lineno"> 5315</span>&#160;#endif <span class="comment">// defined(BETA)</span></div><div class="line"><a name="l05316"></a><span class="lineno"> 5316</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>),</div><div class="line"><a name="l05317"></a><span class="lineno"> 5317</span>&#160; uint src0_stride_z,</div><div class="line"><a name="l05318"></a><span class="lineno"> 5318</span>&#160; uint src1_stride_z,</div><div class="line"><a name="l05319"></a><span class="lineno"> 5319</span>&#160;#<span class="keywordflow">if</span> defined(BETA)</div><div class="line"><a name="l05320"></a><span class="lineno"> 5320</span>&#160; uint src2_stride_z,</div><div class="line"><a name="l05321"></a><span class="lineno"> 5321</span>&#160;#endif <span class="comment">//defined(BETA)</span></div><div class="line"><a name="l05322"></a><span class="lineno"> 5322</span>&#160; uint dst_stride_z</div><div class="line"><a name="l05323"></a><span class="lineno"> 5323</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_INPUT_AS_3D)</div><div class="line"><a name="l05324"></a><span class="lineno"> 5324</span>&#160; ,</div><div class="line"><a name="l05325"></a><span class="lineno"> 5325</span>&#160; uint src_cross_plane_pad</div><div class="line"><a name="l05326"></a><span class="lineno"> 5326</span>&#160;#endif <span class="comment">// REINTERPRET_INPUT_AS_3D</span></div><div class="line"><a name="l05327"></a><span class="lineno"> 5327</span>&#160;#<span class="keywordflow">if</span> defined(REINTERPRET_OUTPUT_AS_3D)</div><div class="line"><a name="l05328"></a><span class="lineno"> 5328</span>&#160; ,</div><div class="line"><a name="l05329"></a><span class="lineno"> 5329</span>&#160; uint dst_cross_plane_pad</div><div class="line"><a name="l05330"></a><span class="lineno"> 5330</span>&#160;#endif <span class="comment">// REINTERPRET_OUTPUT_AS_3D</span></div><div class="line"><a name="l05331"></a><span class="lineno"> 5331</span>&#160; )</div><div class="line"><a name="l05332"></a><span class="lineno"> 5332</span>&#160;{</div><div class="line"><a name="l05333"></a><span class="lineno"> 5333</span>&#160; <span class="keywordtype">int</span> idx = get_global_id(0) * NUM_ELEMS_PROCESSED_PER_THREAD_X;</div><div class="line"><a name="l05334"></a><span class="lineno"> 5334</span>&#160;</div><div class="line"><a name="l05335"></a><span class="lineno"> 5335</span>&#160; <span class="comment">// Compute starting address for matrix A and Matrix B</span></div><div class="line"><a name="l05336"></a><span class="lineno"> 5336</span>&#160; int2 src_addr = ((int2)(src0_offset_first_element_in_bytes, src1_offset_first_element_in_bytes));</div><div class="line"><a name="l05337"></a><span class="lineno"> 5337</span>&#160;</div><div class="line"><a name="l05338"></a><span class="lineno"> 5338</span>&#160; <span class="comment">// Update address for the matrix A</span></div><div class="line"><a name="l05339"></a><span class="lineno"> 5339</span>&#160; src_addr.s0 += get_global_id(1) * src0_stride_y * NUM_ELEMS_PROCESSED_PER_THREAD_Y;</div><div class="line"><a name="l05340"></a><span class="lineno"> 5340</span>&#160;</div><div class="line"><a name="l05341"></a><span class="lineno"> 5341</span>&#160; <span class="comment">// Update address for the matrix B</span></div><div class="line"><a name="l05342"></a><span class="lineno"> 5342</span>&#160; src_addr.s1 += idx * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>);</div><div class="line"><a name="l05343"></a><span class="lineno"> 5343</span>&#160;</div><div class="line"><a name="l05344"></a><span class="lineno"> 5344</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05345"></a><span class="lineno"> 5345</span>&#160; <span class="comment">// Since we load a 2D input tile from a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l05346"></a><span class="lineno"> 5346</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l05347"></a><span class="lineno"> 5347</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l05348"></a><span class="lineno"> 5348</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05349"></a><span class="lineno"> 5349</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l05350"></a><span class="lineno"> 5350</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05351"></a><span class="lineno"> 5351</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l05352"></a><span class="lineno"> 5352</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l05353"></a><span class="lineno"> 5353</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l05354"></a><span class="lineno"> 5354</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l05355"></a><span class="lineno"> 5355</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05356"></a><span class="lineno"> 5356</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l05357"></a><span class="lineno"> 5357</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05358"></a><span class="lineno"> 5358</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l05359"></a><span class="lineno"> 5359</span>&#160;</div><div class="line"><a name="l05360"></a><span class="lineno"> 5360</span>&#160; <span class="comment">// The plane (zin) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l05361"></a><span class="lineno"> 5361</span>&#160; uint4 zin = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l05362"></a><span class="lineno"> 5362</span>&#160; zin = min(DEPTH_GEMM3D - 1, zin);</div><div class="line"><a name="l05363"></a><span class="lineno"> 5363</span>&#160;</div><div class="line"><a name="l05364"></a><span class="lineno"> 5364</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l05365"></a><span class="lineno"> 5365</span>&#160; zin *= (src_cross_plane_pad * src0_stride_y);</div><div class="line"><a name="l05366"></a><span class="lineno"> 5366</span>&#160;</div><div class="line"><a name="l05367"></a><span class="lineno"> 5367</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l05368"></a><span class="lineno"> 5368</span>&#160; <span class="comment">// multiply src0_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l05369"></a><span class="lineno"> 5369</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l05370"></a><span class="lineno"> 5370</span>&#160;</div><div class="line"><a name="l05371"></a><span class="lineno"> 5371</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05372"></a><span class="lineno"> 5372</span>&#160;</div><div class="line"><a name="l05373"></a><span class="lineno"> 5373</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l05374"></a><span class="lineno"> 5374</span>&#160; src_addr.s0 += get_global_id(2) * src0_stride_z;</div><div class="line"><a name="l05375"></a><span class="lineno"> 5375</span>&#160;</div><div class="line"><a name="l05376"></a><span class="lineno"> 5376</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05377"></a><span class="lineno"> 5377</span>&#160;</div><div class="line"><a name="l05378"></a><span class="lineno"> 5378</span>&#160;<span class="preprocessor">#if defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l05379"></a><span class="lineno"> 5379</span>&#160; <span class="comment">// Do not slide matrix B if the matrix B has 3 dimensions and matrix A more than 3</span></div><div class="line"><a name="l05380"></a><span class="lineno"> 5380</span>&#160; src_addr.s1 += (get_global_id(2) % MATRIX_B_DEPTH) * src1_stride_z;</div><div class="line"><a name="l05381"></a><span class="lineno"> 5381</span>&#160;<span class="preprocessor">#else // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l05382"></a><span class="lineno"> 5382</span>&#160; src_addr.s1 += get_global_id(2) * src1_stride_z;</div><div class="line"><a name="l05383"></a><span class="lineno"> 5383</span>&#160;<span class="preprocessor">#endif // defined(MATRIX_B_DEPTH)</span></div><div class="line"><a name="l05384"></a><span class="lineno"> 5384</span>&#160;</div><div class="line"><a name="l05385"></a><span class="lineno"> 5385</span>&#160; half8 acc0 = 0.0h;</div><div class="line"><a name="l05386"></a><span class="lineno"> 5386</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05387"></a><span class="lineno"> 5387</span>&#160; half8 acc1 = 0.0h;</div><div class="line"><a name="l05388"></a><span class="lineno"> 5388</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05389"></a><span class="lineno"> 5389</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05390"></a><span class="lineno"> 5390</span>&#160; half8 acc2 = 0.0h;</div><div class="line"><a name="l05391"></a><span class="lineno"> 5391</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05392"></a><span class="lineno"> 5392</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05393"></a><span class="lineno"> 5393</span>&#160; half8 acc3 = 0.0h;</div><div class="line"><a name="l05394"></a><span class="lineno"> 5394</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05395"></a><span class="lineno"> 5395</span>&#160;</div><div class="line"><a name="l05396"></a><span class="lineno"> 5396</span>&#160; <span class="keywordtype">int</span> i = 0;</div><div class="line"><a name="l05397"></a><span class="lineno"> 5397</span>&#160; <span class="keywordflow">for</span>(; i &lt;= ((int)COLS_A - 4); i += 4)</div><div class="line"><a name="l05398"></a><span class="lineno"> 5398</span>&#160; {</div><div class="line"><a name="l05399"></a><span class="lineno"> 5399</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05400"></a><span class="lineno"> 5400</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05401"></a><span class="lineno"> 5401</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 4, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, a, src0_ptr, src_addr.s0, src0_stride_y, zin.s);</div><div class="line"><a name="l05402"></a><span class="lineno"> 5402</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05403"></a><span class="lineno"> 5403</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05404"></a><span class="lineno"> 5404</span>&#160; half4 a0 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l05405"></a><span class="lineno"> 5405</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05406"></a><span class="lineno"> 5406</span>&#160; half4 a1 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l05407"></a><span class="lineno"> 5407</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05408"></a><span class="lineno"> 5408</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05409"></a><span class="lineno"> 5409</span>&#160; half4 a2 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l05410"></a><span class="lineno"> 5410</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05411"></a><span class="lineno"> 5411</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05412"></a><span class="lineno"> 5412</span>&#160; half4 a3 = vload4(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l05413"></a><span class="lineno"> 5413</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05414"></a><span class="lineno"> 5414</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05415"></a><span class="lineno"> 5415</span>&#160;</div><div class="line"><a name="l05416"></a><span class="lineno"> 5416</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l05417"></a><span class="lineno"> 5417</span>&#160; half8 b0 = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05418"></a><span class="lineno"> 5418</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05419"></a><span class="lineno"> 5419</span>&#160;</div><div class="line"><a name="l05420"></a><span class="lineno"> 5420</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l05421"></a><span class="lineno"> 5421</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a0.s0, acc0);</div><div class="line"><a name="l05422"></a><span class="lineno"> 5422</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05423"></a><span class="lineno"> 5423</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a1.s0, acc1);</div><div class="line"><a name="l05424"></a><span class="lineno"> 5424</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05425"></a><span class="lineno"> 5425</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05426"></a><span class="lineno"> 5426</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a2.s0, acc2);</div><div class="line"><a name="l05427"></a><span class="lineno"> 5427</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05428"></a><span class="lineno"> 5428</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05429"></a><span class="lineno"> 5429</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a3.s0, acc3);</div><div class="line"><a name="l05430"></a><span class="lineno"> 5430</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05431"></a><span class="lineno"> 5431</span>&#160;</div><div class="line"><a name="l05432"></a><span class="lineno"> 5432</span>&#160; b0 = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05433"></a><span class="lineno"> 5433</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05434"></a><span class="lineno"> 5434</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a0.s1, acc0);</div><div class="line"><a name="l05435"></a><span class="lineno"> 5435</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05436"></a><span class="lineno"> 5436</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a1.s1, acc1);</div><div class="line"><a name="l05437"></a><span class="lineno"> 5437</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05438"></a><span class="lineno"> 5438</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05439"></a><span class="lineno"> 5439</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a2.s1, acc2);</div><div class="line"><a name="l05440"></a><span class="lineno"> 5440</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05441"></a><span class="lineno"> 5441</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05442"></a><span class="lineno"> 5442</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a3.s1, acc3);</div><div class="line"><a name="l05443"></a><span class="lineno"> 5443</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05444"></a><span class="lineno"> 5444</span>&#160;</div><div class="line"><a name="l05445"></a><span class="lineno"> 5445</span>&#160; b0 = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05446"></a><span class="lineno"> 5446</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05447"></a><span class="lineno"> 5447</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a0.s2, acc0);</div><div class="line"><a name="l05448"></a><span class="lineno"> 5448</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05449"></a><span class="lineno"> 5449</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a1.s2, acc1);</div><div class="line"><a name="l05450"></a><span class="lineno"> 5450</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05451"></a><span class="lineno"> 5451</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05452"></a><span class="lineno"> 5452</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a2.s2, acc2);</div><div class="line"><a name="l05453"></a><span class="lineno"> 5453</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05454"></a><span class="lineno"> 5454</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05455"></a><span class="lineno"> 5455</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a3.s2, acc3);</div><div class="line"><a name="l05456"></a><span class="lineno"> 5456</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05457"></a><span class="lineno"> 5457</span>&#160;</div><div class="line"><a name="l05458"></a><span class="lineno"> 5458</span>&#160; b0 = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05459"></a><span class="lineno"> 5459</span>&#160; src_addr.s1 += src1_stride_y;</div><div class="line"><a name="l05460"></a><span class="lineno"> 5460</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a0.s3, acc0);</div><div class="line"><a name="l05461"></a><span class="lineno"> 5461</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05462"></a><span class="lineno"> 5462</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a1.s3, acc1);</div><div class="line"><a name="l05463"></a><span class="lineno"> 5463</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05464"></a><span class="lineno"> 5464</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05465"></a><span class="lineno"> 5465</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a2.s3, acc2);</div><div class="line"><a name="l05466"></a><span class="lineno"> 5466</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05467"></a><span class="lineno"> 5467</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05468"></a><span class="lineno"> 5468</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a3.s3, acc3);</div><div class="line"><a name="l05469"></a><span class="lineno"> 5469</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05470"></a><span class="lineno"> 5470</span>&#160;</div><div class="line"><a name="l05471"></a><span class="lineno"> 5471</span>&#160; src_addr.s0 += 4 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>);</div><div class="line"><a name="l05472"></a><span class="lineno"> 5472</span>&#160; }</div><div class="line"><a name="l05473"></a><span class="lineno"> 5473</span>&#160;</div><div class="line"><a name="l05474"></a><span class="lineno"> 5474</span>&#160; <span class="keywordflow">for</span>(; i &lt; (int)COLS_A; ++i)</div><div class="line"><a name="l05475"></a><span class="lineno"> 5475</span>&#160; {</div><div class="line"><a name="l05476"></a><span class="lineno"> 5476</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05477"></a><span class="lineno"> 5477</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05478"></a><span class="lineno"> 5478</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a0 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y + zin.s0));</div><div class="line"><a name="l05479"></a><span class="lineno"> 5479</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05480"></a><span class="lineno"> 5480</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a1 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y + zin.s1));</div><div class="line"><a name="l05481"></a><span class="lineno"> 5481</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05482"></a><span class="lineno"> 5482</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05483"></a><span class="lineno"> 5483</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a2 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y + zin.s2));</div><div class="line"><a name="l05484"></a><span class="lineno"> 5484</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05485"></a><span class="lineno"> 5485</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05486"></a><span class="lineno"> 5486</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a3 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y + zin.s3));</div><div class="line"><a name="l05487"></a><span class="lineno"> 5487</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05488"></a><span class="lineno"> 5488</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05489"></a><span class="lineno"> 5489</span>&#160; <span class="comment">// Load values from matrix A</span></div><div class="line"><a name="l05490"></a><span class="lineno"> 5490</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a0 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 0 * src0_stride_y));</div><div class="line"><a name="l05491"></a><span class="lineno"> 5491</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05492"></a><span class="lineno"> 5492</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a1 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 1 * src0_stride_y));</div><div class="line"><a name="l05493"></a><span class="lineno"> 5493</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05494"></a><span class="lineno"> 5494</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05495"></a><span class="lineno"> 5495</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a2 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 2 * src0_stride_y));</div><div class="line"><a name="l05496"></a><span class="lineno"> 5496</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05497"></a><span class="lineno"> 5497</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05498"></a><span class="lineno"> 5498</span>&#160; <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> a3 = *((__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src0_ptr + src_addr.s0 + 3 * src0_stride_y));</div><div class="line"><a name="l05499"></a><span class="lineno"> 5499</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05500"></a><span class="lineno"> 5500</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_INPUT_AS_3D)</span></div><div class="line"><a name="l05501"></a><span class="lineno"> 5501</span>&#160;</div><div class="line"><a name="l05502"></a><span class="lineno"> 5502</span>&#160; <span class="comment">// Load values from matrix B</span></div><div class="line"><a name="l05503"></a><span class="lineno"> 5503</span>&#160; half8 b0 = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05504"></a><span class="lineno"> 5504</span>&#160;</div><div class="line"><a name="l05505"></a><span class="lineno"> 5505</span>&#160; src_addr += (int2)(<span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>), src1_stride_y);</div><div class="line"><a name="l05506"></a><span class="lineno"> 5506</span>&#160;</div><div class="line"><a name="l05507"></a><span class="lineno"> 5507</span>&#160; <span class="comment">// Accumulate</span></div><div class="line"><a name="l05508"></a><span class="lineno"> 5508</span>&#160; acc0 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a0, acc0); <span class="comment">// b0 * (half8)a0;</span></div><div class="line"><a name="l05509"></a><span class="lineno"> 5509</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05510"></a><span class="lineno"> 5510</span>&#160; acc1 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a1, acc1); <span class="comment">// b0 * (half8)a1;</span></div><div class="line"><a name="l05511"></a><span class="lineno"> 5511</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 1</span></div><div class="line"><a name="l05512"></a><span class="lineno"> 5512</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05513"></a><span class="lineno"> 5513</span>&#160; acc2 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a2, acc2); <span class="comment">// b0 * (half8)a2;</span></div><div class="line"><a name="l05514"></a><span class="lineno"> 5514</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 2</span></div><div class="line"><a name="l05515"></a><span class="lineno"> 5515</span>&#160;<span class="preprocessor">#if NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05516"></a><span class="lineno"> 5516</span>&#160; acc3 = <a class="code" href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">fma</a>(b0, (half8)a3, acc3); <span class="comment">// b0 * (half8)a3;</span></div><div class="line"><a name="l05517"></a><span class="lineno"> 5517</span>&#160;<span class="preprocessor">#endif // NUM_ELEMS_PROCESSED_PER_THREAD_Y &gt; 3</span></div><div class="line"><a name="l05518"></a><span class="lineno"> 5518</span>&#160; }</div><div class="line"><a name="l05519"></a><span class="lineno"> 5519</span>&#160;</div><div class="line"><a name="l05520"></a><span class="lineno"> 5520</span>&#160; <span class="keywordtype">int</span> z = get_global_id(2);</div><div class="line"><a name="l05521"></a><span class="lineno"> 5521</span>&#160;</div><div class="line"><a name="l05522"></a><span class="lineno"> 5522</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l05523"></a><span class="lineno"> 5523</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l05524"></a><span class="lineno"> 5524</span>&#160;</div><div class="line"><a name="l05525"></a><span class="lineno"> 5525</span>&#160; <span class="comment">// Compute dst address</span></div><div class="line"><a name="l05526"></a><span class="lineno"> 5526</span>&#160; __global uchar *dst_addr = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0);</div><div class="line"><a name="l05527"></a><span class="lineno"> 5527</span>&#160;</div><div class="line"><a name="l05528"></a><span class="lineno"> 5528</span>&#160; uint4 zout = 0;</div><div class="line"><a name="l05529"></a><span class="lineno"> 5529</span>&#160;</div><div class="line"><a name="l05530"></a><span class="lineno"> 5530</span>&#160;<span class="preprocessor">#if defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l05531"></a><span class="lineno"> 5531</span>&#160;</div><div class="line"><a name="l05532"></a><span class="lineno"> 5532</span>&#160; <span class="comment">// Since we store a 2D output tile in a 3D tensor, we need to check when the plane changes across the z dimension</span></div><div class="line"><a name="l05533"></a><span class="lineno"> 5533</span>&#160; <span class="comment">// in order to take into account the presence of possible cross plane paddings</span></div><div class="line"><a name="l05534"></a><span class="lineno"> 5534</span>&#160; <span class="comment">//</span></div><div class="line"><a name="l05535"></a><span class="lineno"> 5535</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05536"></a><span class="lineno"> 5536</span>&#160; <span class="comment">// | plane0 |</span></div><div class="line"><a name="l05537"></a><span class="lineno"> 5537</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05538"></a><span class="lineno"> 5538</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l05539"></a><span class="lineno"> 5539</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l05540"></a><span class="lineno"> 5540</span>&#160; <span class="comment">// | cross_plane_pad |</span></div><div class="line"><a name="l05541"></a><span class="lineno"> 5541</span>&#160; <span class="comment">// |******************|</span></div><div class="line"><a name="l05542"></a><span class="lineno"> 5542</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05543"></a><span class="lineno"> 5543</span>&#160; <span class="comment">// | plane1 |</span></div><div class="line"><a name="l05544"></a><span class="lineno"> 5544</span>&#160; <span class="comment">// | |</span></div><div class="line"><a name="l05545"></a><span class="lineno"> 5545</span>&#160; <span class="comment">// |__________________|</span></div><div class="line"><a name="l05546"></a><span class="lineno"> 5546</span>&#160;</div><div class="line"><a name="l05547"></a><span class="lineno"> 5547</span>&#160; <span class="comment">// The plane (zout) is calculated dividing M (get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y) by HEIGHT_GEMM3D</span></div><div class="line"><a name="l05548"></a><span class="lineno"> 5548</span>&#160; zout = ((uint4)(0, 1, 2, 3) + (uint4)(get_global_id(1) * NUM_ELEMS_PROCESSED_PER_THREAD_Y)) / (uint4)HEIGHT_GEMM3D;</div><div class="line"><a name="l05549"></a><span class="lineno"> 5549</span>&#160; zout = min(DEPTH_GEMM3D - 1, zout);</div><div class="line"><a name="l05550"></a><span class="lineno"> 5550</span>&#160;</div><div class="line"><a name="l05551"></a><span class="lineno"> 5551</span>&#160; <span class="comment">// Add offset due to the cross plane paddings</span></div><div class="line"><a name="l05552"></a><span class="lineno"> 5552</span>&#160; zout *= (dst_cross_plane_pad * dst_stride_y);</div><div class="line"><a name="l05553"></a><span class="lineno"> 5553</span>&#160;</div><div class="line"><a name="l05554"></a><span class="lineno"> 5554</span>&#160; <span class="comment">// Add offset for batched GEMM. The batches will be in the fourth dimension and for this reason we</span></div><div class="line"><a name="l05555"></a><span class="lineno"> 5555</span>&#160; <span class="comment">// multiply dst_stride_z by DEPTH_GEMM3D</span></div><div class="line"><a name="l05556"></a><span class="lineno"> 5556</span>&#160; dst_addr += z * dst_stride_z * DEPTH_GEMM3D;</div><div class="line"><a name="l05557"></a><span class="lineno"> 5557</span>&#160;<span class="preprocessor">#else // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l05558"></a><span class="lineno"> 5558</span>&#160; <span class="comment">// Add offset for batched GEMM</span></div><div class="line"><a name="l05559"></a><span class="lineno"> 5559</span>&#160; dst_addr += z * dst_stride_z;</div><div class="line"><a name="l05560"></a><span class="lineno"> 5560</span>&#160;<span class="preprocessor">#endif // defined(REINTERPRET_OUTPUT_AS_3D)</span></div><div class="line"><a name="l05561"></a><span class="lineno"> 5561</span>&#160;</div><div class="line"><a name="l05562"></a><span class="lineno"> 5562</span>&#160; <span class="comment">// Multiply by the weight of matrix-matrix product and store the result</span></div><div class="line"><a name="l05563"></a><span class="lineno"> 5563</span>&#160;<span class="preprocessor">#if defined(ALPHA)</span></div><div class="line"><a name="l05564"></a><span class="lineno"> 5564</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, acc, ALPHA);</div><div class="line"><a name="l05565"></a><span class="lineno"> 5565</span>&#160;<span class="preprocessor">#endif // defined(ALPHA)</span></div><div class="line"><a name="l05566"></a><span class="lineno"> 5566</span>&#160;</div><div class="line"><a name="l05567"></a><span class="lineno"> 5567</span>&#160; <span class="comment">// Add beta*bias</span></div><div class="line"><a name="l05568"></a><span class="lineno"> 5568</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l05569"></a><span class="lineno"> 5569</span>&#160; <a class="code" href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, uint, zero, 0);</div><div class="line"><a name="l05570"></a><span class="lineno"> 5570</span>&#160;</div><div class="line"><a name="l05571"></a><span class="lineno"> 5571</span>&#160;<span class="preprocessor">#if defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l05572"></a><span class="lineno"> 5572</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>));</div><div class="line"><a name="l05573"></a><span class="lineno"> 5573</span>&#160;</div><div class="line"><a name="l05574"></a><span class="lineno"> 5574</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(1, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l05575"></a><span class="lineno"> 5575</span>&#160;</div><div class="line"><a name="l05576"></a><span class="lineno"> 5576</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l05577"></a><span class="lineno"> 5577</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(1, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l05578"></a><span class="lineno"> 5578</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l05579"></a><span class="lineno"> 5579</span>&#160;</div><div class="line"><a name="l05580"></a><span class="lineno"> 5580</span>&#160; <span class="comment">// acc = acc + bias[broadcasted]</span></div><div class="line"><a name="l05581"></a><span class="lineno"> 5581</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, bias0);</div><div class="line"><a name="l05582"></a><span class="lineno"> 5582</span>&#160;</div><div class="line"><a name="l05583"></a><span class="lineno"> 5583</span>&#160;<span class="preprocessor">#else // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l05584"></a><span class="lineno"> 5584</span>&#160; __global uchar *src2_addr = src2_ptr + src2_offset_first_element_in_bytes + (get_global_id(0) * (uint)8 * <span class="keyword">sizeof</span>(<a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>)) + (get_global_id(1) *</div><div class="line"><a name="l05585"></a><span class="lineno"> 5585</span>&#160; (uint)NUM_ELEMS_PROCESSED_PER_THREAD_Y * src2_stride_y) + get_global_id(2) * src2_stride_z;</div><div class="line"><a name="l05586"></a><span class="lineno"> 5586</span>&#160;</div><div class="line"><a name="l05587"></a><span class="lineno"> 5587</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, src2_addr, 0, src2_stride_y, zero);</div><div class="line"><a name="l05588"></a><span class="lineno"> 5588</span>&#160;</div><div class="line"><a name="l05589"></a><span class="lineno"> 5589</span>&#160;<span class="preprocessor">#ifndef UNIT_BETA</span></div><div class="line"><a name="l05590"></a><span class="lineno"> 5590</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>, BETA);</div><div class="line"><a name="l05591"></a><span class="lineno"> 5591</span>&#160;<span class="preprocessor">#endif // UNIT_BIAS</span></div><div class="line"><a name="l05592"></a><span class="lineno"> 5592</span>&#160;</div><div class="line"><a name="l05593"></a><span class="lineno"> 5593</span>&#160; <span class="comment">// acc = acc + bias</span></div><div class="line"><a name="l05594"></a><span class="lineno"> 5594</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, acc, <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">bias</a>);</div><div class="line"><a name="l05595"></a><span class="lineno"> 5595</span>&#160;</div><div class="line"><a name="l05596"></a><span class="lineno"> 5596</span>&#160;<span class="preprocessor">#endif // defined(BROADCAST_BIAS)</span></div><div class="line"><a name="l05597"></a><span class="lineno"> 5597</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l05598"></a><span class="lineno"> 5598</span>&#160;</div><div class="line"><a name="l05599"></a><span class="lineno"> 5599</span>&#160;<span class="preprocessor">#if defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l05600"></a><span class="lineno"> 5600</span>&#160; <a class="code" href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, ACTIVATION_TYPE, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, acc, A_VAL, B_VAL);</div><div class="line"><a name="l05601"></a><span class="lineno"> 5601</span>&#160;<span class="preprocessor">#endif // defined(ACTIVATION_TYPE)</span></div><div class="line"><a name="l05602"></a><span class="lineno"> 5602</span>&#160;</div><div class="line"><a name="l05603"></a><span class="lineno"> 5603</span>&#160; <span class="comment">// Store the output block</span></div><div class="line"><a name="l05604"></a><span class="lineno"> 5604</span>&#160; <a class="code" href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a>(NUM_ELEMS_PROCESSED_PER_THREAD_Y, 8, <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a>, acc, dst_addr, dst_stride_y, zout.s);</div><div class="line"><a name="l05605"></a><span class="lineno"> 5605</span>&#160;}</div><div class="line"><a name="l05606"></a><span class="lineno"> 5606</span>&#160;<span class="preprocessor">#endif // defined(ARM_COMPUTE_OPENCL_FP16_ENABLED)</span></div><div class="line"><a name="l05607"></a><span class="lineno"> 5607</span>&#160;</div><div class="line"><a name="l05608"></a><span class="lineno"> 5608</span>&#160;<span class="preprocessor">#endif // defined(COLS_A) &amp;&amp; defined(NUM_ELEMS_PROCESSED_PER_THREAD_X) &amp;&amp; (NUM_ELEMS_PROCESSED_PER_THREAD_Y)</span></div><div class="line"><a name="l05609"></a><span class="lineno"> 5609</span>&#160;</div><div class="line"><a name="l05610"></a><span class="lineno"> 5610</span>&#160;<span class="preprocessor">#if defined(BETA)</span></div><div class="line"><a name="l05611"></a><span class="lineno"> 5611</span>&#160;</div><div class="line"><a name="l05632"></a><span class="lineno"> 5632</span>&#160;__kernel <span class="keywordtype">void</span> gemm_ma_f32(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>),</div><div class="line"><a name="l05633"></a><span class="lineno"> 5633</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>))</div><div class="line"><a name="l05634"></a><span class="lineno"> 5634</span>&#160;{</div><div class="line"><a name="l05635"></a><span class="lineno"> 5635</span>&#160; <span class="comment">// Compute source and destination addresses</span></div><div class="line"><a name="l05636"></a><span class="lineno"> 5636</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>);</div><div class="line"><a name="l05637"></a><span class="lineno"> 5637</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l05638"></a><span class="lineno"> 5638</span>&#160;</div><div class="line"><a name="l05639"></a><span class="lineno"> 5639</span>&#160; <span class="comment">// Load values from A x B</span></div><div class="line"><a name="l05640"></a><span class="lineno"> 5640</span>&#160; float4 alpha_ab = vload4(0, (__global <span class="keywordtype">float</span> *)<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>.ptr);</div><div class="line"><a name="l05641"></a><span class="lineno"> 5641</span>&#160;</div><div class="line"><a name="l05642"></a><span class="lineno"> 5642</span>&#160; <span class="comment">// Load values from Matrix C</span></div><div class="line"><a name="l05643"></a><span class="lineno"> 5643</span>&#160; float4 c = vload4(0, (__global <span class="keywordtype">float</span> *)<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>.ptr);</div><div class="line"><a name="l05644"></a><span class="lineno"> 5644</span>&#160;</div><div class="line"><a name="l05645"></a><span class="lineno"> 5645</span>&#160; <span class="comment">// Computes alpha * axb + beta * c</span></div><div class="line"><a name="l05646"></a><span class="lineno"> 5646</span>&#160; float4 out = alpha_ab + (float4)BETA * c;</div><div class="line"><a name="l05647"></a><span class="lineno"> 5647</span>&#160;</div><div class="line"><a name="l05648"></a><span class="lineno"> 5648</span>&#160; <span class="comment">// Store final result in axb matrix</span></div><div class="line"><a name="l05649"></a><span class="lineno"> 5649</span>&#160; vstore4(out, 0, (__global <span class="keywordtype">float</span> *)<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>.ptr);</div><div class="line"><a name="l05650"></a><span class="lineno"> 5650</span>&#160;}</div><div class="line"><a name="l05651"></a><span class="lineno"> 5651</span>&#160;</div><div class="line"><a name="l05652"></a><span class="lineno"> 5652</span>&#160;<span class="preprocessor">#if defined(ARM_COMPUTE_OPENCL_FP16_ENABLED)</span></div><div class="line"><a name="l05653"></a><span class="lineno"> 5653</span>&#160;</div><div class="line"><a name="l05674"></a><span class="lineno"> 5674</span>&#160;__kernel <span class="keywordtype">void</span> gemm_ma_f16(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>),</div><div class="line"><a name="l05675"></a><span class="lineno"> 5675</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>))</div><div class="line"><a name="l05676"></a><span class="lineno"> 5676</span>&#160;{</div><div class="line"><a name="l05677"></a><span class="lineno"> 5677</span>&#160; <span class="comment">// Compute source and destination addresses</span></div><div class="line"><a name="l05678"></a><span class="lineno"> 5678</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>);</div><div class="line"><a name="l05679"></a><span class="lineno"> 5679</span>&#160; <a class="code" href="struct_tensor3_d.xhtml">Tensor3D</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l05680"></a><span class="lineno"> 5680</span>&#160;</div><div class="line"><a name="l05681"></a><span class="lineno"> 5681</span>&#160; <span class="comment">// Load values from A x B</span></div><div class="line"><a name="l05682"></a><span class="lineno"> 5682</span>&#160; half8 alpha_ab = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>.ptr);</div><div class="line"><a name="l05683"></a><span class="lineno"> 5683</span>&#160;</div><div class="line"><a name="l05684"></a><span class="lineno"> 5684</span>&#160; <span class="comment">// Load values from Matrix C</span></div><div class="line"><a name="l05685"></a><span class="lineno"> 5685</span>&#160; half8 c = vload8(0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">src</a>.ptr);</div><div class="line"><a name="l05686"></a><span class="lineno"> 5686</span>&#160;</div><div class="line"><a name="l05687"></a><span class="lineno"> 5687</span>&#160; <span class="comment">// Computes alpha * axb + beta * c</span></div><div class="line"><a name="l05688"></a><span class="lineno"> 5688</span>&#160; half8 out = alpha_ab + (half8)BETA * c;</div><div class="line"><a name="l05689"></a><span class="lineno"> 5689</span>&#160;</div><div class="line"><a name="l05690"></a><span class="lineno"> 5690</span>&#160; <span class="comment">// Store final result in axb matrix</span></div><div class="line"><a name="l05691"></a><span class="lineno"> 5691</span>&#160; vstore8(out, 0, (__global <a class="code" href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">half</a> *)<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>.ptr);</div><div class="line"><a name="l05692"></a><span class="lineno"> 5692</span>&#160;}</div><div class="line"><a name="l05693"></a><span class="lineno"> 5693</span>&#160;<span class="preprocessor">#endif // defined(ARM_COMPUTE_OPENCL_FP16_ENABLED)</span></div><div class="line"><a name="l05694"></a><span class="lineno"> 5694</span>&#160;<span class="preprocessor">#endif // defined(BETA)</span></div><div class="line"><a name="l05695"></a><span class="lineno"> 5695</span>&#160;</div><div class="line"><a name="l05696"></a><span class="lineno"> 5696</span>&#160;<span class="preprocessor">#if defined(WIDTH_VECTOR_A)</span></div><div class="line"><a name="l05697"></a><span class="lineno"> 5697</span>&#160;</div><div class="line"><a name="l05724"></a><span class="lineno"> 5724</span>&#160;__kernel <span class="keywordtype">void</span> gemm_lc_vm_f32(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(src0),</div><div class="line"><a name="l05725"></a><span class="lineno"> 5725</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a>(src1),</div><div class="line"><a name="l05726"></a><span class="lineno"> 5726</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>))</div><div class="line"><a name="l05727"></a><span class="lineno"> 5727</span>&#160;{</div><div class="line"><a name="l05728"></a><span class="lineno"> 5728</span>&#160; <span class="keywordtype">int</span> idx = get_global_id(0) * 4;</div><div class="line"><a name="l05729"></a><span class="lineno"> 5729</span>&#160; <span class="keywordtype">int</span> idy = get_global_id(1);</div><div class="line"><a name="l05730"></a><span class="lineno"> 5730</span>&#160;</div><div class="line"><a name="l05731"></a><span class="lineno"> 5731</span>&#160; <span class="comment">// Compute the address for the vector A and matrix B</span></div><div class="line"><a name="l05732"></a><span class="lineno"> 5732</span>&#160; int2 src_addr = ((int2)(src0_offset_first_element_in_bytes + src0_stride_y * idy, src1_offset_first_element_in_bytes + src1_stride_z * idy));</div><div class="line"><a name="l05733"></a><span class="lineno"> 5733</span>&#160; src_addr.s1 += idx * <span class="keyword">sizeof</span>(float);</div><div class="line"><a name="l05734"></a><span class="lineno"> 5734</span>&#160;</div><div class="line"><a name="l05735"></a><span class="lineno"> 5735</span>&#160; <span class="keywordtype">int</span> end_row_vec_a = src_addr.s0 + (WIDTH_VECTOR_A * <span class="keyword">sizeof</span>(float));</div><div class="line"><a name="l05736"></a><span class="lineno"> 5736</span>&#160;</div><div class="line"><a name="l05737"></a><span class="lineno"> 5737</span>&#160; float4 acc = 0.0f;</div><div class="line"><a name="l05738"></a><span class="lineno"> 5738</span>&#160;</div><div class="line"><a name="l05739"></a><span class="lineno"> 5739</span>&#160; <span class="keywordflow">for</span>(; src_addr.s0 &lt;= (end_row_vec_a - 2 * (int)<span class="keyword">sizeof</span>(<span class="keywordtype">float</span>)); src_addr += (int2)(2 * <span class="keyword">sizeof</span>(<span class="keywordtype">float</span>), 2 * src1_stride_y))</div><div class="line"><a name="l05740"></a><span class="lineno"> 5740</span>&#160; {</div><div class="line"><a name="l05741"></a><span class="lineno"> 5741</span>&#160; float2 a0 = vload2(0, (__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0));</div><div class="line"><a name="l05742"></a><span class="lineno"> 5742</span>&#160; float4 b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05743"></a><span class="lineno"> 5743</span>&#160; float4 b1 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1 + src1_stride_y));</div><div class="line"><a name="l05744"></a><span class="lineno"> 5744</span>&#160;</div><div class="line"><a name="l05745"></a><span class="lineno"> 5745</span>&#160; acc += b0 * (float4)a0.s0;</div><div class="line"><a name="l05746"></a><span class="lineno"> 5746</span>&#160; acc += b1 * (float4)a0.s1;</div><div class="line"><a name="l05747"></a><span class="lineno"> 5747</span>&#160; }</div><div class="line"><a name="l05748"></a><span class="lineno"> 5748</span>&#160;</div><div class="line"><a name="l05749"></a><span class="lineno"> 5749</span>&#160; <span class="keywordflow">for</span>(; src_addr.s0 &lt; end_row_vec_a; src_addr += (int2)(<span class="keyword">sizeof</span>(<span class="keywordtype">float</span>), src1_stride_y))</div><div class="line"><a name="l05750"></a><span class="lineno"> 5750</span>&#160; {</div><div class="line"><a name="l05751"></a><span class="lineno"> 5751</span>&#160; <span class="keywordtype">float</span> a0 = *((__global <span class="keywordtype">float</span> *)(src0_ptr + src_addr.s0));</div><div class="line"><a name="l05752"></a><span class="lineno"> 5752</span>&#160; float4 b0 = vload4(0, (__global <span class="keywordtype">float</span> *)(src1_ptr + src_addr.s1));</div><div class="line"><a name="l05753"></a><span class="lineno"> 5753</span>&#160;</div><div class="line"><a name="l05754"></a><span class="lineno"> 5754</span>&#160; acc += b0 * (float4)a0;</div><div class="line"><a name="l05755"></a><span class="lineno"> 5755</span>&#160; }</div><div class="line"><a name="l05756"></a><span class="lineno"> 5756</span>&#160;</div><div class="line"><a name="l05757"></a><span class="lineno"> 5757</span>&#160; <span class="comment">// Compute destination address</span></div><div class="line"><a name="l05758"></a><span class="lineno"> 5758</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> <a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a> = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>);</div><div class="line"><a name="l05759"></a><span class="lineno"> 5759</span>&#160;</div><div class="line"><a name="l05760"></a><span class="lineno"> 5760</span>&#160; vstore4(acc, 0, (__global <span class="keywordtype">float</span> *)(<a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a>(&amp;<a class="code" href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">dst</a>, 0, 0)));</div><div class="line"><a name="l05761"></a><span class="lineno"> 5761</span>&#160;}</div><div class="line"><a name="l05762"></a><span class="lineno"> 5762</span>&#160;<span class="preprocessor">#endif // defined(WIDTH_VECTOR_A)</span></div><div class="line"><a name="l05763"></a><span class="lineno"> 5763</span>&#160;</div><div class="line"><a name="l05780"></a><span class="lineno"> 5780</span>&#160;<span class="preprocessor">#if defined(DATA_TYPE) &amp;&amp; defined(VECTOR_SIZE)</span></div><div class="line"><a name="l05781"></a><span class="lineno"> 5781</span>&#160;__kernel <span class="keywordtype">void</span> gemm_accumulate_biases(</div><div class="line"><a name="l05782"></a><span class="lineno"> 5782</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a>(accum),</div><div class="line"><a name="l05783"></a><span class="lineno"> 5783</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a40a6eb9f2a7712f08d6bb8ff6c9e6ca7">VECTOR_DECLARATION</a>(biases))</div><div class="line"><a name="l05784"></a><span class="lineno"> 5784</span>&#160;{</div><div class="line"><a name="l05785"></a><span class="lineno"> 5785</span>&#160; <a class="code" href="struct_image.xhtml">Image</a> accum = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a>(accum);</div><div class="line"><a name="l05786"></a><span class="lineno"> 5786</span>&#160; <a class="code" href="struct_vector.xhtml">Vector</a> biases = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a527bfdf5eeb306f1cf01c4a8e29f38e0">CONVERT_TO_VECTOR_STRUCT</a>(biases);</div><div class="line"><a name="l05787"></a><span class="lineno"> 5787</span>&#160;</div><div class="line"><a name="l05788"></a><span class="lineno"> 5788</span>&#160; <span class="comment">// Vector size, e.g. number of vector elements.</span></div><div class="line"><a name="l05789"></a><span class="lineno"> 5789</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="softmax__layer_8cl.xhtml#a7c78836761fa3b5b124efea237dac70f">VECTOR_SIZE</a>)</div><div class="line"><a name="l05790"></a><span class="lineno"> 5790</span>&#160; accum_value = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(<a class="code" href="softmax__layer_8cl.xhtml#a7c78836761fa3b5b124efea237dac70f">VECTOR_SIZE</a>)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)accum.<a class="code" href="struct_image.xhtml#acf52c23cbd7424606c10a606524e3e32">ptr</a>);</div><div class="line"><a name="l05791"></a><span class="lineno"> 5791</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a>(<a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a>, <a class="code" href="softmax__layer_8cl.xhtml#a7c78836761fa3b5b124efea237dac70f">VECTOR_SIZE</a>)</div><div class="line"><a name="l05792"></a><span class="lineno"> 5792</span>&#160; biases_value = <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a>(<a class="code" href="softmax__layer_8cl.xhtml#a7c78836761fa3b5b124efea237dac70f">VECTOR_SIZE</a>)(0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)biases.<a class="code" href="struct_vector.xhtml#acf52c23cbd7424606c10a606524e3e32">ptr</a>);</div><div class="line"><a name="l05793"></a><span class="lineno"> 5793</span>&#160; accum_value = biases_value + accum_value;</div><div class="line"><a name="l05794"></a><span class="lineno"> 5794</span>&#160; <span class="comment">// Store result in the accumulate buffer</span></div><div class="line"><a name="l05795"></a><span class="lineno"> 5795</span>&#160; <a class="code" href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#acb282042d1edeeaa3cc979a206f78b54">VSTORE</a>(<a class="code" href="softmax__layer_8cl.xhtml#a7c78836761fa3b5b124efea237dac70f">VECTOR_SIZE</a>)</div><div class="line"><a name="l05796"></a><span class="lineno"> 5796</span>&#160; (accum_value, 0, (__global <a class="code" href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a> *)accum.<a class="code" href="struct_image.xhtml#acf52c23cbd7424606c10a606524e3e32">ptr</a>);</div><div class="line"><a name="l05797"></a><span class="lineno"> 5797</span>&#160;}</div><div class="line"><a name="l05798"></a><span class="lineno"> 5798</span>&#160;<span class="preprocessor">#endif // defined(DATA_TYPE) &amp;&amp; defined(VECTOR_SIZE)</span></div><div class="ttc" id="struct_vector_xhtml"><div class="ttname"><a href="struct_vector.xhtml">Vector</a></div><div class="ttdoc">Structure to hold Vector information.</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00159">helpers.h:159</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a009469e4d9b8fce3b6d5e97d2077827d"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a009469e4d9b8fce3b6d5e97d2077827d">offset</a></div><div class="ttdeci">__global uchar * offset(const Image *img, int x, int y)</div><div class="ttdoc">Get the pointer position of a Image.</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00328">helpers.h:328</a></div></div>
<div class="ttc" id="repeat_8h_xhtml_a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9"><div class="ttname"><a href="repeat_8h.xhtml#a4a61a8df72f9bc1b1dcc4cc4ec6ac3b9">REPEAT_VAR_INIT_TO_CONST</a></div><div class="ttdeci">#define REPEAT_VAR_INIT_TO_CONST(N, TYPE, VAR, VAL)</div><div class="ttdef"><b>Definition:</b> <a href="repeat_8h_source.xhtml#l00081">repeat.h:81</a></div></div>
<div class="ttc" id="repeat_8h_xhtml"><div class="ttname"><a href="repeat_8h.xhtml">repeat.h</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_aebe814363556c244be043b13e7969197"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#aebe814363556c244be043b13e7969197">CONVERT_TO_IMAGE_STRUCT</a></div><div class="ttdeci">#define CONVERT_TO_IMAGE_STRUCT(name)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00129">helpers.h:129</a></div></div>
<div class="ttc" id="convolution3x3_8cl_xhtml_afb8c72ce35c4a1f4a2588d6573e54aa1"><div class="ttname"><a href="convolution3x3_8cl.xhtml#afb8c72ce35c4a1f4a2588d6573e54aa1">DATA_TYPE</a></div><div class="ttdeci">#define DATA_TYPE</div><div class="ttdef"><b>Definition:</b> <a href="convolution3x3_8cl_source.xhtml#l00027">convolution3x3.cl:27</a></div></div>
<div class="ttc" id="namespacearm__compute_1_1test_1_1validation_xhtml_aa76b4a6e74940dabc5b7fc6b2dab3545"><div class="ttname"><a href="namespacearm__compute_1_1test_1_1validation.xhtml#aa76b4a6e74940dabc5b7fc6b2dab3545">arm_compute::test::validation::b</a></div><div class="ttdeci">SimpleTensor&lt; float &gt; b</div><div class="ttdef"><b>Definition:</b> <a href="_c_p_p_2_d_f_t_8cpp_source.xhtml#l00157">DFT.cpp:157</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml"><div class="ttname"><a href="gemm__helpers_8h.xhtml">gemm_helpers.h</a></div></div>
<div class="ttc" id="namespacearm__compute_xhtml_a73e2825fd61d349c5ca2f5313e3c8ea1"><div class="ttname"><a href="namespacearm__compute.xhtml#a73e2825fd61d349c5ca2f5313e3c8ea1">arm_compute::half</a></div><div class="ttdeci">half_float::half half</div><div class="ttdoc">16-bit floating point type</div><div class="ttdef"><b>Definition:</b> <a href="arm__compute_2core_2_types_8h_source.xhtml#l00044">Types.h:44</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a22f42fcf2077d951271df83b55c1a71a"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a22f42fcf2077d951271df83b55c1a71a">IMAGE_DECLARATION</a></div><div class="ttdeci">#define IMAGE_DECLARATION(name)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00093">helpers.h:93</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml_a055f3dc1bfec3462a3c48de0203a2c80"><div class="ttname"><a href="gemm__helpers_8h.xhtml#a055f3dc1bfec3462a3c48de0203a2c80">ADD_BLOCK_BROADCAST</a></div><div class="ttdeci">#define ADD_BLOCK_BROADCAST(N, BASENAME, BIAS)</div><div class="ttdoc">Add elements stored in variables BIAS##0 ...</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00622">gemm_helpers.h:622</a></div></div>
<div class="ttc" id="struct_tensor3_d_xhtml"><div class="ttname"><a href="struct_tensor3_d.xhtml">Tensor3D</a></div><div class="ttdoc">Structure to hold 3D tensor information.</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00176">helpers.h:176</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a527bfdf5eeb306f1cf01c4a8e29f38e0"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a527bfdf5eeb306f1cf01c4a8e29f38e0">CONVERT_TO_VECTOR_STRUCT</a></div><div class="ttdeci">#define CONVERT_TO_VECTOR_STRUCT(name)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00123">helpers.h:123</a></div></div>
<div class="ttc" id="softmax__layer_8cl_xhtml_a7c78836761fa3b5b124efea237dac70f"><div class="ttname"><a href="softmax__layer_8cl.xhtml#a7c78836761fa3b5b124efea237dac70f">VECTOR_SIZE</a></div><div class="ttdeci">#define VECTOR_SIZE</div><div class="ttdef"><b>Definition:</b> <a href="softmax__layer_8cl_source.xhtml#l00060">softmax_layer.cl:60</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml_a47f4d3ee6a1f83a0c3d1744f42da118f"><div class="ttname"><a href="gemm__helpers_8h.xhtml#a47f4d3ee6a1f83a0c3d1744f42da118f">ADD_BLOCK</a></div><div class="ttdeci">#define ADD_BLOCK(N, BASENAME, BIAS)</div><div class="ttdoc">Add BIAS to BASENAME##0 ...</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00552">gemm_helpers.h:552</a></div></div>
<div class="ttc" id="gemmlowp_8cl_xhtml_ae991dacacb5b9b0b176538400a7e8269"><div class="ttname"><a href="gemmlowp_8cl.xhtml#ae991dacacb5b9b0b176538400a7e8269">ARM_DOT_K0XN0</a></div><div class="ttdeci">#define ARM_DOT_K0XN0(n0, k0, a, b, c)</div><div class="ttdef"><b>Definition:</b> <a href="gemmlowp_8cl_source.xhtml#l00184">gemmlowp.cl:184</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a40a6eb9f2a7712f08d6bb8ff6c9e6ca7"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a40a6eb9f2a7712f08d6bb8ff6c9e6ca7">VECTOR_DECLARATION</a></div><div class="ttdeci">#define VECTOR_DECLARATION(name)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00087">helpers.h:87</a></div></div>
<div class="ttc" id="struct_vector_xhtml_acf52c23cbd7424606c10a606524e3e32"><div class="ttname"><a href="struct_vector.xhtml#acf52c23cbd7424606c10a606524e3e32">Vector::ptr</a></div><div class="ttdeci">__global uchar * ptr</div><div class="ttdoc">Pointer to the starting postion of the buffer.</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00161">helpers.h:161</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml_a9fe7ded27e552f48fa285d34fb00427d"><div class="ttname"><a href="gemm__helpers_8h.xhtml#a9fe7ded27e552f48fa285d34fb00427d">LOAD_BLOCK</a></div><div class="ttdeci">#define LOAD_BLOCK(M0, N0, DATA_TYPE, BASENAME, PTR, OFFSET, STRIDE_Y, Z)</div><div class="ttdoc">Load Blocks of M0 consecutive rows and N0 consecutive columns when using Z offset as well Supported c...</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00113">gemm_helpers.h:113</a></div></div>
<div class="ttc" id="namespacearm__compute_1_1test_1_1validation_xhtml_adbf67dcee294e673cf796f1ed8aeb6a4"><div class="ttname"><a href="namespacearm__compute_1_1test_1_1validation.xhtml#adbf67dcee294e673cf796f1ed8aeb6a4">arm_compute::test::validation::dst</a></div><div class="ttdeci">CLTensor dst</div><div class="ttdef"><b>Definition:</b> <a href="_c_l_2_absolute_difference_8cpp_source.xhtml#l00102">AbsoluteDifference.cpp:102</a></div></div>
<div class="ttc" id="namespacearm__compute_1_1test_1_1validation_xhtml_a3a77be8aebd8e00522b32061d46ccdbd"><div class="ttname"><a href="namespacearm__compute_1_1test_1_1validation.xhtml#a3a77be8aebd8e00522b32061d46ccdbd">arm_compute::test::validation::bias</a></div><div class="ttdeci">CLTensor bias</div><div class="ttdef"><b>Definition:</b> <a href="validation_2_c_l_2_convolution_layer_8cpp_source.xhtml#l00181">ConvolutionLayer.cpp:181</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml_ade58486d6b2d13d7941106a9d2ec86d3"><div class="ttname"><a href="gemm__helpers_8h.xhtml#ade58486d6b2d13d7941106a9d2ec86d3">CALCULATE_Z_OFFSET</a></div><div class="ttdeci">#define CALCULATE_Z_OFFSET(M0, DATA_TYPE, Z, Y, HEIGHT_GEMM3D, DEPTH_GEMM3D, CROSS_PLANE_PAD, STRIDE_Y)</div><div class="ttdoc">The Z offsets are expected to have consecutive names, For e.g.</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00181">gemm_helpers.h:181</a></div></div>
<div class="ttc" id="struct_image_xhtml"><div class="ttname"><a href="struct_image.xhtml">Image</a></div><div class="ttdoc">Structure to hold Image information.</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00167">helpers.h:167</a></div></div>
<div class="ttc" id="struct_image_xhtml_acf52c23cbd7424606c10a606524e3e32"><div class="ttname"><a href="struct_image.xhtml#acf52c23cbd7424606c10a606524e3e32">Image::ptr</a></div><div class="ttdeci">__global uchar * ptr</div><div class="ttdoc">Pointer to the starting postion of the buffer.</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00169">helpers.h:169</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a31c8c760f08fb1a331b16b7c204321dc"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a31c8c760f08fb1a331b16b7c204321dc">CONVERT_TO_TENSOR3D_STRUCT</a></div><div class="ttdeci">#define CONVERT_TO_TENSOR3D_STRUCT(name)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00144">helpers.h:144</a></div></div>
<div class="ttc" id="gemm_8cl_xhtml_a2da39e3a865d4f0da99e21e37c2cef85"><div class="ttname"><a href="gemm_8cl.xhtml#a2da39e3a865d4f0da99e21e37c2cef85">ACTIVATION_BLOCK</a></div><div class="ttdeci">#define ACTIVATION_BLOCK(N, ACTIVATION_TYPE, DATA_TYPE, BASENAME, A_VAL, B_VAL)</div><div class="ttdoc">Apply activation to the variables BASENAME##0...</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00692">gemm_helpers.h:692</a></div></div>
<div class="ttc" id="namespacearm__compute_1_1support_1_1cpp11_xhtml_af399bedeaeb8dc177d3a301a12c3a5d0"><div class="ttname"><a href="namespacearm__compute_1_1support_1_1cpp11.xhtml#af399bedeaeb8dc177d3a301a12c3a5d0">arm_compute::support::cpp11::fma</a></div><div class="ttdeci">T fma(T x, T y, T z)</div><div class="ttdoc">Computes (x*y) + z as if to infinite precision and rounded only once to fit the result type.</div><div class="ttdef"><b>Definition:</b> <a href="_toolchain_support_8h_source.xhtml#l00336">ToolchainSupport.h:336</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_acb282042d1edeeaa3cc979a206f78b54"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#acb282042d1edeeaa3cc979a206f78b54">VSTORE</a></div><div class="ttdeci">#define VSTORE(size)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00056">helpers.h:56</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml_ab6d2df39dc14930b9865d6309913dc4d"><div class="ttname"><a href="gemm__helpers_8h.xhtml#ab6d2df39dc14930b9865d6309913dc4d">STORE_BLOCK</a></div><div class="ttdeci">#define STORE_BLOCK(M0, N0, DATA_TYPE, BASENAME, PTR, STRIDE_Y, Z)</div><div class="ttdoc">Store a block of size M0 (rows) x NO (columns).</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00354">gemm_helpers.h:354</a></div></div>
<div class="ttc" id="gemm__helpers_8h_xhtml_a4e4f47c0a33bb8a9ab20b355d7fa84ca"><div class="ttname"><a href="gemm__helpers_8h.xhtml#a4e4f47c0a33bb8a9ab20b355d7fa84ca">SCALE_BLOCK</a></div><div class="ttdeci">#define SCALE_BLOCK(N, DATA_TYPE, BASENAME, SCALE)</div><div class="ttdoc">Scale elements stored in variables BASENAME##0 to BASENAME##(N-1) by SCALE Supported cases N=1,...</div><div class="ttdef"><b>Definition:</b> <a href="gemm__helpers_8h_source.xhtml#l00431">gemm_helpers.h:431</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a287e2fc366c312b468382c95bb90f91f"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a287e2fc366c312b468382c95bb90f91f">VLOAD</a></div><div class="ttdeci">#define VLOAD(size)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00053">helpers.h:53</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a6b83038822d1ae7ab619b684ed3b7fc0"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a6b83038822d1ae7ab619b684ed3b7fc0">TENSOR3D_DECLARATION</a></div><div class="ttdeci">#define TENSOR3D_DECLARATION(name)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00101">helpers.h:101</a></div></div>
<div class="ttc" id="namespacearm__compute_1_1test_1_1validation_xhtml_a989ab3e96426615bb98e04e0235088ca"><div class="ttname"><a href="namespacearm__compute_1_1test_1_1validation.xhtml#a989ab3e96426615bb98e04e0235088ca">arm_compute::test::validation::src</a></div><div class="ttdeci">cast configure &amp; src</div><div class="ttdef"><b>Definition:</b> <a href="_c_l_2_cast_8cpp_source.xhtml#l00169">Cast.cpp:169</a></div></div>
<div class="ttc" id="src_2core_2_c_l_2cl__kernels_2_helpers_8h_xhtml_a36f754c05b6fddf6df0d8d0a74f8159f"><div class="ttname"><a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h.xhtml#a36f754c05b6fddf6df0d8d0a74f8159f">VEC_DATA_TYPE</a></div><div class="ttdeci">#define VEC_DATA_TYPE(type, size)</div><div class="ttdef"><b>Definition:</b> <a href="src_2core_2_c_l_2cl__kernels_2_helpers_8h_source.xhtml#l00073">helpers.h:73</a></div></div>
</div><!-- fragment --></div><!-- contents -->
</div><!-- doc-content -->
<!-- start footer part -->
<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
<ul>
<li class="navelem"><a class="el" href="dir_68267d1309a1af8e8297ef4c3efbcdba.xhtml">src</a></li><li class="navelem"><a class="el" href="dir_aebb8dcc11953d78e620bbef0b9e2183.xhtml">core</a></li><li class="navelem"><a class="el" href="dir_8c278f79c760e5c5fbd911f9870614c1.xhtml">CL</a></li><li class="navelem"><a class="el" href="dir_25885286e9dad4fa105b7b25a8031bbf.xhtml">cl_kernels</a></li><li class="navelem"><a class="el" href="gemm_8cl.xhtml">gemm.cl</a></li>
<li class="footer">Generated on Mon Sep 2 2019 11:47:21 for Compute Library by
<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.15 </li>
</ul>
</div>
</body>
</html>