Home
last modified time | relevance | path

Searched defs:output_y (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c54 const size_t output_y = output_y_x.quotient; in xnn_indirection_init_conv2d() local
116 const size_t output_y = output_y_x.quotient; in xnn_indirection_init_deconv2d() local
170 for (size_t output_y = output_y_start; output_y < output_height; output_y += stride_height) { in xnn_indirection_init_subconv2d() local
225 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_dwconv2d() local
287 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_maxpool2d() local
322 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_maxpool2d() local
369 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
402 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
464 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_resize_bilinear2d_hwc_f32() local
497 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_indirection_init_resize_bilinear2d_hwc_f32() local
[all …]
H A Doperator-run.c643 size_t output_y) in xnn_compute_dwconv_unipass()
678 size_t output_y) in xnn_compute_argmax_pooling_unipass()
697 size_t output_y) in xnn_compute_argmax_pooling_multipass()
719 size_t output_y) in xnn_compute_max_pooling()
757 size_t output_y) in xnn_compute_average_pooling_unipass()
775 size_t output_y) in xnn_compute_average_pooling_multipass()
796 size_t output_y) in xnn_compute_pixelwise_average_pooling_unipass()
816 size_t output_y) in xnn_compute_pixelwise_average_pooling_multipass()
H A Dim2col.c30 for (size_t output_y = 0; output_y < output_height; output_y++) { in xnn_im2col_conv2d() local
/aosp_15_r20/external/XNNPACK/test/
H A Dresize-bilinear-operator-tester.h245 for (size_t output_y = 0; output_y < output_height(); output_y++) { in TestNHWCxF16() local
330 for (size_t output_y = 0; output_y < output_height(); output_y++) { in TestNHWCxF32() local
412 for (size_t output_y = 0; output_y < output_height(); output_y++) { in TestNHWCxS8() local
495 for (size_t output_y = 0; output_y < output_height(); output_y++) { in TestNHWCxU8() local
581 for (size_t output_y = 0; output_y < output_height(); output_y++) { in TestNCHWxF32() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c68 const size_t output_y = y_x.quotient; in pytorch_qnnp_indirection_init_conv3d() local
270 for (size_t output_y = 0; output_y < output_height; output_y++) { in pytorch_qnnp_indirection_init_dwconv() local
317 for (size_t output_y = 0; output_y < output_height; output_y++) { in pytorch_qnnp_indirection_init_dwconv() local
376 const size_t output_y = output_index / output_width; in pytorch_qnnp_indirection_init_deconv2d() local
431 for (size_t output_y = 0; output_y < output_height; output_y++) { in pytorch_qnnp_indirection_init_maxpool2d() local
H A Doperator-run.c445 size_t output_y) { in compute_dwconv2d_unipass()
465 size_t output_y) { in compute_dwconv2d_multiipass()
546 size_t output_y) { in compute_max_pooling()
588 size_t output_y) { in compute_average_pooling_unipass()
610 size_t output_y) { in compute_average_pooling_multipass()
H A Dconv-run.cc245 size_t output_y) { in compute_dwconv2d_unipass()
264 size_t output_y) { in compute_dwconv2d_multiipass()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv3d.cc30 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenConv3DF32()
55 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_conv3d.cc27 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
50 int64_t output_x, int64_t output_y, int64_t output_z, int64_t x_stride, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
H A Druntime_conv_impl.h35 Eigen::Index kernel_filters, Eigen::Index output_x, Eigen::Index output_y, in EigenConv2DImpl()
113 Eigen::Index output_x, Eigen::Index output_y, Eigen::Index output_z, in EigenConv3DImpl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DUpSampleBicubic2d.cu46 const int output_y = index / output_width; in upsample_bicubic2d_out_frame() local
120 const int output_y = index / output_width; in upsample_bicubic2d_backward_out_frame() local
H A DReflectionPad.cu71 auto output_y = output_xy / output_dim_x; in get_index_mapping2d() local
180 int64_t output_y = (output_id / output.size(4)) % output.size(3); in parallel_reflection_pad3d() local
H A DUpSampleBilinear2d.cu489 const int output_y = threadIdx.y + blockIdx.y * blockDim.y; in upsample_gen2d_aa_out_frame() local
583 const int output_y = threadIdx.y + blockIdx.y * blockDim.y; in upsample_gen2d_aa_backward_out_frame() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8dwconv/
H A Dmp8x27-sse2.c54 for (size_t output_y = 0; output_y < output_height; output_y++) { in pytorch_q8dwconv_ukernel_mp8x27__sse2() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dconvolver.hpp117 unsigned int output_y = m_start_output_y; in next_block() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp140 for (const auto output_y : c10::irange(output_height)) { in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/XNNPACK/src/f32-conv-hwc/gen/
H A D3x3s2p1c3x4-neonfma-2x1.c62 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neonfma_2x1() local
H A D3x3s2p0p1c3x4-neonfma-2x1.c62 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neonfma_2x1() local
H A D3x3s2p0p1c3x4-neon-2x1.c60 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x4__neon_2x1() local
H A D3x3s2p1c3x4-neon-2x1.c60 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc_ukernel_3x3s2p1c3x4__neon_2x1() local
/aosp_15_r20/external/XNNPACK/src/f32-conv-hwc2chw/
H A D3x3s2p1c3x4-sse-1x1.c52 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 1) { in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_1x1() local
H A D3x3s2p1c3x4-neon-2x2.c55 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__neon_2x2() local
H A D3x3s2p1c3x4-wasmsimd-2x2.c55 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__wasmsimd_2x2() local
H A D3x3s2p1c3x4-sse-2x2.c55 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f32_conv_hwc2chw_ukernel_3x3s2p1c3x4__sse_2x2() local
/aosp_15_r20/external/XNNPACK/src/f16-conv-hwc2chw/
H A D3x3s2p1c3x4-neonfp16arith-2x2.c55 for (size_t output_y = output_y_start; output_y < output_y_end; output_y += 2) { in xnn_f16_conv_hwc2chw_ukernel_3x3s2p1c3x4__neonfp16arith_2x2() local

12