Home
last modified time | relevance | path

Searched refs:group_input_channels (Results 1 – 18 of 18) sorted by relevance

/external/XNNPACK/test/
Dconvolution-nhwc.cc18 .group_input_channels(23)
28 .group_input_channels(23)
39 .group_input_channels(23)
51 .group_input_channels(23)
62 .group_input_channels(23)
73 .group_input_channels(23)
84 .group_input_channels(23)
95 .group_input_channels(17) in TEST()
106 .group_input_channels(17) in TEST()
118 .group_input_channels(17) in TEST()
[all …]
Ddeconvolution-nhwc.cc26 .group_input_channels(23)
38 .group_input_channels(23)
51 .group_input_channels(23)
64 .group_input_channels(input_channels)
77 .group_input_channels(23)
89 .group_input_channels(23)
101 .group_input_channels(23)
113 .group_input_channels(23)
125 .group_input_channels(23)
138 .group_input_channels(23)
[all …]
Dconvolution-nchw.cc18 .group_input_channels(23)
30 .group_input_channels(23)
43 .group_input_channels(23)
57 .group_input_channels(23)
71 .group_input_channels(input_channels)
85 .group_input_channels(23)
97 .group_input_channels(23)
109 .group_input_channels(23)
122 .group_input_channels(23)
137 .group_input_channels(23) in TEST()
[all …]
Dconvolution-operator-tester.h176 inline ConvolutionOperatorTester& group_input_channels(size_t group_input_channels) { in group_input_channels() argument
177 assert(group_input_channels >= 1); in group_input_channels()
178 this->group_input_channels_ = group_input_channels; in group_input_channels()
182 inline size_t group_input_channels() const { in group_input_channels() function
319 return group_input_channels() * groups(); in input_pixel_stride()
321 assert(this->input_pixel_stride_ >= group_input_channels() * groups()); in input_pixel_stride()
349 return groups() * group_input_channels() * input_height() * input_width(); in input_batch_stride()
351 …assert(this->input_batch_stride_ >= groups() * group_input_channels() * input_height() * input_wid… in input_batch_stride()
543 …put_height() * input_width() - 1) * input_pixel_stride() + groups() * group_input_channels()) + 8); in TestNHWCxQ8()
544 …el(groups() * group_output_channels() * kernel_height() * kernel_width() * group_input_channels()); in TestNHWCxQ8()
[all …]
Ddeconvolution-operator-tester.h155 inline DeconvolutionOperatorTester& group_input_channels(size_t group_input_channels) { in group_input_channels() argument
156 assert(group_input_channels >= 1); in group_input_channels()
157 this->group_input_channels_ = group_input_channels; in group_input_channels()
161 inline size_t group_input_channels() const { in group_input_channels() function
298 return group_input_channels() * groups(); in input_pixel_stride()
300 assert(this->input_pixel_stride_ >= group_input_channels() * groups()); in input_pixel_stride()
437 … * input_height() * input_width() - 1) * input_pixel_stride() + groups() * group_input_channels()); in TestQ8()
438 …el(groups() * group_output_channels() * kernel_height() * kernel_width() * group_input_channels()); in TestQ8()
483 for (size_t ic = 0; ic < group_input_channels(); ic++) { in TestQ8()
485 …height() + iy) * input_width() + ix) * input_pixel_stride() + g * group_input_channels() + ic]) - … in TestQ8()
[all …]
/external/XNNPACK/src/
Dconvolution-nchw.c49 size_t group_input_channels, in xnn_create_convolution2d_nchw_f32() argument
97 if (group_input_channels == 0) { in xnn_create_convolution2d_nchw_f32()
101 group_input_channels); in xnn_create_convolution2d_nchw_f32()
133 if ((flags & XNN_FLAG_DEPTHWISE_CONVOLUTION) != 0 && group_input_channels != 1) { in xnn_create_convolution2d_nchw_f32()
137 group_input_channels); in xnn_create_convolution2d_nchw_f32()
166 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
172 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
178 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
184 …!nhwc_input && group_input_channels == 1 && group_output_channels == 1 && xnn_params.f32.spchw_dwc… in xnn_create_convolution2d_nchw_f32()
213 for (size_t ic = 0; ic < group_input_channels; ic++) { in xnn_create_convolution2d_nchw_f32()
[all …]
Dconvolution-nhwc.c73 size_t group_input_channels, in xnn_create_convolution2d_nhwc_q8() argument
129 if (group_input_channels == 0) { in xnn_create_convolution2d_nhwc_q8()
133 group_input_channels); in xnn_create_convolution2d_nhwc_q8()
145 const size_t input_channels = groups * group_input_channels; in xnn_create_convolution2d_nhwc_q8()
150 input_pixel_stride, groups, group_input_channels); in xnn_create_convolution2d_nhwc_q8()
192 if ((flags & XNN_FLAG_DEPTHWISE_CONVOLUTION) != 0 && group_input_channels != 1) { in xnn_create_convolution2d_nhwc_q8()
196 group_input_channels); in xnn_create_convolution2d_nhwc_q8()
234 if (group_input_channels == 1 && group_output_channels == 1 && groups > 1 && in xnn_create_convolution2d_nhwc_q8()
288 const uint32_t k_stride = round_up_po2(group_input_channels, kr); in xnn_create_convolution2d_nhwc_q8()
302 groups, group_output_channels, group_input_channels, in xnn_create_convolution2d_nhwc_q8()
[all …]
Ddeconvolution-nhwc.c53 size_t group_input_channels, in xnn_create_deconvolution2d_nhwc_q8() argument
108 if (group_input_channels == 0) { in xnn_create_deconvolution2d_nhwc_q8()
112 group_input_channels); in xnn_create_deconvolution2d_nhwc_q8()
124 const size_t input_channels = groups * group_input_channels; in xnn_create_deconvolution2d_nhwc_q8()
129 input_pixel_stride, groups, group_input_channels); in xnn_create_deconvolution2d_nhwc_q8()
196 const uint32_t k_stride = round_up_po2(group_input_channels, kr); in xnn_create_deconvolution2d_nhwc_q8()
236 groups, group_output_channels, kernel_size, group_input_channels, in xnn_create_deconvolution2d_nhwc_q8()
243 groups, group_output_channels, kernel_height, kernel_width, group_input_channels, in xnn_create_deconvolution2d_nhwc_q8()
274 deconvolution_op->group_input_channels = group_input_channels; in xnn_create_deconvolution2d_nhwc_q8()
317 size_t group_input_channels, in xnn_create_deconvolution2d_nhwc_f32() argument
[all …]
Dsubgraph.c126 size_t group_input_channels, in xnn_define_convolution_2d() argument
170 if (group_input_channels == 0) { in xnn_define_convolution_2d()
174 group_input_channels); in xnn_define_convolution_2d()
251 node->params.convolution_2d.group_input_channels = group_input_channels; in xnn_define_convolution_2d()
Dfully-connected-nc.c162 fully_connected_op->group_input_channels = input_channels; in xnn_create_fully_connected_nc_q8()
302 fully_connected_op->group_input_channels = input_channels; in xnn_create_fully_connected_nc_f32()
363 const size_t input_channels = fully_connected_op->group_input_channels; in setup_fully_connected_nc()
Druntime.c92 node->params.convolution_2d.group_input_channels, in xnn_create_runtime_v2()
94 …node->params.convolution_2d.group_input_channels * node->params.convolution_2d.groups /* input_pix… in xnn_create_runtime_v2()
/external/XNNPACK/bench/
Df32-im2col-gemm.cc45 const size_t group_input_channels = state.range(8); in Im2ColGEMMBenchmark() local
61 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); in Im2ColGEMMBenchmark()
63 std::vector<float> a(input_height * input_width * group_input_channels); in Im2ColGEMMBenchmark()
65 std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in Im2ColGEMMBenchmark()
78 xnn_pack_f32_gemm_goi_w(1 /* groups */, group_output_channels, group_input_channels * kernel_size, in Im2ColGEMMBenchmark()
84 …std::vector<float> im2col_buffer(output_size * group_input_channels * kernel_size * group_output_c… in Im2ColGEMMBenchmark()
107 group_input_channels * sizeof(float) /* input channels */, in Im2ColGEMMBenchmark()
108 group_input_channels * sizeof(float) /* input stride */, in Im2ColGEMMBenchmark()
118 mb, nb, kernel_size * group_input_channels * sizeof(float), in Im2ColGEMMBenchmark()
119 …inputData + m * kernel_size * group_input_channels, kernel_size * group_input_channels * sizeof(fl… in Im2ColGEMMBenchmark()
[all …]
Dconvolution.cc51 const size_t group_input_channels = state.range(10); in xnnpack_convolution_q8() local
60 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_q8()
72 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_q8()
100 groups, group_input_channels, group_output_channels, in xnnpack_convolution_q8()
153 groups * group_input_channels * group_output_channels * in xnnpack_convolution_q8()
169 const size_t group_input_channels = state.range(10); in xnnpack_convolution_f32() local
177 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_f32()
189 …loat> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_f32()
217 groups, group_input_channels, group_output_channels, in xnnpack_convolution_f32()
267 groups * group_input_channels * group_output_channels * in xnnpack_convolution_f32()
[all …]
Ddeconvolution.cc40 const size_t group_input_channels = state.range(10); in xnnpack_deconvolution_q8() local
49 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_deconvolution_q8()
61 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_deconvolution_q8()
89 groups, group_input_channels, group_output_channels, in xnnpack_deconvolution_q8()
142 groups * group_input_channels * group_output_channels * in xnnpack_deconvolution_q8()
158 const size_t group_input_channels = state.range(10); in xnnpack_deconvolution_f32() local
166 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_deconvolution_f32()
178 …loat> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_deconvolution_f32()
206 groups, group_input_channels, group_output_channels, in xnnpack_deconvolution_f32()
258 groups * group_input_channels * group_output_channels * in xnnpack_deconvolution_f32()
Df32-igemm.cc49 const size_t group_input_channels = state.range(8); in IGEMMBenchmark() local
57 const size_t input_pixel_stride = group_input_channels; in IGEMMBenchmark()
68 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); in IGEMMBenchmark()
72 std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in IGEMMBenchmark()
77 std::vector<float> z(group_input_channels); in IGEMMBenchmark()
89 1 /* groups */, group_output_channels, kernel_size, group_input_channels, in IGEMMBenchmark()
102 convolution_op.group_input_channels = group_input_channels; in IGEMMBenchmark()
139 mb, nb, group_input_channels * sizeof(float), kernel_size * mr * sizeof(void*), in IGEMMBenchmark()
152 group_input_channels * group_output_channels * in IGEMMBenchmark()
/external/XNNPACK/include/
Dxnnpack.h237 size_t group_input_channels,
603 size_t group_input_channels,
635 size_t group_input_channels,
893 size_t group_input_channels,
1059 size_t group_input_channels,
1097 size_t group_input_channels,
/external/XNNPACK/src/xnnpack/
Dsubgraph.h88 size_t group_input_channels; member
Doperator.h187 size_t group_input_channels; member