Home
last modified time | relevance | path

Searched refs:output_padding_width (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose2d.cpp39 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 DNaiveConvolutionTranspose3d.cpp46 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 DNaiveConvolutionTranspose2d.cu41 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 DNaiveConvolutionTranspose3d.cu46 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 …]