Home
last modified time | relevance | path

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

12

/external/pytorch/aten/src/ATen/native/cuda/
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()
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
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
DIm2Col.cu55 int64_t dilation_height = dilation[0]; in im2col_out_cuda_template() local
DCol2Im.cu67 int64_t dilation_height = dilation[0]; in col2im_out_cuda_template() local
/external/XNNPACK/test/
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()
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() argument
278 inline uint32_t dilation_height() const { in dilation_height() function
/external/pytorch/aten/src/ATen/native/
Dim2col_shape_check.h15 int64_t dilation_height, in col2im_shape_check()
143 int64_t dilation_height, in im2col_shape_check()
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
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
DIm2Col.cpp51 int64_t dilation_height = dilation[0]; in im2col_out_cpu_template() local
DCol2Im.cpp111 int64_t dilation_height = dilation[0]; in col2im_out_cpu_template() local
/external/XNNPACK/src/operators/
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()
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()
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()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
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
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()
Dmax-pooling.c44 uint32_t dilation_height, in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
Ddeconvolution.c47 uint32_t dilation_height, in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
/external/XNNPACK/src/
Dim2col.c20 size_t dilation_height, in xnn_im2col_conv2d()
/external/tensorflow/tensorflow/lite/kernels/
Dconv3d_test.cc38 int32_t dilation_height = 1) { in Conv3dOpModel()
58 int32_t dilation_height = 1) { in Conv3dOpModel()
Dconv3d_transpose_test.cc45 int32_t dilation_height = 1) { in Conv3dTransposeOpModel()
76 int32_t dilation_height = 1) { in Conv3dTransposeOpModel()
/external/XNNPACK/src/subgraph/
Dmax-pooling-2d.c225 uint32_t dilation_height, in xnn_define_max_pooling_2d()
/external/XNNPACK/src/xnnpack/
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
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc55 int dilation_height = 0; in Build() local

12