1 // Tencent is pleased to support the open source community by making ncnn available.
2 //
3 // Copyright (C) 2021 THL A29 Limited, a Tencent company. All rights reserved.
4 //
5 // Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
6 // in compliance with the License. You may obtain a copy of the License at
7 //
8 // https://opensource.org/licenses/BSD-3-Clause
9 //
10 // Unless required by applicable law or agreed to in writing, software distributed
11 // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
12 // CONDITIONS OF ANY KIND, either express or implied. See the License for the
13 // specific language governing permissions and limitations under the License.
14
conv1x1s1_sgemm_int8_neon(const Mat & bottom_blob,Mat & top_blob,const Mat & kernel,const Option & opt)15 static void conv1x1s1_sgemm_int8_neon(const Mat& bottom_blob, Mat& top_blob, const Mat& kernel, const Option& opt)
16 {
17 int w = bottom_blob.w;
18 int h = bottom_blob.h;
19 const int size = w * h;
20
21 Mat bottom_im2col = bottom_blob;
22 bottom_im2col.w = size;
23 bottom_im2col.h = 1;
24
25 im2col_sgemm_int8_neon(bottom_im2col, top_blob, kernel, opt);
26 }
27
conv1x1s2_int8_neon(const Mat & bottom_blob,Mat & top_blob,const Mat & kernel,const Option & opt)28 static void conv1x1s2_int8_neon(const Mat& bottom_blob, Mat& top_blob, const Mat& kernel, const Option& opt)
29 {
30 int w = bottom_blob.w;
31 int channels = bottom_blob.c;
32 size_t elemsize = bottom_blob.elemsize;
33 int elempack = bottom_blob.elempack;
34
35 int outw = top_blob.w;
36 int outh = top_blob.h;
37
38 const int tailstep = w - 2 * outw + w;
39
40 Mat bottom_blob_shrinked;
41 bottom_blob_shrinked.create(outw, outh, channels, elemsize, elempack, opt.workspace_allocator);
42
43 #pragma omp parallel for num_threads(opt.num_threads)
44 for (int p = 0; p < channels; p++)
45 {
46 const signed char* r0 = bottom_blob.channel(p);
47 signed char* outptr = bottom_blob_shrinked.channel(p);
48
49 for (int i = 0; i < outh; i++)
50 {
51 int j = 0;
52 for (; j + 3 < outw; j += 4)
53 {
54 outptr[0] = r0[0];
55 outptr[1] = r0[2];
56 outptr[2] = r0[4];
57 outptr[3] = r0[6];
58
59 r0 += 8;
60 outptr += 4;
61 }
62 for (; j + 1 < outw; j += 2)
63 {
64 outptr[0] = r0[0];
65 outptr[1] = r0[2];
66
67 r0 += 4;
68 outptr += 2;
69 }
70 for (; j < outw; j++)
71 {
72 outptr[0] = r0[0];
73
74 r0 += 2;
75 outptr += 1;
76 }
77
78 r0 += tailstep;
79 }
80 }
81
82 conv1x1s1_sgemm_int8_neon(bottom_blob_shrinked, top_blob, kernel, opt);
83 }
84