Home
last modified time | relevance | path

Searched defs:kernel_width (Results 1 – 25 of 92) sorted by relevance

1234

/aosp_15_r20/external/XNNPACK/test/
H A Dconvolution-test-helpers.cc26 size_t kernel_width, in compute_convolution_qs8_reference_results()
96 size_t kernel_width, in compute_convolution_qs8_reference_results()
150 size_t kernel_width, in compute_convolution_qu8_reference_results()
220 size_t kernel_width, in compute_convolution_qu8_reference_results()
276 size_t kernel_width, in compute_depthwise_convolution_qs8_reference_results()
341 size_t kernel_width, in compute_depthwise_convolution_qs8_reference_results()
393 size_t kernel_width, in compute_depthwise_convolution_qu8_reference_results()
459 size_t kernel_width, in compute_depthwise_convolution_qu8_reference_results()
H A Ddeconvolution-nhwc.cc589 for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) { variable
893 for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) { in TEST() local
1216 for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) { in TEST() local
1541 for (size_t kernel_width = 1; kernel_width <= 4; kernel_width *= 2) { in TEST() local
1927 for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) { variable
2220 for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) { in TEST() local
2530 for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) { in TEST() local
2842 for (size_t kernel_width = 2; kernel_width <= 5; kernel_width++) { in TEST() local
3212 for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) { variable
3416 for (size_t kernel_width = 3; kernel_width <= 5; kernel_width++) { in TEST() local
[all …]
H A Ddwconv2d-microkernel-tester.h114 inline DWConv2DMicrokernelTester& kernel_width(uint32_t kernel_width) { in kernel_width() argument
120 inline uint32_t kernel_width() const { in kernel_width() function
H A Dconv-hwc-microkernel-tester.h177 inline ConvHWCMicrokernelTester& kernel_width(uint32_t kernel_width) { in kernel_width() argument
183 inline uint32_t kernel_width() const { in kernel_width() function
H A Dconv-hwc2chw-microkernel-tester.h179 inline ConvHWC2CHWMicrokernelTester& kernel_width(uint32_t kernel_width) { in kernel_width() function
185 inline uint32_t kernel_width() const { in kernel_width() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp35 const int64_t kernel_width = kernel_size[1]; in compute_columns2d() local
102 int64_t kernel_width, in slow_conv2d_shape_check()
229 int64_t kernel_width, in slow_conv2d_update_output_frame()
292 int64_t kernel_width, in slow_conv2d_backward_update_grad_input_frame()
367 const int64_t kernel_width = kernel_size[1]; in slow_conv2d_backward_out_cpu_template() local
490 const int64_t kernel_width = kernel_size[1]; in slow_conv2d_backward_weight_out_cpu_template() local
555 const int64_t kernel_width = kernel_size[1]; in slow_conv2d_forward_out_cpu() local
H A DConvolutionMM3d.cpp39 const int64_t kernel_width = kernel_size[2]; in compute_columns3d() local
117 int64_t kernel_width, in slow_conv3d_shape_check()
271 int64_t kernel_width, in slow_conv3d_update_output_frame()
319 int64_t kernel_width, in slow_conv3d_backward_update_grad_input_frame()
381 const int64_t kernel_width = kernel_size[2]; in slow_conv3d_backward_out_cpu_template() local
505 const int64_t kernel_width = kernel_size[2]; in slow_conv3d_backward_parameters_out_cpu_template() local
566 const int64_t kernel_width = kernel_size[2]; in slow_conv3d_forward_out_cpu() local
H A DNaiveConvolutionTranspose2d.cpp33 int kernel_width, in slow_conv_transpose2d_shape_check()
183 int64_t kernel_width = kernel_size[1]; in TORCH_META_FUNC() local
255 int64_t kernel_width = kernel_size[1]; in slow_conv_transpose2d_out_cpu_template() local
425 int64_t kernel_width = kernel_size[1]; in slow_conv_transpose2d_backward_out_cpu_template() local
624 int64_t kernel_width = kernel_size[1]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A DNaiveConvolutionTranspose3d.cpp34 int kernel_width, in slow_conv_transpose3d_shape_check()
214 int64_t kernel_width = kernel_size[2]; in slow_conv_transpose3d_out_cpu_template() local
428 int64_t kernel_width = kernel_size[2]; in slow_conv_transpose3d_backward_out_cpu_template() local
633 int64_t kernel_width = kernel_size[2]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
H A Dim2col_shape_check.h14 int64_t kernel_width, in col2im_shape_check()
142 int64_t kernel_width, in im2col_shape_check()
H A Dvol2col.h19 const int64_t kernel_width, in vol2col()
70 const int64_t kernel_width, in col2vol()
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dconvolution-nhwc.c203 uint32_t kernel_width, in create_convolution2d_nhwc()
659 uint32_t kernel_width, in xnn_create_convolution2d_nhwc_qu8()
778 uint32_t kernel_width, in xnn_create_convolution2d_nhwc_qs8()
892 uint32_t kernel_width, in xnn_create_convolution2d_nhwc_qc8()
1014 uint32_t kernel_width, in xnn_create_convolution2d_nhwc_f16()
1127 uint32_t kernel_width, in xnn_create_convolution2d_nhwc_f32()
1239 uint32_t kernel_width, in xnn_create_fused_convolution2d_nhwc_f32()
1533 const size_t kernel_width = convolution_op->kernel_width; in setup_convolution2d_nhwc() local
1708 const size_t kernel_width = convolution_op->kernel_width; in setup_convolution2d_nhwc() local
H A Ddeconvolution-nhwc.c37 uint32_t kernel_width, in create_deconvolution2d_nhwc()
294 uint32_t kernel_width, in xnn_create_deconvolution2d_nhwc_qs8()
390 uint32_t kernel_width, in xnn_create_deconvolution2d_nhwc_qu8()
488 uint32_t kernel_width, in xnn_create_deconvolution2d_nhwc_f16()
583 uint32_t kernel_width, in xnn_create_deconvolution2d_nhwc_f32()
682 const size_t kernel_width = deconvolution_op->kernel_width; in setup_conv_path() local
861 const size_t kernel_width = deconvolution_op->kernel_width; in setup_subconv2d_path() local
/aosp_15_r20/external/ComputeLibrary/src/core/
H A DUtils.cpp377 const unsigned int kernel_width = weights_shape[width_idx]; in calculate_same_pad() local
410 … unsigned int kernel_width, unsigned int kernel_height, in deconvolution_output_dimensions()
430 int kernel_width, int kernel_height, in scaled_dimensions()
464 int kernel_width, int kernel_height, in scaled_dimensions_signed()
493 … int kernel_width, int kernel_height, int kernel_depth, in scaled_3d_dimensions_signed()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dconvolution.c49 uint32_t kernel_width, in pytorch_qnnp_create_convolution_ndhwc_q8()
553 uint32_t kernel_width, in pytorch_qnnp_create_convolution2d_nhwc_q8()
608 uint32_t kernel_width, in pytorch_qnnp_create_convolution3d_ndhwc_q8()
761 const size_t kernel_width = convolution->kernel_width; in pytorch_qnnp_setup_convolution_ndhwc_q8() local
H A Dindirection.c36 const size_t kernel_width = op->kernel_width; in pytorch_qnnp_indirection_init_conv3d() local
240 const size_t kernel_width = op->kernel_width; in pytorch_qnnp_indirection_init_dwconv() local
355 const size_t kernel_width = op->kernel_width; in pytorch_qnnp_indirection_init_deconv2d() local
463 const size_t kernel_width = op->kernel_width; in pytorch_qnnp_indirection_set_step_dimensions() local
H A Ddeconvolution.c44 uint32_t kernel_width, in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
258 const size_t kernel_width = deconvolution->kernel_width; in pytorch_qnnp_setup_deconvolution2d_nhwc_q8() local
/aosp_15_r20/external/XNNPACK/bench/
H A Dconvolution.cc39 const size_t kernel_width = state.range(4); in xnnpack_convolution_qu8() local
159 const size_t kernel_width = state.range(4); in xnnpack_convolution_qs8() local
282 const size_t kernel_width = state.range(4); in xnnpack_convolution_f16() local
398 const size_t kernel_width = state.range(4); in xnnpack_convolution_f32() local
513 const size_t kernel_width = state.range(4); in tflite_convolution_f32() local
H A Ddeconvolution.cc34 const size_t kernel_width = state.range(4); in xnnpack_deconvolution_qu8() local
155 const size_t kernel_width = state.range(4); in xnnpack_deconvolution_f32() local
273 const size_t kernel_width = state.range(4); in tflite_deconvolution_f32() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose2d.cu35 int kernel_width, in slow_conv_transpose2d_shape_check()
169 int64_t kernel_width = kernel_size[1]; in slow_conv_transpose2d_out_cuda_template() local
354 int64_t kernel_width = kernel_size[1]; in slow_conv_transpose2d_backward_out_cuda_template() local
530 int64_t kernel_width = kernel_size[1]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
H A DNaiveConvolutionTranspose3d.cu34 int kernel_width, in slow_conv_transpose3d_shape_check()
210 int64_t kernel_width = kernel_size[2]; in slow_conv_transpose3d_out_cuda_template() local
434 int64_t kernel_width = kernel_size[2]; in slow_conv_transpose3d_backward_out_cuda_template() local
642 int64_t kernel_width = kernel_size[2]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuGemmConv2d.cpp61 const unsigned int kernel_width = weights->dimension(idx_width); in skip_im_col_info() local
279 const unsigned int kernel_width = weights->dimension(idx_width); in configure() local
398 const unsigned int kernel_width = weights->dimension(idx_width); in has_opt_impl() local
445 const unsigned int kernel_width = weights->dimension(idx_width); in validate() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DIm2Col.cpp44 const int kernel_width = kernel_dims.width; in im2col_nchw() local
98 const int kernel_width = kernel_dims.width; in im2col_nhwc() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DDirectConvolution3DFixture.h161 …x, int stride_y, int stride_z, int pad_x, int pad_y, int pad_z, unsigned int kernel_width, int ker… in setup()
174 …x, int stride_y, int stride_z, int pad_x, int pad_y, int pad_z, unsigned int kernel_width, int ker… in setup()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuIm2ColKernel.cpp88 int kernel_width, in linearize_volume_nchw()
186 int kernel_width, in linearize_volume_nhwc()

1234