Searched refs:output_padding_width (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | NaiveConvolutionTranspose2d.cpp | 39 int output_padding_width, in slow_conv_transpose2d_shape_check() argument 62 (output_padding_width < stride_width || in slow_conv_transpose2d_shape_check() 63 output_padding_width < dilation_width) && in slow_conv_transpose2d_shape_check() 69 output_padding_width, in slow_conv_transpose2d_shape_check() 112 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose2d_shape_check() 191 int64_t output_padding_width = output_padding[1]; in TORCH_META_FUNC() local 205 output_padding_width, in TORCH_META_FUNC() 226 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in TORCH_META_FUNC() 263 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_out_cpu_template() local 287 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose2d_out_cpu_template() [all …]
|
H A D | NaiveConvolutionTranspose3d.cpp | 46 int output_padding_width, in slow_conv_transpose3d_shape_check() argument 72 (output_padding_width < stride_width || in slow_conv_transpose3d_shape_check() 73 output_padding_width < dilation_width) && in slow_conv_transpose3d_shape_check() 82 output_padding_width, in slow_conv_transpose3d_shape_check() 142 output_padding_width; in slow_conv_transpose3d_shape_check() 226 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_out_cpu_template() local 246 output_padding_width, in slow_conv_transpose3d_out_cpu_template() 277 output_padding_width; in slow_conv_transpose3d_out_cpu_template() 440 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_backward_out_cpu_template() local 462 output_padding_width, in slow_conv_transpose3d_backward_out_cpu_template() [all …]
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | NaiveConvolutionTranspose2d.cu | 41 int output_padding_width, in slow_conv_transpose2d_shape_check() argument 64 (output_padding_width < stride_width || in slow_conv_transpose2d_shape_check() 65 output_padding_width < dilation_width) && in slow_conv_transpose2d_shape_check() 72 output_padding_width, in slow_conv_transpose2d_shape_check() 115 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose2d_shape_check() 177 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_out_cuda_template() local 200 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose2d_out_cuda_template() 362 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_backward_out_cuda_template() local 376 output_padding_width, in slow_conv_transpose2d_backward_out_cuda_template() 399 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose2d_backward_out_cuda_template() [all …]
|
H A D | NaiveConvolutionTranspose3d.cu | 46 int output_padding_width, in slow_conv_transpose3d_shape_check() argument 72 (output_padding_width < stride_width || in slow_conv_transpose3d_shape_check() 73 output_padding_width < dilation_width) && in slow_conv_transpose3d_shape_check() 82 output_padding_width, in slow_conv_transpose3d_shape_check() 140 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose3d_shape_check() 222 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_out_cuda_template() local 252 output_padding_width, in slow_conv_transpose3d_out_cuda_template() 278 (dilation_width * (kernel_width - 1) + 1) + output_padding_width; in slow_conv_transpose3d_out_cuda_template() 446 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_backward_out_cuda_template() local 478 output_padding_width, in slow_conv_transpose3d_backward_out_cuda_template() [all …]
|