Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h2620 const int output_height_stride = function_params->output_height_stride;
2651 output_block_data + output_height_stride * k_height + 8 * j_depth;
2784 const int output_height_stride = function_params->output_height_stride;
2811 output_block_data + output_height_stride * k_height + 8 * j_depth;
2871 const int output_height_stride = function_params->output_height_stride;
2929 output_block_data + output_height_stride * k_height + 8 * j_depth;
3083 const int output_height_stride = function_params->output_height_stride;
3132 output_block_data + output_height_stride * k_height + 8 * j_depth;
3272 const int output_height_stride = function_params->output_height_stride;
3424 vst1q_lane_8x4(output_data + output_height_stride, acc_u8_all, 1);
[all …]
/external/XNNPACK/src/xnnpack/
Dconv.h34 size_t output_height_stride, \
71 size_t output_height_stride, \
Dcompute.h364 size_t output_height_stride; member
391 size_t output_height_stride; member
450 size_t output_height_stride; member
496 size_t output_height_stride; member
546 size_t output_height_stride; member
580 size_t output_height_stride; member
618 size_t output_height_stride; member
/external/XNNPACK/src/operators/
Dargmax-pooling-nhwc.c254 const size_t output_height_stride = output_width * output_width_stride; in xnn_setup_argmax_pooling2d_nhwc_f32() local
265 .output_batch_stride = output_height * output_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
266 .output_height_stride = output_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
Daverage-pooling-nhwc.c543 const size_t output_height_stride = output_width * output_width_stride; in setup_average_pooling2d() local
585 .output_batch_stride = output_height * output_height_stride, in setup_average_pooling2d()
586 .output_height_stride = output_height_stride, in setup_average_pooling2d()
612 .output_batch_stride = output_height * output_height_stride, in setup_average_pooling2d()
613 .output_height_stride = output_height_stride, in setup_average_pooling2d()
Dmax-pooling-nhwc.c295 const size_t output_height_stride = output_width * output_width_stride; in setup_max_pooling2d_nhwc() local
304 .output_batch_stride = output_height * output_height_stride, in setup_max_pooling2d_nhwc()
305 .output_height_stride = output_height_stride, in setup_max_pooling2d_nhwc()
Ddepth-to-space-nhwc.c141 .output_height_stride = output_width * depth_to_space_op->output_pixel_stride * sizeof(float), in xnn_setup_depth_to_space_nhwc_x32()
/external/XNNPACK/src/
Doperator-run.c363 context->output_height_stride, in xnn_compute_conv2d_hwc2chw()
377 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_dwconv_unipass()
435 (batch_input_y * block_size + block_y) * context->output_height_stride + in xnn_compute_depthtospace2d_hwc_strided()
468 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_argmax_pooling_unipass()
487 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_argmax_pooling_multipass()
509 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_max_pooling()
547 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_average_pooling_unipass()
565 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_average_pooling_multipass()
588 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_pixelwise_average_pooling_unipass()
608 batch_index * context->output_batch_stride + output_y * context->output_height_stride); in xnn_compute_pixelwise_average_pooling_multipass()
/external/XNNPACK/src/f32-conv-hwc/gen/
D3x3s2p0p1c3x4-neon-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neon_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 4)… in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neon_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neon_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neon_2x1()
D3x3s2p1c3x4-neonfma-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 4)… in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x1()
D3x3s2p1c3x4-neon-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neon_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 4)… in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neon_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neon_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neon_2x1()
D3x3s2p0p1c3x4-neonfma-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 4)… in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x1()
D3x3s2p1c3x8-neon-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neon_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 8)… in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neon_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neon_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neon_2x1()
D3x3s2p0p1c3x8-neonfma-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neonfma_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 8)… in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neonfma_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neonfma_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neonfma_2x1()
D3x3s2p1c3x8-neonfma-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 8)… in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x8__neonfma_2x1()
D3x3s2p0p1c3x8-neon-2x1.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neon_2x1() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 8)… in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neon_2x1()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neon_2x1()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neon_2x1()
D3x3s2p0p1c3x4-neonfma-2x2.c30 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x2() argument
43 …const size_t output_height_increment = output_height_stride * 2 - round_up_po2(output_channels, 4)… in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x2()
52 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x2()
53 float* o1 = (float*) ((uintptr_t) o0 + output_height_stride); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x2()
/external/XNNPACK/src/f32-conv-hwc2chw/
D3x3s2p1c3x4-sse-1x1.c25 size_t output_height_stride, in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_1x1() argument
43 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_1x1()
303 output0 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_1x1()
D3x3s2p1c3x4-neonfma-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neonfma_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neonfma_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neonfma_2x2()
645 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neonfma_2x2()
646 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neonfma_2x2()
D3x3s2p1c3x4-neon-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neon_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neon_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neon_2x2()
641 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neon_2x2()
642 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neon_2x2()
D3x3s2p1c3x4-scalar-1x1.c23 size_t output_height_stride, in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__scalar_1x1() argument
41 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__scalar_1x1()
663 output0 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__scalar_1x1()
D3x3s2p1c3x4-wasmsimd-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__wasmsimd_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__wasmsimd_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__wasmsimd_2x2()
644 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__wasmsimd_2x2()
645 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__wasmsimd_2x2()
D3x3s2p1c3x4-sse-2x2.c25 size_t output_height_stride, in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_2x2() argument
45 float* output0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_2x2()
46 float* output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_2x2()
645 output0 = (float*) ((uintptr_t) output1 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_2x2()
646 output1 = (float*) ((uintptr_t) output0 + output_height_stride); in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_2x2()
/external/XNNPACK/src/f32-conv-hwc/
D3x3s2p1c3x4-scalar-1x1.c23 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__scalar_1x1() argument
36 …const size_t output_height_increment = output_height_stride - round_up_po2(output_channels, 4) * s… in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__scalar_1x1()
42 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__scalar_1x1()
D3x3s2p0p1c3x4-scalar-1x1.c23 size_t output_height_stride, in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__scalar_1x1() argument
36 …const size_t output_height_increment = output_height_stride - round_up_po2(output_channels, 4) * s… in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__scalar_1x1()
42 float* o0 = (float*) ((uintptr_t) output + output_height_stride * output_y_start); in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__scalar_1x1()

12