Home
last modified time | relevance | path

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

12

/external/pytorch/aten/src/ATen/native/cuda/
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()
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
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
DIm2Col.cu56 int64_t dilation_width = dilation[1]; in im2col_out_cuda_template() local
DCol2Im.cu68 int64_t dilation_width = dilation[1]; in col2im_out_cuda_template() local
/external/XNNPACK/test/
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()
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
/external/pytorch/aten/src/ATen/native/
Dim2col_shape_check.h16 int64_t dilation_width, in col2im_shape_check()
144 int64_t dilation_width, in im2col_shape_check()
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
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
DIm2Col.cpp52 int64_t dilation_width = dilation[1]; in im2col_out_cpu_template() local
DCol2Im.cpp112 int64_t dilation_width = dilation[1]; in col2im_out_cpu_template() local
/external/XNNPACK/src/operators/
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()
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()
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()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
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
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()
Dmax-pooling.c45 uint32_t dilation_width, in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
Ddeconvolution.c48 uint32_t dilation_width, in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
/external/XNNPACK/src/
Dim2col.c21 size_t dilation_width, in xnn_im2col_conv2d()
/external/tensorflow/tensorflow/lite/kernels/
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()
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()
/external/XNNPACK/src/subgraph/
Dmax-pooling-2d.c226 uint32_t dilation_width, in xnn_define_max_pooling_2d()
/external/XNNPACK/src/xnnpack/
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
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc56 int dilation_width = 0; in Build() local

12