| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | f32-conv-hwc.cc | 24 .output_channels(8) in TEST()
39 .output_channels(8) in TEST()
55 .output_channels(8) in TEST()
71 .output_channels(8) in TEST()
80 for (size_t output_channels = 1; output_channels < 8; output_channels++) { in TEST() local
88 .output_channels(output_channels) in TEST()
98 for (size_t output_channels = 16; output_channels <= 32; output_channels += 8) { in TEST() local
106 .output_channels(output_channels) in TEST()
116 for (size_t output_channels = 9; output_channels < 16; output_channels++) { in TEST() local
124 .output_channels(output_channels) in TEST()
[all …]
|
| H A D | f32-conv-hwc2chw.cc | 24 .output_channels(4) in TEST()
39 .output_channels(4) in TEST()
55 .output_channels(4) in TEST()
71 .output_channels(4) in TEST()
80 for (size_t output_channels = 1; output_channels < 4; output_channels++) { in TEST() local
88 .output_channels(output_channels) in TEST()
98 for (size_t output_channels = 8; output_channels <= 16; output_channels += 4) { in TEST() local
106 .output_channels(output_channels) in TEST()
116 for (size_t output_channels = 5; output_channels < 8; output_channels++) { in TEST() local
124 .output_channels(output_channels) in TEST()
[all …]
|
| H A D | f16-conv-hwc2chw.cc | 24 .output_channels(4) in TEST()
39 .output_channels(4) in TEST()
55 .output_channels(4) in TEST()
71 .output_channels(4) in TEST()
80 for (size_t output_channels = 1; output_channels < 4; output_channels++) { in TEST() local
88 .output_channels(output_channels) in TEST()
98 for (size_t output_channels = 8; output_channels <= 16; output_channels += 4) { in TEST() local
106 .output_channels(output_channels) in TEST()
116 for (size_t output_channels = 5; output_channels < 8; output_channels++) { in TEST() local
124 .output_channels(output_channels) in TEST()
[all …]
|
| H A D | fully-connected-nc.cc | 18 .output_channels(19) in TEST()
27 .output_channels(19) in TEST()
37 .output_channels(19) in TEST()
48 .output_channels(19) in TEST()
57 .output_channels(19) in TEST()
68 .output_channels(19) in TEST()
78 .output_channels(19) in TEST()
87 .output_channels(19) in TEST()
96 .output_channels(19) in TEST()
106 .output_channels(19) in TEST()
[all …]
|
| H A D | depth-to-space-nhwc.cc | 15 .output_channels(17) in TEST()
24 .output_channels(17) in TEST()
34 .output_channels(17) in TEST()
45 .output_channels(17) in TEST()
56 .output_channels(17) in TEST()
62 for (size_t output_channels = 1; output_channels <= 15; output_channels++) { in TEST() local
66 .output_channels(output_channels) in TEST()
77 .output_channels(17) in TEST()
88 .output_channels(17) in TEST()
98 .output_channels(17) in TEST()
[all …]
|
| H A D | fully-connected-operator-tester.h | 45 inline FullyConnectedOperatorTester& output_channels(size_t output_channels) { in output_channels() argument
46 assert(output_channels >= 1); in output_channels()
47 this->output_channels_ = output_channels; in output_channels()
51 inline size_t output_channels() const { in output_channels() function
88 return output_channels(); in output_stride()
90 assert(this->output_stride_ >= output_channels()); in output_stride()
171 std::vector<int8_t> kernel(output_channels() * input_channels()); in TestQS8()
172 std::vector<int32_t> bias(output_channels()); in TestQS8()
173 std::vector<int8_t> output((batch_size() - 1) * output_stride() + output_channels()); in TestQS8()
174 std::vector<int32_t> accumulators(batch_size() * output_channels()); in TestQS8()
[all …]
|
| H A D | conv-hwc2chw-microkernel-tester.h | 138 inline ConvHWC2CHWMicrokernelTester& output_channels(size_t output_channels) { in output_channels() function
139 assert(output_channels >= 1); in output_channels()
140 this->output_channels_ = output_channels; in output_channels()
144 inline size_t output_channels() const { in output_channels() function
149 …return output_channels() % output_channels_tile() == 0 ? output_channels() : output_channels() / o… in packed_output_channels()
243 return output_channels(); in output_pixel_stride()
305 …std::vector<float> kernel(output_channels() * kernel_height() * kernel_width() * input_channels());
306 std::vector<float> bias(output_channels());
307 std::vector<float> output(batch_size() * output_channels() * output_height() * output_width());
308 …std::vector<float> output_ref(batch_size() * output_channels() * output_height() * output_width());
[all …]
|
| H A D | depth-to-space-operator-tester.h | 71 return output_channels() * block_size() * block_size(); in input_channels()
74 inline DepthToSpaceOperatorTester& output_channels(size_t output_channels) { in output_channels() function
75 assert(output_channels != 0); in output_channels()
76 this->output_channels_ = output_channels; in output_channels()
80 inline size_t output_channels() const { in output_channels() function
117 return output_channels(); in output_channels_stride()
119 assert(this->output_channels_stride_ >= output_channels()); in output_channels_stride()
143 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNHWCxX8()
154 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX8()
176 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX8()
[all …]
|
| H A D | depth-to-space-nchw2nhwc.cc | 15 .output_channels(17) in TEST()
24 .output_channels(17) in TEST()
34 .output_channels(17) in TEST()
45 .output_channels(17) in TEST()
56 .output_channels(17) in TEST()
62 for (size_t output_channels = 1; output_channels <= 15; output_channels++) { in TEST() local
66 .output_channels(output_channels) in TEST()
77 .output_channels(17) in TEST()
88 .output_channels(17) in TEST()
98 .output_channels(17) in TEST()
|
| H A D | conv-hwc-microkernel-tester.h | 136 inline ConvHWCMicrokernelTester& output_channels(size_t output_channels) { in output_channels() argument
137 assert(output_channels >= 1); in output_channels()
138 this->output_channels_ = output_channels; in output_channels()
142 inline size_t output_channels() const { in output_channels() function
147 …return output_channels() % output_channels_tile() == 0 ? output_channels() : output_channels() / o… in packed_output_channels()
241 return output_channels(); in output_pixel_stride()
297 …std::vector<float> kernel(output_channels() * kernel_height() * kernel_width() * input_channels());
298 std::vector<float> bias(output_channels());
299 …tch_size() * ((output_height() * output_width() - 1) * output_pixel_stride() + output_channels()));
300 …std::vector<float> output_ref(batch_size() * output_height() * output_width() * output_channels());
[all …]
|
| H A D | deconvolution-nhwc.cc | 73 …for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *=… variable
78 .group_output_channels(output_channels)
202 …for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *=… in TEST() local
208 .group_output_channels(output_channels) in TEST()
337 …for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *=… in TEST() local
343 .group_output_channels(output_channels) in TEST()
476 …for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *=… in TEST() local
483 .group_output_channels(output_channels) in TEST()
713 …for (size_t output_channels = 1; output_channels <= xnn_params.qs8.gemm.nr * 2; output_channels *=… variable
719 .group_output_channels(output_channels)
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | fully_connected_test.cc | 38 const auto output_channels = channels_rng(); variable
43 .OutputChannels(output_channels)
57 const auto output_channels = channels_rng(); variable
62 .OutputChannels(output_channels)
80 const auto output_channels = channels_rng(); variable
85 .OutputChannels(output_channels)
102 const auto output_channels = channels_rng(); variable
107 .OutputChannels(output_channels)
126 const auto output_channels = channels_rng(); variable
131 .OutputChannels(output_channels)
[all …]
|
| H A D | signed_quantized_fully_connected_test.cc | 42 const auto output_channels = channels_rng(); variable
49 .OutputChannels(output_channels)
67 const auto output_channels = channels_rng(); variable
74 .OutputChannels(output_channels)
96 const auto output_channels = channels_rng(); variable
103 .OutputChannels(output_channels)
124 const auto output_channels = channels_rng(); variable
131 .OutputChannels(output_channels)
154 const auto output_channels = channels_rng(); variable
161 .OutputChannels(output_channels)
[all …]
|
| H A D | unsigned_quantized_fully_connected_test.cc | 44 const auto output_channels = channels_rng(); variable
52 .OutputChannels(output_channels)
72 const auto output_channels = channels_rng(); variable
80 .OutputChannels(output_channels)
104 const auto output_channels = channels_rng(); variable
112 .OutputChannels(output_channels)
135 const auto output_channels = channels_rng(); variable
143 .OutputChannels(output_channels)
168 const auto output_channels = channels_rng(); variable
176 .OutputChannels(output_channels)
[all …]
|
| /aosp_15_r20/external/webrtc/audio/utility/ |
| H A D | channel_mixing_matrix_unittest.cc | 55 const int output_channels = ChannelLayoutToChannelCount(output_layout); in TEST() local
66 for (int i = 0; i < output_channels; i++) { in TEST()
182 const int output_channels = ChannelLayoutToChannelCount(output_layout); in TEST() local
184 output_layout, output_channels); in TEST()
197 EXPECT_EQ(static_cast<size_t>(output_channels), matrix.size()); in TEST()
198 for (int n = 0; n < output_channels; n++) { in TEST()
212 const int output_channels = ChannelLayoutToChannelCount(output_layout); in TEST() local
214 output_layout, output_channels); in TEST()
227 EXPECT_EQ(static_cast<size_t>(output_channels), matrix.size()); in TEST()
228 for (int n = 0; n < output_channels; n++) { in TEST()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/benchmarks/ |
| H A D | stateful_conv1d.cpp | 11 const size_t output_channels = static_cast<size_t>(state.range(1)); in stateful_conv1d() local
18 m.register_parameter("weight_1", torch::rand({output_channels, input_channels, kernel}), false); in stateful_conv1d()
19 m.register_parameter("bias_1", torch::rand({output_channels}), false); in stateful_conv1d()
20 m.register_parameter("weight_2", torch::rand({output_channels, output_channels, kernel}), false); in stateful_conv1d()
21 m.register_parameter("bias_2", torch::rand({output_channels}), false); in stateful_conv1d()
22 m.register_parameter("weight_3", torch::rand({output_channels, output_channels, kernel}), false); in stateful_conv1d()
23 m.register_parameter("bias_3", torch::rand({output_channels}), false); in stateful_conv1d()
24 m.register_parameter("weight_4", torch::rand({output_channels, output_channels, kernel}), false); in stateful_conv1d()
25 m.register_parameter("bias_4", torch::rand({output_channels}), false); in stateful_conv1d()
71 for (size_t output_channels = 32; output_channels < 256; output_channels *= 2) { in GenerateSizes() local
[all …]
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/ |
| H A D | etnaviv_ml_nn.c | 190 unsigned new_size = operation->output_channels * 2 * 2 * operation->input_channels; in pointwise_to_2x2()
195 for (unsigned channel = 0; channel < operation->output_channels; channel++) { in pointwise_to_2x2()
217 …unsigned new_size = operation->output_channels * operation->weight_width * operation->weight_heigh… in expand_depthwise()
223 for (unsigned channel = 0; channel < operation->output_channels; channel++) { in expand_depthwise()
224 unsigned in_channel = channel / operation->output_channels; in expand_depthwise()
225 unsigned in_depth = channel % operation->output_channels; in expand_depthwise()
247 unsigned new_size = operation->output_channels * operation->weight_width * \ in transpose()
252 unsigned output_channels = operation->output_channels; in transpose() local
256 output_channels = 1; in transpose()
262 for (unsigned d0 = 0; d0 < output_channels; d0++) in transpose()
[all …]
|
| /aosp_15_r20/external/XNNPACK/src/operators/ |
| H A D | depth-to-space-nhwc.c | 17 size_t output_channels, in create_depth_to_space_nhwc() argument
36 if (output_channels == 0) { in create_depth_to_space_nhwc()
38 xnn_operator_type_to_string(operator_type), output_channels); in create_depth_to_space_nhwc()
42 if (output_channel_stride < output_channels) { in create_depth_to_space_nhwc()
47 output_channel_stride, output_channels); in create_depth_to_space_nhwc()
58 const size_t input_channels = output_channels * block_size * block_size; in create_depth_to_space_nhwc()
78 depth_to_space_op->channels = output_channels; in create_depth_to_space_nhwc()
97 size_t output_channels, in xnn_create_depth_to_space_nhwc_x8() argument
105 output_channels, in xnn_create_depth_to_space_nhwc_x8()
115 size_t output_channels, in xnn_create_depth_to_space_nhwc_x16() argument
[all …]
|
| H A D | fully-connected-nc.c | 29 size_t output_channels, in create_fully_connected_nc() argument
77 if (output_channels == 0) { in create_fully_connected_nc()
80 xnn_operator_type_to_string(operator_type), output_channels); in create_fully_connected_nc()
92 if (output_stride < output_channels) { in create_fully_connected_nc()
96 xnn_operator_type_to_string(operator_type), output_stride, output_channels); in create_fully_connected_nc()
118 const size_t n_stride = round_up(output_channels, nr); in create_fully_connected_nc()
134 output_channels, input_channels, in create_fully_connected_nc()
141 1, output_channels, input_channels, in create_fully_connected_nc()
155 fully_connected_op->group_output_channels = output_channels; in create_fully_connected_nc()
239 const size_t output_channels = fully_connected_op->group_output_channels; in setup_fully_connected_nc() local
[all …]
|
| H A D | transpose-nd.c | 534 size_t output_channels, in xnn_create_depth_to_space_nchw2nhwc_x32() argument
552 if (output_channels == 0) { in xnn_create_depth_to_space_nchw2nhwc_x32()
554 xnn_operator_type_to_string(xnn_operator_type_depth_to_space_nchw2nhwc_x32), output_channels); in xnn_create_depth_to_space_nchw2nhwc_x32()
558 if (output_channel_stride < output_channels) { in xnn_create_depth_to_space_nchw2nhwc_x32()
563 output_channel_stride, output_channels); in xnn_create_depth_to_space_nchw2nhwc_x32()
574 const size_t input_channels = output_channels * block_size * block_size; in xnn_create_depth_to_space_nchw2nhwc_x32()
594 depth_to_space_op->channels = output_channels; in xnn_create_depth_to_space_nchw2nhwc_x32()
681 size_t output_channels, in create_depth_to_space_nhwc() argument
700 if (output_channels == 0) { in create_depth_to_space_nhwc()
702 xnn_operator_type_to_string(operator_type), output_channels); in create_depth_to_space_nhwc()
[all …]
|
| /aosp_15_r20/external/XNNPACK/bench/ |
| H A D | f16-conv-hwc2chw.cc | 40 const size_t output_channels = state.range(2); in f16_conv_hwc2chw() local
57 std::vector<uint16_t> kernel(output_channels * kernel_size * kernel_size * input_channels); in f16_conv_hwc2chw()
59 std::vector<uint16_t> bias(output_channels); in f16_conv_hwc2chw()
65 benchmark::utils::RoundUp<size_t>(output_channels, output_channels_tile); in f16_conv_hwc2chw()
66 const size_t output_elements = output_height * output_width * output_channels; in f16_conv_hwc2chw()
74 output_channels, input_channels, output_channels_tile, in f16_conv_hwc2chw()
102 padding, output_channels, in f16_conv_hwc2chw()
103 output_channels * output_width * sizeof(uint16_t), in f16_conv_hwc2chw()
104 output_channels * sizeof(uint16_t), in f16_conv_hwc2chw()
116 input_channels * output_channels * in f16_conv_hwc2chw()
|
| H A D | f32-conv-hwc2chw.cc | 39 const size_t output_channels = state.range(2); in f32_conv_hwc2chw() local
55 std::vector<float> kernel(output_channels * kernel_size * kernel_size * input_channels); in f32_conv_hwc2chw()
57 std::vector<float> bias(output_channels); in f32_conv_hwc2chw()
63 benchmark::utils::RoundUp<size_t>(output_channels, output_channels_tile); in f32_conv_hwc2chw()
64 const size_t output_elements = output_height * output_width * output_channels; in f32_conv_hwc2chw()
72 output_channels, input_channels, output_channels_tile, in f32_conv_hwc2chw()
100 padding, output_channels, in f32_conv_hwc2chw()
101 output_channels * output_width * sizeof(float), in f32_conv_hwc2chw()
102 output_channels * sizeof(float), in f32_conv_hwc2chw()
114 input_channels * output_channels * in f32_conv_hwc2chw()
|
| H A D | deconvolution.cc | 42 const size_t output_channels = state.range(12); in xnnpack_deconvolution_qu8() local
62 std::vector<uint8_t> kernel(output_channels * kernel_height * kernel_width * input_channels); in xnnpack_deconvolution_qu8()
64 std::vector<int32_t> bias(output_channels); in xnnpack_deconvolution_qu8()
66 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_qu8()
86 /*groups=*/1, input_channels, output_channels, in xnnpack_deconvolution_qu8()
87 /*input_pixel_stride=*/input_channels, /*output_pixel_stride=*/output_channels, in xnnpack_deconvolution_qu8()
144 input_channels * output_channels * in xnnpack_deconvolution_qu8()
163 const size_t output_channels = state.range(12); in xnnpack_deconvolution_f32() local
181 std::vector<float> kernel(output_channels * kernel_height * kernel_width * input_channels); in xnnpack_deconvolution_f32()
183 std::vector<float> bias(output_channels); in xnnpack_deconvolution_f32()
[all …]
|
| H A D | f32-conv-hwc.cc | 37 const size_t output_channels = state.range(2); in f32_conv_hwc() local
53 std::vector<float> kernel(output_channels * kernel_size * kernel_size * input_channels); in f32_conv_hwc()
55 std::vector<float> bias(output_channels); in f32_conv_hwc()
61 benchmark::utils::RoundUp<size_t>(output_channels, output_channels_tile); in f32_conv_hwc()
62 const size_t output_elements = output_height * output_width * output_channels; in f32_conv_hwc()
70 output_channels, input_channels, output_channels_tile, in f32_conv_hwc()
99 padding, output_channels, in f32_conv_hwc()
100 output_channels * output_width * sizeof(float), in f32_conv_hwc()
101 output_channels * sizeof(float), in f32_conv_hwc()
113 input_channels * output_channels * in f32_conv_hwc()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | conv.h | 185 int64_t output_channels, in Conv1dImpl() argument
188 Conv1dOptions(input_channels, output_channels, kernel_size)) {} in Conv1dImpl()
217 int64_t output_channels, in Conv2dImpl() argument
220 Conv2dOptions(input_channels, output_channels, kernel_size)) {} in Conv2dImpl()
252 int64_t output_channels, in Conv3dImpl() argument
255 Conv3dOptions(input_channels, output_channels, kernel_size)) {} in Conv3dImpl()
344 int64_t output_channels, in ConvTranspose1dImpl() argument
348 output_channels, in ConvTranspose1dImpl()
386 int64_t output_channels, in ConvTranspose2dImpl() argument
390 output_channels, in ConvTranspose2dImpl()
[all …]
|