Home
last modified time | relevance | path

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

/external/XNNPACK/test/
Dconvolution-nhwc.cc19 .group_output_channels(19)
29 .group_output_channels(19)
40 .group_output_channels(19)
52 .group_output_channels(19)
63 .group_output_channels(19)
74 .group_output_channels(19)
85 .group_output_channels(19)
96 .group_output_channels(19) in TEST()
107 .group_output_channels(19) in TEST()
119 .group_output_channels(19) in TEST()
[all …]
Ddeconvolution-nhwc.cc27 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
39 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
52 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
65 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
78 .group_output_channels(output_channels)
90 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
102 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
114 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
126 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
139 .group_output_channels(xnn_params.q8.gemm.nr * 2 + 3)
[all …]
Dconvolution-operator-tester.h186 inline ConvolutionOperatorTester& group_output_channels(size_t group_output_channels) { in group_output_channels() argument
187 assert(group_output_channels >= 1); in group_output_channels()
188 this->group_output_channels_ = group_output_channels; in group_output_channels()
192 inline size_t group_output_channels() const { in group_output_channels() function
334 return group_output_channels() * groups(); in output_pixel_stride()
336 assert(this->output_pixel_stride_ >= group_output_channels() * groups()); in output_pixel_stride()
364 return groups() * group_output_channels() * output_height() * output_width(); in output_batch_stride()
366 …assert(this->output_batch_stride_ >= groups() * group_output_channels() * output_height() * output… in output_batch_stride()
544 …std::vector<uint8_t> kernel(groups() * group_output_channels() * kernel_height() * kernel_width() … in TestNHWCxQ8()
545 std::vector<int32_t> bias(groups() * group_output_channels()); in TestNHWCxQ8()
[all …]
Ddeconvolution-operator-tester.h165 inline DeconvolutionOperatorTester& group_output_channels(size_t group_output_channels) { in group_output_channels() function
166 assert(group_output_channels >= 1); in group_output_channels()
167 this->group_output_channels_ = group_output_channels; in group_output_channels()
171 inline size_t group_output_channels() const { in group_output_channels() function
313 return group_output_channels() * groups(); in output_pixel_stride()
315 assert(this->output_pixel_stride_ >= group_output_channels() * groups()); in output_pixel_stride()
438 …std::vector<uint8_t> kernel(groups() * group_output_channels() * kernel_height() * kernel_width() … in TestQ8()
439 std::vector<int32_t> bias(groups() * group_output_channels()); in TestQ8()
440 …utput_height() * output_width() - 1) * output_pixel_stride() + groups() * group_output_channels()); in TestQ8()
441 …accumulators(batch_size() * output_height() * output_width() * groups() * group_output_channels()); in TestQ8()
[all …]
Dconvolution-nchw.cc19 .group_output_channels(19)
31 .group_output_channels(19)
44 .group_output_channels(19)
58 .group_output_channels(19)
72 .group_output_channels(19)
86 .group_output_channels(output_channels)
98 .group_output_channels(19)
110 .group_output_channels(19)
123 .group_output_channels(19)
138 .group_output_channels(19) in TEST()
[all …]
/external/XNNPACK/src/
Dconvolution-nchw.c50 size_t group_output_channels, in xnn_create_convolution2d_nchw_f32() argument
105 if (group_output_channels == 0) { in xnn_create_convolution2d_nchw_f32()
109 group_output_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()
212 for (size_t oc = 0; oc < round_down_po2(group_output_channels, 4); oc += 4) { in xnn_create_convolution2d_nchw_f32()
224 …for (size_t oc = round_down_po2(group_output_channels, 4); oc < round_down_po2(group_output_channe… in xnn_create_convolution2d_nchw_f32()
233 for (size_t oc = round_down_po2(group_output_channels, 2); oc < group_output_channels; oc++) { in xnn_create_convolution2d_nchw_f32()
[all …]
Ddeconvolution-nhwc.c54 size_t group_output_channels, in xnn_create_deconvolution2d_nhwc_q8() argument
116 if (group_output_channels == 0) { in xnn_create_deconvolution2d_nhwc_q8()
120 group_output_channels); in xnn_create_deconvolution2d_nhwc_q8()
133 const size_t output_channels = groups * group_output_channels; in xnn_create_deconvolution2d_nhwc_q8()
138 output_pixel_stride, groups, group_output_channels); in xnn_create_deconvolution2d_nhwc_q8()
195 const uint32_t n_stride = round_up(group_output_channels, nr); 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()
275 deconvolution_op->group_output_channels = group_output_channels; in xnn_create_deconvolution2d_nhwc_q8()
318 size_t group_output_channels, in xnn_create_deconvolution2d_nhwc_f32() argument
[all …]
Dconvolution-nhwc.c74 size_t group_output_channels, in xnn_create_convolution2d_nhwc_q8() argument
137 if (group_output_channels == 0) { in xnn_create_convolution2d_nhwc_q8()
141 group_output_channels); in xnn_create_convolution2d_nhwc_q8()
154 const size_t output_channels = groups * group_output_channels; in xnn_create_convolution2d_nhwc_q8()
159 output_pixel_stride, groups, group_output_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()
287 const uint32_t n_stride = round_up(group_output_channels, nr); in xnn_create_convolution2d_nhwc_q8()
302 groups, group_output_channels, group_input_channels, in xnn_create_convolution2d_nhwc_q8()
316 groups, group_output_channels, kernel_size, in xnn_create_convolution2d_nhwc_q8()
322 groups, group_output_channels, kernel_size, group_input_channels, in xnn_create_convolution2d_nhwc_q8()
[all …]
Dsubgraph.c127 size_t group_output_channels, in xnn_define_convolution_2d() argument
178 if (group_output_channels == 0) { in xnn_define_convolution_2d()
182 group_output_channels); in xnn_define_convolution_2d()
252 node->params.convolution_2d.group_output_channels = group_output_channels; in xnn_define_convolution_2d()
Dfully-connected-nc.c163 fully_connected_op->group_output_channels = output_channels; in xnn_create_fully_connected_nc_q8()
303 fully_connected_op->group_output_channels = output_channels; in xnn_create_fully_connected_nc_f32()
364 const size_t output_channels = fully_connected_op->group_output_channels; in setup_fully_connected_nc()
Druntime.c93 node->params.convolution_2d.group_output_channels, in xnn_create_runtime_v2()
95 …node->params.convolution_2d.group_output_channels * node->params.convolution_2d.groups /* output_p… in xnn_create_runtime_v2()
/external/XNNPACK/bench/
Df32-im2col-gemm.cc46 const size_t group_output_channels = state.range(9); in Im2ColGEMMBenchmark() local
60 const size_t nc_stride = benchmark::utils::RoundUp<size_t>(group_output_channels, nr); in Im2ColGEMMBenchmark()
65 std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in Im2ColGEMMBenchmark()
67 std::vector<float> b(group_output_channels); in Im2ColGEMMBenchmark()
71 const size_t c_elements = output_size * group_output_channels; in Im2ColGEMMBenchmark()
78 xnn_pack_f32_gemm_goi_w(1 /* groups */, group_output_channels, group_input_channels * kernel_size, in Im2ColGEMMBenchmark()
84 …tor<float> im2col_buffer(output_size * group_input_channels * kernel_size * group_output_channels); in Im2ColGEMMBenchmark()
115 for (uint32_t n = 0; n < group_output_channels; n += nr) { in Im2ColGEMMBenchmark()
116 const uint32_t nb = min(group_output_channels - n, nr); in Im2ColGEMMBenchmark()
121 …c.data() + (buffer_index * output_size + m) * group_output_channels + n, group_output_channels * s… in Im2ColGEMMBenchmark()
[all …]
Dconvolution.cc52 const size_t group_output_channels = state.range(11); in xnnpack_convolution_q8() local
59 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_convolution_q8()
72 …std::vector<uint8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_… in xnnpack_convolution_q8()
74 std::vector<int32_t> bias(groups * group_output_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()
170 const size_t group_output_channels = state.range(11); in xnnpack_convolution_f32() local
176 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_convolution_f32()
189 …std::vector<float> kernel(groups * group_output_channels * kernel_height * kernel_width * group_in… in xnnpack_convolution_f32()
191 std::vector<float> bias(groups * group_output_channels); in xnnpack_convolution_f32()
[all …]
Ddeconvolution.cc41 const size_t group_output_channels = state.range(11); in xnnpack_deconvolution_q8() local
48 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_deconvolution_q8()
61 …std::vector<uint8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_… in xnnpack_deconvolution_q8()
63 std::vector<int32_t> bias(groups * group_output_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()
159 const size_t group_output_channels = state.range(11); in xnnpack_deconvolution_f32() local
165 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_deconvolution_f32()
178 …std::vector<float> kernel(groups * group_output_channels * kernel_height * kernel_width * group_in… in xnnpack_deconvolution_f32()
180 std::vector<float> bias(groups * group_output_channels); in xnnpack_deconvolution_f32()
[all …]
Df32-igemm.cc50 const size_t group_output_channels = state.range(9); in IGEMMBenchmark() local
56 const size_t output_pixel_stride = group_output_channels; in IGEMMBenchmark()
67 const size_t nc_stride = benchmark::utils::RoundUp<size_t>(group_output_channels, nr); in IGEMMBenchmark()
72 std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in IGEMMBenchmark()
74 std::vector<float> b(group_output_channels); in IGEMMBenchmark()
89 1 /* groups */, group_output_channels, kernel_size, group_input_channels, in IGEMMBenchmark()
136 for (uint32_t n = 0; n < group_output_channels; n += nr) { in IGEMMBenchmark()
137 const uint32_t nb = min(group_output_channels - n, nr); in IGEMMBenchmark()
142 …c.data() + buffer_index * c_elements + m * group_output_channels + n, group_output_channels * size… in IGEMMBenchmark()
152 group_input_channels * group_output_channels * in IGEMMBenchmark()
/external/XNNPACK/include/
Dxnnpack.h238 size_t group_output_channels,
604 size_t group_output_channels,
636 size_t group_output_channels,
894 size_t group_output_channels,
1060 size_t group_output_channels,
1098 size_t group_output_channels,
/external/XNNPACK/src/xnnpack/
Dsubgraph.h89 size_t group_output_channels; member
Doperator.h188 size_t group_output_channels; member