1 /*
2 * Copyright (c) 2022 Arm Limited.
3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24
25 #include <cstddef>
26
27 namespace arm_conv {
28 namespace winograd {
29 namespace weight_transform {
30
cpp_fp32_1x2_1x7(unsigned int n_channels,const float * inptr,size_t,size_t ld_weight_col,float * outptr,size_t matrix_stride)31 void cpp_fp32_1x2_1x7(
32 unsigned int n_channels,
33 const float* inptr, size_t, size_t ld_weight_col,
34 float *outptr, size_t matrix_stride
35 )
36 {
37 for (; n_channels; n_channels--)
38 {
39 // Matrices used and computed in this kernel
40 float w[7], V[8];
41
42 // Read weights
43 for (int j = 0; j < 7; j++)
44 {
45 w[j] = *(inptr + j*ld_weight_col);
46 }
47
48 // Compute V = w WT
49 V[0] = (w[0]*-1) / 36.0f;
50 V[1] = (w[1]*-1 + w[3]*-1 + w[5]*-1 + w[0]*1 + w[2]*1 + w[4]*1 + w[6]*1) / 48.0f;
51 V[2] = (w[0]*1 + w[1]*1 + w[2]*1 + w[3]*1 + w[4]*1 + w[5]*1 + w[6]*1) / 48.0f;
52 V[3] = (w[0]*-1 + w[6]*-64 + w[4]*-16 + w[2]*-4 + w[1]*2 + w[3]*8 + w[5]*32) / 120.0f;
53 V[4] = (w[0]*-1 + w[6]*-64 + w[5]*-32 + w[4]*-16 + w[3]*-8 + w[2]*-4 + w[1]*-2) / 120.0f;
54 V[5] = (w[5]*-243 + w[3]*-27 + w[1]*-3 + w[2]*9 + w[4]*81 + w[6]*729 + w[0]*1) / 720.0f;
55 V[6] = (w[1]*3 + w[2]*9 + w[3]*27 + w[4]*81 + w[5]*243 + w[6]*729 + w[0]*1) / 720.0f;
56 V[7] = (w[6]*1) / 1.0f;
57
58 // Store the transformed weights
59 for (int j = 0; j < 8; j++)
60 {
61 *(outptr + j*matrix_stride) = V[j];
62 }
63
64 inptr++;
65 outptr++;
66 }
67 }
68
69 } // namespace output_transform
70 } // namespace winograd
71 } // namespace arm_conv
72