Home
last modified time | relevance | path

Searched defs:output_x (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c53 const size_t output_x = output_y_x.remainder; in xnn_indirection_init_conv2d() local
115 const size_t output_x = output_y_x.remainder; in xnn_indirection_init_deconv2d() local
180 const size_t output_x = output_x_start + sliced_output_x * stride_width; in xnn_indirection_init_subconv2d() local
229 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_dwconv2d() local
242 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_dwconv2d() local
300 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_maxpool2d() local
325 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_maxpool2d() local
377 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
409 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_resize_bilinear2d_hwc_f16() local
472 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_indirection_init_resize_bilinear2d_hwc_f32() local
[all …]
H A Dim2col.c31 for (size_t output_x = 0; output_x < output_width; output_x++) { in xnn_im2col_conv2d() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c69 const size_t output_x = y_x.remainder; in pytorch_qnnp_indirection_init_conv3d() local
275 for (size_t output_x = 0; output_x < output_width; output_x++) { in pytorch_qnnp_indirection_init_dwconv() local
300 for (size_t output_x = 0; output_x < output_width; output_x++) { in pytorch_qnnp_indirection_init_dwconv() local
319 for (size_t output_x = 0; output_x < output_width; output_x++) { in pytorch_qnnp_indirection_init_dwconv() local
377 const size_t output_x = output_index % output_width; in pytorch_qnnp_indirection_init_deconv2d() local
437 for (size_t output_x = 0; output_x < output_width; output_x++) { in pytorch_qnnp_indirection_init_maxpool2d() local
/aosp_15_r20/external/XNNPACK/test/
H A Dresize-bilinear-operator-tester.h250 for (size_t output_x = 0; output_x < output_width(); output_x++) { in TestNHWCxF16() local
335 for (size_t output_x = 0; output_x < output_width(); output_x++) { in TestNHWCxF32() local
417 for (size_t output_x = 0; output_x < output_width(); output_x++) { in TestNHWCxS8() local
500 for (size_t output_x = 0; output_x < output_width(); output_x++) { in TestNHWCxU8() local
586 for (size_t output_x = 0; output_x < output_width(); output_x++) { in TestNCHWxF32() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu36 int64_t output_x, in get_index_mapping1d()
70 auto output_x = output_xy % output_dim_x; in get_index_mapping2d() local
100 auto output_x = threadIdx.x + blockIdx.x * blockDim.x; in reflection_pad1d_out_kernel() local
114 auto output_x = threadIdx.x + blockIdx.x * blockDim.x; in reflection_pad1d_backward_out_kernel() local
179 int64_t output_x = output_id % output.size(4); in parallel_reflection_pad3d() local
H A DUpSampleBicubic2d.cu45 const int output_x = index % output_width; in upsample_bicubic2d_out_frame() local
119 const int output_x = index % output_width; in upsample_bicubic2d_backward_out_frame() local
H A DUpSampleBilinear2d.cu488 const int output_x = threadIdx.x + blockIdx.x * blockDim.x; in upsample_gen2d_aa_out_frame() local
582 const int output_x = threadIdx.x + blockIdx.x * blockDim.x; in upsample_gen2d_aa_backward_out_frame() local
/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/quantized/cpu/qnnpack/src/q8dwconv/
H A Dmp8x27-sse2.c56 for (size_t output_x = 0; output_x < output_width; output_x++) { in pytorch_q8dwconv_ukernel_mp8x27__sse2() local
H A Dmp8x27-neon.c57 for (size_t output_x = 0; output_x < output_width; output_x++) { in pytorch_q8dwconv_ukernel_mp8x27__neon() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dconvolver.hpp118 unsigned int output_x = m_start_output_x; in next_block() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSampleBicubic2d.cpp141 for (const auto output_x : c10::irange(output_width)) { in upsample_bicubic2d_backward_out_frame() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/
H A DCpuGemmAssemblyDispatch.cpp264 for(int64_t output_x = 0; output_x < _cp.output_width; output_x++) in prepare_indirect_buffer() local