Home
last modified time | relevance | path

Searched refs:group_input_channels (Results 1 – 22 of 22) 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.h179 inline ConvolutionOperatorTester& group_input_channels(size_t group_input_channels) { in group_input_channels() argument
180 assert(group_input_channels >= 1); in group_input_channels()
181 this->group_input_channels_ = group_input_channels; in group_input_channels()
185 inline size_t group_input_channels() const { in group_input_channels() function
322 return group_input_channels() * groups(); in input_channel_stride()
324 assert(this->input_channel_stride_ >= group_input_channels() * groups()); in input_channel_stride()
517 …t_height() * input_width() - 1) * input_channel_stride() + groups() * group_input_channels()) + 8); in TestNHWCxQS8()
518 …el(groups() * group_output_channels() * kernel_height() * kernel_width() * group_input_channels()); in TestNHWCxQS8()
550 ASSERT_EQ(group_input_channels(), 1); in TestNHWCxQS8()
587 for (size_t ic = 0; ic < group_input_channels(); ic++) { in TestNHWCxQS8()
[all …]
Ddeconvolution-operator-tester.h211 inline DeconvolutionOperatorTester& group_input_channels(size_t group_input_channels) { in group_input_channels() argument
212 assert(group_input_channels >= 1); in group_input_channels()
213 this->group_input_channels_ = group_input_channels; in group_input_channels()
217 inline size_t group_input_channels() const { in group_input_channels() function
354 return group_input_channels() * groups(); in input_pixel_stride()
356 assert(this->input_pixel_stride_ >= group_input_channels() * groups()); in input_pixel_stride()
493 … * input_height() * input_width() - 1) * input_pixel_stride() + groups() * group_input_channels()); in TestQU8()
494 …el(groups() * group_output_channels() * kernel_height() * kernel_width() * group_input_channels()); in TestQU8()
539 for (size_t ic = 0; ic < group_input_channels(); ic++) { in TestQU8()
541 …height() + iy) * input_width() + ix) * input_pixel_stride() + g * group_input_channels() + ic]) - … in TestQU8()
[all …]
Dsubgraph-tester.h80 size_t group_input_channels, size_t group_output_channels, in add_conv() argument
88 groups, group_input_channels, group_output_channels, in add_conv()
/external/XNNPACK/src/operators/
Dconvolution-nchw.c49 size_t group_input_channels, in xnn_create_convolution2d_nchw_f32() argument
99 if (group_input_channels == 0) { in xnn_create_convolution2d_nchw_f32()
102 xnn_operator_type_to_string(xnn_operator_type_convolution_nchw_f32), group_input_channels); in xnn_create_convolution2d_nchw_f32()
113 const size_t input_channels = groups * group_input_channels; in xnn_create_convolution2d_nchw_f32()
119 input_channel_stride, groups, group_input_channels); in xnn_create_convolution2d_nchw_f32()
154 if ((flags & XNN_FLAG_DEPTHWISE_CONVOLUTION) != 0 && group_input_channels != 1) { in xnn_create_convolution2d_nchw_f32()
158 xnn_operator_type_to_string(xnn_operator_type_convolution_nchw_f32), group_input_channels); in xnn_create_convolution2d_nchw_f32()
187 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()
193 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()
199 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()
[all …]
Dconvolution-nhwc.c73 size_t group_input_channels, in create_convolution2d_nhwc() argument
153 if (group_input_channels == 0) { in create_convolution2d_nhwc()
156 xnn_operator_type_to_string(operator_type), group_input_channels); in create_convolution2d_nhwc()
167 const size_t input_channels = groups * group_input_channels; in create_convolution2d_nhwc()
173 input_channel_stride, groups, group_input_channels); in create_convolution2d_nhwc()
187 if ((flags & XNN_FLAG_DEPTHWISE_CONVOLUTION) != 0 && group_input_channels != 1) { in create_convolution2d_nhwc()
191 xnn_operator_type_to_string(operator_type), group_input_channels); in create_convolution2d_nhwc()
222 …if (group_input_channels == 1 && group_output_channels == 1 && kernel_size == 1 && unit_subsamplin… in create_convolution2d_nhwc()
224 } else if (group_input_channels == 1 && group_output_channels == 1 && (dwconv_ukernel = in create_convolution2d_nhwc()
309 const size_t k_stride = round_up_po2(group_input_channels, kr); in create_convolution2d_nhwc()
[all …]
Ddeconvolution-nhwc.c53 size_t group_input_channels, in create_deconvolution2d_nhwc() argument
114 if (group_input_channels == 0) { in create_deconvolution2d_nhwc()
117 xnn_operator_type_to_string(operator_type), group_input_channels); in create_deconvolution2d_nhwc()
128 const size_t input_channels = groups * group_input_channels; in create_deconvolution2d_nhwc()
134 input_pixel_stride, groups, group_input_channels); in create_deconvolution2d_nhwc()
174 const uint32_t k_stride = round_up_po2(group_input_channels, kr); in create_deconvolution2d_nhwc()
218 groups, group_output_channels, kernel_size, group_input_channels, in create_deconvolution2d_nhwc()
225 groups, group_output_channels, kernel_height, kernel_width, group_input_channels, in create_deconvolution2d_nhwc()
257 deconvolution_op->group_input_channels = group_input_channels; in create_deconvolution2d_nhwc()
313 size_t group_input_channels, in xnn_create_deconvolution2d_nhwc_qu8() argument
[all …]
Dfully-connected-nc.c131 fully_connected_op->group_input_channels = input_channels; in create_fully_connected_nc()
193 const size_t input_channels = fully_connected_op->group_input_channels; in setup_fully_connected_nc()
/external/XNNPACK/bench/
Df32-im2col-gemm.cc43 const size_t group_input_channels = state.range(8); in Im2ColGEMMBenchmark() local
59 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); in Im2ColGEMMBenchmark()
61 std::vector<float> a(input_height * input_width * group_input_channels); in Im2ColGEMMBenchmark()
63 std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in Im2ColGEMMBenchmark()
76 xnn_pack_f32_gemm_goi_w(1 /* groups */, group_output_channels, group_input_channels * kernel_size, in Im2ColGEMMBenchmark()
82 …std::vector<float> im2col_buffer(output_size * group_input_channels * kernel_size * group_output_c… in Im2ColGEMMBenchmark()
105 group_input_channels * sizeof(float) /* input channels */, in Im2ColGEMMBenchmark()
106 group_input_channels * sizeof(float) /* input stride */, in Im2ColGEMMBenchmark()
116 mb, nb, kernel_size * group_input_channels * sizeof(float), in Im2ColGEMMBenchmark()
117 …inputData + m * kernel_size * group_input_channels, kernel_size * group_input_channels * sizeof(fl… in Im2ColGEMMBenchmark()
[all …]
Df16-igemm.cc49 const size_t group_input_channels = state.range(8); in IGEMMBenchmark() local
58 const size_t input_pixel_stride = group_input_channels; in IGEMMBenchmark()
69 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); in IGEMMBenchmark()
73 …d::vector<uint16_t> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in IGEMMBenchmark()
78 std::vector<uint16_t> z(group_input_channels); in IGEMMBenchmark()
90 1 /* groups */, group_output_channels, kernel_size, group_input_channels, in IGEMMBenchmark()
103 convolution_op.group_input_channels = group_input_channels; in IGEMMBenchmark()
144 mb, nb, group_input_channels * sizeof(uint16_t), kernel_size * mr * sizeof(void*), in IGEMMBenchmark()
161 group_input_channels * group_output_channels * in IGEMMBenchmark()
Dconvolution.cc52 const size_t group_input_channels = state.range(10); in xnnpack_convolution_qu8() local
61 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_qu8()
73 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_qu8()
97 groups, group_input_channels, group_output_channels, in xnnpack_convolution_qu8()
154 groups * group_input_channels * group_output_channels * in xnnpack_convolution_qu8()
172 const size_t group_input_channels = state.range(10); in xnnpack_convolution_qs8() local
182 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_convolution_qs8()
194 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_convolution_qs8()
218 groups, group_input_channels, group_output_channels, in xnnpack_convolution_qs8()
274 groups * group_input_channels * group_output_channels * in xnnpack_convolution_qs8()
[all …]
Ddeconvolution.cc39 const size_t group_input_channels = state.range(10); in xnnpack_deconvolution_qu8() local
48 const size_t input_pixel_stride = groups * group_input_channels; in xnnpack_deconvolution_qu8()
60 …t8_t> kernel(groups * group_output_channels * kernel_height * kernel_width * group_input_channels); in xnnpack_deconvolution_qu8()
84 groups, group_input_channels, group_output_channels, in xnnpack_deconvolution_qu8()
141 groups * group_input_channels * group_output_channels * in xnnpack_deconvolution_qu8()
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()
202 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.cc44 const size_t group_input_channels = state.range(8); in IGEMMBenchmark() local
52 const size_t input_pixel_stride = group_input_channels; in IGEMMBenchmark()
63 const size_t kc_stride = benchmark::utils::RoundUp<size_t>(group_input_channels, kr); in IGEMMBenchmark()
67 std::vector<float> k(group_output_channels * kernel_height * kernel_width * group_input_channels); in IGEMMBenchmark()
72 std::vector<float> z(group_input_channels); in IGEMMBenchmark()
84 1 /* groups */, group_output_channels, kernel_size, group_input_channels, in IGEMMBenchmark()
97 convolution_op.group_input_channels = group_input_channels; in IGEMMBenchmark()
134 mb, nb, group_input_channels * sizeof(float), kernel_size * mr * sizeof(void*), in IGEMMBenchmark()
151 group_input_channels * group_output_channels * in IGEMMBenchmark()
/external/XNNPACK/src/subgraph/
Ddeconvolution-2d.c31 size_t group_input_channels, in xnn_define_deconvolution_2d() argument
75 if (group_input_channels == 0) { in xnn_define_deconvolution_2d()
78 xnn_node_type_to_string(xnn_node_type_deconvolution_2d), group_input_channels); in xnn_define_deconvolution_2d()
155 node->params.deconvolution_2d.group_input_channels = group_input_channels; in xnn_define_deconvolution_2d()
Dconvolution-2d.c29 size_t group_input_channels, in xnn_define_convolution_2d() argument
73 if (group_input_channels == 0) { in xnn_define_convolution_2d()
76 xnn_node_type_to_string(xnn_node_type_convolution_2d), group_input_channels); in xnn_define_convolution_2d()
183 node->params.convolution_2d.group_input_channels = group_input_channels; in xnn_define_convolution_2d()
/external/XNNPACK/src/xnnpack/
Dsubgraph.h135 size_t group_input_channels; member
152 size_t group_input_channels; member
Doperator.h206 size_t group_input_channels; member
/external/XNNPACK/include/
Dxnnpack.h250 size_t group_input_channels,
305 size_t group_input_channels,
1238 size_t group_input_channels,
1270 size_t group_input_channels,
1659 size_t group_input_channels,
1834 size_t group_input_channels,
1941 size_t group_input_channels,
2033 size_t group_input_channels,
2071 size_t group_input_channels,
/external/XNNPACK/src/
Druntime.c232 node->params.convolution_2d.group_input_channels, in xnn_create_runtime_v2()
234 …node->params.convolution_2d.group_input_channels * node->params.convolution_2d.groups /* input_pix… in xnn_create_runtime_v2()
257 node->params.convolution_2d.group_input_channels, in xnn_create_runtime_v2()
259 …node->params.convolution_2d.group_input_channels * node->params.convolution_2d.groups /* input_pix… in xnn_create_runtime_v2()
308 node->params.deconvolution_2d.group_input_channels, in xnn_create_runtime_v2()
310 …node->params.deconvolution_2d.group_input_channels * node->params.deconvolution_2d.groups /* input… in xnn_create_runtime_v2()
Dsubgraph.c161 if (node->params.convolution_2d.group_input_channels != 3) { in xnn_check_nchw_compatibility()