| /external/XNNPACK/test/ |
| D | average-pooling-operator-tester.h | 298 inline AveragePoolingOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() argument
299 assert(output_pixel_stride != 0); in output_pixel_stride()
300 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
304 inline size_t output_pixel_stride() const { in output_pixel_stride() function
454 …_t> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestQU8()
495 channels(), input_pixel_stride(), output_pixel_stride(), in TestQU8()
520 …uint32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]), ui… in TestQU8()
521 …uint32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]), ui… in TestQU8()
522 …oat(int32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c])), in TestQU8()
538 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestF32()
[all …]
|
| D | max-pooling-operator-tester.h | 340 inline MaxPoolingOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() argument
341 assert(output_pixel_stride != 0); in output_pixel_stride()
342 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
346 inline size_t output_pixel_stride() const { in output_pixel_stride() function
456 …_t> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestU8()
497 channels(), input_pixel_stride(), output_pixel_stride(), in TestU8()
521 …uint32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]), ui… in TestU8()
522 …uint32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]), ui… in TestU8()
524 … uint32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c])) << in TestU8()
539 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestF32()
[all …]
|
| D | deconvolution-operator-tester.h | 361 inline DeconvolutionOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() argument
362 assert(output_pixel_stride >= 1); in output_pixel_stride()
363 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
367 inline size_t output_pixel_stride() const { in output_pixel_stride() function
496 …_t> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + groups(… in TestQU8()
581 input_pixel_stride(), output_pixel_stride(), in TestQU8()
609 …(int32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + g * gro… in TestQU8()
611 …(int32_t(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + g * gro… in TestQU8()
615 …double(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + g * group… in TestQU8()
635 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + groups(… in TestF32()
[all …]
|
| D | unpooling-operator-tester.h | 196 inline UnpoolingOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() function
197 assert(output_pixel_stride != 0); in output_pixel_stride()
198 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
202 inline size_t output_pixel_stride() const { in output_pixel_stride() function
288 …_t> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestX32()
321 channels(), input_pixel_stride(), output_pixel_stride(), in TestX32()
344 … output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]) << in TestX32()
366 (batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channels(), in TestSetupX32()
367 …(next_batch_size() * next_output_height() * next_output_width() - 1) * output_pixel_stride() * cha… in TestSetupX32()
402 channels(), input_pixel_stride(), output_pixel_stride(), in TestSetupX32()
[all …]
|
| D | resize-bilinear-operator-tester.h | 131 inline ResizeBilinearOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() argument
132 assert(output_pixel_stride != 0); in output_pixel_stride()
133 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
137 inline size_t output_pixel_stride() const { in output_pixel_stride() function
233 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestNHWCxF32()
269 channels(), input_pixel_stride(), output_pixel_stride(), in TestNHWCxF32()
293 … ASSERT_NEAR(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c], in TestNHWCxF32()
314 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestNCHWxF32()
354 channels(), input_pixel_stride(), output_pixel_stride(), in TestNCHWxF32()
|
| D | argmax-pooling-operator-tester.h | 247 inline ArgmaxPoolingOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() function
248 assert(output_pixel_stride != 0); in output_pixel_stride()
249 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
253 inline size_t output_pixel_stride() const { in output_pixel_stride() function
337 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestF32()
384 channels(), input_pixel_stride(), output_pixel_stride(), in TestF32()
408 … output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]) << in TestF32()
429 (batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channels(), in TestSetupF32()
430 …(next_batch_size() * next_output_height() * next_output_width() - 1) * output_pixel_stride() + cha… in TestSetupF32()
480 channels(), input_pixel_stride(), output_pixel_stride(), in TestSetupF32()
[all …]
|
| D | conv-hwc-microkernel-tester.h | 241 inline size_t output_pixel_stride() const { in output_pixel_stride() function
300 …t> output(batch_size() * ((output_height() * output_width() - 1) * output_pixel_stride() + output_…
374 output_pixel_stride() * output_width() * sizeof(float),
375 output_pixel_stride() * sizeof(float),
383 …SSERT_GE(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c], out…
385 …SSERT_LE(output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c], out…
389 … output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c],
|
| D | argmax-pooling-nhwc.cc | 180 .output_pixel_stride(5 * channels) in TEST()
189 .output_pixel_stride(5 * channels) in TEST()
354 .output_pixel_stride(5 * channels) in TEST()
363 .output_pixel_stride(5 * channels) in TEST()
430 .output_pixel_stride(5 * channels) in TEST()
439 .output_pixel_stride(5 * channels) in TEST()
509 .output_pixel_stride(5 * channels) in TEST()
518 .output_pixel_stride(5 * channels) in TEST()
|
| D | max-pooling-nhwc.cc | 239 .output_pixel_stride(5 * channels) in TEST()
248 .output_pixel_stride(5 * channels) in TEST()
529 .output_pixel_stride(5 * channels) in TEST()
538 .output_pixel_stride(5 * channels) in TEST()
657 .output_pixel_stride(5 * channels) in TEST()
666 .output_pixel_stride(5 * channels) in TEST()
733 .output_pixel_stride(5 * channels) in TEST()
742 .output_pixel_stride(5 * channels) in TEST()
1055 .output_pixel_stride(5 * channels) in TEST()
1064 .output_pixel_stride(5 * channels) in TEST()
[all …]
|
| /external/XNNPACK/src/operators/ |
| D | resize-bilinear-nhwc.c | 26 size_t output_pixel_stride, in xnn_create_resize_bilinear2d_nhwc_f32() argument
56 if (output_pixel_stride < channels) { in xnn_create_resize_bilinear2d_nhwc_f32()
60 …xnn_operator_type_to_string(xnn_operator_type_resize_bilinear_nhwc_f32), output_pixel_stride, chan… in xnn_create_resize_bilinear2d_nhwc_f32()
76 resize_op->output_pixel_stride = output_pixel_stride; in xnn_create_resize_bilinear2d_nhwc_f32()
195 const size_t output_pixel_stride_in_bytes = resize_op->output_pixel_stride * sizeof(float); in xnn_setup_resize_bilinear2d_nhwc_f32()
203 .output_pixel_stride = output_pixel_stride_in_bytes, in xnn_setup_resize_bilinear2d_nhwc_f32()
|
| D | resize-bilinear-nchw.c | 26 size_t output_pixel_stride, in xnn_create_resize_bilinear2d_nchw_f32() argument
56 if (output_pixel_stride < channels) { in xnn_create_resize_bilinear2d_nchw_f32()
60 …xnn_operator_type_to_string(xnn_operator_type_resize_bilinear_nchw_f32), output_pixel_stride, chan… in xnn_create_resize_bilinear2d_nchw_f32()
76 resize_op->output_pixel_stride = output_pixel_stride; in xnn_create_resize_bilinear2d_nchw_f32()
204 ….output_batch_stride = resize_op->output_pixel_stride * output_height * output_width * sizeof(floa… in xnn_setup_resize_bilinear2d_nchw_f32()
|
| D | max-pooling-nhwc.c | 58 size_t output_pixel_stride, in create_max_pooling2d_nhwc() argument
132 if (output_pixel_stride < channels) { in create_max_pooling2d_nhwc()
136 xnn_operator_type_to_string(operator_type), output_pixel_stride, channels); in create_max_pooling2d_nhwc()
175 max_pooling_op->output_pixel_stride = output_pixel_stride; in create_max_pooling2d_nhwc()
294 …const size_t output_width_stride = max_pooling_op->output_pixel_stride << log2_output_element_size; in setup_max_pooling2d_nhwc()
337 size_t output_pixel_stride, in xnn_create_max_pooling2d_nhwc_u8() argument
356 channels, input_pixel_stride, output_pixel_stride, in xnn_create_max_pooling2d_nhwc_u8()
376 size_t output_pixel_stride, in xnn_create_max_pooling2d_nhwc_f32() argument
409 channels, input_pixel_stride, output_pixel_stride, in xnn_create_max_pooling2d_nhwc_f32()
|
| D | deconvolution-nhwc.c | 56 size_t output_pixel_stride, in create_deconvolution2d_nhwc() argument
139 if (output_pixel_stride < output_channels) { in create_deconvolution2d_nhwc()
144 output_pixel_stride, groups, group_output_channels); in create_deconvolution2d_nhwc()
260 deconvolution_op->output_pixel_stride = output_pixel_stride; in create_deconvolution2d_nhwc()
316 size_t output_pixel_stride, in xnn_create_deconvolution2d_nhwc_qu8() argument
380 input_pixel_stride, output_pixel_stride, in xnn_create_deconvolution2d_nhwc_qu8()
409 size_t output_pixel_stride, in xnn_create_deconvolution2d_nhwc_f32() argument
458 input_pixel_stride, output_pixel_stride, in xnn_create_deconvolution2d_nhwc_f32()
534 .cm_stride = deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
540 .bc_stride = output_size * deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
[all …]
|
| D | unpooling-nhwc.c | 41 size_t output_pixel_stride, in xnn_create_unpooling2d_nhwc_x32() argument
87 if (output_pixel_stride < channels) { in xnn_create_unpooling2d_nhwc_x32()
91 … xnn_operator_type_to_string(xnn_operator_type_unpooling_nhwc_x32), output_pixel_stride, channels); in xnn_create_unpooling2d_nhwc_x32()
114 unpooling_op->output_pixel_stride = output_pixel_stride; in xnn_create_unpooling2d_nhwc_x32()
|
| D | depth-to-space-nhwc.c | 79 depth_to_space_op->output_pixel_stride = output_channel_stride; in xnn_create_depth_to_space_nhwc_x32()
141 .output_height_stride = output_width * depth_to_space_op->output_pixel_stride * sizeof(float), in xnn_setup_depth_to_space_nhwc_x32()
142 .output_width_stride = depth_to_space_op->output_pixel_stride * sizeof(float), in xnn_setup_depth_to_space_nhwc_x32()
145 if (depth_to_space_op->output_pixel_stride == output_channels) { in xnn_setup_depth_to_space_nhwc_x32()
|
| D | argmax-pooling-nhwc.c | 51 size_t output_pixel_stride, in xnn_create_argmax_pooling2d_nhwc_f32() argument
97 if (output_pixel_stride < channels) { in xnn_create_argmax_pooling2d_nhwc_f32()
101 …xnn_operator_type_to_string(xnn_operator_type_argmax_pooling_nhwc_f32), output_pixel_stride, chann… in xnn_create_argmax_pooling2d_nhwc_f32()
140 argmax_pooling_op->output_pixel_stride = output_pixel_stride; in xnn_create_argmax_pooling2d_nhwc_f32()
253 const size_t output_width_stride = argmax_pooling_op->output_pixel_stride * sizeof(float); in xnn_setup_argmax_pooling2d_nhwc_f32()
|
| D | average-pooling-nhwc.c | 54 size_t output_pixel_stride, in xnn_create_average_pooling2d_nhwc_qu8() argument
113 if (output_pixel_stride < channels) { in xnn_create_average_pooling2d_nhwc_qu8()
117 …xnn_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_qu8), output_pixel_stride, chan… in xnn_create_average_pooling2d_nhwc_qu8()
209 average_pooling_op->output_pixel_stride = output_pixel_stride; in xnn_create_average_pooling2d_nhwc_qu8()
250 size_t output_pixel_stride, in xnn_create_average_pooling2d_nhwc_f32() argument
305 if (output_pixel_stride < channels) { in xnn_create_average_pooling2d_nhwc_f32()
309 …xnn_operator_type_to_string(xnn_operator_type_average_pooling_nhwc_f32), output_pixel_stride, chan… in xnn_create_average_pooling2d_nhwc_f32()
379 average_pooling_op->output_pixel_stride = output_pixel_stride; in xnn_create_average_pooling2d_nhwc_f32()
493 .output_batch_stride = average_pooling_op->output_pixel_stride << log2_output_element_size, in setup_average_pooling2d()
542 …const size_t output_width_stride = average_pooling_op->output_pixel_stride << log2_output_element_… in setup_average_pooling2d()
|
| D | depth-to-space-nchw2nhwc.c | 79 depth_to_space_op->output_pixel_stride = output_channel_stride; in xnn_create_depth_to_space_nchw2nhwc_x32()
140 ….output_batch_stride = depth_to_space_op->output_pixel_stride * output_height * output_width * siz… in xnn_setup_depth_to_space_nchw2nhwc_x32()
141 .output_channel_stride = depth_to_space_op->output_pixel_stride, in xnn_setup_depth_to_space_nchw2nhwc_x32()
|
| D | softmax-nc.c | 123 softmax_op->output_pixel_stride = output_stride; in xnn_create_softmax_nc_qu8()
173 .y_stride = softmax_op->output_pixel_stride * sizeof(uint8_t), in xnn_setup_softmax_nc_qu8()
238 softmax_op->output_pixel_stride = output_stride; in xnn_create_softmax_nc_f32()
287 .y_stride = softmax_op->output_pixel_stride * sizeof(float), in xnn_setup_softmax_nc_f32()
|
| D | prelu-nc.c | 84 prelu_op->output_pixel_stride = output_stride; in xnn_create_prelu_nc_f32()
131 .y_stride = prelu_op->output_pixel_stride * sizeof(float), in xnn_setup_prelu_nc_f32()
|
| D | sigmoid-nc.c | 141 sigmoid_op->output_pixel_stride = output_stride; in xnn_create_sigmoid_nc_qu8()
187 const size_t output_stride = sigmoid_op->output_pixel_stride; in xnn_setup_sigmoid_nc_qu8()
|
| D | channel-shuffle-nc.c | 86 channel_shuffle_op->output_pixel_stride = output_stride; in create_channel_shuffle_nc()
167 .y_stride = channel_shuffle_op->output_pixel_stride << log2_element_size, in setup_channel_shuffle_nc()
|
| D | leaky-relu-nc.c | 150 leaky_relu_op->output_pixel_stride = output_stride; in xnn_create_leaky_relu_nc_qu8()
193 const size_t output_stride = leaky_relu_op->output_pixel_stride; in xnn_setup_leaky_relu_nc_qu8()
|
| /external/XNNPACK/bench/ |
| D | deconvolution.cc | 47 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_deconvolution_qu8() local
64 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_qu8()
85 input_pixel_stride, output_pixel_stride, in xnnpack_deconvolution_qu8()
165 const size_t output_pixel_stride = groups * group_output_channels; in xnnpack_deconvolution_f32() local
182 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_deconvolution_f32()
203 input_pixel_stride, output_pixel_stride, in xnnpack_deconvolution_f32()
|
| /external/XNNPACK/include/ |
| D | xnnpack.h | 1142 size_t output_pixel_stride,
1167 size_t output_pixel_stride,
1273 size_t output_pixel_stride,
1416 size_t output_pixel_stride,
1510 size_t output_pixel_stride,
1528 size_t output_pixel_stride,
1787 size_t output_pixel_stride,
2002 size_t output_pixel_stride,
2074 size_t output_pixel_stride,
2236 size_t output_pixel_stride,
|