Home
last modified time | relevance | path

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

/external/XNNPACK/src/
Daverage-pooling-nhwc.c449 const size_t output_height_stride = output_width * output_width_stride; in xnn_setup_average_pooling2d_nhwc_q8() local
458 .output_batch_stride = output_height * output_height_stride, in xnn_setup_average_pooling2d_nhwc_q8()
459 .output_height_stride = output_height_stride, in xnn_setup_average_pooling2d_nhwc_q8()
577 const size_t output_height_stride = output_width * output_width_stride; in xnn_setup_average_pooling2d_nhwc_f32() local
590 .output_batch_stride = output_height * output_height_stride, in xnn_setup_average_pooling2d_nhwc_f32()
591 .output_height_stride = output_height_stride, in xnn_setup_average_pooling2d_nhwc_f32()
648 .output_batch_stride = output_height * output_height_stride, in xnn_setup_average_pooling2d_nhwc_f32()
649 .output_height_stride = output_height_stride, in xnn_setup_average_pooling2d_nhwc_f32()
Dargmax-pooling-nhwc.c247 const size_t output_height_stride = output_width * output_width_stride; in xnn_setup_argmax_pooling2d_nhwc_f32() local
258 .output_batch_stride = output_height * output_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
259 .output_height_stride = output_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
Dmax-pooling-nhwc.c391 const size_t output_height_stride = output_width * output_width_stride; in setup_max_pooling2d() local
400 .output_batch_stride = output_height * output_height_stride, in setup_max_pooling2d()
401 .output_height_stride = output_height_stride, in setup_max_pooling2d()
Doperator-run.c235 context->output_height_stride, in xnn_compute_dconv2d_hwc2spchw()
282 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_argmax_pooling_unipass()
302 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_argmax_pooling_multipass()
325 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_max_pooling()
363 …t->output + batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_average_pooling_unipass()
381 …t->output + batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_average_pooling_multipass()
402 …t->output + batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_pixelwise_average_pooling_unipass()
422 …t->output + batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_pixelwise_average_pooling_multipass()
Dconvolution-nchw.c630 .output_height_stride = output_width << log2_output_element_size, in setup_convolution2d_nchw()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h2622 const int output_height_stride = function_params->output_height_stride;
2653 output_block_data + output_height_stride * k_height + 8 * j_depth;
2786 const int output_height_stride = function_params->output_height_stride;
2813 output_block_data + output_height_stride * k_height + 8 * j_depth;
2873 const int output_height_stride = function_params->output_height_stride;
2931 output_block_data + output_height_stride * k_height + 8 * j_depth;
3085 const int output_height_stride = function_params->output_height_stride;
3134 output_block_data + output_height_stride * k_height + 8 * j_depth;
3274 const int output_height_stride = function_params->output_height_stride;
3426 vst1q_lane_8x4(output_data + output_height_stride, acc_u8_all, 1);
[all …]
Ddepthwiseconv_uint8_3x3_filter.h276 static_assert(offsetof(DepthwiseConvDotProdParams, output_height_stride) ==
10251 const int output_height_stride = output_width * output_depth;
10253 const int output_batch_stride = output_height_stride * output_height;
10266 function_params.output_height_stride = output_height_stride;
10349 row_start * output_height_stride + j_depth * 64 +
10409 output_data_block += output_height_stride * row_count_per_macro;
Ddepthwiseconv_3x3_filter_common.h372 int32 output_height_stride;
/external/XNNPACK/src/xnnpack/
Dconv.h34 size_t output_height_stride, \
56 size_t output_height_stride, \
Dcompute.h220 size_t output_height_stride; member
300 size_t output_height_stride; member
350 size_t output_height_stride; member
386 size_t output_height_stride; member
423 size_t output_height_stride; member
Dparams.h506 size_t output_height_stride,
521 size_t output_height_stride,
536 size_t output_height_stride,
551 size_t output_height_stride,
674 size_t output_height_stride,
686 size_t output_height_stride,
Ddwconv.h157 size_t output_height_stride, \
/external/XNNPACK/src/f32-conv-hwc2spchw/
D3x3s2p1c3x4-neonfma-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__neonfma_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__neonfma_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__neonfma_2x2()
645 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__neonfma_2x2()
646 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__neonfma_2x2()
D3x3s2p1c3x4-scalar-1x1.c23 size_t output_height_stride, in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__scalar_1x1() argument
41 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__scalar_1x1()
663 output0 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2spchw_ukernel_3x3s2p1c3x4__scalar_1x1()
/external/XNNPACK/src/f32-conv-hwc/
D3x3s2p1c3x4-neonfma-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x2()
659 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x2()
660 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x2()
D3x3s2p0p1c3x4-scalar-1x1.c23 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__scalar_1x1() argument
42 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__scalar_1x1()
678 output0 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__scalar_1x1()
D3x3s2p1c3x4-scalar-1x1.c23 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__scalar_1x1() argument
42 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__scalar_1x1()
678 output0 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__scalar_1x1()
D3x3s2p1c3x8-neonfma-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x2()
1016 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x2()
1017 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x2()