Home
last modified time | relevance | path

Searched defs: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()
191 int64_t output_padding_width = output_padding[1]; in TORCH_META_FUNC() local
263 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_out_cpu_template() local
433 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_backward_out_cpu_template() local
632 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A DNaiveConvolutionTranspose3d.cpp46 int 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
440 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_backward_out_cpu_template() local
645 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose2d.cu41 int 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
362 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_backward_out_cuda_template() local
538 int64_t output_padding_width = output_padding[1]; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
H A DNaiveConvolutionTranspose3d.cu46 int 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
446 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_backward_out_cuda_template() local
654 int64_t output_padding_width = output_padding[2]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local