Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dim2col.cuh32 const int64_t dilation_height, in im2col_kernel()
79 const int64_t dilation_height, in im2col()
118 const int64_t dilation_height, in col2im_device()
173 const int64_t dilation_height, in col2im_kernel()
214 const int64_t dilation_height, in col2im()
258 const int64_t dilation_height, in col2im_batched_kernel()
308 const int64_t dilation_height, in col2im_batched()
H A DNaiveConvolutionTranspose2d.cu42 int dilation_height, in slow_conv_transpose2d_shape_check()
170 int64_t dilation_height = dilation[0]; in slow_conv_transpose2d_out_cuda_template() local
355 int64_t dilation_height = dilation[0]; in slow_conv_transpose2d_backward_out_cuda_template() local
531 int64_t dilation_height = dilation[0]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
H A DNaiveConvolutionTranspose3d.cu44 int dilation_height, in slow_conv_transpose3d_shape_check()
212 int64_t dilation_height = dilation[1]; in slow_conv_transpose3d_out_cuda_template() local
436 int64_t dilation_height = dilation[1]; in slow_conv_transpose3d_backward_out_cuda_template() local
644 int64_t dilation_height = dilation[1]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/XNNPACK/test/
H A Dconvolution-test-helpers.cc29 size_t dilation_height, in compute_convolution_qs8_reference_results()
99 size_t dilation_height, in compute_convolution_qs8_reference_results()
153 size_t dilation_height, in compute_convolution_qu8_reference_results()
223 size_t dilation_height, in compute_convolution_qu8_reference_results()
279 size_t dilation_height, in compute_depthwise_convolution_qs8_reference_results()
344 size_t dilation_height, in compute_depthwise_convolution_qs8_reference_results()
396 size_t dilation_height, in compute_depthwise_convolution_qu8_reference_results()
462 size_t dilation_height, 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()
272 inline MaxPoolingOperatorTester& dilation_height(uint32_t dilation_height) { in dilation_height() function
278 inline uint32_t dilation_height() const { in dilation_height() function
H A Ddeconvolution-operator-tester.h227 inline DeconvolutionOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation()
235 inline DeconvolutionOperatorTester& dilation_height(uint32_t dilation_height) { in dilation_height() argument
241 inline uint32_t dilation_height() const { in dilation_height() function
H A Ddeconvolution-nhwc.cc727 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { variable
1040 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { in TEST() local
1363 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { in TEST() local
1697 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { in TEST() local
4802 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { variable
5115 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { in TEST() local
5438 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { in TEST() local
5772 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { in TEST() local
8942 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { variable
9269 for (size_t dilation_height = 2; dilation_height <= 3; dilation_height++) { 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()
266 inline ConvolutionOperatorTester& dilation_height(uint32_t dilation_height) { in dilation_height() argument
272 inline uint32_t dilation_height() const { in dilation_height() function
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dmax-pooling-nhwc.c46 uint32_t dilation_height, in create_max_pooling2d_nhwc()
346 uint32_t dilation_height, in xnn_create_max_pooling2d_nhwc_s8()
386 uint32_t dilation_height, in xnn_create_max_pooling2d_nhwc_u8()
426 uint32_t dilation_height, in xnn_create_max_pooling2d_nhwc_f32()
480 uint32_t dilation_height, in xnn_create_max_pooling2d_nhwc_f16()
H A Dconvolution-nhwc.c206 uint32_t dilation_height, in create_convolution2d_nhwc()
662 uint32_t dilation_height, in xnn_create_convolution2d_nhwc_qu8()
781 uint32_t dilation_height, in xnn_create_convolution2d_nhwc_qs8()
895 uint32_t dilation_height, in xnn_create_convolution2d_nhwc_qc8()
1017 uint32_t dilation_height, in xnn_create_convolution2d_nhwc_f16()
1130 uint32_t dilation_height, in xnn_create_convolution2d_nhwc_f32()
1242 uint32_t dilation_height, in xnn_create_fused_convolution2d_nhwc_f32()
H A Ddeconvolution-nhwc.c40 uint32_t dilation_height, in create_deconvolution2d_nhwc()
297 uint32_t dilation_height, in xnn_create_deconvolution2d_nhwc_qs8()
393 uint32_t dilation_height, in xnn_create_deconvolution2d_nhwc_qu8()
491 uint32_t dilation_height, in xnn_create_deconvolution2d_nhwc_f16()
586 uint32_t dilation_height, in xnn_create_deconvolution2d_nhwc_f32()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose2d.cpp40 int dilation_height, in slow_conv_transpose2d_shape_check()
184 int64_t dilation_height = dilation[0]; in TORCH_META_FUNC() local
256 int64_t dilation_height = dilation[0]; in slow_conv_transpose2d_out_cpu_template() local
426 int64_t dilation_height = dilation[0]; in slow_conv_transpose2d_backward_out_cpu_template() local
625 int64_t dilation_height = dilation[0]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A DNaiveConvolutionTranspose3d.cpp44 int dilation_height, in slow_conv_transpose3d_shape_check()
216 int64_t dilation_height = dilation[1]; in slow_conv_transpose3d_out_cpu_template() local
430 int64_t dilation_height = dilation[1]; in slow_conv_transpose3d_backward_out_cpu_template() local
635 int64_t dilation_height = dilation[1]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
H A Dim2col_shape_check.h15 int64_t dilation_height, in col2im_shape_check()
143 int64_t dilation_height, in im2col_shape_check()
H A DIm2Col.cpp51 int64_t dilation_height = dilation[0]; in im2col_out_cpu_template() local
H A DCol2Im.cpp111 int64_t dilation_height = dilation[0]; in col2im_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c41 const size_t dilation_height = op->dilation_height; in pytorch_qnnp_indirection_init_conv3d() local
245 const size_t dilation_height = op->dilation_height; in pytorch_qnnp_indirection_init_dwconv() local
358 const size_t dilation_height = op->dilation_height; in pytorch_qnnp_indirection_init_deconv2d() local
423 const size_t dilation_height = op->dilation_height; in pytorch_qnnp_indirection_init_maxpool2d() local
H A Dconvolution.c54 uint32_t dilation_height, in pytorch_qnnp_create_convolution_ndhwc_q8()
556 uint32_t dilation_height, in pytorch_qnnp_create_convolution2d_nhwc_q8()
613 uint32_t dilation_height, in pytorch_qnnp_create_convolution3d_ndhwc_q8()
H A Dmax-pooling.c44 uint32_t dilation_height, in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dsubgraph.h189 uint32_t dilation_height; member
206 uint32_t dilation_height; member
221 uint32_t dilation_height; member
238 uint32_t dilation_height; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dconv3d_test.cc38 int32_t dilation_height = 1) { in Conv3dOpModel()
58 int32_t dilation_height = 1) { in Conv3dOpModel()
H A Dconv3d_transpose_test.cc45 int32_t dilation_height = 1) { in Conv3dTransposeOpModel()
76 int32_t dilation_height = 1) { in Conv3dTransposeOpModel()
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c38 const size_t dilation_height = op->dilation_height; in xnn_indirection_init_conv2d() local
98 const size_t dilation_height = op->dilation_height; in xnn_indirection_init_deconv2d() local
220 const size_t dilation_height = op->dilation_height; in xnn_indirection_init_dwconv2d() local
276 const size_t dilation_height = op->dilation_height; in xnn_indirection_init_maxpool2d() local
H A Dim2col.c20 size_t dilation_height, in xnn_im2col_conv2d()

12