Home
last modified time | relevance | path

Searched defs:dilation_width (Results 1 – 25 of 46) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dim2col.cuh33 const int64_t dilation_width, in im2col_kernel()
80 const int64_t dilation_width, in im2col()
119 const int64_t dilation_width, in col2im_device()
174 const int64_t dilation_width, in col2im_kernel()
215 const int64_t dilation_width, in col2im()
259 const int64_t dilation_width, in col2im_batched_kernel()
309 const int64_t dilation_width, in col2im_batched()
H A DNaiveConvolutionTranspose2d.cu43 int dilation_width, in slow_conv_transpose2d_shape_check()
171 int64_t dilation_width = dilation[1]; in slow_conv_transpose2d_out_cuda_template() local
356 int64_t dilation_width = dilation[1]; in slow_conv_transpose2d_backward_out_cuda_template() local
532 int64_t dilation_width = dilation[1]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
H A DNaiveConvolutionTranspose3d.cu43 int dilation_width, in slow_conv_transpose3d_shape_check()
213 int64_t dilation_width = dilation[2]; in slow_conv_transpose3d_out_cuda_template() local
437 int64_t dilation_width = dilation[2]; in slow_conv_transpose3d_backward_out_cuda_template() local
645 int64_t dilation_width = dilation[2]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/XNNPACK/test/
H A Dconvolution-test-helpers.cc30 size_t dilation_width, in compute_convolution_qs8_reference_results()
100 size_t dilation_width, in compute_convolution_qs8_reference_results()
154 size_t dilation_width, in compute_convolution_qu8_reference_results()
224 size_t dilation_width, in compute_convolution_qu8_reference_results()
280 size_t dilation_width, in compute_depthwise_convolution_qs8_reference_results()
345 size_t dilation_width, in compute_depthwise_convolution_qs8_reference_results()
397 size_t dilation_width, in compute_depthwise_convolution_qu8_reference_results()
463 size_t dilation_width, in compute_depthwise_convolution_qu8_reference_results()
H A Dmax-pooling-operator-tester.h264 inline MaxPoolingOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation()
282 inline MaxPoolingOperatorTester& dilation_width(uint32_t dilation_width) { in dilation_width() argument
288 inline uint32_t dilation_width() const { in dilation_width() function
H A Ddeconvolution-operator-tester.h227 inline DeconvolutionOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation()
245 inline DeconvolutionOperatorTester& dilation_width(uint32_t dilation_width) { in dilation_width() function
251 inline uint32_t dilation_width() const { in dilation_width() function
H A Ddeconvolution-nhwc.cc742 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { variable
1056 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
1379 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
1714 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
4817 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { variable
5131 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
5454 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
5789 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
8957 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { variable
9285 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
[all …]
H A Dsubgraph-tester.h241 uint32_t dilation_height, uint32_t dilation_width, uint32_t groups, in AddDeconvolution2D()
313 …uint32_t stride_width, uint32_t dilation_height, uint32_t dilation_width, uint32_t input_id, uint3… in AddMaxPooling2D()
H A Dconvolution-operator-tester.h258 inline ConvolutionOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation()
276 inline ConvolutionOperatorTester& dilation_width(uint32_t dilation_width) { in dilation_width() argument
282 inline uint32_t dilation_width() const { in dilation_width() function
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dmax-pooling-nhwc.c47 uint32_t dilation_width, in create_max_pooling2d_nhwc()
347 uint32_t dilation_width, in xnn_create_max_pooling2d_nhwc_s8()
387 uint32_t dilation_width, in xnn_create_max_pooling2d_nhwc_u8()
427 uint32_t dilation_width, in xnn_create_max_pooling2d_nhwc_f32()
481 uint32_t dilation_width, in xnn_create_max_pooling2d_nhwc_f16()
H A Dconvolution-nhwc.c207 uint32_t dilation_width, in create_convolution2d_nhwc()
663 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_qu8()
782 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_qs8()
896 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_qc8()
1018 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_f16()
1131 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_f32()
1243 uint32_t dilation_width, in xnn_create_fused_convolution2d_nhwc_f32()
H A Ddeconvolution-nhwc.c41 uint32_t dilation_width, in create_deconvolution2d_nhwc()
298 uint32_t dilation_width, in xnn_create_deconvolution2d_nhwc_qs8()
394 uint32_t dilation_width, in xnn_create_deconvolution2d_nhwc_qu8()
492 uint32_t dilation_width, in xnn_create_deconvolution2d_nhwc_f16()
587 uint32_t dilation_width, in xnn_create_deconvolution2d_nhwc_f32()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose2d.cpp41 int dilation_width, in slow_conv_transpose2d_shape_check()
185 int64_t dilation_width = dilation[1]; in TORCH_META_FUNC() local
257 int64_t dilation_width = dilation[1]; in slow_conv_transpose2d_out_cpu_template() local
427 int64_t dilation_width = dilation[1]; in slow_conv_transpose2d_backward_out_cpu_template() local
626 int64_t dilation_width = dilation[1]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A DNaiveConvolutionTranspose3d.cpp43 int dilation_width, in slow_conv_transpose3d_shape_check()
217 int64_t dilation_width = dilation[2]; in slow_conv_transpose3d_out_cpu_template() local
431 int64_t dilation_width = dilation[2]; in slow_conv_transpose3d_backward_out_cpu_template() local
636 int64_t dilation_width = dilation[2]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
H A Dim2col_shape_check.h16 int64_t dilation_width, in col2im_shape_check()
144 int64_t dilation_width, in im2col_shape_check()
H A DIm2Col.cpp52 int64_t dilation_width = dilation[1]; in im2col_out_cpu_template() local
H A DCol2Im.cpp112 int64_t dilation_width = dilation[1]; in col2im_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c42 const size_t dilation_width = op->dilation_width; in pytorch_qnnp_indirection_init_conv3d() local
246 const size_t dilation_width = op->dilation_width; in pytorch_qnnp_indirection_init_dwconv() local
359 const size_t dilation_width = op->dilation_width; in pytorch_qnnp_indirection_init_deconv2d() local
424 const size_t dilation_width = op->dilation_width; in pytorch_qnnp_indirection_init_maxpool2d() local
H A Dconvolution.c55 uint32_t dilation_width, in pytorch_qnnp_create_convolution_ndhwc_q8()
557 uint32_t dilation_width, in pytorch_qnnp_create_convolution2d_nhwc_q8()
614 uint32_t dilation_width, in pytorch_qnnp_create_convolution3d_ndhwc_q8()
H A Dmax-pooling.c45 uint32_t dilation_width, in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dsubgraph.h190 uint32_t dilation_width; member
207 uint32_t dilation_width; member
222 uint32_t dilation_width; member
239 uint32_t dilation_width; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dconv3d_test.cc37 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dOpModel()
57 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dOpModel()
H A Dconv3d_transpose_test.cc44 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dTransposeOpModel()
75 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dTransposeOpModel()
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c39 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_conv2d() local
99 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_deconv2d() local
221 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_dwconv2d() local
277 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_maxpool2d() local
H A Dim2col.c21 size_t dilation_width, in xnn_im2col_conv2d()

12